[u/mrichter/AliRoot.git] / PWGUD / multVScentPbPb / MyAnalysisMacroTrackletMulti.C

Bool_t needRecPoints = kFALSE;

void MyAnalysisMacroTrackletMulti
(TString dataset="/alice/sim/LHC10f8c_130844",
 TString outFName = "trbg.root",
 Int_t   nEvents   = -1,
 Float_t etaMin     =-0.5,        // min eta range to fill in histos
 Float_t etaMax     = 0.5,        // max eta range to fill in histos
 Float_t zMin       = -7,         // process events with Z vertex min
 Float_t zMax       =  7,         //                     max positions
 Int_t   useCentVar = 0,          // centrality variable to use: 0=V0, 1=SPD2corr
 //
 Float_t cutSigNStd  = 1.5,        // cut on weighed distance used to extract signal
 Float_t cutSigDPhiS = -1,        // cut on dPhi-phiBent used to extract signal (if negative -> dphi*sqrt(cutSigNStd)
 Bool_t  useMC  = kTRUE,          // fill MC info (doRec=kTRUE)
 Float_t scaleMCV0 = 0.7520,      // rescale MC V0 to match the data
 //
 Bool_t doRec  = kTRUE,           // fill data histos from new reco
 Bool_t doInj  = kTRUE,           // create Inj. bg
 Bool_t doRot  = kFALSE,          // create Rot. bg
 Bool_t doMix  = kFALSE,//kTRUE,  // create Mix. bg
 // 
 // specific parameters for reconstruction
 float  phiRot      = 3.14159e+00, // angle for bg. generation with rotation
 float  injScale    = 1.,//0.7,    // inject injScale*Ncl(Lr1/Lr2) hits
 Bool_t scaleDTheta = kTRUE,       // scale dTheta by 1/sin^2(theta) in trackleting
 float  nStdDev     = 25.,         // number of st.dev. for tracklet cut to keep
 float  dphi        = 0.06,        // dphi window (sigma of tracklet cut)
 float  dtht        = 0.025,       // dtheta .... (if negative, abs will be used with additional cut on |dthetaX|, apart from w.distance
 float  phishift    = 0.0045,      // bending shift
 Bool_t remOvl      = kTRUE,       
 float  ovlPhiCut   = 0.005, 
 float  ovlZetaCut  = 0.05,
 Int_t  nEventsSkip = 0,
 //----------------------- Ntracklets selection parameters important for mixing, to be tuned
 Float_t ntMin      =   1,         // process events with ESDmult 
 Float_t ntMax      = 20000,       // within this range
 Float_t ntMixBinSz = 20000,       // ESDMult bin size for mixing
 //----------------------- Zv selection parameters important for mixing, to be tuned
 Float_t zMixBinSz  =  14,       //0.1,  // Zv. bin for mixing
 //---------------------------------------------------------------------------------
 //
 Bool_t checkReconstructables = kFALSE//kTRUE, // fill histos for reconstructable (needs useMC and doRec) 
 //
 )
{
  //  
  if (cutSigDPhiS<0) cutSigDPhiS = TMath::Sqrt(cutSigNStd)*dphi;
  //
  needRecPoints = doRec || doInj || doRot || doMix;
    //
  printf("Start Analysis for %s, max %d Events skipping %d, Event Cuts: %.1f<eta<%.1f, %.2f<Zv<%.2f\n",
	 dataset.Data(),nEvents,nEventsSkip,etaMin,etaMax,zMin,zMax);
  printf("Centrality variable: %d\n",useCentVar);
  printf("Tracklet cuts: dPhi:%.3f dTheta:%.3f phiShift:%.4f | Keep %.1f NstDev\n"
	 "Scale dTheta: %s | Signal Selection: NstDev:%.1f, dPhiS: %.3f\n", 
	 dphi,dtht,phishift,nStdDev,scaleDTheta ? "ON":"OFF",
	 cutSigNStd,cutSigDPhiS);
  //
  printf("UseMC: %s. V0 scale: %.4f\n",useMC ? "ON":"OFF",scaleMCV0);
  printf("Operations:  \n"
	 "Reco:%s (RemOvl:%s phi:%.3f zeta:%.3f)\n"
	 "Inj:%s (scale: %.2f)\n"
	 "Rot:%s (phi: %.4f)\n"
	 "Mix:%s (Nt:%.1f:%.1f:%.1f Zv:%.1f:%.1f:%.2f)\n", 
	 doRec ? "ON":"OFF",remOvl? "ON":"OFF",ovlPhiCut,ovlZetaCut,
	 doInj ? "ON":"OFF",injScale,
	 doRot ? "ON":"OFF",phiRot,
	 doMix ? "ON":"OFF",ntMin,ntMax,ntMixBinSz, zMin,zMax,zMixBinSz);
  //
  if (nEvents<0) nEvents = int(1e9);
  TString format = GetFormatFromDataSet(dataset);
  //
  // ALICE stuff
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr) mgr = new AliAnalysisManager("Test train");
  //
  InputHandlerSetup(format,useMC);
  if (doMix) MixHandlerSetup(ntMin,ntMax,ntMixBinSz, zMin,zMax,zMixBinSz);
  // compile our task
  gProof->Load("AliITSMultRecBg.cxx++");
  gProof->Load("AliTrackletTaskMulti.cxx++");
  //
  printf("Loading Centrality task\n");
  gROOT->LoadMacro("$ALICE_ROOT/OADB/macros/AddTaskCentrality.C");
  AliCentralitySelectionTask *taskCentrality = AddTaskCentrality();
  //  taskCentrality->SetDebugLevel(2);
  if (useMC) taskCentrality->SetMCInput();
  //  taskCentrality->Dump();
  //
  // load and run AddTask macro
  gROOT->LoadMacro("AddMultTaskTrackletMulti.C");
  //
  // create our task
  AliTrackletTaskMulti *mltTask = AddMultTaskTrackletMulti(outFName.Data());
  //
  mltTask->SetUseCentralityVar(useCentVar);
  mltTask->SetDoNormalReco(doRec);
  mltTask->SetDoInjection(doInj);
  mltTask->SetDoRotation(doRot);
  mltTask->SetDoMixing(doMix);  
  //
  mltTask->SetUseMC(useMC);
  mltTask->SetCheckReconstructables(checkReconstructables);
  //
  mltTask->SetEtaMin(etaMin);
  mltTask->SetEtaMax(etaMax);
  mltTask->SetZVertexMin(zMin);
  mltTask->SetZVertexMax(zMax);
  //
  mltTask->SetDPhiSCut(cutSigDPhiS);
  mltTask->SetNStdCut(cutSigNStd);
  mltTask->SetScaleMCV0(scaleMCV0);
  //
  mltTask->SetScaleDThetaBySin2T(scaleDTheta);
  mltTask->SetNStdDev(nStdDev);
  mltTask->SetPhiWindow(dphi);
  mltTask->SetThetaWindow(dtht);
  mltTask->SetPhiShift(phishift);
  mltTask->SetPhiOverlapCut(ovlPhiCut);
  mltTask->SetZetaOverlapCut(ovlZetaCut);
  mltTask->SetPhiRot(phiRot);
  mltTask->SetInjScale(injScale);
  mltTask->SetRemoveOverlaps(remOvl);
  //
  printf("new Task: %p\n",mltTask);
  //
  printf("Requesting physics selection in %s mode\n",useMC ? "MC":"Data");
  gROOT->ProcessLine(".L $ALICE_ROOT/OADB/macros/AddTaskPhysicsSelection.C");
  AliPhysicsSelectionTask* physicsSelectionTask = AddTaskPhysicsSelection(useMC,0);
  mltTask->SelectCollisionCandidates();//AliVEvent::kMB);
  //
  // Run analysis
  mgr->InitAnalysis();
  // process dataset  
  mgr->StartAnalysis("proof", dataset.Data(), nEvents, nEventsSkip); 
  //
  TString evstCmd = "if [ -e event_stat.root ]; then \nmv event_stat.root evstat_"; 
  evstCmd += outFName;  evstCmd += " \nfi";
  gSystem->Exec( evstCmd.Data() );
  
}


TString GetFormatFromDataSet(TString dataset) {
  
//   Info("runAAF.C","Detecting format from dataset (may take while, depends on network connection)...");
  TString dsTreeName;
  if (dataset.Contains("#")) {
    Info("runAAF.C",Form("Detecting format from dataset name '%s' ...",dataset.Data()));
    dsTreeName=dataset(dataset.Last('#'),dataset.Length());
  } else {
    Info("runAAF.C",Form("Detecting format from dataset '%s' (may take while, depends on network connection) ...",dataset.Data()));
    TFileCollection *ds = gProof->GetDataSet(dataset.Data());
    if (!ds) {
      Error(Form("Dataset %s doesn't exist on proof cluster!!!!",dataset.Data()));
      return "";
    }
    dsTreeName = ds->GetDefaultTreeName();
  }

  if (dsTreeName.Contains("esdTree")) {
    Info("runAAF.C","ESD input format detected ...");
    return "ESD";
  } else if (dsTreeName.Contains("aodTree"))  {
    Info("runAAF.C","AOD input format detected ...");
    return "AOD";
  } else {
    Error("runAAF.C",Form("Tree %s is not supported !!!",dsTreeName.Data()));
    Error("runAAF.C",Form("Maybe set your DS to %s#esdTree or %s#aodTree",dataset.Data(),dataset.Data()));
  }
  
  return "";
}

Bool_t InputHandlerSetup(TString format = "esd", Bool_t useKine = kTRUE)
{
  format.ToLower();

  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();

  AliAnalysisDataContainer *cin = mgr->GetCommonInputContainer();

  if (cin) return;

  if (!format.CompareTo("esd"))
  {
    AliESDInputHandler *esdInputHandler = dynamic_cast<AliESDInputHandler*>(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());

    if (!esdInputHandler)
    {
      Info("CustomAnalysisTaskInputSetup", "Creating esdInputHandler ...");
      if (needRecPoints)
	esdInputHandler = new AliESDInputHandlerRP();
      else 
	esdInputHandler = new AliESDInputHandler();
      //
      mgr->SetInputEventHandler(esdInputHandler);
    }
    //
    if (useKine)
    {
      AliMCEventHandler* mcInputHandler = dynamic_cast<AliMCEventHandler*>(AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());

      if (!mcInputHandler)
      {
        Info("CustomAnalysisTaskInputSetup", "Creating mcInputHandler ...");
        AliMCEventHandler* mcInputHandler = new AliMCEventHandler();
        mgr->SetMCtruthEventHandler(mcInputHandler);
      }
    }

  }
  else if (!format.CompareTo("aod"))
  {
    AliAODInputHandler *aodInputHandler = dynamic_cast<AliAODInputHandler*>(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());

    if (!aodInputHandler)
    {
      Info("CustomAnalysisTaskInputSetup", "Creating aodInputHandler ...");
      aodInputHandler = new AliAODInputHandler();
      mgr->SetInputEventHandler(aodInputHandler);
    }
  }
  else
  {
    Info("Wrong input format!!! Only ESD and AOD are supported. Skipping Task ...");
    return kFALSE;
  }

  return kTRUE;
}

void MixHandlerSetup(float ntMin,float ntMax,float ntMixBinSz,
		     float zMin, float zMax, float zMixBinSz)
{
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr) return;
  int bufferSize = 1;
  AliESDInputHandlerRP *esdH = dynamic_cast<AliESDInputHandlerRP*>(mgr->GetInputEventHandler());
  if (!esdH) return;
  //
  AliMixEventInputHandler *esdMixH = new AliMixEventInputHandler(bufferSize);
  esdMixH->SetInputHandlerForMixing(esdH);
  AliMixEventPool *evPool = new AliMixEventPool("MyPool");
  AliMixEventCutObj *tracklets = new AliMixEventCutObj(AliMixEventCutObj::kNumberTracklets, ntMin,ntMax,ntMixBinSz);
  AliMixEventCutObj *zvertex = new AliMixEventCutObj(AliMixEventCutObj::kZVertex, zMin,zMax, zMixBinSz);
  //  evPool->AddCut(tracklets);
  evPool->AddCut(zvertex);
  //evPool->Init();
  evPool->Print();
  esdMixH->SetEventPool(evPool);
  esdH->SetMixingHandler(esdMixH);
}

void AddPhysicsSelection(Bool_t isMC)
{
  // physics selection a la Michele
  if(!isMC) {
    //AliPhysicsSelection * physSel = physicsSelectionTask->GetPhysicsSelection();
    //    physSel->AddCollisionTriggerClass("+CMBAC-B-NOPF-ALL");
    /*
    physSel->AddCollisionTriggerClass("+CMBS1C-B-NOPF-ALL");
    physSel->AddCollisionTriggerClass("+CMBS1A-B-NOPF-ALL");
    */
    //
    //    physSel->AddCollisionTriggerClass("+CMBS2C-B-NOPF-ALL");
    //    physSel->AddCollisionTriggerClass("+CMBS2A-B-NOPF-ALL");
    //
    // This are needed only to fill the statistics tables
    //    physSel->AddBGTriggerClass("+CMBAC-C-NOPF-ALL");
    //    physSel->AddBGTriggerClass("+CMBAC-A-NOPF-ALL");
    //    physSel->AddBGTriggerClass("+CMBAC-E-NOPF-ALL");
    //
    /*
    physSel->AddBGTriggerClass("+CMBS1C-C-NOPF-ALL");
    physSel->AddBGTriggerClass("+CMBS1C-A-NOPF-ALL");
    physSel->AddBGTriggerClass("+CMBS1C-E-NOPF-ALL");
    //
    physSel->AddBGTriggerClass("+CMBS1A-C-NOPF-ALL");
    physSel->AddBGTriggerClass("+CMBS1A-A-NOPF-ALL");
    physSel->AddBGTriggerClass("+CMBS1A-E-NOPF-ALL");
    //
    */
    /*
    //
    physSel->AddBGTriggerClass("+CMBS2C-C-NOPF-ALL");
    physSel->AddBGTriggerClass("+CMBS2C-A-NOPF-ALL");
    physSel->AddBGTriggerClass("+CMBS2C-E-NOPF-ALL");
    //
    physSel->AddBGTriggerClass("+CMBS2A-C-NOPF-ALL");
    physSel->AddBGTriggerClass("+CMBS2A-A-NOPF-ALL");
    physSel->AddBGTriggerClass("+CMBS2A-E-NOPF-ALL");
    */
  } 
  // if you use the following line, your task only gets the selected events
  //  task->SelectCollisionCandidates(AliVEvent::kUserDefined);
  //  task->SelectCollisionCandidates();
  //
  //Alternatively, in the UserExec of your task:
  //Bool_t isSelected = (((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected() & AliVEvent::kUserDefined);
  //
}
Commit	Line	Data
db1bbbe1	1	Bool_t needRecPoints = kFALSE;
db1bbbe1	2
a9a39f46	3	void MyAnalysisMacroTrackletMulti
	4	(TString dataset="/alice/sim/LHC10f8c_130844",
	5	TString outFName = "trbg.root",
	6	Int_t nEvents = -1,
6c56e693	7	Float_t etaMin =-0.5, // min eta range to fill in histos
6c56e693	8	Float_t etaMax = 0.5, // max eta range to fill in histos
a9a39f46	9	Float_t zMin = -7, // process events with Z vertex min
	10	Float_t zMax = 7, // max positions
	11	Int_t useCentVar = 0, // centrality variable to use: 0=V0, 1=SPD2corr
	12	//
	13	Float_t cutSigNStd = 1.5, // cut on weighed distance used to extract signal
	14	Float_t cutSigDPhiS = -1, // cut on dPhi-phiBent used to extract signal (if negative -> dphi*sqrt(cutSigNStd)
	15	Bool_t useMC = kTRUE, // fill MC info (doRec=kTRUE)
	16	Float_t scaleMCV0 = 0.7520, // rescale MC V0 to match the data
	17	//
	18	Bool_t doRec = kTRUE, // fill data histos from new reco
	19	Bool_t doInj = kTRUE, // create Inj. bg
	20	Bool_t doRot = kFALSE, // create Rot. bg
	21	Bool_t doMix = kFALSE,//kTRUE, // create Mix. bg
	22	//
	23	// specific parameters for reconstruction
	24	float phiRot = 3.14159e+00, // angle for bg. generation with rotation
	25	float injScale = 1.,//0.7, // inject injScale*Ncl(Lr1/Lr2) hits
	26	Bool_t scaleDTheta = kTRUE, // scale dTheta by 1/sin^2(theta) in trackleting
	27	float nStdDev = 25., // number of st.dev. for tracklet cut to keep
	28	float dphi = 0.06, // dphi window (sigma of tracklet cut)
	29	float dtht = 0.025, // dtheta .... (if negative, abs will be used with additional cut on \|dthetaX\|, apart from w.distance
	30	float phishift = 0.0045, // bending shift
	31	Bool_t remOvl = kTRUE,
	32	float ovlPhiCut = 0.005,
	33	float ovlZetaCut = 0.05,
	34	Int_t nEventsSkip = 0,
	35	//----------------------- Ntracklets selection parameters important for mixing, to be tuned
	36	Float_t ntMin = 1, // process events with ESDmult
	37	Float_t ntMax = 20000, // within this range
	38	Float_t ntMixBinSz = 20000, // ESDMult bin size for mixing
	39	//----------------------- Zv selection parameters important for mixing, to be tuned
	40	Float_t zMixBinSz = 14, //0.1, // Zv. bin for mixing
	41	//---------------------------------------------------------------------------------
	42	//
	43	Bool_t checkReconstructables = kFALSE//kTRUE, // fill histos for reconstructable (needs useMC and doRec)
	44	//
	45	)
	46	{
	47	//
	48	if (cutSigDPhiS<0) cutSigDPhiS = TMath::Sqrt(cutSigNStd)*dphi;
	49	//
db1bbbe1	50	needRecPoints = doRec \|\| doInj \|\| doRot \|\| doMix;
db1bbbe1	51	//
6c56e693	52	printf("Start Analysis for %s, max %d Events skipping %d, Event Cuts: %.1f<eta<%.1f, %.2f<Zv<%.2f\n",
6c56e693	53	dataset.Data(),nEvents,nEventsSkip,etaMin,etaMax,zMin,zMax);
a9a39f46	54	printf("Centrality variable: %d\n",useCentVar);
	55	printf("Tracklet cuts: dPhi:%.3f dTheta:%.3f phiShift:%.4f \| Keep %.1f NstDev\n"
	56	"Scale dTheta: %s \| Signal Selection: NstDev:%.1f, dPhiS: %.3f\n",
	57	dphi,dtht,phishift,nStdDev,scaleDTheta ? "ON":"OFF",
	58	cutSigNStd,cutSigDPhiS);
	59	//
	60	printf("UseMC: %s. V0 scale: %.4f\n",useMC ? "ON":"OFF",scaleMCV0);
	61	printf("Operations: \n"
	62	"Reco:%s (RemOvl:%s phi:%.3f zeta:%.3f)\n"
	63	"Inj:%s (scale: %.2f)\n"
	64	"Rot:%s (phi: %.4f)\n"
	65	"Mix:%s (Nt:%.1f:%.1f:%.1f Zv:%.1f:%.1f:%.2f)\n",
	66	doRec ? "ON":"OFF",remOvl? "ON":"OFF",ovlPhiCut,ovlZetaCut,
	67	doInj ? "ON":"OFF",injScale,
	68	doRot ? "ON":"OFF",phiRot,
	69	doMix ? "ON":"OFF",ntMin,ntMax,ntMixBinSz, zMin,zMax,zMixBinSz);
	70	//
	71	if (nEvents<0) nEvents = int(1e9);
	72	TString format = GetFormatFromDataSet(dataset);
	73	//
	74	// ALICE stuff
	75	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	76	if (!mgr) mgr = new AliAnalysisManager("Test train");
	77	//
	78	InputHandlerSetup(format,useMC);
	79	if (doMix) MixHandlerSetup(ntMin,ntMax,ntMixBinSz, zMin,zMax,zMixBinSz);
	80	// compile our task
	81	gProof->Load("AliITSMultRecBg.cxx++");
	82	gProof->Load("AliTrackletTaskMulti.cxx++");
	83	//
6c56e693	84	printf("Loading Centrality task\n");
1c06a186	85	gROOT->LoadMacro("$ALICE_ROOT/OADB/macros/AddTaskCentrality.C");
6c56e693	86	AliCentralitySelectionTask *taskCentrality = AddTaskCentrality();
db1bbbe1	87	// taskCentrality->SetDebugLevel(2);
6c56e693	88	if (useMC) taskCentrality->SetMCInput();
	89	// taskCentrality->Dump();
	90	//
a9a39f46	91	// load and run AddTask macro
	92	gROOT->LoadMacro("AddMultTaskTrackletMulti.C");
	93	//
	94	// create our task
	95	AliTrackletTaskMulti *mltTask = AddMultTaskTrackletMulti(outFName.Data());
	96	//
	97	mltTask->SetUseCentralityVar(useCentVar);
	98	mltTask->SetDoNormalReco(doRec);
	99	mltTask->SetDoInjection(doInj);
	100	mltTask->SetDoRotation(doRot);
	101	mltTask->SetDoMixing(doMix);
	102	//
	103	mltTask->SetUseMC(useMC);
	104	mltTask->SetCheckReconstructables(checkReconstructables);
	105	//
6c56e693	106	mltTask->SetEtaMin(etaMin);
6c56e693	107	mltTask->SetEtaMax(etaMax);
a9a39f46	108	mltTask->SetZVertexMin(zMin);
	109	mltTask->SetZVertexMax(zMax);
	110	//
	111	mltTask->SetDPhiSCut(cutSigDPhiS);
	112	mltTask->SetNStdCut(cutSigNStd);
	113	mltTask->SetScaleMCV0(scaleMCV0);
	114	//
	115	mltTask->SetScaleDThetaBySin2T(scaleDTheta);
	116	mltTask->SetNStdDev(nStdDev);
	117	mltTask->SetPhiWindow(dphi);
	118	mltTask->SetThetaWindow(dtht);
	119	mltTask->SetPhiShift(phishift);
	120	mltTask->SetPhiOverlapCut(ovlPhiCut);
	121	mltTask->SetZetaOverlapCut(ovlZetaCut);
	122	mltTask->SetPhiRot(phiRot);
	123	mltTask->SetInjScale(injScale);
	124	mltTask->SetRemoveOverlaps(remOvl);
	125	//
	126	printf("new Task: %p\n",mltTask);
	127	//
6c56e693	128	printf("Requesting physics selection in %s mode\n",useMC ? "MC":"Data");
63b6cbd0	129	gROOT->ProcessLine(".L $ALICE_ROOT/OADB/macros/AddTaskPhysicsSelection.C");
6c56e693	130	AliPhysicsSelectionTask* physicsSelectionTask = AddTaskPhysicsSelection(useMC,0);
6c56e693	131	mltTask->SelectCollisionCandidates();//AliVEvent::kMB);
a9a39f46	132	//
	133	// Run analysis
	134	mgr->InitAnalysis();
	135	// process dataset
	136	mgr->StartAnalysis("proof", dataset.Data(), nEvents, nEventsSkip);
	137	//
	138	TString evstCmd = "if [ -e event_stat.root ]; then \nmv event_stat.root evstat_";
	139	evstCmd += outFName; evstCmd += " \nfi";
	140	gSystem->Exec( evstCmd.Data() );
	141
	142	}
	143
	144
	145	TString GetFormatFromDataSet(TString dataset) {
	146
	147	// Info("runAAF.C","Detecting format from dataset (may take while, depends on network connection)...");
	148	TString dsTreeName;
	149	if (dataset.Contains("#")) {
	150	Info("runAAF.C",Form("Detecting format from dataset name '%s' ...",dataset.Data()));
	151	dsTreeName=dataset(dataset.Last('#'),dataset.Length());
	152	} else {
	153	Info("runAAF.C",Form("Detecting format from dataset '%s' (may take while, depends on network connection) ...",dataset.Data()));
	154	TFileCollection *ds = gProof->GetDataSet(dataset.Data());
	155	if (!ds) {
	156	Error(Form("Dataset %s doesn't exist on proof cluster!!!!",dataset.Data()));
	157	return "";
	158	}
	159	dsTreeName = ds->GetDefaultTreeName();
	160	}
	161
	162	if (dsTreeName.Contains("esdTree")) {
	163	Info("runAAF.C","ESD input format detected ...");
	164	return "ESD";
	165	} else if (dsTreeName.Contains("aodTree")) {
	166	Info("runAAF.C","AOD input format detected ...");
	167	return "AOD";
	168	} else {
	169	Error("runAAF.C",Form("Tree %s is not supported !!!",dsTreeName.Data()));
	170	Error("runAAF.C",Form("Maybe set your DS to %s#esdTree or %s#aodTree",dataset.Data(),dataset.Data()));
	171	}
	172
	173	return "";
	174	}
	175
	176	Bool_t InputHandlerSetup(TString format = "esd", Bool_t useKine = kTRUE)
	177	{
	178	format.ToLower();
	179
	180	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	181
	182	AliAnalysisDataContainer *cin = mgr->GetCommonInputContainer();
	183
	184	if (cin) return;
	185
	186	if (!format.CompareTo("esd"))
	187	{
	188	AliESDInputHandler esdInputHandler = dynamic_cast<AliESDInputHandler>(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
	189
	190	if (!esdInputHandler)
	191	{
	192	Info("CustomAnalysisTaskInputSetup", "Creating esdInputHandler ...");
db1bbbe1	193	if (needRecPoints)
	194	esdInputHandler = new AliESDInputHandlerRP();
	195	else
	196	esdInputHandler = new AliESDInputHandler();
	197	//
a9a39f46	198	mgr->SetInputEventHandler(esdInputHandler);
a9a39f46	199	}
db1bbbe1	200	//
a9a39f46	201	if (useKine)
	202	{
	203	AliMCEventHandler* mcInputHandler = dynamic_cast<AliMCEventHandler*>(AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
	204
	205	if (!mcInputHandler)
	206	{
	207	Info("CustomAnalysisTaskInputSetup", "Creating mcInputHandler ...");
	208	AliMCEventHandler* mcInputHandler = new AliMCEventHandler();
	209	mgr->SetMCtruthEventHandler(mcInputHandler);
	210	}
	211	}
	212
	213	}
	214	else if (!format.CompareTo("aod"))
	215	{
	216	AliAODInputHandler aodInputHandler = dynamic_cast<AliAODInputHandler>(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
	217
	218	if (!aodInputHandler)
	219	{
	220	Info("CustomAnalysisTaskInputSetup", "Creating aodInputHandler ...");
	221	aodInputHandler = new AliAODInputHandler();
	222	mgr->SetInputEventHandler(aodInputHandler);
	223	}
	224	}
	225	else
	226	{
	227	Info("Wrong input format!!! Only ESD and AOD are supported. Skipping Task ...");
	228	return kFALSE;
	229	}
	230
	231	return kTRUE;
	232	}
	233
	234	void MixHandlerSetup(float ntMin,float ntMax,float ntMixBinSz,
	235	float zMin, float zMax, float zMixBinSz)
	236	{
	237	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	238	if (!mgr) return;
	239	int bufferSize = 1;
	240	AliESDInputHandlerRP esdH = dynamic_cast<AliESDInputHandlerRP>(mgr->GetInputEventHandler());
	241	if (!esdH) return;
	242	//
	243	AliMixEventInputHandler *esdMixH = new AliMixEventInputHandler(bufferSize);
	244	esdMixH->SetInputHandlerForMixing(esdH);
	245	AliMixEventPool *evPool = new AliMixEventPool("MyPool");
	246	AliMixEventCutObj *tracklets = new AliMixEventCutObj(AliMixEventCutObj::kNumberTracklets, ntMin,ntMax,ntMixBinSz);
	247	AliMixEventCutObj *zvertex = new AliMixEventCutObj(AliMixEventCutObj::kZVertex, zMin,zMax, zMixBinSz);
	248	// evPool->AddCut(tracklets);
	249	evPool->AddCut(zvertex);
	250	//evPool->Init();
	251	evPool->Print();
	252	esdMixH->SetEventPool(evPool);
	253	esdH->SetMixingHandler(esdMixH);
	254	}
	255
	256	void AddPhysicsSelection(Bool_t isMC)
	257	{
	258	// physics selection a la Michele
a9a39f46	259	if(!isMC) {
6c56e693	260	//AliPhysicsSelection * physSel = physicsSelectionTask->GetPhysicsSelection();
6c56e693	261	// physSel->AddCollisionTriggerClass("+CMBAC-B-NOPF-ALL");
a9a39f46	262	/*
	263	physSel->AddCollisionTriggerClass("+CMBS1C-B-NOPF-ALL");
	264	physSel->AddCollisionTriggerClass("+CMBS1A-B-NOPF-ALL");
	265	*/
	266	//
6c56e693	267	// physSel->AddCollisionTriggerClass("+CMBS2C-B-NOPF-ALL");
6c56e693	268	// physSel->AddCollisionTriggerClass("+CMBS2A-B-NOPF-ALL");
a9a39f46	269	//
a9a39f46	270	// This are needed only to fill the statistics tables
6c56e693	271	// physSel->AddBGTriggerClass("+CMBAC-C-NOPF-ALL");
	272	// physSel->AddBGTriggerClass("+CMBAC-A-NOPF-ALL");
	273	// physSel->AddBGTriggerClass("+CMBAC-E-NOPF-ALL");
a9a39f46	274	//
	275	/*
	276	physSel->AddBGTriggerClass("+CMBS1C-C-NOPF-ALL");
	277	physSel->AddBGTriggerClass("+CMBS1C-A-NOPF-ALL");
	278	physSel->AddBGTriggerClass("+CMBS1C-E-NOPF-ALL");
	279	//
	280	physSel->AddBGTriggerClass("+CMBS1A-C-NOPF-ALL");
	281	physSel->AddBGTriggerClass("+CMBS1A-A-NOPF-ALL");
	282	physSel->AddBGTriggerClass("+CMBS1A-E-NOPF-ALL");
	283	//
	284	*/
	285	/*
	286	//
	287	physSel->AddBGTriggerClass("+CMBS2C-C-NOPF-ALL");
	288	physSel->AddBGTriggerClass("+CMBS2C-A-NOPF-ALL");
	289	physSel->AddBGTriggerClass("+CMBS2C-E-NOPF-ALL");
	290	//
	291	physSel->AddBGTriggerClass("+CMBS2A-C-NOPF-ALL");
	292	physSel->AddBGTriggerClass("+CMBS2A-A-NOPF-ALL");
	293	physSel->AddBGTriggerClass("+CMBS2A-E-NOPF-ALL");
	294	*/
	295	}
	296	// if you use the following line, your task only gets the selected events
	297	// task->SelectCollisionCandidates(AliVEvent::kUserDefined);
6c56e693	298	// task->SelectCollisionCandidates();
a9a39f46	299	//
	300	//Alternatively, in the UserExec of your task:
	301	//Bool_t isSelected = (((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected() & AliVEvent::kUserDefined);
	302	//
	303	}