CTPPSGeometry.h

Go to the documentation of this file.

/****************************************************************************
*
* Authors:
*  Jan Kašpar (jan.kaspar@gmail.com)
*
****************************************************************************/

#ifndef Geometry_VeryForwardGeometryBuilder_CTPPSGeometry
#define Geometry_VeryForwardGeometryBuilder_CTPPSGeometry

#include "DataFormats/CTPPSDetId/interface/CTPPSDetId.h"

#include "Geometry/VeryForwardGeometryBuilder/interface/DetGeomDesc.h"
#include "Geometry/VeryForwardGeometryBuilder/interface/CTPPSDDDNames.h"

#include "CLHEP/Vector/ThreeVector.h"
#include "CLHEP/Vector/Rotation.h"

#include <map>
#include <set>

class CTPPSGeometry {
public:
  typedef std::map<unsigned int, const DetGeomDesc*> mapType;
  typedef std::map<int, const DetGeomDesc*> RPDeviceMapType;
  typedef std::map<unsigned int, std::set<unsigned int> > mapSetType;

  CTPPSGeometry() {}
  ~CTPPSGeometry() {}

  CTPPSGeometry(const DetGeomDesc* gd) { build(gd); }

  void build(const DetGeomDesc*);

  //----- setters and getters

  bool addSensor(unsigned int, const DetGeomDesc*&);

  bool addRP(unsigned int id, const DetGeomDesc*&);

  const DetGeomDesc* sensor(unsigned int id) const;
  const DetGeomDesc* sensorNoThrow(unsigned int id) const noexcept;

  const DetGeomDesc* rp(unsigned int id) const;
  const DetGeomDesc* rpNoThrow(unsigned int id) const noexcept;

  //----- objects iterators

  mapType::const_iterator beginSensor() const { return sensors_map_.begin(); }
  mapType::const_iterator endSensor() const { return sensors_map_.end(); }

  RPDeviceMapType::const_iterator beginRP() const { return rps_map_.begin(); }
  RPDeviceMapType::const_iterator endRP() const { return rps_map_.end(); }

  //----- translators

  CLHEP::Hep3Vector localToGlobal(const DetGeomDesc*, const CLHEP::Hep3Vector&) const;
  CLHEP::Hep3Vector globalToLocal(const DetGeomDesc*, const CLHEP::Hep3Vector&) const;
  CLHEP::Hep3Vector localToGlobal(unsigned int, const CLHEP::Hep3Vector&) const;
  CLHEP::Hep3Vector globalToLocal(unsigned int, const CLHEP::Hep3Vector&) const;

  CLHEP::Hep3Vector localToGlobalDirection(unsigned int id, const CLHEP::Hep3Vector&) const;
  CLHEP::Hep3Vector globalToLocalDirection(unsigned int id, const CLHEP::Hep3Vector&) const;

  CLHEP::Hep3Vector sensorTranslation(unsigned int id) const;

  CLHEP::Hep3Vector rpTranslation(unsigned int id) const;

  const std::set<unsigned int>& stationsInArm(unsigned int) const;

  const std::set<unsigned int>& rpsInStation(unsigned int) const;

  const std::set<unsigned int>& sensorsInRP(unsigned int) const;

protected:
  mapType sensors_map_;

  RPDeviceMapType rps_map_;

  mapSetType stations_in_arm_, rps_in_station_, dets_in_rp_;
};

#endif