PWGLF/RESONANCES/macros/lego_train/AddRsnPairsLambda.C

   1 #ifndef __CINT__
   2 #endif
   3 void AddRsnPairsLambda(AliAnalysisTaskSE *task,
   4                       Bool_t isMC,
   5                       Bool_t isMixing,
   6                       AliPID::EParticleType pType1,
   7                       Int_t listID1,
   8                       AliPID::EParticleType pType2,
   9                       Int_t listID2,
  10                       AliRsnCutSet *cutsEvent=0,
  11                       AliRsnCutSet *cutsPair=0,
  12                       TString suffix = "") {
  13
  14    Printf("id1=%d id2=%d",listID1,listID2);
  15
  16    // retrieve mass from PDG database
  17    Int_t    pdg  = 3124;
  18    Double_t mass = 1.5195;
  19
  20    Bool_t valid;
  21    Int_t isRsnMini = AliAnalysisManager::GetGlobalInt("rsnUseMiniPackage",valid);
  22    if (isRsnMini) {
  23       AddPairOutputMiniLambda(task,isMC,isMixing,pType1,listID1,pType2,listID2,pdg,mass,cutsPair,suffix);
  24    } else {
  25       // this function is common and it is located in RsnConfig.C
  26       // as ouptup AddPairOutputLambda from this macro will be taken
  27       AddPair(task,isMC,isMixing,pType1,listID1,pType2,listID2,pdg,mass,cutsEvent,cutsPair,suffix);
  28    }
  29 }
  30 void AddPairOutputLambda(AliRsnLoopPair *pair)
  31 {
  32    Bool_t valid;
  33    Int_t isFullOutput = AliAnalysisManager::GetGlobalInt("rsnOutputFull",valid);
  34
  35    // axes
  36    AliRsnValuePair *axisIM = new AliRsnValuePair("IM", AliRsnValuePair::kInvMass);
  37    AliRsnValuePair *axisPt = new AliRsnValuePair("PT", AliRsnValuePair::kPt);
  38    axisIM     ->SetBins(700, 1.3, 2.0);
  39    axisPt     ->SetBins(120, 0.0, 12.0);
  40
  41    // output: 2D histogram of inv. mass vs. pt
  42    AliRsnListOutput *outPair = 0;
  43    if (!isFullOutput) {
  44       outPair = new AliRsnListOutput("pair", AliRsnListOutput::kHistoDefault);
  45       outPair->AddValue(axisIM);
  46    } else {
  47       outPair = new AliRsnListOutput("pair", AliRsnListOutput::kHistoSparse);
  48       outPair->AddValue(axisIM);
  49       outPair->AddValue(axisPt);
  50    }
  51    // add outputs to loop
  52    pair->AddOutput(outPair);
  53 }
  54
  55 void AddPairOutputMiniLambda(AliAnalysisTaskSE *task,Bool_t isMC,Bool_t isMixing, AliPID::EParticleType pType1,Int_t listID1, AliPID::EParticleType pType2,Int_t listID2, Int_t pdgMother,Double_t massMother, AliRsnCutSet *cutsPair=0,TString suffix = "") {
  56
  57    Bool_t valid;
  58    Int_t isFullOutput = AliAnalysisManager::GetGlobalInt("rsnOutputFull",valid);
  59    Int_t useMixing = AliAnalysisManager::GetGlobalInt("rsnUseMixing",valid);
  60    AliRsnMiniAnalysisTask *taskRsnMini =  (AliRsnMiniAnalysisTask *)task;
  61    /* invariant mass   */ Int_t imID   = taskRsnMini->CreateValue(AliRsnMiniValue::kInvMass, kFALSE);
  62    /* IM resolution    */ Int_t resID  = taskRsnMini->CreateValue(AliRsnMiniValue::kInvMassRes, kTRUE);
  63    /* transv. momentum */ Int_t ptID   = taskRsnMini->CreateValue(AliRsnMiniValue::kPt, kFALSE);
  64    /* centrality       */ Int_t centID = taskRsnMini->CreateValue(AliRsnMiniValue::kMult, kFALSE);
  65
  66    //
  67    // -- Create all needed outputs -----------------------------------------------------------------
  68    //
  69
  70    Int_t iCutK = listID1;
  71    Int_t iCutPi = listID2;
  72
  73    // common definitions
  74    TString outputType = "HIST";
  75    if (isFullOutput) outputType = "SPARSE";
  76
  77    // use an array for more compact writing, which are different on mixing and charges
  78    // [0] = unlike
  79    // [1] = mixing
  80    // [2] = like ++
  81    // [3] = like --
  82    Bool_t  use     [12] = { 1       ,  1       ,  useMixing       ,  useMixing       ,  1      ,  1      ,  isMC   ,   isMC   ,  isMC   ,   isMC   , isMC , isMC };
  83    Bool_t  useIM   [12] = { 1       ,  1       ,  1       ,  1       ,  1      ,  1      ,  1      ,   1      ,  0      ,   0      ,  1      ,   1      };
  84    TString name    [12] = {"Unlike1", "Unlike2", "Mixing1", "Mixing2", "LikePP", "LikeMM", "Trues1",  "Trues2", "Res1"  ,  "Res2"  ,"Mother1", "Mother2"};
  85    TString comp    [12] = {"PAIR"   , "PAIR"   , "MIX"    , "MIX"    , "PAIR"  , "PAIR"  , "TRUE"  ,  "TRUE"  , "TRUE"  ,  "TRUE"  , "MOTHER",  "MOTHER"};
  86    Char_t  charge1 [12] = {'+'      , '-'      , '+'      , '-'      , '+'     , '-'     , '+'     ,  '-'     , '+'     ,  '-'     , '+'     ,  '-'     };
  87    Char_t  charge2 [12] = {'-'      , '+'      , '-'      , '+'      , '+'     , '-'     , '-'     ,  '+'     , '-'     ,  '+'     , '-'     ,  '+'     };
  88    Int_t   cutID1  [12] = { iCutK   ,  iCutK   ,  iCutK   ,  iCutK   ,  iCutK  ,  iCutK  ,  iCutK  ,   iCutK  ,  iCutK  ,   iCutK  ,  iCutK  ,   iCutK  };
  89    Int_t   cutID2  [12] = { iCutPi  ,  iCutPi  ,  iCutPi  ,  iCutPi  ,  iCutPi ,  iCutPi ,  iCutPi ,   iCutPi ,  iCutPi ,   iCutPi ,  iCutPi ,   iCutPi };
  90
  91    for (Int_t i = 0; i < 12; i++) {
  92       if (!use[i]) continue;
  93       // create output
  94       AliRsnMiniOutput *out = taskRsnMini->CreateOutput(Form("%s_%s", suffix.Data(), name[i].Data()), outputType.Data(), comp[i].Data());
  95       // selection settings
  96       out->SetCutID(0, cutID1[i]);
  97       out->SetCutID(1, cutID2[i]);
  98       out->SetDaughter(0, AliRsnDaughter::kProton);
  99       out->SetDaughter(1, AliRsnDaughter::kKaon);
 100       out->SetCharge(0, charge1[i]);
 101       out->SetCharge(1, charge2[i]);
 102       out->SetMotherPDG(pdgMother);
 103       out->SetMotherMass(massMother);
 104       // pair cuts
 105       if (cutsPair) out->SetPairCuts(cutsPair);
 106       // axis X: invmass (or resolution)
 107       if (useIM[i])
 108          out->AddAxis(imID, 700, 1.3, 2.0);
 109       else
 110          out->AddAxis(resID, 200, -0.02, 0.02);
 111
 112       if (isFullOutput) {
 113          // axis Y: transverse momentum
 114          out->AddAxis(ptID, 100, 0.0, 10.0);
 115          // axis Z: centrality
 116          out->AddAxis(centID, 100, 0.0, 100.0);
 117       }
 118    }
 119
 120 }