AliEmcalESDTrackCutsGenerator.cxx

Go to the documentation of this file.

/************************************************************************************
 * Copyright (C) 2017, Copyright Holders of the ALICE Collaboration                 *
 * All rights reserved.                                                             *
 *                                                                                  *
 * Redistribution and use in source and binary forms, with or without               *
 * modification, are permitted provided that the following conditions are met:      *
 *     * Redistributions of source code must retain the above copyright             *
 *       notice, this list of conditions and the following disclaimer.              *
 *     * Redistributions in binary form must reproduce the above copyright          *
 *       notice, this list of conditions and the following disclaimer in the        *
 *       documentation and/or other materials provided with the distribution.       *
 *     * Neither the name of the <organization> nor the                             *
 *       names of its contributors may be used to endorse or promote products       *
 *       derived from this software without specific prior written permission.      *
 *                                                                                  *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND  *
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED    *
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE           *
 * DISCLAIMED. IN NO EVENT SHALL ALICE COLLABORATION BE LIABLE FOR ANY              *
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES       *
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;     *
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND      *
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT       *
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS    *
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.                     *
 ************************************************************************************/

#include <TString.h>
#include <TFormula.h>

#include "AliEmcalESDHybridTrackCuts.h"
#include "AliEmcalTrackSelection.h"
#include "AliEmcalESDTrackCutsGenerator.h"

#include "AliESDtrackCuts.h"

#include "TError.h"

using namespace PWG::EMCAL;

const Int_t AliEmcalESDTrackCutsGenerator::fgkAddCutFactor = 10000;

AliESDtrackCuts* AliEmcalESDTrackCutsGenerator::CreateTrackCutsPWGJE(Int_t cutMode)
{
  //Get standard cuts: last 4 digits of cutMode
  Int_t stdCutMode = cutMode % fgkAddCutFactor;

  //Get additional cut mode: first 4 digits of cutMode
  Int_t addCutMode = (int)((float)cutMode/(float)fgkAddCutFactor);

  return CreateTrackCutsPWGJE(stdCutMode, addCutMode);
}

AliESDtrackCuts* AliEmcalESDTrackCutsGenerator::CreateTrackCutsPWGJE(Int_t stdCutMode, Int_t addCutMode)
{
  AliESDtrackCuts* trackCuts = 0;

  TString tag;

  tag = SetStandardCuts(trackCuts, stdCutMode);
  tag += SetAdditionalCuts(trackCuts, addCutMode);

  ::Info("AliEmcalESDTrackCutsGenerator::CreateTrackCutsPWGJE", "Created track cuts for: %s", tag.Data());

  return trackCuts;
}

AliESDtrackCuts* AliEmcalESDTrackCutsGenerator::CreateTrackCutsPWGJE(Int_t stdCutMode, Int_t addCutMode1, Int_t addCutMode2)
{
  AliESDtrackCuts* trackCuts = 0;

  TString tag;

  tag = SetStandardCuts(trackCuts, stdCutMode);
  tag += SetAdditionalCuts(trackCuts, addCutMode1);
  tag += SetAdditionalCuts(trackCuts, addCutMode2);

  ::Info("AliEmcalESDTrackCutsGenerator::CreateTrackCutsPWGJE", "Created track cuts for: %s", tag.Data());

  return trackCuts;
}

TString AliEmcalESDTrackCutsGenerator::SetStandardCuts(AliESDtrackCuts*& trackCuts, Int_t stdCutMode)
{
  TString tag;

  if (trackCuts) {
    delete trackCuts;
    trackCuts = 0;
  }

  switch (stdCutMode) {
  case kRAA2011:
  {
    trackCuts = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(kTRUE,1);
    trackCuts->SetMinNCrossedRowsTPC(120);
    trackCuts->SetMinRatioCrossedRowsOverFindableClustersTPC(0.8);
    trackCuts->SetMaxChi2PerClusterITS(36);
    trackCuts->SetMaxFractionSharedTPCClusters(0.4);
    trackCuts->SetMaxChi2TPCConstrainedGlobal(36);

    trackCuts->SetEtaRange(-0.9,0.9);
    trackCuts->SetPtRange(0.15, 1e10);

    tag = "Global track RAA analysis QM2011 + Chi2ITS<36";

    break;
  }

  case kGlobalTracksNCls90NoSPD:
  {
    trackCuts = new AliESDtrackCuts("AliESDtrackCuts");
    // TPC
    trackCuts->SetMinNClustersTPC(90);
    trackCuts->SetMaxChi2PerClusterTPC(4);
    trackCuts->SetRequireTPCStandAlone(kTRUE); //cut on NClustersTPC and chi2TPC Iter1
    trackCuts->SetAcceptKinkDaughters(kFALSE);
    trackCuts->SetRequireTPCRefit(kTRUE);
    trackCuts->SetMaxFractionSharedTPCClusters(0.4);
    // ITS
    trackCuts->SetRequireITSRefit(kTRUE);
    //accept secondaries
    trackCuts->SetMaxDCAToVertexXY(2.4);
    trackCuts->SetMaxDCAToVertexZ(3.2);
    trackCuts->SetDCAToVertex2D(kTRUE);
    //reject fakes
    trackCuts->SetMaxChi2PerClusterITS(36);

    trackCuts->SetRequireSigmaToVertex(kFALSE);

    trackCuts->SetEtaRange(-0.9,0.9);
    trackCuts->SetPtRange(0.15, 100.);

    tag = "Global tracks jet analysis with ITSrefit and NclsIter1=90, noSPD requirement";

    break;
  }

  case kGlobalTracksNCls80NoSPD:
  {
    trackCuts = new AliESDtrackCuts("AliESDtrackCuts");
    // TPC
    trackCuts->SetMinNClustersTPC(80);
    trackCuts->SetMaxChi2PerClusterTPC(4);
    trackCuts->SetAcceptKinkDaughters(kFALSE);
    trackCuts->SetRequireTPCRefit(kTRUE);
    trackCuts->SetMaxFractionSharedTPCClusters(0.4);
    // ITS
    trackCuts->SetRequireITSRefit(kTRUE);
    //accept secondaries
    trackCuts->SetMaxDCAToVertexXY(2.4);
    trackCuts->SetMaxDCAToVertexZ(3.2);
    trackCuts->SetDCAToVertex2D(kTRUE);
    //reject fakes
    trackCuts->SetMaxChi2PerClusterITS(36);

    trackCuts->SetRequireSigmaToVertex(kFALSE);

    trackCuts->SetEtaRange(-0.9,0.9);
    trackCuts->SetPtRange(0.15, 100.);

    tag = "Global tracks jet analysis with ITSrefit and Ncls=80, noSPD requirement";

    break;
  }

  case kGlobalTracks2010NCrossRows120:
  {
    trackCuts = new AliESDtrackCuts("AliESDtrackCuts");
    // tight global tracks
    trackCuts = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(kFALSE,1);
    trackCuts->SetMinNClustersTPC(0);
    trackCuts->SetMinNCrossedRowsTPC(120);
    trackCuts->SetMinRatioCrossedRowsOverFindableClustersTPC(0.1);// essentially switches it off
    trackCuts->SetMaxDCAToVertexXY(2.4);
    trackCuts->SetMaxDCAToVertexZ(3.2);
    trackCuts->SetDCAToVertex2D(kTRUE);
    trackCuts->SetMaxChi2PerClusterITS(36);
    trackCuts->SetMaxFractionSharedTPCClusters(0.4);

    tag = "Global tracks ITSTPC2010 + NCrossedRows + loose ITS";

    break;
  }

  case kGlobalTracksNCls70NoSPD:
  {
    trackCuts = new AliESDtrackCuts("AliESDtrackCuts");
    // TPC
    trackCuts->SetMinNClustersTPC(70);
    trackCuts->SetMaxChi2PerClusterTPC(4);
    trackCuts->SetRequireTPCStandAlone(kTRUE); //cut on NClustersTPC and chi2TPC Iter1
    trackCuts->SetAcceptKinkDaughters(kFALSE);
    trackCuts->SetRequireTPCRefit(kTRUE);
    trackCuts->SetMaxFractionSharedTPCClusters(0.4);
    // ITS
    trackCuts->SetRequireITSRefit(kTRUE);
    //accept secondaries
    trackCuts->SetMaxDCAToVertexXY(2.4);
    trackCuts->SetMaxDCAToVertexZ(3.2);
    trackCuts->SetDCAToVertex2D(kTRUE);
    //reject fakes
    trackCuts->SetMaxChi2PerClusterITS(36);

    trackCuts->SetRequireSigmaToVertex(kFALSE);

    trackCuts->SetEtaRange(-0.9,0.9);
    trackCuts->SetPtRange(0.15, 100.);

    tag = "Global tracks jet analysis with ITSrefit and NclsIter1=70, noSPD requirement";

    break;
  }

  case kGlobalTracksNCls70NoSPDNoPtCut:
  {
    trackCuts = new AliESDtrackCuts("AliESDtrackCuts");
    // TPC
    trackCuts->SetMinNClustersTPC(70);
    trackCuts->SetMaxChi2PerClusterTPC(4);
    trackCuts->SetRequireTPCStandAlone(kTRUE); //cut on NClustersTPC and chi2TPC Iter1
    trackCuts->SetAcceptKinkDaughters(kFALSE);
    trackCuts->SetRequireTPCRefit(kTRUE);
    trackCuts->SetMaxFractionSharedTPCClusters(0.4);
    // ITS
    trackCuts->SetRequireITSRefit(kTRUE);
    //accept secondaries
    trackCuts->SetMaxDCAToVertexXY(2.4);
    trackCuts->SetMaxDCAToVertexZ(3.2);
    trackCuts->SetDCAToVertex2D(kTRUE);
    //reject fakes
    trackCuts->SetMaxChi2PerClusterITS(36);

    trackCuts->SetRequireSigmaToVertex(kFALSE);

    trackCuts->SetEtaRange(-0.9,0.9);
    trackCuts->SetPtRange(0.15, 1E+15);

    tag = "Global tracks jet analysis with ITSrefit and NclsIter1=70, noSPD requirement, no upper pt cut";

    break;
  }

  case kGlobalTracksNClsPtDepNoSPDNoPtCut:
  {
    trackCuts = new AliESDtrackCuts("AliESDtrackCuts");
    // TPC
    TFormula *f1NClustersTPCLinearPtDep = new TFormula("f1NClustersTPCLinearPtDep","70.+30./20.*x");
    trackCuts->SetMinNClustersTPCPtDep(f1NClustersTPCLinearPtDep,20.);
    trackCuts->SetMinNClustersTPC(70);
    trackCuts->SetMaxChi2PerClusterTPC(4);
    trackCuts->SetRequireTPCStandAlone(kTRUE); //cut on NClustersTPC and chi2TPC Iter1
    trackCuts->SetAcceptKinkDaughters(kFALSE);
    trackCuts->SetRequireTPCRefit(kTRUE);
    trackCuts->SetMaxFractionSharedTPCClusters(0.4);
    // ITS
    trackCuts->SetRequireITSRefit(kTRUE);
    //accept secondaries
    trackCuts->SetMaxDCAToVertexXY(2.4);
    trackCuts->SetMaxDCAToVertexZ(3.2);
    trackCuts->SetDCAToVertex2D(kTRUE);
    //reject fakes
    trackCuts->SetMaxChi2PerClusterITS(36);
    trackCuts->SetMaxChi2TPCConstrainedGlobal(36);

    trackCuts->SetRequireSigmaToVertex(kFALSE);

    trackCuts->SetEtaRange(-0.9,0.9);
    trackCuts->SetPtRange(0.15, 1E+15);

    tag = "Global tracks jet analysis with ITSrefit and NclsIter1=PtDep, noSPD requirement, no upper pt cut, golden chi2";

    break;
  }

  case kGlobalTracks2011:
  {
    trackCuts = AliESDtrackCuts::GetStandardITSTPCTrackCuts2011(kFALSE,1);
    //accept secondaries
    trackCuts->SetMaxDCAToVertexXY(2.4);
    trackCuts->SetMaxDCAToVertexZ(3.2);
    trackCuts->SetDCAToVertex2D(kTRUE);

    trackCuts->SetMaxChi2TPCConstrainedGlobal(36);

    trackCuts->SetEtaRange(-0.9,0.9);
    trackCuts->SetPtRange(0.15, 1E+15);

    tag = "Global tracks with AliESDtrackCuts::GetStandardITSTPCTrackCuts2011(kFALSE)";

    break;
  }

  case kGlobalTracks2011NoSPD:
  {
    trackCuts = AliESDtrackCuts::GetStandardITSTPCTrackCuts2011(kFALSE,1);
    //accept secondaries
    trackCuts->SetMaxDCAToVertexXY(2.4);
    trackCuts->SetMaxDCAToVertexZ(3.2);
    trackCuts->SetDCAToVertex2D(kTRUE);

    trackCuts->SetMaxChi2TPCConstrainedGlobal(36);
    trackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kNone);

    trackCuts->SetMaxFractionSharedTPCClusters(0.4);

    tag = "Global tracks 2011 with AliESDtrackCuts::GetStandardITSTPCTrackCuts2011(kFALSE) and no SPD requirement";

    break;
  }

  case kGlobalTracksNCls90NoITS:
  {
    trackCuts = new AliESDtrackCuts("AliESDtrackCuts");
    // TPC
    trackCuts->SetMinNClustersTPC(90);
    trackCuts->SetMaxChi2PerClusterTPC(4);
    trackCuts->SetRequireTPCStandAlone(kTRUE); //cut on NClustersTPC and chi2TPC Iter1
    trackCuts->SetAcceptKinkDaughters(kFALSE);
    trackCuts->SetRequireTPCRefit(kTRUE);
    trackCuts->SetMaxFractionSharedTPCClusters(0.4);
    //accept secondaries
    trackCuts->SetMaxDCAToVertexXY(2.4);
    trackCuts->SetMaxDCAToVertexZ(3.2);
    trackCuts->SetDCAToVertex2D(kTRUE);

    trackCuts->SetRequireSigmaToVertex(kFALSE);

    trackCuts->SetEtaRange(-0.9,0.9);
    trackCuts->SetPtRange(0.15, 100.);

    tag = "Global tracks jet analysis, loose cuts, NClsIter1=90, no ITS requirements";

    break;
  }

  case kTPCOnlyTracksNCls70:
  {
    trackCuts = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts();
    // trackCuts->SetRequireTPCRefit(kTRUE);
    trackCuts->SetMinNClustersTPC(70);

    trackCuts->SetEtaRange(-0.9,0.9);
    trackCuts->SetPtRange(0.15, 100.);


    tag = "TPConly track cuts, loose cuts, NCls=70, no ITS requirements";

    break;
  }

  case kTPCOnlyTracksNCrossRows120:
  {
    trackCuts = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts();
    trackCuts->SetMinNClustersTPC(0);
    trackCuts->SetMinNCrossedRowsTPC(120);
    trackCuts->SetMinRatioCrossedRowsOverFindableClustersTPC(0.1);// essentially switches it off

    trackCuts->SetEtaRange(-0.9,0.9);
    trackCuts->SetPtRange(0.15, 100.);

    tag = "TPConly track cuts, loose cuts, NCrossRows=120, no ITS requirements";

    break;
  }

  default:
  {
    Printf("AliEmcalESDTrackCutsGenerator: standard cuts not recognized.");
    break;
  }
  }

  return tag;
}

TString AliEmcalESDTrackCutsGenerator::SetAdditionalCuts(AliESDtrackCuts*& trackCuts, Int_t addCutMode)
{
  TString tag;

  switch (addCutMode) {
  case kSPDAny:
  {
    trackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kAny);

    tag += " + additonal: SPD any requirement";

    break;
  }

  case kSPDNone:
  {
    trackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kNone);

    tag += " + additional: w/o hits in SPD";

    break;
  }

  case kNoITSChi2:
  {
    trackCuts->SetMaxChi2PerClusterITS(1E10);

    tag += " + additional: maxITSChi2=1e10";

    break;
  }

  case kNoMinTPCCls:
  {
    trackCuts->SetMinNClustersTPC(0);
    trackCuts->SetMinNCrossedRowsTPC(0);
    trackCuts->SetMinRatioCrossedRowsOverFindableClustersTPC(0.);

    tag += " + additional: minClusters=0 minCrossedRows=0 minCrossedRowsOverFindable=0";

    break;
  }

  case kNoITSRefit:
  {
    trackCuts->SetRequireITSRefit(kFALSE);

    tag += " + additional: ITSrefit=kFALSE";

    break;
  }

  case kSPDOff:
  {
    trackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kOff);

    tag += " + additional: no SPD requirement (kOff)";

    break;
  }
  }

  return tag;
}

AliEmcalESDTrackCutsGenerator::EDataSet_t AliEmcalESDTrackCutsGenerator::SteerDataSetFromString(TString period)
{
  EDataSet_t dataSet = kUnknown;

  TString strPeriod(period);
  strPeriod.ToLower();

  if (strPeriod == "lhc10h") {
    dataSet = kLHC10h;
  } else if (strPeriod == "lhc11a" || strPeriod == "lhc12a15a") {
    dataSet = kLHC11a;
  } else if (strPeriod == "lhc10b" || strPeriod == "lhc10c" ||
      strPeriod == "lhc10d" || strPeriod == "lhc10e") {
    dataSet = kLHC10bcde;
  } else if (strPeriod == "lhc11a1a" ||  strPeriod == "lhc11a1b" ||
      strPeriod == "lhc11a1c" ||  strPeriod == "lhc11a1d" ||
      strPeriod == "lhc11a1e" ||  strPeriod == "lhc11a1f" ||
      strPeriod == "lhc11a1g" ||  strPeriod == "lhc11a1h" ||
      strPeriod == "lhc11a1i" ||  strPeriod == "lhc11a1j") {
    dataSet = kLHC11a;
  } else if (strPeriod == "lhc11c") {
    dataSet = kLHC11c;
  } else if (strPeriod == "lhc11d") {
    dataSet = kLHC11d;
  } else if (strPeriod == "lhc11h" || strPeriod == "lhc12a15e") {
    dataSet = kLHC11h;
  } else if (strPeriod == "lhc12g") {
    dataSet = kLHC11h;
  } else if (strPeriod == "lhc12") {
    dataSet = kLHC11h;
  } else if (strPeriod == "lhc13b") {
    dataSet = kLHC11h;
  } else if (strPeriod == "lhc13c") {
    dataSet = kLHC11h;
  } else if (strPeriod == "lhc13d") {
    dataSet = kLHC11h;
  } else if (strPeriod == "lhc13e") {
    dataSet = kLHC11h;
  } else if (strPeriod == "lhc13f") {
    dataSet = kLHC11h;
  } else if (strPeriod == "lhc13g") {
    dataSet = kLHC11h;
  } else if (strPeriod == "lhc16q") {
    dataSet = kLHC11h;
  } else if (strPeriod == "lhc16r") {
    dataSet = kLHC11h;
  } else if (strPeriod == "lhc16s") {
    dataSet = kLHC11h;
  } else if (strPeriod == "lhc16t") {
    dataSet = kLHC11h;
  } else if (strPeriod == "lhc17o_trd") {
    dataSet = kLHC17o_TRD;
  } else if (strPeriod == "lhc12a15f") {
    dataSet = kLHC11h;
  } else if (strPeriod == "lhc13b4") {
    dataSet = kLHC11h;
  } else if (strPeriod == "lhc12a15g") {
    dataSet = kLHC11d;
  } else if (strPeriod == "lhc12f2a") {
    dataSet = kLHC11d;
  } else if (strPeriod.BeginsWith("lhc12a17")) {
    dataSet = kLHC11h;
  } else if (strPeriod == "lhc14a1") {
    dataSet = kLHC11h;
  } else if (strPeriod.BeginsWith("lhc15g6")) {
    dataSet = kLHC10bcde;
  } else if (strPeriod.BeginsWith("lhc15i2")) {
    dataSet = kLHC10bcde;
  } else if (strPeriod.BeginsWith("lhc14j4")) {
    dataSet = kLHC10bcde;
  } else {
    ::Error("AliEmcalESDTrackCutsGenerator::SteerDataSetFromString", "Dataset %s not recognized!", period.Data());
  }

  return dataSet;
}

void AliEmcalESDTrackCutsGenerator::AddHybridTrackCuts(AliEmcalTrackSelection* trkSel, EDataSet_t period)
{
  switch (period) {
  case kLHC11c:
  case kLHC11d:
  case kLHC11h:
  {
    auto hybridcuts2011 = new AliEmcalESDHybridTrackCuts("hybridcuts2011NoSPD", AliEmcalESDHybridTrackCuts::kDef2011);
    hybridcuts2011->SetUseNoITSrefitTracks(kFALSE);
    trkSel->AddTrackCuts(hybridcuts2011);
    break;
  }
  case kLHC10h:
  case kLHC11a:
  case kLHC10bcde:
  {
    /* hybrid track cuts*/
    auto hybridcuts2010 = new AliEmcalESDHybridTrackCuts("hybridcuts2010wSPD", AliEmcalESDHybridTrackCuts::kDef2010);
    hybridcuts2010->SetUseNoITSrefitTracks(kTRUE);
    trkSel->AddTrackCuts(hybridcuts2010);
    break;
  }
  case kLHC17o_TRD:
  {
    auto hybridcuts2018TRD = new AliEmcalESDHybridTrackCuts("hybridcuts2018TRD", AliEmcalESDHybridTrackCuts::kDef2018TRD);
    hybridcuts2018TRD->SetUseNoITSrefitTracks(kFALSE);
    trkSel->AddTrackCuts(hybridcuts2018TRD);
    break;
  }
  default:
  {
    ::Error("AliEmcalESDTrackCutsGenerator::AddHybridTrackCuts", "Hybrid track cuts not available for dataset %d", period);
    break;
  }
  }
}

void AliEmcalESDTrackCutsGenerator::AddTPCOnlyTrackCuts(AliEmcalTrackSelection* trkSel, EDataSet_t period)
{
  switch (period) {
  case kLHC11c:
  case kLHC11d:
  case kLHC11h:
  {
    AliESDtrackCuts *cutsp = CreateTrackCutsPWGJE(kTPCOnlyTracksNCls70);
    trkSel->AddTrackCuts(cutsp);
    break;
  }
  default:
  {
    Printf("AliEmcalESDTrackCutsGenerator::AddTPCOnlyTrackCuts: TPC only track cuts not available for dataset %d", period);
    break;
  }
  }
}