EarthModelBinned.h

Go to the documentation of this file.

 
 
 
 
 
 
#ifndef EARTHMODELBINNED_H
#define EARTHMODELBINNED_H
 
#include "EarthModelBase.h"
 
namespace OscProb {
 
  struct TrajConstants {
      TrajConstants(double cosT = 0, double phi = 0, double DetLat = 0,
                    double DetLon = 0, double rDet = 6368)
      {
        UpdateNuAngles(cosT, phi);
        UpdateDetPos(DetLat, DetLon, rDet);
      }
 
      void UpdateNuAngles(
          double cosTheta,
          double phi); 
      void UpdateDetPos(double rDet, double DetLat,
                        double DetLon); 
      void Recalculate(); 
 
      // Variables defining trajectory
      double cosT;      
      double cosA;      
      double sinA;      
      double sinDetLat; 
      double cosDetLon; 
      double sinDetLon; 
      double DetRadius; 
 
      // Derived Variables Calculated by UpdateNuAngles()
      double sinSqT;   
      double sinT;     
      double sinTsinA; 
      double sinTcosA; 
 
      // Derived Variables Calculated by UpdateDetPos()
      double cosDetLat;     
      double rDetSinDetLat; 
      double rDetCosDetLat; 
 
      // Derived Variables Calculated by Recalculate()
      double sinTsinAsinDetLon; 
      double sinTsinAcosDetLon; 
      double cosTcosDetLat;     
      double rDetCosT;          
      double rDetSinT;          
      double alpha;   
      double beta;    
      double gamma;   
      double gammaSq; 
      double
             rDetGammaSinDetLat; 
      double maxSinSqLat; 
      double
          xlatextreme; 
  };
 
  struct EarthBin {
      EarthBin(double r_out = 0, double r_in = 0, double lat = 0,
               double lon = 0, double den = 0, double z = 0.5, int n = 0)
      {
        SetBin(r_out, r_in, lat, lon, den, z, n);
      }
 
      void SetBin(double r_out = 0, double r_in = 0, double lat = 0,
                  double lon = 0, double den = 0, double z = 0.5, int n = 0)
      {
        radius_out = r_out;
        radius_in  = r_in;
        latitude   = lat * M_PI / 180.0; // convert to radians
        longitude  = lon * M_PI / 180.0; // convert to radians
        density    = den;
        zoa        = z;
        index      = n;
      }
 
      double radius_out; 
      double radius_in;  
      double latitude;   
      double longitude;  
      double density;    
      double zoa;        
      int    index;      
  };
 
  class EarthModelBinned : public EarthModelBase {
    public:
      EarthModelBinned(std::string filename = ""); 
      virtual ~EarthModelBinned();                 
 
      void SetDetPos(double rad, double lat = 0,
                     double lon = 0); 
 
      int FillPath(double cosT, double phi = 0); 
 
      virtual void LoadModel(
          std::string filename); 
 
      virtual std::vector<EarthBin> GetEarthBins(); 
 
      virtual void SetRegionZoA(
          int    index,
          double zoa); 
      virtual double GetRegionZoA(
          int index); 
      virtual void ScaleRegionDensity(
          int index, double scalingfactor); 
 
    protected:
      struct LatBinInfo {
          int    bin;             
          int    nextBin;         
          double detDist_nextBin; 
          int sign;        
          double dLat;     
          bool maxreached; 
      };
 
      struct LonBinInfo {
          int    bin;             
          int    nextBin;         
          double detDist_nextBin; 
          double dLon;            
          double min;       
          double max;       
          int    error = 0; 
          std::string
              err_message; 
      };
 
      virtual void ClearModel(); 
 
      virtual void AddBin(
          double radius_out, double radius_in, double latitude,
          double longitude, double density, double zoa,
          double layer); 
 
      virtual void AddPath(
          double length, EarthBin bin); 
 
      virtual double DetDistForNextLatBin(
          int cur_index,
          LatBinInfo&
              L); 
 
      virtual double DetDistForNextLonBin(
          double      prev_lon,
          LonBinInfo& L); 
 
      virtual void RecordLatLonBinCrossings(
          double detDist_nextDbin, double& DetDist, int& index,
          LatBinInfo& latI,
          LonBinInfo&
              lonI); 
 
      virtual int LonBinIndex(
          double longitude); 
 
      TrajConstants fC; 
 
      std::vector<EarthBin> fEarthBins;  
      int                   fnDepthBins; 
      int                   fnLonBins;   
      int                   fnLatBins;   
      double fInvLonBinWidth;            
      double fInvLatBinWidth;            
      double fHalfLonBinWidth;           
      double fHalfLatBinWidth;           
 
      int fmaxRegIndex; 
 
      // For Latitude Calculation Error Message
      std::string fErrorMessage_LonInfo; 
      int fLonError;
 
      // Required for saving in ROOT files
      ClassDef(EarthModelBinned, 1);
  };
 
} // namespace OscProb
 
#endif