1 void AddRsnPairsKStar(AliAnalysisTaskSE *task,
4 AliPID::EParticleType pType1,
6 AliPID::EParticleType pType2,
8 AliRsnCutSet *cutsEvent=0,
9 AliRsnCutSet *cutsPair=0,
10 TString suffix = "") {
12 Printf("id1=%d id2=%d",listID1,listID2);
14 // retrieve mass from PDG database
16 TDatabasePDG *db = TDatabasePDG::Instance();
17 TParticlePDG *part = db->GetParticle(pdg);
18 Double_t mass = part->Mass();
21 Int_t isRsnMini = AliAnalysisManager::GetGlobalInt("rsnUseMiniPackage",valid);
23 AddPairOutputMiniKStar(task,isMC,isMixing,pType1,listID1,pType2,listID2,pdg,mass,cutsPair,suffix);
25 // this function is common and it is located in RsnConfig.C
26 // as ouptup AddPairOutputKStar from this macro will be taken
27 AddPair(task,isMC,isMixing,pType1,listID1,pType2,listID2,pdg,mass,cutsEvent,cutsPair,suffix);
30 void AddPairOutputKStar(AliRsnLoopPair *pair)
33 Int_t isFullOutput = AliAnalysisManager::GetGlobalInt("rsnOutputFull",valid);
36 AliRsnValuePair *axisIM = new AliRsnValuePair("IM", AliRsnValuePair::kInvMass);
37 AliRsnValuePair *axisPt = new AliRsnValuePair("PT", AliRsnValuePair::kPt);
38 axisIM ->SetBins(900, 0.6, 1.5);
39 axisPt ->SetBins(120, 0.0, 12.0);
41 // output: 2D histogram of inv. mass vs. pt
42 AliRsnListOutput *outPair = 0;
44 outPair = new AliRsnListOutput("pair", AliRsnListOutput::kHistoDefault);
45 outPair->AddValue(axisIM);
47 outPair = new AliRsnListOutput("pair", AliRsnListOutput::kHistoSparse);
48 outPair->AddValue(axisIM);
49 outPair->AddValue(axisPt);
51 // add outputs to loop
52 pair->AddOutput(outPair);
55 void AddPairOutputMiniKStar(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 = "") {
58 Int_t isFullOutput = AliAnalysisManager::GetGlobalInt("rsnOutputFull",valid);
59 Int_t useMixing = AliAnalysisManager::GetGlobalInt("rsnUseMixing",valid);
60 AliRsnMiniAnalysisTask *taskRsnMini = (AliRsnMiniAnalysisTask *)task;
61 /* invariant mass */ Int_t imID = taskRsnMini->CreateValue(AliRsnMiniValue::kInvMass, kFALSE);
62 /* IM resolution */ Int_t resID = taskRsnMini->CreateValue(AliRsnMiniValue::kInvMassRes, kTRUE);
63 /* transv. momentum */ Int_t ptID = taskRsnMini->CreateValue(AliRsnMiniValue::kPt, kFALSE);
64 /* centrality */ Int_t centID = taskRsnMini->CreateValue(AliRsnMiniValue::kMult, kFALSE);
67 // -- Create all needed outputs -----------------------------------------------------------------
70 Int_t iCutK = listID1;
71 Int_t iCutPi = listID2;
74 TString outputType = "HIST";
75 if (isFullOutput) outputType = "SPARSE";
77 // use an array for more compact writing, which are different on mixing and charges
82 Bool_t use [10] = { 1 , 1 , useMixing , useMixing , 1 , 1 , isMC , isMC , isMC , isMC };
83 Bool_t useIM [10] = { 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 0 , 0 };
84 TString name [10] = {"Unlike1", "Unlike2", "Mixing1", "Mixing2", "LikePP", "LikeMM", "Trues1", "Trues2", "Res1" , "Res2" };
85 TString comp [10] = {"PAIR" , "PAIR" , "MIX" , "MIX" , "PAIR" , "PAIR" , "TRUE" , "TRUE" , "TRUE" , "TRUE" };
86 Char_t charge1 [10] = {'+' , '-' , '+' , '-' , '+' , '-' , '+' , '-' , '+' , '-' };
87 Char_t charge2 [10] = {'-' , '+' , '-' , '+' , '+' , '-' , '-' , '+' , '-' , '+' };
88 Int_t cutID1 [10] = { iCutK , iCutK , iCutK , iCutK , iCutK , iCutK , iCutK , iCutK , iCutK , iCutK };
89 Int_t cutID2 [10] = { iCutPi , iCutPi , iCutPi , iCutPi , iCutPi , iCutPi , iCutPi , iCutPi , iCutPi , iCutPi };
91 for (Int_t i = 0; i < 10; i++) {
92 if (!use[i]) continue;
94 AliRsnMiniOutput *out = taskRsnMini->CreateOutput(Form("%s_%s", suffix.Data(), name[i].Data()), outputType.Data(), comp[i].Data());
96 out->SetCutID(0, cutID1[i]);
97 out->SetCutID(1, cutID2[i]);
98 out->SetDaughter(0, AliRsnDaughter::kKaon);
99 out->SetDaughter(1, AliRsnDaughter::kPion);
100 out->SetCharge(0, charge1[i]);
101 out->SetCharge(1, charge2[i]);
102 out->SetMotherPDG(pdgMother);
103 out->SetMotherMass(massMother);
105 if (cutsPair) out->SetPairCuts(cutsPair);
106 // axis X: invmass (or resolution)
108 out->AddAxis(imID, 90, 0.6, 1.5);
110 out->AddAxis(resID, 200, -0.02, 0.02);
113 // axis Y: transverse momentum
114 out->AddAxis(ptID, 100, 0.0, 10.0);
115 // axis Z: centrality
116 out->AddAxis(centID, 100, 0.0, 100.0);