Warning fixed

[u/mrichter/AliRoot.git] / PWG3 / muon / AliAODDimuon.h

#ifndef AliAODDimuon_H
#define AliAODDimuon_H

// AliAODDimuon: a class for AODs for the MUON Arm of the ALICE Experiment
// Author: P. Cortese, Universita' del Piemonte Orientale in Alessandria and
// INFN of Torino - Italy
//
// The class defines a dimuon pair object from two AliAODTrack objects.
// AliAODDimuon objects are supposed to be added to the AliAODEvent structure
// during analysis. They would then allow to calculate the dimuon-related
// kinematic variables with a minimal disk occupancy.
// The payload of the class has been reduced to two pointers to the two
// tracks with the addition of a pointer to the AliAODEventInfo. An instance of
// this class has also to be added to the AliAODEvent structure to provide
// additional information that is specific to MUON and therefore has not been
// included into the AOD header.
// Two transient data members are not stored on file as they can be recomputed
// at runtime.
//

// 2007/07/07 v1.00 Initial version
// 2007/12/06 v1.01 Introduction of AliAODEventInfo
// 2007/12/18 v1.02 Corrected CostCS for Like-Sign, added CostKh, CostHe and xf
// 2008/02/01 v1.03 Apply coding conventions

#include "TRef.h"
#include "AliVParticle.h"
#include "AliAODTrack.h"
#include "AliAODEvent.h"
#include "AliAODEventInfo.h"

class TLorentzVector;

class AliAODDimuon: public AliVParticle {
public:
  AliAODDimuon();
  AliAODDimuon(const AliAODDimuon& dimu);
  AliAODDimuon &operator=(const AliAODDimuon& dimu);
  AliAODDimuon(TObject *mu0, TObject *mu1, TObject *ei=0);
  virtual ~AliAODDimuon();

  // Methods to access kinematics
  virtual Double_t Px() const;
  virtual Double_t Py() const;
  virtual Double_t Pz() const;
  virtual Bool_t PxPyPz(Double_t* p) const { p[0]=Px(); p[1]=Py(); p[2]=Pz(); return 1;}
  virtual Double_t Pt() const;
  virtual Double_t P() const;

  virtual Double_t OneOverPt() const {return Pt()>0 ? 1./Pt() : -999999999;}
  virtual Double_t Phi() const;
  virtual Double_t Theta() const;

  virtual Double_t E() const;
  virtual Double_t M() const;
  
  virtual Double_t Eta() const;
  virtual Double_t Y() const;
  
  virtual Short_t Charge() const;

  // Dimuon vertex will be implemented when the muon track covariance matrix 
  // at vertex will be included in the ESD (and AOD)
  // It would require also the information about magnetic field when filling AOD
  virtual Double_t Xv() const {return -999999999;}
  virtual Double_t Yv() const {return -999999999;}
  virtual Double_t Zv() const {return -999999999;}
  virtual Bool_t XvYvZv(Double_t* v) const { v[0]=-999999999; v[1]=-999999999; v[2]=-999999999; return 0;}

  Double_t P();
  Double_t Phi();
  Double_t Theta();
  Double_t M();
  Double_t Mass();
  Double_t Eta();
  Double_t Y();

  // Added functions
  Double_t XF();     // Feynman x
  Double_t CostCS(); // Cosinus of the Collins-Soper polar decay angle
  Double_t CostHe(); // Cosinus of the Helicity polar decay angle
  Int_t AnyPt();
  Int_t LowPt();
  Int_t HighPt();
  Double_t MaxChi2Match();
  // PID
  virtual const Double_t *PID() const {return 0;} // return PID object (to be defined, still)
  //
  Int_t GetLabel() const {return -1;}

  // Additional getters and setters
  AliAODTrack* GetMu(Int_t imu=0) const {return (imu==0||imu==1)&&(fMu[imu]!=0) ? (AliAODTrack*)fMu[imu].GetObject() : 0; } // Get a pointer to a muon
  AliAODTrack* Mu(Int_t imu=0) const {return (imu==0||imu==1)&&(fMu[imu]!=0) ? (AliAODTrack*)fMu[imu].GetObject() : 0; } // Get a pointer to a muon
  AliAODEventInfo* GetEventInfo() const {return (fEi!=0) ? (AliAODEventInfo*)fEi.GetObject() : 0; } // Get a pointer to the AliAODEventInfo
  AliAODEventInfo* GetEi() const { return (fEi!=0) ? (AliAODEventInfo*)fEi.GetObject() : 0; }
  AliAODEventInfo* Ei() const { return (fEi!=0) ? (AliAODEventInfo*)fEi.GetObject() : 0; }

  void SetEi(AliAODEventInfo *ei){ fEi=ei; }
  void SetMu(Int_t imu=0, AliAODTrack *mu=0);
  void SetMuons(AliAODTrack *mu0=0, AliAODTrack *mu1=0);

private:
  Int_t CheckPointers() const;
  void BookP();

  // Data members
  TRef fMu[2];	// Pointers to the reconstructed muons
  TRef fEi;	// Pointer to the EventInfo object
  TLorentzVector *fP; //! TLorentzVector of dimuon momentum (not stored into file)

  // Useful constants
  Double_t fMProton; //! Proton mass (not stored into file)

  ClassDef(AliAODDimuon,1)  // AliAODDimuon track
};

#endif