Updates in the spectra task (Marek)

[u/mrichter/AliRoot.git] / PWG2 / SPECTRA / AliAnalysisChargedHadronSpectraITSTruncatedMeanTask.h

//Class to extract data to do ITS+TPC global Spectra\r
//Autor Marek Chojnacki\r
//emali Marek.Chojnacki@cern.ch\r
\r
\r
\r
#ifndef ALIANALYSISCHARGEDHADRONSPECTRAITSTRUNCATEDMEANTASK_H\r
#define  ALIANALYSISCHARGEDHADRONSPECTRAITSTRUNCATEDMEANTASK_H\r
//#include <fstream>\r
class TH1F;\r
class TH2F;\r
class TH3F;\r
class AliESDEvent;\r
class AliESDtrack;\r
class  AliESDtrackCuts;\r
class AliESDpidCuts;\r
class AliESDpid;\r
class TGraph;\r
class AliStack;\r
#include "AliAnalysisTaskSE.h"\r
//#include "AliESDtrackCuts.h"\r
#include "AliESDpidCuts.h"\r
#include "AliESDpid.h"\r
\r
\r
\r
\r
class AliAnalysisChargedHadronSpectraITSTruncatedMeanTask : public AliAnalysisTaskSE {\r
 public:\r
  AliAnalysisChargedHadronSpectraITSTruncatedMeanTask(const char *name = "AliAnalysisChargedHadronSpectraITSTruncatedMeanTask");\r
  virtual ~AliAnalysisChargedHadronSpectraITSTruncatedMeanTask() {}\r
  \r
  //virtual void   ConnectInputData(Option_t *);\r
  virtual void   UserCreateOutputObjects();\r
  virtual void   UserExec(Option_t *option);\r
  virtual void   Terminate(Option_t *); \r
  virtual void   LocalInit();\r
  AliESDtrackCuts*  GetAliESDtrackCuts() const {return fCuts;}\r
  void SetMCOn(){fMC=kTRUE;\r
   fESDpid->GetTPCResponse().SetBetheBlochParameters(2.15898e+00/50.,1.75295e+01,3.40030e-09,1.96178e+00,3.91720e+00);}\r
  void SetAliESDtrackCuts(AliESDtrackCuts* const cuts ){fCuts=cuts;/*flist->Add(fCuts);*/}\r
  void SetFunctionParam(Double_t* const par);\r
   void SetMultiplicityCut(Int_t low, Int_t up);\r
  void SetCorrectSDD(){fCorrectSDD=kTRUE;}\r
   void SetCorrectSSD(){fCorrectSSD=kTRUE;}\r
  void SetYcut(Float_t value){fYCut=TMath::Abs(value);}\r
  void Setsigmacut(Float_t value){fsigmacut=TMath::Abs(value);}\r
  void SetNsigmaDCAcut(Float_t sigmaxy,Float_t sigmaz){fnsigmaxy=sigmaxy;fnsigmaz=sigmaz;}    \r
  void SetChargeCut(Float_t chargeCut){fchargeCut=TMath::Abs(chargeCut)>50.0?50.0:TMath::Abs(chargeCut);}\r
     void SetTPCPIDCUT(AliESDpidCuts* const cuts){fTPCPIDCUT=cuts;}\r
     void SetWeights(TGraph* const setK0weight, TGraph* const  setlambdaweight,TGraph* const setAntilambdaweight){fK0weight=setK0weight;flambdaweight=setlambdaweight;fAntilambdaweight=setAntilambdaweight;}\r
     void SetDCA2010();\r
     void SetHImode(){fHIsettings=kTRUE;}\r
     void SetCentralityCut(Float_t low, Float_t up); \r
     void SetDoVertexrescuts(){fdovertexrescuts=kTRUE;}\r
 private:\r
 \r
 \r
 AliESDEvent *fESD;    //ESD object    \r
  AliESDtrackCuts *fCuts;//cuts \r
  AliESDtrackCuts *fCutsMul;//cuts for multiplicty \r
  \r
  \r
  \r
  Bool_t fMC;//if TRUE use MC \r
  Int_t fLowMultiplicity;//low Multiplicity cut\r
  Int_t fUpMultiplicity;//up Multiplicity cut\r
  Float_t fLowCentrality;//low Centrality cut\r
  Float_t fUpCentrality;//up  Centrality cut\r
  \r
  \r
  Float_t fYCut;//cut in y\r
  Float_t fsigmacut;//cut in sigma in n-sigma method\r
  Float_t fnsigmaxy; //cut in sigma on xy dca\r
  Float_t fnsigmaz;//cut in sigma on Z dca\r
  Float_t fdcaxypar[3];//parameters for DCAxy cut \r
  Float_t fdcazpar[4];//parameters for DCAz cut \r
  \r
  Float_t fchargeCut;//cut for the low charges\r
  \r
  \r
 Bool_t fCorrectSDD;//In LHC10a3 in some runs dE in SDD had to scaled to SSD flag if this should be done \r
  Bool_t fCorrectSSD;//this same but for dE SSS\r
  \r
  Bool_t fHIsettings;//speciall settings fot HI mode\r
  Bool_t fdovertexrescuts;// check on Vmc-VESD \r
  \r
  \r
  \r
  TGraph* fK0weight ;//weight for pions comming from K0shorts\r
  TGraph* flambdaweight ;//weight for protons comming from lambdas\r
   TGraph* fAntilambdaweight ;//weight for antiprotons comming from antilambdas\r
  \r
   \r
   \r
   \r
TH1F *fHistStats; //histogram with statistic of events\r
TH1F* fHistZVertexBeforeCut; //Z of vertex before cut \r
TH1F* fHistZVertexAfterCut; //Z of vertex after cut\r
TH2F* fHistXYVertexBeforeCut; //XY of vertex before cut \r
TH2F* fHistXYVertexAfterCut; //XY of vertex after cut\r
\r
\r
TH2F* fHistPhiPtBeforeCuts;//phi pt before cuts \r
TH2F* fHistPhiPtAfterCuts;//phi pt after cuts \r
TH2F* fHistEtaPtBeforeCuts;//eta pt before cuts \r
TH2F* fHistEtaPtAfterCuts;//eta pt after cuts \r
\r
TH2F* fHistDCABeforeCuts;//dca hist before cuts \r
TH2F* fHistDCAAfterCuts;//dca hist after cuts \r
\r
TH2F* fHistPminusTPCinPAfterCuts;//differnece between global momentum at primary vetrex and tpc standalone momentum at primry vertex \r
TH2F* fHistPminusTPCinPglobalAfterCuts;//differnece between global momentum at primary vetrex and global momentum at  the inner  wall of the TPC taken from global tracking\r
\r
//positive\r
TH2F* fHistMydEPpositive;//dE in its as function of global p at p.v.\r
TH2F* fHistMydETPCinPpositive;//dE in its as function of TPC p at p.v.\r
TH2F* fHistMydETPCinPglobalpositive;//dE in its as function of p  at  the inner  wall of the TPC taken from global tracking\r
//negative\r
TH2F* fHistMydEPnegative;//dE in its as function of global p at p.v.\r
TH2F* fHistMydETPCinPnegative;//dE in its as function of TPC p at p.v.\r
TH2F* fHistMydETPCinPglobalnegative;//dE in its as function of p  at  the inner  wall of the TPC taken from global tracking\r
\r
//dE   as function of global p at p.v.\r
TH2F* fHistL3dEP;// SDD1 \r
TH2F* fHistL4dEP;// SDD2\r
TH2F* fHistL5dEP;//SSD1\r
TH2F* fHistL6dEP; //SSD2\r
\r
//dE in its as function of TPC p at p.v.\r
TH2F* fHistL3dETPCinP;// SDD1 \r
TH2F* fHistL4dETPCinP;// SDD2\r
TH2F* fHistL5dETPCinP;//SSD1\r
TH2F* fHistL6dETPCinP;//SSD2\r
\r
\r
\r
\r
TH2F* fHistwhichhasmin;// ITS  layer with minimal charged\r
TH1F* fHistMysignalminusESD;// My signal minus ESD\r
\r
\r
//log dE-logdEfit as function of  global p at p.v. for\r
TH2F* fHistminsignalifPionP;//pions\r
TH2F* fHistminsignalifKaonP;//kaons\r
TH2F* fHistminsignalifProtonP;//protons\r
\r
TH2F* fHistminsignalifAntiPionP;//antipions\r
TH2F* fHistminsignalifAntiKaonP;//antikaons\r
TH2F* fHistminsignalifAntiProtonP;//antiprotons\r
\r
\r
//DCA histograms for clean particles with after dca cut\r
TH3F* fDCAXYZforcleanPions;//pions\r
TH3F* fDCAXYZforcleanAntiPions;//antipion\r
TH3F* fDCAXYZforcleanProtons;//kaons\r
TH3F* fDCAXYZforcleanAntiProtons;//antikaons\r
\r
//DCA histograms for clean particles with before dca cut\r
TH3F* fDCAXYZOpenforcleanPions;//pions\r
TH3F* fDCAXYZOpenforcleanAntiPions;//antipions\r
TH3F* fDCAXYZOpenforcleanProtons;//kaons\r
TH3F* fDCAXYZOpenforcleanAntiProtons;//antikaons\r
\r
//pt distibution of track fullfilling some cuts 0-pions, 1-kaons,2-protons\r
TH2F* fHistNtrackwithstandardcuts;//TPC cuts\r
TH2F* fHistNtrackwithITSPIDcuts;//TPC cuts + ITS pid cuts\r
\r
TH2F* fHistSignalinTPCKaonforstandardcuts;//TPC signal for Kaons tpc cuts tracks\r
TH2F* fHistSignalinTPCKaonforITSPIDcuts;//TPC signal for Kaons tpc+itspid cuts tracks\r
\r
TH2F* fHistSignalinTPCAntiKaonforstandardcuts;//TPC signal for AntiKaons tpc cuts tracks\r
TH2F* fHistSignalinTPCAntiKaonforITSPIDcuts; //TPC signal for AntiKaons tpc+itspid cuts tracks\r
\r
\r
TH2F* fHistSignalinTPCProtonforstandardcuts; //TPC signal for Protons tpc cuts tracks\r
TH2F* fHistSignalinTPCProtonforITSPIDcuts;//TPC signal for Protons tpc+itspid cuts tracks\r
\r
TH2F* fHistSignalinTPCAntiProtonforstandardcuts;//TPC signal for AntiProtons tpc cuts tracks\r
TH2F* fHistSignalinTPCAntiProtonforITSPIDcuts; //TPC signal for AntiProtons tpc+itspid cuts tracks\r
\r
\r
//Multiplicity histos\r
TH1F* fHistStandartMul;//number from AliESDtrackCuts::GetReferenceMultiplicity\r
TH1F* fHistMytrackMul;//number of my tracks\r
\r
\r
//log dE-logdEfit as function of  global p at p.v. for primary tracks \r
TH2F* fHistminsignalifPionPPrimary; //pions\r
TH2F* fHistminsignalifKaonPPrimary;//kaons\r
TH2F* fHistminsignalifProtonPPrimary;//protons\r
TH2F* fHistminsignalifProtonPPrimaryfake;//fake protons\r
\r
TH2F* fHistminsignalifAntiPionPPrimary;//antipions\r
TH2F* fHistminsignalifAntiKaonPPrimary;//antikaons\r
TH2F* fHistminsignalifAntiProtonPPrimary;//antiprotons\r
TH2F* fHistminsignalifAntiProtonPPrimaryfake;//antiprotonsfake\r
\r
\r
//log dE-logdEfit as function of  global p at p.v. for secondary and other tracks tracks \r
TH2F* fHistminsignalifPionPSecondary;//pions\r
TH2F* fHistminsignalifKaonPSecondary;//kaon\r
TH2F* fHistminsignalifProtonPSecondaryWD;//protons comming from weak decays \r
TH2F* fHistminsignalifProtonPSecondaryHI;//protons comming from material \r
TH2F* fHistminsignalifProtonPSecondaryRest;//rest contamination \r
\r
TH2F* fHistminsignalifProtonPSecondaryWDfake;//protons fakes comming from weak decays \r
TH2F* fHistminsignalifProtonPSecondaryHIfake;//protons fakes comming from material \r
\r
TH2F* fHistminsignalifAntiPionPSecondary;//antipions\r
TH2F* fHistminsignalifAntiKaonPSecondary;//antikaon\r
\r
TH2F* fHistminsignalifAntiProtonPSecondaryWD;//antiprotons comming from weak decays\r
TH2F* fHistminsignalifAntiProtonPSecondaryHI;//antiprotons fakes comming from material \r
TH2F* fHistminsignalifAntiProtonPSecondaryRest;//rest contamination \r
\r
TH2F* fHistminsignalifAntiProtonPSecondaryWDfake;//antiprotons fakes comming from weak decays \r
TH2F* fHistminsignalifAntiProtonPSecondaryHIfake;//antiprotons fakes comming from material \r
 \r
\r
TH2F* fHistminsignalifMuEPositiveP;//mu+  positronium for pions  \r
TH2F* fHistminsignalifMuENegativeP;//mu-  electrons for antipions \r
\r
 TH2F* fHistminsignalifPionPrimaryfake;//fake pions primary\r
  TH2F* fHistminsignalifKaonPrimaryfake;//fake kaons primary\r
 \r
 TH2F* fHistminsignalifAntiPionPrimaryfake;//fake antipions primary\r
 TH2F* fHistminsignalifAntiKaonPrimaryfake;//fake antikaons primary\r
 \r
 \r
  TH2F* fHistminsignalifPionSecondaryfake;//fake pions\r
  TH2F* fHistminsignalifKaonSecondaryfake;//fake kaons\r
 \r
 TH2F* fHistminsignalifAntiPionSecondaryfake;//fake antipions\r
 TH2F* fHistminsignalifAntiKaonSecondaryfake;//fake antikaons\r
 \r
\r
 //MC particles from Events passing ESD event cuts only pt\r
TH1F* fHistminsignalifPionPMCPrimary;//Pions \r
TH1F* fHistminsignalifKaonPMCPrimary;//Kaons\r
TH1F* fHistminsignalifProtonPMCPrimary;//Protons\r
\r
TH1F* fHistminsignalifAntiPionPMCPrimary;//AntiPions\r
TH1F* fHistminsignalifAntiKaonPMCPrimary;//AntiKaons\r
TH1F* fHistminsignalifAntiProtonPMCPrimary;//AntiProtons\r
\r
\r
//MC particles from all MC events \r
TH1F* fHistminsignalifPionPMCPrimaryBeforeEventCuts;//Pions\r
TH1F* fHistminsignalifKaonPMCPrimaryBeforeEventCuts;//Kaons\r
TH1F* fHistminsignalifProtonPMCPrimaryBeforeEventCuts;//Protons\r
\r
TH1F* fHistminsignalifAntiPionPMCPrimaryBeforeEventCuts;//AntiPions\r
TH1F* fHistminsignalifAntiKaonPMCPrimaryBeforeEventCuts;//AntiKaons\r
TH1F* fHistminsignalifAntiProtonPMCPrimaryBeforeEventCuts;//AntiProtons\r
\r
//MC particles from all MC events  with good vertex in z \r
TH1F* fHistminsignalifPionPMCPrimaryBeforeEventCutswithgoodZvertex;//Pions\r
TH1F* fHistminsignalifKaonPMCPrimaryBeforeEventCutswithgoodZvertex;//Kaons\r
TH1F* fHistminsignalifProtonPMCPrimaryBeforeEventCutswithgoodZvertex;//Protons\r
\r
TH1F* fHistminsignalifAntiPionPMCPrimaryBeforeEventCutswithgoodZvertex;//AntiPions\r
TH1F* fHistminsignalifAntiKaonPMCPrimaryBeforeEventCutswithgoodZvertex;//AntiKaons\r
TH1F* fHistminsignalifAntiProtonPMCPrimaryBeforeEventCutswithgoodZvertex;//AntiProtons\r
\r
\r
//MC particles from  MC events  which ESD event go trought physics selection and has vertex  \r
TH1F* fHistminsignalifPionPMCPrimaryAfterEventCutsBeforeVertexZ;//Pions\r
TH1F* fHistminsignalifKaonPMCPrimaryAfterEventCutsBeforeVertexZ;//Kaons\r
TH1F* fHistminsignalifProtonPMCPrimaryAfterEventCutsBeforeVertexZ;//Protons\r
\r
TH1F* fHistminsignalifAntiPionPMCPrimaryAfterEventCutsBeforeVertexZ;//AntiPions\r
TH1F* fHistminsignalifAntiKaonPMCPrimaryAfterEventCutsBeforeVertexZ;//AntiKaons\r
TH1F* fHistminsignalifAntiProtonPMCPrimaryAfterEventCutsBeforeVertexZ;//AntiProtons\r
\r
\r
//DCA distributions for different parctiles after dca cuts \r
\r
TH3F* fDCAXYZforcleanPionsMCPrimary;//primary pions \r
TH3F* fDCAXYZforcleanAntiPionsMCPrimary;//primary antipions \r
TH3F* fDCAXYZforcleanProtonsMCPrimary;//primary protons\r
TH3F* fDCAXYZforcleanAntiProtonsMCPrimary;//primary antiprotons\r
\r
//Secondrary Pions weak deacy\r
TH3F* fDCAXYZforcleanPionsWD;//pions \r
TH3F* fDCAXYZforcleanAntiPionsWD;// antipions \r
\r
//Secondrary Protons weak deacy + fakes\r
TH3F* fDCAXYZforcleanProtonsWD;//protons\r
TH3F* fDCAXYZforcleanAntiProtonsWD;//antiprotons\r
\r
//Secondrary Pions Hadronic\r
TH3F* fDCAXYZforcleanPionsHI;//pions\r
TH3F* fDCAXYZforcleanAntiPionsHI;//antipions\r
\r
//Secondrary Protons Hadronic+fakes\r
TH3F* fDCAXYZforcleanProtonsHI;//pions\r
TH3F* fDCAXYZforcleanAntiProtonsHI;//antipions\r
\r
//Secondrary Pions mu el\r
TH3F* fDCAXYZforcleanPionsMEPrimary;//posvitive\r
TH3F* fDCAXYZforcleanAntiPionsMEPrimary;//negative\r
TH3F* fDCAXYZforcleanPionsMESecondary;//posvitive\r
TH3F* fDCAXYZforcleanAntiPionsMESecondary;//negative\r
\r
//Secondrary Pions rest source\r
TH3F* fDCAXYZforcleanPionsR;//positive\r
TH3F* fDCAXYZforcleanAntiPionsR;//negative\r
\r
//Secondrary Protons rest\r
TH3F* fDCAXYZforcleanProtonsR;//positive\r
TH3F* fDCAXYZforcleanAntiProtonsR;//negative\r
\r
\r
//DCA distributions for different parctiles before dca cuts \r
\r
TH3F* fDCAXYZOpenforcleanPionsMCPrimary;//primary pions \r
TH3F* fDCAXYZOpenforcleanAntiPionsMCPrimary;//primary antipions \r
TH3F* fDCAXYZOpenforcleanProtonsMCPrimary;//primary protons\r
TH3F* fDCAXYZOpenforcleanAntiProtonsMCPrimary;//primary antiprotons\r
\r
//Secondrary Pions weak deacy\r
TH3F* fDCAXYZOpenforcleanPionsWD;//pions \r
TH3F* fDCAXYZOpenforcleanAntiPionsWD;// antipions \r
\r
//Secondrary Protons weak deacy + fakes\r
TH3F* fDCAXYZOpenforcleanProtonsWD;//protons\r
TH3F* fDCAXYZOpenforcleanAntiProtonsWD;//antiprotons\r
\r
//Secondrary Pions Hadronic\r
TH3F* fDCAXYZOpenforcleanPionsHI;//pions\r
TH3F* fDCAXYZOpenforcleanAntiPionsHI;//antipions\r
\r
//Secondrary Protons Hadronic+fakes\r
TH3F* fDCAXYZOpenforcleanProtonsHI;//pions\r
TH3F* fDCAXYZOpenforcleanAntiProtonsHI;//antipions\r
\r
//Secondrary Pions mu el\r
TH3F* fDCAXYZOpenforcleanPionsMEPrimary;//posvitive\r
TH3F* fDCAXYZOpenforcleanAntiPionsMEPrimary;//negative\r
TH3F* fDCAXYZOpenforcleanPionsMESecondary;//posvitive\r
TH3F* fDCAXYZOpenforcleanAntiPionsMESecondary;//negative\r
\r
//Secondrary Pions rest source\r
TH3F* fDCAXYZOpenforcleanPionsR;//positive\r
TH3F* fDCAXYZOpenforcleanAntiPionsR;//negative\r
\r
//Secondrary Protons rest\r
TH3F* fDCAXYZOpenforcleanProtonsR;//positive\r
TH3F* fDCAXYZOpenforcleanAntiProtonsR;//negative\r
\r
\r
\r
\r
//Electron Muon  source procces\r
TH2F* fElectronsource; //e+\r
TH2F* fAntiElectronsource;//e-\r
\r
TH2F* fMuonsource;//mu+\r
TH2F* fAntiMuonsource;//mu-\r
\r
//N tpc clusters for \r
TH2F* fPionNTPCClusters; //pions tracks\r
TH2F* fAntiPionNTPCClusters;//antipions tracks \r
\r
TH2F* fKaonNTPCClusters; //Kaons tracks\r
TH2F* fAntiKaonNTPCClusters;//antiKaons tracks \r
\r
TH2F* fProtonNTPCClusters; //Protons tracks\r
TH2F* fAntiProtonNTPCClusters;//antiProtons tracks\r
\r
TH2F* fPionchi2; //pions tracks\r
TH2F* fAntiPionchi2;//antipions tracks \r
\r
TH2F* fKaonchi2; //Kaons tracks\r
TH2F* fAntiKaonchi2;//antiKaons tracks \r
\r
TH2F* fProtonchi2; //Protons tracks\r
TH2F* fAntiProtonchi2;//antiProtons tracks\r
 \r
\r
TH2F* fTracksCutmonitoring;// Number of tracks as fun of pt on each step of selection \r
TH3F* fParticlesCutmonitoring;//Number of as particles as fun of pt on each step of selection x 0-pion 1-kaon,2-proton,3-antipion,4-antikaon,5-antiproton\r
TH3F* fVertexshift; //shift of the vertex due to reconstruction\r
\r
TH3F* fPtESDminusPtMCvPtESDafterallcuts;//ptESD -ptMC v ptESD after all cuts \r
TH3F* fPtESDminusPtMCvPtESDafterTPCcuts;//ptESD - ptMC v ptESD after TPC cuts (refit,chi2,nclus);\r
\r
TH3F* fMulESDMulMCVz;//Multiplicty ESD Multiplicty MC Vrt Z \r
\r
\r
\r
//TPC pid objects \r
AliESDpidCuts* fTPCPIDCUT;//cut\r
AliESDpid* fESDpid; // global thing\r
\r
TH1F* fPrimaryElectronsMother; //name says all \r
\r
\r
\r
\r
\r
 TList *flist;//output list\r
	\r
 AliAnalysisChargedHadronSpectraITSTruncatedMeanTask(const AliAnalysisChargedHadronSpectraITSTruncatedMeanTask&); // not implemented\r
AliAnalysisChargedHadronSpectraITSTruncatedMeanTask& operator=(const AliAnalysisChargedHadronSpectraITSTruncatedMeanTask&); // not implemented\r
\r
\r
 Float_t MyITSsignalusing4points(Double_t* const) const;\r
  Float_t MyITSsignalusing3points(Double_t* const) const;\r
void CorrectSDD(Double_t *tmpQESD) const;\r
void CorrectSSD(Double_t *tmpQESD) const;\r
    Bool_t SelectOnImpPar(AliESDtrack* const t) const;\r
    Float_t GetWeight(Int_t type,AliStack* const stack) const;\r
    \r
    \r
    \r
 ClassDef(AliAnalysisChargedHadronSpectraITSTruncatedMeanTask, 2); // example of analysis\r
};\r
\r
#endif\r