RooAbsSelfCachedReal.cxx

Go to the documentation of this file.

 /*****************************************************************************
  * Project: RooFit                                                           *
  *                                                                           *
  * 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 RooAbsSelfCachedReal.cxx
\class RooAbsSelfCachedReal
\ingroup Roofitcore
 
RooAbsSelfCachedReal is an abstract base class for functions whose
output is cached in terms of a histogram in all observables between
getVal() and evaluate(). For certain p.d.f.s that are very
expensive to calculate it may be beneficial to implement them as a
RooAbsSelfCachedReal rather than a RooAbsReal. Class
RooAbsSelfCachedReal is designed to have its interface identical to
that of RooAbsReal, so any p.d.f can make use of its caching
functionality by merely switching its base class.  Existing
RooAbsReal objects can also be cached a posteriori with the
RooCachedReal wrapper function that takes any RooAbsReal object as
input.
**/
 
#include "Riostream.h"
 
#include "RooAbsSelfCachedReal.h"
#include "RooAbsReal.h"
#include "RooMsgService.h"
#include "RooDataHist.h"
#include "RooHistPdf.h"
 
using namespace std ;
 
ClassImp(RooAbsSelfCachedReal);
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Constructor
 
RooAbsSelfCachedReal::RooAbsSelfCachedReal(const char *name, const char *title, Int_t ipOrder) :
  RooAbsCachedReal(name,title,ipOrder)
 {
 }
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Copy constructor
 
RooAbsSelfCachedReal::RooAbsSelfCachedReal(const RooAbsSelfCachedReal& other, const char* name) :
   RooAbsCachedReal(other,name)
 {
 }
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Destructor
 
RooAbsSelfCachedReal::~RooAbsSelfCachedReal()
{
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Fill cache with sampling of function as defined by the evaluate() implementation
 
void RooAbsSelfCachedReal::fillCacheObject(RooAbsCachedReal::FuncCacheElem& cache) const
{
  RooDataHist& cacheHist = *cache.hist() ;
 
  // Make deep clone of self in non-caching mde and attach to dataset observables
  RooArgSet* cloneSet = (RooArgSet*) RooArgSet(*this).snapshot(true) ;
  RooAbsSelfCachedReal* clone2 = (RooAbsSelfCachedReal*) cloneSet->find(GetName()) ;
  clone2->disableCache(true) ;
  clone2->attachDataSet(cacheHist) ;
 
  // Iterator over all bins of RooDataHist and fill weights
  for (Int_t i=0 ; i<cacheHist.numEntries() ; i++) {
    const RooArgSet* obs = cacheHist.get(i) ;
    double wgt = clone2->getVal(obs) ;
    cacheHist.set(i, wgt, 0.);
  }
 
  delete cloneSet ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Defines observables to be cached, given a set of user defined observables
/// Returns the subset of nset that are observables this p.d.f
 
RooFit::OwningPtr<RooArgSet> RooAbsSelfCachedReal::actualObservables(const RooArgSet& nset) const
{
  // Make list of servers
  RooArgSet serverSet;
 
  for (auto server : _serverList) {
    serverSet.add(*server);
  }
 
  // Return servers that are in common with given normalization set
  return RooFit::OwningPtr<RooArgSet>{(RooArgSet*) serverSet.selectCommon(nset)};
 
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Defines parameters on which cache contents depends. Returns
/// subset of variables of self that is not contained in the
/// supplied nset
 
RooFit::OwningPtr<RooArgSet> RooAbsSelfCachedReal::actualParameters(const RooArgSet& nset) const
{
  // Make list of servers
  RooArgSet *serverSet = new RooArgSet;
 
  for (auto server : _serverList) {
    serverSet->add(*server);
  }
 
  // Remove all given observables from server list
  serverSet->remove(nset,true,true);
 
  return RooFit::OwningPtr<RooArgSet>{serverSet};
}