ROOT logo
#ifndef __CINT__
#endif
void AddRsnPairsLambda(AliAnalysisTaskSE *task,
                      Bool_t isMC,
                      Bool_t isMixing,
                      AliPID::EParticleType pType1,
                      Int_t listID1,
                      AliPID::EParticleType pType2,
                      Int_t listID2,
                      AliRsnCutSet *cutsEvent=0,
                      AliRsnCutSet *cutsPair=0,
                      TString suffix = "") {

   Printf("id1=%d id2=%d",listID1,listID2);

   // retrieve mass from PDG database
   Int_t    pdg  = 3124;
   Double_t mass = 1.5195;
   
   Bool_t valid;
   Int_t isRsnMini = AliAnalysisManager::GetGlobalInt("rsnUseMiniPackage",valid);
   if (isRsnMini) {
      AddPairOutputMiniLambda(task,isMC,isMixing,pType1,listID1,pType2,listID2,pdg,mass,cutsPair,suffix);
   } else {
      // this function is common and it is located in RsnConfig.C
      // as ouptup AddPairOutputLambda from this macro will be taken
      AddPair(task,isMC,isMixing,pType1,listID1,pType2,listID2,pdg,mass,cutsEvent,cutsPair,suffix);
   }
}
void AddPairOutputLambda(AliRsnLoopPair *pair)
{
   Bool_t valid;
   Int_t isFullOutput = AliAnalysisManager::GetGlobalInt("rsnOutputFull",valid);

   // axes
   AliRsnValuePair *axisIM = new AliRsnValuePair("IM", AliRsnValuePair::kInvMass);
   AliRsnValuePair *axisPt = new AliRsnValuePair("PT", AliRsnValuePair::kPt);
   axisIM     ->SetBins(700, 1.3, 2.0);
   axisPt     ->SetBins(120, 0.0, 12.0);

   // output: 2D histogram of inv. mass vs. pt
   AliRsnListOutput *outPair = 0;
   if (!isFullOutput) {
      outPair = new AliRsnListOutput("pair", AliRsnListOutput::kHistoDefault);
      outPair->AddValue(axisIM);
   } else {
      outPair = new AliRsnListOutput("pair", AliRsnListOutput::kHistoSparse);
      outPair->AddValue(axisIM);
      outPair->AddValue(axisPt);
   }
   // add outputs to loop
   pair->AddOutput(outPair);
}

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 = "") {

   Bool_t valid;
   Int_t isFullOutput = AliAnalysisManager::GetGlobalInt("rsnOutputFull",valid);
   Int_t useMixing = AliAnalysisManager::GetGlobalInt("rsnUseMixing",valid);
   AliRsnMiniAnalysisTask *taskRsnMini =  (AliRsnMiniAnalysisTask *)task;
   /* invariant mass   */ Int_t imID   = taskRsnMini->CreateValue(AliRsnMiniValue::kInvMass, kFALSE);
   /* IM resolution    */ Int_t resID  = taskRsnMini->CreateValue(AliRsnMiniValue::kInvMassRes, kTRUE);
   /* transv. momentum */ Int_t ptID   = taskRsnMini->CreateValue(AliRsnMiniValue::kPt, kFALSE);
   /* centrality       */ Int_t centID = taskRsnMini->CreateValue(AliRsnMiniValue::kMult, kFALSE);

   //
   // -- Create all needed outputs -----------------------------------------------------------------
   //

   Int_t iCutK = listID1;
   Int_t iCutPi = listID2;

   // common definitions
   TString outputType = "HIST";
   if (isFullOutput) outputType = "SPARSE";

   // use an array for more compact writing, which are different on mixing and charges
   // [0] = unlike
   // [1] = mixing
   // [2] = like ++
   // [3] = like --
   Bool_t  use     [12] = { 1       ,  1       ,  useMixing       ,  useMixing       ,  1      ,  1      ,  isMC   ,   isMC   ,  isMC   ,   isMC   , isMC , isMC };
   Bool_t  useIM   [12] = { 1       ,  1       ,  1       ,  1       ,  1      ,  1      ,  1      ,   1      ,  0      ,   0      ,  1      ,   1      };
   TString name    [12] = {"Unlike1", "Unlike2", "Mixing1", "Mixing2", "LikePP", "LikeMM", "Trues1",  "Trues2", "Res1"  ,  "Res2"  ,"Mother1", "Mother2"};
   TString comp    [12] = {"PAIR"   , "PAIR"   , "MIX"    , "MIX"    , "PAIR"  , "PAIR"  , "TRUE"  ,  "TRUE"  , "TRUE"  ,  "TRUE"  , "MOTHER",  "MOTHER"};
   Char_t  charge1 [12] = {'+'      , '-'      , '+'      , '-'      , '+'     , '-'     , '+'     ,  '-'     , '+'     ,  '-'     , '+'     ,  '-'     };
   Char_t  charge2 [12] = {'-'      , '+'      , '-'      , '+'      , '+'     , '-'     , '-'     ,  '+'     , '-'     ,  '+'     , '-'     ,  '+'     };
   Int_t   cutID1  [12] = { iCutK   ,  iCutK   ,  iCutK   ,  iCutK   ,  iCutK  ,  iCutK  ,  iCutK  ,   iCutK  ,  iCutK  ,   iCutK  ,  iCutK  ,   iCutK  };
   Int_t   cutID2  [12] = { iCutPi  ,  iCutPi  ,  iCutPi  ,  iCutPi  ,  iCutPi ,  iCutPi ,  iCutPi ,   iCutPi ,  iCutPi ,   iCutPi ,  iCutPi ,   iCutPi };

   for (Int_t i = 0; i < 12; i++) {
      if (!use[i]) continue;
      // create output
      AliRsnMiniOutput *out = taskRsnMini->CreateOutput(Form("%s_%s", suffix.Data(), name[i].Data()), outputType.Data(), comp[i].Data());
      // selection settings
      out->SetCutID(0, cutID1[i]);
      out->SetCutID(1, cutID2[i]);
      out->SetDaughter(0, AliRsnDaughter::kProton);
      out->SetDaughter(1, AliRsnDaughter::kKaon);
      out->SetCharge(0, charge1[i]);
      out->SetCharge(1, charge2[i]);
      out->SetMotherPDG(pdgMother);
      out->SetMotherMass(massMother);
      // pair cuts
      if (cutsPair) out->SetPairCuts(cutsPair);
      // axis X: invmass (or resolution)
      if (useIM[i])
         out->AddAxis(imID, 700, 1.3, 2.0);
      else
         out->AddAxis(resID, 200, -0.02, 0.02);

      if (isFullOutput) {
         // axis Y: transverse momentum
         out->AddAxis(ptID, 100, 0.0, 10.0);
         // axis Z: centrality
         out->AddAxis(centID, 100, 0.0, 100.0);
      }
   }

}
 AddRsnPairsLambda.C:1
 AddRsnPairsLambda.C:2
 AddRsnPairsLambda.C:3
 AddRsnPairsLambda.C:4
 AddRsnPairsLambda.C:5
 AddRsnPairsLambda.C:6
 AddRsnPairsLambda.C:7
 AddRsnPairsLambda.C:8
 AddRsnPairsLambda.C:9
 AddRsnPairsLambda.C:10
 AddRsnPairsLambda.C:11
 AddRsnPairsLambda.C:12
 AddRsnPairsLambda.C:13
 AddRsnPairsLambda.C:14
 AddRsnPairsLambda.C:15
 AddRsnPairsLambda.C:16
 AddRsnPairsLambda.C:17
 AddRsnPairsLambda.C:18
 AddRsnPairsLambda.C:19
 AddRsnPairsLambda.C:20
 AddRsnPairsLambda.C:21
 AddRsnPairsLambda.C:22
 AddRsnPairsLambda.C:23
 AddRsnPairsLambda.C:24
 AddRsnPairsLambda.C:25
 AddRsnPairsLambda.C:26
 AddRsnPairsLambda.C:27
 AddRsnPairsLambda.C:28
 AddRsnPairsLambda.C:29
 AddRsnPairsLambda.C:30
 AddRsnPairsLambda.C:31
 AddRsnPairsLambda.C:32
 AddRsnPairsLambda.C:33
 AddRsnPairsLambda.C:34
 AddRsnPairsLambda.C:35
 AddRsnPairsLambda.C:36
 AddRsnPairsLambda.C:37
 AddRsnPairsLambda.C:38
 AddRsnPairsLambda.C:39
 AddRsnPairsLambda.C:40
 AddRsnPairsLambda.C:41
 AddRsnPairsLambda.C:42
 AddRsnPairsLambda.C:43
 AddRsnPairsLambda.C:44
 AddRsnPairsLambda.C:45
 AddRsnPairsLambda.C:46
 AddRsnPairsLambda.C:47
 AddRsnPairsLambda.C:48
 AddRsnPairsLambda.C:49
 AddRsnPairsLambda.C:50
 AddRsnPairsLambda.C:51
 AddRsnPairsLambda.C:52
 AddRsnPairsLambda.C:53
 AddRsnPairsLambda.C:54
 AddRsnPairsLambda.C:55
 AddRsnPairsLambda.C:56
 AddRsnPairsLambda.C:57
 AddRsnPairsLambda.C:58
 AddRsnPairsLambda.C:59
 AddRsnPairsLambda.C:60
 AddRsnPairsLambda.C:61
 AddRsnPairsLambda.C:62
 AddRsnPairsLambda.C:63
 AddRsnPairsLambda.C:64
 AddRsnPairsLambda.C:65
 AddRsnPairsLambda.C:66
 AddRsnPairsLambda.C:67
 AddRsnPairsLambda.C:68
 AddRsnPairsLambda.C:69
 AddRsnPairsLambda.C:70
 AddRsnPairsLambda.C:71
 AddRsnPairsLambda.C:72
 AddRsnPairsLambda.C:73
 AddRsnPairsLambda.C:74
 AddRsnPairsLambda.C:75
 AddRsnPairsLambda.C:76
 AddRsnPairsLambda.C:77
 AddRsnPairsLambda.C:78
 AddRsnPairsLambda.C:79
 AddRsnPairsLambda.C:80
 AddRsnPairsLambda.C:81
 AddRsnPairsLambda.C:82
 AddRsnPairsLambda.C:83
 AddRsnPairsLambda.C:84
 AddRsnPairsLambda.C:85
 AddRsnPairsLambda.C:86
 AddRsnPairsLambda.C:87
 AddRsnPairsLambda.C:88
 AddRsnPairsLambda.C:89
 AddRsnPairsLambda.C:90
 AddRsnPairsLambda.C:91
 AddRsnPairsLambda.C:92
 AddRsnPairsLambda.C:93
 AddRsnPairsLambda.C:94
 AddRsnPairsLambda.C:95
 AddRsnPairsLambda.C:96
 AddRsnPairsLambda.C:97
 AddRsnPairsLambda.C:98
 AddRsnPairsLambda.C:99
 AddRsnPairsLambda.C:100
 AddRsnPairsLambda.C:101
 AddRsnPairsLambda.C:102
 AddRsnPairsLambda.C:103
 AddRsnPairsLambda.C:104
 AddRsnPairsLambda.C:105
 AddRsnPairsLambda.C:106
 AddRsnPairsLambda.C:107
 AddRsnPairsLambda.C:108
 AddRsnPairsLambda.C:109
 AddRsnPairsLambda.C:110
 AddRsnPairsLambda.C:111
 AddRsnPairsLambda.C:112
 AddRsnPairsLambda.C:113
 AddRsnPairsLambda.C:114
 AddRsnPairsLambda.C:115
 AddRsnPairsLambda.C:116
 AddRsnPairsLambda.C:117
 AddRsnPairsLambda.C:118
 AddRsnPairsLambda.C:119
 AddRsnPairsLambda.C:120
 AddRsnPairsLambda.C:121