TotemRPGeometry.h

Go to the documentation of this file.

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

#ifndef Geometry_VeryForwardGeometryBuilder_TotemRPGeometry
#define Geometry_VeryForwardGeometryBuilder_TotemRPGeometry

#include "DataFormats/CTPPSDetId/interface/TotemRPDetId.h"
#include "DataFormats/CTPPSDetId/interface/CTPPSDiamondDetId.h"
#include "CLHEP/Vector/ThreeVector.h"
#include "CLHEP/Vector/Rotation.h"
#include "Geometry/VeryForwardGeometryBuilder/interface/DetGeomDesc.h"

#include <map>
#include <set>

class DetId;

class TotemRPGeometry
{
  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;

    TotemRPGeometry() {}
    ~TotemRPGeometry() {}

    TotemRPGeometry(const DetGeomDesc * gd)
    {
      Build(gd);
    }

    char Build(const DetGeomDesc *);          

    char AddDetector(unsigned int, const DetGeomDesc * &);

    char AddRPDevice(unsigned int id, const DetGeomDesc * &det_geom_desc);

    const DetGeomDesc *GetDetector(unsigned int) const;

    const DetGeomDesc* GetDetector( const TotemRPDetId& id ) const
    {
      return GetDetector( id.rawId() );
    }
    const DetGeomDesc* GetDetector( const CTPPSDetId& id ) const
    {
      return GetDetector( id.rawId() );
    }

    DetGeomDesc const *operator[] (unsigned int id) const
    {
      return GetDetector(id);
    }

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

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

    const DetGeomDesc *GetRPDevice(unsigned int id) const;

    CLHEP::Hep3Vector GetRPThinFoilPosition(int copy_no) const;

    CLHEP::Hep3Vector GetRPThinFoilNormalVector(int copy_no) const;

    mapType::const_iterator beginDet() const
    {
      return theMap.begin();
    }

    mapType::const_iterator endDet() const
    {
      return theMap.end();
    }

    RPDeviceMapType::const_iterator beginRP() const
    {
      return theRomanPotMap.begin();
    }

    RPDeviceMapType::const_iterator endRP() const
    {
      return theRomanPotMap.end();
    }

    void BuildSets();

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

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

    CLHEP::Hep3Vector LocalToGlobal(const DetGeomDesc *gd, const CLHEP::Hep3Vector& r) const;
    CLHEP::Hep3Vector GlobalToLocal(const DetGeomDesc *gd, const CLHEP::Hep3Vector& r) const;
    CLHEP::Hep3Vector LocalToGlobal(unsigned int id, const CLHEP::Hep3Vector& r) const;
    CLHEP::Hep3Vector GlobalToLocal(unsigned int id, const CLHEP::Hep3Vector& r) const;

    CLHEP::Hep3Vector LocalToGlobalDirection(unsigned int id, const CLHEP::Hep3Vector& dir) const;
    CLHEP::Hep3Vector GlobalToLocalDirection(unsigned int id, const CLHEP::Hep3Vector& dir) const;

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

    void GetReadoutDirection(unsigned int id, double &dx, double &dy) const;

    CLHEP::Hep3Vector GetRPGlobalTranslation(int copy_no) const;
    CLHEP::HepRotation GetRPGlobalRotation(int copy_no) const;

    unsigned int NumberOfDetsIncluded() const
    {
      return theMap.size();
    }

  protected:
    mapType theMap;                           
    RPDeviceMapType theRomanPotMap;           

    mapSetType stationsInArm, rpsInStation, detsInRP;
};

#endif