CbmRichMirrorMisalignmentCorrectionUtils.h

Go to the documentation of this file.

/* Copyright (C) 2017-2020 Justus-Liebig-Universitaet Giessen, Giessen
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Jordan Bendarouach [committer] */
 
#ifndef CBM_RICH_MIRROR_MISALIGNMENT_CORRECTION_UTILS
#define CBM_RICH_MIRROR_MISALIGNMENT_CORRECTION_UTILS
 
using namespace std;
using std::string;
#include <Logger.h>
 
#include <fstream>
#include <iostream>
#include <map>
 
class CbmRichMirrorMisalignmentCorrectionUtils {
 public:
  CbmRichMirrorMisalignmentCorrectionUtils() : tBoolCorrection(false) {}
 
  virtual ~CbmRichMirrorMisalignmentCorrectionUtils() {}
 
  //       virtual void Init() { }
 
  void Init(const string& s)
  {
    SetMirrorCorrectionParameterFile(s);
    tBoolCorrection = CheckMirrorCorrectionParameterFile();
    if (tBoolCorrection) {
      ReadAndStoreMirrorCorrectionParamMap(fMirrorMisalignmentCorrectionParameterMap);
    }
    else {
      LOG(info) << "CbmRichProjectionProducerAnalytical::Init() No correction table.";
    }
  }
 
  void SetMirrorCorrectionParameterFile(const string& s) { fPathToMirrorMisalignmentCorrectionParameterFile = s; }
 
  bool CheckMirrorCorrectionParameterFile()
  {
    if (fPathToMirrorMisalignmentCorrectionParameterFile == "/correction_table/correction_param_array.txt") {
      return false;
    }
    ifstream corrFile;
    corrFile.open(fPathToMirrorMisalignmentCorrectionParameterFile);
    if (!corrFile.is_open()) {
      Fatal("CbmRichMirrorMisalignmentCorrectionUtils::"
            "CheckMirrorCorrectionParameterFile:",
            "Wrong correction parameter path given!");
    }
    return true;
  }
 
  void ReadAndStoreMirrorCorrectionParamMap(std::map<string, std::pair<Double_t, Double_t>>& mirrorCorrectionParamMap)
  {
    string mirrorID = "", fileLine = "";
    Double_t misX = 0., misY = 0.;
    ifstream corrFile;
    corrFile.open(fPathToMirrorMisalignmentCorrectionParameterFile, ios::in);
    if (corrFile.is_open()) {
      LOG(info) << "CbmRichProjectionProducerUtils::CheckMirrorCorrectionParameterFile("
                   ") Correction table used at: "
                << fPathToMirrorMisalignmentCorrectionParameterFile;
      cout << "mirrorID: \t\t\t (misY, misX): " << endl;
      while (corrFile >> mirrorID >> misY >> misX) {
        cout << mirrorID << " \t ; \t(" << misY << ", " << misX << ")" << endl;
        mirrorCorrectionParamMap[mirrorID] = std::make_pair(misX, misY);
      }
      corrFile.close();
    }
    else {
      Fatal("CbmRichMirrorMisalignmentCorrectionUtils::"
            "ReadAndStoreMirrorCorrectionParam:",
            "Wrong correction parameter path given!");
    }
 
    cout << endl << "Mapping:" << endl;
    cout << "mirrorID: \t\t\t (misY, misX): " << endl;
    for (std::map<string, std::pair<Double_t, Double_t>>::iterator it =
           fMirrorMisalignmentCorrectionParameterMap.begin();
         it != fMirrorMisalignmentCorrectionParameterMap.end(); ++it) {
      cout << it->first << " \t ; \t(" << it->second.first << ", " << it->second.second << ")" << endl;
    }
    cout << endl;
  }
 
  std::map<string, std::pair<Double_t, Double_t>> GetMirrorCorrectionParamMap()
  {
    return fMirrorMisalignmentCorrectionParameterMap;
  }
 
  bool GetMirrorCorrectionParamBool() { return tBoolCorrection; }
 
 private:
  bool tBoolCorrection;
  string fPathToMirrorMisalignmentCorrectionParameterFile;
  std::map<string, pair<Double_t, Double_t>> fMirrorMisalignmentCorrectionParameterMap;
};
 
#endif