[u/mrichter/AliRoot.git] / PWGLF / RESONANCES / macros / lego_train / AddRsnPairsKStar.C

void AddRsnPairsKStar(AliAnalysisTaskSE *task,
                      Bool_t isMC,
                      Bool_t isMixing,
                      AliPID::EParticleType pType1,
                      Int_t listID1,
                      AliPID::EParticleType pType2,
                      Int_t listID2,
                      AliRsnCutSet *cutsEvent=0,
                      AliRsnCutSet *cutsPair=0,
                      TString suffix = "") {

   Printf("id1=%d id2=%d",listID1,listID2);

   // retrieve mass from PDG database
   Int_t         pdg  = 313;
   TDatabasePDG *db   = TDatabasePDG::Instance();
   TParticlePDG *part = db->GetParticle(pdg);
   Double_t mass = part->Mass();

   Bool_t valid;
   Int_t isRsnMini = AliRsnTrainManager::GetGlobalInt("IsRsnMini",valid);

   if (isRsnMini) {
      AddPairOutputMiniKStar(task,isMC,isMixing,pType1,listID1,pType2,listID2,pdg,mass,cutsPair,suffix);
   } else {
      // this function is common and it is located in RsnConfig.C
      // as ouptup AddPairOutputKStar from this macro will be taken
      AddPair(task,isMC,isMixing,pType1,listID1,pType2,listID2,pdg,mass,cutsEvent,cutsPair,suffix);
   }
}
void AddPairOutputKStar(AliRsnLoopPair *pair)
{
   Bool_t valid;

   Int_t isFullOutput = AliRsnTrainManager::GetGlobalInt("RsnOutputFull",valid);
   Int_t collisionType = AliRsnTrainManager::GetGlobalInt("IsCollisionType",valid);
   Int_t useRapidity = AliRsnTrainManager::GetGlobalInt("RsnUseRapidity",valid);

   // axes
   AliRsnValuePair *axisIM = new AliRsnValuePair("IM", AliRsnValuePair::kInvMass);
   axisIM     ->SetBins(900, 0.6, 1.5);

   AliRsnValuePair *axisPt = new AliRsnValuePair("PT", AliRsnValuePair::kPt);
   axisPt     ->SetBins(120, 0.0, 12.0);

   AliRsnValuePair *axisEta = new AliRsnValuePair("ETA", AliRsnValuePair::kEta);
   axisEta    ->SetBins(400, -0.5, 0.5);

   AliRsnValueEvent *axisCentrality = 0;
   if (collisionType==1) axisCentrality = new AliRsnValueEvent("MULTI",AliRsnValueEvent::kCentralityV0);

   // output: 2D histogram of inv. mass vs. pt
   AliRsnListOutput *outPair = 0;
   if (!isFullOutput) {
      outPair = new AliRsnListOutput("pair", AliRsnListOutput::kHistoDefault);
      outPair->AddValue(axisIM);
   } else {
      outPair = new AliRsnListOutput("pair", AliRsnListOutput::kHistoSparse);
      outPair->AddValue(axisIM);
      outPair->AddValue(axisPt);
      outPair->AddValue(axisEta);
      if (axisCentrality) axisCentrality->SetBins(20,0,100);
   }
   // add outputs to loop
   pair->AddOutput(outPair);
}

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 = "") {

   Bool_t valid;
   Int_t isFullOutput = AliRsnTrainManager::GetGlobalInt("RsnOutputFull",valid);
   Int_t useMixing = AliRsnTrainManager::GetGlobalInt("IsMixing",valid);
   Int_t collisionType = AliRsnTrainManager::GetGlobalInt("IsCollisionType",valid);

   AliRsnMiniAnalysisTask *taskRsnMini =  (AliRsnMiniAnalysisTask *)task;

   if (collisionType==0) taskRsnMini->UseMultiplicity("QUALITY");
   else {
      taskRsnMini->UseCentrality("V0M");
      Int_t multID = taskRsnMini->CreateValue(AliRsnMiniValue::kMult, kFALSE);
      AliRsnMiniOutput *outMult = taskRsnMini->CreateOutput("eventMult", "HIST", "EVENT");
      outMult->AddAxis(multID, 100, 0.0, 100.0);
      Int_t paID = taskRsnMini->CreateValue(AliRsnMiniValue::kPlaneAngle, kFALSE);
      AliRsnMiniOutput *outPa = taskRsnMini->CreateOutput("planeAngle", "HIST", "EVENT");
      outPa->AddAxis(paID, 100, 0, TMath::Pi());
   }

   /* invariant mass   */ Int_t imID   = taskRsnMini->CreateValue(AliRsnMiniValue::kInvMass, kFALSE);
   /* IM resolution    */ Int_t resID  = taskRsnMini->CreateValue(AliRsnMiniValue::kInvMassRes, kTRUE);
   /* transv. momentum */ Int_t ptID   = taskRsnMini->CreateValue(AliRsnMiniValue::kPt, kFALSE);
   /* centrality       */ Int_t centID = taskRsnMini->CreateValue(AliRsnMiniValue::kMult, kFALSE);
   /* eta              */ Int_t etaID = taskRsnMini->CreateValue(AliRsnMiniValue::kEta, kFALSE);
   /* rapidity         */ Int_t yID = taskRsnMini->CreateValue(AliRsnMiniValue::kY, kFALSE);

   Bool_t useRapidity = kTRUE;

   Int_t nIM   = 90; Double_t minIM   = 0.6, maxIM =  1.5;
   Int_t nRes   = 200; Double_t minRes   = -0.02, maxRes =  0.02;
   Int_t nEta   = 400; Double_t minEta   = -0.5, maxEta =  0.5;
   Int_t nY   = 16; Double_t minY   = -0.8, maxY =  0.8;
   Int_t nPt   = 120; Double_t minPt   = 0.0, maxPt = 12.0;
   Int_t nCent = 100; Double_t minCent = 0.0, maxCent = 100.0;
   //
   // -- Create all needed outputs -----------------------------------------------------------------
   //

   Int_t iCutK = listID1;
   Int_t iCutPi = listID2;

   // common definitions
   TString outputType = "HIST";
   if (isFullOutput) outputType = "SPARSE";

   // use an array for more compact writing, which are different on mixing and charges
   // [0] = unlike
   // [1] = mixing
   // [2] = like ++
   // [3] = like --
//    const Int_t num = 12;
   Bool_t  use     [12] = { 1       ,  1       , useMixing, useMixing,  1      ,  1      ,  isMC   ,   isMC   ,  isMC   ,   isMC   ,  isMC    ,   isMC     };
   Bool_t  useIM   [12] = { 1       ,  1       ,  1       ,  1       ,  1      ,  1      ,  1      ,   1      ,  0      ,   0      ,  1       ,   1        };
   TString name    [12] = {"Unlike1", "Unlike2", "Mixing1", "Mixing2", "LikePP", "LikeMM", "Trues1",  "Trues2", "Res1"  ,  "Res2"  , "Mother1",  "Mother2" };
   TString comp    [12] = {"PAIR"   , "PAIR"   , "MIX"    , "MIX"    , "PAIR"  , "PAIR"  , "TRUE"  ,  "TRUE"  , "TRUE"  ,  "TRUE"  , "MOTHER" ,  "MOTHER"  };
   Char_t  charge1 [12] = {'+'      , '-'      , '+'      , '-'      , '+'     , '-'     , '+'     ,  '-'     , '+'     ,  '-'     , '+'      ,  '-'       };
   Char_t  charge2 [12] = {'-'      , '+'      , '-'      , '+'      , '+'     , '-'     , '-'     ,  '+'     , '-'     ,  '+'     , '-'      ,  '+'       };
   Int_t   cutID1  [12] = { iCutK   ,  iCutK   ,  iCutK   ,  iCutK   ,  iCutK  ,  iCutK  ,  iCutK  ,   iCutK  ,  iCutK  ,   iCutK  ,  iCutK  ,   iCutK     };
   Int_t   cutID2  [12] = { iCutPi  ,  iCutPi  ,  iCutPi  ,  iCutPi  ,  iCutPi ,  iCutPi ,  iCutPi ,   iCutPi ,  iCutPi ,   iCutPi ,  iCutPi ,   iCutPi    };

   for (Int_t i = 0; i < 12; i++) {
      if (!use[i]) continue;
      // create output
      AliRsnMiniOutput *out = taskRsnMini->CreateOutput(Form("%s_%s", suffix.Data(), name[i].Data()), outputType.Data(), comp[i].Data());
      // selection settings
      out->SetCutID(0, cutID1[i]);
      out->SetCutID(1, cutID2[i]);
      out->SetDaughter(0, AliRsnDaughter::kKaon);
      out->SetDaughter(1, AliRsnDaughter::kPion);
      out->SetCharge(0, charge1[i]);
      out->SetCharge(1, charge2[i]);
      out->SetMotherPDG(pdgMother);
      out->SetMotherMass(massMother);
      // pair cuts
      if (cutsPair) out->SetPairCuts(cutsPair);
      // axis X: invmass (or resolution)
      if (useIM[i])
         out->AddAxis(imID, nIM, minIM, maxIM);
      else
         out->AddAxis(resID, nRes, minRes, maxRes);

      if (isFullOutput) {
         // axis Y: transverse momentum
         out->AddAxis(ptID, nPt, minPt, maxPt);
         if (useRapidity) out->AddAxis(yID, nY, minY, maxY);
         else  out->AddAxis(etaID, nEta, minEta, maxEta);
         // axis Z: centrality
         if (collisionType==1) out->AddAxis(centID, nCent, minCent, maxCent);
      }
   }

   // -- Create output for MC generated ------------------------------------------------------------
   //

   if (isMC) {
      // create ouput
      AliRsnMiniOutput *outMC = taskRsnMini->CreateOutput(Form("kstar_MCGen1%s", suffix.Data()), outputType.Data(), "MOTHER");
      // selection settings
      outMC->SetDaughter(0, AliRsnDaughter::kPion);
      outMC->SetDaughter(1, AliRsnDaughter::kKaon);
      outMC->SetMotherPDG(313);
      outMC->SetMotherMass(massMother);
      // pair cuts
      if (cutsPair) outMC->SetPairCuts(cutsPair);
      // axis X: invmass
      outMC->AddAxis(imID, nIM, minIM, maxIM);
      if (isFullOutput) {
         // axis Y: transverse momentum
         outMC->AddAxis(ptID, nPt, minPt, maxPt);
         if (useRapidity) outMC->AddAxis(yID, nY, minY, maxY);
         else  outMC->AddAxis(etaID, nEta, minEta, maxEta);
         // axis Z: centrality
         if (collisionType==1) outMC->AddAxis(centID, nCent, minCent, maxCent);
      }
   }


   if (isMC) {
      // create ouput
      AliRsnMiniOutput *outMC1 = taskRsnMini->CreateOutput(Form("phi_MCGen2%s", suffix.Data()), outputType.Data(), "MOTHER");
      // selection settings
      outMC1->SetDaughter(0, AliRsnDaughter::kKaon);
      outMC1->SetDaughter(1, AliRsnDaughter::kPion);
      outMC1->SetMotherPDG(-313);
      outMC1->SetMotherMass(massMother);
      // pair cuts
      if (cutsPair) outMC1->SetPairCuts(cutsPair);
      // axis X: invmass
      outMC1->AddAxis(imID, nIM, minIM, maxIM);
      if (isFullOutput) {
         // axis Y: transverse momentum
         outMC1->AddAxis(ptID, nPt, minPt, maxPt);
         if (useRapidity) outMC1->AddAxis(yID, nY, minY, maxY);
         else  outMC1->AddAxis(etaID, nEta, minEta, maxEta);
         // axis Z: centrality
         if (collisionType==1) outMC1->AddAxis(centID, nCent, minCent, maxCent);
      }
   }


}
Commit	Line	Data
221dc8d3	1	void AddRsnPairsKStar(AliAnalysisTaskSE *task,
	2	Bool_t isMC,
	3	Bool_t isMixing,
	4	AliPID::EParticleType pType1,
	5	Int_t listID1,
	6	AliPID::EParticleType pType2,
	7	Int_t listID2,
	8	AliRsnCutSet *cutsEvent=0,
	9	AliRsnCutSet *cutsPair=0,
	10	TString suffix = "") {
	11
	12	Printf("id1=%d id2=%d",listID1,listID2);
	13
	14	// retrieve mass from PDG database
	15	Int_t pdg = 313;
	16	TDatabasePDG *db = TDatabasePDG::Instance();
	17	TParticlePDG *part = db->GetParticle(pdg);
	18	Double_t mass = part->Mass();
	19
	20	Bool_t valid;
	21	Int_t isRsnMini = AliRsnTrainManager::GetGlobalInt("IsRsnMini",valid);
e22320ab	22
221dc8d3	23	if (isRsnMini) {
	24	AddPairOutputMiniKStar(task,isMC,isMixing,pType1,listID1,pType2,listID2,pdg,mass,cutsPair,suffix);
	25	} else {
	26	// this function is common and it is located in RsnConfig.C
	27	// as ouptup AddPairOutputKStar from this macro will be taken
	28	AddPair(task,isMC,isMixing,pType1,listID1,pType2,listID2,pdg,mass,cutsEvent,cutsPair,suffix);
	29	}
	30	}
	31	void AddPairOutputKStar(AliRsnLoopPair *pair)
	32	{
	33	Bool_t valid;
e22320ab	34
221dc8d3	35	Int_t isFullOutput = AliRsnTrainManager::GetGlobalInt("RsnOutputFull",valid);
	36	Int_t collisionType = AliRsnTrainManager::GetGlobalInt("IsCollisionType",valid);
	37	Int_t useRapidity = AliRsnTrainManager::GetGlobalInt("RsnUseRapidity",valid);
	38
	39	// axes
	40	AliRsnValuePair *axisIM = new AliRsnValuePair("IM", AliRsnValuePair::kInvMass);
	41	axisIM ->SetBins(900, 0.6, 1.5);
	42
	43	AliRsnValuePair *axisPt = new AliRsnValuePair("PT", AliRsnValuePair::kPt);
	44	axisPt ->SetBins(120, 0.0, 12.0);
	45
	46	AliRsnValuePair *axisEta = new AliRsnValuePair("ETA", AliRsnValuePair::kEta);
	47	axisEta ->SetBins(400, -0.5, 0.5);
	48
	49	AliRsnValueEvent *axisCentrality = 0;
	50	if (collisionType==1) axisCentrality = new AliRsnValueEvent("MULTI",AliRsnValueEvent::kCentralityV0);
	51
	52	// output: 2D histogram of inv. mass vs. pt
	53	AliRsnListOutput *outPair = 0;
	54	if (!isFullOutput) {
	55	outPair = new AliRsnListOutput("pair", AliRsnListOutput::kHistoDefault);
	56	outPair->AddValue(axisIM);
	57	} else {
	58	outPair = new AliRsnListOutput("pair", AliRsnListOutput::kHistoSparse);
	59	outPair->AddValue(axisIM);
	60	outPair->AddValue(axisPt);
	61	outPair->AddValue(axisEta);
	62	if (axisCentrality) axisCentrality->SetBins(20,0,100);
	63	}
	64	// add outputs to loop
	65	pair->AddOutput(outPair);
	66	}
	67
	68	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 = "") {
	69
	70	Bool_t valid;
	71	Int_t isFullOutput = AliRsnTrainManager::GetGlobalInt("RsnOutputFull",valid);
	72	Int_t useMixing = AliRsnTrainManager::GetGlobalInt("IsMixing",valid);
	73	Int_t collisionType = AliRsnTrainManager::GetGlobalInt("IsCollisionType",valid);
e22320ab	74
221dc8d3	75	AliRsnMiniAnalysisTask taskRsnMini = (AliRsnMiniAnalysisTask )task;
	76
	77	if (collisionType==0) taskRsnMini->UseMultiplicity("QUALITY");
	78	else {
	79	taskRsnMini->UseCentrality("V0M");
	80	Int_t multID = taskRsnMini->CreateValue(AliRsnMiniValue::kMult, kFALSE);
	81	AliRsnMiniOutput *outMult = taskRsnMini->CreateOutput("eventMult", "HIST", "EVENT");
	82	outMult->AddAxis(multID, 100, 0.0, 100.0);
	83	Int_t paID = taskRsnMini->CreateValue(AliRsnMiniValue::kPlaneAngle, kFALSE);
	84	AliRsnMiniOutput *outPa = taskRsnMini->CreateOutput("planeAngle", "HIST", "EVENT");
	85	outPa->AddAxis(paID, 100, 0, TMath::Pi());
	86	}
	87
	88	/* invariant mass */ Int_t imID = taskRsnMini->CreateValue(AliRsnMiniValue::kInvMass, kFALSE);
	89	/* IM resolution */ Int_t resID = taskRsnMini->CreateValue(AliRsnMiniValue::kInvMassRes, kTRUE);
	90	/* transv. momentum */ Int_t ptID = taskRsnMini->CreateValue(AliRsnMiniValue::kPt, kFALSE);
	91	/* centrality */ Int_t centID = taskRsnMini->CreateValue(AliRsnMiniValue::kMult, kFALSE);
	92	/* eta */ Int_t etaID = taskRsnMini->CreateValue(AliRsnMiniValue::kEta, kFALSE);
	93	/* rapidity */ Int_t yID = taskRsnMini->CreateValue(AliRsnMiniValue::kY, kFALSE);
	94
	95	Bool_t useRapidity = kTRUE;
	96
	97	Int_t nIM = 90; Double_t minIM = 0.6, maxIM = 1.5;
	98	Int_t nRes = 200; Double_t minRes = -0.02, maxRes = 0.02;
	99	Int_t nEta = 400; Double_t minEta = -0.5, maxEta = 0.5;
	100	Int_t nY = 16; Double_t minY = -0.8, maxY = 0.8;
	101	Int_t nPt = 120; Double_t minPt = 0.0, maxPt = 12.0;
	102	Int_t nCent = 100; Double_t minCent = 0.0, maxCent = 100.0;
	103	//
	104	// -- Create all needed outputs -----------------------------------------------------------------
	105	//
	106
	107	Int_t iCutK = listID1;
	108	Int_t iCutPi = listID2;
	109
	110	// common definitions
	111	TString outputType = "HIST";
	112	if (isFullOutput) outputType = "SPARSE";
	113
	114	// use an array for more compact writing, which are different on mixing and charges
	115	// [0] = unlike
	116	// [1] = mixing
	117	// [2] = like ++
	118	// [3] = like --
	119	// const Int_t num = 12;
	120	Bool_t use [12] = { 1 , 1 , useMixing, useMixing, 1 , 1 , isMC , isMC , isMC , isMC , isMC , isMC };
	121	Bool_t useIM [12] = { 1 , 1 , 1 , 1 , 1 , 1 , 1 , 1 , 0 , 0 , 1 , 1 };
	122	TString name [12] = {"Unlike1", "Unlike2", "Mixing1", "Mixing2", "LikePP", "LikeMM", "Trues1", "Trues2", "Res1" , "Res2" , "Mother1", "Mother2" };
	123	TString comp [12] = {"PAIR" , "PAIR" , "MIX" , "MIX" , "PAIR" , "PAIR" , "TRUE" , "TRUE" , "TRUE" , "TRUE" , "MOTHER" , "MOTHER" };
	124	Char_t charge1 [12] = {'+' , '-' , '+' , '-' , '+' , '-' , '+' , '-' , '+' , '-' , '+' , '-' };
	125	Char_t charge2 [12] = {'-' , '+' , '-' , '+' , '+' , '-' , '-' , '+' , '-' , '+' , '-' , '+' };
	126	Int_t cutID1 [12] = { iCutK , iCutK , iCutK , iCutK , iCutK , iCutK , iCutK , iCutK , iCutK , iCutK , iCutK , iCutK };
	127	Int_t cutID2 [12] = { iCutPi , iCutPi , iCutPi , iCutPi , iCutPi , iCutPi , iCutPi , iCutPi , iCutPi , iCutPi , iCutPi , iCutPi };
	128
	129	for (Int_t i = 0; i < 12; i++) {
	130	if (!use[i]) continue;
	131	// create output
	132	AliRsnMiniOutput *out = taskRsnMini->CreateOutput(Form("%s_%s", suffix.Data(), name[i].Data()), outputType.Data(), comp[i].Data());
	133	// selection settings
	134	out->SetCutID(0, cutID1[i]);
	135	out->SetCutID(1, cutID2[i]);
	136	out->SetDaughter(0, AliRsnDaughter::kKaon);
	137	out->SetDaughter(1, AliRsnDaughter::kPion);
	138	out->SetCharge(0, charge1[i]);
139	out->SetCharge(1, charge2[i]);
140	out->SetMotherPDG(pdgMother);
141	out->SetMotherMass(massMother);
142	// pair cuts
143	if (cutsPair) out->SetPairCuts(cutsPair);
144	// axis X: invmass (or resolution)
145	if (useIM[i])
146	out->AddAxis(imID, nIM, minIM, maxIM);
147	else
148	out->AddAxis(resID, nRes, minRes, maxRes);
149
150	if (isFullOutput) {
151	// axis Y: transverse momentum
152	out->AddAxis(ptID, nPt, minPt, maxPt);
153	if (useRapidity) out->AddAxis(yID, nY, minY, maxY);
154	else out->AddAxis(etaID, nEta, minEta, maxEta);
155	// axis Z: centrality
156	if (collisionType==1) out->AddAxis(centID, nCent, minCent, maxCent);
157	}
158	}
159
160	// -- Create output for MC generated ------------------------------------------------------------
161	//
162
163	if (isMC) {
164	// create ouput
165	AliRsnMiniOutput *outMC = taskRsnMini->CreateOutput(Form("kstar_MCGen1%s", suffix.Data()), outputType.Data(), "MOTHER");
166	// selection settings
167	outMC->SetDaughter(0, AliRsnDaughter::kPion);
168	outMC->SetDaughter(1, AliRsnDaughter::kKaon);
169	outMC->SetMotherPDG(313);
170	outMC->SetMotherMass(massMother);
171	// pair cuts
172	if (cutsPair) outMC->SetPairCuts(cutsPair);
173	// axis X: invmass
174	outMC->AddAxis(imID, nIM, minIM, maxIM);
175	if (isFullOutput) {
176	// axis Y: transverse momentum
177	outMC->AddAxis(ptID, nPt, minPt, maxPt);
178	if (useRapidity) outMC->AddAxis(yID, nY, minY, maxY);
179	else outMC->AddAxis(etaID, nEta, minEta, maxEta);
180	// axis Z: centrality
181	if (collisionType==1) outMC->AddAxis(centID, nCent, minCent, maxCent);
182	}
183	}
184
185
186
187	if (isMC) {
188	// create ouput
189	AliRsnMiniOutput *outMC1 = taskRsnMini->CreateOutput(Form("phi_MCGen2%s", suffix.Data()), outputType.Data(), "MOTHER");
190	// selection settings
191	outMC1->SetDaughter(0, AliRsnDaughter::kKaon);
192	outMC1->SetDaughter(1, AliRsnDaughter::kPion);
193	outMC1->SetMotherPDG(-313);
194	outMC1->SetMotherMass(massMother);
195	// pair cuts
196	if (cutsPair) outMC1->SetPairCuts(cutsPair);
197	// axis X: invmass
198	outMC1->AddAxis(imID, nIM, minIM, maxIM);
199	if (isFullOutput) {
200	// axis Y: transverse momentum
201	outMC1->AddAxis(ptID, nPt, minPt, maxPt);
202	if (useRapidity) outMC1->AddAxis(yID, nY, minY, maxY);
203	else outMC1->AddAxis(etaID, nEta, minEta, maxEta);
204	// axis Z: centrality
205	if (collisionType==1) outMC1->AddAxis(centID, nCent, minCent, maxCent);
206	}
207	}
208
209
210	}
211