EvtRecTrack.h

Go to the documentation of this file.

#ifndef Event_EvtRecTrack_H
#define Event_EvtRecTrack_H
 
#include "GaudiKernel/ObjectVector.h"
#include "GaudiKernel/ContainedObject.h"
#include "GaudiKernel/SmartRef.h"
#include "GaudiKernel/SmartRefVector.h"
#include "EventModel/EventModel.h"
#include "MdcRecEvent/RecMdcTrack.h"
#include "MdcRecEvent/RecMdcKalTrack.h"
#include "MdcRecEvent/RecMdcDedx.h"
#include "CgemRecEvent/RecCgemTrack.h"
#include "CgemRecEvent/RecCgemSegment.h"
#include "CgemRecEvent/RecCgemKalTrack.h"
#include "CgemRecEvent/RecCgemDedx.h"
#include "ExtEvent/RecExtTrack.h"
#include "TofRecEvent/RecTofTrack.h"
#include "EmcRecEventModel/RecEmcShower.h"
#include "MucRecEvent/RecMucTrack.h"
 
using namespace EventModel;
 
extern const CLID &CLID_EvtRecTrack;
 
class EvtRecTrack:virtual public ContainedObject {
 public:
  virtual const CLID& clID() const {return  EvtRecTrack::classID();}
  static  const CLID& classID() { return CLID_EvtRecTrack;}
  
  EvtRecTrack();
  ~EvtRecTrack() {;}
  
  EvtRecTrack(const EvtRecTrack&);
  
  //get Track ID, Particle ID and Data Quality
  int trackId() const {return m_trackId;}
  int partId() const {return m_partId;}
  int quality() const { return m_quality; }
 
  bool isElectron() const { return (m_partId & 0x01); }
  bool isMuon()     const { return (m_partId & 0x02); }
  bool isPion()     const { return (m_partId & 0x04); }
  bool isKaon()     const { return (m_partId & 0x08); }
  bool isProton()   const { return (m_partId & 0x10); }
 
  //Sub detector infomation is Valid or not
  bool isMdcTrackValid() {return (m_originMdcTrack != 0);}
  bool isMdcKalTrackValid() {return (m_originMdcKalTrack != 0);}
  bool isMdcDedxValid() {return (m_originMdcDedx != 0);}
  bool isCgemTrackValid() {return (m_originCgemTrack != 0);}
  bool isCgemSegmentValid() {return (m_originCgemSegment != 0);}
  bool isCgemKalTrackValid() {return (m_originCgemKalTrack != 0);}
  bool isCgemDedxValid() {return (m_originCgemDedx != 0);}
  bool isTofTrackValid() {return (!m_originTofTrack.empty());}
  bool isEmcShowerValid() {return (m_originEmcShower!= 0);}
  bool isMucTrackValid() {return (m_originMucTrack != 0);}
  bool isExtTrackValid() {return (m_originExtTrack != 0);}
  
  //pointer to sub detector infomation
  
  RecMdcTrack* mdcTrack()  {return m_originMdcTrack;}
  RecMdcKalTrack* mdcKalTrack()  {return m_originMdcKalTrack;}
  RecMdcDedx*  mdcDedx()  {return m_originMdcDedx;}
  RecCgemTrack* cgemTrack()  {return m_originCgemTrack;}
  RecCgemSegment* cgemSegment()  {return m_originCgemSegment;}
  RecCgemKalTrack* cgemKalTrack()  {return m_originCgemKalTrack;}
  RecCgemDedx*  cgemDedx()  {return m_originCgemDedx;}
  RecExtTrack* extTrack()  {return m_originExtTrack;}
  SmartRefVector<RecTofTrack> tofTrack()  {return m_originTofTrack;}
  RecEmcShower* emcShower()  {return m_originEmcShower;}
  RecMucTrack* mucTrack()  {return m_originMucTrack;}
 
 
  // set Track ID, Particle ID and Data Quality
 
  void setTrackId(const int trkId) {m_trackId = trkId;}
  void setPartId(const int partId) {m_partId = partId;}
  void setQuality(const int quality) { m_quality = quality; }
 
  void tagElectron() { m_partId |= 0x01; }
  void tagMuon()     { m_partId |= 0x02; }
  void tagPion()     { m_partId |= 0x04; }
  void tagKaon()     { m_partId |= 0x08; }
  void tagProton()   { m_partId |= 0x10; }
  
  // set sub detector pointer
  
  void setMdcTrack(const RecMdcTrack* trk) {m_originMdcTrack = trk;}
  void setMdcKalTrack(const RecMdcKalTrack* trk) {m_originMdcKalTrack = trk;}
  void setMdcDedx(const RecMdcDedx* trk) {m_originMdcDedx = trk;}
  void setCgemTrack(const RecCgemTrack* trk) {m_originCgemTrack = trk;}
  void setCgemSegment(const RecCgemSegment* trk) {m_originCgemSegment = trk;}
  void setCgemKalTrack(const RecCgemKalTrack* trk) {m_originCgemKalTrack = trk;}
  void setCgemDedx(const RecCgemDedx* trk) {m_originCgemDedx = trk;}
  void setExtTrack(const RecExtTrack* trk) {m_originExtTrack = trk;}
  void setTofTrack(const SmartRefVector<RecTofTrack> trk) {m_originTofTrack = trk;}
  void addTofTrack(const SmartRef<RecTofTrack> trk) {m_originTofTrack.push_back(trk);}
  void setEmcShower(const RecEmcShower* shower) {m_originEmcShower = shower;}
  void setMucTrack(const RecMucTrack* trk) {m_originMucTrack = trk;}
 
  // assign operator
        
  EvtRecTrack & operator = (const EvtRecTrack &);
 
        
        
 private:
 
  int m_trackId;
  // Pid: 0x01 - electron, 0x02 - muon, 0x04 - pion, 0x08 - kaon, 0x10 - proton
  int m_partId;
  //data quality: 1 - very good, 2 - available, 3 - normal
  int m_quality;
 
  SmartRef<RecMdcTrack> m_originMdcTrack;
  SmartRef<RecMdcKalTrack> m_originMdcKalTrack;
  SmartRef<RecMdcDedx> m_originMdcDedx;
  SmartRef<RecCgemTrack> m_originCgemTrack;
  SmartRef<RecCgemSegment> m_originCgemSegment;
  SmartRef<RecCgemKalTrack> m_originCgemKalTrack;
  SmartRef<RecCgemDedx> m_originCgemDedx;
  SmartRef<RecExtTrack> m_originExtTrack;
  SmartRefVector<RecTofTrack> m_originTofTrack;
  SmartRef<RecEmcShower> m_originEmcShower;
  SmartRef<RecMucTrack> m_originMucTrack;
 
};
 
typedef ObjectVector<EvtRecTrack> EvtRecTrackCol;
 
typedef EvtRecTrackCol::iterator EvtRecTrackIterator;
 
 
 
#endif