c3be81bf9922e21d8f8f13186a62dd1a6ba33ba9
[u/mrichter/AliRoot.git] / RALICE / AliInvmass.h
1 #ifndef ALIINVMASS_H
2 #define ALIINVMASS_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 // $Id$
7
8 #include <math.h>
9  
10 #include "TObject.h"
11 #include "TObjArray.h"
12
13 #include "AliRandom.h"
14 #include "AliTrack.h"
15
16 class AliInvmass : public TObject
17 {
18  public:
19   AliInvmass();                                    // Default constructor
20   ~AliInvmass();                                   // Destructor
21   void SetStorageMode(Int_t m);                    // Set storage mode (1=single, 2=multiple)
22   void SetThetaSwitch(Int_t i=1);                  // Enable (1/0) new theta for comb. bkg. reco.
23   void SetPhiSwitch(Int_t i=1);                    // Enable (1/0) new phi for comb. bkg. reco.
24   Int_t GetStorageMode();                          // Provide storage mode
25   Int_t GetThetaSwitch();                          // Provide theta switch flag
26   Int_t GetPhiSwitch();                            // Provide phi switch flag
27   TObjArray* Invmass(TObjArray* a1,TObjArray* a2); // Two-particle inv. mass reco.
28   TObjArray* CombBkg(TObjArray* a1,TObjArray* a2); // Two-particle comb. background reco.
29
30  protected:
31   Double_t fPi;     // Value of pi
32   Int_t fMode;      // Storage mode for signal and bkg. results (2=separate arrays)
33   Int_t fBkg;       // Flag to denote comb. background processing
34   AliRandom fRndm;  // The random number generator for the comb. bkg. reconstruction
35   Int_t fNewtheta;  // Flag to denote enabling of switching theta for comb. bkg. reco.
36   Int_t fNewphi;    // Flag to denote enabling of switching phi for comb. bkg. reco.
37   TObjArray* fMinv; // Array with reconstructed invariant mass 'tracks'
38   TObjArray* fMbkg; // Array with reconstructed comb. background 'tracks'
39
40  private:
41   void Combine(TObjArray* a1,TObjArray* a2); // Make two-particle combinations
42
43  ClassDef(AliInvmass,1) // Construction of invariant mass and combinatorial background.
44 };
45 #endif