Calibrator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
/********************************************************************
 
   NAME: Calibrator.h
PACKAGE: TRTCalibTools
 
AUTHORS: Johan Lundquist
CREATED: 27-03-2009
 
PURPOSE: Class for calibrating a TRT sub-level
 
********************************************************************/
 
#include "CxxUtils/checker_macros.h"
#include <vector>
#include <map>
#include <set>
#include <string>
#include <array>
 
class TH1F;
class TH2F;
class TF1;
class TH1D;
class TDirectory;
class TGraphErrors;
 
class RtGraph{
public: RtGraph(TH2F*, int, const char*, bool, TDirectory*); ~RtGraph();
  //explicitly delete these, code to implement them would be horrible due to use of raw arrays RtGraph(const RtGraph & other) =  delete; RtGraph & operator=(const RtGraph & other) = delete; //TH1D** hslizes; std::vector<double> rval; std::vector<double> tval; TGraphErrors* trgr; TGraphErrors* rtgr; int npoints; float mintime;
private: std::string m_chname; std::string m_chtit;
  /***/ enum  bintype {LOW, GOOD, HIGH, EMPTY};
  /***/ bintype* m_btype;
  /***/ float *m_rightsig;
  /***/ float *m_leftsig;
  /***/ float *m_maxval;
  /***/ float *m_leftval;
  /***/ float *m_rightval;
  /***/ float m_mean;
  /***/ float m_mindistance;
  /***/ int *m_maxbin;
  /***/ int m_ipoint;
  /***/ float m_t;
  /***/ float m_d;
  /***/ float m_et;
  /***/ float m_ed;
  /***/ float *m_tv;
  /***/ float *m_dv;
  /***/ float *m_etv;
  /***/ float *m_edv;
//  /***/ float width;
//  /***/ TF1* m_ff;
};
 
class databundle{
 public: int det; int lay; int mod; int brd; int chp; int stl; int stw; int sid; int ievt; float tres; float weight; float res; float t; float r; float t0; float rt0; std::array<float, 4> rtpar{}; float x; float y; float z;
};
 
class caldata{
 public:
  caldata();
  caldata(bool,int,int);
  ~caldata(); int det; int lay; int mod; int brd; int chp; int stl; int stw; int sid; int ntres; int nrt; int nres; int t0fittype; float res; float resMean; float reserr; float tres; float tresMean; float t0; float t0err; float reft0; float t0off; float rtt0; std::array<float, 4> oldrtpar{}; std::array<float, 4> rtpar{}; float nhits; float x; float y; float z; double oldt02; double sumt0; double sumx; double sumy; double sumz; bool calflag; bool rtflag; bool t0flag; std::vector<float> m_treshist; std::vector<float> reshist; std::vector<float> rthist; RtGraph* rtgraph;
};
 
//class caldata : public caldata{
// public:
//  /**the 1D time residual histogram (100 bins)*/ float treshist[100];
//  /**the 1D residual histogram (100 bins)*/ float reshist[100];
//  /**the 2D rt histogram (20x32 bins)*/ float rthist[640];
//};
 
class Calibrator{
public:
  Calibrator();
 
  Calibrator(int,const std::string&,int,int,const std::string&,const std::string&,float);
 
  Calibrator (const Calibrator&) = delete;
  Calibrator& operator= (const Calibrator&) = delete;
 
   ~Calibrator();
 
  int Simple1dHist(float, float, int, float);
 
  int Simple2dHist(float, float, int, float, float, int, float, float);
 
  float AccumulativeMean(float, float, float);
 
  int AddHit(const std::string&, const databundle &,int*,bool);
 
  TDirectory* Calibrate ATLAS_NOT_THREAD_SAFE (TDirectory*, std::string, const std::string&, caldata*);
 
  float FitRt ATLAS_NOT_THREAD_SAFE (const std::string&,const std::string&,TH2F*,TDirectory*);
 
  float FitTimeResidual ATLAS_NOT_THREAD_SAFE (const std::string&,TH1F*);
 
  float FitResidual ATLAS_NOT_THREAD_SAFE (const std::string&,TH1F*);
 
  void WriteStat(TDirectory*);
 
  void DumpConstants();
 
  std::string GetBinnedRt(const std::string&);
 
  std::string PrintStat();
 
  std::string PrintInfo();
 
  bool HasKey(const std::string&) const;
 
  int UpdateOldConstants();
 
  bool CheckSelection(int);
 
  bool Skip();
 
  std::string GetOptString() const;
 
  int GetNDirs();
 
  std::string GetSelString();
 
  float oldt0(const std::string& key) {return data[key].sumt0/float(data[key].ntres);}
  float xmean(const std::string& key) {return data[key].sumx/float(data.size());}
  float ymean(const std::string& key) {return data[key].sumy/float(data.size());}
  float zmean(const std::string& key) {return data[key].sumz/float(data.size());}
 
  std::map<std::string,caldata> data;
  bool dort;  bool dot0;  bool dores;  bool nort;  bool not0;  bool usebref;  bool bequiet;  bool printlog;  bool printt0;  bool printrt;  bool usep0;  bool floatp3;  bool useshortstraws; std::set<int> selection; int level;
 
private:
 std::string m_name; std::string m_rtbinning; int m_minrtstat; int m_mint0stat; float m_t0shift; int m_nbinsr; int m_nbinst; int m_nbinstres; int m_nbinsres; float m_minr; float m_maxr; float m_mint; float m_maxt; float m_mintres; float m_maxtres; float m_minres; float m_maxres; std::map<std::string,TDirectory*> m_hdirs; std::map<std::string,TH1F*> m_resHists; std::map<std::string,TH1F*> m_tresHists; std::map<std::string,TH2F*> m_rtHists;
  //  /**...*/ std::map<std::string,RtGraph*> m_rtgraphs; int m_ntreshits; int m_nreshits; int m_nrthits; int m_nhits;
 
 
 bool m_isdines;
 
};