AliAODPid.h

Go to the documentation of this file.

#ifndef AliAODPID_H
#define AliAODPID_H
/* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */


#include <TObject.h>
#include "AliPID.h"
#include "AliTPCdEdxInfo.h"

class AliAODPid : public TObject {

 public:
  AliAODPid();
  virtual ~AliAODPid();
  AliAODPid(const AliAODPid& pid); 
  AliAODPid& operator=(const AliAODPid& pid);
  
  enum{kTRDnPlanes=6};

 //setters
  void      SetITSsignal(Double_t its)                         {fITSsignal=its;}
  void      SetITSdEdxSamples(const Double_t s[4]);
  void      SetTPCsignal(Double_t tpc)                         {fTPCsignal=tpc;}
  void      SetTPCsignalN(UShort_t tpcN)                       {fTPCsignalN=(UChar_t)((tpcN<160)?tpcN:160);}
  void      SetTPCmomentum(Double_t tpcMom)                    {fTPCmomentum=tpcMom;}
  void      SetTPCTgl(Double_t tpcTgl)                         {fTPCTgl=tpcTgl;}
  void      SetTPCdEdxInfo(AliTPCdEdxInfo * dEdxInfo);
  void      SetTRDslices(Int_t nslices, const Double_t * const trdslices);  
  void      SetTRDsignal(Double_t sig)                         {fTRDsignal = sig;}
  void      SetTRDmomentum(Int_t nplane, Float_t trdMom)       {fTRDmomentum[nplane]=trdMom;}
  void      SetTRDncls(UChar_t ncls, Int_t layer = -1);
  void      SetTRDntrackletsPID(UChar_t ntls) {fTRDntls = ntls;}
  void      SetTRDChi2(Double_t chi2)          {fTRDChi2 = chi2;}
  void      SetTOFsignal(Double_t tof)                         {fTOFesdsignal=tof;}
  void      SetTOFpidResolution(Double_t tofPIDres[5]);
  void      SetIntegratedTimes(Double_t timeint[5]);

  Double_t  GetITSsignal()       const {return  fITSsignal;}
  void      GetITSdEdxSamples(Double_t *s) const;
  Double_t  GetITSdEdxSample(Int_t i) const {
    if(i>=0 && i<4) return fITSdEdxSamples[i];
    else return 0.;
  }
  Double_t  GetTPCsignal()       const {return  fTPCsignal;}
  UShort_t  GetTPCsignalN()      const {return  (UShort_t)fTPCsignalN;}
  Bool_t  GetTPCdEdxInfo( AliTPCdEdxInfo &v) const {
    if( fTPCdEdxInfo ){ v = *fTPCdEdxInfo; return kTRUE;}
    return kFALSE;
  }

  Double_t  GetTPCmomentum()     const {return  fTPCmomentum;}
  Double_t  GetTPCTgl()          const {return  fTPCTgl;}
  Int_t     GetTRDnSlices()      const {return  fTRDnSlices/6;}
  Double_t  GetTRDsignal()       const {return  fTRDsignal;}
  Double_t* GetTRDslices()       const {return  fTRDslices;}
  Double_t  GetTRDChi2()         const {return fTRDChi2;}
  const Double_t*  GetTRDmomentum() const {return  fTRDmomentum;}
  UChar_t   GetTRDncls(UChar_t layer) const { if(layer > 5) return 0; return fTRDncls[layer];}
  UChar_t   GetTRDncls() const;
  UChar_t   GetTRDntrackletsPID() const {return fTRDntls;}
  Double_t  GetTOFsignal()       const {return  fTOFesdsignal;}

  void      GetIntegratedTimes(Double_t *timeint, Int_t nspec=AliPID::kSPECIES)  const; 
  void      GetTOFpidResolution (Double_t tofRes[5]) const;

 private :
  Double32_t  fITSsignal;        //[0.,0.,10]
  Double32_t  fITSdEdxSamples[4];//[0.,0.,10]

  Double32_t  fTPCsignal;        //[0.,0.,10]
  UChar_t     fTPCsignalN;       
  Double32_t  fTPCmomentum;      //[0.,0.,20]
  Double32_t  fTPCTgl;           //[0.,0.,10]
  
  Int_t       fTRDnSlices;       
  UChar_t     fTRDntls;          
  UChar_t     fTRDncls[6];       

  Double32_t* fTRDslices;        //[fTRDnSlices][0.,0.,10]
  Double32_t  fTRDsignal;        //[0.,0.,10]
  Double32_t  fTRDmomentum[6];   //[0.,0.,10]
  Double32_t  fTRDChi2;          //[0.,0.,10]

  Double32_t  fTOFesdsignal;     //[0.,0.,20]
  Double32_t  fTOFpidResolution[5]; //[0.,0.,20]
  Double32_t  fIntTime[5];       //[0.,0.,20]
 
  AliTPCdEdxInfo * fTPCdEdxInfo; 

  ClassDef(AliAODPid, 15);
};

//_____________________________________________________________
inline void AliAODPid::SetTRDslices(Int_t nslices, const Double_t * const trdslices) {

  if(fTRDslices && fTRDnSlices != nslices) {
    delete [] fTRDslices; fTRDslices = NULL;
  };
  if(!fTRDslices) fTRDslices = new Double32_t[nslices];
  fTRDnSlices = nslices; 
  for(Int_t is = 0; is < fTRDnSlices; is++) fTRDslices[is] = trdslices[is];
}

//_____________________________________________________________
inline void AliAODPid::SetTRDncls(UChar_t ncls, Int_t layer) { 

  if(layer > 5) return; 
  if(layer < 0) fTRDncls[0] = ncls;
  else fTRDncls[layer] = ncls;
}

//_____________________________________________________________
inline UChar_t AliAODPid::GetTRDncls() const {

  UChar_t ncls = 0;
  for(Int_t ily = 0; ily < 6; ily++) ncls += fTRDncls[ily];
  return ncls;
}
#endif