[u/mrichter/AliRoot.git] / PWGCF / FLOW / macros / runFlowTaskCentralityPIDTrain.C

enum anaModes {mLocal,mPROOF,mGrid};
Int_t binfirst = 0;  //where do we start numbering bins
Int_t binlast = 8;  //where do we stop numbering bins
const Int_t numberOfCentralityBins = 9;
Float_t centralityArray[numberOfCentralityBins+1] = {0.,5.,10.,20.,30.,40.,50.,60.,70.,80.}; // in centrality percentile
//Int_t centralityArray[numberOfCentralityBins+1] = {41,80,146,245,384,576,835,1203,1471,10000}; // in terms of TPC only reference multiplicity

TString commonOutputFileName = "outputCentrality"; // e.g.: result for centrality bin 0 will be in the file "outputCentrality0.root", etc

void runFlowTaskCentralityPIDTrain( Int_t mode = mLocal,
                                    Bool_t useFlowParFiles = kFALSE,
                                    Bool_t DATA = kTRUE,
                                    const Char_t* dataDir="fileList",
                                    Int_t nEvents = 1e4,
                                    Int_t offset=0 )
{
  // Time:
  TStopwatch timer;
  timer.Start();

  // Load needed libraries:
  LoadLibraries(mode,useFlowParFiles);

  // Create analysis manager:
  AliAnalysisManager *mgr = new AliAnalysisManager("FlowAnalysisManager");

  // Chains:
  if(mode == mLocal)
  {
    gROOT->LoadMacro("$ALICE_ROOT/PWGUD/macros/CreateESDChain.C");
    TChain* chain = CreateESDChain(dataDir, nEvents, offset);
    //TChain* chain = CreateAODChain(dataDir, nEvents, offset);
  }

  // Connect plug-in to the analysis manager:
  if(mode == mGrid)
  {
    gROOT->LoadMacro("CreateAlienHandler.C");
    AliAnalysisGrid *alienHandler = CreateAlienHandler(useFlowParFiles);
    if(!alienHandler) return;
    mgr->SetGridHandler(alienHandler);
  }

  // Event handlers:
  AliVEventHandler* esdH = new AliESDInputHandler;
  mgr->SetInputEventHandler(esdH);
  if (!DATA)
  {
    AliMCEventHandler *mc = new AliMCEventHandler();
    mgr->SetMCtruthEventHandler(mc);
  }

  // Task to check the offline trigger:
  gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
  AddTaskPhysicsSelection(!DATA);

  //Add the centrality determination task
  gROOT->LoadMacro("$ALICE_ROOT/OADB/macros/AddTaskCentrality.C");
  AliCentralitySelectionTask* centSelTask = AddTaskCentrality();
  if (!DATA) centSelTask->SetMCInput();

  //add the PID response task
  gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
  AliAnalysisTaskPIDResponse* pidresponsetask = AddTaskPIDResponse(!DATA);

  //Add the TOF tender
  //gROOT->LoadMacro("$ALICE_ROOT/PWGCF/FLOW/macros/AddTaskTenderTOF.C");
  //AddTaskTenderTOF();

  // Setup analysis per centrality bin:
  gROOT->LoadMacro("AddTaskFlowCentralityPID.C");
  for (Int_t i=binfirst; i<binlast+1; i++)
  {
    Float_t lowCentralityBinEdge = centralityArray[i];
    Float_t highCentralityBinEdge = centralityArray[i+1];
    AddTaskFlowCentralityPID( lowCentralityBinEdge,
                              highCentralityBinEdge,
                              commonOutputFileName,
                              AliPID::kPion,
                              AliFlowTrackCuts::kTOFbeta,
                              -1,2,kTRUE );
    AddTaskFlowCentralityPID( lowCentralityBinEdge,
                              highCentralityBinEdge,
                              commonOutputFileName,
                              AliPID::kPion,
                              AliFlowTrackCuts::kTOFbeta,
                              1,2,kTRUE );
    AddTaskFlowCentralityPID( lowCentralityBinEdge,
                              highCentralityBinEdge,
                              commonOutputFileName,
                              AliPID::kKaon,
                              AliFlowTrackCuts::kTOFbeta,
                              -1,2,kTRUE );
    AddTaskFlowCentralityPID( lowCentralityBinEdge,
                              highCentralityBinEdge,
                              commonOutputFileName,
                              AliPID::kKaon,
                              AliFlowTrackCuts::kTOFbeta,
                              1,2,kTRUE );
    AddTaskFlowCentralityPID( lowCentralityBinEdge,
                              highCentralityBinEdge,
                              commonOutputFileName,
                              AliPID::kProton,
                              AliFlowTrackCuts::kTOFbeta,
                              -1,2,kTRUE );
    AddTaskFlowCentralityPID( lowCentralityBinEdge,
                              highCentralityBinEdge,
                              commonOutputFileName,
                              AliPID::kProton,
                              AliFlowTrackCuts::kTOFbeta,
                              1,2,kTRUE );
    AddTaskFlowCentralityPID( lowCentralityBinEdge,
                              highCentralityBinEdge,
                              commonOutputFileName,
                              AliPID::kPion,
                              AliFlowTrackCuts::kTOFbeta,
                              -1,3 );
    AddTaskFlowCentralityPID( lowCentralityBinEdge,
                              highCentralityBinEdge,
                              commonOutputFileName,
                              AliPID::kPion,
                              AliFlowTrackCuts::kTOFbeta,
                              1,3 );
    AddTaskFlowCentralityPID( lowCentralityBinEdge,
                              highCentralityBinEdge,
                              commonOutputFileName,
                              AliPID::kKaon,
                              AliFlowTrackCuts::kTOFbeta,
                              -1,3 );
    AddTaskFlowCentralityPID( lowCentralityBinEdge,
                              highCentralityBinEdge,
                              commonOutputFileName,
                              AliPID::kKaon,
                              AliFlowTrackCuts::kTOFbeta,
                              1,3 );
    AddTaskFlowCentralityPID( lowCentralityBinEdge,
                              highCentralityBinEdge,
                              commonOutputFileName,
                              AliPID::kProton,
                              AliFlowTrackCuts::kTOFbeta,
                              -1,3 );
    AddTaskFlowCentralityPID( lowCentralityBinEdge,
                              highCentralityBinEdge,
                              commonOutputFileName,
                              AliPID::kProton,
                              AliFlowTrackCuts::kTOFbeta,
                              1,3 );
  } // end of for (Int_t i=0; i<numberOfCentralityBins; i++)

  // Enable debug printouts:
  mgr->SetDebugLevel(2);
  // Run the analysis:
  if(!mgr->InitAnalysis()) return;
  mgr->PrintStatus();
  if(mode == mLocal)
  {
    mgr->StartAnalysis("local",chain);
  }
  else if(mode == mPROOF)
  {
    mgr->StartAnalysis("proof",dataDir,nEvents,offset);
  }
  else if(mode == mGrid)
  {
    mgr->StartAnalysis("grid");
  }

  // Print real and CPU time used for analysis:
  timer.Stop();
  timer.Print();

} // end of void runFlowTaskCentralityPIDTrain(...)

//===============================================================================================
/*
void CrossCheckUserSettings(Bool_t bData)
{
 // Check in this method if the user settings make sense.
 if(LYZ1SUM && LYZ2SUM) {cout<<" WARNING: You cannot run LYZ1 and LYZ2 at the same time! LYZ2 needs the output from LYZ1 !!!!"<<endl; exit(0); }
 if(LYZ1PROD && LYZ2PROD) {cout<<" WARNING: You cannot run LYZ1 and LYZ2 at the same time! LYZ2 needs the output from LYZ1 !!!!"<<endl; exit(0); }
 if(LYZ2SUM && LYZEP) {cout<<" WARNING: You cannot run LYZ2 and LYZEP at the same time! LYZEP needs the output from LYZ2 !!!!"<<endl; exit(0); }
 if(LYZ1SUM && LYZEP) {cout<<" WARNING: You cannot run LYZ1 and LYZEP at the same time! LYZEP needs the output from LYZ2 !!!!"<<endl; exit(0); }
} // end of void CrossCheckUserSettings()
*/
//===============================================================================================

void LoadLibraries(const anaModes mode, Bool_t useFlowParFiles )
{
  //--------------------------------------
  // Load the needed libraries most of them already loaded by aliroot
  //--------------------------------------

  gSystem->Load("libCore");
  gSystem->Load("libTree");
  gSystem->Load("libGeom");
  gSystem->Load("libVMC");
  gSystem->Load("libXMLIO");
  gSystem->Load("libPhysics");
  gSystem->Load("libXMLParser");
  gSystem->Load("libProof");
  gSystem->Load("libMinuit");

  if (mode==mLocal || mode==mGrid)
  {
    gSystem->Load("libSTEERBase");
    gSystem->Load("libCDB");
    gSystem->Load("libRAWDatabase");
    gSystem->Load("libRAWDatarec");
    gSystem->Load("libESD");
    gSystem->Load("libAOD");
    gSystem->Load("libSTEER");
    gSystem->Load("libANALYSIS");
    gSystem->Load("libANALYSISalice");
    gSystem->Load("libTPCbase");
    gSystem->Load("libTOFbase");
    gSystem->Load("libTOFrec");
    gSystem->Load("libTRDbase");
    gSystem->Load("libVZERObase");
    gSystem->Load("libVZEROrec");
    gSystem->Load("libT0base");
    gSystem->Load("libT0rec");
    gSystem->Load("libTender");
    gSystem->Load("libTenderSupplies");

    if (useFlowParFiles)
    {
      AliAnalysisAlien::SetupPar("PWGflowBase");
      AliAnalysisAlien::SetupPar("PWGflowTasks");
    }
    else
    {
      gSystem->Load("libPWGflowBase");
      gSystem->Load("libPWGflowTasks");
    }
  }
  else if (mode==mPROOF)
  {
    TList* list = new TList();
    list->Add(new TNamed("ALIROOT_MODE", "ALIROOT"));
    if (useFlowParFiles)
      list->Add(new TNamed("ALIROOT_EXTRA_LIBS", "ANALYSIS:ANALYSISalice:Tender:TenderSupplies"));
    else
      list->Add(new TNamed("ALIROOT_EXTRA_LIBS", "ANALYSIS:ANALYSISalice:Tender:TenderSupplies:PWGflowBase:PWGflowTasks"));

    //list->Add(new TNamed("ALIROOT_EXTRA_INCLUDES","PWG/FLOW/Base:PWG/FLOW/Tasks"));

    // Connect to proof
    printf("*** Connect to PROOF ***\n");
    gEnv->SetValue("XSec.GSI.DelegProxy","2");
    //TProof* proof = TProof::Open("alice-caf.cern.ch");
    TProof* proof = TProof::Open("skaf.saske.sk");

    // list the data available
    //gProof->ShowDataSets("/*/*");
    //gProof->ShowDataSets("/alice/sim/"); //for MC Data
    //gProof->ShowDataSets("/alice/data/"); //for REAL Data

    proof->ClearPackages();
    proof->EnablePackage("VO_ALICE@AliRoot::v4-21-14-AN",list);

    if (useFlowParFiles)
    {
      gProof->UploadPackage("PWGflowBase.par");
      gProof->UploadPackage("PWGflowTasks.par");
    }

    // Show enables Packages
    gProof->ShowEnabledPackages();
  }
} // end of void LoadLibraries(const anaModes mode)

//===============================================================================================

TChain* CreateAODChain(const char* aDataDir, Int_t aRuns, Int_t offset)
{
  // creates chain of files in a given directory or file containing a list.
  // In case of directory the structure is expected as:
  // <aDataDir>/<dir0>/AliAOD.root
  // <aDataDir>/<dir1>/AliAOD.root
  // ...

  if (!aDataDir)
    return 0;

  Long_t id, size, flags, modtime;
  if (gSystem->GetPathInfo(aDataDir, &id, &size, &flags, &modtime))
  {
    printf("%s not found.\n", aDataDir);
    return 0;
  }

  TChain* chain = new TChain("aodTree");
  TChain* chaingAlice = 0;

  if (flags & 2)
  {
    TString execDir(gSystem->pwd());
    TSystemDirectory* baseDir = new TSystemDirectory(".", aDataDir);
    TList* dirList            = baseDir->GetListOfFiles();
    Int_t nDirs               = dirList->GetEntries();
    gSystem->cd(execDir);

    Int_t count = 0;

    for (Int_t iDir=0; iDir<nDirs; ++iDir)
    {
      TSystemFile* presentDir = (TSystemFile*) dirList->At(iDir);
      if (!presentDir || !presentDir->IsDirectory() || strcmp(presentDir->GetName(), ".") == 0 || strcmp(presentDir->GetName(), "..") == 0)
        continue;

      if (offset > 0)
      {
        --offset;
        continue;
      }

      if (count++ == aRuns)
        break;

      TString presentDirName(aDataDir);
      presentDirName += "/";
      presentDirName += presentDir->GetName();
      chain->Add(presentDirName + "/AliAOD.root/aodTree");
      // cerr<<presentDirName<<endl;
    }

  }
  else
  {
    // Open the input stream
    ifstream in;
    in.open(aDataDir);

    Int_t count = 0;

    // Read the input list of files and add them to the chain
    TString aodfile;
    while(in.good())
    {
      in >> aodfile;
      if (!aodfile.Contains("root")) continue; // protection

      if (offset > 0)
      {
        --offset;
        continue;
      }

      if (count++ == aRuns)
        break;

      // add aod file
      chain->Add(aodfile);
    }

    in.close();
  }

  return chain;

} // end of TChain* CreateAODChain(const char* aDataDir, Int_t aRuns, Int_t offset)
Commit	Line	Data
f8892375	1	enum anaModes {mLocal,mPROOF,mGrid};
f8892375	2	Int_t binfirst = 0; //where do we start numbering bins
	3	Int_t binlast = 8; //where do we stop numbering bins
	4	const Int_t numberOfCentralityBins = 9;
	5	Float_t centralityArray[numberOfCentralityBins+1] = {0.,5.,10.,20.,30.,40.,50.,60.,70.,80.}; // in centrality percentile
	6	//Int_t centralityArray[numberOfCentralityBins+1] = {41,80,146,245,384,576,835,1203,1471,10000}; // in terms of TPC only reference multiplicity
	7
	8	TString commonOutputFileName = "outputCentrality"; // e.g.: result for centrality bin 0 will be in the file "outputCentrality0.root", etc
	9
08c6e3ef	10	void runFlowTaskCentralityPIDTrain( Int_t mode = mLocal,
08c6e3ef	11	Bool_t useFlowParFiles = kFALSE,
f8892375	12	Bool_t DATA = kTRUE,
f21bdf48	13	const Char_t* dataDir="fileList",
f8892375	14	Int_t nEvents = 1e4,
	15	Int_t offset=0 )
	16	{
	17	// Time:
	18	TStopwatch timer;
	19	timer.Start();
f8892375	20
	21	// Load needed libraries:
	22	LoadLibraries(mode,useFlowParFiles);
	23
	24	// Create analysis manager:
	25	AliAnalysisManager *mgr = new AliAnalysisManager("FlowAnalysisManager");
	26
	27	// Chains:
	28	if(mode == mLocal)
	29	{
08c6e3ef	30	gROOT->LoadMacro("$ALICE_ROOT/PWGUD/macros/CreateESDChain.C");
f8892375	31	TChain* chain = CreateESDChain(dataDir, nEvents, offset);
	32	//TChain* chain = CreateAODChain(dataDir, nEvents, offset);
	33	}
	34
	35	// Connect plug-in to the analysis manager:
	36	if(mode == mGrid)
	37	{
	38	gROOT->LoadMacro("CreateAlienHandler.C");
	39	AliAnalysisGrid *alienHandler = CreateAlienHandler(useFlowParFiles);
	40	if(!alienHandler) return;
	41	mgr->SetGridHandler(alienHandler);
	42	}
	43
	44	// Event handlers:
	45	AliVEventHandler* esdH = new AliESDInputHandler;
	46	mgr->SetInputEventHandler(esdH);
	47	if (!DATA)
	48	{
	49	AliMCEventHandler *mc = new AliMCEventHandler();
	50	mgr->SetMCtruthEventHandler(mc);
	51	}
	52
	53	// Task to check the offline trigger:
	54	gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
	55	AddTaskPhysicsSelection(!DATA);
	56
	57	//Add the centrality determination task
1c06a186	58	gROOT->LoadMacro("$ALICE_ROOT/OADB/macros/AddTaskCentrality.C");
f21bdf48	59	AliCentralitySelectionTask* centSelTask = AddTaskCentrality();
f21bdf48	60	if (!DATA) centSelTask->SetMCInput();
f8892375	61
2047680a	62	//add the PID response task
2047680a	63	gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
08c6e3ef	64	AliAnalysisTaskPIDResponse* pidresponsetask = AddTaskPIDResponse(!DATA);
2047680a	65
f8892375	66	//Add the TOF tender
08c6e3ef	67	//gROOT->LoadMacro("$ALICE_ROOT/PWGCF/FLOW/macros/AddTaskTenderTOF.C");
08c6e3ef	68	//AddTaskTenderTOF();
f8892375	69
	70	// Setup analysis per centrality bin:
	71	gROOT->LoadMacro("AddTaskFlowCentralityPID.C");
	72	for (Int_t i=binfirst; i<binlast+1; i++)
	73	{
	74	Float_t lowCentralityBinEdge = centralityArray[i];
	75	Float_t highCentralityBinEdge = centralityArray[i+1];
	76	AddTaskFlowCentralityPID( lowCentralityBinEdge,
	77	highCentralityBinEdge,
	78	commonOutputFileName,
	79	AliPID::kPion,
2047680a	80	AliFlowTrackCuts::kTOFbeta,
f8892375	81	-1,2,kTRUE );
	82	AddTaskFlowCentralityPID( lowCentralityBinEdge,
	83	highCentralityBinEdge,
	84	commonOutputFileName,
	85	AliPID::kPion,
2047680a	86	AliFlowTrackCuts::kTOFbeta,
f8892375	87	1,2,kTRUE );
	88	AddTaskFlowCentralityPID( lowCentralityBinEdge,
	89	highCentralityBinEdge,
	90	commonOutputFileName,
	91	AliPID::kKaon,
2047680a	92	AliFlowTrackCuts::kTOFbeta,
f8892375	93	-1,2,kTRUE );
	94	AddTaskFlowCentralityPID( lowCentralityBinEdge,
	95	highCentralityBinEdge,
	96	commonOutputFileName,
	97	AliPID::kKaon,
2047680a	98	AliFlowTrackCuts::kTOFbeta,
f8892375	99	1,2,kTRUE );
	100	AddTaskFlowCentralityPID( lowCentralityBinEdge,
	101	highCentralityBinEdge,
	102	commonOutputFileName,
	103	AliPID::kProton,
2047680a	104	AliFlowTrackCuts::kTOFbeta,
f8892375	105	-1,2,kTRUE );
	106	AddTaskFlowCentralityPID( lowCentralityBinEdge,
	107	highCentralityBinEdge,
	108	commonOutputFileName,
	109	AliPID::kProton,
2047680a	110	AliFlowTrackCuts::kTOFbeta,
f8892375	111	1,2,kTRUE );
	112	AddTaskFlowCentralityPID( lowCentralityBinEdge,
	113	highCentralityBinEdge,
	114	commonOutputFileName,
	115	AliPID::kPion,
2047680a	116	AliFlowTrackCuts::kTOFbeta,
f8892375	117	-1,3 );
	118	AddTaskFlowCentralityPID( lowCentralityBinEdge,
	119	highCentralityBinEdge,
	120	commonOutputFileName,
	121	AliPID::kPion,
2047680a	122	AliFlowTrackCuts::kTOFbeta,
f8892375	123	1,3 );
	124	AddTaskFlowCentralityPID( lowCentralityBinEdge,
	125	highCentralityBinEdge,
	126	commonOutputFileName,
	127	AliPID::kKaon,
2047680a	128	AliFlowTrackCuts::kTOFbeta,
f8892375	129	-1,3 );
	130	AddTaskFlowCentralityPID( lowCentralityBinEdge,
	131	highCentralityBinEdge,
	132	commonOutputFileName,
	133	AliPID::kKaon,
2047680a	134	AliFlowTrackCuts::kTOFbeta,
f8892375	135	1,3 );
	136	AddTaskFlowCentralityPID( lowCentralityBinEdge,
	137	highCentralityBinEdge,
	138	commonOutputFileName,
	139	AliPID::kProton,
2047680a	140	AliFlowTrackCuts::kTOFbeta,
f8892375	141	-1,3 );
	142	AddTaskFlowCentralityPID( lowCentralityBinEdge,
	143	highCentralityBinEdge,
	144	commonOutputFileName,
	145	AliPID::kProton,
2047680a	146	AliFlowTrackCuts::kTOFbeta,
f8892375	147	1,3 );
	148	} // end of for (Int_t i=0; i<numberOfCentralityBins; i++)
	149
	150	// Enable debug printouts:
	151	mgr->SetDebugLevel(2);
	152	// Run the analysis:
	153	if(!mgr->InitAnalysis()) return;
	154	mgr->PrintStatus();
	155	if(mode == mLocal)
	156	{
	157	mgr->StartAnalysis("local",chain);
	158	}
	159	else if(mode == mPROOF)
	160	{
	161	mgr->StartAnalysis("proof",dataDir,nEvents,offset);
	162	}
	163	else if(mode == mGrid)
	164	{
	165	mgr->StartAnalysis("grid");
	166	}
	167
	168	// Print real and CPU time used for analysis:
	169	timer.Stop();
	170	timer.Print();
	171
	172	} // end of void runFlowTaskCentralityPIDTrain(...)
	173
	174	//===============================================================================================
	175	/*
	176	void CrossCheckUserSettings(Bool_t bData)
	177	{
	178	// Check in this method if the user settings make sense.
	179	if(LYZ1SUM && LYZ2SUM) {cout<<" WARNING: You cannot run LYZ1 and LYZ2 at the same time! LYZ2 needs the output from LYZ1 !!!!"<<endl; exit(0); }
	180	if(LYZ1PROD && LYZ2PROD) {cout<<" WARNING: You cannot run LYZ1 and LYZ2 at the same time! LYZ2 needs the output from LYZ1 !!!!"<<endl; exit(0); }
	181	if(LYZ2SUM && LYZEP) {cout<<" WARNING: You cannot run LYZ2 and LYZEP at the same time! LYZEP needs the output from LYZ2 !!!!"<<endl; exit(0); }
	182	if(LYZ1SUM && LYZEP) {cout<<" WARNING: You cannot run LYZ1 and LYZEP at the same time! LYZEP needs the output from LYZ2 !!!!"<<endl; exit(0); }
	183	} // end of void CrossCheckUserSettings()
	184	*/
	185	//===============================================================================================
	186
	187	void LoadLibraries(const anaModes mode, Bool_t useFlowParFiles )
	188	{
	189	//--------------------------------------
	190	// Load the needed libraries most of them already loaded by aliroot
	191	//--------------------------------------
	192
	193	gSystem->Load("libCore");
	194	gSystem->Load("libTree");
	195	gSystem->Load("libGeom");
	196	gSystem->Load("libVMC");
	197	gSystem->Load("libXMLIO");
	198	gSystem->Load("libPhysics");
	199	gSystem->Load("libXMLParser");
	200	gSystem->Load("libProof");
	201	gSystem->Load("libMinuit");
	202
	203	if (mode==mLocal \|\| mode==mGrid)
	204	{
	205	gSystem->Load("libSTEERBase");
	206	gSystem->Load("libCDB");
	207	gSystem->Load("libRAWDatabase");
	208	gSystem->Load("libRAWDatarec");
	209	gSystem->Load("libESD");
	210	gSystem->Load("libAOD");
211	gSystem->Load("libSTEER");
212	gSystem->Load("libANALYSIS");
213	gSystem->Load("libANALYSISalice");
cd62a2a8	214	gSystem->Load("libTPCbase");
f8892375	215	gSystem->Load("libTOFbase");
	216	gSystem->Load("libTOFrec");
	217	gSystem->Load("libTRDbase");
	218	gSystem->Load("libVZERObase");
	219	gSystem->Load("libVZEROrec");
	220	gSystem->Load("libT0base");
	221	gSystem->Load("libT0rec");
230b7aff	222	gSystem->Load("libTender");
230b7aff	223	gSystem->Load("libTenderSupplies");
f8892375	224
	225	if (useFlowParFiles)
	226	{
2e311896	227	AliAnalysisAlien::SetupPar("PWGflowBase");
2e311896	228	AliAnalysisAlien::SetupPar("PWGflowTasks");
f8892375	229	}
	230	else
	231	{
2e311896	232	gSystem->Load("libPWGflowBase");
2e311896	233	gSystem->Load("libPWGflowTasks");
f8892375	234	}
	235	}
	236	else if (mode==mPROOF)
	237	{
	238	TList* list = new TList();
	239	list->Add(new TNamed("ALIROOT_MODE", "ALIROOT"));
	240	if (useFlowParFiles)
230b7aff	241	list->Add(new TNamed("ALIROOT_EXTRA_LIBS", "ANALYSIS:ANALYSISalice:Tender:TenderSupplies"));
f8892375	242	else
230b7aff	243	list->Add(new TNamed("ALIROOT_EXTRA_LIBS", "ANALYSIS:ANALYSISalice:Tender:TenderSupplies:PWGflowBase:PWGflowTasks"));
f8892375	244
2e311896	245	//list->Add(new TNamed("ALIROOT_EXTRA_INCLUDES","PWG/FLOW/Base:PWG/FLOW/Tasks"));
f8892375	246
	247	// Connect to proof
	248	printf("* Connect to PROOF *\n");
	249	gEnv->SetValue("XSec.GSI.DelegProxy","2");
	250	//TProof* proof = TProof::Open("alice-caf.cern.ch");
	251	TProof* proof = TProof::Open("skaf.saske.sk");
	252
	253	// list the data available
	254	//gProof->ShowDataSets("//");
	255	//gProof->ShowDataSets("/alice/sim/"); //for MC Data
	256	//gProof->ShowDataSets("/alice/data/"); //for REAL Data
	257
	258	proof->ClearPackages();
	259	proof->EnablePackage("VO_ALICE@AliRoot::v4-21-14-AN",list);
	260
	261	if (useFlowParFiles)
	262	{
2e311896	263	gProof->UploadPackage("PWGflowBase.par");
2e311896	264	gProof->UploadPackage("PWGflowTasks.par");
f8892375	265	}
	266
	267	// Show enables Packages
	268	gProof->ShowEnabledPackages();
	269	}
	270	} // end of void LoadLibraries(const anaModes mode)
	271
	272	//===============================================================================================
	273
	274	TChain* CreateAODChain(const char* aDataDir, Int_t aRuns, Int_t offset)
	275	{
	276	// creates chain of files in a given directory or file containing a list.
	277	// In case of directory the structure is expected as:
	278	// <aDataDir>/<dir0>/AliAOD.root
	279	// <aDataDir>/<dir1>/AliAOD.root
	280	// ...
	281
	282	if (!aDataDir)
	283	return 0;
	284
	285	Long_t id, size, flags, modtime;
	286	if (gSystem->GetPathInfo(aDataDir, &id, &size, &flags, &modtime))
	287	{
	288	printf("%s not found.\n", aDataDir);
	289	return 0;
	290	}
	291
	292	TChain* chain = new TChain("aodTree");
	293	TChain* chaingAlice = 0;
	294
	295	if (flags & 2)
	296	{
	297	TString execDir(gSystem->pwd());
	298	TSystemDirectory* baseDir = new TSystemDirectory(".", aDataDir);
	299	TList* dirList = baseDir->GetListOfFiles();
	300	Int_t nDirs = dirList->GetEntries();
	301	gSystem->cd(execDir);
	302
	303	Int_t count = 0;
	304
	305	for (Int_t iDir=0; iDir<nDirs; ++iDir)
	306	{
	307	TSystemFile* presentDir = (TSystemFile*) dirList->At(iDir);
	308	if (!presentDir \|\| !presentDir->IsDirectory() \|\| strcmp(presentDir->GetName(), ".") == 0 \|\| strcmp(presentDir->GetName(), "..") == 0)
	309	continue;
	310
	311	if (offset > 0)
	312	{
	313	--offset;
	314	continue;
	315	}
	316
	317	if (count++ == aRuns)
	318	break;
	319
	320	TString presentDirName(aDataDir);
	321	presentDirName += "/";
	322	presentDirName += presentDir->GetName();
	323	chain->Add(presentDirName + "/AliAOD.root/aodTree");
	324	// cerr<<presentDirName<<endl;
	325	}
	326
	327	}
	328	else
329	{
330	// Open the input stream
331	ifstream in;
332	in.open(aDataDir);
333
334	Int_t count = 0;
335
336	// Read the input list of files and add them to the chain
337	TString aodfile;
338	while(in.good())
339	{
340	in >> aodfile;
341	if (!aodfile.Contains("root")) continue; // protection
342
343	if (offset > 0)
344	{
345	--offset;
346	continue;
347	}
348
349	if (count++ == aRuns)
350	break;
351
352	// add aod file
353	chain->Add(aodfile);
354	}
355
356	in.close();
357	}
358
359	return chain;
360
361	} // end of TChain* CreateAODChain(const char* aDataDir, Int_t aRuns, Int_t offset)
362