adding the option for hybrid tracks global+global constrained as well as global+TPC...
[u/mrichter/AliRoot.git] / PWG4 / macros / ConfigAnalysisMCExample.C
1 /* $Id:  $ */
2 /* $Log$ */
3
4 //------------------------------------
5 // Configuration macro example:
6 //
7 // Do example analysis with MC
8 //
9 // Author : Gustavo Conesa Balbastre (INFN-LNF)
10 //------------------------------------
11
12 AliAnaPartCorrMaker*  ConfigAnalysis()
13 {
14   //
15   // Configuration goes here
16   // 
17   printf("======================== \n");
18   printf("ConfigAnalysis() \n");
19   printf("======================== \n");
20   
21   
22   //Detector Fiducial Cuts
23   AliFiducialCut * fidCut = new AliFiducialCut();
24   fidCut->DoCTSFiducialCut(kFALSE) ;
25   fidCut->DoEMCALFiducialCut(kFALSE) ;
26   fidCut->DoPHOSFiducialCut(kTRUE) ;
27   
28   //fidCut->SetSimpleCTSFiducialCut(0.9,0.,360.);
29   //fidCut->SetSimpleEMCALFiducialCut(0.7,80.,190.);
30   fidCut->SetSimplePHOSFiducialCut(0.13,220.,320.);
31   
32   //   //Fiducial cut EMCAL, 5 regions
33   //   Float_t etamax[]={0.67,0.51,0.16,-0.21,-0.61};
34   //   TArrayF etamaxarr(5,etamax);
35   //   fidCut->AddEMCALFidCutMaxEtaArray(etamaxarr);
36   //   Float_t etamin[]={0.61,0.21,-0.16,-0.51,-0.67};
37   //   TArrayF etaminarr(5,etamin);
38   //   fidCut->AddEMCALFidCutMinEtaArray(etaminarr);
39   //   Float_t phimax[]={99.*TMath::DegToRad(), 119.*TMath::DegToRad(), 139.*TMath::DegToRad(), 
40   //                159.*TMath::DegToRad(), 179.*TMath::DegToRad(), 189.*TMath::DegToRad()};
41   //   TArrayF phimaxarr(6,phimax);
42   //   fidCut->AddEMCALFidCutMaxPhiArray(phimaxarr);
43   //   Float_t phimin[]={81.*TMath::DegToRad(), 101.*TMath::DegToRad(), 121.*TMath::DegToRad(), 
44   //                141.*TMath::DegToRad(), 161.*TMath::DegToRad(), 181.*TMath::DegToRad()};
45   //   TArrayF phiminarr(6,phimin);
46   //   fidCut->AddEMCALFidCutMinPhiArray(phiminarr);
47   
48   fidCut->Print("");
49   
50   //-----------------------------------------------------------  
51   // Reader
52   //-----------------------------------------------------------
53   AliCaloTrackMCReader *reader = new AliCaloTrackMCReader();
54   reader->SetDebug(-1);
55   
56  //Switch on or off the detectors information that you want
57   reader->SwitchOffEMCAL();
58   reader->SwitchOffCTS();
59   reader->SwitchOnPHOS();
60   reader->SwitchOffEMCALCells();
61   reader->SwitchOffPHOSCells();
62
63   //Min particle pT
64   reader->SetEMCALPtMin(0.); 
65   reader->SetPHOSPtMin(0.);
66   reader->SetCTSPtMin(.2);
67  
68   reader->SetFiducialCut(fidCut);
69
70   //Remove the temporal AODs we create. 
71   reader->SwitchOnCleanStdAOD();        
72         
73   reader->Print("");
74   
75   
76   //---------------------------------------------------------------------
77   // Analysis algorithm
78   //---------------------------------------------------------------------
79   
80   //Detector Fiducial Cuts for analysis part
81   AliFiducialCut * fidCut2 = new AliFiducialCut();
82   fidCut2->DoCTSFiducialCut(kFALSE) ;
83   fidCut2->DoEMCALFiducialCut(kFALSE) ;
84   fidCut2->DoPHOSFiducialCut(kTRUE) ;
85   fidCut2->SetSimplePHOSFiducialCut(0.1,240.,280.);
86
87   AliCaloPID * pid = new AliCaloPID();
88   // use selection with simple weights
89   pid->SetPHOSPhotonWeight(0.7);    pid->SetPHOSPi0Weight(0.7); 
90   pid->SetEMCALPhotonWeight(0.7);    pid->SetEMCALPi0Weight(0.7);
91   // use more complicated selection, particle weight depending on cluster energy
92 //   pid->UsePHOSPIDWeightFormula(kTRUE);
93 //   TFormula * photonF = new TFormula("photonWeight","0.98*(x<40)+ 0.68*(x>=100)+(x>=40 && x<100)*(0.98+x*(6e-3)-x*x*(2e-04)+x*x*x*(1.1e-06))");
94 //   TFormula * pi0F = new TFormula("pi0Weight","0.98*(x<65)+ 0.915*(x>=100)+(x>=65 && x-x*(1.95e-3)-x*x*(4.31e-05)+x*x*x*(3.61e-07))");
95 //   pid->SetPHOSPhotonWeightFormula(photonF);
96 //   pid->SetPHOSPi0WeightFormula(pi0F);
97
98
99   AliAnaExample *ana = new AliAnaExample();
100   ana->SetDebug(-1);
101   ana->SetCaloPID(pid);
102   ana->SetFiducialCut(fidCut2);
103   ana->SetDetector("PHOS");
104   //ana->SwitchOnDataMC();
105   ana->SetMinPt(0.);
106   ana->SetOutputAODName("Example");
107   ana->SetOutputAODClassName("AliAODPWG4Particle"); //Or AliAODPWG4ParticleCorrelation
108   //Set Histrograms bins and ranges
109 //      ana->SetHistoPtRangeAndNBins(0, 50, 100) ;
110 //      ana->SetHistoPhiRangeAndNBins(0, TMath::TwoPi(), 100) ;
111 //      ana->SetHistoEtaRangeAndNBins(-0.7, 0.7, 100) ;  
112   
113   //---------------------------------------------------------------------
114   // Set  analysis algorithm and reader
115   //---------------------------------------------------------------------
116   maker = new AliAnaPartCorrMaker();
117   maker->SetReader(reader);//pointer to reader
118   maker->AddAnalysis(ana,0);
119   maker->SetAnaDebug(-1)  ;
120   maker->SwitchOnHistogramsMaker()  ;  
121   maker->SwitchOnAODsMaker()  ;
122   
123   maker->Print("");
124   //
125   printf("======================== \n");
126   printf("END ConfigAnalysis() \n");
127   printf("======================== \n");
128   return maker ;
129 }