GammaEfficiencyAnalysis.h

Go to the documentation of this file.

#include "SpectrumAnalysis.h"
#include "TFile.h"
#include "TTree.h"
#include "TChain.h"
#include "TH2.h"
#include "TCanvas.h"
#include <vector>
#include <string>
#include <map>
#include <random>
using std::vector;
using std::string;
using std::map;
using std::pair;
#include <Eigen/Dense>
class GammaEfficiencyAnalysis
{
public:
  GammaEfficiencyAnalysis();
  void setVerbosity(int a_verbosity);
  void loadExperimentalSpectra(string a_fileName,
                              string a_isotope,
                              double a_countTime = 1,
                              double a_activity = 1,
                              int a_numLines = 5);
  void useTFIntDiffCorrection(bool a_useCorr);
  void loadBackgroundSpectra(string a_fileName,
                              double a_countTime);
  int addSimulatedData(string a_fileName, int a_numSimEvents);
  void setSimHistNumBins(int a_bins);
  void setSimHistMaxE(double a_maxEnergy);
  void processSimulatedData(string a_configFileName,
                            bool a_resampleEffCorr=false,
                            double a_acceptanceWindow=2.5e-3);
  TH1F* getSimHist(DetType a_detType,
                    int a_detID,
                    int a_elemID=0,
                    bool a_addBack=false);
  double getSimEff(double a_gammaEnergy,
                    DetType a_detType,
                    int a_detID,
                    int a_elemID=0,
                    bool a_addBack=false);
  void fitTriggerProb(string a_fileNameHighTF,
                      string a_fileNameLowTF,
                      double a_countTimeHigh,
                      double a_countTimeLow,
                      string a_fileNameBKG="",
                      double a_countTimeBKG=-1,
                      string a_isotope="Eu152",
                      int a_numLines = 16);
  double calcTriggerProb(DetType a_detType,
                          int a_detID,
                          int a_elemID,
                          double a_energy);
  void generateNewTriggerProbs();
  void computeAllExpEffs(bool a_drawFits=false);
  void drawEffPeakFits(DetType a_detType,
                      int a_detID,
                      int a_elemID=0,
                      bool a_addBack=false);
  vector<vector<double>> getExpEffs(DetType a_detType,
                      int a_detID,
                      int a_elemID=0,
                      bool a_addBack=false);
  map<int, vector<vector<double>>> calcAddBackFactors();
  void plotAddBackSimExpComp();
  void plotSimExpComp(DetType a_detType,
                      int a_detID,
                      int a_elemID=0,
                      bool a_addBack=false);
  map<double, PeakInfo> fitIsotopePeaks(TH1* a_hist,
                      string a_isotope,   
                      double a_activity,  
                      double a_countTime, 
                      int a_numLines,      
                      bool a_draw=false
                      );
  map<double, std::pair<double,double>> calcEfficiency(TH1* a_hist,
                      string a_isotope,   
                      double a_activity,  
                      double a_countTime, 
                      int a_numLines,     
                      vector<PeakInfo>& a_peakFits
                      );  
  map<double, std::pair<double,double>> calcEfficiency(
                      string a_isotope,   
                      double a_activity,  
                      double a_countTime, 
                      int a_numLines,    
                      vector<PeakInfo>& a_peakFits
                      );  
  map<double, std::pair<double,double>> calcEfficiency(map<double, PeakInfo> a_peakFits,
                      string a_isotope,   
                      double a_activity,  
                      double a_countTime, 
                      int a_numLines      
                      );  
  void setNewPeakFit(PeakInfo a_peakInfo,
                      DetType a_detType,
                      int a_detID,
                      int a_elemID=0,
                      bool a_addBack=false);
  vector<PeakInfo> getPeakFits(DetType a_detType,
                                int a_detID,
                                int a_elemID=0,
                                bool a_addBack=false);
  map<double, PeakInfo> m_lastPeakFits;
  vector<double> m_gammaEnergies;
  map<DetType, map<int, map<int, Eigen::VectorXf>>> m_corrParamsD0;
  map<DetType, map<int, map<int, Eigen::MatrixXf>>> m_corrParamsDCov;
  map<DetType, map<int, map<int, Eigen::VectorXf>>> m_corrParamsD;
  map<DetType, map<int, map<int, TH2F*>>> m_angEffHists;
private:
  SpectrumAnalysis m_specAna;
  int m_verbosity;
  TChain* m_simTree;
  bool m_hasSimData;
  int m_numSimEvents;
  int m_numSimEnergyBins;
  double m_maxSimEnergy;
  int m_numResamples;
  map<string, map<string, vector<double>>> m_expData;
  // map<DetType, map<int, map<int, Eigen::VectorXf>>> m_corrParamsD;
  // map<DetType, map<int, map<int, Eigen::MatrixXf>>> m_corrParamsDCov;
  // map<DetType, map<int, map<int, Eigen::VectorXf>>> m_corrParamsD0;
  map<DetType, map<int, map<int, vector<double>>>> m_simHistsM1;
  map<DetType, map<int, map<int, vector<double>>>> m_simHistsM2;
  map<DetType, map<int, map<int, vector<vector<double>>>>> m_allExpEffs;
  map<DetType, map<int, map<int, vector<PeakInfo>>>> m_allExpPeakFits;
 
  std::mt19937_64 m_randGen;
};