BesEvent.h

Go to the documentation of this file.

//
// BesEvent.h
//
 
#ifndef BES_EVENT
#define BES_EVENT
 
#include <ctime>        //Long Peixun's update: time.h -> ctime
#include <TObject.h>
#include <TString.h>
#include <TDatabasePDG.h>
 
#include "BesEventHeader.h"
#include "BesGeoTrack.h"
#include "BesMcTrack.h"
 
#include "RootEventData/TDigiEvent.h"
#include "RootEventData/TRecTrackEvent.h"
#include "RootEventData/TEvtHeader.h"
#include "RootEventData/TDisTrack.h"
#include "RootEventData/TTrigEvent.h"
#include "RootEventData/TTrigData.h"
#include "RootEventData/TRecEvTime.h"
#include "RootEventData/TRecCgemCluster.h"  //Long Peixun's update: for CGEM clusters
#include "RootEventData/TMcEvent.h"
 
class BesEvent : public TObject {
 
private:
    Double_t        f_Magnetic;
    BesEventHeader      fEventHeader;         // Event header
    TDigiEvent          *fDigiEvent;
    TEvtHeader          *fEvtHeader;
    TTrigEvent          *fTrigEvent;
    TRecEvTime          *fRecEvTime;
    TMcEvent            *fMcEvent;
 
    TObjArray           *fCgemClusterCol;       //Long Peixun's update: for CGEM clusters
    TObjArray           *fCgemClusterUFCol[3];  //Long Peixun's update: for CGEM clusters in UF view
    TObjArray           *fMdcTrackCol;
    TObjArray           *fIdealTrackCol; // by Wang Liangliang for ideal tracks
    TObjArray           *fTofTrackCol;
    TObjArray           *fEmcTrackCol;
    TObjArray           *fMucTrackCol;
    TObjArray           *fExtTrackCol;
    TObjArray           *fMcParticleCol;
 
    bool        Is_tofBarrel(UInt_t status);
    bool        Is_tofCounter(UInt_t status);
public:
    BesEvent();
    virtual ~BesEvent();
 
    virtual void Clear(Option_t *option = "");
    virtual void Delete(Option_t *option = "");
    virtual void Print(Option_t *option = "");
 
    //virtual void SetEvent(TDigiEvent *digiEvent, TDisTrack *recEvent, TEvtHeader *evtHeader, TTrigEvent *trigEvent);
    //virtual void SetEvent(TDigiEvent *digiEvent, TDisTrack *recEvent, TEvtHeader *evtHeader, TRecEvTime *recEvTime);//, TTrigEvent *trigEvent);
    virtual void SetEvent(TDigiEvent *digiEvent, TDisTrack *recEvent, TEvtHeader *evtHeader, TRecEvTime *recEvTime, Bool_t isRec, TMcEvent *mcEvt = 0);//, TTrigEvent *trigEvent);
    virtual void SetHits();
    virtual void ClearHits();   //Long Peixun's update: Clear all hits in detector
 
    virtual void SetCgemClusters(const TRecCgemCluster* recCluster, int ncluster);  //Long Peixun's update: for CGEM clusters
    virtual void SetTracks(TDisTrack *recEvent);
    virtual void SetMdcTracks(TDisTrack *recEvent);
    virtual void SetTofTracks(TDisTrack *recEvent);
    virtual void SetEmcShowers(TDisTrack *recEvent);
    virtual void SetMucTracks(TDisTrack *recEvent);
    virtual void SetExtTracks(TDisTrack *recEvent);
    virtual void SetMcEvent(TMcEvent *mcEvt);
    virtual void ConstructMdcTrackFromRec(BesGeoTrack *mdcTrack, const TRecMdcTrack *recTrack,TDisTrack *recEvent);
    virtual void ConstructMucTrackFromRec(BesGeoTrack *mucTrack, const TRecMucTrack *recTrack);
    virtual void ConstructEmcTrackFromRec(BesGeoTrack *emcTrack, const TRecEmcShower *recTrack);
    //virtual void ConstructTofTrackFromRec(BesGeoTrack *tofTrack, const TRecBTofTrack *recBTrack);
    //virtual void ConstructTofTrackFromRec(BesGeoTrack *tofTrack, const TRecETofTrack *recETrack);
    virtual void ConstructTofTrackFromRec(BesGeoTrack *tofTrack, const TRecTofTrack *recTrack,TDisTrack *recEvent);
    virtual void ConstructMcTrack(BesMcTrack *mcTrack, const TMcParticle *mcPart);
    //virtual void ConstructExtTrackFromRec(BesGeoTrack *extTrack, const TExtTrack *recTrack);
    virtual void DrawHits(Option_t *option);
    virtual void DrawTracks(Option_t *option);
    virtual void DrawMcParticles(Option_t *option);
    void SetMagnetic(Double_t input) {f_Magnetic = input;}
 
    virtual void DrawCgem3DHits(Option_t* option);      //Long Peixun's update: hits in CGEM 3D display
    virtual void DrawCgemUFHits(Option_t* option);      //Long Peixun's update: hits in CGEM unfolded display
    virtual void DrawCgemClusters(Option_t* option);    //Long Peixun's update: clusters in CGEM display
    virtual void DrawCgemUFClusters(int layer, Option_t* option);  //Long Peixun's update: clusters in CGEM unfolded display
    
 
    //Long Peixun's update: for CGEM hits array
    const TObjArray* GetCgemDigiCol() const
    {
        if (fDigiEvent) return fDigiEvent->getCgemDigiCol();
        else return 0;
    }
    const Int_t      GetCgemDigiNum() const      {
        if (fDigiEvent) return fDigiEvent->getCgemDigiCol()->GetEntries();
        else return -1;
    }
    const TCgemDigi*  GetCgemDigi(Int_t i) const  {
        if (fDigiEvent) return fDigiEvent->getCgemDigi(i);
        else return 0;
    }
 
    const TObjArray* GetMdcDigiCol() const      {
        if (fDigiEvent) return fDigiEvent->getMdcDigiCol();
        else return 0;
    }
    const TObjArray* GetTofDigiCol() const      {
        if (fDigiEvent) return fDigiEvent->getTofDigiCol();
        else return 0;
    }
    const TObjArray* GetEmcDigiCol() const      {
        if (fDigiEvent) return fDigiEvent->getEmcDigiCol();
        else return 0;
    }
    const TObjArray* GetMucDigiCol() const      {
        if (fDigiEvent) return fDigiEvent->getMucDigiCol();
        else return 0;
    }
    const Int_t      GetMdcDigiNum() const      {
        if (fDigiEvent) return fDigiEvent->getMdcDigiCol()->GetEntries();
        else return -1;
    }
    const Int_t      GetTofDigiNum() const      {
        if (fDigiEvent) return fDigiEvent->getTofDigiCol()->GetEntries();
        else return -1;
    }
    const Int_t      GetEmcDigiNum() const      {
        if (fDigiEvent) return fDigiEvent->getEmcDigiCol()->GetEntries();
        else return -1;
    }
    const Int_t      GetMucDigiNum() const      {
        if (fDigiEvent) return fDigiEvent->getMucDigiCol()->GetEntries();
        else return -1;
    }
    const TMdcDigi*  GetMdcDigi(Int_t i) const  {
        if (fDigiEvent) return fDigiEvent->getMdcDigi(i);
        else return 0;
    }   //Long Peixun's update: Remove semicolon
    const TTofDigi*  GetTofDigi(Int_t i) const  {
        if (fDigiEvent) return fDigiEvent->getTofDigi(i);
        else return 0;
    }
    const TEmcDigi*  GetEmcDigi(Int_t i) const  {
        if (fDigiEvent) return fDigiEvent->getEmcDigi(i);
        else return 0;
    }
    const TMucDigi*  GetMucDigi(Int_t i) const  {
        if (fDigiEvent) return fDigiEvent->getMucDigi(i);
        else return 0;
    }
 
    //Long Peixun TODO: Add CGEM tracks code
 
    const TObjArray* GetMdcTrackCol(TDisTrack *recEvent) const       {
        if (recEvent)
            return recEvent->getRecMdcTrackCol();
        else return 0;
    }
    //const TObjArray* GetBTofTrackCol() const       {
    //    if (fRecEvent)
    //        return fRecEvent->getTofTrackCol();
    //    else return 0;
    //}
    const TObjArray* GetTofTrackCol(TDisTrack *recEvent) const       {
        if (recEvent)
            return recEvent->getTofTrackCol();
        else 
      return 0;
    }
    const TObjArray* GetEmcShowerCol(TDisTrack *recEvent) const       {
      //  if (recEvent)
      //      return recEvent->getEmcShowerCol();
      //  else 
      return 0;
    }
    const TObjArray* GetMucTrackCol(TDisTrack *recEvent) const       {
        if (recEvent)
            return recEvent->getMucTrackCol();
        else 
        return 0;
    }
    const TObjArray* GetExtTrackCol() const       {
         return 0;  // for test 
        //if (fRecEvent)
        //    return fRecEvent->getExtTrackCol();
        //else return 0;
    }
    const TObjArray* GetDedxCol(TDisTrack *recEvent)     const       {
     //   if (recEvent)
      //      return recEvent->getRecMdcDedxCol();
      //  else 
      return 0;
    }
    const Int_t      GetMdcTrackNum(TDisTrack *recEvent) const       {
       if (recEvent)
//            return recEvent->getRecMdcTrackCol()->GetEntries();
       return recEvent->getMdcTrackNum();
     else 
      return -1;
    }
    //const Int_t      GetBTofTrackNum() const       {
    //    if (fRecEvent)
    //        return fRecEvent->getTofTrackCol()->GetEntries();
    //    else return -1;
    //};
    const Int_t      GetTofTrackNum(TDisTrack *recEvent) const       {
        if (recEvent)
   //         return recEvent->getTofTrackCol()->GetEntries();
       return recEvent->getTofTrackNum();
        else 
      return -1;
    }
    const Int_t      GetEmcShowerNum(TDisTrack *recEvent) const       {
      if (recEvent)
   //     return recEvent->getEmcShowerCol()->GetEntries();
     return recEvent->getEmcShowerNum(); 
       else 
      return -1;
    }
    const Int_t      GetMucTrackNum(TDisTrack *recEvent) const       {
        if (recEvent)
         return recEvent->getMucTrackNum();
     //       return recEvent->getMucTrackCol()->GetEntries();
        else 
      return -1;
    }
    const Int_t      GetExtTrackNum() const       {
        return -1;// for test 
        //if (fRecEvent)
        //    return fRecEvent->getExtTrackCol()->GetEntries();
        //else return -1;
    }
    const Int_t      GetDedxNum(TDisTrack *recEvent)     const       {
      //  if (fRecEvent)
      //      return fRecEvent->getRecMdcDedxCol()->GetEntries();
       // else 
      return -1;
    }
    const TRecMdcTrack*  GetMdcTrack(Int_t i,TDisTrack *recEvent) const  {
        if (recEvent)
            return recEvent->getRecMdcTrack(i);
        else return 0;
    }
    //const TRecBTofTrack*  GetBTofTrack(Int_t i) const  {
    //    if (fRecEvent)
    //        return fRecEvent->getBTofTrack(i);
    //    else return 0;
    //};
    const TRecTofTrack*  GetTofTrack(Int_t i,TDisTrack *recEvent) const  {
        if (recEvent)
            return recEvent->getTofTrack(i);
        else
      return 0;
    }
    const TRecEmcShower*  GetEmcShower(Int_t i,TDisTrack *recEvent) const  {
        if (recEvent)
            return recEvent->getEmcShower(i);
        else 
      return 0;
    }
    const TRecMucTrack*  GetMucTrack(Int_t i,TDisTrack *recEvent) const  {
        if (recEvent)
           return recEvent->getMucTrack(i);
        else 
      return 0;
    }
    //const TExtTrack*  GetExtTrack(Int_t i) const  {
    //    if (fRecEvent)
    //        return fRecEvent->getExtTrack(i);
    //    else return 0;
    //};
    const TRecMdcDedx*      GetDedx(Int_t i)     const  {
       // if (fRecEvent)
       //     return fRecEvent->getRecMdcDedx(i);
      //  else
      return 0;
    }
 
    // header access functions
    inline Long64_t GetRun()   {
        return fEventHeader.GetRunNumber();
    }
    inline Long64_t GetEvent() {
        return fEventHeader.GetEventNumber();
    }
    inline Int_t GetYear()  {
        return fEventHeader.GetEventYear();
    }
    inline Int_t GetMonth() {
        return fEventHeader.GetEventMonth();
    }
    inline Int_t GetDay()   {
        return fEventHeader.GetEventDay();
    }
    inline Int_t GetHour()  {
        return fEventHeader.GetEventHour();
    }
    inline Int_t GetMin()   {
        return fEventHeader.GetEventMin();
    }
    inline Int_t GetSec()   {
        return fEventHeader.GetEventSec();
    }
    inline Int_t GetMC()    {
        return fEventHeader.GetEventMC();
    }
    inline BesEventHeader GetHeader(){
        return fEventHeader;
    }
 
    TDatabasePDG* pdgTable;
    ClassDef(BesEvent,0) // BesVis Event
};
 
R__EXTERN BesEvent *gEvent;
 
#endif