[u/mrichter/AliRoot.git] / PWGJE / macros / AddTaskJetCorePP.C



AliAnalysisTaskJetCorePP* AddTaskJetCorePP(
   const Char_t* branchPrefix="clustersAOD",
   const Char_t* jetAlgo="ANTIKT",   
   Float_t jetParameterR = 0.4,  //jet R 
   UInt_t  trkFilterMask = 272, 
   Float_t trackLowPtCut = 0.15,
   const Char_t* jetbgAlgo="ANTIKT",    //background jet algo
   Float_t bgjetParameterR = 0.3,  //R of jet to be removed while bg calc 
   Float_t bgMaxJetPt = 8.0, //max jet pt to be accepted to bg 
   Int_t  rndTrials = 2000, //number of trials to get jet free cell area
   Float_t jetFreeAreaFrac = 0.75, //cell area free of jets
   Float_t bgConeR = 0.4,  //R of perp cone jet R 
   Int_t   collisionSystem = 0, //pp=0, pPb=1
   Int_t   offlineTriggerMask=AliVEvent::kMB, //MinBias=0 
   Int_t   minContribVtx = 1,
   Float_t vtxZMin = -10.0,
   Float_t vtxZMax = 10.0,
   Float_t centMin = 0.0,
   Float_t centMax = 100.0,
   Float_t triggerEtaCut = 0.9,
   Float_t trackEtaCut = 0.9,
   const Char_t* nonStdFile="",
   const Char_t* mcFullFlag="",  // real="", all jets= "MC"    
   const Char_t* mcChargFlag="",  // real="", charged jets = "MC2" 
   Bool_t bfillrespmx=0,  // 0=dont fill resp mx histos, 1=fill histos
   Bool_t bDiceEff=0,  // 0=leave efficiency as it is,  1= reduce efficiency by constant amount via SetFixedEfficiency
   Bool_t bMomSmear=0,  // 0=leave efficiency as it is,  1= reduce efficiency by constant amount via SetFixedEfficiency
   Bool_t bDoubleBinning=0,  // 0= 2GeV bin size  1= 1GeV/bin size
   Bool_t bUseExchContainer=0, //
   Int_t triggerType=0,  //0=single incl trigger, 1=leading track, 2=hadron pt>10 
   Int_t evtRangeLow=0,   //last digit of range of ESD event number
   Int_t evtRangeHigh=9,  //first digit of range of ESD event number
   Float_t trigRangeLow=0,  //trigger pT low bin boreder works with triggType=0
   Float_t trigRangeHigh=50  //trigger pT high border works with triggType=0
  ){ 
   Printf("adding task jet response\n");

   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
   if(!mgr){
      ::Error("AddTaskJetCorePP", "No analysis manager to connect to.");
      return NULL;
   }
   if(!mgr->GetInputEventHandler()){
      ::Error("AddTaskJetCorePP", "This task requires an input event handler.");
      return NULL;
   }

   Float_t jetEtaMin = -0.9 + jetParameterR;
   Float_t jetEtaMax =  0.9 - jetParameterR; 
    
   TString analBranch="";
   TString analBranchBg="";   //jet to be removed from bg 
   TString analBranchFullMC="";          //full jets MC to be used with charged jets MC 
   TString analBranchChargMC="";         //charged jets
   TString analBranchBgChargMC="";       //charged jets background 
   TString analBranchKine="";            //kine jets
   TString analBranchBgKine="";          //kine jets background 

   TString mcFullSuffix="";    //MC = all jets
   TString mcChargSuffix="";  //MC2= charged jets,  MC = all jets
 

   TString bpfx(branchPrefix);
   TString stJetAlgo(jetAlgo);
   TString stJetBgAlgo(jetbgAlgo);
   stJetAlgo.ToUpper();
   stJetBgAlgo.ToUpper();

   TString jet="";
   TString jetbg="";
   TString otherparams="";

   jet   = jet   + "_" + stJetAlgo   + Form("%02d",(Int_t) (10*jetParameterR));  // _ANTIKT02
   jetbg = jetbg + "_" + stJetBgAlgo + Form("%02d",(Int_t) (10*bgjetParameterR));
 
   if(!bpfx.Contains("KINE")){

      analBranch   = bpfx;
      analBranchBg = bpfx; 
 
      otherparams = otherparams + "_B0"; //bg mode
      otherparams = otherparams + Form("_Filter%05d",(UInt_t) trkFilterMask);
      otherparams = otherparams + Form("_Cut%05d",(Int_t) (1000*trackLowPtCut));

      if(analBranch.BeginsWith("clustersAOD")){
         otherparams = otherparams + Form("_Skip%02d",0);
      }
   
      analBranch   = analBranch   + jet   + otherparams; //antikt jet 
      analBranchBg = analBranchBg + jetbg + otherparams; //kt bg jet

      if(bDiceEff || bMomSmear){ //dicing efficiency relates rec only
         analBranch   = analBranch   + Form("Detector%d%dFr0",(Int_t) bDiceEff,(Int_t) bMomSmear); //dice=1, smear=0, change eff fraction =0
         analBranchBg = analBranchBg + Form("Detector%d%dFr0",(Int_t) bDiceEff,(Int_t) bMomSmear); 
      }

      //clustersAOD_ANTIKT04_B0_Filter00272_Cut00150_Skip00   
      //clustersAOD_ANTIKT04_B0_Filter00272_Cut00150_Skip00_Detector10Fr0   
      //Skip00 none of the most energetic jets is ommited
      //Cut00150  pT min cut on track
      //Filter00272
   
      mcFullSuffix=Form("%s",mcFullFlag);    //MC = all jets
      mcChargSuffix=Form("%s",mcChargFlag);  //MC2= charged jets,  MC = all jets
   
      if(mcChargSuffix.Length()>0 && mcChargSuffix=="MC2"){  //charged jets generator level
         analBranchChargMC   = bpfx + mcChargSuffix + jet   + otherparams; 
         analBranchBgChargMC = bpfx + mcChargSuffix + jetbg + otherparams; 
      }
   
      if(mcFullSuffix.Length()>0 && mcFullSuffix=="MC"){ //full jets generator level
         analBranchFullMC    = bpfx + mcFullSuffix  + jet   + otherparams; 
      }
   }else{ //KINE JETS
      //clustersKINECHARGED_ANTIKT02_Cut00150
      otherparams = otherparams + Form("_Cut%05d",TMath::Nint(1000*trackLowPtCut));

      analBranchKine   = analBranchKine   + bpfx + jet   + otherparams;            
      analBranchBgKine = analBranchBgKine + bpfx + jetbg + otherparams;         
   }

   TString analBranchTmp = analBranch;

   if(bpfx.Contains("KINE")){
      analBranchTmp=analBranchKine;
   }
   AliAnalysisTaskJetCorePP *task = new AliAnalysisTaskJetCorePP(Form("JetCorePP_%s_%s_%d",analBranchTmp.Data(), mcChargSuffix.Data(), offlineTriggerMask));

   task->SetBranchName(analBranch.Data());
   task->SetBranchNameFullMC(analBranchFullMC.Data());
   task->SetBranchNameChargMC(analBranchChargMC.Data());
   task->SetBranchNameKine(analBranchKine.Data());
   task->SetBranchNameBg(analBranchBg.Data()); //kine jets
   task->SetBranchNameBgChargMC(analBranchBgChargMC.Data()); //jet to be removed from bg 
   task->SetBranchNameBgKine(analBranchBgKine.Data()); //kine jets
   task->SetNonStdFile(nonStdFile);
   task->SetSystem(collisionSystem); 
   task->SetJetR(jetParameterR);
   task->SetBgJetR(bgjetParameterR);
   task->SetBgMaxJetPt(bgMaxJetPt);
   task->SetRndTrials(rndTrials);
   task->SetFreeAreaFrac(jetFreeAreaFrac);
   task->SetBgConeR(bgConeR); 
   task->SetOfflineTrgMask(offlineTriggerMask);
   task->SetMinContribVtx(minContribVtx);
   task->SetVtxZMin(vtxZMin);
   task->SetVtxZMax(vtxZMax);
   task->SetFilterMask(trkFilterMask);
   task->SetCentMin(centMin);
   task->SetCentMax(centMax);
   task->SetJetEtaMin(jetEtaMin);
   task->SetJetEtaMax(jetEtaMax);
   task->SetTriggerEtaCut(triggerEtaCut);
   task->SetTrackEtaCut(trackEtaCut);
   task->SetTrackLowPtCut(trackLowPtCut);
   task->SetTriggerType(triggerType); 
   task->SetEventNumberRangeLow(evtRangeLow);
   task->SetEventNumberRangeHigh(evtRangeHigh);
   task->SetTriggerPtRangeLow(trigRangeLow);
   task->SetTriggerPtRangeHigh(trigRangeHigh); 
   task->SetFillResponseMatrix(bfillrespmx);
   task->SetBinning(bDoubleBinning);
   task->SetUseExchangeContainerInput(bUseExchContainer);

   task->SetDebugLevel(0); //No debug messages 0
   mgr->AddTask(task);

   //E=  range of last two decimal numbers in event numbers
   //Ptt range of the cosidered trigger bin
   AliAnalysisDataContainer *coutputJetCorePP = mgr->CreateContainer(
      Form("pwgjejetcorepp_%s_%s%02d_%s_%d_T%d_E%d_%d_Ptt%.0f_%.0f",analBranchTmp.Data(),jetbgAlgo,TMath::Nint(10*bgjetParameterR),mcChargSuffix.Data(),offlineTriggerMask,triggerType,evtRangeLow,evtRangeHigh,trigRangeLow,trigRangeHigh), 
      TList::Class(),
      AliAnalysisManager::kOutputContainer,
      Form("%s:PWGJE_jetcorepp_%s_%s%02d_%s_%d_T%d_E%d_%d_Ptt%.0f_%.0f",AliAnalysisManager::GetCommonFileName(),analBranchTmp.Data(),jetbgAlgo,TMath::Nint(10*bgjetParameterR),mcChargSuffix.Data(),offlineTriggerMask,triggerType,evtRangeLow,evtRangeHigh,trigRangeLow,trigRangeHigh)
   );

   mgr->ConnectInput (task, 0, mgr->GetCommonInputContainer());
   //FK//mgr->ConnectOutput(task, 0, mgr->GetCommonOutputContainer());

   if(bUseExchContainer){

      if(analBranchKine.Length()>0){ //kine jets = input slot 1
         AliAnalysisDataContainer* containerKineJets = (AliAnalysisDataContainer*)mgr->GetContainers()->FindObject(analBranchKine.Data());
        mgr->ConnectInput(task, 1, containerKineJets); 
      }
      if(analBranchBgKine.Length()>0){ //kine jets to be removed from bg = input slot 2
         AliAnalysisDataContainer* containerKineJetsBg = (AliAnalysisDataContainer*)mgr->GetContainers()->FindObject(analBranchBgKine.Data());
         mgr->ConnectInput(task, 2, containerKineJetsBg);  
      }
   }

   mgr->ConnectOutput(task, 1, coutputJetCorePP);
   return task;
}
Commit	Line	Data
ad869500	1
	2
	3	AliAnalysisTaskJetCorePP* AddTaskJetCorePP(
	4	const Char_t* branchPrefix="clustersAOD",
255cb71c	5	const Char_t* jetAlgo="ANTIKT",
ad869500	6	Float_t jetParameterR = 0.4, //jet R
ad869500	7	UInt_t trkFilterMask = 272,
ad869500	8	Float_t trackLowPtCut = 0.15,
8aa19603	9	const Char_t* jetbgAlgo="ANTIKT", //background jet algo
	10	Float_t bgjetParameterR = 0.3, //R of jet to be removed while bg calc
	11	Float_t bgMaxJetPt = 8.0, //max jet pt to be accepted to bg
d1405a52 ML	12	Int_t rndTrials = 2000, //number of trials to get jet free cell area
d1405a52 ML	13	Float_t jetFreeAreaFrac = 0.75, //cell area free of jets
8aa19603	14	Float_t bgConeR = 0.4, //R of perp cone jet R
ad869500	15	Int_t collisionSystem = 0, //pp=0, pPb=1
	16	Int_t offlineTriggerMask=AliVEvent::kMB, //MinBias=0
	17	Int_t minContribVtx = 1,
	18	Float_t vtxZMin = -10.0,
	19	Float_t vtxZMax = 10.0,
	20	Float_t centMin = 0.0,
	21	Float_t centMax = 100.0,
	22	Float_t triggerEtaCut = 0.9,
	23	Float_t trackEtaCut = 0.9,
80ac66f6	24	const Char_t* nonStdFile="",
255cb71c	25	const Char_t* mcFullFlag="", // real="", all jets= "MC"
255cb71c	26	const Char_t* mcChargFlag="", // real="", charged jets = "MC2"
8aa19603	27	Bool_t bfillrespmx=0, // 0=dont fill resp mx histos, 1=fill histos
1afc79e9	28	Bool_t bDiceEff=0, // 0=leave efficiency as it is, 1= reduce efficiency by constant amount via SetFixedEfficiency
08d97d90	29	Bool_t bMomSmear=0, // 0=leave efficiency as it is, 1= reduce efficiency by constant amount via SetFixedEfficiency
49dc2855	30	Bool_t bDoubleBinning=0, // 0= 2GeV bin size 1= 1GeV/bin size
49dc2855	31	Bool_t bUseExchContainer=0, //
8e103a56	32	Int_t triggerType=0, //0=single incl trigger, 1=leading track, 2=hadron pt>10
	33	Int_t evtRangeLow=0, //last digit of range of ESD event number
	34	Int_t evtRangeHigh=9, //first digit of range of ESD event number
	35	Float_t trigRangeLow=0, //trigger pT low bin boreder works with triggType=0
	36	Float_t trigRangeHigh=50 //trigger pT high border works with triggType=0
ad869500	37	){
ad869500	38	Printf("adding task jet response\n");
	39
	40	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	41	if(!mgr){
	42	::Error("AddTaskJetCorePP", "No analysis manager to connect to.");
	43	return NULL;
	44	}
	45	if(!mgr->GetInputEventHandler()){
	46	::Error("AddTaskJetCorePP", "This task requires an input event handler.");
	47	return NULL;
	48	}
	49
	50	Float_t jetEtaMin = -0.9 + jetParameterR;
	51	Float_t jetEtaMax = 0.9 - jetParameterR;
	52
49dc2855	53	TString analBranch="";
	54	TString analBranchBg=""; //jet to be removed from bg
	55	TString analBranchFullMC=""; //full jets MC to be used with charged jets MC
	56	TString analBranchChargMC=""; //charged jets
	57	TString analBranchBgChargMC=""; //charged jets background
	58	TString analBranchKine=""; //kine jets
	59	TString analBranchBgKine=""; //kine jets background
	60
	61	TString mcFullSuffix=""; //MC = all jets
	62	TString mcChargSuffix=""; //MC2= charged jets, MC = all jets
	63
255cb71c	64
255cb71c	65
49dc2855	66	TString bpfx(branchPrefix);
ad869500	67	TString stJetAlgo(jetAlgo);
255cb71c	68	TString stJetBgAlgo(jetbgAlgo);
ad869500	69	stJetAlgo.ToUpper();
255cb71c	70	stJetBgAlgo.ToUpper();
80ac66f6	71
255cb71c	72	TString jet="";
	73	TString jetbg="";
	74	TString otherparams="";
49dc2855	75
49dc2855	76	jet = jet + "_" + stJetAlgo + Form("%02d",(Int_t) (10*jetParameterR)); // _ANTIKT02
8aa19603	77	jetbg = jetbg + "_" + stJetBgAlgo + Form("%02d",(Int_t) (10*bgjetParameterR));
49dc2855	78
49dc2855	79	if(!bpfx.Contains("KINE")){
80ac66f6	80
49dc2855	81	analBranch = bpfx;
	82	analBranchBg = bpfx;
	83
	84	otherparams = otherparams + "_B0"; //bg mode
	85	otherparams = otherparams + Form("_Filter%05d",(UInt_t) trkFilterMask);
	86	otherparams = otherparams + Form("_Cut%05d",(Int_t) (1000*trackLowPtCut));
255cb71c	87
49dc2855	88	if(analBranch.BeginsWith("clustersAOD")){
	89	otherparams = otherparams + Form("_Skip%02d",0);
	90	}
	91
	92	analBranch = analBranch + jet + otherparams; //antikt jet
	93	analBranchBg = analBranchBg + jetbg + otherparams; //kt bg jet
	94
08d97d90	95	if(bDiceEff \|\| bMomSmear){ //dicing efficiency relates rec only
	96	analBranch = analBranch + Form("Detector%d%dFr0",(Int_t) bDiceEff,(Int_t) bMomSmear); //dice=1, smear=0, change eff fraction =0
	97	analBranchBg = analBranchBg + Form("Detector%d%dFr0",(Int_t) bDiceEff,(Int_t) bMomSmear);
49dc2855	98	}
	99
	100	//clustersAOD_ANTIKT04_B0_Filter00272_Cut00150_Skip00
	101	//clustersAOD_ANTIKT04_B0_Filter00272_Cut00150_Skip00_Detector10Fr0
	102	//Skip00 none of the most energetic jets is ommited
	103	//Cut00150 pT min cut on track
	104	//Filter00272
	105
	106	mcFullSuffix=Form("%s",mcFullFlag); //MC = all jets
	107	mcChargSuffix=Form("%s",mcChargFlag); //MC2= charged jets, MC = all jets
	108
	109	if(mcChargSuffix.Length()>0 && mcChargSuffix=="MC2"){ //charged jets generator level
	110	analBranchChargMC = bpfx + mcChargSuffix + jet + otherparams;
	111	analBranchBgChargMC = bpfx + mcChargSuffix + jetbg + otherparams;
	112	}
	113
	114	if(mcFullSuffix.Length()>0 && mcFullSuffix=="MC"){ //full jets generator level
	115	analBranchFullMC = bpfx + mcFullSuffix + jet + otherparams;
	116	}
	117	}else{ //KINE JETS
	118	//clustersKINECHARGED_ANTIKT02_Cut00150
	119	otherparams = otherparams + Form("_Cut%05d",TMath::Nint(1000*trackLowPtCut));
	120
	121	analBranchKine = analBranchKine + bpfx + jet + otherparams;
	122	analBranchBgKine = analBranchBgKine + bpfx + jetbg + otherparams;
1afc79e9 ML	123	}
1afc79e9 ML	124
49dc2855	125	TString analBranchTmp = analBranch;
255cb71c	126
49dc2855	127	if(bpfx.Contains("KINE")){
49dc2855	128	analBranchTmp=analBranchKine;
80ac66f6	129	}
49dc2855	130	AliAnalysisTaskJetCorePP *task = new AliAnalysisTaskJetCorePP(Form("JetCorePP_%s_%s_%d",analBranchTmp.Data(), mcChargSuffix.Data(), offlineTriggerMask));
ad869500	131
ad869500	132	task->SetBranchName(analBranch.Data());
255cb71c	133	task->SetBranchNameFullMC(analBranchFullMC.Data());
255cb71c	134	task->SetBranchNameChargMC(analBranchChargMC.Data());
49dc2855	135	task->SetBranchNameKine(analBranchKine.Data());
	136	task->SetBranchNameBg(analBranchBg.Data()); //kine jets
	137	task->SetBranchNameBgChargMC(analBranchBgChargMC.Data()); //jet to be removed from bg
	138	task->SetBranchNameBgKine(analBranchBgKine.Data()); //kine jets
ad869500	139	task->SetNonStdFile(nonStdFile);
	140	task->SetSystem(collisionSystem);
	141	task->SetJetR(jetParameterR);
8aa19603	142	task->SetBgJetR(bgjetParameterR);
8aa19603	143	task->SetBgMaxJetPt(bgMaxJetPt);
d1405a52 ML	144	task->SetRndTrials(rndTrials);
d1405a52 ML	145	task->SetFreeAreaFrac(jetFreeAreaFrac);
8aa19603	146	task->SetBgConeR(bgConeR);
ad869500	147	task->SetOfflineTrgMask(offlineTriggerMask);
	148	task->SetMinContribVtx(minContribVtx);
	149	task->SetVtxZMin(vtxZMin);
	150	task->SetVtxZMax(vtxZMax);
	151	task->SetFilterMask(trkFilterMask);
	152	task->SetCentMin(centMin);
	153	task->SetCentMax(centMax);
	154	task->SetJetEtaMin(jetEtaMin);
	155	task->SetJetEtaMax(jetEtaMax);
	156	task->SetTriggerEtaCut(triggerEtaCut);
	157	task->SetTrackEtaCut(trackEtaCut);
	158	task->SetTrackLowPtCut(trackLowPtCut);
8e103a56	159	task->SetTriggerType(triggerType);
	160	task->SetEventNumberRangeLow(evtRangeLow);
	161	task->SetEventNumberRangeHigh(evtRangeHigh);
	162	task->SetTriggerPtRangeLow(trigRangeLow);
	163	task->SetTriggerPtRangeHigh(trigRangeHigh);
8aa19603	164	task->SetFillResponseMatrix(bfillrespmx);
49dc2855	165	task->SetBinning(bDoubleBinning);
49dc2855	166	task->SetUseExchangeContainerInput(bUseExchContainer);
8e103a56	167
ad869500	168	task->SetDebugLevel(0); //No debug messages 0
ad869500	169	mgr->AddTask(task);
49dc2855	170
8e103a56	171	//E= range of last two decimal numbers in event numbers
8e103a56	172	//Ptt range of the cosidered trigger bin
ad869500	173	AliAnalysisDataContainer *coutputJetCorePP = mgr->CreateContainer(
49dc2855	174	Form("pwgjejetcorepp_%s_%s%02d_%s_%d_T%d_E%d_%d_Ptt%.0f_%.0f",analBranchTmp.Data(),jetbgAlgo,TMath::Nint(10*bgjetParameterR),mcChargSuffix.Data(),offlineTriggerMask,triggerType,evtRangeLow,evtRangeHigh,trigRangeLow,trigRangeHigh),
ad869500	175	TList::Class(),
ad869500	176	AliAnalysisManager::kOutputContainer,
49dc2855	177	Form("%s:PWGJE_jetcorepp_%s_%s%02d_%s_%d_T%d_E%d_%d_Ptt%.0f_%.0f",AliAnalysisManager::GetCommonFileName(),analBranchTmp.Data(),jetbgAlgo,TMath::Nint(10*bgjetParameterR),mcChargSuffix.Data(),offlineTriggerMask,triggerType,evtRangeLow,evtRangeHigh,trigRangeLow,trigRangeHigh)
ad869500	178	);
	179
	180	mgr->ConnectInput (task, 0, mgr->GetCommonInputContainer());
49dc2855	181	//FK//mgr->ConnectOutput(task, 0, mgr->GetCommonOutputContainer());
	182
	183	if(bUseExchContainer){
	184
	185	if(analBranchKine.Length()>0){ //kine jets = input slot 1
	186	AliAnalysisDataContainer* containerKineJets = (AliAnalysisDataContainer*)mgr->GetContainers()->FindObject(analBranchKine.Data());
	187	mgr->ConnectInput(task, 1, containerKineJets);
	188	}
	189	if(analBranchBgKine.Length()>0){ //kine jets to be removed from bg = input slot 2
3d1fac8a	190	AliAnalysisDataContainer* containerKineJetsBg = (AliAnalysisDataContainer*)mgr->GetContainers()->FindObject(analBranchBgKine.Data());
3d1fac8a	191	mgr->ConnectInput(task, 2, containerKineJetsBg);
49dc2855	192	}
49dc2855	193	}
ad869500	194
49dc2855	195	mgr->ConnectOutput(task, 1, coutputJetCorePP);
ad869500	196	return task;
ad869500	197	}