g4simulation/g4detectors/PHG4BlockGeomContainer.h

0001 // Tell emacs that this is a C++ source
0002 //  -*- C++ -*-.
0003 #ifndef G4DETECTORS_PHG4BLOCKGEOMCONTAINER_H
0004 #define G4DETECTORS_PHG4BLOCKGEOMCONTAINER_H
0005
0006 #include <phool/PHObject.h>
0007
0008 #include <iostream>  // for cout, ostream
0009 #include <map>
0010 #include <utility>  // for make_pair, pair
0011
0012 class PHG4BlockGeom;
0013
0014 class PHG4BlockGeomContainer : public PHObject
0015 {
0016  public:
0017   typedef std::map<int, PHG4BlockGeom *> Map;
0018   typedef Map::iterator Iterator;
0019   typedef Map::const_iterator ConstIterator;
0020   typedef std::pair<Iterator, Iterator> Range;
0021   typedef std::pair<ConstIterator, ConstIterator> ConstRange;
0022
0023   PHG4BlockGeomContainer();
0024   ~PHG4BlockGeomContainer() override;
0025
0026   // from PHObject
0027   void identify(std::ostream &os = std::cout) const override;
0028
0029   int AddLayerGeom(const int i, PHG4BlockGeom *mygeom);
0030   int AddLayerGeom(PHG4BlockGeom *mygeom);
0031   PHG4BlockGeom *GetLayerGeom(const int i);
0032   int get_NLayers() const { return _layergeoms.size(); }
0033   std::pair<std::map<int, PHG4BlockGeom *>::const_iterator, std::map<int, PHG4BlockGeom *>::const_iterator> get_begin_end() const { return std::make_pair(_layergeoms.begin(), _layergeoms.end()); }
0034
0035  protected:
0036   std::map<int, PHG4BlockGeom *> _layergeoms;
0037   float _magfield;
0038
0039   ClassDefOverride(PHG4BlockGeomContainer, 1)
0040 };
0041
0042 #endif