OscProbCalcer_CUDAProb3.h

Go to the documentation of this file.

#ifndef __OSCILLATOR_CUDAPROB3_H__
#define __OSCILLATOR_CUDAPROB3_H__
 
#include "OscProbCalcerBase.h"
#include "TH2D.h"
#include "TH3D.h"
#include "TFile.h"
 
#include <memory>
 
// ==================================================================
namespace cudaprob3 { template<typename T> class Propagator;}
// ==================================================================
 
class OscProbCalcerCUDAProb3 : public OscProbCalcerBase {
 public:
  OscProbCalcerCUDAProb3(YAML::Node Config_);
 
  OscProbCalcerCUDAProb3(std::string ConfigName_) : OscProbCalcerCUDAProb3(YAML::LoadFile(ConfigName_)) {}
 
  virtual ~OscProbCalcerCUDAProb3();
 
 private:
  // ========================================================================================================================================================================
  // Functions which need implementation specific code
 
  void SetupPropagator() final;
 
  void CalculateProbabilities() final;
 
  int ReturnWeightArrayIndex(int NuTypeIndex, int OscChanIndex, int EnergyIndex, int CosineZIndex=-1) final;
 
  long DefineWeightArraySize() final;
  
  // ========================================================================================================================================================================
  //Functions which help setup implementation specific code
 
  void SetProductionHeightsAveraging();
  
  void ApplyEarthModelSystematics();
  // ========================================================================================================================================================================
  // Variables which are needed for implementation specific code
 
  enum OscParams{kTH12, kTH23, kTH13, kDM12, kDM23, kDCP, kPRODH, kNOscParams};
 
  enum NuType{Nubar=-1, Nu=1};
 
  std::vector<int> OscChannels;
 
  int nThreads;
 
  std::unique_ptr< cudaprob3::Propagator< FLOAT_T > > propagator;
 
  std::string EarthDensityFile;
 
  bool UseProductionHeightsAve;
  
  std::string ProductionHeightsFile;
 
  std::vector<std::string> ProductionHeightsHistFlavourSuffixes;
 
  bool UseEarthModelSystematics;
 
  int nLayers;
 
  int CopyArrSize;
 
  FLOAT_T* CopyArr;
 
};
 
#endif