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

AliAnalysisTaskJetCluster *AddTaskJetCluster(char* bRec = "AOD",char* bGen = "",UInt_t filterMask = 16, UInt_t iPhysicsSelectionFlag = AliVEvent::kAny,Char_t *jf = "KT", Float_t radius = 0.4,Int_t nSkip = 0,Int_t kWriteAOD = kFALSE,char* deltaFile = "",Float_t ptTrackCut = 0.15, Float_t etaTrackWindow = 0.9,Float_t vertexWindow = 10.,Int_t nSkipCone = 2,Int_t dice=0,Int_t smear=0,Bool_t useOADB=kFALSE,Double_t changeEfficiencyFraction=0.);
AliAnalysisTaskJetCluster *AddTaskJetClusterDelta(UInt_t filterMask = 16,Bool_t kUseAODMC = kFALSE,UInt_t iPhysicsSelectionFlag = AliVEvent::kMB,Char_t *jf = "KT", UInt_t iFlag);

Int_t kBackgroundModeCl = 0;
Float_t kPtTrackCutCl = 0.15;
Float_t kTrackEtaWindowCl = 0.8;
Float_t kVertexWindowCl = 10;

AliAnalysisTaskJetCluster *AddTaskJetClusterDelta(UInt_t filterMask,Bool_t kUseAODMC,UInt_t iPhysicsSelectionFlag,Char_t *jf, UInt_t iFlag){
   AliAnalysisTaskJetCluster *js = 0;
   if(kUseAODMC&&false){// do not use the MC info yet
     if(iFlag&(1<<0))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.00001);
     if(iFlag&(1<<1))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.1);
     if(iFlag&(1<<2))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.2);
     if(iFlag&(1<<4))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.4);
     if(iFlag&(1<<6))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.6);
     if(iFlag&(1<<8))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.8);
     if(iFlag&(1<<10))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,1.0);
   }
   else{
     if(iFlag&(1<<0))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.00001);
     if(iFlag&(1<<1))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.1);
     if(iFlag&(1<<2))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.2);
     if(iFlag&(1<<4))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.4);
     if(iFlag&(1<<6))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.6);
     if(iFlag&(1<<8))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.8);
     if(iFlag&(1<<10))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,1.0);
   }

   return js;
 }


AliAnalysisTaskJetCluster *AddTaskJetCluster(char* bRec,char* bGen ,UInt_t filterMask,UInt_t iPhysicsSelectionFlag,Char_t *jf,Float_t radius,Int_t nSkip,Int_t kWriteAOD,char *deltaFile,Float_t ptTrackCut,Float_t etaTrackWindow,Float_t vertexWindow,Int_t nSkipCone,Int_t dice,Int_t smear,Bool_t useOADB,Double_t changeEfficiencyFraction)
 {
 // Creates a jet fider task, configures it and adds it to the analysis manager.
   kPtTrackCutCl = ptTrackCut;
   kTrackEtaWindowCl = etaTrackWindow;
   kVertexWindowCl = vertexWindow;

   TString outputFile(deltaFile);
    // Get the pointer to the existing analysis manager via the static access method.
    //==============================================================================
    AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
    if (!mgr) {
       ::Error("AddTaskJetCluster", "No analysis manager to connect to.");
       return NULL;
    }  

    // Check the analysis type using the event handlers connected to the analysis manager.
    //==============================================================================
    if (!mgr->GetInputEventHandler()) {
      ::Error("AddTaskJetCluster", "This task requires an input event handler");
       return NULL;
    }

    TString type = mgr->GetInputEventHandler()->GetDataType();
    TString typeRec(bRec);
    TString typeGen(bGen);
    if(!typeRec.Contains("AODextra")) {
      typeGen.ToUpper();
      typeRec.ToUpper();
    }
    cout << "typeRec: " << typeRec << endl;
    // Create the task and configure it.
    //===========================================================================


    TString cAdd = "";
    cAdd += Form("%02d_",(int)((radius+0.01)*10.));
    cAdd += Form("B%d",(int)kBackgroundModeCl);
    cAdd += Form("_Filter%05d",filterMask);
    cAdd += Form("_Cut%05d",(int)(1000.*kPtTrackCutCl));
    cAdd += Form("_Skip%02d",nSkip);
    if(dice>0 || smear>0)
      cAdd += Form("Detector%d%dFr%d",dice,smear,(int)(changeEfficiencyFraction*100.));
    

    Printf("%s %E %d %d",cAdd.Data(),kPtTrackCutCl,dice,smear);
    AliAnalysisTaskJetCluster* clus = new  AliAnalysisTaskJetCluster(Form("JetCluster%s_%s%s",bRec,jf,cAdd.Data()));
      
   // or a config file
   // clus->SetAnalysisType(AliAnalysisTaskJetCluster::kAnaMC);
   // if(iAODanalysis)clus->SetAODInput(kTRUE);
   // clus->SetDebugLevel(11); 
   clus->SetFilterMask(filterMask); 
   //   clus->SetUseGlobalSelection(kTRUE); 
   clus->SetVtxCuts(kVertexWindowCl,1);
   if(type == "AOD"){
     // Assume all jet are produced already
     clus->SetAODTrackInput(kTRUE);
     clus->SetAODMCInput(kTRUE);
   }

   if(typeRec.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD
     clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCChargedAcceptance);
     clus->SetTrackPtCut(kPtTrackCutCl);
     clus->SetTrackEtaWindow(kTrackEtaWindowCl);
   }
   else if (typeRec.Contains("AODMC2")){
     clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCCharged);
     clus->SetTrackPtCut(kPtTrackCutCl);
     clus->SetTrackEtaWindow(5);
   }
   else if (typeRec.Contains("AODMC")){
     clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCAll);
     clus->SetTrackPtCut(kPtTrackCutCl);
     clus->SetTrackEtaWindow(5);
   }
   else if (typeRec.Contains("AODextraonly")) {
     clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODextraonly);
     clus->SetTrackPtCut(kPtTrackCutCl);
     clus->SetTrackEtaWindow(kTrackEtaWindowCl);
   }
   else if (typeRec.Contains("AODextra")) {
     cout << "AliAnalysisTaskJetCluster::kTrackAODextra: " << AliAnalysisTaskJetCluster::kTrackAODextra << endl;
     clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODextra);
     clus->SetTrackPtCut(kPtTrackCutCl);
     clus->SetTrackEtaWindow(kTrackEtaWindowCl);
   }
   else if (typeRec.Contains("AOD")) {
     clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAOD);
     clus->SetTrackPtCut(kPtTrackCutCl);
     clus->SetTrackEtaWindow(kTrackEtaWindowCl);
   }

   clus->SetRparam(radius);
   clus->SetGhostArea(0.005);
   clus->SetGhostEtamax(kTrackEtaWindowCl);

   switch (jf) {
   case "ANTIKT":
     clus->SetAlgorithm(2); // antikt from fastjet/JetDefinition.hh
     break;
   case "CA":
     clus->SetAlgorithm(1); // CA from fastjet/JetDefinition.hh
     break;
   case "KT":
     clus->SetAlgorithm(0); // kt from fastjet/JetDefinition.hh
     break;
   default:
     ::Error("AddTaskJetCluster", "Wrong jet finder selected\n");
     return 0;
   }

   
   if(kWriteAOD){
     if(outputFile.Length())clus->SetJetOutputFile(outputFile);
     clus->SetJetOutputBranch(Form("clusters%s_%s%s",bRec,jf,cAdd.Data()));
     clus->SetJetOutputMinPt(0); // store only jets / clusters above a certain threshold
   }

   clus->SetNSkipLeadingRan(nSkip);
   clus->SetNSkipLeadingCone(nSkipCone);

   if(iPhysicsSelectionFlag)clus->SelectCollisionCandidates(iPhysicsSelectionFlag);

   if(useOADB) {
     clus->SetUseTrResolutionFromOADB();
     clus->SetUseTrEfficiencyFromOADB();
     clus->SetChangeEfficiencyFraction(changeEfficiencyFraction);
   }

   mgr->AddTask(clus);

   // Create ONLY the output containers for the data produced by the task.
   // Get and connect other common input/output containers via the manager as below
   //==============================================================================
   AliAnalysisDataContainer *coutput1_clus = mgr->CreateContainer(Form("cluster_%s_%s_%s%s",bRec,bGen,jf,cAdd.Data()), TList::Class(),AliAnalysisManager::kOutputContainer,Form("%s:PWGJE_cluster_%s_%s_%s%s",AliAnalysisManager::GetCommonFileName(),bRec,bGen,jf,cAdd.Data()));

   mgr->ConnectInput  (clus, 0, mgr->GetCommonInputContainer());
   mgr->ConnectOutput (clus, 0, mgr->GetCommonOutputContainer());
   mgr->ConnectOutput (clus,  1, coutput1_clus );
   
   return clus;
}
Commit	Line	Data
fa7c34ba	1	AliAnalysisTaskJetCluster AddTaskJetCluster(char bRec = "AOD",char* bGen = "",UInt_t filterMask = 16, UInt_t iPhysicsSelectionFlag = AliVEvent::kAny,Char_t jf = "KT", Float_t radius = 0.4,Int_t nSkip = 0,Int_t kWriteAOD = kFALSE,char deltaFile = "",Float_t ptTrackCut = 0.15, Float_t etaTrackWindow = 0.9,Float_t vertexWindow = 10.,Int_t nSkipCone = 2,Int_t dice=0,Int_t smear=0,Bool_t useOADB=kFALSE,Double_t changeEfficiencyFraction=0.);
353203e1	2	AliAnalysisTaskJetCluster AddTaskJetClusterDelta(UInt_t filterMask = 16,Bool_t kUseAODMC = kFALSE,UInt_t iPhysicsSelectionFlag = AliVEvent::kMB,Char_t jf = "KT", UInt_t iFlag);
a1d44e31	3
aa1ee58e	4	Int_t kBackgroundModeCl = 0;
	5	Float_t kPtTrackCutCl = 0.15;
	6	Float_t kTrackEtaWindowCl = 0.8;
	7	Float_t kVertexWindowCl = 10;
a1d44e31	8
353203e1	9	AliAnalysisTaskJetCluster AddTaskJetClusterDelta(UInt_t filterMask,Bool_t kUseAODMC,UInt_t iPhysicsSelectionFlag,Char_t jf, UInt_t iFlag){
a1d44e31	10	AliAnalysisTaskJetCluster *js = 0;
	11	if(kUseAODMC&&false){// do not use the MC info yet
	12	if(iFlag&(1<<0))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.00001);
	13	if(iFlag&(1<<1))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.1);
	14	if(iFlag&(1<<2))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.2);
	15	if(iFlag&(1<<4))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.4);
	16	if(iFlag&(1<<6))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.6);
	17	if(iFlag&(1<<8))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.8);
	18	if(iFlag&(1<<10))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,1.0);
	19	}
	20	else{
	21	if(iFlag&(1<<0))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.00001);
	22	if(iFlag&(1<<1))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.1);
	23	if(iFlag&(1<<2))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.2);
	24	if(iFlag&(1<<4))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.4);
	25	if(iFlag&(1<<6))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.6);
	26	if(iFlag&(1<<8))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.8);
	27	if(iFlag&(1<<10))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,1.0);
	28	}
	29
	30	return js;
	31	}
	32
	33
fa7c34ba	34	AliAnalysisTaskJetCluster AddTaskJetCluster(char bRec,char* bGen ,UInt_t filterMask,UInt_t iPhysicsSelectionFlag,Char_t jf,Float_t radius,Int_t nSkip,Int_t kWriteAOD,char deltaFile,Float_t ptTrackCut,Float_t etaTrackWindow,Float_t vertexWindow,Int_t nSkipCone,Int_t dice,Int_t smear,Bool_t useOADB,Double_t changeEfficiencyFraction)
a1d44e31	35	{
a1d44e31	36	// Creates a jet fider task, configures it and adds it to the analysis manager.
aa1ee58e	37	kPtTrackCutCl = ptTrackCut;
	38	kTrackEtaWindowCl = etaTrackWindow;
	39	kVertexWindowCl = vertexWindow;
a1d44e31	40
	41	TString outputFile(deltaFile);
	42	// Get the pointer to the existing analysis manager via the static access method.
	43	//==============================================================================
	44	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	45	if (!mgr) {
	46	::Error("AddTaskJetCluster", "No analysis manager to connect to.");
	47	return NULL;
	48	}
	49
	50	// Check the analysis type using the event handlers connected to the analysis manager.
	51	//==============================================================================
	52	if (!mgr->GetInputEventHandler()) {
	53	::Error("AddTaskJetCluster", "This task requires an input event handler");
	54	return NULL;
	55	}
	56
	57	TString type = mgr->GetInputEventHandler()->GetDataType();
	58	TString typeRec(bRec);
	59	TString typeGen(bGen);
	60	if(!typeRec.Contains("AODextra")) {
	61	typeGen.ToUpper();
	62	typeRec.ToUpper();
	63	}
	64	cout << "typeRec: " << typeRec << endl;
	65	// Create the task and configure it.
	66	//===========================================================================
	67
	68
	69
	70
	71	TString cAdd = "";
	72	cAdd += Form("%02d_",(int)((radius+0.01)*10.));
aa1ee58e	73	cAdd += Form("B%d",(int)kBackgroundModeCl);
a1d44e31	74	cAdd += Form("_Filter%05d",filterMask);
aa1ee58e	75	cAdd += Form("_Cut%05d",(int)(1000.*kPtTrackCutCl));
a1d44e31	76	cAdd += Form("_Skip%02d",nSkip);
28c36b48	77	if(dice>0 \|\| smear>0)
fa7c34ba	78	cAdd += Form("Detector%d%dFr%d",dice,smear,(int)(changeEfficiencyFraction*100.));
fa7c34ba	79
28c36b48	80
28c36b48	81	Printf("%s %E %d %d",cAdd.Data(),kPtTrackCutCl,dice,smear);
5c4489e2	82	AliAnalysisTaskJetCluster* clus = new AliAnalysisTaskJetCluster(Form("JetCluster%s_%s%s",bRec,jf,cAdd.Data()));
a1d44e31	83
a1d44e31	84	// or a config file
5c4489e2	85	// clus->SetAnalysisType(AliAnalysisTaskJetCluster::kAnaMC);
	86	// if(iAODanalysis)clus->SetAODInput(kTRUE);
	87	// clus->SetDebugLevel(11);
	88	clus->SetFilterMask(filterMask);
	89	// clus->SetUseGlobalSelection(kTRUE);
aa1ee58e	90	clus->SetVtxCuts(kVertexWindowCl,1);
a1d44e31	91	if(type == "AOD"){
a1d44e31	92	// Assume all jet are produced already
5c4489e2	93	clus->SetAODTrackInput(kTRUE);
5c4489e2	94	clus->SetAODMCInput(kTRUE);
a1d44e31	95	}
	96
	97	if(typeRec.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD
5c4489e2	98	clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCChargedAcceptance);
aa1ee58e	99	clus->SetTrackPtCut(kPtTrackCutCl);
aa1ee58e	100	clus->SetTrackEtaWindow(kTrackEtaWindowCl);
a1d44e31	101	}
a1d44e31	102	else if (typeRec.Contains("AODMC2")){
5c4489e2	103	clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCCharged);
aa1ee58e	104	clus->SetTrackPtCut(kPtTrackCutCl);
5c4489e2	105	clus->SetTrackEtaWindow(5);
a1d44e31	106	}
a1d44e31	107	else if (typeRec.Contains("AODMC")){
5c4489e2	108	clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCAll);
aa1ee58e	109	clus->SetTrackPtCut(kPtTrackCutCl);
5c4489e2	110	clus->SetTrackEtaWindow(5);
a1d44e31	111	}
a1d44e31	112	else if (typeRec.Contains("AODextraonly")) {
5c4489e2	113	clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODextraonly);
aa1ee58e	114	clus->SetTrackPtCut(kPtTrackCutCl);
aa1ee58e	115	clus->SetTrackEtaWindow(kTrackEtaWindowCl);
a1d44e31	116	}
	117	else if (typeRec.Contains("AODextra")) {
	118	cout << "AliAnalysisTaskJetCluster::kTrackAODextra: " << AliAnalysisTaskJetCluster::kTrackAODextra << endl;
5c4489e2	119	clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODextra);
aa1ee58e	120	clus->SetTrackPtCut(kPtTrackCutCl);
aa1ee58e	121	clus->SetTrackEtaWindow(kTrackEtaWindowCl);
a1d44e31	122	}
a1d44e31	123	else if (typeRec.Contains("AOD")) {
5c4489e2	124	clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAOD);
aa1ee58e	125	clus->SetTrackPtCut(kPtTrackCutCl);
aa1ee58e	126	clus->SetTrackEtaWindow(kTrackEtaWindowCl);
a1d44e31	127	}
a1d44e31	128
5c4489e2	129	clus->SetRparam(radius);
5c4489e2	130	clus->SetGhostArea(0.005);
aa1ee58e	131	clus->SetGhostEtamax(kTrackEtaWindowCl);
a1d44e31	132
	133	switch (jf) {
	134	case "ANTIKT":
5c4489e2	135	clus->SetAlgorithm(2); // antikt from fastjet/JetDefinition.hh
a1d44e31	136	break;
a1d44e31	137	case "CA":
5c4489e2	138	clus->SetAlgorithm(1); // CA from fastjet/JetDefinition.hh
a1d44e31	139	break;
a1d44e31	140	case "KT":
5c4489e2	141	clus->SetAlgorithm(0); // kt from fastjet/JetDefinition.hh
a1d44e31	142	break;
	143	default:
	144	::Error("AddTaskJetCluster", "Wrong jet finder selected\n");
	145	return 0;
	146	}
	147
	148
	149	if(kWriteAOD){
5c4489e2	150	if(outputFile.Length())clus->SetJetOutputFile(outputFile);
	151	clus->SetJetOutputBranch(Form("clusters%s_%s%s",bRec,jf,cAdd.Data()));
	152	clus->SetJetOutputMinPt(0); // store only jets / clusters above a certain threshold
a1d44e31	153	}
a1d44e31	154
5c4489e2	155	clus->SetNSkipLeadingRan(nSkip);
5c4489e2	156	clus->SetNSkipLeadingCone(nSkipCone);
a1d44e31	157
5c4489e2	158	if(iPhysicsSelectionFlag)clus->SelectCollisionCandidates(iPhysicsSelectionFlag);
a1d44e31	159
fa7c34ba	160	if(useOADB) {
	161	clus->SetUseTrResolutionFromOADB();
	162	clus->SetUseTrEfficiencyFromOADB();
	163	clus->SetChangeEfficiencyFraction(changeEfficiencyFraction);
	164	}
	165
5c4489e2	166	mgr->AddTask(clus);
a1d44e31	167
	168	// Create ONLY the output containers for the data produced by the task.
	169	// Get and connect other common input/output containers via the manager as below
	170	//==============================================================================
28c36b48	171	AliAnalysisDataContainer *coutput1_clus = mgr->CreateContainer(Form("cluster_%s_%s_%s%s",bRec,bGen,jf,cAdd.Data()), TList::Class(),AliAnalysisManager::kOutputContainer,Form("%s:PWGJE_cluster_%s_%s_%s%s",AliAnalysisManager::GetCommonFileName(),bRec,bGen,jf,cAdd.Data()));
a1d44e31	172
5c4489e2	173	mgr->ConnectInput (clus, 0, mgr->GetCommonInputContainer());
5c4489e2	174	mgr->ConnectOutput (clus, 0, mgr->GetCommonOutputContainer());
28c36b48	175	mgr->ConnectOutput (clus, 1, coutput1_clus );
a1d44e31	176
5c4489e2	177	return clus;
a1d44e31	178	}