OscProbCalcer_CUDAProb3Linear.h

Go to the documentation of this file.

#ifndef __OSCILLATOR_CUDAPROB3LINEAR_H__
#define __OSCILLATOR_CUDAPROB3LINEAR_H__
 
#include "OscProbCalcerBase.h"
#include "Constants/OscillatorConstants.h"
 
#include <memory>
 
// ==================================================================
namespace cudaprob3linear { template<typename T> class Propagator;}
// ==================================================================
 
class OscProbCalcerCUDAProb3Linear : public OscProbCalcerBase {
 public:
  OscProbCalcerCUDAProb3Linear(YAML::Node Config_);
 
  OscProbCalcerCUDAProb3Linear(std::string ConfigName_) : OscProbCalcerCUDAProb3Linear(YAML::LoadFile(ConfigName_)) {}
 
  virtual ~OscProbCalcerCUDAProb3Linear();
 
 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
 
  // ========================================================================================================================================================================
  // Variables which are needed for implementation specific code
 
  enum OscParams{kTH12, kTH23, kTH13, kDM12, kDM23, kDCP, kPATHL, kDENS, kNOscParams};
 
  enum NuType{Nubar=-1, Nu=1};
 
  std::vector<int> OscChannels;
 
  int nThreads;
 
  std::unique_ptr< cudaprob3linear::Propagator< FLOAT_T > > propagator;
};
 
#endif