RooAbsAnaConvPdf.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)             *
 *****************************************************************************/
 
//////////////////////////////////////////////////////////////////////////////
///  \class RooAbsAnaConvPdf
///  \ingroup Roofitcore
///
///  RooAbsAnaConvPdf is the base class for PDFs that represent a
///  physics model that can be analytically convolved with a resolution model.
///
///  To achieve factorization between the physics model and the resolution
///  model, each physics model must be able to be written in the form
///  \f[
///    \mathrm{Phys}(x, \bar{a}, \bar{b}) = \sum_k \mathrm{coef}_k(\bar{a}) * \mathrm{basis}_k(x,\bar{b})
///  \f]
///
///  where \f$ \mathrm{basis}_k \f$ are a limited number of functions in terms of the variable
///  to be convoluted, and \f$ \mathrm{coef}_k \f$ are coefficients independent of the convolution
///  variable.
///
///  Classes derived from RooResolutionModel implement
///  \f[
///     R_k(x,\bar{b},\bar{c}) = \int \mathrm{basis}_k(x', \bar{b}) \cdot \mathrm{resModel}(x-x',\bar{c}) \; \mathrm{d}x',
///  \f]
///
///  which RooAbsAnaConvPdf uses to construct the pdf for [ Phys (x) R ] :
///  \f[
///     \mathrm{PDF}(x,\bar{a},\bar{b},\bar{c}) = \sum_k \mathrm{coef}_k(\bar{a}) * R_k(x,\bar{b},\bar{c})
///  \f]
///
///  A minimal implementation of a RooAbsAnaConvPdf physics model consists of
///
///  - A constructor that declares the required basis functions using the declareBasis() method.
///    The declareBasis() function assigns a unique identifier code to each declare basis
///
///  - An implementation of `coefficient(Int_t code)` returning the coefficient value for each
///    declared basis function
///
///  Optionally, analytical integrals can be provided for the coefficient functions. The
///  interface for this is quite similar to that for integrals of regular PDFs. Two functions,
///  \code{.cpp}
///   Int_t getCoefAnalyticalIntegral(Int_t coef, RooArgSet& allVars, RooArgSet& analVars, const char* rangeName) const
///   double coefAnalyticalIntegral(Int_t coef, Int_t code, const char* rangeName) const
///  \endcode
///
///  advertise the coefficient integration capabilities and implement them respectively.
///  Please see RooAbsPdf for additional details. Advertised analytical integrals must be
///  valid for all coefficients.
 
#include "RooAbsAnaConvPdf.h"
 
#include "RooNormalizedPdf.h"
#include "RooMsgService.h"
#include "Riostream.h"
#include "RooResolutionModel.h"
#include "RooRealVar.h"
#include "RooFormulaVar.h"
#include "RooConvGenContext.h"
#include "RooGenContext.h"
#include "RooTruthModel.h"
#include "RooConvCoefVar.h"
#include "RooNameReg.h"
 
using namespace std;
 
ClassImp(RooAbsAnaConvPdf);
 
 
////////////////////////////////////////////////////////////////////////////////
/// Default constructor, required for persistence
 
RooAbsAnaConvPdf::RooAbsAnaConvPdf() :
  _isCopy(false),
  _coefNormMgr(this,10)
{
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Constructor. The supplied resolution model must be constructed with the same
/// convoluted variable as this physics model ('convVar')
 
RooAbsAnaConvPdf::RooAbsAnaConvPdf(const char *name, const char *title,
               const RooResolutionModel& model, RooRealVar& cVar) :
  RooAbsPdf(name,title), _isCopy(false),
  _model("!model","Original resolution model",this,(RooResolutionModel&)model,false,false),
  _convVar("!convVar","Convolution variable",this,cVar,false,false),
  _convSet("!convSet","Set of resModel X basisFunc convolutions",this),
  _coefNormMgr(this,10),
  _codeReg(10)
{
  _model.absArg()->setAttribute("NOCacheAndTrack") ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
 
RooAbsAnaConvPdf::RooAbsAnaConvPdf(const RooAbsAnaConvPdf& other, const char* name) :
  RooAbsPdf(other,name), _isCopy(true),
  _model("!model",this,other._model),
  _convVar("!convVar",this,other._convVar),
  _convSet("!convSet",this,other._convSet),
  _coefNormMgr(other._coefNormMgr,this),
  _codeReg(other._codeReg)
{
  // Copy constructor
  if (_model.absArg()) {
    _model.absArg()->setAttribute("NOCacheAndTrack") ;
  }
  other._basisList.snapshot(_basisList);
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Destructor
 
RooAbsAnaConvPdf::~RooAbsAnaConvPdf()
{
  if (!_isCopy) {
    std::vector<RooAbsArg*> tmp(_convSet.begin(), _convSet.end());
 
    for (auto arg : tmp) {
      _convSet.remove(*arg) ;
      delete arg ;
    }
  }
 
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Declare a basis function for use in this physics model. The string expression
/// must be a valid RooFormulVar expression representing the basis function, referring
/// to the convolution variable as '@0', and any additional parameters (supplied in
/// 'params' as '@1','@2' etc.
///
/// The return value is a unique identifier code, that will be passed to coefficient()
/// to identify the basis function for which the coefficient is requested. If the
/// resolution model used does not support the declared basis function, code -1 is
/// returned.
///
 
Int_t RooAbsAnaConvPdf::declareBasis(const char* expression, const RooArgList& params)
{
  // Sanity check
  if (_isCopy) {
    coutE(InputArguments) << "RooAbsAnaConvPdf::declareBasis(" << GetName() << "): ERROR attempt to "
           << " declare basis functions in a copied RooAbsAnaConvPdf" << endl ;
    return -1 ;
  }
 
  // Resolution model must support declared basis
  if (!((RooResolutionModel*)_model.absArg())->isBasisSupported(expression)) {
    coutE(InputArguments) << "RooAbsAnaConvPdf::declareBasis(" << GetName() << "): resolution model "
           << _model.absArg()->GetName()
           << " doesn't support basis function " << expression << endl ;
    return -1 ;
  }
 
  // Instantiate basis function
  RooArgList basisArgs(_convVar.arg()) ;
  basisArgs.add(params) ;
 
  TString basisName(expression) ;
  for (const auto arg : basisArgs) {
    basisName.Append("_") ;
    basisName.Append(arg->GetName()) ;
  }
 
  auto basisFunc = std::make_unique<RooFormulaVar>(basisName, expression, basisArgs);
  basisFunc->setAttribute("RooWorkspace::Recycle") ;
  basisFunc->setAttribute("NOCacheAndTrack") ;
  basisFunc->setOperMode(operMode()) ;
 
  // Instantiate resModel x basisFunc convolution
  RooAbsReal* conv = static_cast<RooResolutionModel*>(_model.absArg())->convolution(basisFunc.get(),this);
  _basisList.addOwned(std::move(basisFunc));
  if (!conv) {
    coutE(InputArguments) << "RooAbsAnaConvPdf::declareBasis(" << GetName() << "): unable to construct convolution with basis function '"
           << expression << "'" << endl ;
    return -1 ;
  }
  _convSet.add(*conv) ;
 
  return _convSet.index(conv) ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Change the current resolution model to newModel
 
bool RooAbsAnaConvPdf::changeModel(const RooResolutionModel& newModel)
{
  RooArgList newConvSet ;
  bool allOK(true) ;
  for (auto convArg : _convSet) {
    auto conv = static_cast<RooResolutionModel*>(convArg);
 
    // Build new resolution model
    std::unique_ptr<RooResolutionModel> newConv{newModel.convolution(const_cast<RooFormulaVar*>(&conv->basis()),this)};
    if (!newConvSet.addOwned(std::move(newConv))) {
      allOK = false ;
      break ;
    }
  }
 
  // Check if all convolutions were successfully built
  if (!allOK) {
    return true ;
  }
 
  // Replace old convolutions with new set
  _convSet.removeAll() ;
  _convSet.addOwned(std::move(newConvSet));
 
  const std::string attrib = std::string("ORIGNAME:") + _model->GetName();
  const bool oldAttrib = newModel.getAttribute(attrib.c_str());
  const_cast<RooResolutionModel&>(newModel).setAttribute(attrib.c_str());
 
  redirectServers(RooArgSet{newModel}, false, true);
 
  // reset temporary attribute for server redirection
  const_cast<RooResolutionModel&>(newModel).setAttribute(attrib.c_str(), oldAttrib);
 
  return false ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Create a generator context for this p.d.f. If both the p.d.f and the resolution model
/// support internal generation of the convolution observable on an infinite domain,
/// deploy a specialized convolution generator context, which generates the physics distribution
/// and the smearing separately, adding them a posteriori. If this is not possible return
/// a (slower) generic generation context that uses accept/reject sampling
 
RooAbsGenContext* RooAbsAnaConvPdf::genContext(const RooArgSet &vars, const RooDataSet *prototype,
                      const RooArgSet* auxProto, bool verbose) const
{
  // Check if the resolution model specifies a special context to be used.
  RooResolutionModel* conv = dynamic_cast<RooResolutionModel*>(_model.absArg());
  assert(conv);
 
  std::unique_ptr<RooArgSet> modelDep {_model->getObservables(&vars)};
  modelDep->remove(*convVar(),true,true) ;
  Int_t numAddDep = modelDep->getSize() ;
 
  // Check if physics PDF and resolution model can both directly generate the convolution variable
  RooArgSet dummy ;
  bool pdfCanDir = (getGenerator(*convVar(),dummy) != 0) ;
  bool resCanDir = conv && (conv->getGenerator(*convVar(),dummy)!=0) && conv->isDirectGenSafe(*convVar()) ;
 
  if (numAddDep>0 || !pdfCanDir || !resCanDir) {
    // Any resolution model with more dependents than the convolution variable
    // or pdf or resmodel do not support direct generation
    string reason ;
    if (numAddDep>0) reason += "Resolution model has more observables than the convolution variable. " ;
    if (!pdfCanDir) reason += "PDF does not support internal generation of convolution observable. " ;
    if (!resCanDir) reason += "Resolution model does not support internal generation of convolution observable. " ;
 
    coutI(Generation) << "RooAbsAnaConvPdf::genContext(" << GetName() << ") Using regular accept/reject generator for convolution p.d.f because: " << reason.c_str() << endl ;
    return new RooGenContext(*this,vars,prototype,auxProto,verbose) ;
  }
 
  RooAbsGenContext* context = conv->modelGenContext(*this, vars, prototype, auxProto, verbose);
  if (context) return context;
 
  // Any other resolution model: use specialized generator context
  return new RooConvGenContext(*this,vars,prototype,auxProto,verbose) ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return true if it is safe to generate the convolution observable
/// from the internal generator (this is the case if the chosen resolution
/// model is the truth model)
 
bool RooAbsAnaConvPdf::isDirectGenSafe(const RooAbsArg& arg) const
{
 
  // All direct generation of convolution arg if model is truth model
  if (!TString(_convVar.absArg()->GetName()).CompareTo(arg.GetName()) &&
      dynamic_cast<RooTruthModel*>(_model.absArg())) {
    return true ;
  }
 
  return RooAbsPdf::isDirectGenSafe(arg) ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return a pointer to the convolution variable instance used in the resolution model
 
RooAbsRealLValue* RooAbsAnaConvPdf::convVar()
{
  auto* conv = static_cast<RooResolutionModel*>(_convSet.at(0));
  if (!conv) return nullptr;
  return &conv->convVar() ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Calculate the current unnormalized value of the PDF
///
/// PDF = sum_k coef_k * [ basis_k (x) ResModel ]
///
 
double RooAbsAnaConvPdf::evaluate() const
{
  double result(0) ;
 
  Int_t index(0) ;
  for (auto convArg : _convSet) {
    auto conv = static_cast<RooAbsPdf*>(convArg);
    double coef = coefficient(index++) ;
    if (coef!=0.) {
      const double c = conv->getVal(nullptr);
      cxcoutD(Eval) << "RooAbsAnaConvPdf::evaluate(" << GetName() << ") val += coef*conv [" << index-1 << "/"
          << _convSet.size() << "] coef = " << coef << " conv = " << c << endl ;
      result += c * coef;
    } else {
      cxcoutD(Eval) << "RooAbsAnaConvPdf::evaluate(" << GetName() << ") [" << index-1 << "/" << _convSet.size() << "] coef = 0" << endl ;
    }
  }
 
  return result ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Advertise capability to perform (analytical) integrals
/// internally. For a given integration request over allVars while
/// normalized over normSet2 and in range 'rangeName', returns
/// largest subset that can be performed internally in analVars
/// Return code is unique integer code identifying integration scenario
/// to be passed to analyticalIntegralWN() to calculate requeste integral
///
/// Class RooAbsAnaConv defers analytical integration request to
/// resolution model and/or coefficient implementations and
/// aggregates results into composite configuration with a unique
/// code assigned by RooAICRegistry
 
Int_t RooAbsAnaConvPdf::getAnalyticalIntegralWN(RooArgSet& allVars,
                       RooArgSet& analVars, const RooArgSet* normSet2, const char* /*rangeName*/) const
{
  // Handle trivial no-integration scenario
  if (allVars.empty()) return 0 ;
 
  if (_forceNumInt) return 0 ;
 
  // Select subset of allVars that are actual dependents
  RooArgSet allDeps;
  getObservables(&allVars, allDeps);
  std::unique_ptr<RooArgSet> normSet{normSet2 ? getObservables(normSet2) : nullptr};
 
  RooArgSet intSetAll{allDeps,"intSetAll"};
 
  // Split intSetAll in coef/conv parts
  auto intCoefSet = std::make_unique<RooArgSet>("intCoefSet");
  auto intConvSet = std::make_unique<RooArgSet>("intConvSet");
 
  for (RooAbsArg * arg : intSetAll) {
    bool ok(true) ;
    for (RooAbsArg * conv : _convSet) {
      if (conv->dependsOn(*arg)) ok=false ;
    }
 
    if (ok) {
      intCoefSet->add(*arg) ;
    } else {
      intConvSet->add(*arg) ;
    }
 
  }
 
  // Split normSetAll in coef/conv parts
  auto normCoefSet = std::make_unique<RooArgSet>("normCoefSet");
  auto normConvSet = std::make_unique<RooArgSet>("normConvSet");
  if (normSet) {
    for (RooAbsArg * arg : *normSet) {
      bool ok(true) ;
      for (RooAbsArg * conv : _convSet) {
        if (conv->dependsOn(*arg)) ok=false ;
      }
 
      if (ok) {
        normCoefSet->add(*arg) ;
      } else {
        normConvSet->add(*arg) ;
      }
 
    }
  }
 
  if (intCoefSet->empty()) intCoefSet.reset();
  if (intConvSet->empty()) intConvSet.reset();
  if (normCoefSet->empty()) normCoefSet.reset();
  if (normConvSet->empty()) normConvSet.reset();
 
 
  // Store integration configuration in registry
  Int_t masterCode(0) ;
  std::vector<Int_t> tmp(1, 0) ;
 
  // takes ownership of all sets
  masterCode = _codeReg.store(tmp,
                              intCoefSet.release(),
                              intConvSet.release(),
                              normCoefSet.release(),
                              normConvSet.release()) + 1;
 
  analVars.add(allDeps) ;
 
  return masterCode  ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return analytical integral defined by given code, which is returned
/// by getAnalyticalIntegralWN()
///
/// For unnormalized integrals the returned value is
/// \f[
///     \mathrm{PDF} = \sum_k \int \mathrm{coef}_k \; \mathrm{d}\bar{x}
///         \cdot \int \mathrm{basis}_k (x) \mathrm{ResModel} \; \mathrm{d}\bar{y},
/// \f]
/// where \f$ \bar{x} \f$ is the set of coefficient dependents to be integrated,
/// and \f$ \bar{y} \f$ the set of basis function dependents to be integrated.
///
/// For normalized integrals this becomes
/// \f[
///   \mathrm{PDF} = \frac{\sum_k \int \mathrm{coef}_k \; \mathrm{d}x
///         \cdot \int \mathrm{basis}_k (x) \mathrm{ResModel} \; \mathrm{d}y}
///     {\sum_k \int \mathrm{coef}_k \; \mathrm{d}v
///         \cdot \int \mathrm{basis}_k (x) \mathrm{ResModel} \; \mathrm{d}w},
/// \f]
/// where
/// * \f$ x \f$ is the set of coefficient dependents to be integrated,
/// * \f$ y \f$ the set of basis function dependents to be integrated,
/// * \f$ v \f$ is the set of coefficient dependents over which is normalized and
/// * \f$ w \f$ is the set of basis function dependents over which is normalized.
///
/// Set \f$ x \f$ must be contained in \f$ v \f$ and set \f$ y \f$ must be contained in \f$ w \f$.
///
 
double RooAbsAnaConvPdf::analyticalIntegralWN(Int_t code, const RooArgSet* normSet, const char* rangeName) const
{
  // WVE needs adaptation to handle new rangeName feature
 
  // Handle trivial passthrough scenario
  if (code==0) return getVal(normSet) ;
 
  // Unpack master code
  RooArgSet *intCoefSet, *intConvSet, *normCoefSet, *normConvSet ;
  _codeReg.retrieve(code-1,intCoefSet,intConvSet,normCoefSet,normConvSet) ;
 
  Int_t index(0) ;
  double answer(0) ;
 
  if (normCoefSet==0&&normConvSet==0) {
 
    // Integral over unnormalized function
    double integral(0) ;
    const TNamed *_rangeName = RooNameReg::ptr(rangeName);
    for (auto convArg : _convSet) {
      auto conv = static_cast<RooAbsPdf*>(convArg);
      double coef = getCoefNorm(index++,intCoefSet,_rangeName) ;
      //cout << "coefInt[" << index << "] = " << coef << " " ; intCoefSet->Print("1") ;
      if (coef!=0) {
   integral += coef* conv->getNormObj(0,intConvSet,_rangeName)->getVal();
   cxcoutD(Eval) << "RooAbsAnaConv::aiWN(" << GetName() << ") [" << index-1 << "] integral += " << conv->getNorm(intConvSet) << endl ;
      }
 
    }
    answer = integral ;
 
  } else {
 
    // Integral over normalized function
    double integral(0) ;
    double norm(0) ;
    const TNamed *_rangeName = RooNameReg::ptr(rangeName);
    for (auto convArg : _convSet) {
      auto conv = static_cast<RooAbsPdf*>(convArg);
 
      double coefInt = getCoefNorm(index,intCoefSet,_rangeName) ;
      //cout << "coefInt[" << index << "] = " << coefInt << "*" << term << " " << (intCoefSet?*intCoefSet:RooArgSet()) << endl ;
      if (coefInt!=0) {
   double term = conv->getNormObj(0,intConvSet,_rangeName)->getVal();
   integral += coefInt*term ;
      }
 
      double coefNorm = getCoefNorm(index,normCoefSet) ;
      //cout << "coefNorm[" << index << "] = " << coefNorm << "*" << term << " " << (normCoefSet?*normCoefSet:RooArgSet()) << endl ;
      if (coefNorm!=0) {
   double term = conv->getNormObj(0,normConvSet)->getVal();
   norm += coefNorm*term ;
      }
 
      index++ ;
    }
    answer = integral/norm ;
  }
 
  return answer ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Default implementation of function advertising integration capabilities. The interface is
/// similar to that of getAnalyticalIntegral except that an integer code is added that
/// designates the coefficient number for which the integration capabilities are requested
///
/// This default implementation advertises that no internal integrals are supported.
 
Int_t RooAbsAnaConvPdf::getCoefAnalyticalIntegral(Int_t /* coef*/, RooArgSet& /*allVars*/, RooArgSet& /*analVars*/, const char* /*rangeName*/) const
{
  return 0 ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Default implementation of function implementing advertised integrals. Only
/// the pass-through scenario (no integration) is implemented.
 
double RooAbsAnaConvPdf::coefAnalyticalIntegral(Int_t coef, Int_t code, const char* /*rangeName*/) const
{
  if (code==0) return coefficient(coef) ;
  coutE(InputArguments) << "RooAbsAnaConvPdf::coefAnalyticalIntegral(" << GetName() << ") ERROR: unrecognized integration code: " << code << endl ;
  assert(0) ;
  return 1 ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// This function forces RooRealIntegral to offer all integration dependents
/// to RooAbsAnaConvPdf::getAnalyticalIntegralWN() for consideration for
/// internal integration, if RooRealIntegral considers this to be unsafe (e.g. due
/// to hidden Jacobian terms).
///
/// RooAbsAnaConvPdf will not attempt to actually integrate all these dependents
/// but feed them to the resolution models integration interface, which will
/// make the final determination on how to integrate these dependents.
 
bool RooAbsAnaConvPdf::forceAnalyticalInt(const RooAbsArg& /*dep*/) const
{
  return true ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Returns the normalization integral value of the coefficient with number coefIdx over normalization
/// set nset in range rangeName
 
double RooAbsAnaConvPdf::getCoefNorm(Int_t coefIdx, const RooArgSet* nset, const TNamed* rangeName) const
{
  if (nset==0) return coefficient(coefIdx) ;
 
  CacheElem* cache = (CacheElem*) _coefNormMgr.getObj(nset,0,0,rangeName) ;
  if (!cache) {
 
    cache = new CacheElem ;
 
    // Make list of coefficient normalizations
    Int_t i ;
    makeCoefVarList(cache->_coefVarList) ;
 
    for (i=0 ; i<cache->_coefVarList.getSize() ; i++) {
      cache->_normList.addOwned(std::unique_ptr<RooAbsReal>{static_cast<RooAbsReal&>(*cache->_coefVarList.at(i)).createIntegral(*nset,RooNameReg::str(rangeName))});
    }
 
    _coefNormMgr.setObj(nset,0,cache,rangeName) ;
  }
 
  return ((RooAbsReal*)cache->_normList.at(coefIdx))->getVal() ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Build complete list of coefficient variables
 
void RooAbsAnaConvPdf::makeCoefVarList(RooArgList& varList) const
{
  // Instantate a coefficient variables
  for (Int_t i=0 ; i<_convSet.getSize() ; i++) {
    auto cvars = coefVars(i);
    varList.addOwned(std::make_unique<RooConvCoefVar>(Form("%s_coefVar_%d",GetName(),i),"coefVar",*this,i,&*cvars));
  }
 
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return set of parameters with are used exclusively by the coefficient functions
 
RooFit::OwningPtr<RooArgSet> RooAbsAnaConvPdf::coefVars(Int_t /*coefIdx*/) const
{
  auto cVars = getParameters(static_cast<RooArgSet*>(nullptr));
  std::vector<RooAbsArg*> tmp;
  for (auto arg : *cVars) {
    for (auto convSetArg : _convSet) {
      if (convSetArg->dependsOn(*arg)) {
        tmp.push_back(arg);
      }
    }
  }
 
  cVars->remove(tmp.begin(), tmp.end(), true, true);
 
  return RooFit::OwningPtr<RooArgSet>{std::move(cVars)};
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Print info about this object to the specified stream. In addition to the info
/// from RooAbsPdf::printStream() we add:
///
///   Verbose : detailed information on convolution integrals
 
void RooAbsAnaConvPdf::printMultiline(ostream& os, Int_t contents, bool verbose, TString indent) const
{
  RooAbsPdf::printMultiline(os,contents,verbose,indent);
 
  os << indent << "--- RooAbsAnaConvPdf ---" << endl;
  for (RooAbsArg * conv : _convSet) {
    conv->printMultiline(os,contents,verbose,indent) ;
  }
}
 
 
///////////////////////////////////////////////////////////////////////////////
/// Label OK'ed components with cache-and-track
void RooAbsAnaConvPdf::setCacheAndTrackHints(RooArgSet& trackNodes)
{
  for (auto const* carg : static_range_cast<RooAbsArg*>(_convSet)) {
    if (carg->canNodeBeCached()==Always) {
      trackNodes.add(*carg) ;
      //cout << "tracking node RooAddPdf component " << carg->ClassName() << "::" << carg->GetName() << endl ;
    }
  }
}
 
std::unique_ptr<RooAbsArg>
RooAbsAnaConvPdf::compileForNormSet(RooArgSet const &normSet, RooFit::Detail::CompileContext &ctx) const
{
   // If there is only one component in the linear sum of convolutions, we can
   // just return that one, normalized.
   if(_convSet.size() == 1) {
      if (normSet.empty()) {
         return _convSet[0].compileForNormSet(normSet, ctx);
      }
      std::unique_ptr<RooAbsPdf> pdfClone(static_cast<RooAbsPdf *>(_convSet[0].Clone()));
      ctx.compileServers(*pdfClone, normSet);
 
      auto newArg = std::make_unique<RooNormalizedPdf>(*pdfClone, normSet);
 
      // The direct servers are this pdf and the normalization integral, which
      // don't need to be compiled further.
      for (RooAbsArg *server : newArg->servers()) {
         server->setAttribute("_COMPILED");
      }
      newArg->setAttribute("_COMPILED");
      newArg->addOwnedComponents(std::move(pdfClone));
      return newArg;
   }
 
   // Here, we can't use directly the function from the RooAbsPdf base class,
   // because the convolution argument servers need to be evaluated
   // unnormalized, even if they are pdfs.
 
   if (normSet.empty()) {
      return RooAbsPdf::compileForNormSet(normSet, ctx);
   }
   std::unique_ptr<RooAbsAnaConvPdf> pdfClone(static_cast<RooAbsAnaConvPdf *>(this->Clone()));
 
   // The actual resolution model is not serving the RooAbsAnaConvPdf
   // in the evaluation. It was only used get the convolutions with a given
   // basis. We can remove it for the compiled model.
   pdfClone->removeServer(const_cast<RooAbsReal &>(pdfClone->_model.arg()), true);
 
   // The other servers will be compiled with the original normSet, but the
   // _convSet has to be evaluated unnormalized.
   RooArgList convArgClones;
   for (RooAbsArg *convArg : _convSet) {
      if (auto convArgClone = ctx.compile(*convArg, *pdfClone, {})) {
         convArgClones.add(*convArgClone);
      }
   }
   pdfClone->redirectServers(convArgClones, false, true);
 
   // Compile remaining servers that are evaluated normalized
   ctx.compileServers(*pdfClone, normSet);
 
   // Finally, this RooAbsAnaConvPdf needs to be normalized
   auto newArg = std::make_unique<RooNormalizedPdf>(*pdfClone, normSet);
 
   // The direct servers are this pdf and the normalization integral, which
   // don't need to be compiled further.
   for (RooAbsArg *server : newArg->servers()) {
      server->setAttribute("_COMPILED");
   }
   newArg->setAttribute("_COMPILED");
   newArg->addOwnedComponents(std::move(pdfClone));
   return newArg;
}