#include "RooFit.h"
#include "Riostream.h"
#include "Riostream.h"
#include <fstream>
#include "TTree.h"
#include "TH2.h"
#include "RooDataSet.h"
#include "RooPlot.h"
#include "RooAbsReal.h"
#include "Roo1DTable.h"
#include "RooCategory.h"
#include "RooFormulaVar.h"
#include "RooArgList.h"
#include "RooAbsRealLValue.h"
#include "RooRealVar.h"
#include "RooDataHist.h"
#include "TROOT.h"
#if (__GNUC__==3&&__GNUC_MINOR__==2&&__GNUC_PATCHLEVEL__==3)
char* operator+( streampos&, char* );
#endif
ClassImp(RooDataSet)
;
RooDataSet::RooDataSet() {}
RooDataSet::RooDataSet(const char *name, const char *title, const RooArgSet& vars, const char* wgtVarName) :
RooTreeData(name,title,vars)
{
appendToDir(this,kTRUE) ;
initialize(wgtVarName) ;
}
RooDataSet::RooDataSet(const char *name, const char *title, RooDataSet *dset,
const RooArgSet& vars, const char *cuts, const char* wgtVarName) :
RooTreeData(name,title,dset,vars,cuts)
{
appendToDir(this,kTRUE) ;
if (wgtVarName) {
initialize(wgtVarName) ;
} else {
if (dset->_wgtVar && vars.find(dset->_wgtVar->GetName())) {
initialize(dset->_wgtVar->GetName()) ;
} else {
initialize(0) ;
}
}
}
RooDataSet::RooDataSet(const char *name, const char *title, RooDataSet *t,
const RooArgSet& vars, const RooFormulaVar& cutVar, const char* wgtVarName) :
RooTreeData(name,title,t,vars,cutVar)
{
appendToDir(this,kTRUE) ;
if (wgtVarName) {
initialize(wgtVarName) ;
} else {
if (t->_wgtVar && vars.find(t->_wgtVar->GetName())) {
initialize(t->_wgtVar->GetName()) ;
} else {
initialize(0) ;
}
}
}
RooDataSet::RooDataSet(const char *name, const char *title, TTree *t,
const RooArgSet& vars, const RooFormulaVar& cutVar, const char* wgtVarName) :
RooTreeData(name,title,t,vars,cutVar)
{
appendToDir(this,kTRUE) ;
initialize(wgtVarName) ;
}
RooDataSet::RooDataSet(const char *name, const char *title, TTree *ntuple,
const RooArgSet& vars, const char *cuts, const char* wgtVarName) :
RooTreeData(name,title,ntuple,vars,cuts)
{
appendToDir(this,kTRUE) ;
initialize(wgtVarName) ;
}
RooDataSet::RooDataSet(const char *name, const char *filename, const char *treename,
const RooArgSet& vars, const char *cuts, const char* wgtVarName) :
RooTreeData(name,filename,treename,vars,cuts)
{
appendToDir(this,kTRUE) ;
initialize(wgtVarName) ;
}
RooDataSet::RooDataSet(RooDataSet const & other, const char* newname) :
RooTreeData(other,newname), RooDirItem()
{
appendToDir(this,kTRUE) ;
initialize(other._wgtVar?other._wgtVar->GetName():0) ;
}
RooArgSet RooDataSet::addWgtVar(const RooArgSet& origVars, const RooAbsArg* wgtVar)
{
RooArgSet tmp(origVars) ;
if (wgtVar) tmp.add(*wgtVar) ;
return tmp ;
}
RooDataSet::RooDataSet(const char *name, const char *title, RooDataSet *ntuple,
const RooArgSet& vars, const RooFormulaVar* cutVar, const char* cutRange,
Int_t nStart, Int_t nStop, Bool_t copyCache) :
RooTreeData(name,title,ntuple,addWgtVar(vars,ntuple->_wgtVar),cutVar,cutRange,nStart,nStop,copyCache)
{
appendToDir(this,kTRUE) ;
initialize(ntuple->_wgtVar?ntuple->_wgtVar->GetName():0) ;
}
RooAbsData* RooDataSet::cacheClone(const RooArgSet* newCacheVars, const char* newName)
{
RooDataSet* dset = new RooDataSet(newName?newName:GetName(),GetTitle(),this,_vars,(RooFormulaVar*)0,0,0,2000000000,kTRUE) ;
if (_wgtVar) dset->setWeightVar(_wgtVar->GetName()) ;
RooArgSet* selCacheVars = (RooArgSet*) newCacheVars->selectCommon(dset->_cachedVars) ;
dset->initCache(*selCacheVars) ;
delete selCacheVars ;
return dset ;
}
RooAbsData* RooDataSet::emptyClone(const char* newName, const char* newTitle, const RooArgSet* vars) const
{
RooDataSet* dset = new RooDataSet(newName?newName:GetName(),newTitle?newTitle:GetTitle(),vars?*vars:*get()) ;
if (_wgtVar) dset->setWeightVar(_wgtVar->GetName()) ;
return dset ;
}
void RooDataSet::initialize(const char* wgtVarName)
{
_varsNoWgt.removeAll() ;
_varsNoWgt.add(_vars) ;
_wgtVar = 0 ;
if (wgtVarName) {
RooAbsArg* wgt = _varsNoWgt.find(wgtVarName) ;
if (!wgt) {
cout << "RooDataSet::RooDataSet(" << GetName() << ") WARNING: designated weight variable "
<< wgtVarName << " not found in set of variables, no weighting will be assigned" << endl ;
} else if (!dynamic_cast<RooRealVar*>(wgt)) {
cout << "RooDataSet::RooDataSet(" << GetName() << ") WARNING: designated weight variable "
<< wgtVarName << " is not of type RooRealVar, no weighting will be assigned" << endl ;
} else {
_varsNoWgt.remove(*wgt) ;
_wgtVar = (RooRealVar*) wgt ;
}
}
}
RooAbsData* RooDataSet::reduceEng(const RooArgSet& varSubset, const RooFormulaVar* cutVar, const char* cutRange,
Int_t nStart, Int_t nStop, Bool_t copyCache)
{
checkInit() ;
return new RooDataSet(GetName(), GetTitle(), this, varSubset, cutVar, cutRange, nStart, nStop, copyCache) ;
}
RooDataSet::~RooDataSet()
{
removeFromDir(this) ;
}
RooDataHist* RooDataSet::binnedClone(const char* newName, const char* newTitle) const
{
TString title, name ;
if (newName) {
name = newName ;
} else {
name = Form("%s_binned",GetName()) ;
}
if (newTitle) {
title = newTitle ;
} else {
title = Form("%s_binned",GetTitle()) ;
}
return new RooDataHist(name,title,*get(),*this) ;
}
void RooDataSet::setWeightVar(const char* name)
{
_varsNoWgt.removeAll() ;
initialize(name) ;
}
Double_t RooDataSet::weight() const
{
return _wgtVar ? _wgtVar->getVal() : 1. ;
}
const RooArgSet* RooDataSet::get(Int_t index) const
{
const RooArgSet* ret = RooTreeData::get(index) ;
return ret ? &_varsNoWgt : 0 ;
}
Int_t RooDataSet::numEntries(Bool_t useWeights) const
{
if (!useWeights || !_wgtVar) return (Int_t) GetEntries() ;
return (Int_t)sumEntries() ;
}
Double_t RooDataSet::sumEntries(const char* cutSpec, const char* cutRange) const
{
RooFormula* select = 0 ;
if (cutSpec) {
select = new RooFormula("select",cutSpec,*get()) ;
}
Double_t sumw(0) ;
Int_t i ;
for (i=0 ; i<GetEntries() ; i++) {
get(i) ;
if (select && select->eval()==0.) continue ;
if (cutRange && !_vars.allInRange(cutRange)) continue ;
sumw += weight() ;
}
if (select) delete select ;
return sumw ;
}
const RooArgSet* RooDataSet::get() const
{
return &_varsNoWgt ;
}
void RooDataSet::add(const RooArgSet& data, Double_t weight)
{
checkInit() ;
_varsNoWgt = data;
if (_wgtVar) _wgtVar->setVal(weight) ;
Fill();
}
Bool_t RooDataSet::merge(RooDataSet* data1, RooDataSet* data2, RooDataSet* data3,
RooDataSet* data4, RooDataSet* data5, RooDataSet* data6)
{
TList dsetList ;
dsetList.Add(data1) ;
if (data2) {
dsetList.Add(data2) ;
if (data3) {
dsetList.Add(data3) ;
if (data4) {
dsetList.Add(data4) ;
if (data5) {
dsetList.Add(data5) ;
if (data6) {
dsetList.Add(data6) ;
}
}
}
}
}
return merge(dsetList) ;
}
Bool_t RooDataSet::merge(const TList& dsetList)
{
checkInit() ;
TIterator* iter = dsetList.MakeIterator() ;
RooDataSet* data ;
while((data=(RooDataSet*)iter->Next())) {
if (numEntries()!=data->numEntries()) {
cout << "RooDataSet::merge(" << GetName() << " ERROR: datasets have different size" << endl ;
delete iter ;
return kTRUE ;
}
}
RooTreeData* cloneData = (RooTreeData*) Clone() ;
iter->Reset() ;
while((data=(RooDataSet*)iter->Next())) {
data->_iterator->Reset() ;
RooAbsArg* arg ;
while ((arg=(RooAbsArg*)data->_iterator->Next())) {
RooAbsArg* clone = _vars.addClone(*arg,kTRUE) ;
if (clone) clone->attachToTree(*_tree,_defTreeBufSize) ;
}
}
Reset() ;
for (int i=0 ; i<cloneData->numEntries() ; i++) {
_vars = *cloneData->get(i) ;
iter->Reset() ;
while((data=(RooDataSet*)iter->Next())) {
_vars = *data->get(i) ;
}
Fill() ;
}
delete cloneData ;
delete iter ;
initialize(_wgtVar?_wgtVar->GetName():0) ;
return kFALSE ;
}
void RooDataSet::append(RooTreeData& data) {
checkInit() ;
loadValues(data._tree,(RooFormulaVar*)0) ;
}
RooAbsArg* RooDataSet::addColumn(RooAbsArg& var)
{
RooAbsArg* ret = RooTreeData::addColumn(var) ;
initialize(_wgtVar?_wgtVar->GetName():0) ;
return ret ;
}
RooArgSet* RooDataSet::addColumns(const RooArgList& varList)
{
RooArgSet* ret = RooTreeData::addColumns(varList) ;
initialize(_wgtVar?_wgtVar->GetName():0) ;
return ret ;
}
TH2F* RooDataSet::createHistogram(const RooAbsRealLValue& var1, const RooAbsRealLValue& var2, const char* cuts, const char *name) const
{
return createHistogram(var1, var2, var1.getBins(), var2.getBins(), cuts, name);
}
TH2F* RooDataSet::createHistogram(const RooAbsRealLValue& var1, const RooAbsRealLValue& var2, Int_t nx, Int_t ny, const char* cuts, const char *name) const
{
static Int_t counter(0) ;
Bool_t ownPlotVarX(kFALSE) ;
RooAbsReal* plotVarX= (RooAbsReal*)_vars.find(var1.GetName());
if(0 == plotVarX) {
if (!var1.dependsOn(_vars)) {
cout << GetName() << "::createHistogram: Argument " << var1.GetName()
<< " is not in dataset and is also not dependent on data set" << endl ;
return 0 ;
}
plotVarX = (RooAbsReal*) var1.Clone() ;
ownPlotVarX = kTRUE ;
plotVarX->redirectServers(const_cast<RooArgSet&>(_vars)) ;
}
Bool_t ownPlotVarY(kFALSE) ;
RooAbsReal* plotVarY= (RooAbsReal*)_vars.find(var2.GetName());
if(0 == plotVarY) {
if (!var2.dependsOn(_vars)) {
cout << GetName() << "::createHistogram: Argument " << var2.GetName()
<< " is not in dataset and is also not dependent on data set" << endl ;
return 0 ;
}
plotVarY = (RooAbsReal*) var2.Clone() ;
ownPlotVarY = kTRUE ;
plotVarY->redirectServers(const_cast<RooArgSet&>(_vars)) ;
}
RooFormula* select = 0;
if(0 != cuts && strlen(cuts)) {
select=new RooFormula(cuts,cuts,_vars);
if (!select || !select->ok()) {
delete select;
return 0 ;
}
}
TString histName(name);
histName.Prepend("_");
histName.Prepend(fName);
histName.Append("_") ;
histName.Append(Form("%08x",counter++)) ;
TH2F* histogram=new TH2F(histName.Data(), "Events", nx, var1.getMin(), var1.getMax(),
ny, var2.getMin(), var2.getMax());
if(!histogram) {
cout << fName << "::createHistogram: unable to create a new histogram" << endl;
return 0;
}
Int_t nevent= (Int_t)_tree->GetEntries();
for(Int_t i=0; i < nevent; ++i)
{
Int_t entryNumber=_tree->GetEntryNumber(i);
if (entryNumber<0) break;
get(entryNumber);
if (select && select->eval()==0) continue ;
histogram->Fill(plotVarX->getVal(), plotVarY->getVal()) ;
}
if (ownPlotVarX) delete plotVarX ;
if (ownPlotVarY) delete plotVarY ;
if (select) delete select ;
return histogram ;
}
RooDataSet *RooDataSet::read(const char *fileList, const RooArgList &varList,
const char *verbOpt, const char* commonPath,
const char* indexCatName) {
RooArgList variables(varList) ;
Bool_t ownIsBlind(kTRUE) ;
RooAbsArg* blindState = variables.find("blindState") ;
if (!blindState) {
blindState = new RooCategory("blindState","Blinding State") ;
variables.add(*blindState) ;
} else {
ownIsBlind = kFALSE ;
if (blindState->IsA()!=RooCategory::Class()) {
cout << "RooDataSet::read: ERROR: variable list already contains"
<< "a non-RooCategory blindState member" << endl ;
return 0 ;
}
cout << "RooDataSet::read: WARNING: recycling existing "
<< "blindState category in variable list" << endl ;
}
RooCategory* blindCat = (RooCategory*) blindState ;
blindCat->setAttribute("Dynamic") ;
blindCat->defineType("Normal",0) ;
blindCat->defineType("Blind",1) ;
TString opts= verbOpt;
opts.ToLower();
Bool_t verbose= !opts.Contains("q");
Bool_t debug= opts.Contains("d");
Bool_t haveRefBlindString(false) ;
RooDataSet *data= new RooDataSet("dataset", fileList, variables);
if (ownIsBlind) { variables.remove(*blindState) ; delete blindState ; }
if(!data) {
cout << "RooDataSet::read: unable to create a new dataset"
<< endl;
return 0;
}
blindCat = (RooCategory*) data->_vars.find("blindState") ;
RooCategory *indexCat = 0;
if (indexCatName) {
RooAbsArg* tmp = 0;
tmp = data->_vars.find(indexCatName) ;
if (!tmp) {
cout << "RooDataSet::read: no index category named "
<< indexCatName << " in supplied variable list" << endl ;
return 0 ;
}
if (tmp->IsA()!=RooCategory::Class()) {
cout << "RooDataSet::read: variable " << indexCatName
<< " is not a RooCategory" << endl ;
return 0 ;
}
indexCat = (RooCategory*)tmp ;
indexCat->setAttribute("Dynamic") ;
}
Int_t outOfRange(0) ;
char fileList2[10240] ;
strcpy(fileList2,fileList) ;
char *filename = strtok(fileList2,", ") ;
Int_t fileSeqNum(0) ;
while (filename) {
if (indexCat) {
char *catname = strchr(filename,':') ;
if (catname) {
*catname=0 ;
catname++ ;
const RooCatType* type = indexCat->lookupType(catname,kFALSE) ;
if (type) {
indexCat->setIndex(type->getVal()) ;
} else {
indexCat->defineType(catname,fileSeqNum) ;
indexCat->setIndex(fileSeqNum) ;
}
} else {
char newLabel[128] ;
sprintf(newLabel,"file%03d",fileSeqNum) ;
if (indexCat->defineType(newLabel,fileSeqNum)) {
cout << "RooDataSet::read: Error, cannot register automatic type name " << newLabel
<< " in index category " << indexCat->GetName() << endl ;
return 0 ;
}
indexCat->setIndex(fileSeqNum) ;
}
}
cout << "RooDataSet::read: reading file " << filename << endl ;
TString fullName(commonPath) ;
fullName.Append(filename) ;
ifstream file(fullName) ;
if(!file.good()) {
cout << "RooDataSet::read: unable to open '"
<< filename << "', skipping" << endl;
}
Int_t line(0) ;
Bool_t haveBlindString(false) ;
while(file.good() && !file.eof()) {
line++;
if(debug) cout << "reading line " << line << endl;
if (file.peek() == '#')
{
if(debug) cout << "skipping comment on line " << line << endl;
TString line ;
line.ReadLine(file) ;
if (line.Contains("#BLIND#")) {
haveBlindString = true ;
if (haveRefBlindString) {
TString curBlindString(line(7,line.Length()-7)) ;
if (debug) cout << "Found blind string " << curBlindString << endl ;
if (curBlindString != data->_blindString) {
cout << "RooDataSet::read: ERROR blinding string mismatch, abort" << endl ;
return 0 ;
}
} else {
data->_blindString=TString(line(7,line.Length()-7)) ;
if (debug) cout << "Storing ref blind string " << data->_blindString << endl ;
haveRefBlindString = true ;
}
}
}
else {
Bool_t readError = variables.readFromStream(file,kTRUE,verbose) ;
data->_vars = variables ;
if(file.eof()) break ;
if(!file.good()) {
cout << "RooDataSet::read(static): read error at line " << line << endl ;
break;
}
if (readError) {
outOfRange++ ;
continue ;
}
blindCat->setIndex(haveBlindString) ;
data->Fill();
}
}
file.close();
filename = strtok(0," ,") ;
fileSeqNum++ ;
}
if (indexCat) {
RooCategory* origIndexCat = (RooCategory*) variables.find(indexCatName) ;
TIterator* tIter = indexCat->typeIterator() ;
RooCatType* type = 0;
while ((type=(RooCatType*)tIter->Next())) {
origIndexCat->defineType(type->GetName(),type->getVal()) ;
}
}
cout << "RooDataSet::read: read " << data->GetEntries()
<< " events (ignored " << outOfRange << " out of range events)" << endl;
return data;
}
Bool_t RooDataSet::write(const char* filename)
{
checkInit() ;
ofstream ofs(filename) ;
if (ofs.fail()) {
cout << "RooDataSet::write(" << GetName() << ") cannot create file " << filename << endl ;
return kTRUE ;
}
cout << "RooDataSet::write(" << GetName() << ") writing ASCII file " << filename << endl ;
Int_t i ;
for (i=0 ; i<numEntries() ; i++) {
RooArgList list(*get(i),"line") ;
list.writeToStream(ofs,kTRUE) ;
}
if (ofs.fail()) {
cout << "RooDataSet::write(" << GetName() << "): WARNING error(s) have occured in writing" << endl ;
}
return ofs.fail() ;
}
void RooDataSet::printToStream(ostream& os, PrintOption opt, TString indent) const {
RooTreeData::printToStream(os,opt,indent) ;
if (opt>=Shape && _wgtVar) {
os << indent << " Dataset variable \"" << _wgtVar->GetName() << "\" is interpreted as the event weight" << endl ;
}
}
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.