PWGCF/FEMTOSCOPY/macros/Train/pPbpionFemto/AOD/3D/pippim/ConfigFemtoAnalysis.C

   1
   2 /*********************************************************************
   3  *                                                                   *
   4  * ConfigFemtoAnalysis.C - configuration macro for the femtoscopic   *
   5  * analysis, meant as a QA process for two-particle effects          *
   6  *                                                                   *
   7  * Author: Adam Kisiel (Adam.Kisiel@cern.ch)                         *
   8  *                                                                   *
   9  *********************************************************************/
  10
  11 #if !defined(__CINT__) || defined(__MAKECINT_)
  12 #include "AliFemtoManager.h"
  13 #include "AliFemtoEventReaderESDChain.h"
  14 #include "AliFemtoEventReaderESDChainKine.h"
  15 #include "AliFemtoEventReaderAODChain.h"
  16 #include "AliFemtoSimpleAnalysis.h"
  17 #include "AliFemtoBasicEventCut.h"
  18 #include "AliFemtoESDTrackCut.h"
  19 #include "AliFemtoCorrFctn.h"
  20 #include "AliFemtoCutMonitorParticleYPt.h"
  21 #include "AliFemtoCutMonitorParticleVertPos.h"
  22 #include "AliFemtoCutMonitorParticleMomRes.h"
  23 #include "AliFemtoCutMonitorParticlePID.h"
  24 #include "AliFemtoCutMonitorEventMult.h"
  25 #include "AliFemtoCutMonitorEventVertex.h"
  26 #include "AliFemtoShareQualityTPCEntranceSepPairCut.h"
  27 #include "AliFemtoPairCutAntiGamma.h"
  28 #include "AliFemtoPairCutRadialDistance.h"
  29 #include "AliFemtoQinvCorrFctn.h"
  30 #include "AliFemtoShareQualityCorrFctn.h"
  31 #include "AliFemtoTPCInnerCorrFctn.h"
  32 #include "AliFemtoVertexMultAnalysis.h"
  33 #include "AliFemtoCorrFctn3DSpherical.h"
  34 #include "AliFemtoChi2CorrFctn.h"
  35 #include "AliFemtoCorrFctnTPCNcls.h"
  36 #include "AliFemtoBPLCMS3DCorrFctn.h"
  37 #include "AliFemtoCorrFctn3DLCMSSym.h"
  38 #include "AliFemtoModelBPLCMSCorrFctn.h"
  39 #include "AliFemtoModelCorrFctn3DSpherical.h"
  40 #include "AliFemtoModelGausLCMSFreezeOutGenerator.h"
  41 #include "AliFemtoModelGausRinvFreezeOutGenerator.h"
  42 #include "AliFemtoModelManager.h"
  43 #include "AliFemtoModelWeightGeneratorBasic.h"
  44 #include "AliFemtoModelWeightGeneratorLednicky.h"
  45 #include "AliFemtoCorrFctnDirectYlm.h"
  46 #include "AliFemtoModelCorrFctnDirectYlm.h"
  47 #include "AliFemtoModelCorrFctnSource.h"
  48 #include "AliFemtoCutMonitorParticlePtPDG.h"
  49 #include "AliFemtoKTPairCut.h"
  50 #include "AliFemtoCutMonitorCollections.h"
  51 #endif
  52
  53 //________________________________________________________________________
  54 AliFemtoManager* ConfigFemtoAnalysis() {
  55
  56   double PionMass = 0.13956995;
  57   double KaonMass = 0.493677;
  58
  59   //multiplicity bins
  60   int runmults[10] = {1, 1, 1, 1, 0, 0, 0, 0, 0, 0};
  61   int multbins[11] = {0.01, 200, 400, 600, 900, 950, 500, 600, 700, 800, 900};
  62
  63   int runch[3] = {0, 0, 1};
  64   const char *chrgs[3] = { "pip", "pim", "pippim" };
  65
  66   int runktdep = 1;
  67   double ktrng[8] = {0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 1.0};
  68
  69   int run3d = 1; // Do 3D cartesian analysis?
  70   int runshlcms = 1;
  71
  72   //PhysicsSelection set only in runBatch with trigger + Physics Selection Task
  73
  74
  75   double shqmax;
  76   int nbinssh = 200;
  77
  78   if (runshlcms) shqmax = 2.0;
  79   else shqmax = 0.9;
  80
  81   AliFemtoEventReaderAODChain *Reader = new AliFemtoEventReaderAODChain();
  82   Reader->SetFilterBit(0);
  83   //Reader->SetCentralityPreSelection(500, 950);
  84
  85   //AliFemtoEventReaderESDChainKine* Reader=new AliFemtoEventReaderESDChainKine();
  86   //Reader->SetConstrained(true);
  87   // Reader->SetUseTPCOnly(false);
  88
  89   //AliFemtoEventReaderESDChain *Reader = new AliFemtoEventReaderESDChain();
  90   //Reader->SetUseMultiplicity(AliFemtoEventReaderESDChain::kV0Centrality);
  91   //Reader->SetReadTrackType(AliFemtoEventReaderESDChain::kGlobal);
  92
  93
  94   AliFemtoManager* Manager=new AliFemtoManager();
  95   Manager->SetEventReader(Reader);
  96
  97   AliFemtoVertexMultAnalysis    *anetaphitpc[320];
  98   AliFemtoBasicEventCut         *mecetaphitpc[320];
  99   AliFemtoCutMonitorEventMult   *cutPassEvMetaphitpc[320];
 100   AliFemtoCutMonitorEventMult   *cutFailEvMetaphitpc[320];
 101   AliFemtoCutMonitorEventVertex *cutPassEvVetaphitpc[320];
 102   AliFemtoCutMonitorEventVertex *cutFailEvVetaphitpc[320];
 103   AliFemtoCutMonitorCollections   *cutPassColletaphitpc[320];
 104   AliFemtoCutMonitorCollections   *cutFailColletaphitpc[320];
 105   AliFemtoESDTrackCut           *dtc1etaphitpc[320];
 106   AliFemtoESDTrackCut           *dtc2etaphitpc[320];
 107   AliFemtoCutMonitorParticleYPt *cutPass1YPtetaphitpc[320];
 108   AliFemtoCutMonitorParticleYPt *cutFail1YPtetaphitpc[320];
 109   AliFemtoCutMonitorParticlePID *cutPass1PIDetaphitpc[320];
 110   AliFemtoCutMonitorParticlePID *cutFail1PIDetaphitpc[320];
 111   AliFemtoCutMonitorParticleYPt *cutPass2YPtetaphitpc[320];
 112   AliFemtoCutMonitorParticleYPt *cutFail2YPtetaphitpc[320];
 113   AliFemtoCutMonitorParticlePID *cutPass2PIDetaphitpc[320];
 114   AliFemtoCutMonitorParticlePID *cutFail2PIDetaphitpc[320];
 115   //  AliFemtoPairCutAntiGamma      *sqpcetaphitpc[320];
 116   AliFemtoShareQualityTPCEntranceSepPairCut      *sqpcetaphitpc[320];
 117   //AliFemtoPairCutRadialDistance      *sqpcetaphitpc[320];
 118   AliFemtoCorrFctnDirectYlm     *cylmetaphitpc[320];
 119   AliFemtoCorrFctnDEtaDPhi      *cdedpetaphi[320];
 120   AliFemtoChi2CorrFctn          *cchiqinvetaphitpc[320];
 121   AliFemtoKTPairCut             *ktpcuts[320*7];
 122   AliFemtoCorrFctnDirectYlm     *cylmkttpc[320*7];
 123   AliFemtoQinvCorrFctn          *cqinvkttpc[320*7];
 124   AliFemtoCorrFctn3DLCMSSym     *cq3dlcmskttpc[320*7];
 125   AliFemtoCorrFctnTPCNcls       *cqinvnclstpc[320];
 126   AliFemtoShareQualityCorrFctn  *cqinvsqtpc[320*10];
 127   AliFemtoChi2CorrFctn          *cqinvchi2tpc[320];
 128   AliFemtoTPCInnerCorrFctn      *cqinvinnertpc[320*10];
 129
 130   // *** Begin pion-pion analysis ***
 131   int aniter = 0;
 132
 133   for (int imult=0; imult<10; imult++) {
 134     if (runmults[imult]) {
 135       for (int ichg=0; ichg<3; ichg++) {
 136         if (runch[ichg]) {
 137           aniter = ichg*10+imult;
 138
 139           anetaphitpc[aniter] = new AliFemtoVertexMultAnalysis(10, -10.0, 10.0, 4, multbins[imult], multbins[imult+1]);
 140           anetaphitpc[aniter]->SetNumEventsToMix(5);
 141           anetaphitpc[aniter]->SetMinSizePartCollection(1);
 142
 143           mecetaphitpc[aniter] = new AliFemtoBasicEventCut();
 144           mecetaphitpc[aniter]->SetEventMult(0,10000);
 145           mecetaphitpc[aniter]->SetVertZPos(-10,10);
 146
 147           cutPassEvMetaphitpc[aniter] = new AliFemtoCutMonitorEventMult(Form("cutPass%stpcM%i", chrgs[ichg], imult),500);
 148           cutFailEvMetaphitpc[aniter] = new AliFemtoCutMonitorEventMult(Form("cutFail%stpcM%i", chrgs[ichg], imult),500);
 149           mecetaphitpc[aniter]->AddCutMonitor(cutPassEvMetaphitpc[aniter], cutFailEvMetaphitpc[aniter]);
 150
 151           cutPassEvVetaphitpc[aniter] = new AliFemtoCutMonitorEventVertex(Form("cutPass%stpcM%i", chrgs[ichg], imult));
 152           cutFailEvVetaphitpc[aniter] = new AliFemtoCutMonitorEventVertex(Form("cutFail%stpcM%i", chrgs[ichg], imult));
 153           mecetaphitpc[aniter]->AddCutMonitor(cutPassEvVetaphitpc[aniter], cutFailEvVetaphitpc[aniter]);
 154
 155           cutPassColletaphitpc[aniter] = new AliFemtoCutMonitorCollections(Form("cutPass%stpcM%i", chrgs[ichg], imult));
 156           cutFailColletaphitpc[aniter] = new AliFemtoCutMonitorCollections(Form("cutFail%stpcM%i", chrgs[ichg], imult));
 157           mecetaphitpc[aniter]->AddCutMonitor(cutPassColletaphitpc[aniter], cutFailColletaphitpc[aniter]);
 158
 159           dtc1etaphitpc[aniter] = new AliFemtoESDTrackCut();
 160
 161           dtc1etaphitpc[aniter]->SetCharge(1.0);
 162
 163           dtc1etaphitpc[aniter]->SetPt(0.12,4.0);
 164           dtc1etaphitpc[aniter]->SetEta(-1.2,1.2);
 165
 166           //PID method
 167           dtc1etaphitpc[aniter]->SetMass(PionMass);
 168           dtc1etaphitpc[aniter]->SetMostProbablePion();
 169           //dtc1etaphitpc[aniter]->SetPIDMethod(AliFemtoESDTrackCut::kContour);
 170
 171           //Track quality cuts
 172           dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit|AliESDtrack::kITSrefit);
 173           dtc1etaphitpc[aniter]->SetminTPCncls(50);
 174           dtc1etaphitpc[aniter]->SetRemoveKinks(kTRUE);
 175
 176
 177           dtc1etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
 178           dtc1etaphitpc[aniter]->SetMaxITSChiNdof(36);
 179           dtc1etaphitpc[aniter]->SetLabel(kFALSE);
 180
 181           //primary particles: hits in ITS + DCA cut
 182           dtc1etaphitpc[aniter]->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
 183                                          AliESDtrackCuts::kAny);
 184           dtc1etaphitpc[aniter]->SetMaxImpactZ(2.0);
 185           //dtc1etaphitpc[aniter]->SetMaxImpactXY(2.4);
 186           dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0105, 0.0350, -1.1);
 187           //dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0182, 0.0350, -1.01);
 188           //dtc1etaphitpc[aniter]->SetMaxSigmaToVertex(6.0);
 189
 190           dtc2etaphitpc[aniter] = new AliFemtoESDTrackCut();
 191
 192           dtc2etaphitpc[aniter]->SetCharge(-1.0);
 193
 194           dtc2etaphitpc[aniter]->SetPt(0.12,4.0);
 195           dtc2etaphitpc[aniter]->SetEta(-1.2,1.2);
 196
 197           //PID method
 198           dtc2etaphitpc[aniter]->SetMass(PionMass);
 199           dtc2etaphitpc[aniter]->SetMostProbablePion();
 200           //dtc2etaphitpc[aniter]->SetPIDMethod(AliFemtoESDTrackCut::kContour);
 201
 202           //Track quality cuts
 203           dtc2etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit|AliESDtrack::kITSrefit);
 204           dtc2etaphitpc[aniter]->SetminTPCncls(50);
 205           dtc2etaphitpc[aniter]->SetRemoveKinks(kTRUE);
 206
 207
 208           dtc2etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
 209           dtc2etaphitpc[aniter]->SetMaxITSChiNdof(36);
 210           dtc2etaphitpc[aniter]->SetLabel(kFALSE);
 211
 212           //primary particles: hits in ITS + DCA cut
 213           dtc2etaphitpc[aniter]->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
 214                                          AliESDtrackCuts::kAny);
 215           dtc2etaphitpc[aniter]->SetMaxImpactZ(2.0);
 216           //dtc2etaphitpc[aniter]->SetMaxImpactXY(2.4);
 217           dtc2etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0105, 0.0350, -1.1);
 218           //dtc2etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0182, 0.0350, -1.01);
 219           //dtc2etaphitpc[aniter]->SetMaxSigmaToVertex(6.0);
 220
 221           cutPass1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutPass1%stpcM%i", chrgs[ichg], imult), 0.13957);
 222           cutFail1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutFail1%stpcM%i", chrgs[ichg], imult), 0.13957);
 223           dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1YPtetaphitpc[aniter], cutFail1YPtetaphitpc[aniter]);
 224
 225           cutPass1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutPass1%stpcM%i", chrgs[ichg], imult),0);
 226           cutFail1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutFail1%stpcM%i", chrgs[ichg], imult),0);
 227           dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1PIDetaphitpc[aniter], cutFail1PIDetaphitpc[aniter]);
 228
 229           //sqpcetaphitpc[aniter] = new AliFemtoPairCutAntiGamma();
 230           sqpcetaphitpc[aniter] = new AliFemtoShareQualityTPCEntranceSepPairCut();
 231           //sqpcetaphitpc[aniter] = new AliFemtoPairCutRadialDistance();
 232           sqpcetaphitpc[aniter]->SetShareQualityMax(1.0);
 233           sqpcetaphitpc[aniter]->SetShareFractionMax(0.05);
 234           sqpcetaphitpc[aniter]->SetRemoveSameLabel(kFALSE);
 235           //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.016);
 236           //sqpcetaphitpc[aniter]->SetPhiStarDifferenceMinimum(0.02);
 237           //runtype==0
 238           // sqpcetaphitpc[aniter]->SetMaxEEMinv(0.0);
 239           // sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.0);
 240           // sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(1.5);
 241           // sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
 242           // sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(0.12, 0.03);
 243           //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
 244           //runtype==1
 245           //        sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(5.0);
 246           //        sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
 247           //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(1.2, 0.03);
 248           //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
 249           //runtype==2
 250           //        sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(1.0);
 251           //        sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
 252           //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(1.2, 0.045);
 253           //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.016);
 254           //sqpcetaphitpc[aniter]->SetPhiStarDifferenceMinimum(0.02);
 255
 256           anetaphitpc[aniter]->SetEventCut(mecetaphitpc[aniter]);
 257           if(ichg==0)
 258             {
 259               anetaphitpc[aniter]->SetFirstParticleCut(dtc1etaphitpc[aniter]);
 260               anetaphitpc[aniter]->SetSecondParticleCut(dtc1etaphitpc[aniter]);
 261             }
 262           else if(ichg==1)
 263             {
 264               anetaphitpc[aniter]->SetFirstParticleCut(dtc2etaphitpc[aniter]);
 265               anetaphitpc[aniter]->SetSecondParticleCut(dtc2etaphitpc[aniter]);
 266             }
 267           else if(ichg==2)
 268             {
 269               anetaphitpc[aniter]->SetFirstParticleCut(dtc1etaphitpc[aniter]);
 270               anetaphitpc[aniter]->SetSecondParticleCut(dtc2etaphitpc[aniter]);
 271             }
 272           anetaphitpc[aniter]->SetPairCut(sqpcetaphitpc[aniter]);
 273
 274           //Correlation functions
 275
 276           //Spherical harmonics (without kT bins)
 277           //cylmetaphitpc[aniter] = new AliFemtoCorrFctnDirectYlm(Form("cylm%stpcM%i", chrgs[ichg], imult),3,nbinssh,0.0,shqmax,runshlcms);
 278           //anetaphitpc[aniter]->AddCorrFctn(cylmetaphitpc[aniter]);
 279
 280           //Qinv (without kT bins)
 281           //cqinvkttpc[aniter] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
 282           //anetaphitpc[aniter]->AddCorrFctn(cqinvkttpc[aniter]);
 283
 284           //3D cartesian (without kT bins)
 285           if(run3d){
 286             cq3dlcmskttpc[aniter] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%i", chrgs[ichg], imult),100,0.5);
 287             anetaphitpc[aniter]->AddCorrFctn(cq3dlcmskttpc[aniter]);
 288           }
 289
 290           // cqinvnclstpc[aniter] = new AliFemtoCorrFctnTPCNcls(Form("cqinvncls%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
 291           // anetaphitpc[aniter]->AddCorrFctn(cqinvnclstpc[aniter]);
 292
 293           // cqinvchi2tpc[aniter] = new AliFemtoChi2CorrFctn(Form("cqinvchi2%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
 294           // anetaphitpc[aniter]->AddCorrFctn(cqinvchi2tpc[aniter]);
 295
 296           if (runktdep) {
 297             int ktm;
 298             for (int ikt=0; ikt<7; ikt++) {
 299               ktm = aniter*7 + ikt;
 300               ktpcuts[ktm] = new AliFemtoKTPairCut(ktrng[ikt], ktrng[ikt+1]);
 301
 302               //cylmkttpc[ktm] = new AliFemtoCorrFctnDirectYlm(Form("cylm%stpcM%ikT%i", chrgs[ichg], imult, ikt),3,
 303               //                                             nbinssh, 0.0, shqmax, runshlcms);
 304               //cylmkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 305               //anetaphitpc[aniter]->AddCorrFctn(cylmkttpc[ktm]);
 306
 307               //cqinvkttpc[ktm] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0, shqmax);
 308               //cqinvkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 309               //anetaphitpc[aniter]->AddCorrFctn(cqinvkttpc[ktm]);
 310
 311               //cqinvsqtpc[ktm] = new AliFemtoShareQualityCorrFctn(Form("cqinvsq%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
 312               //cqinvsqtpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 313               //anetaphitpc[aniter]->AddCorrFctn(cqinvsqtpc[ktm]);
 314
 315               //cqinvinnertpc[ktm] = new AliFemtoTPCInnerCorrFctn(Form("cqinvinner%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
 316               //cqinvinnertpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 317               //cqinvinnertpc[ktm]->SetRadius(1.2);
 318               //anetaphitpc[aniter]->AddCorrFctn(cqinvinnertpc[ktm]);
 319
 320               if (run3d) {
 321                 //              cq3dlcmskttpc[ktm] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%ikT%i", chrgs[ichg], imult, ikt),60,(imult>3)?((imult>6)?((imult>7)?0.6:0.4):0.25):0.15);
 322                 cq3dlcmskttpc[ktm] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%ikT%i", chrgs[ichg], imult, ikt),100,0.5);
 323                 cq3dlcmskttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 324                 anetaphitpc[aniter]->AddCorrFctn(cq3dlcmskttpc[ktm]);
 325               }
 326             }
 327           }
 328
 329           //cdedpetaphi[aniter] = new AliFemtoCorrFctnDEtaDPhi(Form("cdedp%stpcM%i", chrgs[ichg], imult),39, 39);
 330           //anetaphitpc[aniter]->AddCorrFctn(cdedpetaphi[aniter]);
 331
 332           Manager->AddAnalysis(anetaphitpc[aniter]);
 333         }
 334       }
 335     }
 336   }
 337   // *** End pion-pion analysis
 338
 339   return Manager;
 340 }