Add const to SolenoidField() method for correct overwriting from the parent class
[u/mrichter/AliRoot.git] / EVGEN / AliDimuCombinator.cxx
index 861a32d..12f70e2 100644 (file)
 
 /*
 $Log$
+Revision 1.10  2001/03/27 11:14:54  morsch
+Weight calculation for correlated particles updated:
+- Decay probability is counted once if muons are decay products
+of the same mother particle. Otherwise,  it's counted twice.
+
+Revision 1.9  2001/03/08 13:30:43  morsch
+Make it work with particle stack of V3.05.
+
+Revision 1.8  2000/12/21 16:24:06  morsch
+Coding convention clean-up
+
+Revision 1.7  2000/10/02 15:16:37  morsch
+Correct coding rule violation for member data names of type fi -> fI.
+
 Revision 1.6  2000/06/14 15:19:47  morsch
 Include clean-up (IH)
 
@@ -29,57 +43,67 @@ Introduction of the Copyright and cvs Log
 
 */
 
-//
-//
-//
-//
+/*  
+ Class for dimuon analysis and fast dimuon simulation.
+ It provides single and dimuon iterators, cuts, weighting, kinematic
+ It uses the AliRun particle tree.
+ Comments and suggestions to 
+ andreas.morsch@cern.ch
+*/
+
+
 #include "AliDimuCombinator.h" 
 #include "AliPDG.h" 
+#include "AliRun.h" 
 #include <TRandom.h>
 #include <TClonesArray.h>
 #include <TParticle.h>
+#include <TTree.h>
 
 //
 ClassImp(AliDimuCombinator)
-    AliDimuCombinator::AliDimuCombinator(TClonesArray* Partarray) 
+    AliDimuCombinator::AliDimuCombinator() 
 {
 // Constructor
-    fPartArray=Partarray;
-    fNParticle=fPartArray->GetEntriesFast();
-    
-    fImuon1 =0;
-    fImuon2 =0;
-    fMuon1  =0;
-    fMuon2  =0;
+    fNParticle = (Int_t) (gAlice->TreeK())->GetEntries();
+    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.;
+    fPtMin  = 0;
+    fEtaMin = -10;
+    fEtaMax = -10;
+    fRate1  = 1.;
+    fRate2  = 1.;
 }
 
 AliDimuCombinator::AliDimuCombinator(const AliDimuCombinator & combinator)
 {
-// copy constructor
+// Dummy copy constructor
 }
 
 
 //
 //                       Iterators
 // 
+TParticle* AliDimuCombinator::Particle(Int_t i)
+{
+    return gAlice->Particle(i);
+}
+
 TParticle* AliDimuCombinator::FirstMuon()
 {
 // Single muon iterator: initialisation
-    fImuon1=fImin1;
-    fMuon1 = (TParticle*) fPartArray->UncheckedAt(fImuon1);
-    while(Type(fMuon1)!=kMuonPlus && Type(fMuon1)!=kMuonMinus) {
+    fImuon1 = fImin1;
+    fMuon1  = Particle(fImuon1);
+    while(Type(fMuon1) != kMuonPlus && Type(fMuon1) != kMuonMinus) {
        fImuon1++;
-       if (fImuon1 >= fImax1) {fMuon1=0; break;}
-       fMuon1 = (TParticle*) fPartArray->UncheckedAt(fImuon1);
+       if (fImuon1 >= fImax1) {fMuon1 = 0; break;}
+       fMuon1 = Particle(fImuon1);
     }
     return fMuon1;
 }
@@ -87,8 +111,8 @@ TParticle* AliDimuCombinator::FirstMuon()
 TParticle* AliDimuCombinator::FirstMuonSelected()
 {
 // Single selected muon iterator: initialisation
-    TParticle * muon=FirstMuon();
-    while(muon!=0 && !Selected(muon)) {muon=NextMuon();}
+    TParticle* muon = FirstMuon();
+    while(muon != 0 && !Selected(muon)) {muon = NextMuon();}
     return muon;
 }
 
@@ -97,13 +121,13 @@ TParticle* AliDimuCombinator::NextMuon()
 {
 // Single muon iterator: increment
     fImuon1++;
-    if (fImuon1>=fNParticle) {fMuon1 = 0; return fMuon1;}
+    if (fImuon1 >= fNParticle) {fMuon1 = 0; return fMuon1;}
     
-    fMuon1 = (TParticle*) fPartArray->UncheckedAt(fImuon1);
-    while(Type(fMuon1)!=kMuonPlus && Type(fMuon1)!=kMuonMinus) {
+    fMuon1 = Particle(fImuon1);
+    while(Type(fMuon1) != kMuonPlus && Type(fMuon1) != kMuonMinus) {
        fImuon1++;
-       if (fImuon1>=fImax1) {fMuon1 = 0; break;}
-       fMuon1 = (TParticle*) fPartArray->UncheckedAt(fImuon1);
+       if (fImuon1 >= fImax1) {fMuon1 = 0; break;}
+       fMuon1 = Particle(fImuon1);
     }
     return fMuon1;
 }
@@ -111,8 +135,8 @@ TParticle* AliDimuCombinator::NextMuon()
 TParticle* AliDimuCombinator::NextMuonSelected()
 {
 // Single selected muon iterator: increment
-    TParticle * muon=NextMuon();
-    while(muon !=0 && !Selected(muon)) {muon=NextMuon();}
+    TParticle * muon = NextMuon();
+    while(muon !=0 && !Selected(muon)) {muon = NextMuon();}
     return muon;
 }
 
@@ -120,17 +144,17 @@ TParticle* AliDimuCombinator::NextMuonSelected()
 void AliDimuCombinator::FirstPartner()
 {
 // Helper for  dimuon iterator: initialisation
-    if (fImin1==fImin2) {
-       fImuon2=fImuon1+1;
+    if (fImin1 == fImin2) {
+       fImuon2 = fImuon1+1;
     } else {
-       fImuon2=fImin2;
+       fImuon2 = fImin2;
     }
-    if (fImuon2 >= fImax2) {fMuon2=0; return;}
-    fMuon2 = (TParticle*) fPartArray->UncheckedAt(fImuon2);
-    while(Type(fMuon2)!=kMuonPlus && Type(fMuon2)!=kMuonMinus) {
+    if (fImuon2 >= fImax2) {fMuon2 = 0; return;}
+    fMuon2 = Particle(fImuon2);
+    while(Type(fMuon2) != kMuonPlus && Type(fMuon2) != kMuonMinus) {
        fImuon2++;
-       if (fImuon2 >= fImax2) {fMuon2=0; break;}
-       fMuon2 = (TParticle*) fPartArray->UncheckedAt(fImuon2);
+       if (fImuon2 >= fImax2) {fMuon2 = 0; break;}
+       fMuon2 = Particle(fImuon2);
     }
 }
 
@@ -146,15 +170,15 @@ void AliDimuCombinator::NextPartner()
 {
 // Helper for dimuon iterator: increment    
     fImuon2++;
-    if (fImuon2>=fImax2) {fMuon2 = 0; return;}
+    if (fImuon2 >= fImax2) {fMuon2 = 0; return;}
     
     
-    fMuon2 = (TParticle*) fPartArray->UncheckedAt(fImuon2);
+    fMuon2 = Particle(fImuon2);
     
-    while(Type(fMuon2)!=kMuonPlus && Type(fMuon2)!=kMuonMinus) {
+    while(Type(fMuon2) != kMuonPlus && Type(fMuon2) != kMuonMinus) {
        fImuon2++;
-       if (fImuon2>=fImax2) {fMuon2 = 0; break;}
-       fMuon2 = (TParticle*) fPartArray->UncheckedAt(fImuon2);
+       if (fImuon2 >= fImax2) {fMuon2 = 0; break;}
+       fMuon2 = Particle(fImuon2);
     }
 }
 
@@ -177,8 +201,8 @@ void AliDimuCombinator::FirstMuonPair(TParticle* & muon1, TParticle* & muon2)
 // Dimuon iterator: initialisation
     FirstMuon();
     FirstPartner();
-    muon1=fMuon1;
-    muon2=fMuon2;       
+    muon1 = fMuon1;
+    muon2 = fMuon2;     
 }
 
 void AliDimuCombinator::NextMuonPair(TParticle* & muon1, TParticle* & muon2)
@@ -189,8 +213,8 @@ void AliDimuCombinator::NextMuonPair(TParticle* & muon1, TParticle* & muon2)
        NextMuon();
        FirstPartner();
     }
-    muon1=fMuon1;
-    muon2=fMuon2;       
+    muon1 = fMuon1;
+    muon2 = fMuon2;     
 }
 void AliDimuCombinator::FirstMuonPairSelected(TParticle* & muon1, 
                                              TParticle* & muon2)
@@ -198,8 +222,8 @@ void AliDimuCombinator::FirstMuonPairSelected(TParticle* & muon1,
 // Selected dimuon iterator: initialisation    
     FirstMuonSelected();
     FirstPartnerSelected();
-    muon1=fMuon1;
-    muon2=fMuon2;       
+    muon1 = fMuon1;
+    muon2 = fMuon2;     
 }
 
 void AliDimuCombinator::NextMuonPairSelected(TParticle* & muon1, 
@@ -211,31 +235,31 @@ void AliDimuCombinator::NextMuonPairSelected(TParticle* & muon1,
        NextMuonSelected();
        FirstPartnerSelected();
     }
-    muon1=fMuon1;
-    muon2=fMuon2;       
+    muon1 = fMuon1;
+    muon2 = fMuon2;     
 }
 
 void AliDimuCombinator::ResetRange()
 {
 // Reset index ranges for single muons
-    fImin1=fImin2=0;
-    fImax1=fImax2=fNParticle;
+    fImin1 = fImin2 = 0;
+    fImax1 = fImax2 = fNParticle;
 }
 
 void AliDimuCombinator::SetFirstRange(Int_t from, Int_t to)
 {
 // Reset index range for first muon
-    fImin1=from;
-    fImax1=to;
-    if (fImax1 > fNParticle) fImax1=fNParticle;
+    fImin1 = from;
+    fImax1 = to;
+    if (fImax1 > fNParticle) fImax1 = fNParticle;
 }
 
 void AliDimuCombinator::SetSecondRange(Int_t from, Int_t to)
 {
 // Reset index range for second muon
-    fImin2=from;
-    fImax2=to;
-    if (fImax2 > fNParticle) fImax2=fNParticle;
+    fImin2 = from;
+    fImax2 = to;
+    if (fImax2 > fNParticle) fImax2 = fNParticle;
 }
 //
 //                       Selection
@@ -245,9 +269,9 @@ Bool_t AliDimuCombinator::Selected(TParticle* part)
 {
 // Selection cut for single muon 
 //
-    if (part==0) {return 0;}
+    if (part == 0) {return 0;}
     
-    if (part->Pt() > fPtMin && part->Eta()>fEtaMin && part->Eta()<fEtaMax) {
+    if (part->Pt() > fPtMin && part->Eta() > fEtaMin && part->Eta() < fEtaMax) {
        return 1;
     } else {
        return 0;
@@ -268,11 +292,11 @@ Float_t AliDimuCombinator::Mass(TParticle* part1, TParticle* part2)
 // Invariant mass
 //
     Float_t px,py,pz,e;
-    px=part1->Px()+part2->Px();
-    py=part1->Py()+part2->Py();
-    pz=part1->Pz()+part2->Pz();    
-    e =part1->Energy()+part2->Energy();
-    Float_t p=px*px+py*py+pz*pz;
+    px = part1->Px()+part2->Px();
+    py = part1->Py()+part2->Py();
+    pz = part1->Pz()+part2->Pz();    
+    e  = part1->Energy()+part2->Energy();
+    Float_t p = px*px+py*py+pz*pz;
     if (e*e < p) {
        return -1; 
     } else {
@@ -285,8 +309,8 @@ Float_t AliDimuCombinator::PT(TParticle* part1, TParticle* part2)
 // Transverse momentum of dimuons
 //
     Float_t px,py;
-    px=part1->Px()+part2->Px();
-    py=part1->Py()+part2->Py();
+    px = part1->Px()+part2->Px();
+    py = part1->Py()+part2->Py();
     return TMath::Sqrt(px*px+py*py);
 }
 
@@ -302,8 +326,8 @@ Float_t AliDimuCombinator::Y(TParticle* part1, TParticle* part2)
 // Rapidity of dimuon system
 //
     Float_t pz,e;
-    pz=part1->Pz()+part2->Pz();
-    e =part1->Energy()+part2->Energy();
+    pz = part1->Pz()+part2->Pz();
+    e  = part1->Energy()+part2->Energy();
     return 0.5*TMath::Log((e+pz)/(e-pz));
 }
 //                  Response
@@ -321,15 +345,16 @@ Float_t AliDimuCombinator::DecayProbability(TParticle* part)
 {
 // Calculate decay probability for muons from pion and kaon decays
 // 
+
     Float_t d, h, theta, cTau;
     TParticle* parent = Parent(part);
-    Int_t ipar=Type(parent);
-    if (ipar==kPiPlus || ipar==kPiMinus) {
+    Int_t ipar = Type(parent);
+    if (ipar == kPiPlus || ipar == kPiMinus) {
        cTau=780.4;
-    } else if (ipar==kKPlus || ipar==kKMinus) {
-       cTau=370.9;
+    } else if (ipar == kKPlus || ipar == kKMinus) {
+       cTau = 370.9;
     } else {
-       cTau=0;
+       cTau = 0;
     }
     
     
@@ -337,8 +362,10 @@ Float_t AliDimuCombinator::DecayProbability(TParticle* part)
 //
 // this part is still very ALICE muon-arm specific
 //
-    theta=parent->Theta();
-    h=90*TMath::Tan(theta);
+
+
+    theta = parent->Theta();
+    h = 90*TMath::Tan(theta);
     
     if (h<4) {
        d=4/TMath::Sin(theta);
@@ -353,19 +380,82 @@ Float_t AliDimuCombinator::DecayProbability(TParticle* part)
     }
 }
 
+//Begin_Html
+/*
+<p> In the the code above :
+<P>If h is less than 4 cm, pions or kaons go in the beam pipe and can have a long way
+<BR>If h is greater than 4 cm, pions or kaons crash into the front absorber
+<P><IMG SRC="absorbeur.jpg" HEIGHT=292 WIDTH=819>
+*/
+//End_Html
+
+
 Float_t AliDimuCombinator::Weight(TParticle* part1, TParticle* part2)
 {
 // Dimuon weight
 
-    Float_t wgt=(part1->GetWeight())*(part2->GetWeight());
+    Float_t wgt = (part1->GetWeight())*(part2->GetWeight());
     
     if (Correlated(part1, part2)) {
-       return wgt/(Parent(part1)->GetWeight())*fRate1;
+       if ( part1->GetFirstMother() == part2->GetFirstMother()) {
+           return part1->GetWeight()*fRate1;
+       } else {
+           return wgt/(Parent(part1)->GetWeight())*fRate1;
+       }
     } else {
        return wgt*fRate1*fRate2;
     }
 } 
 
+//Begin_Html
+/*
+<p>Some clarifications on the calculation of the dimuons weight :
+<P>We must keep in mind that if we force the meson decay in muons and we put
+lot of mesons (J/psi, upsilon, ...) to have a good statistic we are
+obliged to calculate different weights to correct the number
+of muons
+<BR>&nbsp;
+<P>First -->
+<BR>The particle weight is given by w=R*M*Br
+<BR>&nbsp;with&nbsp; :
+<UL>R&nbsp;&nbsp; =&nbsp; the rate by event. This number gives the number
+of produced J/psi, upsilon, pion ... in a collision.
+<BR>It corresponds of the weight 0.06 given for example in&nbsp; gener->AddGenerator(jpsi,"J/Psi",
+0.06); from the config.C macro.
+<BR>In this example R=0.06
+
+<P>M&nbsp; = the rate of the mother production. This number depend on :
+<BR>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp; - the number of generated events --> fParentWeight=1./Float_t(fNpart) in AliGenPythia.cxx . This
+is a normalization to 1 of the number of generated particles.
+<BR>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp; - the kinematic bias coming
+from the y and Pt cuts.&nbsp; Method&nbsp; AliGenPythia::AdjustWeights() in AliGenPythia.cxx
+<BR>(in AliGenParam.cxx this 2 things are taken into account in fParentWeight
+= fYWgt*fPtWgt*phiWgt/fNpart )
+
+<P>Br = the branching ratio in muon from the mother decay</UL>
+
+<P><BR>In this method, part->GetWeight() = M*Br
+<UL>&nbsp;</UL>
+Next -->
+<BR>The weight of the dimuon depends on the correlation between muons
+<BR>&nbsp;
+<UL>If the muons are correlated and come from a resonance (for example
+J/psi -> mu+ mu-) , the weight of the dimuon is the weight of one muon then
+<BR>w12= R*M*Br = w1* R1 (in this method this gives part1->GetWeight()*fRate1)
+
+<P>If the muons are correlated and come from a charm or a bottom pair then
+w12 = M*R*Br1*Br2 = w1*w2*R1/M1
+<BR>(in this method this gives wgt/(Parent(part1)->GetWeight())*fRate1).
+Indeed the 2 muons come from the same mother so the
+<BR>weight of a DD~ or BB~ is M*Br and they are no correlation in the decay
+(Br1*Br2)
+
+<P>If the muons are not correlated w12 = M1*M2*R1*R2*Br1*Br2 = w1*w2*R1*R2
+(in this method this gives wgt*fRate1*fRate2)
+<BR>&nbsp;</UL>
+*/
+//End_Html
+
 
 Float_t AliDimuCombinator::Weight(TParticle* part)
 {
@@ -377,7 +467,15 @@ Bool_t  AliDimuCombinator::Correlated(TParticle* part1, TParticle* part2)
 {
 // Check if muons are correlated
 //
-    if (Origin(part1) == Origin(part2)) {
+    if ((Origin(part1) >= 0) && Origin(part1) == Origin(part2)) {
+/*
+       printf("\n origin %d %d ", 
+              Type(Particle(Origin(part1))),
+              Type(Particle(Origin(part2))));
+       printf("\n parent %d %d \n \n ", 
+              Type(Parent(part1)),
+              Type(Parent(part2)));
+*/     
        return kTRUE;
     } else {
        return kFALSE;
@@ -388,7 +486,7 @@ TParticle* AliDimuCombinator::Parent(TParticle* part)
 {
 // Return pointer to parent
 //
-    return (TParticle*) (fPartArray->UncheckedAt(part->GetFirstMother()));
+    return Particle(part->GetFirstMother());
 }
 
 Int_t AliDimuCombinator::Origin(TParticle* part)
@@ -399,11 +497,11 @@ Int_t AliDimuCombinator::Origin(TParticle* part)
     if (iparent < 0) return iparent;
     Int_t ip;
     while(1) {
-       ip=((TParticle*) fPartArray->UncheckedAt(iparent))->GetFirstMother();
+       ip = (Particle(iparent))->GetFirstMother();
        if (ip < 0) {
            break;
        } else {
-           iparent=ip;
+           iparent = ip;
        }
     }
     return iparent;
@@ -422,7 +520,13 @@ AliDimuCombinator& AliDimuCombinator::operator=(const  AliDimuCombinator& rhs)
 }
 
 
-
+void AliDimuCombinator::Copy(AliDimuCombinator &combi) const
+{
+  //
+  // Copy *this onto lego -- not implemented
+  //
+  Fatal("Copy","Not implemented!\n");
+}