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

#ifndef __CINT__
#include <AliRsnListOutput.h>
#include <AliRsnValuePair.h>
#include <AliRsnValueEvent.h>
#include <AliRsnMiniAnalysisTask.h>
#endif

void AddRsnPairsPhi(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  = 333;
   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) {
      AddPairOutputMiniPhi(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 AddPairOutputPhi from this macro will be taken
      AddPair(task,isMC,isMixing,pType1,listID1,pType2,listID2,pdg,mass,cutsEvent,cutsPair,suffix);
   }
}
void AddPairOutputPhi(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);
   AliRsnValuePair *axisPt = new AliRsnValuePair("PT", AliRsnValuePair::kPt);
   AliRsnValuePair *axisEta = new AliRsnValuePair("ETA", AliRsnValuePair::kEta);
   AliRsnValuePair *axisRapidity = new AliRsnValuePair("Y", AliRsnValuePair::kY);

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


   axisIM     ->SetBins(300, 0.9, 1.2);
//   axisIM     ->SetBins(600, 0.9, 1.2);
//   axisIM     ->SetBins(1000, 0.9, 1.9);
   axisPt     ->SetBins(120, 0.0, 12.0);
//    axisEta    ->SetBins(400, -2.0, 2.0);
   axisEta    ->SetBins(400, -0.5, 0.5);
   axisRapidity    ->SetBins(10, -0.5, 0.5);

   if (axisCentrality) axisCentrality->SetBins(20,0,100);

   // 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);
      if (useRapidity) outPair->AddValue(axisRapidity);
      else outPair->AddValue(axisEta);
      if (axisCentrality) outPair->AddValue(axisCentrality);
   }
   // add outputs to loop
   pair->AddOutput(outPair);
}

void AddPairOutputMiniPhi(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);

   Int_t useRapidity = AliRsnTrainManager::GetGlobalInt("RsnUseRapidity",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::kInvMassDiff, 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);


   // use an array for more compact writing, which are different on mixing and charges
   // [0] = unlike
   // [1] = mixing
   // [2] = like ++
   // [3] = like --
   Bool_t  use     [5] = { 1      ,  useMixing      ,  1      ,  1      ,  isMC  };
   TString name    [5] = {"Unlike", "Mixing", "LikePP", "LikeMM", "Trues"};
   TString comp    [5] = {"PAIR"  , "MIX"   , "PAIR"  , "PAIR"  , "TRUE" };
   Char_t  charge1 [5] = {'+'     , '+'     , '+'     , '-'     , '+'    };
   Char_t  charge2 [5] = {'-'     , '-'     , '+'     , '-'     , '-'    };

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

   Int_t nIM   = 215; Double_t minIM   = 0.985, maxIM =  1.2;
//    Int_t nEta   = 400; Double_t minEta   = -2.0, maxEta =  2.0;
   Int_t nEta   = 400; Double_t minEta   = -0.5, maxEta =  0.5;
   Int_t nY   = 1; Double_t minY   = -0.5, maxY =  0.5;
//   Int_t nIM   = 1000; Double_t minIM   = 0.9, maxIM =  1.9;
   Int_t nPt   = 120; Double_t minPt   = 0.0, maxPt = 12.0;
   if (collisionType==0) nPt = 240;minPt=0.0;maxPt=24.0;
   Int_t nCent = 20; Double_t minCent = 0.0, maxCent = 100.0;
   Int_t nRes  = 200; Double_t maxRes  = 0.01;

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


   Printf(suffix.Data());
   // create standard outputs
   for (Int_t i = 0; i < 5; 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, listID1);
      out->SetCutID(1, listID1);
      out->SetDaughter(0, AliRsnDaughter::kKaon);
      out->SetDaughter(1, AliRsnDaughter::kKaon);
      out->SetCharge(0, charge1[i]);
      out->SetCharge(1, charge2[i]);
      out->SetMotherPDG(pdg);
      out->SetMotherMass(part->Mass());
      // pair cuts
      if (cutsPair) out->SetPairCuts(cutsPair);
      // axis X: invmass
      out->AddAxis(imID, nIM, minIM, maxIM);

      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);
      }
   }

   // add output for resolution
   if (isMC) {
      AliRsnMiniOutput *outRes = taskRsnMini->CreateOutput(Form("phi_Res%s", suffix.Data()), outputType.Data(), "TRUE");
      // selection settings
      outRes->SetCutID(0, listID1);
      outRes->SetCutID(1, listID1);
      outRes->SetDaughter(0, AliRsnDaughter::kKaon);
      outRes->SetDaughter(1, AliRsnDaughter::kKaon);
      outRes->SetCharge(0, '+');
      outRes->SetCharge(1, '-');
      outRes->SetMotherPDG(pdg);
      outRes->SetMotherMass(part->Mass());
      // pair cuts
      if (cutsPair) outRes->SetPairCuts(cutsPair);
      // axis X: resolution
      outRes->AddAxis(resID, nRes, -maxRes, maxRes);

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

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

   if (isMC) {
      // create ouput
      AliRsnMiniOutput *outMC = taskRsnMini->CreateOutput(Form("phi_MCGen%s", suffix.Data()), outputType.Data(), "MOTHER");
      // selection settings
      outMC->SetDaughter(0, AliRsnDaughter::kKaon);
      outMC->SetDaughter(1, AliRsnDaughter::kKaon);
      outMC->SetMotherPDG(pdg);
      outMC->SetMotherMass(part->Mass());
      // 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);
      }
   }


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