#include "RooFit.h"
#include "RooAbsGenContext.h"
#include "RooAbsGenContext.h"
#include "RooAbsPdf.h"
#include "RooDataSet.h"
ClassImp(RooAbsGenContext)
;
RooAbsGenContext::RooAbsGenContext(const RooAbsPdf& model, const RooArgSet &vars,
const RooDataSet *prototype, const RooArgSet* auxProto, Bool_t verbose) :
TNamed(model),
_prototype(prototype),
_theEvent(0),
_isValid(kTRUE),
_verbose(verbose),
_protoOrder(0)
{
if (model.recursiveCheckObservables(&vars)) {
cout << "RooAbsGenContext::ctor: Error in PDF dependents" << endl ;
_isValid = kFALSE ;
return ;
}
_theEvent= (RooArgSet*)vars.snapshot(kFALSE);
_nextProtoIndex= 0;
if(0 != _prototype) {
TIterator *protoIterator= _prototype->get()->createIterator();
const RooAbsArg *proto = 0;
while((proto= (const RooAbsArg*)protoIterator->Next())) {
if(!_theEvent->contains(*proto)) {
_protoVars.add(*proto);
_theEvent->addClone(*proto);
}
}
delete protoIterator;
}
if (auxProto) {
_protoVars.add(*auxProto) ;
_theEvent->addClone(*auxProto) ;
}
_extendMode = model.extendMode() ;
if (model.canBeExtended()) {
_expectedEvents= (Int_t)(model.expectedEvents(_theEvent) + 0.5);
} else {
_expectedEvents= 0 ;
}
}
RooAbsGenContext::~RooAbsGenContext()
{
if(0 != _theEvent) delete _theEvent;
if (_protoOrder) delete[] _protoOrder ;
}
RooDataSet *RooAbsGenContext::generate(Int_t nEvents) {
if(!isValid()) {
cout << ClassName() << "::" << GetName() << ": context is not valid" << endl;
return 0;
}
if(nEvents <= 0) {
if(_prototype) {
nEvents= (Int_t)_prototype->numEntries();
}
else {
if (_extendMode == RooAbsPdf::CanNotBeExtended) {
cout << ClassName() << "::" << GetName()
<< ":generate: PDF not extendable: cannot calculate expected number of events" << endl;
return 0;
}
nEvents= _expectedEvents;
}
if(nEvents <= 0) {
cout << ClassName() << "::" << GetName()
<< ":generate: cannot calculate expected number of events" << endl;
return 0;
}
else if(_verbose) {
cout << ClassName() << "::" << GetName() << ":generate: will generate "
<< nEvents << " events" << endl;
}
}
if(_prototype) {
const RooArgSet *vars= _prototype->get();
TIterator *iterator= _protoVars.createIterator();
const RooAbsArg *arg = 0;
Bool_t ok(kTRUE);
while((arg= (const RooAbsArg*)iterator->Next())) {
if(vars->contains(*arg)) continue;
cout << ClassName() << "::" << GetName() << ":generate: prototype dataset is missing \""
<< arg->GetName() << "\"" << endl;
}
delete iterator;
if(!ok) return 0;
}
if (_verbose) Print("v") ;
TString name(GetName()),title(GetTitle());
name.Append("Data");
title.Prepend("Generated From ");
RooDataSet *data= new RooDataSet(name.Data(), title.Data(), *_theEvent);
initGenerator(*_theEvent);
for(Int_t evt= 0; evt < nEvents; evt++) {
if(0 != _prototype) {
if(_nextProtoIndex >= _prototype->numEntries()) _nextProtoIndex= 0;
Int_t actualProtoIdx = _protoOrder ? _protoOrder[_nextProtoIndex] : _nextProtoIndex ;
const RooArgSet *subEvent= _prototype->get(actualProtoIdx);
_nextProtoIndex++;
if(0 != subEvent) {
*_theEvent= *subEvent;
}
else {
cout << ClassName() << "::" << GetName() << ":generate: cannot load event "
<< actualProtoIdx << " from prototype dataset" << endl;
return 0;
}
}
generateEvent(*_theEvent, nEvents - evt);
data->add(*_theEvent);
}
return data;
}
void RooAbsGenContext::initGenerator(const RooArgSet&) {
}
void RooAbsGenContext::printToStream(ostream &os, PrintOption opt, TString indent) const
{
oneLinePrint(os,*this);
if(opt >= Standard) {
PrintOption less= lessVerbose(opt);
TString deeper(indent);
indent.Append(" ");
os << indent << " Generator of ";
_theEvent->printToStream(os,less,deeper);
os << indent << " Prototype variables are ";
_protoVars.printToStream(os,less,deeper);
}
}
void RooAbsGenContext::setProtoDataOrder(Int_t* lut)
{
if (_protoOrder) {
delete[] _protoOrder ;
_protoOrder = 0 ;
}
if (lut && _prototype) {
Int_t n = _prototype->numEntries() ;
_protoOrder = new Int_t[n] ;
Int_t i ;
for (i=0 ; i<n ; i++) {
_protoOrder[i] = lut[i] ;
}
}
}
ROOT page - Class index - Class Hierarchy - Top of the page
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.