Updated cut classes and macro for Sigma* analysis (M. Venaruzzo)
[u/mrichter/AliRoot.git] / PWGLF / RESONANCES / macros / lego_train / old / AddRsnPairsRho.C
1 #ifndef __CINT__
2 #include <AliRsnValuePair.h>
3 #endif
4
5 void AddRsnPairsRho(AliAnalysisTaskSE *task,
6                     Bool_t isMC,
7                     Bool_t isMixing,
8                     AliPID::EParticleType pType1,
9                     Int_t listID1,
10                     AliPID::EParticleType pType2,
11                     Int_t listID2,
12                     AliRsnCutSet *cutsEvent=0,
13                     AliRsnCutSet *cutsPair=0,
14                     TString suffix = "") {
15
16    Printf("id1=%d id2=%d",listID1,listID2);
17
18    // retrieve mass from PDG database
19    Int_t         pdg  = 113;
20    TDatabasePDG *db   = TDatabasePDG::Instance();
21    TParticlePDG *part = db->GetParticle(pdg);
22    Double_t mass = part->Mass();
23    Bool_t valid;
24    Int_t isRsnMini = AliAnalysisManager::GetGlobalInt("rsnUseMiniPackage",valid);
25
26    if (isRsnMini) {
27       AddPairOutputMiniRho(task,isMC,isMixing,pType1,listID1,pType2,listID2,pdg,mass,cutsPair,suffix);
28    } else {
29       // this function is common and it is located in RsnConfig.C
30       // as ouptup AddPairOutputPhi from this macro will be taken
31       AddPair(task,isMC,isMixing,pType1,listID1,pType2,listID2,pdg,mass,cutsEvent,cutsPair,suffix);
32    }
33 }
34 void AddPairOutputRho(AliRsnLoopPair *pair)
35 {
36    Bool_t valid;
37    Int_t isFullOutput = AliAnalysisManager::GetGlobalInt("rsnOutputFull",valid);
38    // axes
39    AliRsnValuePair *axisIM = new AliRsnValuePair("IM", AliRsnValuePair::kInvMass);
40    AliRsnValuePair *axisPt = new AliRsnValuePair("PT", AliRsnValuePair::kPt);
41    AliRsnValuePair *axisEta = new AliRsnValuePair("ETA", AliRsnValuePair::kEta);
42    axisIM     ->SetBins(1000, 0.2, 1.2);
43 //   axisIM     ->SetBins(1000, 0.9, 1.9);
44    axisPt     ->SetBins(120, 0.0, 12.0);
45    axisEta    ->SetBins(400, -2.0, 2.0);
46
47    // output: 2D histogram of inv. mass vs. pt
48    AliRsnListOutput *outPair = 0;
49    if (!isFullOutput) {
50       outPair = new AliRsnListOutput("pair", AliRsnListOutput::kHistoDefault);
51       outPair->AddValue(axisIM);
52    } else {
53       outPair = new AliRsnListOutput("pair", AliRsnListOutput::kHistoSparse);
54       outPair->AddValue(axisIM);
55       outPair->AddValue(axisPt);
56       outPair->AddValue(axisEta);
57    }
58    // add outputs to loop
59    pair->AddOutput(outPair);
60 }
61
62 void AddPairOutputMiniRho(AliAnalysisTaskSE *task, Bool_t isMC,Bool_t isMixing, AliPID::EParticleType pType1,Int_t listID1, AliPID::EParticleType pType2,Int_t listID2, Int_t pdgMother,Double_t massMother, AliRsnCutSet *cutsPair=0,TString suffix = "") {
63
64    Bool_t valid;
65    Int_t isFullOutput = AliAnalysisManager::GetGlobalInt("rsnOutputFull",valid);
66    Int_t useMixing = AliAnalysisManager::GetGlobalInt("rsnUseMixing",valid);
67    Int_t isPP = AliAnalysisManager::GetGlobalInt("rsnIsPP",valid);
68
69    AliRsnMiniAnalysisTask *taskRsnMini =  (AliRsnMiniAnalysisTask *)task;
70
71    /* invariant mass   */ Int_t imID   = taskRsnMini->CreateValue(AliRsnMiniValue::kInvMass, kFALSE);
72    /* IM resolution    */ Int_t resID  = taskRsnMini->CreateValue(AliRsnMiniValue::kInvMassDiff, kTRUE);
73    /* transv. momentum */ Int_t ptID   = taskRsnMini->CreateValue(AliRsnMiniValue::kPt, kFALSE);
74    /* centrality       */ Int_t centID = taskRsnMini->CreateValue(AliRsnMiniValue::kMult, kFALSE);
75    /* eta              */ Int_t etaID = taskRsnMini->CreateValue(AliRsnMiniValue::kEta, kFALSE);
76
77
78    // use an array for more compact writing, which are different on mixing and charges
79    // [0] = unlike
80    // [1] = mixing
81    // [2] = like ++
82    // [3] = like --
83    Bool_t  use     [5] = { 1      ,  useMixing      ,  1      ,  1      ,  isMC  };
84    TString name    [5] = {"Unlike", "Mixing", "LikePP", "LikeMM", "Trues"};
85    TString comp    [5] = {"PAIR"  , "MIX"   , "PAIR"  , "PAIR"  , "TRUE" };
86    Char_t  charge1 [5] = {'+'     , '+'     , '+'     , '-'     , '+'    };
87    Char_t  charge2 [5] = {'-'     , '-'     , '+'     , '-'     , '-'    };
88
89    // common definitions
90    TString outputType = "HIST";
91    if (isFullOutput) outputType = "SPARSE";
92
93    Int_t nIM   = 1000; Double_t minIM   = 0.2, maxIM =  1.2;
94    Int_t nEta   = 400; Double_t minEta   = -2.0, maxEta =  2.0;
95 //   Int_t nIM   = 1000; Double_t minIM   = 0.9, maxIM =  1.9;
96    Int_t nPt   = 120; Double_t minPt   = 0.0, maxPt = 12.0;
97    Int_t nCent = 100; Double_t minCent = 0.0, maxCent = 100.0;
98    Int_t nRes  = 200; Double_t maxRes  = 0.01;
99
100    // retrieve mass from PDG database
101    Int_t         pdg  = 113;
102    TDatabasePDG *db   = TDatabasePDG::Instance();
103    TParticlePDG *part = db->GetParticle(pdg);
104
105
106    Printf(suffix.Data());
107    // create standard outputs
108    for (Int_t i = 0; i < 5; i++) {
109       if (!use[i]) continue;
110       // create output
111       AliRsnMiniOutput *out = taskRsnMini->CreateOutput(Form("%s_%s", suffix.Data(),name[i].Data() ), outputType.Data(), comp[i].Data());
112       // selection settings
113       out->SetCutID(0, listID1);
114       out->SetCutID(1, listID1);
115       out->SetDaughter(0, AliRsnDaughter::kPion);
116       out->SetDaughter(1, AliRsnDaughter::kPion);
117       out->SetCharge(0, charge1[i]);
118       out->SetCharge(1, charge2[i]);
119       out->SetMotherPDG(pdg);
120       out->SetMotherMass(part->Mass());
121       // pair cuts
122       if (cutsPair) out->SetPairCuts(cutsPair);
123       // axis X: invmass
124       out->AddAxis(imID, nIM, minIM, maxIM);
125
126       if (isFullOutput) {
127          // axis Y: transverse momentum
128          out->AddAxis(ptID, nPt, minPt, maxPt);
129
130          out->AddAxis(etaID, nEta, minEta, maxEta);
131          // axis Z: centrality
132          if (!isPP) out->AddAxis(centID, nCent, minCent, maxCent);
133       }
134    }
135
136    // add output for resolution
137    if (isMC) {
138       AliRsnMiniOutput *outRes = taskRsnMini->CreateOutput(Form("rho_Res%s", suffix.Data()), outputType.Data(), "TRUE");
139       // selection settings
140       outRes->SetCutID(0, listID1);
141       outRes->SetCutID(1, listID1);
142       outRes->SetDaughter(0, AliRsnDaughter::kPion);
143       outRes->SetDaughter(1, AliRsnDaughter::kPion);
144       outRes->SetCharge(0, '+');
145       outRes->SetCharge(1, '-');
146       outRes->SetMotherPDG(pdg);
147       outRes->SetMotherMass(part->Mass());
148       // pair cuts
149       if (cutsPair) outRes->SetPairCuts(cutsPair);
150       // axis X: resolution
151       outRes->AddAxis(resID, nRes, -maxRes, maxRes);
152
153       if (isFullOutput) {
154          // axis Y: transverse momentum
155          outRes->AddAxis(ptID, nPt, minPt, maxPt);
156          outRes->AddAxis(etaID, nEta, minEta, maxEta);
157          // axis Z: centrality
158          if (!isPP) outRes->AddAxis(centID, nCent, minCent, maxCent);
159       }
160    }
161
162    //
163    // -- Create output for MC generated ------------------------------------------------------------
164    //
165
166    if (isMC) {
167       // create ouput
168       AliRsnMiniOutput *outMC = taskRsnMini->CreateOutput(Form("rho_MCGen%s", suffix.Data()), outputType.Data(), "MOTHER");
169       // selection settings
170       outMC->SetDaughter(0, AliRsnDaughter::kPion);
171       outMC->SetDaughter(1, AliRsnDaughter::kPion);
172       outMC->SetMotherPDG(pdg);
173       outMC->SetMotherMass(part->Mass());
174       // pair cuts
175       if (cutsPair) outMC->SetPairCuts(cutsPair);
176       // axis X: invmass
177       outMC->AddAxis(imID, nIM, minIM, maxIM);
178       if (isFullOutput) {
179          // axis Y: transverse momentum
180          outMC->AddAxis(ptID, nPt, minPt, maxPt);
181          outMC->AddAxis(etaID, nEta, minEta, maxEta);
182          // axis Z: centrality
183          if (!isPP) outMC->AddAxis(centID, nCent, minCent, maxCent);
184       }
185    }
186
187
188 }