RooDataProjBinding.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 RooDataProjBinding.cxx
\class RooDataProjBinding
\ingroup Roofitcore
 
adaptor that projects a real function via summation of states
provided in a dataset. The real function must be attached to the
dataset before creating this binding object.
 
If the dataset only contains category variables, the summation is optimized
performing a weighted sum over the states of a RooSuperCategory that is
constructed from all the categories in the dataset
 
**/
 
#include "RooFit.h"
 
#include "RooDataProjBinding.h"
#include "RooAbsReal.h"
#include "RooAbsData.h"
#include "Roo1DTable.h"
#include "RooSuperCategory.h"
#include "RooCategory.h"
#include "RooAbsPdf.h"
#include "RooMsgService.h"
 
#include <iostream>
#include <cassert>
 
using namespace std;
 
ClassImp(RooDataProjBinding);
;
 
 
////////////////////////////////////////////////////////////////////////////////
/// Constructor of a data weighted average function binding with
/// variables 'vars' for function 'real' and dataset 'data' with
/// weights.
 
RooDataProjBinding::RooDataProjBinding(const RooAbsReal &real, const RooAbsData& data, 
                   const RooArgSet &vars, const RooArgSet* nset) :
  RooRealBinding(real,vars,0), _first(kTRUE), _real(&real), _data(&data), _nset(nset), 
  _superCat(0), _catTable(0)
{  
  // Determine if dataset contains only categories
  TIterator* iter = data.get()->createIterator() ;
  Bool_t allCat(kTRUE) ;
  RooAbsArg* arg ;
  while((arg=(RooAbsArg*)iter->Next())) {
    if (!dynamic_cast<RooCategory*>(arg)) allCat = kFALSE ;
  }
  delete iter ;
 
  // Determine weights of various super categories fractions
  if (allCat) {
     _superCat = new RooSuperCategory("superCat","superCat",*data.get()) ;
     _catTable = data.table(*_superCat) ;
  }
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Destructor, delete owned objects
 
RooDataProjBinding::~RooDataProjBinding() 
{
  if (_superCat) delete _superCat ;
  if (_catTable) delete _catTable ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Evaluate data-projected values of the bound real function.
 
Double_t RooDataProjBinding::operator()(const Double_t xvector[]) const 
{
  assert(isValid());
  loadValues(xvector);    
 
  //RooAbsArg::setDirtyInhibit(kTRUE) ;
 
  Double_t result(0) ;
  Double_t wgtSum(0) ;  
 
  if (_catTable) {
 
    // Data contains only categories, sum over weighted supercategory states
    for (const auto& nameIdx : *_superCat) {
      // Backprop state to data set so that _real takes appropriate value
      _superCat->setIndex(nameIdx) ;
 
      // Add weighted sum
      Double_t wgt = _catTable->get(nameIdx.first.c_str());
      if (wgt) {
   result += wgt * _real->getVal(_nset) ;
   wgtSum += wgt ;
      }
    }
    
  } else {
 
    // Data contains reals, sum over all entries
    Int_t i ;
    Int_t nEvt = _data->numEntries() ;
 
    // Procedure might be lengthy, give some progress indication
    if (_first) {
      oocoutW(_real,Eval) << "RooDataProjBinding::operator() projecting over " << nEvt << " events" << endl ;
      _first = kFALSE ;
    } else {
      if (oodologW(_real,Eval)) {
   ooccoutW(_real,Eval) << "." ; cout.flush() ;
      }
    }
 
//     _real->Print("v") ;
//     ((RooAbsReal*)_real)->printCompactTree() ;
 
//     RooArgSet* params = _real->getObservables(_data->get()) ;
 
    for (i=0 ; i<nEvt ; i++) {
      _data->get(i) ;
 
      Double_t wgt = _data->weight() ;
      Double_t ret ;
      if (wgt) {  
   ret = _real->getVal(_nset) ;
   result += wgt * ret ;
//    cout << "ret[" << i << "] = " ;
//    params->printStream(cout,RooPrintable::kName|RooPrintable::kValue,RooPrintable::kStandard) ;
//    cout << " = " << ret << endl ;
   wgtSum += wgt ;
      }      
    }
  }
 
  //RooAbsArg::setDirtyInhibit(kFALSE) ;
 
  if (wgtSum==0) return 0 ;
  return result / wgtSum ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Evaluate the function at the specified values of the dependents.
RooSpan<const double> RooDataProjBinding::getValues(std::vector<RooSpan<const double>> coordinates) const {
  assert(isValid());
 
  if (!_batchBuffer)
    _batchBuffer.reset(new std::vector<double>());
  _batchBuffer->resize(coordinates.front().size());
 
  std::unique_ptr<double[]> xVec( new double[coordinates.size()] );
 
  for (std::size_t i=0; i < coordinates.front().size(); ++i) {
    for (unsigned int dim=0; dim < coordinates.size(); ++dim) {
      xVec.get()[dim] = coordinates[dim][i];
    }
 
    (*_batchBuffer)[i] = this->operator()(xVec.get());
  }
 
  return {*_batchBuffer};
}