Coding convention clean-up
[u/mrichter/AliRoot.git] / EVGEN / AliDimuCombinator.h
1 #ifndef ALIDIMUCOMBINATOR_H
2 #define ALIDIMUCOMBINATOR_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
9 //  Class for dimuon analysis and fast dimuon simulation.
10 //  It uses the AliRun particle tree.
11 //  Comments and suggestions to andreas.morsch@cern.ch
12
13
14 #include <TObject.h>
15
16 class TClonesArray;
17 class TParticle;
18
19
20 class AliDimuCombinator:
21     public TObject 
22 {
23 public:
24     AliDimuCombinator(TClonesArray* Partarray);
25     void  Copy(AliDimuCombinator &combi) const;
26 //    
27 //  Iterators
28 //  Single muons
29     TParticle* FirstMuon();
30     TParticle* NextMuon();
31 //  Single muons selected
32     TParticle* FirstMuonSelected();
33     TParticle* NextMuonSelected();
34 //  Dimuons    
35     void FirstMuonPair(TParticle* & muon1, TParticle* & muon2);
36     void NextMuonPair(TParticle* & muon1, TParticle* & muon2);
37 //  Dimuons selected    
38     void FirstMuonPairSelected(TParticle* & muon1, TParticle* & muon2);
39     void NextMuonPairSelected(TParticle* & muon1, TParticle* & muon2);
40 //  Loop over all prticles    
41     void ResetRange();
42 //  Set two ranges for dimuon loop    
43     void SetFirstRange (Int_t from, Int_t to);
44     void SetSecondRange(Int_t from, Int_t to);    
45 //  Cuts
46     void SetPtMin(Float_t ptmin) {fPtMin=ptmin;}
47     void SetEtaCut(Float_t etamin, Float_t etamax)
48         {fEtaMin=etamin; fEtaMax=etamax;}
49     Bool_t Selected(TParticle* part);
50     Bool_t Selected(TParticle* part1, TParticle* part2);
51 // Kinematics
52     Float_t Mass(TParticle* part1, TParticle* part);
53     Float_t PT(TParticle* part1, TParticle* part);
54     Float_t Pz(TParticle* part1, TParticle* part);
55     Float_t Y(TParticle* part1, TParticle* part);
56 // Response
57     void SmearGauss(Float_t width, Float_t & value);
58 // Weight
59     Bool_t  Correlated(TParticle* part1, TParticle* part2);
60     void    SetRate(Float_t rate) {fRate1=rate;}
61     void    SetRate(Float_t rate1, Float_t rate2 ) {fRate1=rate1; fRate2=rate2;}
62     Float_t Weight(TParticle* part);
63     Float_t Weight(TParticle* part1, TParticle* part);
64     Float_t DecayProbability(TParticle* part);
65     
66  private:
67     void FirstPartner();
68     void NextPartner();
69     void FirstPartnerSelected();
70     void NextPartnerSelected();
71     Int_t Origin(TParticle* part);
72     TParticle* Parent(TParticle* part);
73     TParticle* Partner();
74     Int_t Type(TParticle *part);
75     AliDimuCombinator(const AliDimuCombinator &combinator);
76     AliDimuCombinator & operator=(const AliDimuCombinator & rhs);
77
78  private:
79     TClonesArray *fPartArray;      // MC Event
80     Int_t fNParticle;              // Number of particles
81     Int_t fImuon1;                 // Index of first muon
82     Int_t fImuon2;                 // Index of second muon
83     Int_t fImin1;                  // Lowest index for first   muon  
84     Int_t fImin2;                  // Lowest index for second  muon 
85     Int_t fImax1;                  // Highest index for first  muon  
86     Int_t fImax2;                  // Highest index for second muon 
87     Float_t fRate1;                // weight factor  
88     Float_t fRate2;                // weight factor
89     TParticle *fMuon1;             // First muon
90     TParticle *fMuon2;             // Second muon
91     Float_t fPtMin;                // pT-cut 
92     Float_t fEtaMin;               // Minimum pseudorapidity cut
93     Float_t fEtaMax;               // Maximum pseudorapidity cut
94     
95     ClassDef(AliDimuCombinator,1)  // Tools for dimuon combinatoric studies
96 };
97 #endif
98
99
100
101
102
103
104
105