PWGLF/RESONANCES/macros/lego_train/AddRsnPairsPhi.C

   1 void AddRsnPairsPhi(AliAnalysisTaskSE *task,
   2                     Bool_t isMC,
   3                     Bool_t isMixing,
   4                     AliPID::EParticleType pType1,
   5                     Int_t listID1,
   6                     AliPID::EParticleType pType2,
   7                     Int_t listID2,
   8                     AliRsnCutSet *cutsEvent=0,
   9                     AliRsnCutSet *cutsPair=0,
  10                     TString suffix = "") {
  11
  12    Printf("id1=%d id2=%d",listID1,listID2);
  13
  14    // retrieve mass from PDG database
  15    Int_t         pdg  = 333;
  16    TDatabasePDG *db   = TDatabasePDG::Instance();
  17    TParticlePDG *part = db->GetParticle(pdg);
  18    Double_t mass = part->Mass();
  19    Bool_t valid;
  20    Int_t isRsnMini = AliAnalysisManager::GetGlobalInt("rsnUseMiniPackage",valid);
  21
  22    if (isRsnMini) {
  23       AddPairOutputMiniPhi(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 AddPairOutputPhi from this macro will be taken
  27       AddPair(task,isMC,isMixing,pType1,listID1,pType2,listID2,pdg,mass,cutsEvent,cutsPair,suffix);
  28    }
  29 }
  30 void AddPairOutputPhi(AliRsnLoopPair *pair)
  31 {
  32    Bool_t valid;
  33    Int_t isFullOutput = AliAnalysisManager::GetGlobalInt("rsnOutputFull",valid);
  34    // axes
  35    AliRsnValuePair *axisIM = new AliRsnValuePair("IM", AliRsnValuePair::kInvMass);
  36    AliRsnValuePair *axisPt = new AliRsnValuePair("PT", AliRsnValuePair::kPt);
  37    axisIM     ->SetBins(300, 0.9, 1.2);
  38 //   axisIM     ->SetBins(1000, 0.9, 1.9);
  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 AddPairOutputMiniPhi(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    Int_t isPP = AliAnalysisManager::GetGlobalInt("rsnIsPP",valid);
  61
  62    AliRsnMiniAnalysisTask *taskRsnMini =  (AliRsnMiniAnalysisTask *)task;
  63
  64    /* invariant mass   */ Int_t imID   = taskRsnMini->CreateValue(AliRsnMiniValue::kInvMass, kFALSE);
  65    /* IM resolution    */ Int_t resID  = taskRsnMini->CreateValue(AliRsnMiniValue::kInvMassDiff, kTRUE);
  66    /* transv. momentum */ Int_t ptID   = taskRsnMini->CreateValue(AliRsnMiniValue::kPt, kFALSE);
  67    /* centrality       */ Int_t centID = taskRsnMini->CreateValue(AliRsnMiniValue::kMult, kFALSE);
  68    /* eta              */ Int_t etaID = taskRsnMini->CreateValue(AliRsnMiniValue::kEta, kFALSE);
  69
  70
  71    // use an array for more compact writing, which are different on mixing and charges
  72    // [0] = unlike
  73    // [1] = mixing
  74    // [2] = like ++
  75    // [3] = like --
  76    Bool_t  use     [5] = { 1      ,  useMixing      ,  1      ,  1      ,  isMC  };
  77    TString name    [5] = {"Unlike", "Mixing", "LikePP", "LikeMM", "Trues"};
  78    TString comp    [5] = {"PAIR"  , "MIX"   , "PAIR"  , "PAIR"  , "TRUE" };
  79    Char_t  charge1 [5] = {'+'     , '+'     , '+'     , '-'     , '+'    };
  80    Char_t  charge2 [5] = {'-'     , '-'     , '+'     , '-'     , '-'    };
  81
  82    // common definitions
  83    TString outputType = "HIST";
  84    if (isFullOutput) outputType = "SPARSE";
  85
  86    Int_t nIM   = 300; Double_t minIM   = 0.9, maxIM =  1.2;
  87    Int_t nEta   = 200; Double_t minEta   = -1.0, maxEta =  1.0;
  88 //   Int_t nIM   = 1000; Double_t minIM   = 0.9, maxIM =  1.9;
  89    Int_t nPt   = 120; Double_t minPt   = 0.0, maxPt = 12.0;
  90    Int_t nCent = 100; Double_t minCent = 0.0, maxCent = 100.0;
  91    Int_t nRes  = 200; Double_t maxRes  = 0.01;
  92
  93    // retrieve mass from PDG database
  94    Int_t         pdg  = 333;
  95    TDatabasePDG *db   = TDatabasePDG::Instance();
  96    TParticlePDG *part = db->GetParticle(pdg);
  97
  98
  99    Printf(suffix.Data());
 100    // create standard outputs
 101    for (Int_t i = 0; i < 5; i++) {
 102       if (!use[i]) continue;
 103       // create output
 104       AliRsnMiniOutput *out = taskRsnMini->CreateOutput(Form("%s_%s", suffix.Data(),name[i].Data() ), outputType.Data(), comp[i].Data());
 105       // selection settings
 106       out->SetCutID(0, listID1);
 107       out->SetCutID(1, listID1);
 108       out->SetDaughter(0, AliRsnDaughter::kKaon);
 109       out->SetDaughter(1, AliRsnDaughter::kKaon);
 110       out->SetCharge(0, charge1[i]);
 111       out->SetCharge(1, charge2[i]);
 112       out->SetMotherPDG(pdg);
 113       out->SetMotherMass(part->Mass());
 114       // pair cuts
 115       if (cutsPair) out->SetPairCuts(cutsPair);
 116       // axis X: invmass
 117       out->AddAxis(imID, nIM, minIM, maxIM);
 118
 119       if (isFullOutput) {
 120          // axis Y: transverse momentum
 121          out->AddAxis(ptID, nPt, minPt, maxPt);
 122
 123          out->AddAxis(etaID, nEta, minEta, maxEta);
 124          // axis Z: centrality
 125          if (!isPP) out->AddAxis(centID, nCent, minCent, maxCent);
 126       }
 127    }
 128
 129    // add output for resolution
 130    if (isMC) {
 131       AliRsnMiniOutput *outRes = taskRsnMini->CreateOutput(Form("phi_Res%s", suffix.Data()), outputType.Data(), "TRUE");
 132       // selection settings
 133       outRes->SetCutID(0, listID1);
 134       outRes->SetCutID(1, listID1);
 135       outRes->SetDaughter(0, AliRsnDaughter::kKaon);
 136       outRes->SetDaughter(1, AliRsnDaughter::kKaon);
 137       outRes->SetCharge(0, '+');
 138       outRes->SetCharge(1, '-');
 139       outRes->SetMotherPDG(pdg);
 140       outRes->SetMotherMass(part->Mass());
 141       // pair cuts
 142       if (cutsPair) outRes->SetPairCuts(cutsPair);
 143       // axis X: resolution
 144       outRes->AddAxis(resID, nRes, -maxRes, maxRes);
 145
 146       if (isFullOutput) {
 147          // axis Y: transverse momentum
 148          outRes->AddAxis(ptID, nPt, minPt, maxPt);
 149          outRes->AddAxis(etaID, nEta, minEta, maxEta);
 150          // axis Z: centrality
 151          if (!isPP) outRes->AddAxis(centID, nCent, minCent, maxCent);
 152       }
 153    }
 154
 155    //
 156    // -- Create output for MC generated ------------------------------------------------------------
 157    //
 158
 159    if (isMC) {
 160       // create ouput
 161       AliRsnMiniOutput *outMC = taskRsnMini->CreateOutput(Form("phi_MCGen%s", suffix.Data()), outputType.Data(), "MOTHER");
 162       // selection settings
 163       outMC->SetDaughter(0, AliRsnDaughter::kKaon);
 164       outMC->SetDaughter(1, AliRsnDaughter::kKaon);
 165       outMC->SetMotherPDG(pdg);
 166       outMC->SetMotherMass(part->Mass());
 167       // pair cuts
 168       if (cutsPair) outMC->SetPairCuts(cutsPair);
 169       // axis X: invmass
 170       outMC->AddAxis(imID, nIM, minIM, maxIM);
 171       if (isFullOutput) {
 172          // axis Y: transverse momentum
 173          outMC->AddAxis(ptID, nPt, minPt, maxPt);
 174          outMC->AddAxis(etaID, nEta, minEta, maxEta);
 175          // axis Z: centrality
 176          if (!isPP) outMC->AddAxis(centID, nCent, minCent, maxCent);
 177       }
 178    }
 179
 180
 181 }