PndLmdAlignQA.h

Go to the documentation of this file.

/*
 * PndLmdAlignQA.h
 *
 *  Created on: Jul 6, 2015
 *      Author: Roman Klasen, roklasen@uni-mainz.de or klasen@kph.uni-mainz.de
 *
 *      helper class that plots stuff, use one instance for desired plot, then delete and create new.
 */

#ifndef LMD_LMDSENSORALIGNMENT_PNDLMDALIGNQA_H_
#define LMD_LMDSENSORALIGNMENT_PNDLMDALIGNQA_H_

#include <sstream>
#include <string>
#include <vector>

#include "PndLmdAlignManager.h"
#include <PndLmdGeometryHelper.h>

#include <TGeoMatrix.h>

#include <matrix.h>

struct histParams {
  bool printCMPXinPathName = true;
  double scaleFactor = 1.0;
  std::string fileName, title, xtitle, ytitle, path;
  int bins = 30;
  double xMin = -1;
  double xMax = -1;
  int vectorIndex = 0;
};

class PndLmdAlignQA {

 private:
  std::vector<std::string> _inputFiles;
  std::string outputPath, pdfOutPath, binaryMatPath, LMDMatPath;
  double infoMomentum;
  bool infoAbsolute, infoRelative, byPlane, _inCentimeters;
  bool alignOptionBool;
  int pairsRequired;
  PndLmdAlignManager manager;
  PndLmdGeometryHelper *helper;

  // contains number of pairs on overlap area
  std::map<int, int> matrixInfo;

  // set of misaligned matrices read from disk
  std::map<std::string, TGeoHMatrix> *matricesMisaligned;

  void createHist(std::vector<std::vector<double>> &vec, histParams &parameters);

  double calculateOverlappingArea(int id1, int id2);

  int noOfPairs(int overlapID);

 public:
  PndLmdAlignQA();
  virtual ~PndLmdAlignQA();
  void init();

  // ========== QA functions
  void checkCombined();
  // TODO: reimplement these with new geometry
  void checkCyclicMatrices(bool inCentimeters = true);
  void checkCombinedMatrices(bool inCentimeters = true);

  // ========== helper functions
  std::map<std::string, TGeoHMatrix> *readRootMatrices(TString &filename);
  TGeoHMatrix baseTransformation(TGeoHMatrix &input, TGeoHMatrix &toBaseMatrix);
  TGeoHMatrix getMatrixSensorToSensor(int sensorOne, int sensorTwo);
  TGeoHMatrix getOverlapMatrixLikeICP(PndLmdOverlapInfo &info);
  TGeoHMatrix getMisalignedOverlapFromGeoManager(PndLmdOverlapInfo &info);
  TGeoHMatrix getMisalignedOverlapFromICP(PndLmdOverlapInfo &info, std::string ICPmatrix);
  std::vector<double> getMatrixDiffCM(PndLmdOverlapInfo &info, std::string &icpFile);
  PndLmdOverlapInfo &getSmallOverlapInfo(std::vector<PndLmdOverlapInfo> &infos, int smallOverlap);

  void readMatrixInfo();
  void checkIOpaths();
  bool checkForMatrixFiles();
  void calculateOverlapingAreas();

  // ========== settings for this QA task
  void setInCentimeters(bool inCentimeters) { this->_inCentimeters = inCentimeters; }
  void setLmdMatPath(const std::string &path) { LMDMatPath = path; }
  void setBinaryMatPath(const std::string &path) { binaryMatPath = path; }
  void setPdfOutPath(const std::string &path) { pdfOutPath = path; }
  void setAlignedGeometry(bool aligned) { alignOptionBool = aligned; }
  void setPairsRequired(int number) { pairsRequired = number; }
  void setOutputPath(std::string path) { outputPath = path; }
  void setInfoAbsolute(bool info) { infoAbsolute = info; }
  void setInfoMomentum(double info) { infoMomentum = info; }
  void setInfoRelative(bool info) { infoRelative = info; }
};

#endif /* LMD_LMDSENSORALIGNMENT_PNDLMDALIGNQA_H_ */