AlignmentTask.h

Go to the documentation of this file.

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

#ifndef CalibPPS_AlignmentRelative_AlignmentTask_h
#define CalibPPS_AlignmentRelative_AlignmentTask_h

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "CalibPPS/AlignmentRelative/interface/AlignmentGeometry.h"
class AlignmentConstraint;
class CTPPSGeometry;

#include <vector>

class AlignmentTask {
public:
  // -------------------- config file parameters --------------------

  bool resolveShR;

  bool resolveShZ;

  bool resolveRotZ;

  bool oneRotZPerPot;

  bool useEqualMeanUMeanVRotZConstraints;

  edm::ParameterSet fixedDetectorsConstraints;

  edm::ParameterSet standardConstraints;

  // -------------------- geometry-related members --------------------

  AlignmentGeometry geometry;

  static void buildGeometry(const std::vector<unsigned int> &rpDecIds,
                            const std::vector<unsigned int> &excludedSensors,
                            const CTPPSGeometry *,
                            double z0,
                            AlignmentGeometry &geometry);

  // -------------------- quantity-class-related members --------------------

  enum QuantityClass {
    qcShR1,  
    qcShR2,  
    qcShZ,   
    qcRotZ,  
  };

  std::vector<QuantityClass> quantityClasses;

  std::string quantityClassTag(QuantityClass) const;

  struct DetIdDirIdxPair {
    unsigned int detId;
    unsigned int dirIdx;

    bool operator<(const DetIdDirIdxPair &other) const {
      if (detId < other.detId)
        return true;
      if (detId > other.detId)
        return false;
      if (dirIdx < other.dirIdx)
        return true;

      return false;
    }
  };

  std::map<QuantityClass, std::map<DetIdDirIdxPair, unsigned int>> mapMeasurementIndeces;

  std::map<QuantityClass, std::map<unsigned int, unsigned int>> mapQuantityIndeces;

  void buildIndexMaps();

  unsigned int measurementsOfClass(QuantityClass) const;

  unsigned int quantitiesOfClass(QuantityClass) const;

  signed int getMeasurementIndex(QuantityClass cl, unsigned int detId, unsigned int dirIdx) const;

  signed int getQuantityIndex(QuantityClass cl, unsigned int detId) const;

  // -------------------- constraint-related members --------------------

  void buildFixedDetectorsConstraints(std::vector<AlignmentConstraint> &) const;

  void buildStandardConstraints(std::vector<AlignmentConstraint> &) const;

  void buildOneRotZPerPotConstraints(std::vector<AlignmentConstraint> &) const;

  void buildEqualMeanUMeanVRotZConstraints(std::vector<AlignmentConstraint> &constraints) const;

  // -------------------- constructors --------------------

  AlignmentTask() {}

  AlignmentTask(const edm::ParameterSet &ps);
};

#endif