New classes plus some renamed for a better user friendly interface
[u/mrichter/AliRoot.git] / PWG2 / RESONANCES / AliRsnSimpleAnalysis.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notice   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15
16 //-------------------------------------------------------------------------
17 //                     Class AliRsnSimpleAnalysis
18 //-------------------------------------------------------------------------
19 // This class is a manager to process one or more pair analysis with a
20 // given event sample, specified as a TTree passed to this object.
21 // Each single pair analysis must be defined with its specifications, like
22 // histogram binning, cuts and everything else.
23 // This class utiliy consists in defining a unique event sample which is
24 // used for all pairs analysis, and all kinds of event mixing (if required).
25 // All histograms computed in a single execution, are then saved into a file.
26 // This object contains a two TObjArray's:
27 //  - one to contain single event pair analysis objects (signal, like-sign)
28 //  - one to contain all event-mixing pair analysis objects
29 // When a new pair is added, it must be then specified in what container it
30 // must be placed, in order to avoid meaningless results.
31 //
32 // author: A. Pulvirenti
33 // email : alberto.pulvirenti@ct.infn.it
34 //-------------------------------------------------------------------------
35
36 #include <TH1.h>
37 #include <TH2.h>
38 #include <TTree.h>
39 #include <TFile.h>
40 #include <TArray.h>
41 #include <TClonesArray.h>
42
43 #include "AliLog.h"
44 #include "AliRsnPID.h"
45 #include "AliRsnDaughter.h"
46 #include "AliRsnEvent.h"
47 #include "AliRsnEventBuffer.h"
48 #include "AliRsnSimpleFunction.h"
49 #include "AliRsnSimpleAnalyzer.h"
50 #include "AliRsnSimpleAnalysis.h"
51
52 ClassImp(AliRsnSimpleAnalysis)
53
54 //_____________________________________________________________________________
55 AliRsnSimpleAnalysis::AliRsnSimpleAnalysis(AliRsnSimpleAnalyzer *ana, AliRsnPID *pid) :
56   TObject(),
57   fInitialized(kFALSE),
58   fStep(1000),
59   fTree(0x0),
60   fPID(pid),
61   fAnalyzer(ana)
62 {
63 //
64 // Constructor
65 // Initializes all pointers and collections to NULL.
66 //
67
68     strcpy(fFileName, "default.root");
69 }
70
71
72 //_____________________________________________________________________________
73 void AliRsnSimpleAnalysis::Clear(Option_t* /*option*/)
74 {
75 //
76 // Clear heap
77 //
78 }
79
80 //_____________________________________________________________________________
81 void AliRsnSimpleAnalysis::SetEventsTree(TTree *tree)
82 {
83 //
84 // Set the tree containing the events to be processed.
85 // Counts also the number of events and stores it in a private datamember.
86 // This can avoid the time-wasting entries count in a long TChain.
87 //
88     fTree = tree;
89     AliInfo(Form("Total number of events to be processed: %d", fTree->GetEntries()));
90 }
91
92 //_____________________________________________________________________________
93 Bool_t AliRsnSimpleAnalysis::Initialize()
94 {
95 //
96 // Various initialization processes
97 //
98     // check process objects
99     if (!fAnalyzer) {
100         AliError("Analyzer not initialized");
101         return kFALSE;
102     }
103
104     // initialize analyzer
105     fAnalyzer->Init();
106
107     // at the end, update flag for initialization
108     fInitialized = kTRUE;
109
110     return kTRUE;
111 }
112
113 //_____________________________________________________________________________
114 Stat_t AliRsnSimpleAnalysis::Process()
115 {
116 //
117 // Computes all invariant mass distributions defined in the AliRsnPair objects collected.
118 // Depending on the kind of background evaluation method, computes also this one.
119 //
120
121     // check initialization
122     if (!fInitialized) {
123         AliError("Analysis not initialized. Use method 'Initialize()'");
124         return 0.0;
125     }
126
127     // set cursor object
128     AliRsnEvent *event = 0x0;
129     fTree->SetBranchAddress("rsnEvents", &event);
130
131     // output counter
132     Stat_t nPairs = 0.0;
133
134         // loop on events
135         Int_t i, nEvents = (Int_t)fTree->GetEntries();
136     for (i = 0; i < nEvents; i++) {
137         // message
138         if ((i % fStep) == 0) AliInfo(Form("Processing event %d", i));
139         // get entry
140         fTree->GetEntry(i);
141         if (!event) continue;
142         fPID->Process(event);
143         fAnalyzer->Process(event);
144     }
145
146     return nPairs;
147 }
148
149 //_____________________________________________________________________________
150 void AliRsnSimpleAnalysis::SaveOutput() const
151 {
152 //
153 // Writes histograms in current directory
154 //
155     TFile *file = TFile::Open(fFileName, "RECREATE");
156     AliRsnSimpleFunction *pair = 0;
157     TH1D *h1D = 0;
158     TH2D *h2D = 0;
159     TObjArrayIter pairIterator(fAnalyzer->GetSingle());
160     while ( (pair = (AliRsnSimpleFunction*)pairIterator.Next()) ) {
161         h1D = pair->GetHistogram1D();
162         h2D = pair->GetHistogram2D();
163         if (h1D) h1D->Write();
164         if (h2D) h2D->Write();
165     }
166     if (fAnalyzer->GetMix() && !fAnalyzer->GetMix()->IsEmpty()) {
167         TObjArrayIter mixIterator(fAnalyzer->GetMix());
168         while ( (pair = (AliRsnSimpleFunction*)mixIterator.Next()) ) {
169             h1D = pair->GetHistogram1D();
170             h2D = pair->GetHistogram2D();
171             if (h1D) h1D->Write();
172             if (h2D) h2D->Write();
173         }
174         }
175         file->Close();
176 }