[u/mrichter/AliRoot.git] / PWGCF / FEMTOSCOPY / macros / Train / MC_KK_pPb_V0M / ConfigFemtoAnalysis.C


/*********************************************************************
 *                                                                   *
 * ConfigFemtoAnalysis.C - configuration macro for the femtoscopic   *
 * analysis, meant as a QA process for two-particle effects          *
 *                                                                   *
 * Author: Adam Kisiel (Adam.Kisiel@cern.ch)                         *
 *                                                                   *
 *********************************************************************/

#if !defined(__CINT__) || defined(__MAKECINT_)
#include "AliFemtoManager.h"
#include "AliFemtoEventReaderESDChain.h"
#include "AliFemtoEventReaderESDChainKine.h"
#include "AliFemtoEventReaderAODChain.h"
#include "AliFemtoSimpleAnalysis.h"
#include "AliFemtoBasicEventCut.h"
#include "AliFemtoESDTrackCut.h"
#include "AliFemtoKKTrackCut.h"
#include "AliFemtoCorrFctn.h"
#include "AliFemtoCutMonitorParticleYPt.h"
#include "AliFemtoCutMonitorParticleVertPos.h"
#include "AliFemtoCutMonitorParticleMomRes.h"
#include "AliFemtoCutMonitorParticlePID.h"
#include "AliFemtoCutMonitorEventMult.h"
#include "AliFemtoCutMonitorEventVertex.h"
#include "AliFemtoShareQualityTPCEntranceSepPairCut.h"
#include "AliFemtoPairCutAntiGamma.h"
#include "AliFemtoPairCutRadialDistance.h"
#include "AliFemtoQinvCorrFctn.h"
#include "AliFemtoShareQualityCorrFctn.h"
#include "AliFemtoTPCInnerCorrFctn.h"
#include "AliFemtoVertexMultAnalysis.h"
#include "AliFemtoCorrFctn3DSpherical.h"
#include "AliFemtoChi2CorrFctn.h"
#include "AliFemtoCorrFctnTPCNcls.h"
#include "AliFemtoBPLCMS3DCorrFctn.h"
#include "AliFemtoCorrFctn3DLCMSSym.h"
#include "AliFemtoModelBPLCMSCorrFctn.h"
#include "AliFemtoModelCorrFctn3DSpherical.h"
#include "AliFemtoModelGausLCMSFreezeOutGenerator.h"
#include "AliFemtoModelGausRinvFreezeOutGenerator.h"
#include "AliFemtoModelManager.h"
#include "AliFemtoModelWeightGeneratorBasic.h"
#include "AliFemtoModelWeightGeneratorLednicky.h"
#include "AliFemtoCorrFctnDirectYlm.h"
#include "AliFemtoModelCorrFctnDirectYlm.h"
#include "AliFemtoModelCorrFctnSource.h"
#include "AliFemtoCutMonitorParticlePtPDG.h"
#include "AliFemtoKTPairCut.h"
#include "AliFemtoCutMonitorCollections.h"
#endif

//________________________________________________________________________
AliFemtoManager* ConfigFemtoAnalysis() {

  double PionMass = 0.13956995;
  double KaonMass = 0.493677;
	
  //multiplicity bins
 // int runmults[10] = {1, 1, 1, 1, 0, 0, 0, 0, 0, 0};
 // int multbins[11] = {0.01, 200, 400, 600, 900, 950, 500, 600, 700, 800, 900};
   int runmults[3] = {1, 1, 1};
    int multbins[4] = {0.01, 200, 400, 900};

  // int runmults[6] = {1, 1, 1, 1, 1, 1};
  // int multbins[7] = {0.01, 50, 100, 200, 400, 600, 900};
  // int runmults[4] = {1, 1, 1, 1};
  // int multbins[5] = {0.01, 200, 400, 600, 900};

  int runch[2] = {1, 1};
  // const char *chrgs[2] = { "pip", "pim" };
  const char *chrgs[2] = { "Kp", "Km"};

  int runktdep = 1;
  //double ktrng[8] = {0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 1.0};
  double ktrng[3] = {0.2, 0.5, 1.0};

  int run3d = 0; // Do 3D cartesian analysis?
  //int runshlcms = 1;
  int runshlcms = 0;

  //PhysicsSelection set only in runBatch with trigger + Physics Selection Task


  double shqmax;
  //int nbinssh = 200;
  int nbinssh = 100;

  //if (runshlcms) shqmax = 2.0;
  if (runshlcms) shqmax = 0.25;
  else shqmax = 2.0;

  AliFemtoEventReaderAODChain *Reader = new AliFemtoEventReaderAODChain();
<<<<<<< HEAD
  Reader->SetUseMultiplicity(AliFemtoEventReaderAODChain::kCentralityV0);
=======
  //V0M estimator:
  Reader->SetUseMultiplicity(AliFemtoEventReaderAODChain::kCentrality);
>>>>>>> new macros for Kaon Femto in pPb
  Reader->SetFilterBit(7);
  Reader->SetpA2013(kTRUE);
  //Reader->SetUseMultiplicity(AliFemtoEventReaderAODChain::kCentralityV0A);
 // Reader->SetCentralityPreSelection(0.01, 900);

  //AliFemtoEventReaderESDChainKine* Reader=new AliFemtoEventReaderESDChainKine();
  //Reader->SetConstrained(true);
  // Reader->SetUseTPCOnly(false);

 // AliFemtoEventReaderESDChain *Reader = new AliFemtoEventReaderESDChain();
 // Reader->SetUseMultiplicity(AliFemtoEventReaderESDChain::kV0Centrality);
 // Reader->SetReadTrackType(AliFemtoEventReaderESDChain::kGlobal);


  AliFemtoManager* Manager=new AliFemtoManager();
  Manager->SetEventReader(Reader);

  AliFemtoVertexMultAnalysis    *anetaphitpc[20];
  AliFemtoBasicEventCut         *mecetaphitpc[20];
  AliFemtoCutMonitorEventMult   *cutPassEvMetaphitpc[20];
  AliFemtoCutMonitorEventMult   *cutFailEvMetaphitpc[20];
  AliFemtoCutMonitorEventVertex *cutPassEvVetaphitpc[20];
  AliFemtoCutMonitorEventVertex *cutFailEvVetaphitpc[20];
  AliFemtoCutMonitorCollections   *cutPassColletaphitpc[20];
  AliFemtoCutMonitorCollections   *cutFailColletaphitpc[20];
  AliFemtoKKTrackCut           *dtc1etaphitpc[20];
  AliFemtoKKTrackCut           *dtc2etaphitpc[20];
  //AliFemtoESDTrackCut           *dtc1etaphitpc[20];
  //AliFemtoESDTrackCut           *dtc2etaphitpc[20];
  AliFemtoCutMonitorParticleYPt *cutPass1YPtetaphitpc[20];
  AliFemtoCutMonitorParticleYPt *cutFail1YPtetaphitpc[20];
  AliFemtoCutMonitorParticlePID *cutPass1PIDetaphitpc[20];
  AliFemtoCutMonitorParticlePID *cutFail1PIDetaphitpc[20];
  AliFemtoCutMonitorParticleYPt *cutPass2YPtetaphitpc[20];
  AliFemtoCutMonitorParticleYPt *cutFail2YPtetaphitpc[20];
  AliFemtoCutMonitorParticlePID *cutPass2PIDetaphitpc[20];
  AliFemtoCutMonitorParticlePID *cutFail2PIDetaphitpc[20];
  //  AliFemtoPairCutAntiGamma      *sqpcetaphitpc[20];
  AliFemtoShareQualityPairCut      *sqpcetaphitpc[20];
  //AliFemtoPairCutRadialDistance      *sqpcetaphitpc[20];
  AliFemtoCorrFctnDirectYlm     *cylmetaphitpc[20];
  AliFemtoCorrFctnDEtaDPhi      *cdedpetaphi[20];
  AliFemtoChi2CorrFctn          *cchiqinvetaphitpc[20];
  //AliFemtoKTPairCut             *ktpcuts[20*7];
  //AliFemtoCorrFctnDirectYlm     *cylmkttpc[20*7];
  //AliFemtoQinvCorrFctn          *cqinvkttpc[20*7];
  //AliFemtoCorrFctn3DLCMSSym     *cq3dlcmskttpc[20*7];
  AliFemtoKTPairCut             *ktpcuts[20*8];
  AliFemtoCorrFctnDirectYlm     *cylmkttpc[20*8];
  AliFemtoQinvCorrFctn          *cqinvkttpc[20*8];
  AliFemtoCorrFctn3DLCMSSym     *cq3dlcmskttpc[20*8];
  AliFemtoCorrFctnTPCNcls       *cqinvnclstpc[20];
  AliFemtoShareQualityCorrFctn  *cqinvsqtpc[20*10];
  AliFemtoChi2CorrFctn          *cqinvchi2tpc[20];
  AliFemtoTPCInnerCorrFctn      *cqinvinnertpc[20*10];

  // *** Begin pion-pion analysis ***
  int aniter = 0;

  for (int imult=0; imult<3; imult++) {
    if (runmults[imult]) {
      for (int ichg=0; ichg<2; ichg++) {
	if (runch[ichg]) {
	  aniter = ichg*3+imult;

	  anetaphitpc[aniter] = new AliFemtoVertexMultAnalysis(10, -10.0, 10.0, 4, multbins[imult], multbins[imult+1]);
	  anetaphitpc[aniter]->SetNumEventsToMix(5);
	  anetaphitpc[aniter]->SetMinSizePartCollection(1);
	  anetaphitpc[aniter]->SetVerboseMode(kTRUE);
      
	  mecetaphitpc[aniter] = new AliFemtoBasicEventCut();
	  mecetaphitpc[aniter]->SetEventMult(0,10000);
	  mecetaphitpc[aniter]->SetVertZPos(-10,10);

	  cutPassEvMetaphitpc[aniter] = new AliFemtoCutMonitorEventMult(Form("cutPass%stpcM%i", chrgs[ichg], imult));
	  cutFailEvMetaphitpc[aniter] = new AliFemtoCutMonitorEventMult(Form("cutFail%stpcM%i", chrgs[ichg], imult));
	  mecetaphitpc[aniter]->AddCutMonitor(cutPassEvMetaphitpc[aniter], cutFailEvMetaphitpc[aniter]);
	  
	  cutPassEvVetaphitpc[aniter] = new AliFemtoCutMonitorEventVertex(Form("cutPass%stpcM%i", chrgs[ichg], imult));
	  cutFailEvVetaphitpc[aniter] = new AliFemtoCutMonitorEventVertex(Form("cutFail%stpcM%i", chrgs[ichg], imult));
	  mecetaphitpc[aniter]->AddCutMonitor(cutPassEvVetaphitpc[aniter], cutFailEvVetaphitpc[aniter]);
	  
	  cutPassColletaphitpc[aniter] = new AliFemtoCutMonitorCollections(Form("cutPass%stpcM%i", chrgs[ichg], imult));
          cutFailColletaphitpc[aniter] = new AliFemtoCutMonitorCollections(Form("cutFail%stpcM%i", chrgs[ichg], imult));
          mecetaphitpc[aniter]->AddCutMonitor(cutPassColletaphitpc[aniter], cutFailColletaphitpc[aniter]);

	 // dtc1etaphitpc[aniter] = new AliFemtoESDTrackCut();
	 dtc1etaphitpc[aniter] = new AliFemtoKKTrackCut();

	  if (ichg == 0)
	    dtc1etaphitpc[aniter]->SetCharge(1.0);
	  else if (ichg == 1)
	    dtc1etaphitpc[aniter]->SetCharge(-1.0);

	  //  dtc1etaphitpc[aniter]->SetPt(0.12,4.0);
	  // dtc1etaphitpc[aniter]->SetEta(-1.2,1.2);
	   dtc1etaphitpc[aniter]->SetPt(0.14,1.5);
	   dtc1etaphitpc[aniter]->SetEta(-0.8,0.8);

	  //PID method
	   // dtc1etaphitpc[aniter]->SetMass(PionMass);
	   // dtc1etaphitpc[aniter]->SetMostProbablePion();
	   dtc1etaphitpc[aniter]->SetMass(KaonMass);
	   dtc1etaphitpc[aniter]->SetMostProbableKaon();
	  //dtc1etaphitpc[aniter]->SetPIDMethod(AliFemtoESDTrackCut::kContour);
//------------------- November 2013 -----------------------------------< 
	  // new cuts to remove electron (do not take into analysis if 400<p<500) 
	 dtc1etaphitpc[aniter]->SetNsigmaTPCle250(2.0);
	 dtc1etaphitpc[aniter]->SetNsigmaTPC250_400(2.0);
	  dtc1etaphitpc[aniter]->SetNsigmaTPC400_450(2.0);
	  dtc1etaphitpc[aniter]->SetNsigmaTPC450_500(2.0);
	  dtc1etaphitpc[aniter]->SetNsigmaTPCge500(3.0);    
	  // new cuts are stronger, better separation of pion in TOF 
	  // when momentum is greater then 800 MeV/c
	  dtc1etaphitpc[aniter]->SetNsigmaTOF500_800(2.0);
	  dtc1etaphitpc[aniter]->SetNsigmaTOF800_1000(1.5);
	  dtc1etaphitpc[aniter]->SetNsigmaTOFge1000(1.0);
	  //------------------- November 2013 ----------------------------------->
	  //Track quality cuts
	 dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit|AliESDtrack::kITSrefit);
	  dtc1etaphitpc[aniter]->SetminTPCncls(50);
	  dtc1etaphitpc[aniter]->SetRemoveKinks(kTRUE);


	  dtc1etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
	  dtc1etaphitpc[aniter]->SetMaxITSChiNdof(36);	  
	  dtc1etaphitpc[aniter]->SetLabel(kFALSE);
	  
	  //primary particles: hits in ITS + DCA cut
	  //dtc1etaphitpc[aniter]->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
	  //				 AliESDtrackCuts::kAny);
	  dtc1etaphitpc[aniter]->SetMaxImpactZ(2.0);
	  dtc1etaphitpc[aniter]->SetMaxImpactXY(2.4);
	  //dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0105, 0.0350, -1.1);
	  //dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0182, 0.0350, -1.01);
	  //dtc1etaphitpc[aniter]->SetMaxSigmaToVertex(6.0);

	  //cutPass1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutPass1%stpcM%i", chrgs[ichg], imult), 0.13957);
	  //cutFail1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutFail1%stpcM%i", chrgs[ichg], imult), 0.13957);
	  cutPass1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutPass1%stpcM%i", chrgs[ichg], imult), 0.493677);
	  cutFail1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutFail1%stpcM%i", chrgs[ichg], imult), 0.493677);
	  dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1YPtetaphitpc[aniter], cutFail1YPtetaphitpc[aniter]);
	  
	  //cutPass1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutPass1%stpcM%i", chrgs[ichg], imult),0);
	  //cutFail1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutFail1%stpcM%i", chrgs[ichg], imult),0);
	  cutPass1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutPass1%stpcM%i", chrgs[ichg], imult),1);
	  cutFail1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutFail1%stpcM%i", chrgs[ichg], imult),1);
	  dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1PIDetaphitpc[aniter], cutFail1PIDetaphitpc[aniter]);
	  
      dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCin); ///my

      
	  //sqpcetaphitpc[aniter] = new AliFemtoPairCutAntiGamma();
	  sqpcetaphitpc[aniter] = new AliFemtoShareQualityPairCut();
	  //sqpcetaphitpc[aniter] = new AliFemtoShareQualityTPCEntranceSepPairCut();
	  //sqpcetaphitpc[aniter] = new AliFemtoPairCutRadialDistance();
	  sqpcetaphitpc[aniter]->SetShareQualityMax(1.0);
	  sqpcetaphitpc[aniter]->SetShareFractionMax(0.05);
	  sqpcetaphitpc[aniter]->SetRemoveSameLabel(kFALSE);
	  //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.016);
	  //sqpcetaphitpc[aniter]->SetPhiStarDifferenceMinimum(0.02);
	  //runtype==0
	  // sqpcetaphitpc[aniter]->SetMaxEEMinv(0.0);
	  // sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.0);
	  // sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(1.5);
	  // sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
	  // sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(0.12, 0.03);
	  //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
	  //runtype==1
	  //	    sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(5.0);
	  //	    sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
	  //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(1.2, 0.03);
	  //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
	  //runtype==2
	  //	    sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(1.0);
	  //	    sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
	  //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(1.2, 0.045);
	  //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.016);
	  //sqpcetaphitpc[aniter]->SetPhiStarDifferenceMinimum(0.02);
	  
	  anetaphitpc[aniter]->SetEventCut(mecetaphitpc[aniter]);
	  anetaphitpc[aniter]->SetFirstParticleCut(dtc1etaphitpc[aniter]);
	  anetaphitpc[aniter]->SetSecondParticleCut(dtc1etaphitpc[aniter]);
	  anetaphitpc[aniter]->SetPairCut(sqpcetaphitpc[aniter]);
	  
	  //Correlation functions

	  //Spherical harmonics (without kT bins)
	  //cylmetaphitpc[aniter] = new AliFemtoCorrFctnDirectYlm(Form("cylm%stpcM%i", chrgs[ichg], imult),3,nbinssh,0.0,shqmax,runshlcms);
	  //anetaphitpc[aniter]->AddCorrFctn(cylmetaphitpc[aniter]);

	  //Qinv (without kT bins)
	  cqinvkttpc[aniter] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
	  anetaphitpc[aniter]->AddCorrFctn(cqinvkttpc[aniter]);

	  //3D cartesian (without kT bins)
	  if(run3d){
	    //cq3dlcmskttpc[aniter] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%i", chrgs[ichg], imult),60,0.5);
	    cq3dlcmskttpc[aniter] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%i", chrgs[ichg], imult),100,0.5);
	    anetaphitpc[aniter]->AddCorrFctn(cq3dlcmskttpc[aniter]);
	  }
	  
	  // cqinvnclstpc[aniter] = new AliFemtoCorrFctnTPCNcls(Form("cqinvncls%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
	  // anetaphitpc[aniter]->AddCorrFctn(cqinvnclstpc[aniter]);

	  // cqinvchi2tpc[aniter] = new AliFemtoChi2CorrFctn(Form("cqinvchi2%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
	  // anetaphitpc[aniter]->AddCorrFctn(cqinvchi2tpc[aniter]);

	  if (runktdep) {
	    int ktm;
	    for (int ikt=0; ikt<2; ikt++) {
	      ktm = aniter*2 + ikt;
	      ktpcuts[ktm] = new AliFemtoKTPairCut(ktrng[ikt], ktrng[ikt+1]);
	      
  	      //cylmkttpc[ktm] = new AliFemtoCorrFctnDirectYlm(Form("cylm%stpcM%ikT%i", chrgs[ichg], imult, ikt),3,
	      //							     nbinssh, 0.0, shqmax, runshlcms);
	    //cylmkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
	    //anetaphitpc[aniter]->AddCorrFctn(cylmkttpc[ktm]);
	      
	      cqinvkttpc[ktm] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0, shqmax);
	      cqinvkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
	      anetaphitpc[aniter]->AddCorrFctn(cqinvkttpc[ktm]);

	      cqinvsqtpc[ktm] = new AliFemtoShareQualityCorrFctn(Form("cqinvsq%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
	      cqinvsqtpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
	      anetaphitpc[aniter]->AddCorrFctn(cqinvsqtpc[ktm]);

	      cqinvinnertpc[ktm] = new AliFemtoTPCInnerCorrFctn(Form("cqinvinner%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
	      cqinvinnertpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
	      cqinvinnertpc[ktm]->SetRadius(1.2);
	      anetaphitpc[aniter]->AddCorrFctn(cqinvinnertpc[ktm]);

	      if (run3d) {
		//		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);
		cq3dlcmskttpc[ktm] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%ikT%i", chrgs[ichg], imult, ikt),60,0.5);
		cq3dlcmskttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
		anetaphitpc[aniter]->AddCorrFctn(cq3dlcmskttpc[ktm]);
	      }
	    }
	  }
	  
	  //cdedpetaphi[aniter] = new AliFemtoCorrFctnDEtaDPhi(Form("cdedp%stpcM%i", chrgs[ichg], imult),39, 39);
	  //anetaphitpc[aniter]->AddCorrFctn(cdedpetaphi[aniter]);
	  
	  Manager->AddAnalysis(anetaphitpc[aniter]);	
	}
      }
    }
  }
  // *** End pion-pion analysis

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