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