PWGLF/RESONANCES/macros/mini/ConfigLambdaStarPbPbMC.C

   1 /***************************************************************************
   2               fbellini@cern.ch - last modified on 06/08/2012
   3
   4 // *** Configuration script for Lambda*, anti-Lambda* analysis with 2011 PbPb runs ***
   5 //
   6 // A configuration script for RSN package needs to define the followings:
   7 //
   8 // (1) decay tree of each resonance to be studied, which is needed to select
   9 //     true pairs and to assign the right mass to all candidate daughters
  10 // (2) cuts at all levels: single daughters, tracks, events
  11 // (3) output objects: histograms or trees
  12 ****************************************************************************/
  13
  14 Bool_t ConfigLambdaStarPbPbMC
  15 (
  16     AliRsnMiniAnalysisTask *task,
  17     Bool_t                 isMC,
  18     Bool_t                 isPP,
  19     const char             *suffix,
  20     AliRsnCutSet           *cutsPair,
  21     Int_t                  aodFilterBit = 5,
  22     AliRsnCutSetDaughterParticle::ERsnDaughterCutSet cutPrCandidate = AliRsnCutSetDaughterParticle::kTPCTOFpidKstarPP2010,
  23     AliRsnCutSetDaughterParticle::ERsnDaughterCutSet cutKaCandidate = AliRsnCutSetDaughterParticle::kTPCTOFpidKstarPP2010,
  24     Float_t                nsigmaPr = 2.0,
  25     Float_t                nsigmaKa = 2.0,
  26     Bool_t                 enableMonitor = kTRUE,
  27     Bool_t                 IsMcTrueOnly = kFALSE,
  28     Int_t                  Pdg = 3124,
  29     Int_t                  aodN = 0
  30 )
  31 {
  32   // manage suffix
  33   if (strlen(suffix) > 0) suffix = Form("_%s", suffix);
  34
  35   // set daughter cuts
  36   AliRsnCutSetDaughterParticle * cutSetQ;
  37   AliRsnCutSetDaughterParticle * cutSetPr;
  38   AliRsnCutSetDaughterParticle * cutSetK;
  39
  40   cutSetQ  = new AliRsnCutSetDaughterParticle("cutQuality", AliRsnCutSetDaughterParticle::kQualityStd2011, AliPID::kKaon, -1.0, aodFilterBit);
  41   cutSetPr = new AliRsnCutSetDaughterParticle(Form("cutProtonTPCpp2010_%2.1fsigma",nsigmaPr), cutPrCandidate, AliPID::kProton, nsigmaPr, aodFilterBit);
  42   cutSetK  = new AliRsnCutSetDaughterParticle(Form("cutKaonTPCpp2010_%2.1f2sigma",nsigmaKa), cutKaCandidate, AliPID::kKaon, nsigmaKa, aodFilterBit);
  43
  44   Int_t iCutQ = task->AddTrackCuts(cutSetQ);
  45   Int_t iCutPr = task->AddTrackCuts(cutSetPr);
  46   Int_t iCutK = task->AddTrackCuts(cutSetK);
  47
  48   if(enableMonitor){
  49     Printf("======== Monitoring cut AliRsnCutSetDaughterParticle enabled");
  50     gROOT->LoadMacro("$ALICE_ROOT/PWGLF/RESONANCES/macros/mini/AddMonitorOutput.C");
  51     AddMonitorOutput(isMC, cutSetQ->GetMonitorOutput());
  52     AddMonitorOutput(isMC, cutSetPr->GetMonitorOutput());
  53     AddMonitorOutput(isMC, cutSetK->GetMonitorOutput());
  54   }
  55
  56   // -- Values ------------------------------------------------------------------------------------
  57   /* invariant mass   */ Int_t imID   = task->CreateValue(AliRsnMiniValue::kInvMass, kFALSE);
  58   /* IM resolution    */ Int_t resID  = task->CreateValue(AliRsnMiniValue::kInvMassRes, kTRUE);
  59   /* transv. momentum */ Int_t ptID   = task->CreateValue(AliRsnMiniValue::kPt, kFALSE);
  60   /* centrality       */ Int_t centID = task->CreateValue(AliRsnMiniValue::kMult, kFALSE);
  61   /* pseudorapidity   */ Int_t etaID  = task->CreateValue(AliRsnMiniValue::kEta, kFALSE);
  62   /* rapidity         */ Int_t yID    = task->CreateValue(AliRsnMiniValue::kY, kFALSE);
  63
  64   // -- Create all needed outputs -----------------------------------------------------------------
  65   // use an array for more compact writing, which are different on mixing and charges
  66   // [0] = unlike
  67   // [1] = mixing
  68   // [2] = like ++
  69   // [3] = like --
  70   Bool_t  use     [10] = { !IsMcTrueOnly,  !IsMcTrueOnly,  !IsMcTrueOnly,  !IsMcTrueOnly ,  !IsMcTrueOnly, !IsMcTrueOnly,  isMC   ,   isMC   ,  isMC   ,   isMC   };
  71   Bool_t  useIM   [10] = { 1       ,  1       ,  1       ,  1       ,  1      ,  1      ,  1      ,   1      ,  0      ,   0      };
  72   TString name    [10] = {"UnlikePM", "UnlikeMP", "MixingPM", "MixingMP", "LikePP", "LikeMM", "TruesPM",  "TruesMP", "ResPM"  ,  "ResMP"  };
  73   TString comp    [10] = {"PAIR"   , "PAIR"   , "MIX"    , "MIX"    , "PAIR"  , "PAIR"  , "TRUE"  ,  "TRUE"  , "TRUE"  ,  "TRUE"  };
  74   //TString output  [10] = {"HIST"   , "HIST"   , "HIST"   , "HIST"   , "HIST"  , "HIST"  , "HIST"  ,  "HIST"  , "HIST"  ,  "HIST"  };
  75   TString output  [10] = {"SPARSE"   , "SPARSE"   , "SPARSE"   , "SPARSE"   , "SPARSE"  , "SPARSE"  , "SPARSE"  ,  "SPARSE"  , "SPARSE"  ,  "SPARSE"  };
  76   Char_t  charge1 [10] = {'+'      , '-'      , '+'      , '-'      , '+'     , '-'     , '+'     ,  '-'     , '+'     ,  '-'     };
  77   Char_t  charge2 [10] = {'-'      , '+'      , '-'      , '+'      , '+'     , '-'     , '-'     ,  '+'     , '-'     ,  '+'     };
  78   Int_t   cutID1  [10] = { iCutK   ,  iCutK   ,  iCutK   ,  iCutK   ,  iCutK  ,  iCutK  ,  iCutK  ,   iCutK  ,  iCutK  ,   iCutK  };
  79   Int_t   cutID2  [10] = { iCutPr  ,  iCutPr  ,  iCutPr  ,  iCutPr  ,  iCutPr ,  iCutPr ,  iCutPr ,   iCutPr ,  iCutPr ,   iCutPr };
  80
  81   for (Int_t i = 0; i < 10; i++) {
  82     if (!use[i]) continue;
  83     if(Pdg > 0) AliRsnMiniOutput *out = task->CreateOutput(Form("lstar1_%s%s", name[i].Data(), suffix), output[i].Data(), comp[i].Data());
  84     if(Pdg < 0) AliRsnMiniOutput *out = task->CreateOutput(Form("lstar2_%s%s", name[i].Data(), suffix), output[i].Data(), comp[i].Data());
  85     out->SetCutID(0, cutID1[i]);
  86     out->SetCutID(1, cutID2[i]);
  87     out->SetDaughter(0, AliRsnDaughter::kKaon);
  88     out->SetDaughter(1, AliRsnDaughter::kProton);
  89     out->SetCharge(0, charge1[i]);
  90     out->SetCharge(1, charge2[i]);
  91     out->SetMotherPDG(Pdg);//2122
  92     out->SetMotherMass(1.520);//GeV
  93     out->SetPairCuts(cutsPair);
  94
  95     // axis X: invmass (or resolution)
  96     if (useIM[i])
  97       out->AddAxis(imID, 900, 1.3, 2.2);
  98     //else
  99     //out->AddAxis(resID, 200, -0.02, 0.02);
 100
 101     // axis Y: transverse momentum
 102     out->AddAxis(ptID, 500, 0.0, 50.0);
 103
 104     // axis Z: centrality-multiplicity
 105     if (!isPP)
 106       out->AddAxis(centID, 100, 0.0, 100.0);
 107     else
 108       out->AddAxis(centID, 400, 0.0, 400.0);
 109
 110     // axis W: pseudorapidity
 111     //    out->AddAxis(etaID, 20, -1.0, 1.0);
 112     // axis J: rapidity
 113     //out->AddAxis(yID, 32, -0.8, 0.8);
 114
 115   }
 116
 117   if (isMC){
 118     // create output
 119
 120     if(Pdg > 0) {AliRsnMiniOutput *outm = task->CreateOutput(Form("lstar_Mother1%s", suffix), "SPARSE", "MOTHER");}
 121     if(Pdg < 0) {AliRsnMiniOutput *outm = task->CreateOutput(Form("lstar_Mother2%s", suffix), "SPARSE", "MOTHER");}
 122     outm->SetDaughter(0, AliRsnDaughter::kKaon);
 123     outm->SetDaughter(1, AliRsnDaughter::kProton);
 124     outm->SetMotherPDG(Pdg);//313
 125     outm->SetMotherMass(1.520);
 126     // pair cuts
 127     outm->SetPairCuts(cutsPair);
 128     // binnings
 129     outm->AddAxis(imID, 900, 1.3, 2.2);
 130     outm->AddAxis(ptID, 500, 0.0, 50.0);
 131     if (!isPP){
 132       outm->AddAxis(centID, 100, 0.0, 100.0);
 133     }   else {
 134       outm->AddAxis(centID, 400, 0.0, 400.0);
 135     }
 136     //outm->AddAxis(yID, 32, -0.8, 0.8);
 137   }
 138   return kTRUE;
 139 }