Logo ROOT  
Reference Guide
RooStudyPackage.cxx
Go to the documentation of this file.
1/*****************************************************************************
2 * Project: RooFit *
3 * Package: RooFitCore *
4 * @(#)root/roofitcore:$Id$
5 * Authors: *
6 * WV, Wouter Verkerke, UC Santa Barbara, verkerke@slac.stanford.edu *
7 * DK, David Kirkby, UC Irvine, dkirkby@uci.edu *
8 * *
9 * Copyright (c) 2000-2005, Regents of the University of California *
10 * and Stanford University. All rights reserved. *
11 * *
12 * Redistribution and use in source and binary forms, *
13 * with or without modification, are permitted according to the terms *
14 * listed in LICENSE (http://roofit.sourceforge.net/license.txt) *
15 *****************************************************************************/
16
17/**
18\file RooStudyPackage.cxx
19\class RooStudyPackage
20\ingroup Roofitcore
21
22RooStudyPackage is a utility class to manage studies that consist of
23repeated applications of generate-and-fit operations on a workspace
24
25**/
26
27
28
29#include "Riostream.h"
30
31#include "RooStudyPackage.h"
32#include "RooWorkspace.h"
33#include "RooAbsStudy.h"
34#include "RooDataSet.h"
35#include "RooMsgService.h"
36#include "TFile.h"
37#include "TRandom2.h"
38#include "RooRandom.h"
39#include "TMath.h"
40#include "TEnv.h"
41
42using namespace std ;
43
45 ;
46
47
48
49////////////////////////////////////////////////////////////////////////////////
50
52{
53}
54
55
56
57////////////////////////////////////////////////////////////////////////////////
58
60{
61}
62
63
64
65////////////////////////////////////////////////////////////////////////////////
66
67RooStudyPackage::RooStudyPackage(const RooStudyPackage& other) : TNamed(other), _ws(new RooWorkspace(*other._ws))
68{
69 list<RooAbsStudy*>::const_iterator iter = other._studies.begin() ;
70 for (;iter!=other._studies.end() ; ++iter) {
71 _studies.push_back((*iter)->clone()) ;
72 }
73}
74
75
76
77////////////////////////////////////////////////////////////////////////////////
78
80{
81 _studies.push_back(&study) ;
82}
83
84
85
86////////////////////////////////////////////////////////////////////////////////
87
89{
90 initialize() ;
91 run(nExperiments) ;
92 finalize() ;
93}
94
95
96
97////////////////////////////////////////////////////////////////////////////////
98/// Make iterator over copy of studies attached to workspace
99
101{
102 for (list<RooAbsStudy*>::iterator iter=_studies.begin() ; iter!=_studies.end() ; ++iter) {
103 (*iter)->attach(*_ws) ;
104 (*iter)->initialize() ;
105 }
106
107}
108
109
110////////////////////////////////////////////////////////////////////////////////
111
112void RooStudyPackage::run(Int_t nExperiments)
113{
114 // Run the requested number of experiments
115 Int_t prescale = nExperiments>100 ? Int_t(nExperiments/100) : 1 ;
116 for (Int_t i=0 ; i<nExperiments ; i++) {
117 if (i%prescale==0) {
118 coutP(Generation) << "RooStudyPackage::run(" << GetName() << ") processing experiment " << i << "/" << nExperiments << endl ;
119 }
120 runOne() ;
121 }
122}
123
124
125
126////////////////////////////////////////////////////////////////////////////////
127
129{
130 for (list<RooAbsStudy*>::iterator iter=_studies.begin() ; iter!=_studies.end() ; ++iter) {
131 (*iter)->execute() ;
132 }
133}
134
135
136
137
138////////////////////////////////////////////////////////////////////////////////
139/// Finalize all studies
140
142{
143 for (list<RooAbsStudy*>::iterator iter=_studies.begin() ; iter!=_studies.end() ; ++iter) {
144 (*iter)->finalize() ;
145 }
146}
147
148
149
150
151////////////////////////////////////////////////////////////////////////////////
152
154{
155 for (list<RooAbsStudy*>::iterator iter=_studies.begin() ; iter!=_studies.end() ; ++iter) {
156
157 (*iter)->finalize() ;
158
159 RooDataSet* summaryData = (*iter)->summaryData() ;
160 if (summaryData) {
161 summaryData->SetName(Form("%s_%d",summaryData->GetName(),seqno)) ;
162 cout << "registering summary dataset: " ; summaryData->Print() ;
163 olist->Add(summaryData) ;
164 }
165
166 RooLinkedList* detailedData = (*iter)->detailedData() ;
167 if (detailedData && detailedData->GetSize()>0) {
168
169 detailedData->SetName(Form("%s_%d",detailedData->GetName(),seqno)) ;
170 cout << "registering detailed dataset " << detailedData->IsA()->GetName() << "::"
171 << detailedData->GetName() << " with " << detailedData->GetSize() << " elements" << endl ;
172 TIterator* diter = detailedData->MakeIterator() ;
173 TNamed* dobj ;
174 while((dobj=(TNamed*)diter->Next())) {
175 dobj->SetName(Form("%s_%d",dobj->GetName(),seqno)) ;
176 }
177 delete diter ;
178 olist->Add(detailedData) ;
179 (*iter)->releaseDetailData() ;
180 }
181 }
182}
183
184
185
186////////////////////////////////////////////////////////////////////////////////
187/// Choose random seed for this process
188/// in case pass a definite seed to have it deterministic
189/// use also worker number
190
192{
193 TRandom2 random(0);
194 //gRandom->SetSeed(0) ;
195 Int_t seed = random.Integer(TMath::Limits<Int_t>::Max()) ;
196
197 // get worker number
198 TString worknumber = gEnv->GetValue("ProofServ.Ordinal","undef");
199 int iworker = -1;
200 if (worknumber != "undef")
201 iworker = int( worknumber.Atof()*10 + 0.1);
202
203 if (iworker >= 0) {
204 for (int i = 0; i <= iworker; ++i )
205 seed = random.Integer( TMath::Limits<Int_t>::Max() );
206 }
207
209 gRandom->SetSeed(seed) ;
210
211 return seed ;
212}
213
214
215
216////////////////////////////////////////////////////////////////////////////////
217/// Read in study package
218
219void RooStudyPackage::processFile(const char* studyName, Int_t nexp)
220{
221 string name_fin = Form("study_data_%s.root",studyName) ;
222 TFile fin(name_fin.c_str()) ;
223 RooStudyPackage* pkg = dynamic_cast<RooStudyPackage*>(fin.Get("studypack")) ;
224 if (!pkg) {
225 cout << "RooStudyPackage::processFile() ERROR input file " << name_fin << " does not contain a RooStudyPackage named 'studypack'" << endl ;
226 return ;
227 }
228
229 // Initialize random seed
230 Int_t seqno = pkg->initRandom() ;
231 cout << "RooStudyPackage::processFile() Initial random seed for this run is " << seqno << endl ;
232
233 // Run study
234 pkg->driver(nexp) ;
235
236 // Save result
237 TList res ;
238 pkg->exportData(&res,seqno) ;
239 TFile fout(Form("study_result_%s_%d.root",studyName,seqno),"RECREATE") ;
240 res.Write() ;
241 fout.Close() ;
242}
#define coutP(a)
Definition: RooMsgService.h:31
int Int_t
Definition: RtypesCore.h:45
#define ClassImp(name)
Definition: Rtypes.h:375
R__EXTERN TEnv * gEnv
Definition: TEnv.h:170
R__EXTERN TRandom * gRandom
Definition: TRandom.h:62
char * Form(const char *fmt,...)
Formats a string in a circular formatting buffer.
Definition: TString.cxx:2447
void Print(Option_t *options=0) const override
Print TNamed name and title.
Definition: RooAbsData.h:237
RooAbsStudy is an abstract base class for RooStudyManager modules.
Definition: RooAbsStudy.h:33
RooDataSet is a container class to hold unbinned data.
Definition: RooDataSet.h:55
void SetName(const char *name) override
Change the name of this dataset into the given name.
RooLinkedList is an collection class for internal use, storing a collection of RooAbsArg pointers in ...
Definition: RooLinkedList.h:38
Int_t GetSize() const
Definition: RooLinkedList.h:63
void SetName(const char *name)
Definition: RooLinkedList.h:99
const char * GetName() const override
Returns name of object.
Definition: RooLinkedList.h:98
TClass * IsA() const override
TIterator * MakeIterator(Bool_t forward=kTRUE) const
Create a TIterator for this list.
static TRandom * randomGenerator()
Return a pointer to a singleton random-number generator implementation.
Definition: RooRandom.cxx:51
RooStudyPackage is a utility class to manage studies that consist of repeated applications of generat...
void driver(Int_t nExperiments)
Int_t initRandom()
Choose random seed for this process in case pass a definite seed to have it deterministic use also wo...
std::list< RooAbsStudy * > _studies
static void processFile(const char *infile, Int_t nexp)
Read in study package.
void addStudy(RooAbsStudy &study)
void finalize()
Finalize all studies.
void initialize()
Make iterator over copy of studies attached to workspace.
void exportData(TList *olist, Int_t seqno)
RooWorkspace * _ws
void run(Int_t nExperiments)
The RooWorkspace is a persistable container for RooFit projects.
Definition: RooWorkspace.h:43
Int_t Write(const char *name=nullptr, Int_t option=0, Int_t bufsize=0) override
Write all objects in this collection.
TObject * Get(const char *namecycle) override
Return pointer to object identified by namecycle.
virtual Int_t GetValue(const char *name, Int_t dflt) const
Returns the integer value for a resource.
Definition: TEnv.cxx:491
A ROOT file is a suite of consecutive data records (TKey instances) with a well defined format.
Definition: TFile.h:54
void Close(Option_t *option="") override
Close a file.
Definition: TFile.cxx:891
Iterator abstract base class.
Definition: TIterator.h:30
virtual TObject * Next()=0
A doubly linked list.
Definition: TList.h:44
void Add(TObject *obj) override
Definition: TList.h:87
The TNamed class is the base class for all named ROOT classes.
Definition: TNamed.h:29
const char * GetName() const override
Returns name of object.
Definition: TNamed.h:47
virtual void SetName(const char *name)
Set the name of the TNamed.
Definition: TNamed.cxx:140
Random number generator class based on the maximally quidistributed combined Tausworthe generator by ...
Definition: TRandom2.h:27
virtual void SetSeed(ULong_t seed=0)
Set the random generator seed.
Definition: TRandom.cxx:608
virtual UInt_t Integer(UInt_t imax)
Returns a random integer uniformly distributed on the interval [ 0, imax-1 ].
Definition: TRandom.cxx:360
Basic string class.
Definition: TString.h:136
Double_t Atof() const
Return floating-point value contained in string.
Definition: TString.cxx:2012
@ Generation
Definition: RooGlobalFunc.h:63