[u/mrichter/AliRoot.git] / PWGPP / EVCHAR / runProofSPDdNdEta.C

void runProofSPDdNdEta_CorrRef4(
                       TString  proofCluster  = "mnicassi@alice-caf.cern.ch", //skaf.saske.sk",
                       TString  alirootVer    = "VO_ALICE@AliRoot::v4-21-04-AN", 
                       TString  rootVer       = "VO_ALICE@ROOT::v5-27-06a-1",
//                     TString  dataset       = "/alice/data/LHC10h_000137161_p1_plusplusplus", 
                       TString  dataset       = "/alice/sim/LHC10h8_000137161",
                       TString  outFileCorr   = "SPDdNdEtaCorr.root",
//                       TString  outFileCorr   = "SPDdNdEtaCorrRot.root",
                       TString  outFileData   = "SPDdNdEtaData.root",
                       TString  percentFile   = "./AliCentralityBy1D_LHC10g2a_100.root",
                       TString  percentFile2  = "./AliCentralityByFunction_LHC10g2a_100.root",
                       Bool_t   useMC         = kTRUE, 
                       Bool_t   readtr        = kFALSE, 
                       Bool_t   recotracklets = kTRUE, 
                       Bool_t   dataonalien   = kFALSE,
                       Float_t  centrlowlim   = 0., 
                       Float_t  centruplim    = 5., 
                       Bool_t   centrest      = kFALSE,
                       Int_t    minClMultLay2 = 4300, 
//                       Int_t    minClMultLay2 = -1,  
                       Int_t    maxClMultLay2 = 1.0*1e5, 
                       Int_t    minV0Mult     = 14674.5,
                       Float_t vtxlim         = 7.,
                       Bool_t partsp          = kTRUE, 
                       Float_t  phiWindow     = 0.8,
                       Float_t  thetaWindow   = 0.025,
                       Float_t  phiShift      = 0.0045,
                       Float_t  removeClFOvl  = kFALSE,
                       Float_t  phiOvlCut     = 0.005,
                       Float_t  zetaOvlCut    = 0.05,
                       Float_t  phiRotAngle   = 0.,  
//                       Float_t  phiRotAngle   = TMath::Pi(),
                       Float_t  phiWindowAna  = 0.08,
                       Int_t    nEvents       = 1.0*1e7, 
                       Int_t    nEventsSkip   = 0) { //1.0*1e7

  gEnv->SetValue("XSec.GSI.DelegProxy","2");

  TString alirootMode = "";    // STEERBase,ESD,AOD,ANALYSIS,ANALYSISalice (default aliroot mode)
  TString extraLibs;
  TList *list = new TList();
  if (recotracklets) {
    alirootMode="REC";     // $ALICE_ROOT/macros/loadlibsrec.C
    extraLibs= "ITSrec:CDB:Geom:"; // not needed in default aliroot mode
  } else alirootMode="ALIROOT";
  extraLibs+= "ANALYSIS:ANALYSISalice";

  // sets $ALIROOT_MODE on each worker to let proof to know to run in special mode
  list->Add(new TNamed("ALIROOT_MODE", alirootMode.Data()));
  list->Add(new TNamed("ALIROOT_EXTRA_LIBS", extraLibs.Data()));
  if (recotracklets||dataonalien) list->Add(new TNamed("ALIROOT_ENABLE_ALIEN", "1"));

  // REM: same version of AliRoot on client!
//  TProof::Mgr(proofCluster.Data())->SetROOTVersion(rootVer.Data()); //If not using the default version
  TProof::Open(proofCluster.Data());
  // enable n workers per machine
//  TProof::Open(proofCluster.Data(),"workers=nx")       
  // enable less workers
//  TProof::Open(proofCluster.Data(),"workers=20"); //For performance reasons, try to avoid it.
  if (!gProof) {
    Error("runSPDdNdEtaAna.C","Connection to AF failed.");
    return;
  }
  gProof->EnablePackage(alirootVer.Data(), list);

//  gROOT->LoadMacro("AnalysisMacroa.C");
  Analysis(dataset.Data(), outFileCorr, outFileData, percentFile, percentFile2, 
           useMC, readtr, recotracklets, 
           nEvents, nEventsSkip, centrlowlim, centruplim, centrest, minClMultLay2, maxClMultLay2, minV0Mult, vtxlim, partsp,
           phiWindow, thetaWindow, phiShift, removeClFOvl, phiOvlCut, zetaOvlCut, phiRotAngle,phiWindowAna);

}

//________________________________________________________________________
void Analysis(TString dataset, TString outFileCorr, TString outFileData, TString percentFile, TString percentFile2, 
              Bool_t useMC, Bool_t readtr, Bool_t recotracklets, 
              Int_t nEvents, Int_t nEventsSkip, 
              Float_t centrlowlim, Float_t centruplim, Bool_t centrest, Int_t minClMultLay2, Int_t maxClMultLay2, Int_t minV0Mult, Float_t vtxlim, Bool_t partsp,
              Float_t phiWindow, Float_t thetaWindow, Float_t phiShift, Bool_t removeClFOvl, 
              Float_t phiOvlCut, Float_t zetaOvlCut, Float_t phiRotAngle, Float_t phiWindowAna) {

  TString format = GetFormatFromDataSet(dataset);

  // ALICE stuff
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr) mgr = new AliAnalysisManager("Test train");

  InputHandlerSetup(format,useMC,recotracklets);

  // compile the tracklet reconstruction class
  gProof->Load("AliTrackletAlg.cxx++");
  // compile analysis task
  gProof->Load("AliAnalysisTaskSPDdNdEta.cxx++");

  // create manager
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr) mgr = new AliAnalysisManager("SPD analysis");

  // create task
  AliAnalysisTaskSPDdNdEta *task = new AliAnalysisTaskSPDdNdEta("AliAnalysisTaskSPDdNdEta");

  AliTriggerAnalysis::Trigger trigg = AliTriggerAnalysis::kAcceptAll; // to be changed every time
  AliAnalysisTaskSPDdNdEta::MCCentralityBin kmccentrbin = AliAnalysisTaskSPDdNdEta::kall; // idem

  task->SetReadMC(useMC);
  task->SetTrigger(trigg);
  task->SetReadTrackRefs(readtr);
  task->SetRecoTracklets(recotracklets);
  task->SetMCCentralityBin(kmccentrbin);
  task->SetCentralityLowLim(centrlowlim);
  task->SetCentralityUpLim(centruplim);
  task->SetCentralityEst(centrest);
  task->SetMinClusterMultLay2(minClMultLay2);
  task->SetMaxClusterMultLay2(maxClMultLay2);
  task->SetMinV0Mult(minV0Mult); 
  task->SetVertexRange(vtxlim);
  task->SetPartSpecies(partsp);

  task->SetPhiWindow(phiWindow);
  task->SetThetaWindow(thetaWindow); 
  task->SetPhiShift(phiShift);
  task->SetRemoveClustersFromOverlaps(removeClFOvl); 
  task->SetPhiOverlapCut(phiOvlCut);
  task->SetZetaOverlapCut(zetaOvlCut);
  task->SetPhiRotationAngle(phiRotAngle);
  task->SetPhiWindowAna(phiWindowAna);

  // physics selection
  gROOT->ProcessLine(".L $ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
  AliPhysicsSelectionTask* physSelTask=AddTaskPhysicsSelection(useMC,kFALSE);
  if (!useMC) {
    AliPhysicsSelection * physSel = physSelTask->GetPhysicsSelection();
  
/*    if (dataset=="/alice/data/LHC10h_000137045_p1_plus") {
      physSel->AddCollisionTriggerClass("+CMBS1A-B-NOPF-ALL");
      physSel->AddCollisionTriggerClass("+CMBS1C-B-NOPF-ALL");
      physSel->AddCollisionTriggerClass("+CMBAC-B-NOPF-ALL");

    } else {*/
      physSel->AddCollisionTriggerClass("+CMBS2A-B-NOPF-ALL");
      physSel->AddCollisionTriggerClass("+CMBS2C-B-NOPF-ALL");
      physSel->AddCollisionTriggerClass("+CMBAC-B-NOPF-ALL");
//    }

    task->SelectCollisionCandidates(AliVEvent::kUserDefined);
  } else {

    task->SelectCollisionCandidates(AliVEvent::kMB);
  }

  // centrality selection
  gProof->Load("./AliCentralitySelectionTask.cxx++g");
  AliCentralitySelectionTask *taskCentr = new AliCentralitySelectionTask("CentralitySelection");
  taskCentr->SetPercentileFile(percentFile);
  taskCentr->SetPercentileFile2(percentFile2);

  mgr->AddTask(taskCentr);
  mgr->ConnectInput (taskCentr,0, mgr->GetCommonInputContainer());

  // create output container
  if (useMC) AliAnalysisDataContainer *output = mgr->CreateContainer("cOutput", TList::Class(), AliAnalysisManager::kOutputContainer, outFileCorr);
  else       AliAnalysisDataContainer *output = mgr->CreateContainer("cOutput", TList::Class(), AliAnalysisManager::kOutputContainer, outFileData);

  // add task to the manager
  mgr->AddTask(task);

  // connect input and output
  mgr->ConnectInput(task, 0, mgr->GetCommonInputContainer());
  mgr->ConnectOutput(task, 1, output);

  // run analysis
  mgr->InitAnalysis();
  // process dataset  
  mgr->StartAnalysis("proof", dataset.Data(), nEvents, nEventsSkip);

}

//________________________________________________________________________
TString GetFormatFromDataSet(TString dataset) {

//  Info("runAAF.C","Detecting format from dataset (may take while, depends on network connection)...");
  TString dsTreeName;
  if (dataset.Contains("#")) {
    Info("runSPDdNdEta.C",Form("Detecting format from dataset name '%s' ...",dataset.Data()));
    dsTreeName=dataset(dataset.Last('#'),dataset.Length());
  } else {
    Info("runSPDdNdEta.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("runSPDdNdEta.C","ESD input format detected ...");
    return "ESD";
  } else if (dsTreeName.Contains("aodTree"))  {
    Info("runSPDdNdEta.C","AOD input format detected ...");
    return "AOD";
  } else {
    Error("runSPDdNdEta.C",Form("Tree %s is not supported !!!",dsTreeName.Data()));
    Error("runSPDdNdEta.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, Bool_t recotracklets)
{
  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(recotracklets) 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
  {
    AliWarning("Wrong input format!!! Only ESD and AOD are supported. Skipping Task ...");
    return kFALSE;
  }

  return kTRUE;
}

//________________________________________________________________________
// MEMO DATASETS ...
/*
/alice/data/LHC10h_000137045_p1           |      60 | /esdTree     | 5.144e+05|    12 GB |  100 %
/alice/data/LHC10h_000137045_p1_lowflux   |      50 | /esdTree     | 4.144e+05|    10 GB |  100 %
/alice/data/LHC10h_000137045_p1_plus      |      53 | /esdTree     | 4.444e+05|    10 GB |  100 %
/alice/data/LHC10h_000137124_p1           |      89 | /esdTree     | 7.152e+04|     3 GB |  100 %
/alice/data/LHC10h_000137125_p1           |      41 | /esdTree     | 1.537e+04|     1 GB |  100 %
/alice/data/LHC10h_000137132_p1           |      87 | /esdTree     | 8.881e+04|     8 GB |  100 %
/alice/data/LHC10h_000137133_p1           |     306 | /esdTree     | 3.521e+05|    29 GB |  100 %
/alice/data/LHC10h_000137161              |    3348 | /esdTree     | 6.279e+05|   407 GB |   97 %

/alice/sim/LHC10h1_000137045              |    1198 | /esdTree     | 1.198e+04|   138 GB |  100 %
/alice/sim/LHC10h1a_000137045             |    1022 | /esdTree     | 1.006e+04|   124 GB |   98 %

*/
//________________________________________________________________________
Commit	Line	Data
ccfafc80	1	void runProofSPDdNdEta_CorrRef4(
	2	TString proofCluster = "mnicassi@alice-caf.cern.ch", //skaf.saske.sk",
	3	TString alirootVer = "VO_ALICE@AliRoot::v4-21-04-AN",
	4	TString rootVer = "VO_ALICE@ROOT::v5-27-06a-1",
	5	// TString dataset = "/alice/data/LHC10h_000137161_p1_plusplusplus",
	6	TString dataset = "/alice/sim/LHC10h8_000137161",
86841765	7	TString outFileCorr = "SPDdNdEtaCorr.root",
ccfafc80	8	// TString outFileCorr = "SPDdNdEtaCorrRot.root",
86841765	9	TString outFileData = "SPDdNdEtaData.root",
	10	TString percentFile = "./AliCentralityBy1D_LHC10g2a_100.root",
	11	TString percentFile2 = "./AliCentralityByFunction_LHC10g2a_100.root",
ccfafc80	12	Bool_t useMC = kTRUE,
86841765	13	Bool_t readtr = kFALSE,
	14	Bool_t recotracklets = kTRUE,
	15	Bool_t dataonalien = kFALSE,
	16	Float_t centrlowlim = 0.,
	17	Float_t centruplim = 5.,
ccfafc80	18	Bool_t centrest = kFALSE,
	19	Int_t minClMultLay2 = 4300,
	20	// Int_t minClMultLay2 = -1,
	21	Int_t maxClMultLay2 = 1.0*1e5,
	22	Int_t minV0Mult = 14674.5,
	23	Float_t vtxlim = 7.,
	24	Bool_t partsp = kTRUE,
86841765	25	Float_t phiWindow = 0.8,
	26	Float_t thetaWindow = 0.025,
	27	Float_t phiShift = 0.0045,
	28	Float_t removeClFOvl = kFALSE,
	29	Float_t phiOvlCut = 0.005,
	30	Float_t zetaOvlCut = 0.05,
ccfafc80	31	Float_t phiRotAngle = 0.,
ccfafc80	32	// Float_t phiRotAngle = TMath::Pi(),
86841765	33	Float_t phiWindowAna = 0.08,
	34	Int_t nEvents = 1.0*1e7,
	35	Int_t nEventsSkip = 0) { //1.0*1e7
	36
	37	gEnv->SetValue("XSec.GSI.DelegProxy","2");
	38
	39	TString alirootMode = ""; // STEERBase,ESD,AOD,ANALYSIS,ANALYSISalice (default aliroot mode)
	40	TString extraLibs;
	41	TList *list = new TList();
	42	if (recotracklets) {
	43	alirootMode="REC"; // $ALICE_ROOT/macros/loadlibsrec.C
	44	extraLibs= "ITSrec:CDB:Geom:"; // not needed in default aliroot mode
	45	} else alirootMode="ALIROOT";
	46	extraLibs+= "ANALYSIS:ANALYSISalice";
	47
	48	// sets $ALIROOT_MODE on each worker to let proof to know to run in special mode
	49	list->Add(new TNamed("ALIROOT_MODE", alirootMode.Data()));
	50	list->Add(new TNamed("ALIROOT_EXTRA_LIBS", extraLibs.Data()));
	51	if (recotracklets\|\|dataonalien) list->Add(new TNamed("ALIROOT_ENABLE_ALIEN", "1"));
	52
	53	// REM: same version of AliRoot on client!
	54	// TProof::Mgr(proofCluster.Data())->SetROOTVersion(rootVer.Data()); //If not using the default version
	55	TProof::Open(proofCluster.Data());
	56	// enable n workers per machine
	57	// TProof::Open(proofCluster.Data(),"workers=nx")
	58	// enable less workers
	59	// TProof::Open(proofCluster.Data(),"workers=20"); //For performance reasons, try to avoid it.
	60	if (!gProof) {
	61	Error("runSPDdNdEtaAna.C","Connection to AF failed.");
	62	return;
	63	}
	64	gProof->EnablePackage(alirootVer.Data(), list);
c6788f58	65
86841765	66	// gROOT->LoadMacro("AnalysisMacroa.C");
	67	Analysis(dataset.Data(), outFileCorr, outFileData, percentFile, percentFile2,
	68	useMC, readtr, recotracklets,
ccfafc80	69	nEvents, nEventsSkip, centrlowlim, centruplim, centrest, minClMultLay2, maxClMultLay2, minV0Mult, vtxlim, partsp,
86841765	70	phiWindow, thetaWindow, phiShift, removeClFOvl, phiOvlCut, zetaOvlCut, phiRotAngle,phiWindowAna);
c6788f58	71
	72	}
	73
86841765	74	//________________________________________________________________________
	75	void Analysis(TString dataset, TString outFileCorr, TString outFileData, TString percentFile, TString percentFile2,
	76	Bool_t useMC, Bool_t readtr, Bool_t recotracklets,
	77	Int_t nEvents, Int_t nEventsSkip,
ccfafc80	78	Float_t centrlowlim, Float_t centruplim, Bool_t centrest, Int_t minClMultLay2, Int_t maxClMultLay2, Int_t minV0Mult, Float_t vtxlim, Bool_t partsp,
86841765	79	Float_t phiWindow, Float_t thetaWindow, Float_t phiShift, Bool_t removeClFOvl,
86841765	80	Float_t phiOvlCut, Float_t zetaOvlCut, Float_t phiRotAngle, Float_t phiWindowAna) {
c6788f58	81
	82	TString format = GetFormatFromDataSet(dataset);
	83
	84	// ALICE stuff
	85	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	86	if (!mgr) mgr = new AliAnalysisManager("Test train");
	87
	88	InputHandlerSetup(format,useMC,recotracklets);
	89
	90	// compile the tracklet reconstruction class
	91	gProof->Load("AliTrackletAlg.cxx++");
	92	// compile analysis task
	93	gProof->Load("AliAnalysisTaskSPDdNdEta.cxx++");
	94
	95	// create manager
	96	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	97	if (!mgr) mgr = new AliAnalysisManager("SPD analysis");
	98
	99	// create task
	100	AliAnalysisTaskSPDdNdEta *task = new AliAnalysisTaskSPDdNdEta("AliAnalysisTaskSPDdNdEta");
	101
	102	AliTriggerAnalysis::Trigger trigg = AliTriggerAnalysis::kAcceptAll; // to be changed every time
	103	AliAnalysisTaskSPDdNdEta::MCCentralityBin kmccentrbin = AliAnalysisTaskSPDdNdEta::kall; // idem
	104
	105	task->SetReadMC(useMC);
	106	task->SetTrigger(trigg);
c6788f58	107	task->SetReadTrackRefs(readtr);
	108	task->SetRecoTracklets(recotracklets);
	109	task->SetMCCentralityBin(kmccentrbin);
	110	task->SetCentralityLowLim(centrlowlim);
	111	task->SetCentralityUpLim(centruplim);
	112	task->SetCentralityEst(centrest);
86841765	113	task->SetMinClusterMultLay2(minClMultLay2);
ccfafc80	114	task->SetMaxClusterMultLay2(maxClMultLay2);
	115	task->SetMinV0Mult(minV0Mult);
	116	task->SetVertexRange(vtxlim);
	117	task->SetPartSpecies(partsp);
86841765	118
	119	task->SetPhiWindow(phiWindow);
	120	task->SetThetaWindow(thetaWindow);
	121	task->SetPhiShift(phiShift);
	122	task->SetRemoveClustersFromOverlaps(removeClFOvl);
	123	task->SetPhiOverlapCut(phiOvlCut);
	124	task->SetZetaOverlapCut(zetaOvlCut);
	125	task->SetPhiRotationAngle(phiRotAngle);
	126	task->SetPhiWindowAna(phiWindowAna);
c6788f58	127
c6788f58	128	// physics selection
86841765	129	gROOT->ProcessLine(".L $ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
	130	AliPhysicsSelectionTask* physSelTask=AddTaskPhysicsSelection(useMC,kFALSE);
	131	if (!useMC) {
	132	AliPhysicsSelection * physSel = physSelTask->GetPhysicsSelection();
	133
ccfafc80	134	/* if (dataset=="/alice/data/LHC10h_000137045_p1_plus") {
86841765	135	physSel->AddCollisionTriggerClass("+CMBS1A-B-NOPF-ALL");
	136	physSel->AddCollisionTriggerClass("+CMBS1C-B-NOPF-ALL");
	137	physSel->AddCollisionTriggerClass("+CMBAC-B-NOPF-ALL");
	138
ccfafc80	139	} else {*/
86841765	140	physSel->AddCollisionTriggerClass("+CMBS2A-B-NOPF-ALL");
	141	physSel->AddCollisionTriggerClass("+CMBS2C-B-NOPF-ALL");
	142	physSel->AddCollisionTriggerClass("+CMBAC-B-NOPF-ALL");
ccfafc80	143	// }
86841765	144
	145	task->SelectCollisionCandidates(AliVEvent::kUserDefined);
	146	} else {
	147
	148	task->SelectCollisionCandidates(AliVEvent::kMB);
	149	}
c6788f58	150
c6788f58	151	// centrality selection
86841765	152	gProof->Load("./AliCentralitySelectionTask.cxx++g");
c6788f58	153	AliCentralitySelectionTask *taskCentr = new AliCentralitySelectionTask("CentralitySelection");
86841765	154	taskCentr->SetPercentileFile(percentFile);
	155	taskCentr->SetPercentileFile2(percentFile2);
	156
c6788f58	157	mgr->AddTask(taskCentr);
	158	mgr->ConnectInput (taskCentr,0, mgr->GetCommonInputContainer());
	159
c6788f58	160	// create output container
86841765	161	if (useMC) AliAnalysisDataContainer *output = mgr->CreateContainer("cOutput", TList::Class(), AliAnalysisManager::kOutputContainer, outFileCorr);
86841765	162	else AliAnalysisDataContainer *output = mgr->CreateContainer("cOutput", TList::Class(), AliAnalysisManager::kOutputContainer, outFileData);
c6788f58	163
	164	// add task to the manager
	165	mgr->AddTask(task);
	166
	167	// connect input and output
	168	mgr->ConnectInput(task, 0, mgr->GetCommonInputContainer());
	169	mgr->ConnectOutput(task, 1, output);
	170
	171	// run analysis
	172	mgr->InitAnalysis();
	173	// process dataset
	174	mgr->StartAnalysis("proof", dataset.Data(), nEvents, nEventsSkip);
	175
	176	}
	177
86841765	178	//________________________________________________________________________
c6788f58	179	TString GetFormatFromDataSet(TString dataset) {
c6788f58	180
86841765	181	// Info("runAAF.C","Detecting format from dataset (may take while, depends on network connection)...");
c6788f58	182	TString dsTreeName;
	183	if (dataset.Contains("#")) {
	184	Info("runSPDdNdEta.C",Form("Detecting format from dataset name '%s' ...",dataset.Data()));
	185	dsTreeName=dataset(dataset.Last('#'),dataset.Length());
	186	} else {
	187	Info("runSPDdNdEta.C",Form("Detecting format from dataset '%s' (may take while, depends on network connection) ...",dataset.Data()));
	188	TFileCollection *ds = gProof->GetDataSet(dataset.Data());
	189	if (!ds) {
	190	Error(Form("Dataset %s doesn't exist on proof cluster!!!!",dataset.Data()));
	191	return "";
	192	}
	193	dsTreeName = ds->GetDefaultTreeName();
	194	}
	195
	196	if (dsTreeName.Contains("esdTree")) {
	197	Info("runSPDdNdEta.C","ESD input format detected ...");
	198	return "ESD";
	199	} else if (dsTreeName.Contains("aodTree")) {
	200	Info("runSPDdNdEta.C","AOD input format detected ...");
	201	return "AOD";
	202	} else {
	203	Error("runSPDdNdEta.C",Form("Tree %s is not supported !!!",dsTreeName.Data()));
	204	Error("runSPDdNdEta.C",Form("Maybe set your DS to %s#esdTree or %s#aodTree",dataset.Data(),dataset.Data()));
	205	}
	206
	207	return "";
	208	}
	209
86841765	210	//________________________________________________________________________
c6788f58	211	Bool_t InputHandlerSetup(TString format = "esd", Bool_t useKine = kTRUE, Bool_t recotracklets)
	212	{
	213	format.ToLower();
	214
	215	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	216
	217	AliAnalysisDataContainer *cin = mgr->GetCommonInputContainer();
	218
	219	if (cin) return;
	220
	221	if (!format.CompareTo("esd"))
	222	{
	223	AliESDInputHandler esdInputHandler = dynamic_cast<AliESDInputHandler>(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
	224
	225	if (!esdInputHandler)
	226	{
	227	Info("CustomAnalysisTaskInputSetup", "Creating esdInputHandler ...");
	228	if(recotracklets) esdInputHandler = new AliESDInputHandlerRP();
	229	else esdInputHandler = new AliESDInputHandler();
	230	mgr->SetInputEventHandler(esdInputHandler);
	231	}
	232
	233	if (useKine)
	234	{
	235	AliMCEventHandler* mcInputHandler = dynamic_cast<AliMCEventHandler*>(AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
	236
	237	if (!mcInputHandler)
	238	{
	239	Info("CustomAnalysisTaskInputSetup", "Creating mcInputHandler ...");
	240	AliMCEventHandler* mcInputHandler = new AliMCEventHandler();
	241	mgr->SetMCtruthEventHandler(mcInputHandler);
	242	}
	243	}
	244
	245	}
	246	else if (!format.CompareTo("aod"))
	247	{
	248	AliAODInputHandler aodInputHandler = dynamic_cast<AliAODInputHandler>(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
	249
	250	if (!aodInputHandler)
	251	{
	252	Info("CustomAnalysisTaskInputSetup", "Creating aodInputHandler ...");
	253	aodInputHandler = new AliAODInputHandler();
	254	mgr->SetInputEventHandler(aodInputHandler);
	255	}
	256	}
	257	else
	258	{
	259	AliWarning("Wrong input format!!! Only ESD and AOD are supported. Skipping Task ...");
	260	return kFALSE;
	261	}
	262
	263	return kTRUE;
ea23c361	264	}
86841765	265
	266	//________________________________________________________________________
	267	// MEMO DATASETS ...
	268	/*
	269	/alice/data/LHC10h_000137045_p1 \| 60 \| /esdTree \| 5.144e+05\| 12 GB \| 100 %
	270	/alice/data/LHC10h_000137045_p1_lowflux \| 50 \| /esdTree \| 4.144e+05\| 10 GB \| 100 %
	271	/alice/data/LHC10h_000137045_p1_plus \| 53 \| /esdTree \| 4.444e+05\| 10 GB \| 100 %
	272	/alice/data/LHC10h_000137124_p1 \| 89 \| /esdTree \| 7.152e+04\| 3 GB \| 100 %
	273	/alice/data/LHC10h_000137125_p1 \| 41 \| /esdTree \| 1.537e+04\| 1 GB \| 100 %
	274	/alice/data/LHC10h_000137132_p1 \| 87 \| /esdTree \| 8.881e+04\| 8 GB \| 100 %
	275	/alice/data/LHC10h_000137133_p1 \| 306 \| /esdTree \| 3.521e+05\| 29 GB \| 100 %
	276	/alice/data/LHC10h_000137161 \| 3348 \| /esdTree \| 6.279e+05\| 407 GB \| 97 %
	277
	278	/alice/sim/LHC10h1_000137045 \| 1198 \| /esdTree \| 1.198e+04\| 138 GB \| 100 %
	279	/alice/sim/LHC10h1a_000137045 \| 1022 \| /esdTree \| 1.006e+04\| 124 GB \| 98 %
	280
	281	*/
ccfafc80	282	//________________________________________________________________________