Changed to new naming scheme for histograms, added functionality to the analysis...

[u/mrichter/AliRoot.git] / PWG4 / GammaConv / AliV0Reader.h

#ifndef ALIV0READER_H\r
#define ALIV0READER_H\r
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
 * See cxx source for full Copyright notice     */\r
\r
////////////////////////////////////////////////\r
//--------------------------------------------- \r
// Class used to do analysis on conversion pairs\r
//---------------------------------------------\r
////////////////////////////////////////////////\r
\r
// --- ROOT system ---\r
#include "TObject.h" \r
#include "AliESDv0.h"\r
#include "AliESDEvent.h"\r
#include "AliKFParticle.h"\r
#include "TParticle.h"\r
#include "AliGammaConversionHistograms.h"\r
#include <vector>\r
\r
class TClonesArray; \r
class TFormula;\r
class Riostream;\r
class TChain;\r
//--- AliRoot system ---\r
\r
class AliStack;\r
class AliESDEvent; \r
class AliMCEventHandler;\r
class AliESDInputHandler;\r
class AliESDVertex;\r
class AliLog;\r
class TChain;\r
class TChain;\r
\r
\r
\r
class AliV0Reader : public TObject {\r
\r
 public: \r
\r
  AliV0Reader();                                        //constructor\r
  AliV0Reader(const AliV0Reader & g);                   //copy constructor\r
  AliV0Reader & operator = (const AliV0Reader & g);     //assignment operator\r
  virtual ~AliV0Reader() {;}                            //virtual destructor\r
  /*\r
   *Initialize the reader\r
   */\r
  void Initialize();\r
\r
  /*\r
   *Returns the number of v0s in the event, no cuts applied.\r
   */\r
  Int_t GetNumberOfV0s() const{return fESDEvent->GetNumberOfV0s();}\r
\r
  /*\r
   * Check if there are any more good v0s left in the v0 stack\r
   * if so, fCurrent v0 is set to this v0 and can be retrieved\r
   * by GetCurrentV0 function.\r
   * returns kFALSE if there is no more good v0s in the v0 stack\r
   */\r
  Bool_t NextV0();\r
  \r
  /*\r
   * Returns the v0 at the given index, no checks are done on the v0. \r
   */\r
  AliESDv0* GetV0(Int_t index);\r
\r
  /*\r
   * Returns the current v0\r
   */\r
  AliESDv0* GetCurrentV0() const{return fCurrentV0;}\r
\r
  /*\r
   * Returns the negative ESD track which belongs to fCurrentV0\r
   */\r
  AliESDtrack* GetNegativeESDTrack(){return fESDEvent->GetTrack(fCurrentV0->GetNindex());}\r
\r
  /*\r
   * Returns the positive ESD track which belongs to fCurrentV0\r
   */\r
  AliESDtrack* GetPositiveESDTrack(){return fESDEvent->GetTrack(fCurrentV0->GetPindex());}\r
\r
  /*\r
   * Returns the negative KF particle which belongs to fCurrentV0\r
   */\r
  AliKFParticle* GetNegativeKFParticle() const{return fCurrentNegativeKFParticle;}\r
\r
  /*\r
   * Returns the positive KF particle which belongs to fCurrentV0\r
   */\r
  AliKFParticle* GetPositiveKFParticle() const{return fCurrentPositiveKFParticle;}\r
\r
  /*\r
   * Returns the KFParticle object of the 2 tracks.\r
   */\r
  AliKFParticle* GetMotherCandidateKFCombination() const{return fCurrentMotherKFCandidate;}\r
\r
  /*\r
   * Checks the probablity that the PID of the particle is what we want it to be.\r
   */\r
  Bool_t CheckPIDProbability(Double_t negProbCut, Double_t posProbCut);\r
  \r
  /*\r
   * Checks if the PID of the two particles are within our cuts.\r
   */\r
  void GetPIDProbability(Double_t &negPIDProb, Double_t &posPIDProb);\r
\r
  /*\r
   *Get the negative MC TParticle from the stack \r
   */\r
  TParticle * GetNegativeMCParticle() const{return fNegativeMCParticle;}\r
\r
  /*\r
   *Get the positive MC TParticle from the stack \r
   */\r
  TParticle * GetPositiveMCParticle() const{return fPositiveMCParticle;}\r
\r
  /*\r
   *Get the mother MC TParticle from the stack \r
   */\r
  TParticle * GetMotherMCParticle() const{return fMotherMCParticle;}\r
\r
  /*\r
   * Flag to see if the v0 particles share the same mother\r
   */\r
  Bool_t HasSameMCMother();\r
\r
\r
  /*\r
   *Get the PID of the MC mother particle\r
   */\r
  Int_t GetMotherMCParticlePDGCode() const{if(fMotherMCParticle != NULL){ cout<<"MCParticle exists"<<endl;} return fMotherMCParticle->GetPdgCode();}\r
\r
  /*\r
   *Get the MC stack \r
   */\r
  AliStack* GetMCStack() const{return fMCStack;}\r
\r
  /*\r
   *Get the magnetic field from the ESD event \r
   */\r
  Double_t GetMagneticField() const{return fESDEvent->GetMagneticField();}\r
\r
  /*\r
   *Get the primary vertex from the esd event\r
   */\r
  const AliESDVertex *GetPrimaryVertex() const {return fESDEvent->GetPrimaryVertex();}\r
\r
  /*\r
   * Set the PID of the negative track\r
   */\r
  void SetNegativeTrackPID(Int_t negTrackPID){fNegativeTrackPID=negTrackPID;}\r
\r
  /*\r
   * Set the PID of the positive track\r
   */\r
  void SetPositiveTrackPID(Int_t posTrackPID){fPositiveTrackPID=posTrackPID;}\r
\r
  /*\r
   * Set the flag to use the kfparticle class. Will also disable the use of esd tracks\r
   */\r
  void UseKFParticle(){fUseKFParticle = kTRUE; fUseESDTrack = kFALSE;}\r
\r
  /*\r
   *  Set the flag to use the esd track class. Will also disable the use of kf particles\r
   */\r
  void UseESDTrack(){fUseESDTrack = kTRUE; fUseKFParticle = kFALSE;}\r
\r
  /*\r
   *  Set the flag to use improved vertex or not\r
   */\r
  void SetUseImprovedVertex(Bool_t useImprovedVertex){fUseImprovedVertex=useImprovedVertex;}\r
\r
  /*\r
   * Return the number in the species array belonging to the negative or positive track pid.\r
   */\r
  Int_t GetSpeciesIndex(Int_t chargeOfTrack);\r
\r
  /*\r
   * Return the x coordinate of the v0\r
   */\r
  Double_t GetX() const{return fCurrentXValue;}\r
\r
  /*\r
   * Return the y coordinate of the v0\r
   */\r
  Double_t GetY() const{return fCurrentYValue;}\r
\r
  /*\r
   * Return the Z coordinate of the v0\r
   */\r
  Double_t GetZ() const{return fCurrentZValue;}\r
\r
  /*\r
   * Return the radius of the v0\r
   */\r
  Double_t GetXYRadius() const{return sqrt((Double_t)(fCurrentXValue*fCurrentXValue + fCurrentYValue*fCurrentYValue));}\r
\r
  /*\r
   * Get the opening angle between the two tracks\r
   */\r
  Double_t GetOpeningAngle(){return fNegativeTrackLorentzVector->Angle(fPositiveTrackLorentzVector->Vect());}\r
\r
  /*\r
   * Gets the Energy of the negative track.\r
   */\r
  Double_t GetNegativeTrackEnergy() const{return fCurrentNegativeKFParticle->E();}\r
\r
  /*\r
   * Gets the Energy of the positive track.\r
   */\r
  Double_t GetPositiveTrackEnergy() const{return fCurrentPositiveKFParticle->E();}\r
\r
  /*\r
   * Gets the Energy of the mother candidate.\r
   */\r
  Double_t GetMotherCandidateEnergy() const{return fCurrentMotherKFCandidate->E();}\r
\r
  /*\r
   * Gets the Pt of the negative track.\r
   */\r
  Double_t GetNegativeTrackPt() const{return fNegativeTrackLorentzVector->Pt();}\r
\r
  /*\r
   * Gets the Pt of the positive track.\r
   */\r
  Double_t GetPositiveTrackPt() const{return fPositiveTrackLorentzVector->Pt();}\r
\r
  /*\r
   * Gets the Pt of the mother candidate.\r
   */\r
  Double_t GetMotherCandidatePt() const{return fMotherCandidateLorentzVector->Pt();}\r
\r
  /*\r
   * Gets the Eta of the negative track.\r
   */\r
  Double_t GetNegativeTrackEta() const{return fNegativeTrackLorentzVector->Eta();}\r
  /*\r
   * Gets the Eta of the positive track.\r
   */\r
  Double_t GetPositiveTrackEta() const{return fPositiveTrackLorentzVector->Eta();}\r
  /*\r
   * Gets the Eta of the mother candidate.\r
   */\r
  Double_t GetMotherCandidateEta() const{return fMotherCandidateLorentzVector->Eta();}\r
\r
  /*\r
   * Gets the NDF of the mother candidate.\r
   */\r
  Double_t GetMotherCandidateNDF() const{return fCurrentMotherKFCandidate->GetNDF();}\r
\r
  /*\r
   * Gets the Chi2 of the mother candidate.\r
   */\r
  Double_t GetMotherCandidateChi2() const{return fCurrentMotherKFCandidate->GetChi2();}\r
\r
  /*\r
   * Gets the Mass of the mother candidate.\r
   */\r
  Double_t GetMotherCandidateMass() const{return fMotherCandidateKFMass;}\r
\r
  /*\r
   * Gets the Width of the mother candidate.\r
   */\r
  Double_t GetMotherCandidateWidth() const{return fMotherCandidateKFWidth;}\r
\r
  /*\r
   * Gets the Phi of the negative track.\r
   */\r
  Double_t GetNegativeTrackPhi() const;\r
\r
  /*\r
   * Gets the Phi of the positive track.\r
   */\r
  Double_t GetPositiveTrackPhi() const;\r
\r
  /*\r
   * Gets the Phi of the mother candidate.\r
   */\r
  Double_t GetMotherCandidatePhi() const;\r
\r
  /*\r
   * Gets the Rapidity of the mother candidate.\r
   */\r
  Double_t GetMotherCandidateRapidity() const;\r
\r
\r
  /*\r
   * Update data which need to be updated every event.\r
   */\r
  void UpdateEventByEventData();\r
  \r
  /*\r
   * Gets the MaxRCut value.\r
   */\r
  Double_t GetMaxRCut() const{return fMaxR;}\r
\r
  /*\r
   * Gets the Eta cut value.\r
   */\r
  Double_t GetEtaCut() const{return fEtaCut;}\r
\r
  /*\r
   * Gets the Pt cut value.\r
   */\r
  Double_t GetPtCut() const{return fPtCut;}\r
\r
  /*\r
   * Gets the line cut values.\r
   */\r
  Double_t GetLineCutZRSlope() const{return fLineCutZRSlope;}\r
  Double_t GetLineCutZValue() const{return fLineCutZValue;}\r
  \r
  /*\r
   * Gets the Chi2 cut value for the conversions.\r
   */\r
  Double_t GetChi2CutConversion() const{return fChi2CutConversion;}\r
\r
  /*\r
   * Gets the Chi2 cut value for the mesons.\r
   */\r
  Double_t GetChi2CutMeson() const{return fChi2CutMeson;}\r
\r
  Double_t GetPositiveTrackLength() const{return fCurrentPositiveESDTrack->GetIntegratedLength();}\r
  Double_t GetNegativeTrackLength() const{return fCurrentNegativeESDTrack->GetIntegratedLength();}\r
\r
  Double_t GetPositiveNTPCClusters() const{return fCurrentPositiveESDTrack->GetTPCNcls();}\r
  Double_t GetNegativeNTPCClusters() const{return fCurrentNegativeESDTrack->GetTPCNcls();}\r
\r
  /*\r
   * Sets the MaxRCut value.\r
   */\r
  void SetMaxRCut(Double_t maxR){fMaxR=maxR;}\r
\r
  /*\r
   * Sets the EtaCut value.\r
   */\r
  void SetEtaCut(Double_t etaCut){fEtaCut=etaCut;}\r
\r
  /*\r
   * Sets the PtCut value.\r
   */\r
  void SetPtCut(Double_t ptCut){fPtCut=ptCut;}\r
\r
  /*\r
   * Sets the LineCut values.\r
   */\r
  void SetLineCutZRSlope(Double_t LineCutZRSlope){fLineCutZRSlope=LineCutZRSlope;}\r
  void SetLineCutZValue(Double_t LineCutZValue){fLineCutZValue=LineCutZValue;}\r
\r
  /*\r
   * Sets the Chi2Cut value for conversions.\r
   */\r
  void SetChi2CutConversion(Double_t chi2){fChi2CutConversion=chi2;}\r
\r
  /*\r
   * Sets the Chi2Cut for the mesons.\r
   */\r
  void SetChi2CutMeson(Double_t chi2){fChi2CutMeson=chi2;}\r
  \r
  /*\r
   * Sets the XVertexCut value.\r
   */\r
  void SetXVertexCut(Double_t xVtx){fCurrentXValue=xVtx;}\r
\r
  /*\r
   * Sets the YVertexCut value.\r
   */\r
  void SetYVertexCut(Double_t yVtx){fCurrentYValue=yVtx;}\r
\r
  /*\r
   * Sets the ZVertexCut value.\r
   */\r
  void SetZVertexCut(Double_t zVtx){fCurrentZValue=zVtx;}\r
\r
  /*\r
   * Sets the PIDProbabilityCut value for track particles.\r
   */\r
  void SetPIDProbability(Double_t pidProb){fPIDProbabilityCutPositiveParticle=pidProb; fPIDProbabilityCutNegativeParticle=pidProb;}\r
\r
  /*\r
   * Sets the PIDProbability cut value for the negative track.\r
   */\r
  void SetPIDProbabilityNegativeParticle(Double_t pidProb){fPIDProbabilityCutNegativeParticle=pidProb;}\r
\r
  /*\r
   * Sets the PIDProbability cut value for the positive track.\r
   */\r
  void SetPIDProbabilityPositiveParticle(Double_t pidProb){fPIDProbabilityCutPositiveParticle=pidProb;}\r
\r
  /*\r
   * Sets the SigmaMassCut value.\r
   */\r
  void SetSigmaMass(Double_t sigmaMass){fNSigmaMass=sigmaMass;}\r
\r
  /*\r
   * Sets the flag to enable/disable the usage of MC information. \r
   */\r
  void SetDoMCTruth(Bool_t doMC){fDoMC = doMC;}\r
\r
  /*\r
   * Updates the V0 information of the current V0.\r
   */\r
  Bool_t UpdateV0Information();\r
\r
  /*\r
   * Resets the V0 index.\r
   */\r
  void ResetV0IndexNumber(){fCurrentV0IndexNumber=0;}\r
\r
  /*\r
   * Sets the histograms.\r
   */\r
  void SetHistograms(AliGammaConversionHistograms *histograms){fHistograms=histograms;}\r
\r
  /*\r
   * Check for primary vertex.\r
   */\r
  Bool_t CheckForPrimaryVertex();\r
\r
  /*\r
   * Gets a vector of good v0s.\r
   */\r
  vector<AliKFParticle> GetCurrentEventGoodV0s() const{return fCurrentEventGoodV0s;}\r
\r
  /*\r
   * Gets the vector of previous events v0s (for bacground analysis)\r
   */\r
  vector<AliKFParticle> GetPreviousEventGoodV0s() const{return fPreviousEventGoodV0s;}\r
\r
 private:\r
  AliStack * fMCStack;           // pointer to MonteCarlo particle stack \r
  AliMCEventHandler* fMCTruth;   // pointer to the MC event handler\r
  TChain * fChain;               // pointer to the TChain\r
  \r
  AliESDInputHandler* fESDHandler;      //! pointer to esd object\r
  AliESDEvent *fESDEvent;               //! pointer to esd object\r
\r
  AliGammaConversionHistograms *fHistograms;\r
  \r
  Int_t fCurrentV0IndexNumber;\r
  AliESDv0 * fCurrentV0;                //! pointer to the current v0\r
  AliKFParticle * fCurrentNegativeKFParticle;  //! pointer to the negative KF particle\r
  AliKFParticle * fCurrentPositiveKFParticle;  //! pointer to the positive KF particle\r
  AliKFParticle * fCurrentMotherKFCandidate;   //! pointer to the positive KF particle\r
\r
  AliESDtrack * fCurrentNegativeESDTrack;      //! pointer to the negative ESD track\r
  AliESDtrack * fCurrentPositiveESDTrack;      //! pointer to the positive ESD track\r
 \r
  TLorentzVector * fNegativeTrackLorentzVector; //! pointer to the negative Track Lorentz Vector\r
  TLorentzVector * fPositiveTrackLorentzVector; //! pointer to the positive Track Lorentz Vector\r
  TLorentzVector * fMotherCandidateLorentzVector;   //! pointer to the mother candidate Track Lorentz Vector\r
\r
  Double_t fCurrentXValue;\r
  Double_t fCurrentYValue;\r
  Double_t fCurrentZValue;\r
\r
  Int_t fPositiveTrackPID;\r
  Int_t fNegativeTrackPID;\r
\r
  TParticle *fNegativeMCParticle;      //!\r
  TParticle *fPositiveMCParticle;      //!\r
  TParticle *fMotherMCParticle;        //!\r
\r
  Double_t fMotherCandidateKFMass;\r
  Double_t fMotherCandidateKFWidth;\r
\r
  Bool_t fUseKFParticle;\r
  Bool_t fUseESDTrack;\r
  Bool_t fDoMC;\r
\r
  //cuts\r
  Double_t fMaxR;\r
  Double_t fEtaCut;\r
  Double_t fPtCut;\r
  Double_t fLineCutZRSlope;\r
  Double_t fLineCutZValue;\r
  Double_t fChi2CutConversion;\r
  Double_t fChi2CutMeson;\r
  Double_t fPIDProbabilityCutNegativeParticle;\r
  Double_t fPIDProbabilityCutPositiveParticle;\r
  Double_t fXVertexCut;\r
  Double_t fYVertexCut;\r
  Double_t fZVertexCut;\r
  \r
  Double_t fNSigmaMass;\r
  \r
  Bool_t fUseImprovedVertex;\r
  \r
  vector<AliKFParticle> fCurrentEventGoodV0s;\r
  vector<AliKFParticle> fPreviousEventGoodV0s;\r
\r
  ClassDef(AliV0Reader,1)\r
};\r
\r
\r
#endif\r
\r
\r
\r