TrkrClusterContainer.h

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

#ifndef TRACKBASE_TRKRCLUSTERCONTAINER_H
#define TRACKBASE_TRKRCLUSTERCONTAINER_H

#include "TrkrDefs.h"

#include <phool/PHObject.h>

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

class TrkrCluster;

class TrkrClusterContainer : public PHObject
{
 public:


  using Map = std::map<TrkrDefs::cluskey, TrkrCluster *>;
  using Iterator = Map::iterator;
  using ConstIterator = Map::const_iterator;
  using Range = std::pair<Iterator, Iterator>;
  using ConstRange = std::pair<ConstIterator, ConstIterator>;

  void Reset() override {}

  void identify(std::ostream &/*os*/ = std::cout) const override {}

  virtual ConstIterator addCluster(TrkrCluster*);

  virtual ConstIterator addClusterSpecifyKey(const TrkrDefs::cluskey, TrkrCluster* );

  virtual void removeCluster(TrkrDefs::cluskey) {}

  virtual void removeCluster(TrkrCluster* ) {}

  virtual Iterator findOrAddCluster(TrkrDefs::cluskey);
  
  virtual ConstRange getClusters() const;

  virtual ConstRange getClusters(TrkrDefs::hitsetkey) const;

  virtual Map* getClusterMap(TrkrDefs::hitsetkey) { return nullptr; }

  virtual TrkrCluster* findCluster(TrkrDefs::cluskey) const { return nullptr; }

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

  protected:
  TrkrClusterContainer() = default;

  private:

  ClassDefOverride(TrkrClusterContainer, 1)

};

#endif //TRACKBASE_TRKRCLUSTERCONTAINER_H