JanAlignmentAlgorithm.h

Go to the documentation of this file.

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

#ifndef CalibPPS_AlignmentRelative_JanAlignmentAlgorithm_h
#define CalibPPS_AlignmentRelative_JanAlignmentAlgorithm_h

#include "CalibPPS/AlignmentRelative/interface/AlignmentAlgorithm.h"
#include "CalibPPS/AlignmentRelative/interface/SingularMode.h"

#include "TMatrixD.h"
#include "TVectorD.h"
#include "TH1D.h"
#include "TH2D.h"
#include "TGraph.h"

#include <vector>
#include <map>
#include <string>

class JanAlignmentAlgorithm : public AlignmentAlgorithm {
public:
  struct ScatterPlot {
    TGraph *g;
    TH2D *h;
  };

  struct DetStat {
    TH1D *m_dist;
    TH1D *R_dist;
    std::vector<TH1D *> coefHist;
    std::vector<TGraph *> resVsCoef;
    std::map<std::set<unsigned int>, ScatterPlot> resVsCoefRot_perRPSet;
  };

private:
  TMatrixD **Sc;

  TVectorD *Mc;

  TMatrixD S;

  TVectorD M;

  TVectorD S_eigVal;

  TMatrixD S_eigVec;

  std::vector<SingularMode> singularModes;

  bool stopOnSingularModes;

  double weakLimit;

  unsigned int events;

  std::map<unsigned int, DetStat> statistics;

  bool buildDiagnosticPlots;

public:
  JanAlignmentAlgorithm() {}

  JanAlignmentAlgorithm(const edm::ParameterSet &ps, AlignmentTask *_t);

  ~JanAlignmentAlgorithm() override;

  std::string getName() override { return "Jan"; }

  bool hasErrorEstimate() override { return true; }

  void begin(const CTPPSGeometry *geometryReal, const CTPPSGeometry *geometryMisaligned) override;

  void feed(const HitCollection &, const LocalTrackFit &) override;

  void saveDiagnostics(TDirectory *) override;

  void analyze() override;

  unsigned int solve(const std::vector<AlignmentConstraint> &,
                     std::map<unsigned int, AlignmentResult> &results,
                     TDirectory *dir) override;

  void end() override;
};

#endif