macro to run standalone ESD filtering

[u/mrichter/AliRoot.git] / PWGPP / macros / runFilteringTask.C

void runFilteringTask( const char* esdList,   
        Float_t scalingTracks,
        Float_t scalingV0,
        const char* ocdb = "local:///cvmfs/alice.gsi.de/alice/data/2010/OCDB/" ,
        Int_t nFiles = 1000000,  
        Int_t firstFile=0, 
        Int_t nEvents=30000000, 
        Int_t firstEvent =0,
        const char* esdFileName="AliESDs.root",
        Bool_t mc=kFALSE)
{
    TStopwatch timer;
    timer.Start();

    printf("\n\n\n");
    printf("scalingTracks=%d\n",scalingTracks);
    printf("scalingV0=%d\n",scalingV0);
    printf("nFiles=%d\n",nFiles);

    gSystem->SetIncludePath("-I$ROOTSYS/include -I$ALICE_ROOT/include -I$ALICE_ROOT/ITS -I$ALICE_ROOT -I$ALICE_ROOT/TRD");

    gSystem->Load("libANALYSIS");
    gSystem->Load("libANALYSISalice");
    gSystem->Load("libTENDER");
    gSystem->Load("libCORRFW");
    gSystem->Load("libPWGUDbase");
    gSystem->Load("libTPCcalib");
    gSystem->Load("libPWGPP");
    gSystem->Load("libPWGLFspectra");

    //____________________________________________//
    // Make the analysis manager
    AliAnalysisManager *mgr = new AliAnalysisManager("TestManager");
    mgr->SetDebugLevel(0);

    AliESDInputHandler* esdH = new AliESDInputHandler();
    //esdH->SetReadFriends(1);
    //esdH->SetReadFriends(1);
    mgr->SetInputEventHandler(esdH);  

    // Enable MC event handler
    AliMCEventHandler* handlerMC = new AliMCEventHandler;
    //handler->SetReadTR(kFALSE);
    if (mc) mgr->SetMCtruthEventHandler(handlerMC);

    gROOT->LoadMacro("$ALICE_ROOT/PWGPP/PilotTrain/AddTaskCDBconnect.C");
    AddTaskCDBconnect(ocdb);

    if (gSystem->AccessPathName("localOCDBaccessConfig.C", kFileExists)==0) {
      gROOT->LoadMacro("localOCDBaccessConfig.C");
      localOCDBaccessConfig();
    }

    // Create input chain
    //gROOT->LoadMacro("$ALICE_ROOT/PWGUD/macros/CreateESDChain.C");
    //TChain* chain = CreateESDChain(esdList, nFiles,firstFile);
    AliXRDPROOFtoolkit toolkit;
    AliXRDPROOFtoolkit::FilterList(esdList,Form("%s esdTree",esdFileName),0);
    TChain * chain = toolkit.MakeChain(Form("%s.Good",esdList),"esdTree","asd",nFiles,firstFile);



    if(!chain) {
        printf("ERROR: chain cannot be created\n");
        return;
    }
    chain->Lookup();

    //
    // Wagons to run 
    //


    gROOT->LoadMacro("$ALICE_ROOT/PWGPP/macros/AddTaskFilteredTree.C");
    AliAnalysisTaskFilteredTree* task = (AliAnalysisTaskFilteredTree*)AddTaskFilteredTree("Filtered.root");
    task->SetLowPtTrackDownscaligF(scalingTracks);
    task->SetLowPtV0DownscaligF(scalingV0);

    // Init
    if (!mgr->InitAnalysis()) 
        mgr->PrintStatus();

    //
    // Run on dataset
    mgr->StartAnalysis("local",chain,nEvents, firstEvent);

    timer.Stop();
    timer.Print();

    delete mgr;
}