GEMGeometry.h

Go to the documentation of this file.

#ifndef GEMGeometry_GEMGeometry_h
#define GEMGeometry_GEMGeometry_h

#include "DataFormats/DetId/interface/DetId.h"
#include "Geometry/CommonDetUnit/interface/TrackingGeometry.h"
#include "Geometry/GEMGeometry/interface/GEMEtaPartition.h"
#include "Geometry/GEMGeometry/interface/GEMChamber.h"
#include "Geometry/GEMGeometry/interface/GEMSuperChamber.h"
#include "Geometry/GEMGeometry/interface/GEMRing.h"
#include "Geometry/GEMGeometry/interface/GEMStation.h"
#include "Geometry/GEMGeometry/interface/GEMRegion.h"
#include <vector>
#include <map>

class GeomDetType;

class GEMGeometry : public TrackingGeometry {

 public:
  GEMGeometry();

  virtual ~GEMGeometry();

  // Return a vector of all det types
  virtual const DetTypeContainer&  detTypes() const override;

  // Return a vector of all GeomDetUnit
  virtual const DetUnitContainer& detUnits() const override;

  // Return a vector of all GeomDet
  virtual const DetContainer& dets() const override;
  
  // Return a vector of all GeomDetUnit DetIds
  virtual const DetIdContainer& detUnitIds() const override;

  // Return a vector of all GeomDet DetIds
  virtual const DetIdContainer& detIds() const override;

  // Return the pointer to the GeomDetUnit corresponding to a given DetId
  virtual const GeomDetUnit* idToDetUnit(DetId) const override;

  // Return the pointer to the GeomDet corresponding to a given DetId
  virtual const GeomDet* idToDet(DetId) const override;


  //---- Extension of the interface

  const std::vector<const GEMRegion*>& regions() const;

  const std::vector<const GEMStation*>& stations() const;

  const std::vector<const GEMRing*>& rings() const;

  const std::vector<const GEMSuperChamber*>& superChambers() const;

  const std::vector<const GEMChamber*>& chambers() const;

  const std::vector<const GEMEtaPartition*>& etaPartitions() const;

  // Return a GEMRegion 
  const GEMRegion* region(int region) const;

  // Return a GEMStation
  const GEMStation* station(int region, int station) const;

  const GEMRing* ring(int region, int station, int ring) const;

  // Return a GEMSuperChamber given its id
  const GEMSuperChamber* superChamber(GEMDetId id) const;

  // Return a GEMChamber given its id
  const GEMChamber* chamber(GEMDetId id) const;

  const GEMEtaPartition* etaPartition(GEMDetId id) const;

  void add(GEMRegion* region);

  void add(GEMStation* station);

  void add(GEMRing* ring);

  void add(GEMSuperChamber* sch);
 
  void add(GEMChamber* ch);

  void add(GEMEtaPartition* etaPartition);

 private:
  DetUnitContainer theEtaPartitions;
  DetContainer theDets;
  DetTypeContainer theEtaPartitionTypes;
  DetIdContainer theEtaPartitionIds;
  DetIdContainer theDetIds;
  
  // Map for efficient lookup by DetId 
  mapIdToDet theMap;

  std::vector<const GEMEtaPartition*> allEtaPartitions; // Are not owned by this class; are owned by their chamber.
  std::vector<const GEMChamber*> allChambers; // Are not owned by this class; are owned by their chamber.
  std::vector<const GEMSuperChamber*> allSuperChambers; // Are owned by this class.
  std::vector<const GEMRing*> allRings; // Are owned by this class.
  std::vector<const GEMStation*> allStations; // Are owned by this class.
  std::vector<const GEMRegion*> allRegions; // Are owned by this class.
};

#endif