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* getSensor( unsigned int id ) const;
    const DetGeomDesc* getSensorNoThrow( unsigned int id ) const noexcept;

    const DetGeomDesc* getRP( unsigned int id ) const;
    const DetGeomDesc* getRPNoThrow( 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 getSensorTranslation( unsigned int id ) const;

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

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

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

  protected:
    mapType sensors_map_;
    
    RPDeviceMapType rps_map_;

    mapSetType stations_in_arm_, rps_in_station_, dets_in_rp_;
};

#endif