TrkrHitSetContainer.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file TrkrHitSetContainer.h

#ifndef TRACKBASE_TRKRHITSETCONTAINER_H
#define TRACKBASE_TRKRHITSETCONTAINER_H

#include "TrkrDefs.h"        // for hitsetkey, TrkrId

#include <phool/PHObject.h>

#include <iostream>          // for cout, ostream
#include <map>
#include <utility>           // for pair

class TrkrHitSet;

class TrkrHitSetContainer : public PHObject
{
 public:
  
  using Map = std::map<TrkrDefs::hitsetkey, TrkrHitSet *>;
  using Iterator = Map::iterator;
  using ConstIterator = Map::const_iterator;
  using Range = std::pair<Iterator, Iterator>;
  using ConstRange = std::pair<ConstIterator, ConstIterator>;

  ~TrkrHitSetContainer() override = default;

  void Reset()  override;
  
  virtual ConstIterator addHitSet(TrkrHitSet*);

  virtual ConstIterator addHitSetSpecifyKey(const TrkrDefs::hitsetkey, TrkrHitSet*);

  virtual void removeHitSet(TrkrDefs::hitsetkey)
  {}

  virtual void removeHitSet(TrkrHitSet*)
  {}

  virtual Iterator findOrAddHitSet(TrkrDefs::hitsetkey);

  virtual ConstRange getHitSets(const TrkrDefs::TrkrId) const;

  virtual ConstRange getHitSets(const TrkrDefs::TrkrId trackerid, const uint8_t layer) const;

  virtual ConstRange getHitSets() const;

  virtual TrkrHitSet *findHitSet(TrkrDefs::hitsetkey)
  { return nullptr; }

  virtual unsigned int size() const
  { return 0; }

  protected:
  TrkrHitSetContainer() = default;

  private:

  ClassDefOverride(TrkrHitSetContainer, 1)
};

#endif //TRACKBASE_TRKRHITSETCONTAINER_H