StGmtCollection.h

#ifndef StGmtCollection_hh
#define StGmtCollection_hh

// StRoot headers
#include "StObject.h"
#include "StGmtStripCollection.h"
#include "StGmtHitCollection.h"
#include "StGmtPointCollection.h"
#include "StEnumerations.h"

//________________
class StGmtCollection : public StObject {
 public:
  StGmtCollection();
  // StGmtCollection( const StGmtCollection& other );            ---> use default
  // StGmtCollection& operator=( const StGmtCollection& other ); ---> use default 
    
  ~StGmtCollection();

  size_t getNumModules() const                             { return kGmtNumModules; }
  size_t getNumStrips() const;
  size_t getNumStrips( unsigned short moduleIdx) const
  { return (moduleIdx < kGmtNumModules ? mStripCollection[moduleIdx].getNumStrips() : 0 ); }
  size_t getNumHits() const;
  size_t getNumHits( unsigned short moduleIdx ) const
  { return (moduleIdx < kGmtNumModules ? mHitCollection[moduleIdx].getNumHits() : 0 ); }
  size_t getNumPoints() const
  { return mPointCollection.getNumPoints(); }
    
  // note: ownership of all pointers is retained by the containers.
  // Do not delete any pointers received from this class.
  
  StGmtStripCollection* getStripCollection( unsigned short moduleIdx )
  { return (moduleIdx < kGmtNumModules ? &mStripCollection[moduleIdx] : 0 ); }
  const StGmtStripCollection* getStripCollection( unsigned short moduleIdx ) const
  { return (moduleIdx < kGmtNumModules ? &mStripCollection[moduleIdx] : 0 ); }

  StGmtHitCollection* getHitCollection( unsigned short moduleIdx )
  { return (moduleIdx < kGmtNumModules ? &mHitCollection[moduleIdx] : 0 ); }
  const StGmtHitCollection* getHitCollection( unsigned short moduleIdx ) const
  { return (moduleIdx < kGmtNumModules ? &mHitCollection[moduleIdx] : 0 ); }
  
  StGmtPointCollection* getPointCollection()             { return &mPointCollection; }
  const StGmtPointCollection* getPointCollection() const { return &mPointCollection; }
  
  void Clear( Option_t *opts = "" );
    
 protected:

  friend class StMuDstMaker;

  StGmtStripCollection mStripCollection[kGmtNumModules];
  StGmtHitCollection mHitCollection[kGmtNumModules];
  StGmtPointCollection mPointCollection;
    
 private:   
  ClassDef(StGmtCollection,1)
}; 

#endif // #define StGmtCollection_hh