PWGCF/FEMTOSCOPY/macros/Train/ProtonFemtoscopy/MagPlus/ConfigFemtoAnalysis.C

   1 /*********************************************************************
   2  *                                                                   *
   3  * Configfemtoanalysis.C - configuration macro for the femtoscopic   *
   4  * analysis, meant as a QA process for two-particle effects          *
   5  *                                                                   *
   6  * Author: Adam Kisiel (Adam.Kisiel@cern.ch)                         *
   7  *                                                                   *
   8  *********************************************************************/
   9
  10 #if !defined(__CINT__) || defined(__MAKECINT_)
  11 #include "AliFemtoManager.h"
  12 #include "AliFemtoEventReaderESDChain.h"
  13 #include "AliFemtoEventReaderESDChainKine.h"
  14 #include "AliFemtoEventReaderAODChain.h"
  15 #include "AliFemtoSimpleAnalysis.h"
  16 #include "AliFemtoBasicEventCut.h"
  17 #include "AliFemtoESDTrackCut.h"
  18 #include "AliFemtoCorrFctn.h"
  19 #include "AliFemtoCutMonitorParticleYPt.h"
  20 #include "AliFemtoCutMonitorParticleVertPos.h"
  21 #include "AliFemtoCutMonitorParticleMomRes.h"
  22 #include "AliFemtoCutMonitorParticlePID.h"
  23 #include "AliFemtoCutMonitorEventMult.h"
  24 #include "AliFemtoCutMonitorEventVertex.h"
  25 #include "AliFemtoShareQualityTPCEntranceSepPairCut.h"
  26 #include "AliFemtoPairCutAntiGamma.h"
  27 #include "AliFemtoPairCutRadialDistance.h"
  28 #include "AliFemtoQinvCorrFctn.h"
  29 #include "AliFemtoCorrFctnNonIdDR.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 #endif
  51
  52 //________________________________________________________________________
  53 AliFemtoManager* ConfigFemtoAnalysis() {
  54
  55         double PionMass = 0.13956995;
  56         double KaonMass = 0.493677;
  57         double ProtonMass = 0.938272013;
  58
  59         int runmults[10] = {1, 1, 0, 0, 0, 0, 0, 0, 0, 0};
  60         int multbins[11] = {0.001, 50, 100, 200, 300, 400, 500, 600, 700, 800, 900};
  61
  62         int runch[3] = {1, 1, 1};
  63         const char *chrgs[3] = { "PP", "APAP", "PAP" };
  64
  65         int runktdep = 1;
  66         double ktrng[5] = {0.1, 0.3, 0.6, 1.0, 10.0};
  67
  68         int numOfMultBins = 10;
  69         int numOfChTypes = 3;
  70         int numOfkTbins = 4;
  71
  72         int runqinv = 1;
  73         int runshlcms = 0; // 0:PRF(PAP), 1:LCMS(PP,APAP)
  74
  75         int runtype = 2; // Types 0 - global, 1 - ITS only, 2 - TPC Inner
  76         int isrealdata = 1;
  77
  78         //  int gammacut = 1;
  79
  80         double shqmax = 2.0;
  81         int nbinssh = 200;
  82
  83         AliFemtoEventReaderAODChain *Reader = new AliFemtoEventReaderAODChain();
  84         Reader->SetFilterBit(7);
  85         Reader->SetCentralityPreSelection(0.001, 510);
  86
  87         AliFemtoManager* Manager = new AliFemtoManager();
  88         Manager->SetEventReader(Reader);
  89
  90         AliFemtoVertexMultAnalysis    *anetaphitpc[10*3];
  91         AliFemtoBasicEventCut         *mecetaphitpc[10*3];
  92
  93         AliFemtoCutMonitorEventMult   *cutPassEvMetaphitpc[10*3];
  94         AliFemtoCutMonitorEventMult   *cutFailEvMetaphitpc[10*3];
  95
  96         AliFemtoCutMonitorEventVertex *cutPassEvVetaphitpc[10*3];
  97         AliFemtoCutMonitorEventVertex *cutFailEvVetaphitpc[10*3];
  98
  99         AliFemtoESDTrackCut           *dtc1etaphitpc[10*3];
 100         AliFemtoESDTrackCut           *dtc2etaphitpc[10*3];
 101
 102         AliFemtoCutMonitorParticleYPt *cutPass1YPtetaphitpc[10*3];
 103         AliFemtoCutMonitorParticleYPt *cutFail1YPtetaphitpc[10*3];
 104
 105         AliFemtoCutMonitorParticlePID *cutPass1PIDetaphitpc[10*3];
 106         AliFemtoCutMonitorParticlePID *cutFail1PIDetaphitpc[10*3];
 107
 108         AliFemtoPairCutRadialDistance      *sqpcetaphitpc[10*3];
 109
 110         AliFemtoKTPairCut             *ktpcuts[100*3*4];
 111         AliFemtoCorrFctnDirectYlm     *cylmkttpc[100*3];
 112
 113         AliFemtoQinvCorrFctn          *cqinvkttpc[100*3*4];
 114         AliFemtoQinvCorrFctn          *cqinvtpc[100*3];
 115
 116         AliFemtoCorrFctnNonIdDR       *ckstartpc[100*3];
 117         AliFemtoCorrFctnNonIdDR       *ckstarkttpc[100*3*4];
 118
 119         // *** Third QA task - HBT analysis with all pair cuts off, TPC only ***
 120         // *** Begin pion-pion (positive) analysis ***
 121         int aniter = 0;
 122
 123         for (int imult = 0; imult < numOfMultBins; imult++) {
 124                 if (runmults[imult]) {
 125
 126                         for (int ichg = 0; ichg < numOfChTypes; ichg++) {
 127                                 if (runch[ichg]) {
 128
 129                                         aniter = ichg * numOfMultBins + imult;
 130
 131                                         anetaphitpc[aniter] = new AliFemtoVertexMultAnalysis(8, -8.0, 8.0, 4, multbins[imult], multbins[imult+1]);
 132                                         anetaphitpc[aniter]->SetNumEventsToMix(10);
 133                                         anetaphitpc[aniter]->SetMinSizePartCollection(1);
 134
 135                                         mecetaphitpc[aniter] = new AliFemtoBasicEventCut();
 136                                         mecetaphitpc[aniter]->SetEventMult(0.001,100000);
 137                                         mecetaphitpc[aniter]->SetVertZPos(-8,8);
 138
 139                                         if (isrealdata)
 140                                                 mecetaphitpc[aniter]->SetAcceptOnlyPhysics(kTRUE);
 141
 142                                         cutPassEvMetaphitpc[aniter] = new AliFemtoCutMonitorEventMult(Form("cutPass%stpcM%i", chrgs[ichg], imult));
 143                                         cutFailEvMetaphitpc[aniter] = new AliFemtoCutMonitorEventMult(Form("cutFail%stpcM%i", chrgs[ichg], imult));
 144                                         mecetaphitpc[aniter]->AddCutMonitor(cutPassEvMetaphitpc[aniter], cutFailEvMetaphitpc[aniter]);
 145
 146                                         // cutPassEvVetaphitpc[aniter] = new AliFemtoCutMonitorEventVertex(Form("cutPass%stpcM%i", chrgs[ichg], imult));
 147                                         // cutFailEvVetaphitpc[aniter] = new AliFemtoCutMonitorEventVertex(Form("cutFail%stpcM%i", chrgs[ichg], imult));
 148                                         // mecetaphitpc[aniter]->AddCutMonitor(cutPassEvVetaphitpc[aniter], cutFailEvVetaphitpc[aniter]);
 149
 150                                         dtc1etaphitpc[aniter] = new AliFemtoESDTrackCut();
 151                                         dtc2etaphitpc[aniter] = new AliFemtoESDTrackCut();
 152
 153                                         if (ichg == 0) { // pp
 154                                                 dtc1etaphitpc[aniter]->SetCharge(1.0);
 155                                                 dtc1etaphitpc[aniter]->SetPt(0.5,5.0);
 156                                                 dtc1etaphitpc[aniter]->SetEta(-0.8,0.8);
 157                                                 dtc1etaphitpc[aniter]->SetMass(ProtonMass);
 158                                                 dtc1etaphitpc[aniter]->SetMostProbableProton();
 159
 160                                         }
 161
 162                                         else if (ichg == 1) { // apap
 163                                                 dtc1etaphitpc[aniter]->SetCharge(-1.0);
 164                                                 dtc2etaphitpc[aniter]->SetPt(0.2,5.0);
 165                                                 dtc1etaphitpc[aniter]->SetEta(-0.8,0.8);
 166                                                 dtc1etaphitpc[aniter]->SetMass(ProtonMass);
 167                                                 dtc1etaphitpc[aniter]->SetMostProbableProton();
 168
 169
 170                                         }
 171                                         else if (ichg == 2) { // pap
 172                                                 dtc1etaphitpc[aniter]->SetCharge(-1.0);
 173                                                 dtc1etaphitpc[aniter]->SetPt(0.2,5.0);
 174                                                 dtc1etaphitpc[aniter]->SetEta(-0.8,0.8);
 175                                                 dtc1etaphitpc[aniter]->SetMass(ProtonMass);
 176                                                 dtc1etaphitpc[aniter]->SetMostProbableProton();
 177
 178                                                 dtc2etaphitpc[aniter]->SetCharge(1.0);
 179                                                 dtc2etaphitpc[aniter]->SetPt(0.5,5.0);
 180                                                 dtc2etaphitpc[aniter]->SetEta(-0.8,0.8);
 181                                                 dtc2etaphitpc[aniter]->SetMass(ProtonMass);
 182                                                 dtc2etaphitpc[aniter]->SetMostProbableProton();
 183                                         }
 184
 185                                         // Track quality cuts
 186
 187                                         if (runtype == 0) {
 188                                                 dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit|AliESDtrack::kITSrefit);
 189                                                 //          dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit);
 190                                                 //    dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kITSrefit);
 191                                                 dtc1etaphitpc[aniter]->SetminTPCncls(80);
 192                                                 dtc1etaphitpc[aniter]->SetRemoveKinks(kTRUE);
 193                                                 dtc1etaphitpc[aniter]->SetLabel(kFALSE);
 194                                                 //    dtc1etaphitpc[aniter]->SetMaxITSChiNdof(6.0);
 195                                                 dtc1etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
 196                                                 dtc1etaphitpc[aniter]->SetMaxImpactXY(0.2);
 197                                                 //            dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0182, 0.0350, -1.01);
 198                                                 dtc1etaphitpc[aniter]->SetMaxImpactZ(0.15);
 199                                                 //      dtc1etaphitpc[aniter]->SetMaxSigmaToVertex(6.0);
 200                                         }
 201                                         else if (runtype == 1) {
 202                                                 //      dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit|AliESDtrack::kITSrefit);
 203                                                 //    dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit);
 204                                                 //          dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kITSrefit|AliESDtrack::kITSpureSA);
 205                                                 //      dtc1etaphitpc[aniter]->SetminTPCncls(70);
 206                                                 dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kITSrefit);
 207                                                 dtc1etaphitpc[aniter]->SetRemoveKinks(kTRUE);
 208                                                 dtc1etaphitpc[aniter]->SetLabel(kFALSE);
 209                                                 //    dtc1etaphitpc[aniter]->SetMaxITSChiNdof(6.0);
 210                                                 //      dtc1etaphitpc[aniter]->SetMaxTPCChiNdof(6.0);
 211                                                 dtc1etaphitpc[aniter]->SetMaxImpactXY(0.2);
 212                                                 dtc1etaphitpc[aniter]->SetMaxImpactZ(0.25);
 213                                                 //      dtc1etaphitpc[aniter]->SetMaxSigmaToVertex(6.0);
 214                                         }
 215
 216                                         else if (runtype == 2) {
 217
 218                                                 dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCin);
 219                                                 dtc1etaphitpc[aniter]->SetminTPCncls(70);
 220                                                 dtc1etaphitpc[aniter]->SetRemoveKinks(kTRUE);
 221                                                 dtc1etaphitpc[aniter]->SetLabel(kFALSE);
 222                                                 dtc1etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
 223                                                 // dtc1etaphitpc[aniter]->SetMaxImpactXY(0.1); // 2.4 0.1
 224                                                 dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.018, 0.035, -1.01);        //      DCA xy
 225                                                 dtc1etaphitpc[aniter]->SetMaxImpactZ(2.0); // 2.0 0.1
 226
 227                                                 if (ichg == 2) {
 228
 229                                                         dtc2etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCin);
 230                                                         dtc2etaphitpc[aniter]->SetminTPCncls(70);
 231                                                         dtc2etaphitpc[aniter]->SetRemoveKinks(kTRUE);
 232                                                         dtc2etaphitpc[aniter]->SetLabel(kFALSE);
 233                                                         dtc2etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
 234                                                         // dtc2etaphitpc[aniter]->SetMaxImpactXY(0.1); // 2.4 0.1
 235                                                         dtc2etaphitpc[aniter]->SetMaxImpactXYPtDep(0.018, 0.035, -1.01);        //      DCA xy
 236                                                         dtc2etaphitpc[aniter]->SetMaxImpactZ(2.0); // 2.0 0.1
 237                                                 }
 238
 239                                         }
 240
 241                                         cutPass1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutPass1%stpcM%i", chrgs[ichg], imult),2);//0-pion,1-kaon,2-proton
 242                                         cutFail1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutFail1%stpcM%i", chrgs[ichg], imult),2);
 243                                         dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1PIDetaphitpc[aniter], cutFail1PIDetaphitpc[aniter]);
 244
 245
 246
 247                                         cutPass1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutPass1%stpcM%i", chrgs[ichg], imult),ProtonMass);
 248                                         cutFail1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutFail1%stpcM%i", chrgs[ichg], imult),ProtonMass);
 249                                         dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1YPtetaphitpc[aniter], cutFail1YPtetaphitpc[aniter]);
 250
 251
 252
 253                                         // sqpcetaphitpc[aniter] = new AliFemtoPairCutAntiGamma();
 254                                         sqpcetaphitpc[aniter] = new AliFemtoPairCutRadialDistance();
 255
 256                                         if (runtype == 0) {
 257                                                 sqpcetaphitpc[aniter]->SetShareQualityMax(1.0);
 258                                                 sqpcetaphitpc[aniter]->SetShareFractionMax(0.05);
 259                                                 sqpcetaphitpc[aniter]->SetRemoveSameLabel(kFALSE);
 260                                                 // sqpcetaphitpc[aniter]->SetMaxEEMinv(0.0);
 261                                                 // sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.0);
 262                                                 //          sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(1.5);
 263                                                 //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(0.12, 0.03);
 264                                                 //          sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
 265                                         }
 266                                         else if (runtype == 1) {
 267                                                 sqpcetaphitpc[aniter]->SetShareQualityMax(1.0);
 268                                                 sqpcetaphitpc[aniter]->SetShareFractionMax(1.05);
 269                                                 sqpcetaphitpc[aniter]->SetRemoveSameLabel(kFALSE);
 270                                                 // sqpcetaphitpc[aniter]->SetMaxEEMinv(0.002);
 271                                                 // sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.008);
 272                                                 //          sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(5.0);
 273                                                 //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(1.2, 0.03);
 274                                                 //          sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
 275                                         }
 276                                         else if (runtype == 2) {
 277
 278                                                 sqpcetaphitpc[aniter]->SetShareQualityMax(1.0);
 279                                                 sqpcetaphitpc[aniter]->SetShareFractionMax(0.05);
 280                                                 sqpcetaphitpc[aniter]->SetRemoveSameLabel(kFALSE);
 281
 282                                                 //          if (gammacut == 0) {
 283                                                 //sqpcetaphitpc[aniter]->SetMaxEEMinv(0.0);
 284                                                 //sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.0);
 285                                                 //}
 286                                                 //else if (gammacut == 1) {
 287                                                 //sqpcetaphitpc[aniter]->SetMaxEEMinv(0.002);
 288                                                 //sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.008);
 289                                                 //}
 290
 291                                                 sqpcetaphitpc[aniter]->SetMagneticFieldSign(1); // field1 -1, field3 +1
 292                                                 sqpcetaphitpc[aniter]->SetPhiStarDifferenceMinimum(0.012); // 0.012 - pions, 0.017 - kaons, 0.018
 293                                                 sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.015); // 0.017 - pions, 0.015 - kaons
 294
 295                                         }
 296
 297                                         anetaphitpc[aniter]->SetEventCut(mecetaphitpc[aniter]);
 298
 299                                         if (ichg == 2) {
 300                                                 anetaphitpc[aniter]->SetFirstParticleCut(dtc1etaphitpc[aniter]);
 301                                                 anetaphitpc[aniter]->SetSecondParticleCut(dtc2etaphitpc[aniter]);
 302                                         }
 303                                         else {
 304                                                 anetaphitpc[aniter]->SetFirstParticleCut(dtc1etaphitpc[aniter]);
 305                                                 anetaphitpc[aniter]->SetSecondParticleCut(dtc1etaphitpc[aniter]);
 306                                         }
 307
 308                                         anetaphitpc[aniter]->SetPairCut(sqpcetaphitpc[aniter]);
 309
 310
 311                                         if (ichg == 2) {
 312                                                 ckstartpc[aniter] = new AliFemtoCorrFctnNonIdDR(Form("ckstar%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
 313                                                 anetaphitpc[aniter]->AddCorrFctn(ckstartpc[aniter]);
 314                                         }
 315                                         else {
 316                                                 cqinvtpc[aniter] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
 317                                                 anetaphitpc[aniter]->AddCorrFctn(cqinvtpc[aniter]);
 318                                         }
 319
 320                                         cylmkttpc[aniter] = new AliFemtoCorrFctnDirectYlm(Form("cylm%stpcM%i", chrgs[ichg], imult),2,nbinssh, 0.0,shqmax,runshlcms);
 321                                         anetaphitpc[aniter]->AddCorrFctn(cylmkttpc[aniter]);
 322
 323
 324                                         if (runktdep) {
 325                                                 int ktm;
 326                                                 for (int ikt=0; ikt<numOfkTbins; ikt++) {
 327
 328                                                         ktm = aniter * numOfkTbins + ikt;
 329                                                         ktpcuts[ktm] = new AliFemtoKTPairCut(ktrng[ikt], ktrng[ikt+1]);
 330
 331                                                         cylmkttpc[ktm] = new AliFemtoCorrFctnDirectYlm(Form("cylm%stpcM%ikT%i", chrgs[ichg], imult, ikt),2,nbinssh,0.0,shqmax,0);
 332                                                         cylmkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 333                                                         anetaphitpc[aniter]->AddCorrFctn(cylmkttpc[ktm]);
 334
 335                                                         if (ichg == 2) {
 336                                                                 ckstarkttpc[ktm] = new AliFemtoCorrFctnNonIdDR(Form("ckstar%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
 337                                                                 ckstarkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 338                                                                 anetaphitpc[aniter]->AddCorrFctn(ckstarkttpc[ktm]);
 339
 340
 341                                                         }
 342                                                         else {
 343                                                                 cqinvkttpc[ktm] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
 344                                                                 cqinvkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 345                                                                 anetaphitpc[aniter]->AddCorrFctn(cqinvkttpc[ktm]);
 346                                                         }
 347
 348                                                 }
 349                                         }
 350
 351
 352                                         Manager->AddAnalysis(anetaphitpc[aniter]);
 353                                 }
 354                         }
 355                 }
 356         }
 357         // *** End  analysis
 358
 359         return Manager;
 360 }