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 "CLHEP/Vector/ThreeVector.h"
#include "CLHEP/Vector/Rotation.h"
//#include "HepMC/SimpleVector.h"
#include "Geometry/VeryForwardGeometryBuilder/interface/DetGeomDesc.h"

#include <map>
#include <set>

class DetId;

class TotemRPGeometry
{
  public:
    typedef std::map<unsigned int, DetGeomDesc* > mapType;
    typedef std::map<int, 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 AddDetector(const DetId & id, const DetGeomDesc * &gd) { return AddDetector(id.rawId(), gd); }

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

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

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

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

    DetGeomDesc *GetRPDevice(int copy_no) 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 > StationsInArm(unsigned int) const;

    std::set < unsigned int > RPsInStation(unsigned int) const;
    
    double GetStationCentreZPosition(unsigned int) const;

    std::set<unsigned int> DetsInRP(unsigned int) const;
    std::set<unsigned int> DetsInRP(const DetId & id) const { return DetsInRP(id.rawId()); }


    CLHEP::Hep3Vector LocalToGlobal(DetGeomDesc *gd, const CLHEP::Hep3Vector r) const;
    CLHEP::Hep3Vector GlobalToLocal(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