]>
Commit | Line | Data |
---|---|---|
d88f97cc | 1 | #ifndef ALIINVMASS_H |
2 | #define ALIINVMASS_H | |
3da30618 | 3 | /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * |
4 | * See cxx source for full Copyright notice */ | |
5 | ||
6 | /* $Id$ */ | |
7 | ||
d88f97cc | 8 | //////////////////////////////////////////////////////////////////////////////// |
9 | // Class AliInvmass | |
10 | // Construction of invariant mass and combinatorial background. | |
11 | // | |
12 | // Example : | |
13 | // --------- | |
14 | // | |
15 | // TObjArray* photons=new TObjArray(); // Array with photon tracks for pi0 rec. | |
16 | // | |
17 | // // Code to create some photon tracks from pi0 decays | |
18 | // Int_t ntracks=200; | |
19 | // for (Int_t i=0; i<ntracks; i++) | |
20 | // { | |
21 | // photons->Add(new Alitrack); | |
22 | // ... | |
23 | // ... | |
24 | // ... | |
25 | // } | |
26 | // | |
27 | // // Perform the invariant mass and comb. bkg. reconstruction | |
28 | // | |
29 | // TObjArray* allm=q.Invmass(photons,photons); // All reconstructed invariant masses | |
30 | // | |
31 | // TH1F* hall=new TH1F("hall","hall",200,0,2); // Histo with M_inv of all combinations | |
32 | // | |
33 | // Int_t nall=0; | |
34 | // if (allm) nall=allm->GetEntries(); | |
35 | // | |
36 | // AliTrack* t; | |
37 | // Float_t minv; | |
38 | // for (Int_t j=0; j<nall; j++) | |
39 | // { | |
40 | // t=(AliTrack*)allm->At(j); | |
41 | // if (t) | |
42 | // { | |
43 | // minv=t->GetMass(); | |
44 | // hall->Fill(minv); | |
45 | // } | |
46 | // } | |
47 | // | |
48 | // TObjArray* bkgm=q.CombBkg(photons,photons); // Reconstructed comb. background | |
49 | // | |
50 | // TH1F* hbkg=new TH1F("hbkg","hbkg",200,0,2); // Histo with M_inv. of comb. background | |
51 | // | |
52 | // Int_t nbkg=0; | |
53 | // if (bkgm) nbkg=bkgm->GetEntries(); | |
54 | // | |
55 | // for (Int_t j=0; j<nbkg; j++) | |
56 | // { | |
57 | // t=(AliTrack*)bkgm->At(j); | |
58 | // if (t) | |
59 | // { | |
60 | // minv=t->GetMass(); | |
61 | // hbkg->Fill(minv); | |
62 | // } | |
63 | // } | |
64 | // | |
65 | // TH1F* hsig=new TH1F("sig","sig",200,0,2); // Histo with the bkg. subtracted signal | |
66 | // hsig->Sumw2(); | |
67 | // hsig->Add(hall,hbkg,1,-1); | |
68 | // | |
69 | // | |
70 | // Note : By default the storage of the reconstructed information is performed | |
71 | // in separate TObjArrays for the signal and comb. background resp. | |
72 | // In order to limit the memory usage, AliInvmass::SetStorageMode(1) may be | |
73 | // used to activate only a single TObjArray to store the reconstructed information. | |
74 | // Consequently, the following statements | |
75 | // | |
76 | // TObjArray* allm=q.Invmass(photons,photons); | |
77 | // TObjArray* bkgm=q.CombBkg(photons,photons); | |
78 | // | |
79 | // will result in the fact that after he invokation of CombBkg | |
80 | // the information of "allm" is lost due to the fact that the storage is | |
81 | // is re-used for "bkgm" in case the "single storage" option has been selected. | |
82 | // Usage of the, in that case invalid, pointer "allm" may cause your | |
83 | // program to crash. | |
84 | // | |
85 | // * Thus : In case of single storage usage, all invokations of the returned | |
86 | // array pointer have to be completed before invoking any memberfunction | |
87 | // of the same AliInvmass object again. | |
88 | // | |
89 | // | |
90 | // | |
91 | //--- NvE 12-apr-1999 UU-SAP Utrecht | |
92 | //////////////////////////////////////////////////////////////////////////////// | |
93 | ||
94 | #include <iostream.h> | |
95 | #include <math.h> | |
96 | ||
97 | #include "TObject.h" | |
98 | #include "TObjArray.h" | |
99 | ||
100 | #include "AliRandom.h" | |
101 | #include "AliTrack.h" | |
102 | ||
103 | class AliInvmass : public TObject | |
104 | { | |
105 | public: | |
106 | AliInvmass(); // Default constructor | |
107 | ~AliInvmass(); // Destructor | |
108 | void SetStorageMode(Int_t m); // Set storage mode (1=single, 2=multiple) | |
109 | void SetThetaSwitch(Int_t i=1); // Enable (1/0) new theta for comb. bkg. reco. | |
110 | void SetPhiSwitch(Int_t i=1); // Enable (1/0) new phi for comb. bkg. reco. | |
111 | Int_t GetStorageMode(); // Provide storage mode | |
112 | Int_t GetThetaSwitch(); // Provide theta switch flag | |
113 | Int_t GetPhiSwitch(); // Provide phi switch flag | |
114 | TObjArray* Invmass(TObjArray* a1,TObjArray* a2); // Two-particle inv. mass reco. | |
115 | TObjArray* CombBkg(TObjArray* a1,TObjArray* a2); // Two-particle comb. background reco. | |
116 | ||
117 | protected: | |
118 | Double_t fPi; // Value of pi | |
119 | Int_t fMode; // Storage mode for signal and bkg. results (2=separate arrays) | |
120 | Int_t fBkg; // Flag to denote comb. background processing | |
121 | AliRandom fRndm; // The random number generator for the comb. bkg. reconstruction | |
122 | Int_t fNewtheta; // Flag to denote enabling of switching theta for comb. bkg. reco. | |
123 | Int_t fNewphi; // Flag to denote enabling of switching phi for comb. bkg. reco. | |
124 | TObjArray* fMinv; // Array with reconstructed invariant mass 'tracks' | |
125 | TObjArray* fMbkg; // Array with reconstructed comb. background 'tracks' | |
126 | ||
127 | private: | |
128 | void Combine(TObjArray* a1,TObjArray* a2); // Make two-particle combinations | |
129 | ||
130 | ClassDef(AliInvmass,1) // Class definition to enable ROOT I/O | |
131 | }; | |
132 | #endif |