JCORRAN code update from DongJo

[u/mrichter/AliRoot.git] / PWGCF / Correlations / JCORRAN / AliJTrack.h

diff --git a/PWGCF/Correlations/JCORRAN/AliJTrack.h b/PWGCF/Correlations/JCORRAN/AliJTrack.h
old mode 100755 (executable)
new mode 100644 (file)
index fc8b233..c93045c
--- a/PWGCF/Correlations/JCORRAN/AliJTrack.h
+++ b/PWGCF/Correlations/JCORRAN/AliJTrack.h
@@ -1,3 +1,8 @@
+/* Copyright(c) 1998-2014, ALICE Experiment at CERN, All rights reserved. *
+ * See cxx source for full Copyright notice */
+
+// Short comment describing what this class does needed!
+
 // $Id: AliJTrack.h,v 1.3 2008/01/21 11:56:39 djkim Exp $
 ////////////////////////////////////////////////////
 /*!
@@ -33,53 +38,66 @@ public:
    };
 
    enum AliJTrkPIDmethod {
-    kTOF=0, kTPC=1, kTPCTOF=2, kNAliJTrkPIDmethod
+    //kTOF=0, kTPC=1, kTPCTOF=2, kNAliJTrkPIDmethod
+    kTOF=0, kNAliJTrkPIDmethod
    };
 
-  AliJTrack();                  // default constructor
+  AliJTrack();             // default constructor
   AliJTrack(const AliJTrack& a); // copy constructor
-  ~AliJTrack(){;}              //destructor
+  ~AliJTrack(){;}    //destructor
 
   Double32_t  GetPID(AliJTrkPID p, AliJTrkPIDmethod m) const { return fTrkPID[p][m]; }
   UInt_t      GetFilterMap() const { return fFilterMap; }
   Bool_t      IsFiltered( int i ) const { return TESTBIT( fFilterMap, i); }
   Bool_t      IsFilteredMask( UInt_t mask ) const { return ((Bool_t)(((fFilterMap) & mask) != 0)); };
   int         GetTPCnClust() const {return fTPCnClust;}
-  Float_t     GetTPCdEdx()  const {return fTPCdEdx; }
-  Float_t  GetTOFbeta() const {return fTOFbeta;}
-  Float_t  GetExpectedTOFbeta( AliJTrkPID p ) const {return fExpTOFbeta[p];}
-  Double_t  GetExpectedTPCdEdx( AliJTrkPID p ) const {return fExpTPCdEdx[p];}
-  Double_t  GetTPCsigma( AliJTrkPID p) const {return fTPCsigma[p];}
-  Double_t  GetTOFsigma( AliJTrkPID p) const {return fTOFsigma[p];}
+  Double32_t     GetTPCdEdx()  const {return fTPCdEdx; }
+  Double32_t GetTOFsignal() const {return fTOFsignal;}
+  Double32_t GetExpectedTOFsignal( AliJTrkPID p ) const {return fExpTOFsignal[p];}
+  Double32_t GetTPCmomentum() const {return fTPCmom;}
+  void GetTrackPos( Double32_t *p ) const { p[0] = fTrackPos[0]; p[1] = fTrackPos[1]; p[2] = fTrackPos[2]; }
 
   void SetPID(AliJTrkPID p, double pro, AliJTrkPIDmethod m){ fTrkPID[p][m] = pro; }
   void SetFilterMap( UInt_t map ){ fFilterMap = map; }
   void SetFilterMap( int i, bool t ){ t?SETBIT(fFilterMap,i):CLRBIT(fFilterMap,i); } 
   void SetTPCnClust(int ival) {fTPCnClust = ival;}
-  void SetTPCdEdx( Float_t dedx ){ fTPCdEdx = dedx; }
-  void SetTOFbeta(Float_t beta) {fTOFbeta = beta;}
-  void SetExpectedTOFbeta(AliJTrkPID p, float exbeta) {fExpTOFbeta[p] = exbeta;}
-  void SetExpectedTPCdEdx(AliJTrkPID p, Double_t exdEdx) {fExpTPCdEdx[p] = exdEdx;}
-  void SetTPCsigma(AliJTrkPID p, Double_t TPCsigma) {fTPCsigma[p] = TPCsigma;}
-  void SetTOFsigma(AliJTrkPID p, Double_t TOFsigma) {fTOFsigma[p] = TOFsigma;} // Return the expected sigma of the PID signal for the specified  particle type.
-
-  void SetTPCTrack( Float_t px,  Float_t py,  Float_t pz ){ fTPCTrack[0]=px;fTPCTrack[1]=py;fTPCTrack[2]=pz;}
+  void SetTPCdEdx(Double32_t dedx ){ fTPCdEdx = dedx; }
+  void SetTOFsignal(Double_t tofsig) {fTOFsignal = tofsig;}
+  void SetExpectedTOFsignal(AliJTrkPID p, double extime) {fExpTOFsignal[p] = extime;}
+  void SetTPCmomentum(Double_t momTPC) {fTPCmom = momTPC;}
+  void SetTPCTrack( Double32_t px,  Double32_t py,  Double32_t pz ){ fTPCTrack[0]=px;fTPCTrack[1]=py;fTPCTrack[2]=pz;}
+  void SetGCGTrack( Double32_t px,  Double32_t py,  Double32_t pz ){ fGCGTrack[0]=px;fGCGTrack[1]=py;fGCGTrack[2]=pz;}
+  void SetTrackPos( Double32_t *p ){ fTrackPos[0]=p[0];fTrackPos[1]=p[1];fTrackPos[2]=p[2];}
   void SetUseTPCTrack(){ SetPxPyPzE( fTPCTrack[0], fTPCTrack[1], fTPCTrack[2], 0 ); } 
-  Float_t* GetTPCTrack(){ return fTPCTrack; }
+  void SetUseGCGTrack(){ SetPxPyPzE( fGCGTrack[0], fGCGTrack[1], fGCGTrack[2], 0 ); } 
+  Double32_t * GetTPCTrack(){ return fTPCTrack; }
+  Double32_t * GetGCGTrack(){ return fGCGTrack; }
+
+  Double32_t  PtTPC(){ return TMath::Sqrt( fTPCTrack[0]*fTPCTrack[0] +  fTPCTrack[1]*fTPCTrack[1] ); }
+  Double32_t  PtGCG(){ return TMath::Sqrt( fGCGTrack[0]*fGCGTrack[0] +  fGCGTrack[1]*fGCGTrack[1] ); }
+
 
   AliJTrack& operator=(const AliJTrack& trk);
 
+  //Double32_t GetDCAtoVertexXY(){ return fDCAtoVertexXY; }
+  //void SetDCAtoVertexXY( double dxy ){ fDCAtoVertexXY = dxy; }
+  //Double32_t GetDCAtoVertexZ(){ return fDCAtoVertexZ; }
+  //void SetDCAtoVertexZ( double dxy ){ fDCAtoVertexZ = dxy; }
+
 private:
-  Double32_t  fTrkPID[kNAliJTrkPID][kNAliJTrkPIDmethod];    //[0.,1.,8] Array for PID. 
-  UInt_t      fFilterMap;               // bit serious of cuts
-  Short_t     fTPCnClust;               // track TPC nclusters 
-  Float_t     fTPCdEdx;
-  Float_t    fTOFbeta;                    //!
-  Float_t    fExpTOFbeta[kNAliJTrkPID];   //!
-  Double_t   fExpTPCdEdx[kNAliJTrkPID];   //!
-  Double_t   fTPCsigma[kNAliJTrkPID];     //!
-  Double_t   fTOFsigma[kNAliJTrkPID];     //!
-  Float_t    fTPCTrack[3];              // px, py, pz for TPCTrack;
+  Double32_t    fTrkPID[kNAliJTrkPID][kNAliJTrkPIDmethod];   //[0.,1.,8] Array for PID. 
+  UInt_t        fFilterMap;                 // bit serious of cuts
+  Short_t       fTPCnClust;                 // track TPC nclusters 
+  Double32_t    fTPCdEdx;                   // TPC dEdx
+  Double32_t   fTOFsignal;             // TOF time
+  Double32_t   fExpTOFsignal[kNAliJTrkPID];    // expected TOF time
+  Double32_t   fTPCmom;                // TPC momentum to calculate expected TPC dEdx
+  Double32_t    fTPCTrack[3];               // px, py, pz for TPCTrack;
+  Double32_t    fGCGTrack[3];               // px, py, pz for GCGTrack;
+  Double32_t    fTrackPos[3];               // track position
+
+  //Double32_t    fDCAtoVertexXY;             //!
+  //Double32_t    fDCAtoVertexZ;             //!
 
   ClassDef(AliJTrack,1)
 };