Reader: Add option to remove or not event with primary vertex not reconstructed

[u/mrichter/AliRoot.git] / PWG4 / macros / electrons / ConfigJetAnalysisFastJet.C

//------------------------------------\r
// Configuration macro:\r
//\r
// Configure JETAN FastJet analysis.\r
//\r
// Modified by: K. Read\r
//\r
//------------------------------------\r
\r
AliJetFinder*  ConfigJetAnalysis()\r
{\r
    //\r
    // Configuration goes here\r
    // \r
    printf("========================== \n");\r
    printf("ConfigJetAnalysisFastJet() \n");\r
    printf("========================== \n");\r
\r
    Bool_t kInputIsESD = kTRUE;     //uncomment for input ESD\r
  //Bool_t kInputIsESD = kFALSE;    //uncomment for input AODs\r
    Bool_t kFollowsFilter = kTRUE;  //uncomment if follows ESD filter task\r
  //Bool_t kFollowsFilter = kFALSE; //uncomment if no ESD filter task\r
\r
    //Alternatively, select input via anaInputData environment variable.\r
    if (gSystem->Getenv("anaInputData")){\r
      TString kInputData = gSystem->Getenv("anaInputData");\r
      if( kInputData == "AOD" ){\r
        kInputIsESD = kFALSE;\r
        kFollowsFilter = kFALSE;\r
      }\r
    }\r
\r
    // Define the grids\r
    AliJetGrid *grid = new AliJetGrid(419,119,0.,2*TMath::Pi(),-0.9,0.9); \r
    grid->SetGridType(1);\r
    grid->InitParams(80.*TMath::Pi()/180,190.*TMath::Pi()/180,-0.7,0.7); \r
    grid->SetMatrixIndexes();\r
    grid->SetIndexIJ();\r
    AliJetGrid *grid2 = new AliJetGrid(131,95,80.*TMath::Pi()/180.,190.*TMath::Pi()/180.,-0.7,0.7); \r
    grid2->SetGridType(0);\r
    grid2->SetMatrixIndexes();\r
    grid2->SetIndexIJ();\r
\r
    // Define reader header\r
    if(kInputIsESD && !kFollowsFilter) AliJetESDReaderHeader *jrh = new AliJetESDReaderHeader();\r
    else            AliJetAODReaderHeader *jrh = new AliJetAODReaderHeader();\r
    jrh->SetComment("Testing");\r
    if(kInputIsESD && !kFollowsFilter) jrh->SetReadSignalOnly(kFALSE);\r
\r
    // Detector options: 0 = Charged particles only (MomentumArray)\r
    //                   1 = Charged particles only (UnitArray)\r
    //                   2 = Neutral cells only (UnitArray)\r
    //                   3 = Charged particles + neutral cells (UnitArray)\r
    jrh->SetDetector(3);\r
    //jrh->SetDebug(-1);\r
    //jrh->SetFiducialEta(-0.7,0.7);\r
    //jrh->SetFiducialPhi(80.*TMath::Pi()/180,190.*TMath::Pi()/180);\r
    jrh->SetPtCut(0.1);\r
    jrh->SetFiducialEta(-0.9,0.9);    //fiducial range used by AliJetFillUnitArrayTracks\r
    jrh->SetFiducialPhi(0,2*TMath::Pi()); //fiducial range used by AliJetFillUnitArrayTracks\r
    \r
    // Define reader and set its header\r
    if(kInputIsESD && !kFollowsFilter) AliJetESDReader *er = new AliJetESDReader();\r
    else            AliJetAODReader *er = new AliJetAODReader();\r
    er->SetReaderHeader(jrh);\r
    er->SetTPCGrid(grid);\r
    er->SetEMCalGrid(grid2);\r
    er->SetApplyMIPCorrection(kFALSE);\r
    //hadronic correction\r
    er->SetApplyFractionHadronicCorrection(kTRUE);\r
    er->SetFractionHadronicCorrection(1.0);\r
\r
    // Define jet header\r
    AliFastJetHeaderV1 *jh=new AliFastJetHeaderV1();\r
    Double_t R=0.4;\r
    Double_t Rbkg=0.2;\r
\r
    // AliFastJetHeaderV1 *jh=new AliFastJetHeaderV1();\r
    jh->SetComment("Fast jet code with default parameters");\r
    //jh->SetDebug(-1);\r
    //jh->SetBGMode(1); //Do BG Subtraction\r
    jh->SetBGMode(0); //No BG Subtraction.  Store AOD track refs.\r
    jh->SetRparam(R); // setup parameters\r
    jh->SetRparamBkg(Rbkg); // setup parameters\r
    jh->SetPtMin(0.2);\r
    //jh->SetGhostEtaMax(0.9);\r
    jh->SetGhostArea(0.01);\r
    jh->SetGhostEtaMax(0.7);//used to set the rap_min and rap_max, that are then used by FJ\r
    jh->SetPhiRange(80.*TMath::Pi()/180+R,190.*TMath::Pi()/180-R);//used in AliFastJetFinder for the range\r
\r
    // Define jet finder. Set its header and reader\r
    jetFinder = new AliFastJetFinder();\r
    jetFinder->SetJetHeader(jh);\r
    jetFinder->SetJetReader(er);\r
    //jetFinder->SetPlotMode(kTRUE);\r
\r
    printf("============================== \n");\r
    printf("END ConfigJetAnalysisFastJet() \n");\r
    printf("============================== \n");\r
 \r
    return jetFinder;\r
}\r