AlignmentGeometry.h

Go to the documentation of this file.

/****************************************************************************
* Authors: 
*  Jan Kašpar (jan.kaspar@gmail.com) 
****************************************************************************/
 
#ifndef CalibPPS_AlignmentRelative_AlignmentGeometry_h
#define CalibPPS_AlignmentRelative_AlignmentGeometry_h
 
#include "FWCore/Utilities/interface/Exception.h"
 
#include <map>
#include <set>
#include <string>
 
//----------------------------------------------------------------------------------------------------
 
struct DetGeometry {
  double z;  
 
  double sx, sy;  
 
  struct DirectionData {
    double dx, dy, dz;  
    double s;           
  };
 
  std::map<unsigned int, DirectionData> directionData;
 
  bool isU;  
 
  DetGeometry(double _z = 0., double _sx = 0., double _sy = 0., bool _isU = false)
      : z(_z), sx(_sx), sy(_sy), isU(_isU) {}
 
  void setDirection(unsigned int idx, double dx, double dy, double dz) {
    directionData[idx] = {dx, dy, dz, dx * sx + dy * sy};
  }
 
  const DirectionData& getDirectionData(unsigned int idx) const {
    auto it = directionData.find(idx);
    if (it == directionData.end())
      throw cms::Exception("PPS") << "direction index " << idx << " not in the mapping.";
 
    return it->second;
  }
};
 
//----------------------------------------------------------------------------------------------------
 
class AlignmentGeometry {
protected:
  std::map<unsigned int, DetGeometry> sensorGeometry;
 
public:
  double z0;
 
  void insert(unsigned int id, const DetGeometry& g);
 
  const DetGeometry& get(unsigned int id) const;
 
  const std::map<unsigned int, DetGeometry>& getSensorMap() const { return sensorGeometry; }
 
  unsigned int getNumberOfDetectors() const { return sensorGeometry.size(); }
 
  bool isValidSensorId(unsigned int id) const { return (sensorGeometry.find(id) != sensorGeometry.end()); }
 
  void print() const;
};
 
#endif