FluxMatrixAna.h

Go to the documentation of this file.

#ifndef _FLUX_MATRIX_ANA_H_
#define _FLUX_MATRIX_ANA_H_
 
//ROOT includes
#include "TH1.h"
#include "SpectrumAnalysis.h"
#include "TH2.h"
#include "TMatrixD.h"
#include "TFitResult.h"
#include "TRandom3.h"
#include "TALYSUtils.h"
//c++includes
#include <cstdint>
#include <vector>
#include <utility>
#include <deque>
#include <random>
using std::vector;
#include <map>
using std::map;
using std::pair;
 
// #include "../../BasicSupport/external/eigen/Eigen/Dense"
#include <Eigen/Dense>
 
 
class FluxAna
{
public:
  FluxAna();
 
 
  //setters
  //all times are ns, distances are cm, energies are MeV
  void setRFPeriod(double a_RFPeriod);
  void setTimeResolution(double a_timeResolution);
  void setReqTimeBinWidth(double a_reqTimeBinWidth);
  void setFlightPath(double a_flightPath);
  void setEMin(double a_eMin);
  void setEMax(double a_eMax);
  double getRFPeriod();
  double getTimeResolution();
  double getReqTimeBinWidth();
  double getFlightPath();
  double getEMin();
  double getEMax();
  //OriginalFlux readFlux(double a_minEnergy);
  void readBinWiseFluxCSV(std::string a_fileName);
  void loadFluxHist(std::string a_fileName,std::string a_histName);
  void loadFluxHistCovMat(std::string a_fileName,std::string a_histName);
  void setFlxHist(TH1* a_fluxHist);
  TH1* getFluxHist();
 
  vector<double> readDeutronTimeProfile(TH1* a_rfHist,
                                        double a_distToBreakup,
                                        double a_distToDet,
                                        double a_timeResolution=-1);
 
  vector<double> getEnergySpecLowerBinEdges();
  
  std::vector<double> buildTimeSinceRFAxis();
  void setTimeSinceRFAxis(std::vector<double> a_binEdges);
  std::vector<double> buildEnergyAxis();
  std::pair<int,int> findRFPeriodPlusBinIndex(double a_En);
  TH1* generateNew1DFlux();
  TH1* rebinFluxHist();
  TH1* getRebinnedFluxHistPerMeV();
  TH2* buildFluxMatrix();
 
 
  TH2F* smearFluxMatrix();
  TH2* getSmearFluxMatrix();
  vector<Eigen::MatrixXf> getFluxMatrix();
  vector<Eigen::MatrixXf> getSmearedFluxMatrix();
  //requires that the rebinned energy histogram has been generated
  std::vector<double> getEnBinCenters();
 
  std::pair<double, double> getFluxAndError(double a_en);
 
  std::pair<double, double> getFluxAndError(double a_lowEn,double a_highEn);
  vector<vector<double>> getFluxAtTimeSinceRFBin(int a_tofBin);
  ~FluxAna();
 
private:
  TH1* m_fluxHisto;
  TH1* m_fluxHistoOriginal;
  Eigen::MatrixXf m_fluxCovarianceMatrix;
  Eigen::MatrixXf m_cholDecompFluxCov;
  TH1* m_fluxHistoRebin;
  TH2* m_rootFluxMatrix;
  TH2* m_rootFluxMatrixConv;
  Eigen::MatrixXf m_fullFluxMat;
  Eigen::MatrixXf m_fullFluxMatErr;
  Eigen::MatrixXf m_fullFluxMatConv;
  Eigen::MatrixXf m_fullFluxMatErrConv;
  vector<double> m_orgBinEdges;
  vector<double> m_orgBinFlux;
  vector<double> m_orgBinDFlux;
 
  std::vector<double> m_timeSinceRFAxis;
  std::vector<double> m_energyAxis;
 
  vector<double> m_timeSmearing;
  double m_RFPeriod;
  double m_timeResolution;
  double m_reqTimeBinWidth;
  double m_timeBinWidth;
  double m_flightPath;
  double m_eMin;
  double m_eMax;
  bool m_hasFluxUncMat;
  TALYSUtils m_talysUtil;  
  std::mt19937_64 m_randGen;
};
 
#endif