#include "RooFit.h"
#include "RooMsgService.h"
#include "TClass.h"
#include "Riostream.h"
#include "TMath.h"
#include "TObjString.h"
#include "TPaveText.h"
#include "TList.h"
#include "TH1.h"
#include "TH2.h"
#include "RooAbsPdf.h"
#include "RooDataSet.h"
#include "RooArgSet.h"
#include "RooArgProxy.h"
#include "RooRealProxy.h"
#include "RooRealVar.h"
#include "RooGenContext.h"
#include "RooPlot.h"
#include "RooCurve.h"
#include "RooNLLVar.h"
#include "RooMinuit.h"
#include "RooCategory.h"
#include "RooNameReg.h"
#include "RooCmdConfig.h"
#include "RooGlobalFunc.h"
#include "RooAddition.h"
#include "RooRandom.h"
#include "RooNumIntConfig.h"
#include "RooProjectedPdf.h"
#include "RooInt.h"
#include "RooCustomizer.h"
#include "RooConstraintSum.h"
#include "RooParamBinning.h"
#include "RooNumCdf.h"
#include <string>
ClassImp(RooAbsPdf)
;
Int_t RooAbsPdf::_verboseEval = 0;
Bool_t RooAbsPdf::_evalError = kFALSE ;
RooAbsPdf::RooAbsPdf() : _norm(0), _normSet(0)
{
}
RooAbsPdf::RooAbsPdf(const char *name, const char *title) :
RooAbsReal(name,title), _norm(0), _normSet(0), _normMgr(this,10), _selectComp(kTRUE)
{
resetErrorCounters() ;
setTraceCounter(0) ;
}
RooAbsPdf::RooAbsPdf(const char *name, const char *title,
Double_t plotMin, Double_t plotMax) :
RooAbsReal(name,title,plotMin,plotMax), _norm(0), _normSet(0), _normMgr(this,10), _selectComp(kTRUE)
{
resetErrorCounters() ;
setTraceCounter(0) ;
}
RooAbsPdf::RooAbsPdf(const RooAbsPdf& other, const char* name) :
RooAbsReal(other,name), _norm(0), _normSet(0), _normMgr(other._normMgr,this), _selectComp(other._selectComp)
{
resetErrorCounters() ;
setTraceCounter(other._traceCount) ;
}
RooAbsPdf::~RooAbsPdf()
{
}
Double_t RooAbsPdf::getVal(const RooArgSet* nset) const
{
if (!nset) {
Double_t val = evaluate() ;
Bool_t error = traceEvalPdf(val) ;
cxcoutD(Tracing) << IsA()->GetName() << "::getVal(" << GetName()
<< "): value = " << val << " (unnormalized)" << endl ;
if (error) {
raiseEvalError() ;
return 0 ;
}
return val ;
}
Bool_t nsetChanged(kFALSE) ;
if (nset!=_normSet || _norm==0) {
nsetChanged = syncNormalization(nset) ;
}
if ((isValueDirty() || nsetChanged || _norm->isValueDirty()) && operMode()!=AClean) {
Double_t rawVal = evaluate() ;
Bool_t error = traceEvalPdf(rawVal) ;
Double_t normVal(_norm->getVal()) ;
Double_t normError(kFALSE) ;
if (normVal==0.) {
normError=kTRUE ;
logEvalError("p.d.f normalization integral is zero") ;
}
if (normError||error) raiseEvalError() ;
_value = normError ? 0 : (rawVal / normVal) ;
cxcoutD(Tracing) << "RooAbsPdf::getVal(" << GetName() << ") new value with norm " << _norm->GetName() << " = " << rawVal << "/" << normVal << " = " << _value << endl ;
clearValueDirty() ;
clearShapeDirty() ;
}
if (_traceCount>0) {
cxcoutD(Tracing) << "[" << _traceCount << "] " ;
Int_t tmp = _traceCount ;
_traceCount = 0 ;
printStream(ccoutD(Tracing),kName|kValue|kArgs,kSingleLine) ;
_traceCount = tmp-1 ;
}
return _value ;
}
Double_t RooAbsPdf::analyticalIntegralWN(Int_t code, const RooArgSet* normSet, const char* rangeName) const
{
cxcoutD(Eval) << "RooAbsPdf::analyticalIntegralWN(" << GetName() << ") code = " << code << " normset = " << (normSet?*normSet:RooArgSet()) << endl ;
if (code==0) return getVal(normSet) ;
if (normSet) {
return analyticalIntegral(code,rangeName) / getNorm(normSet) ;
} else {
return analyticalIntegral(code,rangeName) ;
}
}
Bool_t RooAbsPdf::traceEvalPdf(Double_t value) const
{
Bool_t error= isnan(value) || (value < 0);
if (isnan(value)) {
logEvalError(Form("p.d.f value is Not-a-Number (%f), forcing value to zero",value)) ;
}
if (value<0) {
logEvalError(Form("p.d.f value is less than zero (%f), forcing value to zero",value)) ;
}
if(!error) return error ;
if(++_errorCount <= 10) {
cxcoutD(Tracing) << "*** Evaluation Error " << _errorCount << " ";
if(_errorCount == 10) cxcoutD(Tracing) << "(no more will be printed) ";
}
else {
return error ;
}
Print() ;
return error ;
}
Double_t RooAbsPdf::getNorm(const RooArgSet* nset) const
{
if (!nset) return 1 ;
syncNormalization(nset,kTRUE) ;
if (_verboseEval>1) cxcoutD(Tracing) << IsA()->GetName() << "::getNorm(" << GetName() << "): norm(" << _norm << ") = " << _norm->getVal() << endl ;
Double_t ret = _norm->getVal() ;
if (ret==0.) {
if(++_errorCount <= 10) {
coutW(Eval) << "RooAbsPdf::getNorm(" << GetName() << ":: WARNING normalization is zero, nset = " ; nset->Print("1") ;
if(_errorCount == 10) coutW(Eval) << "RooAbsPdf::getNorm(" << GetName() << ") INFO: no more messages will be printed " << endl ;
}
}
return ret ;
}
const RooAbsReal* RooAbsPdf::getNormObj(const RooArgSet* nset, const RooArgSet* iset, const TNamed* rangeName) const
{
CacheElem* cache = (CacheElem*) _normMgr.getObj(nset,iset,0,rangeName) ;
if (cache) {
return cache->_norm ;
}
RooArgSet* depList = getObservables(iset) ;
RooAbsReal* norm = createIntegral(*depList,*nset, *getIntegratorConfig(), RooNameReg::str(rangeName)) ;
delete depList ;
cache = new CacheElem(*norm) ;
_normMgr.setObj(nset,iset,cache,rangeName) ;
return norm ;
}
Bool_t RooAbsPdf::syncNormalization(const RooArgSet* nset, Bool_t adjustProxies) const
{
_normSet = (RooArgSet*) nset ;
CacheElem* cache = (CacheElem*) _normMgr.getObj(nset) ;
if (cache) {
Bool_t nsetChanged = (_norm!=cache->_norm) ;
_norm = cache->_norm ;
if (nsetChanged && adjustProxies) {
((RooAbsPdf*) this)->setProxyNormSet(nset) ;
}
return nsetChanged ;
}
if (adjustProxies) {
((RooAbsPdf*) this)->setProxyNormSet(nset) ;
}
RooArgSet* depList = getObservables(nset) ;
if (_verboseEval>0) {
if (!selfNormalized()) {
cxcoutD(Tracing) << IsA()->GetName() << "::syncNormalization(" << GetName()
<< ") recreating normalization integral " << endl ;
if (depList) depList->printStream(ccoutD(Tracing),kName|kValue|kArgs,kSingleLine) ; else ccoutD(Tracing) << "<none>" << endl ;
} else {
cxcoutD(Tracing) << IsA()->GetName() << "::syncNormalization(" << GetName() << ") selfNormalized, creating unit norm" << endl;
}
}
if (selfNormalized() || !dependsOn(*depList)) {
TString ntitle(GetTitle()) ; ntitle.Append(" Unit Normalization") ;
TString nname(GetName()) ; nname.Append("_UnitNorm") ;
_norm = new RooRealVar(nname.Data(),ntitle.Data(),1) ;
} else {
_norm = createIntegral(*depList,*getIntegratorConfig()) ;
}
cache = new CacheElem(*_norm) ;
_normMgr.setObj(nset,cache) ;
delete depList ;
return kTRUE ;
}
Bool_t RooAbsPdf::traceEvalHook(Double_t value) const
{
Bool_t error= isnan(value) || (value < 0);
if(!error && _traceCount <= 0) return error ;
if(error && ++_errorCount <= 10) {
cxcoutD(Tracing) << "*** Evaluation Error " << _errorCount << " ";
if(_errorCount == 10) ccoutD(Tracing) << "(no more will be printed) ";
}
else if(_traceCount > 0) {
ccoutD(Tracing) << '[' << _traceCount-- << "] ";
}
else {
return error ;
}
Print() ;
return error ;
}
void RooAbsPdf::resetErrorCounters(Int_t resetValue)
{
_errorCount = resetValue ;
_negCount = resetValue ;
}
void RooAbsPdf::setTraceCounter(Int_t value, Bool_t allNodes)
{
if (!allNodes) {
_traceCount = value ;
return ;
} else {
RooArgList branchList ;
branchNodeServerList(&branchList) ;
TIterator* iter = branchList.createIterator() ;
RooAbsArg* arg ;
while((arg=(RooAbsArg*)iter->Next())) {
RooAbsPdf* pdf = dynamic_cast<RooAbsPdf*>(arg) ;
if (pdf) pdf->setTraceCounter(value,kFALSE) ;
}
delete iter ;
}
}
Double_t RooAbsPdf::getLogVal(const RooArgSet* nset) const
{
Double_t prob = getVal(nset) ;
if(prob <= 0) {
logEvalError("getLogVal() top-level p.d.f evaluates to zero or negative number") ;
return 0;
}
return log(prob);
}
Double_t RooAbsPdf::extendedTerm(UInt_t observed, const RooArgSet* nset) const
{
if(!canBeExtended()) {
coutE(InputArguments) << fName << ": this PDF does not support extended maximum likelihood"
<< endl;
return 0;
}
Double_t expected= expectedEvents(nset);
if(expected < 0) {
coutE(InputArguments) << fName << ": calculated negative expected events: " << expected
<< endl;
return 0;
}
Double_t extra= expected - observed*log(expected);
Bool_t trace(kFALSE) ;
if(trace) {
cxcoutD(Tracing) << fName << "::extendedTerm: expected " << expected << " events, got "
<< observed << " events. extendedTerm = " << extra << endl;
}
return extra;
}
RooAbsReal* RooAbsPdf::createNLL(RooAbsData& data, RooCmdArg arg1, RooCmdArg arg2, RooCmdArg arg3, RooCmdArg arg4,
RooCmdArg arg5, RooCmdArg arg6, RooCmdArg arg7, RooCmdArg arg8)
{
RooLinkedList l ;
l.Add((TObject*)&arg1) ; l.Add((TObject*)&arg2) ;
l.Add((TObject*)&arg3) ; l.Add((TObject*)&arg4) ;
l.Add((TObject*)&arg5) ; l.Add((TObject*)&arg6) ;
l.Add((TObject*)&arg7) ; l.Add((TObject*)&arg8) ;
return createNLL(data,l) ;
}
RooAbsReal* RooAbsPdf::createNLL(RooAbsData& data, const RooLinkedList& cmdList)
{
RooCmdConfig pc(Form("RooAbsPdf::createNLL(%s)",GetName())) ;
pc.defineString("rangeName","RangeWithName",0,"",kTRUE) ;
pc.defineString("addCoefRange","SumCoefRange",0,"") ;
pc.defineDouble("rangeLo","Range",0,-999.) ;
pc.defineDouble("rangeHi","Range",1,-999.) ;
pc.defineInt("splitRange","SplitRange",0,0) ;
pc.defineInt("ext","Extended",0,2) ;
pc.defineInt("numcpu","NumCPU",0,1) ;
pc.defineInt("verbose","Verbose",0,0) ;
pc.defineInt("optConst","Optimize",0,1) ;
pc.defineObject("projDepSet","ProjectedObservables",0,0) ;
pc.defineObject("cPars","Constrain",0,0) ;
pc.defineObject("extCons","ExternalConstraints",0,0) ;
pc.defineMutex("Range","RangeWithName") ;
pc.process(cmdList) ;
if (!pc.ok(kTRUE)) {
return 0 ;
}
const char* rangeName = pc.getString("rangeName",0,kTRUE) ;
const char* addCoefRangeName = pc.getString("addCoefRange",0,kTRUE) ;
Int_t ext = pc.getInt("ext") ;
Int_t numcpu = pc.getInt("numcpu") ;
Int_t splitr = pc.getInt("splitRange") ;
Bool_t verbose = pc.getInt("verbose") ;
Int_t optConst = pc.getInt("optConst") ;
const RooArgSet* cPars = static_cast<RooArgSet*>(pc.getObject("cPars")) ;
const RooArgSet* extCons = static_cast<RooArgSet*>(pc.getObject("extCons")) ;
if (ext==2) {
ext = ((extendMode()==CanBeExtended || extendMode()==MustBeExtended)) ? 1 : 0 ;
if (ext) {
coutI(Minimization) << "p.d.f. provides expected number of events, including extended term in likelihood." << endl ;
}
}
if (pc.hasProcessed("Range")) {
Double_t rangeLo = pc.getDouble("rangeLo") ;
Double_t rangeHi = pc.getDouble("rangeHi") ;
RooArgSet* obs = getObservables(&data) ;
TIterator* iter = obs->createIterator() ;
RooAbsArg* arg ;
while((arg=(RooAbsArg*)iter->Next())) {
RooRealVar* rrv = dynamic_cast<RooRealVar*>(arg) ;
if (rrv) rrv->setRange("fit",rangeLo,rangeHi) ;
}
rangeName = "fit" ;
}
RooArgSet projDeps ;
RooArgSet* tmp = (RooArgSet*) pc.getObject("projDepSet") ;
if (tmp) {
projDeps.add(*tmp) ;
}
RooAbsReal::enableEvalErrorLogging(kTRUE) ;
RooAbsReal* nll ;
if (!rangeName || strchr(rangeName,',')==0) {
nll = new RooNLLVar("nll","-log(likelihood)",*this,data,projDeps,ext,rangeName,addCoefRangeName,numcpu,kFALSE,verbose,splitr) ;
} else {
RooArgList nllList ;
char* buf = new char[strlen(rangeName)+1] ;
strcpy(buf,rangeName) ;
char* token = strtok(buf,",") ;
while(token) {
RooAbsReal* nllComp = new RooNLLVar(Form("nll_%s",token),"-log(likelihood)",*this,data,projDeps,ext,token,addCoefRangeName,numcpu,kFALSE,verbose,splitr) ;
nllList.add(*nllComp) ;
token = strtok(0,",") ;
}
delete[] buf ;
nll = new RooAddition("nll","-log(likelihood)",nllList,kTRUE) ;
}
RooAbsReal::enableEvalErrorLogging(kFALSE) ;
RooArgSet allConstraints ;
if (cPars) {
RooArgSet* constraints = getAllConstraints(*data.get(),*cPars) ;
allConstraints.add(*constraints) ;
delete constraints ;
}
if (extCons) {
allConstraints.add(*extCons) ;
}
RooAbsReal* nllCons(0) ;
if (allConstraints.getSize()>0) {
coutI(Minimization) << " Including the following contraint terms in minimization: " << allConstraints << endl ;
nllCons = new RooConstraintSum("nllCons","nllCons",allConstraints) ;
RooAbsReal* orignll = nll ;
nll = new RooAddition("nllWithCons","nllWithCons",RooArgSet(*nll,*nllCons)) ;
nll->addOwnedComponents(RooArgSet(*orignll,*nllCons)) ;
}
if (optConst) {
nll->constOptimizeTestStatistic(RooAbsArg::Activate) ;
}
return nll ;
}
RooFitResult* RooAbsPdf::fitTo(RooAbsData& data, RooCmdArg arg1, RooCmdArg arg2, RooCmdArg arg3, RooCmdArg arg4,
RooCmdArg arg5, RooCmdArg arg6, RooCmdArg arg7, RooCmdArg arg8)
{
RooLinkedList l ;
l.Add((TObject*)&arg1) ; l.Add((TObject*)&arg2) ;
l.Add((TObject*)&arg3) ; l.Add((TObject*)&arg4) ;
l.Add((TObject*)&arg5) ; l.Add((TObject*)&arg6) ;
l.Add((TObject*)&arg7) ; l.Add((TObject*)&arg8) ;
return fitTo(data,l) ;
}
RooFitResult* RooAbsPdf::fitTo(RooAbsData& data, const RooLinkedList& cmdList)
{
RooCmdConfig pc(Form("RooAbsPdf::fitTo(%s)",GetName())) ;
pc.defineString("fitOpt","FitOptions",0,"") ;
pc.defineString("rangeName","RangeWithName",0,"",kTRUE) ;
pc.defineString("addCoefRange","SumCoefRange",0,"") ;
pc.defineDouble("rangeLo","Range",0,-999.) ;
pc.defineDouble("rangeHi","Range",1,-999.) ;
pc.defineInt("splitRange","SplitRange",0,0) ;
pc.defineInt("optConst","Optimize",0,1) ;
pc.defineInt("verbose","Verbose",0,0) ;
pc.defineInt("doSave","Save",0,0) ;
pc.defineInt("doTimer","Timer",0,0) ;
pc.defineInt("plevel","PrintLevel",0,1) ;
pc.defineInt("strat","Strategy",0,1) ;
pc.defineInt("initHesse","InitialHesse",0,0) ;
pc.defineInt("hesse","Hesse",0,1) ;
pc.defineInt("minos","Minos",0,0) ;
pc.defineInt("ext","Extended",0,2) ;
pc.defineInt("numcpu","NumCPU",0,1) ;
pc.defineInt("numee","PrintEvalErrors",0,10) ;
pc.defineInt("doEEWall","EvalErrorWall",0,1) ;
pc.defineInt("doWarn","Warnings",0,1) ;
pc.defineObject("projDepSet","ProjectedObservables",0,0) ;
pc.defineObject("minosSet","Minos",0,0) ;
pc.defineObject("cPars","Constrain",0,0) ;
pc.defineObject("extCons","ExternalConstraints",0,0) ;
pc.defineMutex("FitOptions","Verbose") ;
pc.defineMutex("FitOptions","Save") ;
pc.defineMutex("FitOptions","Timer") ;
pc.defineMutex("FitOptions","Strategy") ;
pc.defineMutex("FitOptions","InitialHesse") ;
pc.defineMutex("FitOptions","Hesse") ;
pc.defineMutex("FitOptions","Minos") ;
pc.defineMutex("Range","RangeWithName") ;
pc.process(cmdList) ;
if (!pc.ok(kTRUE)) {
return 0 ;
}
const char* fitOpt = pc.getString("fitOpt",0,kTRUE) ;
const char* rangeName = pc.getString("rangeName",0,kTRUE) ;
const char* addCoefRangeName = pc.getString("addCoefRange",0,kTRUE) ;
Int_t optConst = pc.getInt("optConst") ;
Int_t verbose = pc.getInt("verbose") ;
Int_t doSave = pc.getInt("doSave") ;
Int_t doTimer = pc.getInt("doTimer") ;
Int_t plevel = pc.getInt("plevel") ;
Int_t strat = pc.getInt("strat") ;
Int_t initHesse= pc.getInt("initHesse") ;
Int_t hesse = pc.getInt("hesse") ;
Int_t minos = pc.getInt("minos") ;
Int_t ext = pc.getInt("ext") ;
Int_t numcpu = pc.getInt("numcpu") ;
Int_t splitr = pc.getInt("splitRange") ;
Int_t numee = pc.getInt("numee") ;
Int_t doEEWall = pc.getInt("doEEWall") ;
Int_t doWarn = pc.getInt("doWarn") ;
const RooArgSet* minosSet = static_cast<RooArgSet*>(pc.getObject("minosSet")) ;
const RooArgSet* cPars = static_cast<RooArgSet*>(pc.getObject("cPars")) ;
const RooArgSet* extCons = static_cast<RooArgSet*>(pc.getObject("extCons")) ;
if (ext==2) {
ext = ((extendMode()==CanBeExtended || extendMode()==MustBeExtended)) ? 1 : 0 ;
if (ext) {
coutI(Minimization) << "p.d.f. provides expected number of events, including extended term in likelihood." << endl ;
}
}
if (pc.hasProcessed("Range")) {
Double_t rangeLo = pc.getDouble("rangeLo") ;
Double_t rangeHi = pc.getDouble("rangeHi") ;
RooArgSet* obs = getObservables(&data) ;
TIterator* iter = obs->createIterator() ;
RooAbsArg* arg ;
while((arg=(RooAbsArg*)iter->Next())) {
RooRealVar* rrv = dynamic_cast<RooRealVar*>(arg) ;
if (rrv) rrv->setRange("fit",rangeLo,rangeHi) ;
}
rangeName = "fit" ;
}
RooArgSet projDeps ;
RooArgSet* tmp = (RooArgSet*) pc.getObject("projDepSet") ;
if (tmp) {
projDeps.add(*tmp) ;
}
RooAbsReal::enableEvalErrorLogging(kTRUE) ;
RooAbsReal* nll ;
if (!rangeName || strchr(rangeName,',')==0) {
nll = new RooNLLVar("nll","-log(likelihood)",*this,data,projDeps,ext,rangeName,addCoefRangeName,numcpu,kFALSE,plevel!=-1,splitr) ;
} else {
RooArgList nllList ;
char* buf = new char[strlen(rangeName)+1] ;
strcpy(buf,rangeName) ;
char* token = strtok(buf,",") ;
while(token) {
RooAbsReal* nllComp = new RooNLLVar(Form("nll_%s",token),"-log(likelihood)",*this,data,projDeps,ext,token,addCoefRangeName,numcpu,kFALSE,plevel!=-1,splitr) ;
nllList.add(*nllComp) ;
token = strtok(0,",") ;
}
delete[] buf ;
nll = new RooAddition("nll","-log(likelihood)",nllList,kTRUE) ;
}
RooAbsReal::enableEvalErrorLogging(kFALSE) ;
RooArgSet allConstraints ;
if (cPars) {
RooArgSet* constraints = getAllConstraints(*data.get(),*cPars) ;
allConstraints.add(*constraints) ;
delete constraints ;
}
if (extCons) {
allConstraints.add(*extCons) ;
}
RooAbsReal* nllCons(0) ;
if (allConstraints.getSize()>0) {
coutI(Minimization) << " Including the following contraint terms in minimization: " << allConstraints << endl ;
nllCons = new RooConstraintSum("nllCons","nllCons",allConstraints) ;
RooAbsReal* orignll = nll ;
nll = new RooAddition("nllWithCons","nllWithCons",RooArgSet(*nll,*nllCons)) ;
nll->addOwnedComponents(RooArgSet(*orignll,*nllCons)) ;
}
RooMinuit m(*nll) ;
m.setEvalErrorWall(doEEWall) ;
if (doWarn==0) {
m.setNoWarn() ;
}
m.setPrintEvalErrors(numee) ;
if (plevel!=1) {
m.setPrintLevel(plevel) ;
}
if (optConst) {
m.optimizeConst(1) ;
}
RooFitResult *ret = 0 ;
if (fitOpt) {
ret = m.fit(fitOpt) ;
} else {
if (verbose) {
m.setVerbose(1) ;
}
if (doTimer) {
m.setProfile(1) ;
}
if (strat!=1) {
m.setStrategy(strat) ;
}
if (initHesse) {
m.hesse() ;
}
m.migrad() ;
if (hesse) {
m.hesse() ;
}
if (minos) {
if (minosSet) {
m.minos(*minosSet) ;
} else {
m.minos() ;
}
}
if (doSave) {
ret = m.save() ;
}
}
delete nll ;
return ret ;
}
RooFitResult* RooAbsPdf::fitTo(RooAbsData& data, Option_t *fitOpt, Option_t *optOpt, const char* fitRange)
{
return fitTo(data,RooArgSet(),fitOpt,optOpt,fitRange) ;
}
RooFitResult* RooAbsPdf::fitTo(RooAbsData& data, const RooArgSet& projDeps, Option_t *fitOpt, Option_t *optOpt, const char* fitRange)
{
TString fopt(fitOpt) ;
TString oopt(optOpt) ;
fopt.ToLower() ;
oopt.ToLower() ;
Bool_t extended = fopt.Contains("e") ;
Bool_t cOpt = oopt.Contains("p") ||
oopt.Contains("c") ;
Bool_t blindfit = fopt.Contains("b") ;
Int_t ncpu = 1 ;
if (oopt.Contains("2")) ncpu=2 ;
if (oopt.Contains("3")) ncpu=3 ;
if (oopt.Contains("4")) ncpu=4 ;
if (oopt.Contains("5")) ncpu=5 ;
if (oopt.Contains("6")) ncpu=6 ;
if (oopt.Contains("7")) ncpu=7 ;
if (oopt.Contains("8")) ncpu=8 ;
if (oopt.Contains("9")) ncpu=9 ;
RooAbsReal::enableEvalErrorLogging(kTRUE) ;
RooNLLVar nll("nll","-log(likelihood)",*this,data,projDeps,extended,fitRange,0,ncpu) ;
RooAbsReal::enableEvalErrorLogging(kFALSE) ;
RooMinuit m(nll) ;
if(blindfit)
m.setPrintLevel(-1);
if (cOpt) m.optimizeConst(1) ;
return m.fit(fopt) ;
}
void RooAbsPdf::printValue(ostream& os) const
{
if (_norm) {
os << evaluate() << "/" << _norm->getVal() ;
} else {
os << evaluate() ;
}
}
void RooAbsPdf::printMultiline(ostream& os, Int_t contents, Bool_t verbose, TString indent) const
{
RooAbsReal::printMultiline(os,contents,verbose,indent);
os << indent << "--- RooAbsPdf ---" << endl;
os << indent << "Cached value = " << _value << endl ;
if (_norm) {
os << indent << " Normalization integral: " << endl ;
TString moreIndent(indent) ; moreIndent.Append(" ") ;
_norm->printStream(os,kName|kAddress|kTitle|kValue|kArgs,kSingleLine,moreIndent.Data()) ;
}
}
RooAbsGenContext* RooAbsPdf::genContext(const RooArgSet &vars, const RooDataSet *prototype,
const RooArgSet* auxProto, Bool_t verbose) const
{
return new RooGenContext(*this,vars,prototype,auxProto,verbose) ;
}
RooDataSet *RooAbsPdf::generate(const RooArgSet& whatVars, Int_t nEvents, const RooCmdArg& arg1,
const RooCmdArg& arg2, const RooCmdArg& arg3,const RooCmdArg& arg4, const RooCmdArg& arg5)
{
return generate(whatVars,RooFit::NumEvents(nEvents),arg1,arg2,arg3,arg4,arg5) ;
}
RooDataSet *RooAbsPdf::generate(const RooArgSet& whatVars, const RooCmdArg& arg1,const RooCmdArg& arg2,
const RooCmdArg& arg3,const RooCmdArg& arg4, const RooCmdArg& arg5,const RooCmdArg& arg6)
{
RooCmdConfig pc(Form("RooAbsPdf::generate(%s)",GetName())) ;
pc.defineObject("proto","PrototypeData",0,0) ;
pc.defineString("dsetName","Name",0,"") ;
pc.defineInt("randProto","PrototypeData",0,0) ;
pc.defineInt("resampleProto","PrototypeData",1,0) ;
pc.defineInt("verbose","Verbose",0,0) ;
pc.defineInt("extended","Extended",0,0) ;
pc.defineInt("nEvents","NumEvents",0,0) ;
pc.process(arg1,arg2,arg3,arg4,arg5,arg6) ;
if (!pc.ok(kTRUE)) {
return 0 ;
}
RooDataSet* protoData = static_cast<RooDataSet*>(pc.getObject("proto",0)) ;
const char* dsetName = pc.getString("dsetName") ;
Int_t nEvents = pc.getInt("nEvents") ;
Bool_t verbose = pc.getInt("verbose") ;
Bool_t randProto = pc.getInt("randProto") ;
Bool_t resampleProto = pc.getInt("resampleProto") ;
Bool_t extended = pc.getInt("extended") ;
if (extended) {
nEvents = RooRandom::randomGenerator()->Poisson(nEvents==0?expectedEvents(&whatVars):nEvents) ;
cxcoutI(Generation) << " Extended mode active, number of events generated (" << nEvents << ") is Poisson fluctuation on "
<< GetName() << "::expectedEvents() = " << nEvents << endl ;
if (nEvents==0) {
return 0 ;
}
} else if (nEvents==0) {
cxcoutI(Generation) << "No number of events specified , number of events generated is "
<< GetName() << "::expectedEvents() = " << expectedEvents(&whatVars)<< endl ;
}
if (extended && protoData && !randProto) {
cxcoutI(Generation) << "WARNING Using generator option Extended() (Poisson distribution of #events) together "
<< "with a prototype dataset implies incomplete sampling or oversampling of proto data. "
<< "Set randomize flag in ProtoData() option to randomize prototype dataset order and thus "
<< "to randomize the set of over/undersampled prototype events for each generation cycle." << endl ;
}
RooDataSet* data ;
if (protoData) {
data = generate(whatVars,*protoData,nEvents,verbose,randProto,resampleProto) ;
} else {
data = generate(whatVars,nEvents,verbose) ;
}
if (dsetName && strlen(dsetName)>0) {
data->SetName(dsetName) ;
}
return data ;
}
RooDataSet *RooAbsPdf::generate(const RooArgSet &whatVars, Int_t nEvents, Bool_t verbose) const
{
RooDataSet *generated = 0;
RooAbsGenContext *context= genContext(whatVars,0,0,verbose);
if(0 != context && context->isValid()) {
generated= context->generate(nEvents);
}
else {
coutE(Generation) << "RooAbsPdf::generate(" << GetName() << ") cannot create a valid context" << endl;
}
if(0 != context) delete context;
return generated;
}
RooDataSet *RooAbsPdf::generate(const RooArgSet &whatVars, const RooDataSet &prototype,
Int_t nEvents, Bool_t verbose, Bool_t randProtoOrder, Bool_t resampleProto) const
{
RooDataSet *generated = 0;
RooAbsGenContext *context= genContext(whatVars,&prototype,0,verbose);
if (resampleProto) {
randProtoOrder=kTRUE ;
}
if (randProtoOrder && prototype.numEntries()!=nEvents) {
coutI(Generation) << "RooAbsPdf::generate (Re)randomizing event order in prototype dataset (Nevt=" << nEvents << ")" << endl ;
Int_t* newOrder = randomizeProtoOrder(prototype.numEntries(),nEvents,resampleProto) ;
context->setProtoDataOrder(newOrder) ;
delete[] newOrder ;
}
if(0 != context && context->isValid()) {
generated= context->generate(nEvents);
}
else {
coutE(Generation) << "RooAbsPdf::generate(" << GetName() << ") cannot create a valid context" << endl;
}
if(0 != context) delete context;
return generated;
}
Int_t* RooAbsPdf::randomizeProtoOrder(Int_t nProto, Int_t, Bool_t resampleProto) const
{
RooLinkedList l ;
Int_t i ;
for (i=0 ; i<nProto ; i++) {
l.Add(new RooInt(i)) ;
}
Int_t* lut = new Int_t[nProto] ;
if (!resampleProto) {
for (i=0 ; i<nProto ; i++) {
Int_t iran = RooRandom::integer(nProto-i) ;
RooInt* sample = (RooInt*) l.At(iran) ;
lut[i] = *sample ;
l.Remove(sample) ;
delete sample ;
}
} else {
for (i=0 ; i<nProto ; i++) {
lut[i] = RooRandom::integer(nProto);
}
}
return lut ;
}
Int_t RooAbsPdf::getGenerator(const RooArgSet &, RooArgSet &, Bool_t ) const
{
return 0 ;
}
void RooAbsPdf::initGenerator(Int_t )
{
}
void RooAbsPdf::generateEvent(Int_t )
{
}
Bool_t RooAbsPdf::isDirectGenSafe(const RooAbsArg& arg) const
{
if (!findServer(arg.GetName())) return kFALSE ;
TIterator* sIter = serverIterator() ;
const RooAbsArg *server = 0;
while((server=(const RooAbsArg*)sIter->Next())) {
if(server == &arg) continue;
if(server->dependsOn(arg)) {
delete sIter ;
return kFALSE ;
}
}
delete sIter ;
return kTRUE ;
}
RooPlot* RooAbsPdf::plotOn(RooPlot* frame, RooLinkedList& cmdList) const
{
RooCmdArg* plotRange(0) ;
RooCmdArg* normRange(0) ;
if (getStringAttribute("fitrange") && !cmdList.FindObject("Range") &&
!cmdList.FindObject("RangeWithName")) {
plotRange = (RooCmdArg*) RooFit::Range(getStringAttribute("fitrange")).Clone() ;
cmdList.Add(plotRange) ;
}
if (getStringAttribute("fitrange") && !cmdList.FindObject("NormRange")) {
normRange = (RooCmdArg*) RooFit::NormRange(getStringAttribute("fitrange")).Clone() ;
cmdList.Add(normRange) ;
}
if (plotRange || normRange) {
coutI(Plotting) << "RooAbsPdf::plotOn(" << GetName() << ") p.d.f was fitted in range and no explicit "
<< (plotRange?"plot":"") << ((plotRange&&normRange)?",":"")
<< (normRange?"norm":"") << " range was specified, using fit range as default" << endl ;
}
if (plotSanityChecks(frame)) return frame ;
RooCmdConfig pc(Form("RooAbsPdf::plotOn(%s)",GetName())) ;
pc.defineDouble("scaleFactor","Normalization",0,1.0) ;
pc.defineInt("scaleType","Normalization",0,RooAbsPdf::Relative) ;
pc.defineObject("compSet","SelectCompSet",0) ;
pc.defineString("compSpec","SelectCompSpec",0) ;
pc.defineObject("asymCat","Asymmetry",0) ;
pc.defineDouble("rangeLo","Range",0,-999.) ;
pc.defineDouble("rangeHi","Range",1,-999.) ;
pc.defineString("rangeName","RangeWithName",0,"") ;
pc.defineString("normRangeName","NormRange",0,"") ;
pc.defineInt("rangeAdjustNorm","Range",0,0) ;
pc.defineInt("rangeWNAdjustNorm","RangeWithName",0,0) ;
pc.defineMutex("SelectCompSet","SelectCompSpec") ;
pc.defineMutex("Range","RangeWithName") ;
pc.allowUndefined() ;
pc.process(cmdList) ;
if (!pc.ok(kTRUE)) {
return frame ;
}
ScaleType stype = (ScaleType) pc.getInt("scaleType") ;
Double_t scaleFactor = pc.getDouble("scaleFactor") ;
const RooAbsCategoryLValue* asymCat = (const RooAbsCategoryLValue*) pc.getObject("asymCat") ;
const char* compSpec = pc.getString("compSpec") ;
const RooArgSet* compSet = (const RooArgSet*) pc.getObject("compSet") ;
Bool_t haveCompSel = (strlen(compSpec)>0 || compSet) ;
TString nameSuffix ;
if (compSpec && strlen(compSpec)>0) {
nameSuffix.Append("_Comp[") ;
nameSuffix.Append(compSpec) ;
nameSuffix.Append("]") ;
} else if (compSet) {
nameSuffix.Append("_Comp[") ;
nameSuffix.Append(compSet->contentsString().c_str()) ;
nameSuffix.Append("]") ;
}
pc.stripCmdList(cmdList,"SelectCompSet,SelectCompSpec") ;
if (asymCat) {
RooCmdArg cnsuffix("CurveNameSuffix",0,0,0,0,nameSuffix.Data(),0,0,0) ;
cmdList.Add(&cnsuffix);
return RooAbsReal::plotOn(frame,cmdList) ;
}
Double_t nExpected(1) ;
if (stype==RelativeExpected) {
if (!canBeExtended()) {
coutE(Plotting) << "RooAbsPdf::plotOn(" << GetName()
<< "): ERROR the 'Expected' scale option can only be used on extendable PDFs" << endl ;
return frame ;
}
nExpected = expectedEvents(frame->getNormVars()) ;
}
if (stype != Raw) {
if (frame->getFitRangeNEvt() && stype==Relative) {
Bool_t hasCustomRange(kFALSE), adjustNorm(kFALSE) ;
list<pair<Double_t,Double_t> > rangeLim ;
if (pc.hasProcessed("Range")) {
Double_t rangeLo = pc.getDouble("rangeLo") ;
Double_t rangeHi = pc.getDouble("rangeHi") ;
rangeLim.push_back(make_pair(rangeLo,rangeHi)) ;
adjustNorm = pc.getInt("rangeAdjustNorm") ;
hasCustomRange = kTRUE ;
coutI(Plotting) << "RooAbsPdf::plotOn(" << GetName() << ") only plotting range ["
<< rangeLo << "," << rangeHi << "]" ;
if (!pc.hasProcessed("NormRange")) {
ccoutI(Plotting) << ", curve is normalized to data in " << (adjustNorm?"given":"full") << " given range" << endl ;
} else {
ccoutI(Plotting) << endl ;
}
nameSuffix.Append(Form("_Range[%d_%d]",rangeLo,rangeHi)) ;
} else if (pc.hasProcessed("RangeWithName")) {
char tmp[1024] ;
strcpy(tmp,pc.getString("rangeName",0,kTRUE)) ;
char* rangeNameToken = strtok(tmp,",") ;
while(rangeNameToken) {
Double_t rangeLo = frame->getPlotVar()->getMin(rangeNameToken) ;
Double_t rangeHi = frame->getPlotVar()->getMax(rangeNameToken) ;
rangeLim.push_back(make_pair(rangeLo,rangeHi)) ;
rangeNameToken = strtok(0,",") ;
}
adjustNorm = pc.getInt("rangeWNAdjustNorm") ;
hasCustomRange = kTRUE ;
coutI(Plotting) << "RooAbsPdf::plotOn(" << GetName() << ") only plotting range '" << pc.getString("rangeName",0,kTRUE) << "'" ;
if (!pc.hasProcessed("NormRange")) {
ccoutI(Plotting) << ", curve is normalized to data in " << (adjustNorm?"given":"full") << " given range" << endl ;
} else {
ccoutI(Plotting) << endl ;
}
nameSuffix.Append(Form("_Range[%s]",pc.getString("rangeName"))) ;
}
if (pc.hasProcessed("NormRange")) {
char tmp[1024] ;
strcpy(tmp,pc.getString("normRangeName",0,kTRUE)) ;
char* rangeNameToken = strtok(tmp,",") ;
rangeLim.clear() ;
while(rangeNameToken) {
Double_t rangeLo = frame->getPlotVar()->getMin(rangeNameToken) ;
Double_t rangeHi = frame->getPlotVar()->getMax(rangeNameToken) ;
rangeLim.push_back(make_pair(rangeLo,rangeHi)) ;
rangeNameToken = strtok(0,",") ;
}
adjustNorm = kTRUE ;
hasCustomRange = kTRUE ;
coutI(Plotting) << "RooAbsPdf::plotOn(" << GetName() << ") p.d.f. curve is normalized using explicit choice of ranges '" << pc.getString("normRangeName",0,kTRUE) << "'" << endl ;
nameSuffix.Append(Form("_NormRange[%s]",pc.getString("rangeName"))) ;
}
if (hasCustomRange && adjustNorm) {
Double_t rangeNevt(0) ;
list<pair<Double_t,Double_t> >::iterator riter = rangeLim.begin() ;
for (;riter!=rangeLim.end() ; ++riter) {
Double_t nevt= frame->getFitRangeNEvt(riter->first,riter->second) ;
rangeNevt += nevt ;
}
scaleFactor *= rangeNevt/nExpected ;
} else {
scaleFactor *= frame->getFitRangeNEvt()/nExpected ;
}
} else if (stype==RelativeExpected) {
scaleFactor *= nExpected ;
} else if (stype==NumEvent) {
scaleFactor /= nExpected ;
}
scaleFactor *= frame->getFitRangeBinW() ;
}
frame->updateNormVars(*frame->getPlotVar()) ;
RooCmdArg tmp = RooFit::Normalization(scaleFactor,Raw) ;
cmdList.Add(&tmp) ;
if (haveCompSel) {
RooArgSet branchNodeSet ;
branchNodeServerList(&branchNodeSet) ;
TIterator* iter = branchNodeSet.createIterator() ;
RooAbsArg* arg ;
while((arg=(RooAbsArg*)iter->Next())) {
if (!dynamic_cast<RooAbsReal*>(arg)) {
branchNodeSet.remove(*arg) ;
}
}
delete iter ;
RooArgSet* dirSelNodes ;
if (compSet) {
dirSelNodes = (RooArgSet*) branchNodeSet.selectCommon(*compSet) ;
} else {
dirSelNodes = (RooArgSet*) branchNodeSet.selectByName(compSpec) ;
}
coutI(Plotting) << "RooAbsPdf::plotOn(" << GetName() << ") directly selected PDF components: " << *dirSelNodes << endl ;
plotOnCompSelect(dirSelNodes) ;
delete dirSelNodes ;
}
RooCmdArg cnsuffix("CurveNameSuffix",0,0,0,0,nameSuffix.Data(),0,0,0) ;
cmdList.Add(&cnsuffix);
RooPlot* ret = RooAbsReal::plotOn(frame,cmdList) ;
if (haveCompSel) plotOnCompSelect(0) ;
if (plotRange) {
delete plotRange ;
}
if (normRange) {
delete normRange ;
}
return ret ;
}
void RooAbsPdf::plotOnCompSelect(RooArgSet* selNodes) const
{
RooArgSet branchNodeSet ;
branchNodeServerList(&branchNodeSet) ;
TIterator* iter = branchNodeSet.createIterator() ;
RooAbsArg* arg ;
while((arg=(RooAbsArg*)iter->Next())) {
if (!dynamic_cast<RooAbsReal*>(arg)) {
branchNodeSet.remove(*arg) ;
}
}
if (!selNodes) {
iter->Reset() ;
while((arg=(RooAbsArg*)iter->Next())) {
((RooAbsReal*)arg)->selectComp(kTRUE) ;
}
delete iter ;
return ;
}
iter->Reset() ;
TIterator* sIter = selNodes->createIterator() ;
RooArgSet tmp ;
while((arg=(RooAbsArg*)iter->Next())) {
sIter->Reset() ;
RooAbsArg* selNode ;
while((selNode=(RooAbsArg*)sIter->Next())) {
if (selNode->dependsOn(*arg)) {
tmp.add(*arg,kTRUE) ;
}
}
}
delete sIter ;
iter->Reset() ;
while((arg=(RooAbsArg*)iter->Next())) {
if (arg->dependsOn(*selNodes)) {
tmp.add(*arg,kTRUE) ;
}
}
tmp.remove(*selNodes,kTRUE) ;
tmp.remove(*this) ;
selNodes->add(tmp) ;
coutI(Plotting) << "RooAbsPdf::plotOn(" << GetName() << ") indirectly selected PDF components: " << tmp << endl ;
iter->Reset() ;
while((arg=(RooAbsArg*)iter->Next())) {
Bool_t select = selNodes->find(arg->GetName()) ? kTRUE : kFALSE ;
((RooAbsReal*)arg)->selectComp(select) ;
}
delete iter ;
}
RooPlot* RooAbsPdf::plotOn(RooPlot *frame, PlotOpt o) const
{
if (plotSanityChecks(frame)) return frame ;
Double_t nExpected(1) ;
if (o.stype==RelativeExpected) {
if (!canBeExtended()) {
coutE(Plotting) << "RooAbsPdf::plotOn(" << GetName()
<< "): ERROR the 'Expected' scale option can only be used on extendable PDFs" << endl ;
return frame ;
}
nExpected = expectedEvents(frame->getNormVars()) ;
}
if (o.stype != Raw) {
if (frame->getFitRangeNEvt() && o.stype==Relative) {
o.scaleFactor *= frame->getFitRangeNEvt()/nExpected ;
} else if (o.stype==RelativeExpected) {
o.scaleFactor *= nExpected ;
} else if (o.stype==NumEvent) {
o.scaleFactor /= nExpected ;
}
o.scaleFactor *= frame->getFitRangeBinW() ;
}
frame->updateNormVars(*frame->getPlotVar()) ;
return RooAbsReal::plotOn(frame,o) ;
}
RooPlot* RooAbsPdf::paramOn(RooPlot* frame, const RooCmdArg& arg1, const RooCmdArg& arg2,
const RooCmdArg& arg3, const RooCmdArg& arg4, const RooCmdArg& arg5,
const RooCmdArg& arg6, const RooCmdArg& arg7, const RooCmdArg& arg8)
{
RooLinkedList cmdList;
cmdList.Add(const_cast<RooCmdArg*>(&arg1)) ; cmdList.Add(const_cast<RooCmdArg*>(&arg2)) ;
cmdList.Add(const_cast<RooCmdArg*>(&arg3)) ; cmdList.Add(const_cast<RooCmdArg*>(&arg4)) ;
cmdList.Add(const_cast<RooCmdArg*>(&arg5)) ; cmdList.Add(const_cast<RooCmdArg*>(&arg6)) ;
cmdList.Add(const_cast<RooCmdArg*>(&arg7)) ; cmdList.Add(const_cast<RooCmdArg*>(&arg8)) ;
RooCmdConfig pc(Form("RooAbsPdf::paramOn(%s)",GetName())) ;
pc.defineString("label","Label",0,"") ;
pc.defineDouble("xmin","Layout",0,0.50) ;
pc.defineDouble("xmax","Layout",1,0.99) ;
pc.defineInt("ymaxi","Layout",0,Int_t(0.95*10000)) ;
pc.defineInt("showc","ShowConstants",0,0) ;
pc.defineObject("params","Parameters",0,0) ;
pc.defineString("formatStr","Format",0,"NELU") ;
pc.defineInt("sigDigit","Format",0,2) ;
pc.defineInt("dummy","FormatArgs",0,0) ;
pc.defineMutex("Format","FormatArgs") ;
pc.process(cmdList) ;
if (!pc.ok(kTRUE)) {
return frame ;
}
const char* label = pc.getString("label") ;
Double_t xmin = pc.getDouble("xmin") ;
Double_t xmax = pc.getDouble("xmax") ;
Double_t ymax = pc.getInt("ymaxi") / 10000. ;
Int_t showc = pc.getInt("showc") ;
const char* formatStr = pc.getString("formatStr") ;
Int_t sigDigit = pc.getInt("sigDigit") ;
RooArgSet* params = static_cast<RooArgSet*>(pc.getObject("params")) ;
if (!params) {
params = getParameters(frame->getNormVars()) ;
if (pc.hasProcessed("FormatArgs")) {
const RooCmdArg* formatCmd = static_cast<RooCmdArg*>(cmdList.FindObject("FormatArgs")) ;
paramOn(frame,*params,showc,label,0,0,xmin,xmax,ymax,formatCmd) ;
} else {
paramOn(frame,*params,showc,label,sigDigit,formatStr,xmin,xmax,ymax) ;
}
delete params ;
} else {
RooArgSet* pdfParams = getParameters(frame->getNormVars()) ;
RooArgSet* selParams = static_cast<RooArgSet*>(pdfParams->selectCommon(*params)) ;
if (pc.hasProcessed("FormatArgs")) {
const RooCmdArg* formatCmd = static_cast<RooCmdArg*>(cmdList.FindObject("FormatArgs")) ;
paramOn(frame,*selParams,showc,label,0,0,xmin,xmax,ymax,formatCmd) ;
} else {
paramOn(frame,*selParams,showc,label,sigDigit,formatStr,xmin,xmax,ymax) ;
}
delete selParams ;
delete pdfParams ;
}
return frame ;
}
RooPlot* RooAbsPdf::paramOn(RooPlot* frame, const RooAbsData* data, const char *label,
Int_t sigDigits, Option_t *options, Double_t xmin,
Double_t xmax ,Double_t ymax)
{
RooArgSet* params = getParameters(data) ;
TString opts(options) ;
paramOn(frame,*params,opts.Contains("c"),label,sigDigits,options,xmin,xmax,ymax) ;
delete params ;
return frame ;
}
RooPlot* RooAbsPdf::paramOn(RooPlot* frame, const RooArgSet& params, Bool_t showConstants, const char *label,
Int_t sigDigits, Option_t *options, Double_t xmin,
Double_t xmax ,Double_t ymax, const RooCmdArg* formatCmd)
{
TString opts = options;
opts.ToLower();
Bool_t showLabel= (label != 0 && strlen(label) > 0);
TIterator* pIter = params.createIterator() ;
Int_t nPar= params.getSize();
Double_t ymin(ymax), dy(0.06);
Int_t index(nPar);
RooRealVar *var = 0;
while((var=(RooRealVar*)pIter->Next())) {
if(showConstants || !var->isConstant()) ymin-= dy;
}
if(showLabel) ymin-= dy;
TPaveText *box= new TPaveText(xmin,ymax,xmax,ymin,"BRNDC");
box->SetName(Form("%s_paramBox",GetName())) ;
if(!box) return 0;
box->SetFillColor(0);
box->SetBorderSize(1);
box->SetTextAlign(12);
box->SetTextSize(0.04F);
box->SetFillStyle(1001);
box->SetFillColor(0);
TText *text = 0;
index= nPar;
pIter->Reset() ;
while((var=(RooRealVar*)pIter->Next())) {
if(var->isConstant() && !showConstants) continue;
TString *formatted= options ? var->format(sigDigits, options) : var->format(*formatCmd) ;
text= box->AddText(formatted->Data());
delete formatted;
}
if(showLabel) text= box->AddText(label);
frame->addObject(box) ;
delete pIter ;
return frame ;
}
Double_t RooAbsPdf::expectedEvents(const RooArgSet*) const
{
return 0 ;
}
void RooAbsPdf::verboseEval(Int_t stat)
{
_verboseEval = stat ;
}
Int_t RooAbsPdf::verboseEval()
{
return _verboseEval ;
}
void RooAbsPdf::CacheElem::operModeHook(RooAbsArg::OperMode)
{
}
RooAbsPdf::CacheElem::~CacheElem()
{
if (_owner) {
RooAbsPdf* pdfOwner = static_cast<RooAbsPdf*>(_owner) ;
if (pdfOwner->_norm == _norm) {
pdfOwner->_norm = 0 ;
}
}
delete _norm ;
}
RooAbsPdf* RooAbsPdf::createProjection(const RooArgSet& iset)
{
TString name(GetName()) ;
name.Append("_Proj[") ;
if (iset.getSize()>0) {
TIterator* iter = iset.createIterator() ;
RooAbsArg* arg ;
Bool_t first(kTRUE) ;
while((arg=(RooAbsArg*)iter->Next())) {
if (first) {
first=kFALSE ;
} else {
name.Append(",") ;
}
name.Append(arg->GetName()) ;
}
delete iter ;
}
name.Append("]") ;
return new RooProjectedPdf(name.Data(),name.Data(),*this,iset) ;
}
RooAbsReal* RooAbsPdf::createCdf(const RooArgSet& iset, const RooArgSet& nset)
{
return createCdf(iset,RooFit::SupNormSet(nset)) ;
}
RooAbsReal* RooAbsPdf::createCdf(const RooArgSet& iset, const RooCmdArg arg1, const RooCmdArg arg2,
const RooCmdArg arg3, const RooCmdArg arg4, const RooCmdArg arg5,
const RooCmdArg arg6, const RooCmdArg arg7, const RooCmdArg arg8)
{
RooCmdConfig pc(Form("RooAbsReal::createCdf(%s)",GetName())) ;
pc.defineObject("supNormSet","SupNormSet",0,0) ;
pc.defineInt("numScanBins","ScanParameters",0,1000) ;
pc.defineInt("intOrder","ScanParameters",1,2) ;
pc.defineInt("doScanNum","ScanNumCdf",0,1) ;
pc.defineInt("doScanAll","ScanAllCdf",0,0) ;
pc.defineInt("doScanNon","ScanNoCdf",0,0) ;
pc.defineMutex("ScanNumCdf","ScanAllCdf","ScanNoCdf") ;
pc.process(arg1,arg2,arg3,arg4,arg5,arg6,arg7,arg8) ;
if (!pc.ok(kTRUE)) {
return 0 ;
}
const RooArgSet* snset = static_cast<const RooArgSet*>(pc.getObject("supNormSet",0)) ;
RooArgSet nset ;
if (snset) {
nset.add(*snset) ;
}
Int_t numScanBins = pc.getInt("numScanBins") ;
Int_t intOrder = pc.getInt("intOrder") ;
Int_t doScanNum = pc.getInt("doScanNum") ;
Int_t doScanAll = pc.getInt("doScanAll") ;
Int_t doScanNon = pc.getInt("doScanNon") ;
if (doScanNon) {
return createIntRI(iset,nset) ;
}
if (doScanAll) {
return createScanCdf(iset,nset,numScanBins,intOrder) ;
}
if (doScanNum) {
RooRealIntegral* tmp = (RooRealIntegral*) createIntegral(iset) ;
Int_t isNum= (tmp->numIntRealVars().getSize()>0) ;
delete tmp ;
if (isNum) {
coutI(NumIntegration) << "RooAbsPdf::createCdf(" << GetName() << ") integration over observable(s) " << iset << " involves numeric integration," << endl
<< " constructing cdf though numeric integration of sampled pdf in " << numScanBins << " bins and applying order "
<< intOrder << " interpolation on integrated histogram." << endl
<< " To override this choice of technique use argument ScanNone(), to change scan parameters use ScanParameters(nbins,order) argument" << endl ;
}
return isNum ? createScanCdf(iset,nset,numScanBins,intOrder) : createIntRI(iset,nset) ;
}
return 0 ;
}
RooAbsReal* RooAbsPdf::createScanCdf(const RooArgSet& iset, const RooArgSet& nset, Int_t numScanBins, Int_t intOrder)
{
string name = string(GetName()) + "_NUMCDF_" + integralNameSuffix(iset,&nset).Data() ;
RooRealVar* ivar = (RooRealVar*) iset.first() ;
ivar->setBins(numScanBins,"numcdf") ;
RooNumCdf* ret = new RooNumCdf(name.c_str(),name.c_str(),*this,*ivar,"numcdf") ;
ret->setInterpolationOrder(intOrder) ;
return ret ;
}
RooArgSet* RooAbsPdf::getAllConstraints(const RooArgSet& observables, const RooArgSet& constrainedParams) const
{
RooArgSet* ret = new RooArgSet("AllConstraints") ;
RooArgSet* comps = getComponents() ;
TIterator* iter = comps->createIterator() ;
RooAbsArg *arg ;
while((arg=(RooAbsArg*)iter->Next())) {
RooAbsPdf* pdf = dynamic_cast<RooAbsPdf*>(arg) ;
if (pdf) {
RooArgSet* compRet = pdf->getConstraints(observables,constrainedParams) ;
if (compRet) {
ret->add(*compRet,kFALSE) ;
delete compRet ;
}
}
}
delete iter ;
delete comps ;
return ret ;
}
void RooAbsPdf::clearEvalError()
{
_evalError = kFALSE ;
}
Bool_t RooAbsPdf::evalError()
{
return _evalError ;
}
void RooAbsPdf::raiseEvalError()
{
_evalError = kTRUE ;
}
Last change: Mon Dec 15 13:02:46 2008
Last generated: 2008-12-15 13:02
This page has been automatically generated. If you have any comments or suggestions about the page layout send a mail to ROOT support, or contact the developers with any questions or problems regarding ROOT.