Changed DimuonCombinator into AliDimuCombinator to follow ALICE coding

[u/mrichter/AliRoot.git] / EVGEN / AliDimuCombinator.h

diff --git a/EVGEN/AliDimuCombinator.h b/EVGEN/AliDimuCombinator.h
new file mode 100644 (file)
index 0000000..79dde52
--- /dev/null
+++ b/EVGEN/AliDimuCombinator.h
@@ -0,0 +1,101 @@
+#ifndef _AliDimuCombinator_H
+#define _AliDimuCombinator_H
+#include "GParticle.h"
+#include <TBrowser.h>
+#include <TList.h>
+#include <TTree.h>
+#include <TROOT.h>
+
+
+class AliDimuCombinator:
+    public TObject 
+{
+public:
+    AliDimuCombinator(TClonesArray* Partarray){
+       fPartArray=Partarray;
+       fNParticle=fPartArray->GetEntriesFast();
+       
+       fimuon1 =0;
+       fimuon2 =0;
+       fmuon1  =0;
+       fmuon2  =0;
+       fimin1  = 0;
+       fimin2  = 0;
+       fimax1  = fNParticle;
+       fimax2  = fNParticle;
+       fPtMin  =0;
+       fEtaMin =-10;
+       fEtaMax =-10;
+       fRate1=1.;
+       fRate2=1.;
+    }
+//    
+//  Iterators
+//  Single muons
+    GParticle* FirstMuon();
+    GParticle* NextMuon();
+//  Single muons selected
+    GParticle* FirstMuonSelected();
+    GParticle* NextMuonSelected();
+//  Dimuons    
+    void FirstMuonPair(GParticle* & muon1, GParticle* & muon2);
+    void NextMuonPair(GParticle* & muon1, GParticle* & muon2);
+//  Dimuons selected    
+    void FirstMuonPairSelected(GParticle* & muon1, GParticle* & muon2);
+    void NextMuonPairSelected(GParticle* & muon1, GParticle* & muon2);
+//  Loop over all prticles    
+    void ResetRange();
+//  Set two ranges for dimuon loop    
+    void SetFirstRange (Int_t from, Int_t to);
+    void SetSecondRange(Int_t from, Int_t to);    
+//  Cuts
+    void SetPtMin(Float_t ptmin) {fPtMin=ptmin;}
+    void SetEtaCut(Float_t etamin, Float_t etamax){fEtaMin=etamin; fEtaMax=etamax;}      Bool_t Selected(GParticle* part);
+    Bool_t Selected(GParticle* part1, GParticle* part2);
+// Kinematics
+    Float_t Mass(GParticle* part1, GParticle* part);
+    Float_t PT(GParticle* part1, GParticle* part);
+    Float_t Pz(GParticle* part1, GParticle* part);
+    Float_t Y(GParticle* part1, GParticle* part);
+// Response
+    void SmearGauss(Float_t width, Float_t & value);
+// Weight
+    Bool_t  Correlated(GParticle* part1, GParticle* part2);
+    void    SetRate(Float_t rate){fRate1=rate;}
+    void    SetRate(Float_t rate1, Float_t rate2 ){fRate1=rate1; fRate2=rate2;}
+    Float_t Weight(GParticle* part);
+    Float_t Weight(GParticle* part1, GParticle* part);
+    Float_t Decay_Prob(GParticle* part);
+    
+ private:
+    void FirstPartner();
+    void NextPartner();
+    void FirstPartnerSelected();
+    void NextPartnerSelected();
+    Int_t Origin(GParticle* part);
+    GParticle* Parent(GParticle* part);
+    GParticle* Partner();
+    Int_t Type(GParticle *part){return part->GetKF();}
+private:
+    TClonesArray *fPartArray;
+    Int_t fNParticle;
+    Int_t fimuon1;
+    Int_t fimuon2;
+    Int_t fimin1;
+    Int_t fimin2;
+    Int_t fimax1;
+    Int_t fimax2;
+    Float_t fRate1;
+    Float_t fRate2;
+    GParticle *fmuon1;
+    GParticle *fmuon2;
+    Float_t fPtMin;
+    Float_t fEtaMin;
+    Float_t fEtaMax;
+  ClassDef(AliDimuCombinator,1) // Dimuon Combinator
+};
+#endif
+
+
+
+