major update in LRC code (Igor Altsybeev <Igor.Altsybeev@cern.ch>)

[u/mrichter/AliRoot.git] / PWGCF / EBYE / macros / AddTaskLRCtrain.C

AliAnalysisTaskLRC *AddTaskLRCtrain(
          Bool_t RunKine = kFALSE
        , TString strPrefixTaskName = "testTask"
        , TString strRunMode = "default"
        , int nEtaWindows = 8
        , int nPhiWindows = 8
        , double etaWinWidth = 0.2
        , double etaWindowStep = 0.1
        , double phiWinWidth = TMath::Pi()/4
        , int nEventPhiRotations = 8
        , double ptMin = 0.3
        , double ptMax = 1.5
        , int multMin = 0
        , int multMax = 1000
        , double vertexZmax = 7.0
        )
{
    gROOT->LoadMacro("configLRCAnalysis.C");
    //gROOT->LoadMacro("AliAnalysisTaskIA.cxx+g");

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

    // B. Check the analysis type using the event handlers connected to the analysis
    //    manager. The availability of MC handler cann also be checked here.
    //==============================================================================
    if (!mgr->GetInputEventHandler()) {
        Error("AddTaskLRC", "This task requires an input event handler");
        return NULL;
    }
    TString type = mgr->GetInputEventHandler()->GetDataType();
    cout << " # TaskLRC - input :" << type << "\n";

    // ########## Task LRC
    AliAnalysisTaskLRC* taskLRC = 0x0;
    taskLRC = createLRCtaskSkeleton("Task_LRC_" + strPrefixTaskName /*+ "_MB_Global2"*/,RunKine);
    //addAliLRCProcessors(taskLRC,windowsConfigId, nPhiSectors,gapAsPartOfPi);//, whichParticleToFill, whichPidCondition);

    tuneEtaPhiWindows( taskLRC
                       , nEtaWindows
                       , nPhiWindows
                       , etaWinWidth
                       , etaWindowStep
                       , 0
                       , phiWinWidth
                       );

    taskLRC->SetNumberOfPhiSectors( nEventPhiRotations );
    taskLRC->SetAnalysisLevel( type );

    taskLRC->SetAODtrackCutBit( 1 );//128 );//128 );

    taskLRC->SetVtxDiamond(0.4,0.4,vertexZmax);
    taskLRC->SetMaxPtLimit(ptMax);
    taskLRC->SetMinPtLimit(ptMin);
    //taskLRC->SetMinNumberOfSPDtracklets( systConfig->minNtracklets );
    taskLRC->SetNchCuts( multMin, multMax ); // !!! for mult studies
    taskLRC->SetTrackCuts(createAliLRCcuts( "StandardTPCOnlyTrackCuts" )); // "Global2" ));
    setHistMultRange( taskLRC,  0, 0, 15 );
    setHistPtRange( taskLRC, ptMin, ptMax, 0.01, 5 );

//    taskLRC->SetNumberOfPhiSectorsByHand(8);

    taskLRC->SelectCollisionCandidates(AliVEvent::kMB); // if physics selection performed in UserExec(), this line should be commented
    mgr->AddTask(taskLRC);

    configureLRCtaskOutput(taskLRC, /*strOutputRootFolder,*/ strPrefixTaskName, strRunMode);

    return taskLRC;
}