RooDataHistSliceIter.cxx

Go to the documentation of this file.

/*****************************************************************************
 * Project: RooFit                                                           *
 * Package: RooFitCore                                                       *
 * @(#)root/roofitcore:$Id$
 * Authors:                                                                  *
 *   WV, Wouter Verkerke, UC Santa Barbara, verkerke@slac.stanford.edu       *
 *   DK, David Kirkby,    UC Irvine,         dkirkby@uci.edu                 *
 *                                                                           *
 * Copyright (c) 2000-2005, Regents of the University of California          *
 *                          and Stanford University. All rights reserved.    *
 *                                                                           *
 * Redistribution and use in source and binary forms,                        *
 * with or without modification, are permitted according to the terms        *
 * listed in LICENSE (http://roofit.sourceforge.net/license.txt)             *
 *****************************************************************************/

/**
\file RooDataHistSliceIter.cxx
\class RooDataHistSliceIter
\ingroup Roofitcore

RooDataHistSliceIter iterates over all bins in a RooDataHist that
occur in a slice defined by the bin coordinates of the input
sliceSet.
**/

#include "RooFit.h"

#include "RooDataHist.h"
#include "RooArgSet.h"
#include "RooAbsLValue.h"
#include "RooDataHistSliceIter.h"

using namespace std;

ClassImp(RooDataHistSliceIter);
;



////////////////////////////////////////////////////////////////////////////////
/// Construct an iterator over all bins of RooDataHist 'hist' in the slice defined
/// by the values of the arguments in 'sliceArg'

RooDataHistSliceIter::RooDataHistSliceIter(RooDataHist& hist, RooAbsArg& sliceArg) : _hist(&hist), _sliceArg(&sliceArg)
{
  // Calculate base index (for 0th bin) for slice    
  RooAbsArg* sliceArgInt = hist.get()->find(sliceArg.GetName()) ;
  dynamic_cast<RooAbsLValue&>(*sliceArgInt).setBin(0) ;

  if (hist._vars.getSize()>1) {
    _baseIndex = hist.calcTreeIndex() ;  
  } else {
    _baseIndex = 0 ;
  }

  _nStep = dynamic_cast<RooAbsLValue&>(*sliceArgInt).numBins() ;

//   cout << "RooDataHistSliceIter" << endl ;
//   hist.Print() ;
//   cout << "hist._iterator = " << hist._iterator << endl ;

  Int_t i=0 ;
  for (const auto arg : hist._vars) {
    if (arg==sliceArgInt) break ;
    i++ ;
  }
  _stepSize = hist._idxMult[i] ;  
  _curStep = 0 ;  

}



////////////////////////////////////////////////////////////////////////////////
/// Copy constructor

RooDataHistSliceIter::RooDataHistSliceIter(const RooDataHistSliceIter& other) : 
  TIterator(other), 
  _hist(other._hist), 
  _sliceArg(other._sliceArg),   
  _baseIndex(other._baseIndex),
  _stepSize(other._stepSize),
  _nStep(other._nStep), 
  _curStep(other._curStep)
{
}



////////////////////////////////////////////////////////////////////////////////
/// Destructor

RooDataHistSliceIter::~RooDataHistSliceIter() 
{
}



////////////////////////////////////////////////////////////////////////////////
/// Dummy

const TCollection* RooDataHistSliceIter::GetCollection() const 
{
  return 0 ;
}




////////////////////////////////////////////////////////////////////////////////
/// Iterator increment operator

TObject* RooDataHistSliceIter::Next() 
{  
  if (_curStep==_nStep) {
    return 0 ;
  }
  
  // Select appropriate entry in RooDataHist 
  _hist->get(_baseIndex + _curStep*_stepSize) ;

  // Increment iterator position 
  _curStep++ ;

  return _sliceArg ;
}



////////////////////////////////////////////////////////////////////////////////
/// Reset iterator position to beginning

void RooDataHistSliceIter::Reset() 
{
  _curStep=0 ;
}



////////////////////////////////////////////////////////////////////////////////
/// Iterator dereference operator, not functional for this iterator

TObject *RooDataHistSliceIter::operator*() const
{
   Int_t step = _curStep == 0 ? _curStep : _curStep - 1;
   // Select appropriate entry in RooDataHist 
   _hist->get(_baseIndex + step*_stepSize) ;

   return _sliceArg ;
}


////////////////////////////////////////////////////////////////////////////////
/// Returns true if position of this iterator differs from position
/// of iterator 'aIter'

bool RooDataHistSliceIter::operator!=(const TIterator &aIter) const
{
   if ((aIter.IsA() == RooDataHistSliceIter::Class())) {
      const RooDataHistSliceIter &iter(dynamic_cast<const RooDataHistSliceIter &>(aIter));
      return (_curStep != iter._curStep);
   }
   
   return false;
}