Updates to AddTask macro from Chris
[u/mrichter/AliRoot.git] / MUON / MUONTriggerEfficiency.C
1 /**************************************************************************
2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notice   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15
16 // $Id$
17
18 /// \ingroup macros
19 /// \file MUONTriggerEfficiency.C
20 /// \brief Macro for checking trigger integrated efficiency for dimuons.
21 ///
22 /// The efficiency is calculated with respect to the 3/4 coincidence.
23 ///
24 /// \author Fabien Guerin, LPC Clermont-Ferrand, Jan. 2006
25
26 #if !defined(__CINT__) || defined(__MAKECINT__)
27 // ROOT includes
28 #include "TBranch.h"
29 #include "TClonesArray.h"
30 #include "TObject.h"
31 #include "TFile.h"
32 #include "TH1.h"
33 #include "TH2.h"
34 #include "TTree.h"
35 #include "TParticle.h"
36
37 // STEER includes
38 #include "AliRun.h"
39 #include "AliRunLoader.h"
40 #include "AliHeader.h"
41 #include "AliLoader.h"
42 #include "AliStack.h"
43 #include "AliCDBManager.h"
44
45 // MUON includes
46 #include "AliMUON.h"
47 #include "AliMUONHit.h"
48 #include "AliMUONConstants.h"
49 #include "AliMUONDigit.h"
50 #include "AliMUONGlobalTrigger.h"
51 #include "AliMUONTrack.h"
52
53 #include "AliMUONDataInterface.h"
54 #include "AliMUONMCDataInterface.h"
55 #include "AliMUONVHitStore.h"
56 #include "AliMUONVTriggerStore.h"
57
58 #endif
59
60 // Upsilon(1S)
61
62
63 void MUONTriggerEfficiency(const char* filenameSim="galice_sim.root", 
64                            const char* filenameRec="galice.root",  
65                            Bool_t readFromRP = 0)
66 {
67   
68   // Set default CDB storage
69   AliCDBManager* man = AliCDBManager::Instance();
70   man->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
71   man->SetRun(0);
72   
73   // output file
74   
75   AliMUONMCDataInterface diSim(filenameSim);
76   AliMUONDataInterface diRec(filenameRec);
77   
78   if (!diSim.IsValid())
79   {
80     cerr << "Cannot access " << filenameSim << endl;
81     return;
82   }
83   
84   if (!diRec.IsValid())
85   {
86     cerr << "Cannot access " << filenameRec << endl;
87     return;
88   }
89   
90   FILE* fdat = fopen("MUONTriggerEfficiency.out","w");          
91   if (!fdat)
92   {
93     cerr << "Cannot create output file" << endl;
94     return;
95   }
96   
97   Int_t coincmuon,muonlpt,muonhpt;
98   Int_t CoincMuPlus,CoincMuMinus;
99   coincmuon=0;
100   muonlpt=0;  
101   muonhpt=0;
102   
103   Int_t nevents = diSim.NumberOfEvents();          
104   
105   for (Int_t ievent=0; ievent<nevents; ++ievent) 
106   {   
107     CoincMuPlus=0;
108     CoincMuMinus=0;
109     
110     if (ievent%1000==0) printf("\t Event = %d\n",ievent);    
111     
112     // Hits
113     
114     Int_t NbHits[2][4];
115     for (Int_t j=0; j<2; j++) 
116     {
117       for (Int_t jj=0; jj<4; jj++) 
118       {
119         NbHits[j][jj]=0;
120       }
121     } 
122     
123     Int_t ntracks = (Int_t) diSim.NumberOfTracks(ievent);
124     
125     for ( Int_t itrack=0; itrack<ntracks; ++itrack ) 
126     {
127       AliMUONVHitStore* hitStore = diSim.HitStore(ievent,itrack);      
128       AliMUONHit* mHit;
129       TIter next(hitStore->CreateIterator());
130       
131       while ( ( mHit = static_cast<AliMUONHit*>(next()) ) )
132       {
133         Int_t Nch = mHit->Chamber(); 
134         Int_t hittrack = mHit->Track();
135         Float_t IdPart = mHit->Particle();
136         
137         for (Int_t j=0;j<4;j++) 
138         {
139           Int_t kch=11+j;
140           if (hittrack==1 || hittrack==2) 
141           {
142             if (Nch==kch && IdPart==-13 && NbHits[0][j]==0) NbHits[0][j]++; 
143             if (Nch==kch && IdPart==13 && NbHits[1][j]==0) NbHits[1][j]++; 
144           }       
145         }
146       }
147     } // end track loop     
148     
149     // 3/4 coincidence
150     Int_t SumNbHits=NbHits[0][0]+NbHits[0][1]+NbHits[0][2]+NbHits[0][3];
151     if (SumNbHits==3 || SumNbHits==4) CoincMuPlus=1;
152     
153     SumNbHits=NbHits[1][0]+NbHits[1][1]+NbHits[1][2]+NbHits[1][3];
154     if (SumNbHits==3 || SumNbHits==4) CoincMuMinus=1;
155     
156     if (CoincMuPlus==1 && CoincMuMinus==1) coincmuon++;         
157           
158     TString tree("D");
159     if ( readFromRP ) tree = "R";
160     
161     AliMUONVTriggerStore* triggerStore = diRec.TriggerStore(ievent,tree.Data());
162
163     AliMUONGlobalTrigger* gloTrg = triggerStore->Global();
164     
165     if (gloTrg->PairUnlikeLpt()>=1) muonlpt++;
166     if (gloTrg->PairUnlikeHpt()>=1) muonhpt++;
167
168   } // end loop on event  
169  
170   // calculate efficiency with as a ref. at least 3/4 planes fired
171   Float_t efficiencylpt,efficiencyhpt;
172   Double_t coincmu,lptmu,hptmu;  
173   Float_t error;  
174   coincmu=Double_t(coincmuon);
175   lptmu=Double_t(muonlpt); 
176   hptmu=Double_t(muonhpt); 
177
178   // output
179   fprintf(fdat,"\n"); 
180   fprintf(fdat," Number of events = %d \n",nevents);  
181   fprintf(fdat," Number of events with 3/4 coinc = %d \n",coincmuon);  
182   fprintf(fdat," Nomber of dimuons with 3/4 coinc Lpt cut = %d \n",muonlpt);  
183   fprintf(fdat," Number of dimuons with 3/4 coinc Hpt cut = %d \n",muonhpt);  
184   fprintf(fdat,"\n");
185   
186   efficiencylpt=lptmu/coincmu;  
187   error=efficiencylpt*TMath::Sqrt((lptmu+coincmu)/(lptmu*coincmu));  
188   fprintf(fdat," Efficiency Lpt cut = %4.4f +/- %4.4f\n",efficiencylpt,error); 
189  
190   efficiencyhpt=hptmu/coincmu; 
191   error=efficiencyhpt*TMath::Sqrt((hptmu+coincmu)/(hptmu*coincmu));          
192   fprintf(fdat," Efficiency Hpt cut = %4.4f +/- %4.4f\n",efficiencyhpt,error);
193       
194
195   fclose(fdat);  
196 }
197
198
199
200
201