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

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