PHG4CylinderCellContainer.h

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

// Tell emacs that this is a C++ source
//  -*- C++ -*-.
#ifndef G4DETECTORS_PHG4CYLINDERCELLCONTAINER_H
#define G4DETECTORS_PHG4CYLINDERCELLCONTAINER_H

#include "PHG4CylinderCellDefs.h"  // for keytype

#include <phool/PHObject.h>

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

class PHG4CylinderCell;

class PHG4CylinderCellContainer : public PHObject
{
 public:
  typedef std::map<PHG4CylinderCellDefs::keytype, PHG4CylinderCell *> Map;
  typedef Map::iterator Iterator;
  typedef Map::const_iterator ConstIterator;
  typedef std::pair<Iterator, Iterator> Range;
  typedef std::pair<ConstIterator, ConstIterator> ConstRange;
  typedef std::set<int>::const_iterator LayerIter;
  typedef std::pair<LayerIter, LayerIter> LayerRange;

  PHG4CylinderCellContainer() {}

  ~PHG4CylinderCellContainer() override {}

  // from PHObject
  void identify(std::ostream &os = std::cout) const override;
  void Reset() override;

  ConstIterator AddCylinderCell(const unsigned int detid, PHG4CylinderCell *newcylinderCell);
  ConstIterator AddCylinderCellSpecifyKey(const PHG4CylinderCellDefs::keytype key, PHG4CylinderCell *newcylinderCell);

  void RemoveCylinderCell(PHG4CylinderCellDefs::keytype key)
  {
    cellmap.erase(key);
  }

  void RemoveCylinderCell(PHG4CylinderCell *cell)
  {
    Iterator its = cellmap.begin();
    Iterator last = cellmap.end();
    for (; its != last;)
    {
      if (its->second == cell)
      {
        cellmap.erase(its++);
      }
      else
      {
        ++its;
      }
    }
  }

  Iterator findOrAddCylinderCell(PHG4CylinderCellDefs::keytype key);

  PHG4CylinderCellDefs::keytype genkey(const unsigned int detid);

  ConstRange getCylinderCells(const unsigned int detid) const;

  ConstRange getCylinderCells(void) const;

  PHG4CylinderCell *findCylinderCell(PHG4CylinderCellDefs::keytype key);

  unsigned int size(void) const
  {
    return cellmap.size();
  }
  unsigned int num_layers(void) const
  {
    return layers.size();
  }
  LayerRange getLayers() const
  {
    return make_pair(layers.begin(), layers.end());
  }

  double getTotalEdep() const;

 protected:
  Map cellmap;
  std::set<int> layers;  // layers is not reset since layers must not change event by event

  ClassDefOverride(PHG4CylinderCellContainer, 1)
};

#endif