PWGLF/RESONANCES/macros/lego_train/AddRsnPairsKStar.C

   1 void AddRsnPairsKStar(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  = 313;
  16    TDatabasePDG *db   = TDatabasePDG::Instance();
  17    TParticlePDG *part = db->GetParticle(pdg);
  18    Double_t mass = part->Mass();
  19
  20    Bool_t valid;
  21    Int_t isRsnMini = AliRsnTrainManager::GetGlobalInt("IsRsnMini",valid);
  22
  23    if (isRsnMini) {
  24       AddPairOutputMiniKStar(task,isMC,isMixing,pType1,listID1,pType2,listID2,pdg,mass,cutsPair,suffix);
  25    } else {
  26       // this function is common and it is located in RsnConfig.C
  27       // as ouptup AddPairOutputKStar from this macro will be taken
  28       AddPair(task,isMC,isMixing,pType1,listID1,pType2,listID2,pdg,mass,cutsEvent,cutsPair,suffix);
  29    }
  30 }
  31 void AddPairOutputKStar(AliRsnLoopPair *pair)
  32 {
  33    Bool_t valid;
  34
  35    Int_t isFullOutput = AliRsnTrainManager::GetGlobalInt("RsnOutputFull",valid);
  36    Int_t collisionType = AliRsnTrainManager::GetGlobalInt("IsCollisionType",valid);
  37    Int_t useRapidity = AliRsnTrainManager::GetGlobalInt("RsnUseRapidity",valid);
  38
  39    // axes
  40    AliRsnValuePair *axisIM = new AliRsnValuePair("IM", AliRsnValuePair::kInvMass);
  41    axisIM     ->SetBins(900, 0.6, 1.5);
  42
  43    AliRsnValuePair *axisPt = new AliRsnValuePair("PT", AliRsnValuePair::kPt);
  44    axisPt     ->SetBins(120, 0.0, 12.0);
  45
  46    AliRsnValuePair *axisEta = new AliRsnValuePair("ETA", AliRsnValuePair::kEta);
  47    axisEta    ->SetBins(400, -0.5, 0.5);
  48
  49    AliRsnValueEvent *axisCentrality = 0;
  50    if (collisionType==1) axisCentrality = new AliRsnValueEvent("MULTI",AliRsnValueEvent::kCentralityV0);
  51
  52    // output: 2D histogram of inv. mass vs. pt
  53    AliRsnListOutput *outPair = 0;
  54    if (!isFullOutput) {
  55       outPair = new AliRsnListOutput("pair", AliRsnListOutput::kHistoDefault);
  56       outPair->AddValue(axisIM);
  57    } else {
  58       outPair = new AliRsnListOutput("pair", AliRsnListOutput::kHistoSparse);
  59       outPair->AddValue(axisIM);
  60       outPair->AddValue(axisPt);
  61       outPair->AddValue(axisEta);
  62       if (axisCentrality) axisCentrality->SetBins(20,0,100);
  63    }
  64    // add outputs to loop
  65    pair->AddOutput(outPair);
  66 }
  67
  68 void AddPairOutputMiniKStar(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 = "") {
  69
  70    Bool_t valid;
  71    Int_t isFullOutput = AliRsnTrainManager::GetGlobalInt("RsnOutputFull",valid);
  72    Int_t useMixing = AliRsnTrainManager::GetGlobalInt("IsMixing",valid);
  73    Int_t collisionType = AliRsnTrainManager::GetGlobalInt("IsCollisionType",valid);
  74
  75    AliRsnMiniAnalysisTask *taskRsnMini =  (AliRsnMiniAnalysisTask *)task;
  76
  77    if (collisionType==0) taskRsnMini->UseMultiplicity("QUALITY");
  78    else {
  79       taskRsnMini->UseCentrality("V0M");
  80       Int_t multID = taskRsnMini->CreateValue(AliRsnMiniValue::kMult, kFALSE);
  81       AliRsnMiniOutput *outMult = taskRsnMini->CreateOutput("eventMult", "HIST", "EVENT");
  82       outMult->AddAxis(multID, 100, 0.0, 100.0);
  83       Int_t paID = taskRsnMini->CreateValue(AliRsnMiniValue::kPlaneAngle, kFALSE);
  84       AliRsnMiniOutput *outPa = taskRsnMini->CreateOutput("planeAngle", "HIST", "EVENT");
  85       outPa->AddAxis(paID, 100, 0, TMath::Pi());
  86    }
  87
  88    /* invariant mass   */ Int_t imID   = taskRsnMini->CreateValue(AliRsnMiniValue::kInvMass, kFALSE);
  89    /* IM resolution    */ Int_t resID  = taskRsnMini->CreateValue(AliRsnMiniValue::kInvMassRes, kTRUE);
  90    /* transv. momentum */ Int_t ptID   = taskRsnMini->CreateValue(AliRsnMiniValue::kPt, kFALSE);
  91    /* centrality       */ Int_t centID = taskRsnMini->CreateValue(AliRsnMiniValue::kMult, kFALSE);
  92    /* eta              */ Int_t etaID = taskRsnMini->CreateValue(AliRsnMiniValue::kEta, kFALSE);
  93    /* rapidity         */ Int_t yID = taskRsnMini->CreateValue(AliRsnMiniValue::kY, kFALSE);
  94
  95    Bool_t useRapidity = kTRUE;
  96
  97    Int_t nIM   = 90; Double_t minIM   = 0.6, maxIM =  1.5;
  98    Int_t nRes   = 200; Double_t minRes   = -0.02, maxRes =  0.02;
  99    Int_t nEta   = 400; Double_t minEta   = -0.5, maxEta =  0.5;
 100    Int_t nY   = 16; Double_t minY   = -0.8, maxY =  0.8;
 101    Int_t nPt   = 120; Double_t minPt   = 0.0, maxPt = 12.0;
 102    Int_t nCent = 100; Double_t minCent = 0.0, maxCent = 100.0;
 103    //
 104    // -- Create all needed outputs -----------------------------------------------------------------
 105    //
 106
 107    Int_t iCutK = listID1;
 108    Int_t iCutPi = listID2;
 109
 110    // common definitions
 111    TString outputType = "HIST";
 112    if (isFullOutput) outputType = "SPARSE";
 113
 114    // use an array for more compact writing, which are different on mixing and charges
 115    // [0] = unlike
 116    // [1] = mixing
 117    // [2] = like ++
 118    // [3] = like --
 119 //    const Int_t num = 12;
 120    Bool_t  use     [12] = { 1       ,  1       , useMixing, useMixing,  1      ,  1      ,  isMC   ,   isMC   ,  isMC   ,   isMC   ,  isMC    ,   isMC     };
 121    Bool_t  useIM   [12] = { 1       ,  1       ,  1       ,  1       ,  1      ,  1      ,  1      ,   1      ,  0      ,   0      ,  1       ,   1        };
 122    TString name    [12] = {"Unlike1", "Unlike2", "Mixing1", "Mixing2", "LikePP", "LikeMM", "Trues1",  "Trues2", "Res1"  ,  "Res2"  , "Mother1",  "Mother2" };
 123    TString comp    [12] = {"PAIR"   , "PAIR"   , "MIX"    , "MIX"    , "PAIR"  , "PAIR"  , "TRUE"  ,  "TRUE"  , "TRUE"  ,  "TRUE"  , "MOTHER" ,  "MOTHER"  };
 124    Char_t  charge1 [12] = {'+'      , '-'      , '+'      , '-'      , '+'     , '-'     , '+'     ,  '-'     , '+'     ,  '-'     , '+'      ,  '-'       };
 125    Char_t  charge2 [12] = {'-'      , '+'      , '-'      , '+'      , '+'     , '-'     , '-'     ,  '+'     , '-'     ,  '+'     , '-'      ,  '+'       };
 126    Int_t   cutID1  [12] = { iCutK   ,  iCutK   ,  iCutK   ,  iCutK   ,  iCutK  ,  iCutK  ,  iCutK  ,   iCutK  ,  iCutK  ,   iCutK  ,  iCutK  ,   iCutK     };
 127    Int_t   cutID2  [12] = { iCutPi  ,  iCutPi  ,  iCutPi  ,  iCutPi  ,  iCutPi ,  iCutPi ,  iCutPi ,   iCutPi ,  iCutPi ,   iCutPi ,  iCutPi ,   iCutPi    };
 128
 129    for (Int_t i = 0; i < 12; i++) {
 130       if (!use[i]) continue;
 131       // create output
 132       AliRsnMiniOutput *out = taskRsnMini->CreateOutput(Form("%s_%s", suffix.Data(), name[i].Data()), outputType.Data(), comp[i].Data());
 133       // selection settings
 134       out->SetCutID(0, cutID1[i]);
 135       out->SetCutID(1, cutID2[i]);
 136       out->SetDaughter(0, AliRsnDaughter::kKaon);
 137       out->SetDaughter(1, AliRsnDaughter::kPion);
 138       out->SetCharge(0, charge1[i]);
 139       out->SetCharge(1, charge2[i]);
 140       out->SetMotherPDG(pdgMother);
 141       out->SetMotherMass(massMother);
 142       // pair cuts
 143       if (cutsPair) out->SetPairCuts(cutsPair);
 144       // axis X: invmass (or resolution)
 145       if (useIM[i])
 146          out->AddAxis(imID, nIM, minIM, maxIM);
 147       else
 148          out->AddAxis(resID, nRes, minRes, maxRes);
 149
 150       if (isFullOutput) {
 151          // axis Y: transverse momentum
 152          out->AddAxis(ptID, nPt, minPt, maxPt);
 153          if (useRapidity) out->AddAxis(yID, nY, minY, maxY);
 154          else  out->AddAxis(etaID, nEta, minEta, maxEta);
 155          // axis Z: centrality
 156          if (collisionType==1) out->AddAxis(centID, nCent, minCent, maxCent);
 157       }
 158    }
 159
 160    // -- Create output for MC generated ------------------------------------------------------------
 161    //
 162
 163    if (isMC) {
 164       // create ouput
 165       AliRsnMiniOutput *outMC = taskRsnMini->CreateOutput(Form("kstar_MCGen1%s", suffix.Data()), outputType.Data(), "MOTHER");
 166       // selection settings
 167       outMC->SetDaughter(0, AliRsnDaughter::kPion);
 168       outMC->SetDaughter(1, AliRsnDaughter::kKaon);
 169       outMC->SetMotherPDG(313);
 170       outMC->SetMotherMass(massMother);
 171       // pair cuts
 172       if (cutsPair) outMC->SetPairCuts(cutsPair);
 173       // axis X: invmass
 174       outMC->AddAxis(imID, nIM, minIM, maxIM);
 175       if (isFullOutput) {
 176          // axis Y: transverse momentum
 177          outMC->AddAxis(ptID, nPt, minPt, maxPt);
 178          if (useRapidity) outMC->AddAxis(yID, nY, minY, maxY);
 179          else  outMC->AddAxis(etaID, nEta, minEta, maxEta);
 180          // axis Z: centrality
 181          if (collisionType==1) outMC->AddAxis(centID, nCent, minCent, maxCent);
 182       }
 183    }
 184
 185
 186
 187    if (isMC) {
 188       // create ouput
 189       AliRsnMiniOutput *outMC1 = taskRsnMini->CreateOutput(Form("phi_MCGen2%s", suffix.Data()), outputType.Data(), "MOTHER");
 190       // selection settings
 191       outMC1->SetDaughter(0, AliRsnDaughter::kKaon);
 192       outMC1->SetDaughter(1, AliRsnDaughter::kPion);
 193       outMC1->SetMotherPDG(-313);
 194       outMC1->SetMotherMass(massMother);
 195       // pair cuts
 196       if (cutsPair) outMC1->SetPairCuts(cutsPair);
 197       // axis X: invmass
 198       outMC1->AddAxis(imID, nIM, minIM, maxIM);
 199       if (isFullOutput) {
 200          // axis Y: transverse momentum
 201          outMC1->AddAxis(ptID, nPt, minPt, maxPt);
 202          if (useRapidity) outMC1->AddAxis(yID, nY, minY, maxY);
 203          else  outMC1->AddAxis(etaID, nEta, minEta, maxEta);
 204          // axis Z: centrality
 205          if (collisionType==1) outMC1->AddAxis(centID, nCent, minCent, maxCent);
 206       }
 207    }
 208
 209
 210 }
 211