StPicoDst.h

#ifndef StPicoDst_h
#define StPicoDst_h

// ROOT headers
#include "TClonesArray.h"

// PicoDst headers
#include "StPicoArrays.h"

// Forward declarations
class StPicoEvent;
class StPicoTrack;
class StPicoEmcTrigger;
class StPicoMtdTrigger;
class StPicoBTowHit;
class StPicoBTofHit;
class StPicoMtdHit;
class StPicoEpdHit;
class StPicoBbcHit;
class StPicoFmsHit;
class StPicoBEmcPidTraits;
class StPicoBTofPidTraits;
class StPicoMtdPidTraits;
class StPicoTrackCovMatrix;
class StPicoBEmcSmdEHit;
class StPicoBEmcSmdPHit;
class StPicoETofHit;
class StPicoETofPidTraits;
class StPicoMcVertex;
class StPicoMcTrack;

//_________________
class StPicoDst {

 public:

#if defined (__TFG__VERSION__)
  StPicoDst() { fgPicoDst = this;}
  virtual ~StPicoDst() {fgPicoDst = 0;}
  virtual Bool_t IsGoodTrigger() const;
#else /* ! __TFG__VERSION__ */
  StPicoDst() { /* emtpy */}
  ~StPicoDst() { /* empty*/ }
#endif

  static void set(TClonesArray**);
  static void unset();
  static TClonesArray* picoArray(Int_t type) { return picoArrays[type]; }

  static StPicoEvent* event() { return (StPicoEvent*)picoArrays[StPicoArrays::Event]->UncheckedAt(0); }
  static StPicoTrack* track(Int_t i) { return (StPicoTrack*)picoArrays[StPicoArrays::Track]->UncheckedAt(i); }
  static StPicoEmcTrigger* emcTrigger(Int_t i) { return (StPicoEmcTrigger*)picoArrays[StPicoArrays::EmcTrigger]->UncheckedAt(i); }
  static StPicoMtdTrigger* mtdTrigger(Int_t i) { return (StPicoMtdTrigger*)picoArrays[StPicoArrays::MtdTrigger]->UncheckedAt(i); }
  static StPicoBTowHit* btowHit(Int_t i) { return (StPicoBTowHit*)picoArrays[StPicoArrays::BTowHit]->UncheckedAt(i); }
  static StPicoBTofHit* btofHit(Int_t i) { return (StPicoBTofHit*)picoArrays[StPicoArrays::BTofHit]->UncheckedAt(i); }
  static StPicoMtdHit*  mtdHit(Int_t i) { return (StPicoMtdHit*)picoArrays[StPicoArrays::MtdHit]->UncheckedAt(i); }
  static StPicoBbcHit* bbcHit(Int_t i) { return (StPicoBbcHit*)picoArrays[StPicoArrays::BbcHit]->UncheckedAt(i); }
  static StPicoEpdHit* epdHit(Int_t i) { return (StPicoEpdHit*)picoArrays[StPicoArrays::EpdHit]->UncheckedAt(i); }
  static StPicoFmsHit*  fmsHit(Int_t i) { return (StPicoFmsHit*)picoArrays[StPicoArrays::FmsHit]->UncheckedAt(i); }
  static StPicoBEmcPidTraits* bemcPidTraits(Int_t i) { return (StPicoBEmcPidTraits*)picoArrays[StPicoArrays::BEmcPidTraits]->UncheckedAt(i); }
  static StPicoBTofPidTraits* btofPidTraits(Int_t i) { return (StPicoBTofPidTraits*)picoArrays[StPicoArrays::BTofPidTraits]->UncheckedAt(i); }
  static StPicoMtdPidTraits* mtdPidTraits(Int_t i) { return (StPicoMtdPidTraits*)picoArrays[StPicoArrays::MtdPidTraits]->UncheckedAt(i); }
  static StPicoTrackCovMatrix* trackCovMatrix(Int_t i) { return (StPicoTrackCovMatrix*)picoArrays[StPicoArrays::TrackCovMatrix]->UncheckedAt(i); }
  static StPicoBEmcSmdEHit* bemcSmdEHit(Int_t i) { return (StPicoBEmcSmdEHit*)picoArrays[StPicoArrays::BEmcSmdEHit]->UncheckedAt(i); }
  static StPicoBEmcSmdPHit* bemcSmdPHit(Int_t i) { return (StPicoBEmcSmdPHit*)picoArrays[StPicoArrays::BEmcSmdPHit]->UncheckedAt(i); }
  static StPicoETofHit* etofHit(Int_t i) { return (StPicoETofHit*)picoArrays[StPicoArrays::ETofHit]->UncheckedAt(i); }
  static StPicoETofPidTraits* etofPidTraits(Int_t i) { return (StPicoETofPidTraits*)picoArrays[StPicoArrays::ETofPidTraits]->UncheckedAt(i); }
  static StPicoMcVertex* mcVertex(Int_t i) { return (StPicoMcVertex*)picoArrays[StPicoArrays::McVertex]->UncheckedAt(i); }
  static StPicoMcTrack* mcTrack(Int_t i) { return (StPicoMcTrack*)picoArrays[StPicoArrays::McTrack]->UncheckedAt(i); }

  static UInt_t numberOfTracks() { return picoArrays[StPicoArrays::Track]->GetEntriesFast(); }
  static UInt_t numberOfEmcTriggers() { return picoArrays[StPicoArrays::EmcTrigger]->GetEntriesFast(); }
  static UInt_t numberOfMtdTriggers() { return picoArrays[StPicoArrays::MtdTrigger]->GetEntriesFast(); }
  static UInt_t numberOfBTowHits() { return picoArrays[StPicoArrays::BTowHit]->GetEntriesFast(); }
  static UInt_t numberOfBTofHits() { return picoArrays[StPicoArrays::BTofHit]->GetEntriesFast(); }
  static UInt_t numberOfMtdHits() { return picoArrays[StPicoArrays::MtdHit]->GetEntriesFast(); }
  static UInt_t numberOfBbcHits() { return picoArrays[StPicoArrays::BbcHit]->GetEntriesFast(); }
  static UInt_t numberOfEpdHits() { return picoArrays[StPicoArrays::EpdHit]->GetEntriesFast(); }
  static UInt_t numberOfFmsHits() { return picoArrays[StPicoArrays::FmsHit]->GetEntriesFast(); }
  static UInt_t numberOfBEmcPidTraits() { return picoArrays[StPicoArrays::BEmcPidTraits]->GetEntriesFast(); }
  static UInt_t numberOfBTofPidTraits() { return picoArrays[StPicoArrays::BTofPidTraits]->GetEntriesFast(); }
  static UInt_t numberOfMtdPidTraits() { return picoArrays[StPicoArrays::MtdPidTraits]->GetEntriesFast(); }
  static UInt_t numberOfTrackCovMatrices() { return picoArrays[StPicoArrays::TrackCovMatrix]->GetEntriesFast(); }
  static UInt_t numberOfBEmcSmdEHits() { return picoArrays[StPicoArrays::BEmcSmdEHit]->GetEntriesFast(); }
  static UInt_t numberOfBEmcSmdPHits() { return picoArrays[StPicoArrays::BEmcSmdPHit]->GetEntriesFast(); }
  static UInt_t numberOfETofHits() { return picoArrays[StPicoArrays::ETofHit]->GetEntriesFast(); }
  static UInt_t numberOfETofPidTraits() { return picoArrays[StPicoArrays::ETofPidTraits]->GetEntriesFast(); }
  static UInt_t numberOfMcVertices() { return picoArrays[StPicoArrays::McVertex]->GetEntriesFast(); }
  static UInt_t numberOfMcTracks() { return picoArrays[StPicoArrays::McTrack]->GetEntriesFast(); }

  void print() const;
  static void printTracks();
  static void printTriggers();
  static void printBTowHits();
  static void printBTofHits();
  static void printMtdHits();
  static void printFmsHits();
  static void printBEmcPidTraits();
  static void printBTofPidTraits();
  static void printMtdPidTraits();
  static void printTrackCovMatrices();
  static void printBEmcSmdEHits();
  static void printBEmcSmdPHits();
  static void printETofHits();
  static void printETofPidTraits();
  static void printMcVertices();
  static void printMcTracks();

#if defined (__TFG__VERSION__)
  static StPicoDst *instance() {return fgPicoDst;}
#endif /* __TFG__VERSION__ */

 private:

  static TClonesArray** picoArrays;

#if defined (__TFG__VERSION__)
  static StPicoDst *fgPicoDst; 
#endif /* __TFG__VERSION__ */
};

#endif