StRoot: StHbtMaker/Infrastructure/StHbtTrack.hh Source File

00001 /***************************************************************************
00002  *
00003  * $Id: StHbtTrack.hh,v 1.24 2005/07/15 17:40:57 kopytin Exp $
00004  *
00005  * Author: Mike Lisa, Ohio State, lisa@mps.ohio-state.edu
00006  ***************************************************************************
00007  *
00008  * Description: part of STAR HBT Framework: StHbtMaker package
00009  *    Intermediate format for particle.  This is built from the
00010  *    input particle format (e.g. StTrack of StEvent) and presented to
00011  *    the Analyses for ParticleCuts.
00012  *
00013  ***************************************************************************
00014  *
00015  * $Log: StHbtTrack.hh,v $
00016  * Revision 1.24  2005/07/15 17:40:57  kopytin
00017  * initialized mHiddenInfo to null in StHbtTrack()
00018  *
00019  * Revision 1.23  2003/09/02 17:58:32  perev
00020  * gcc 3.2 updates + WarnOff
00021  *
00022  * Revision 1.22  2003/03/18 14:41:48  kisiel
00023  * Bugfix update for the theoretical part of the code. Reverts the changes to the Lednicky weight calculator, as the previuos one had problems with strong interaction
00024  *
00025  * Revision 1.21  2002/03/21 18:49:31  laue
00026  * updated for new MuDst reader
00027  *
00028  * Revision 1.20  2001/12/14 23:11:30  fretiere
00029  * Add class HitMergingCut. Add class fabricesPairCut = HitMerginCut + pair purity cuts. Add TpcLocalTransform function which convert to local tpc coord (not pretty). Modify StHbtTrack, StHbtParticle, StHbtHiddenInfo, StHbtPair to handle the hit information and cope with my code
00030  *
00031  * Revision 1.17  2001/06/21 19:15:48  laue
00032  * Modified fiels:
00033  *   CTH.hh : new constructor added
00034  *   StHbtEvent, StHbtKink, StHbtTrack : constructors from the persistent
00035  *                                   (TTree) classes added
00036  *   StHbtLikeSignAnalysis : minor changes, for debugging
00037  *   StHbtTypes: split into different files
00038  * Added files: for the new TTree muDst's
00039  *   StExceptions.cxx StExceptions.hh StHbtEnumeration.hh
00040  *   StHbtHelix.hh StHbtHisto.hh StHbtString.hh StHbtTFile.hh
00041  *   StHbtTTreeEvent.cxx StHbtTTreeEvent.h StHbtTTreeKink.cxx
00042  *   StHbtTTreeKink.h StHbtTTreeTrack.cxx StHbtTTreeTrack.h
00043  *   StHbtTTreeV0.cxx StHbtTTreeV0.h StHbtVector.hh
00044  *
00045  * Revision 1.16  2001/05/25 23:23:59  lisa
00046  * Added in StHbtKink stuff
00047  *
00048  * Revision 1.15  2001/04/05 22:25:15  kisiel
00049  *   Correct the problems detected by Insure++.
00050  *
00051  * Revision 1.14  2001/04/03 21:04:36  kisiel
00052  *
00053  *
00054  *   Changes needed to make the Theoretical code
00055  *   work. The main code is the ThCorrFctn directory.
00056  *   The most visible change is the addition of the
00057  *   HiddenInfo to StHbtPair.
00058  *
00059  * Revision 1.13  2000/05/03 17:44:43  laue
00060  * StHbtEvent, StHbtTrack & StHbtV0 declared friend to StHbtIOBinary
00061  * StHbtParticle updated for V0 pos,neg track Id
00062  *
00063  * Revision 1.12  2000/04/03 16:21:51  laue
00064  * some include files changed
00065  * Multi track cut added
00066  *
00067  * Revision 1.11  2000/02/18 21:32:24  laue
00068  * franksTrackCut changed. If mCharge is set to '0' there will be no cut
00069  * on charge. This is important for front-loaded cuts.
00070  *
00071  * copy constructor implemented for StHbtEvent, StHbtTrack and StHbtV0.
00072  *
00073  * franks1HistoD.cxx franks1HistoD.h franks2HistoD.cxx franks2HistoD.h
00074  * removed. We can now (CC5 on Solaris) use the versions (no D)
00075  *
00076  * Revision 1.10  2000/01/25 17:35:17  laue
00077  * I. In order to run the stand alone version of the StHbtMaker the following
00078  * changes have been done:
00079  * a) all ClassDefs and ClassImps have been put into #ifdef __ROOT__ statements
00080  * b) unnecessary includes of StMaker.h have been removed
00081  * c) the subdirectory StHbtMaker/doc/Make has been created including everything
00082  * needed for the stand alone version
00083  *
00084  * II. To reduce the amount of compiler warning
00085  * a) some variables have been type casted
00086  * b) some destructors have been declared as virtual
00087  *
00088  * Revision 1.9  2000/01/07 22:16:14  laue
00089  * missing 'const' in Helix() added
00090  *
00091  * Revision 1.8  2000/01/06 17:36:38  laue
00092  * mNHits changed from 'int' to 'unsigned short'
00093  * 'int' is inconsistend with the corresponding getters and setters
00094  *
00095  * Revision 1.7  1999/09/03 22:39:16  lisa
00096  * Readers now MUST have Report() methods and MAY have WriteHbtEvent() methods
00097  *
00098  * Revision 1.6  1999/09/01 19:04:54  lisa
00099  * update Particle class AND add parity cf and Randys Coulomb correction
00100  *
00101  * Revision 1.5  1999/07/27 20:21:10  lisa
00102  * Franks fixes of StTrack and subsequent changes to particleCut and EventReader
00103  *
00104  * Revision 1.4  1999/07/19 14:24:06  hardtke
00105  * modifications to implement uDST
00106  *
00107  * Revision 1.3  1999/07/06 22:33:23  lisa
00108  * Adjusted all to work in pro and new - dev itself is broken
00109  *
00110  * Revision 1.2  1999/06/29 17:50:27  fisyak
00111  * formal changes to account new StEvent, does not complie yet
00112  *
00113  * Revision 1.1.1.1  1999/06/29 16:02:57  lisa
00114  * Installation of StHbtMaker
00115  *
00116  **************************************************************************/
00117 
00118 #ifndef StHbtTrack_hh
00119 #define StHbtTrack_hh
00120 
00121 #include "Stiostream.h"
00122 
00123 #include "StHbtMaker/Infrastructure/StHbtTypes.hh"
00124 #include "StPhysicalHelixD.hh"
00125 /* Th stuff */
00126 #include "StHbtMaker/Base/StHbtHiddenInfo.hh"
00127 /***/
00128 
00129 class StEvent;
00130 class StTrack;
00131 class StHbtTTreeTrack;
00132 class StHbtTTreeEvent;
00133 class StMuDst;
00134 class StMuTrack;
00135 
00136 class StHbtTrack{
00137 public:
00138   StHbtTrack();
00139   StHbtTrack(const StHbtTrack&);// copy constructor
00140 #ifdef __ROOT__
00141   StHbtTrack(const StTrack*, StHbtThreeVector);   // c-tor from StTrack of STAR DSTs
00142   StHbtTrack(const StHbtTTreeEvent* ev, const StHbtTTreeTrack* t);
00143   StHbtTrack(const StMuDst* dst, const StMuTrack* t);
00144 #endif
00145   StHbtTrack(const StEvent*, const StTrack*);
00146   ~StHbtTrack();
00147 //    ~StHbtTrack(){/* no-op*/};
00148 
00149   short TrackType() const;
00150   short Charge() const;
00151   short NHits() const;
00152   short NHitsPossible() const;
00153   short NHitsDedx() const;
00154   const float* NSigma() const; // Fab private
00155   float NSigmaElectron() const;
00156   float NSigmaPion() const;
00157   float NSigmaKaon() const;
00158   float NSigmaProton() const;
00159   float PidProbElectron() const;
00160   float PidProbPion() const;
00161   float PidProbKaon() const;
00162   float PidProbProton() const;
00163   float dEdx() const;
00164   float DCAz() const;
00165   float DCAxy() const;
00166   float DCAzGlobal() const;
00167   float DCAxyGlobal() const;
00168   float ChiSquaredXY() const;
00169   float ChiSquaredZ() const;
00170   StHbtThreeVector P() const;
00171   float Pt() const;
00172   StHbtThreeVector PGlobal() const;
00173   float PtGlobal() const;
00174   const StPhysicalHelixD& Helix() const;
00175   const StPhysicalHelixD& HelixGlobal() const;
00176   unsigned int TopologyMap(const unsigned int word) const;
00177   short TrackId() const;
00178 
00179   void SetTrackType(const short&);
00180   void SetCharge(const short&);
00181   void SetNHits(const short&);
00182   void SetNHitsPossible(const short&);
00183   void SetNHitsDedx(const short&);
00184   void SetNSigmaElectron(const float&);
00185   void SetNSigmaPion(const float&);
00186   void SetNSigmaKaon(const float&);
00187   void SetNSigmaProton(const float&);
00188   void SetPidProbElectron(const float&);
00189   void SetPidProbPion(const float&);
00190   void SetPidProbKaon(const float&);
00191   void SetPidProbProton(const float&);
00192   void SetdEdx(const float&);
00193   void SetDCAxy(const float&);
00194   void SetDCAz(const float&);
00195   void SetDCAxyGlobal(const float&);
00196   void SetDCAzGlobal(const float&);
00197   void SetChiSquaredXY(const float&);
00198   void SetChiSquaredZ(const float&);
00199   void SetP(const StHbtThreeVector&);
00200   void SetPt(const float&);
00201   void SetPGlobal(const StHbtThreeVector&);
00202   void SetPtGlobal(const float&);
00203   void SetHelix(const StPhysicalHelixD&);
00204   void SetHelixGlobal(const StPhysicalHelixD&);
00205   void SetTopologyMap(const int word, const unsigned int map);
00206   void SetTrackId(const short&);
00207 
00208   /* Th stuff */
00209   void SetHiddenInfo(StHbtHiddenInfo* aHiddenInfo);
00210   bool ValidHiddenInfo() const;
00211   // Fab private : (official : const StHbtHiddenInfo* HiddenInfo() const;
00212   StHbtHiddenInfo* getHiddenInfo() const;
00213   /***/
00214 
00215   // For I/O of this object -- functions defined in StHbtIO.cc
00216   friend ostream& operator<<(ostream& out, StHbtTrack& trk);
00217   friend istream& operator>>(istream& in,  StHbtTrack& trk);
00218 
00219 private:
00220   short mTrackType;
00221   char mCharge;
00222   unsigned short mNHits;
00223   unsigned short mNHitsPoss; 
00224   unsigned short mNHitsDedx;
00225   float mNSigmaElectron;
00226   float mNSigmaPion;
00227   float mNSigmaKaon;
00228   float mNSigmaProton;
00229   float mPidProbElectron; // new
00230   float mPidProbPion; // new
00231   float mPidProbKaon; // new
00232   float mPidProbProton; // new
00233   float mdEdx;
00234   float mDCAxy;
00235   float mDCAz; 
00236   float mDCAxyGlobal;
00237   float mDCAzGlobal; 
00238   float mChiSqXY;
00239   float mChiSqZ;
00240   unsigned int mMap[2];
00241   unsigned int mTrackId;
00242 
00243 
00244   StHbtThreeVector mP;
00245   float mPt;
00246   StHbtThreeVector mPGlobal;
00247   float mPtGlobal;
00248   StPhysicalHelixD mHelix;
00249   StPhysicalHelixD mHelixGlobal;
00250 
00251   /* Th stuff */
00252   // Fab private : add mutable
00253   //  mutable 
00254   StHbtHiddenInfo* mHiddenInfo; 
00255   /***/
00256 
00257   friend class StHbtIOBinary;
00258   friend class StHbtTTreeEvent;
00259   friend class StHbtTTreeTrack;
00260 };
00261 
00262 inline const float* StHbtTrack::NSigma() const 
00263 {return &mNSigmaElectron;} // Fab private 
00264 inline float StHbtTrack::PidProbElectron() const {return mPidProbElectron;}
00265 inline float StHbtTrack::PidProbPion() const {return mPidProbPion;}
00266 inline float StHbtTrack::PidProbKaon() const {return mPidProbKaon;}
00267 inline float StHbtTrack::PidProbProton() const {return mPidProbProton;}
00268 inline StHbtThreeVector StHbtTrack::PGlobal() const {return mPGlobal;}
00269 inline float StHbtTrack::PtGlobal() const {return mPtGlobal;}   
00270 
00271 #endif