StGmtStripCollection.h

#ifndef StGmtStripCollection_hh
#define StGmtStripCollection_hh

// STAR headers
#include "StObject.h"
#include "StContainers.h"
#include "StGmtStrip.h"

//________________
class StGmtStripCollection : public StObject {
 public:
  StGmtStripCollection( short module = 0 );
    
  ~StGmtStripCollection();
    
  // WARNING: never use getStripVec().push_back() or equivelants. 
  // Instead use StGmtStripCollection::getStrip to add a new strip.

  StSPtrVecGmtStrip& getStripVec()             { return mStripVec; }
  const StSPtrVecGmtStrip& getStripVec() const { return mStripVec; }
    
    // sort internal vector by geoId
    void sortByGeoId();

    // sort internal vector by coordinate number
    void sortByCoord();
    // sort internal vector by coordinate number
    void partialSortByCoord();

    // sort internal vector by layer (X first then Y)
    void sortByLayer();

    // remove all hits with negative geoIds
    void removeFlagged();
    
    // size of internal vector
    size_t getNumStrips() const;
    
    // modify/access the moduleId
    short getModule() const;
    void setModule( short module );
    
    // Clear
    void Clear( Option_t *opt = "" );
    
    // Get pointer to a strip -- note: this is the only way to modify a
    // strip.  New strip is created if it does not exist, but only
    // using StGmtStrip() constructor.  Ownership is retained by the
    // collection.

    StGmtStrip* getStrip( int Id );
    StGmtStrip* getSortedStrip( int Id );
    
 protected:  
  static bool hitGeoIdLessThan( const StGmtStrip* h1, const StGmtStrip* h2 ); 
  static bool hitCoordLessThan( const StGmtStrip* h1, const StGmtStrip* h2 ); 
  static bool hitLayerLessThan( const StGmtStrip* h1, const StGmtStrip* h2 ); 
     
  Short_t mModule;
  StSPtrVecGmtStrip mStripVec;    
    
  StPtrVecGmtStrip mStripElecIdVec;
  StPtrVecGmtStrip mStripGeoIdVec; 
    
 private:   
  ClassDef(StGmtStripCollection,1)
}; 

inline size_t StGmtStripCollection::getNumStrips() const {
    return mStripVec.size();
};

inline void StGmtStripCollection::setModule( short moduleId ) {
    mModule = moduleId;
};

inline short StGmtStripCollection::getModule() const {
    return mModule;
};

#endif // #define StGmtStripCollection_hh