ERF
Energy Research and Forecasting: An Atmospheric Modeling Code
ERF_RadiationInterface.H
Go to the documentation of this file.
1 #ifndef ERF_RADIATION_INTERFACE_H
2 #define ERF_RADIATION_INTERFACE_H
3 
4 #include <memory>
5 #include <iostream>
6 
7 #include <AMReX_BoxArray.H>
8 #include <AMReX_Geometry.H>
9 #include <AMReX_MultiFab.H>
10 #include <AMReX_ParmParse.H>
11 
12 // Basic interface class for wrapping different radiation models
14 {
15 public:
16  virtual ~IRadiation () = default;
17 
18  virtual
19  void
20  Init (const amrex::Geometry& geom,
21  const amrex::BoxArray& ba,
22  amrex::MultiFab* cons_in) = 0;
23 
24  virtual
25  void
26  Run (int& level,
27  int& step,
28  double& time,
29  const double& dt,
30  const amrex::BoxArray& ba,
31  amrex::Geometry& geom,
32  amrex::MultiFab* cons_in,
33  amrex::iMultiFab* lmask,
34  amrex::MultiFab* t_surf,
35  amrex::Vector<amrex::MultiFab*>& lsm_input_ptrs,
36  amrex::Vector<amrex::MultiFab*>& lsm_output_ptrs,
37  amrex::MultiFab* qheating_rates,
38  amrex::MultiFab* rad_fluxes,
39  amrex::MultiFab* z_phys,
40  amrex::MultiFab* lat,
41  amrex::MultiFab* lon,
42  const bool updated_lsm) = 0;
43 
44  virtual
45  void
46  WriteDataLog (const double& time) = 0;
47 
48  // Get names input varnames for lsm
49  virtual
50  amrex::Vector<std::string>
52  {
53  amrex::Vector<std::string> empty = {" "};
54  return empty;
55  }
56 
57  // Get names output varnames for lsm
58  virtual
59  amrex::Vector<std::string>
61  {
62  amrex::Vector<std::string> empty = {" "};
63  return empty;
64  }
65 
66  // Check if this is a nested patch (doesn't reach model top)
67  virtual
68  bool
69  is_nested_patch () const
70  {
71  return false; // Default: not a nested patch
72  }
73 
74  void
76  {
77  amrex::ParmParse pp("erf.rad");
78  if (pp.contains("datalog")) {
79  pp.queryAdd("datalog", datalogname);
80 
82  }
83  }
84 
85  void
86  setDataLogFrequency (const int nstep) {
87  datalog_int = nstep;
88  }
89 
90  bool
91  hasDatalog () { return (datalog != nullptr && datalog->good()); }
92 
93 protected:
94  std::unique_ptr<std::fstream> datalog = nullptr;
95  std::string datalogname;
96 
97  int datalog_int = -1; // datalog frequency in steps
98 
99 private:
100  void
101  setRecordDataInfo (const std::string& filename)
102  {
103  datalog = std::make_unique<std::fstream>();
104  if (amrex::ParallelDescriptor::IOProcessor())
105  {
106  datalog->open(filename.c_str(),std::ios::out|std::ios::app);
107  if (!datalog->good()) {
108  amrex::FileOpenFailed(filename);
109  }
110  datalogname = filename;
111  }
112  amrex::ParallelDescriptor::Barrier("ERF::Radiation::setRecordDataInfo");
113  }
114 };
115 
116 #endif
ParmParse pp("prob")
Definition: ERF_RadiationInterface.H:14
virtual void Run(int &level, int &step, double &time, const double &dt, const amrex::BoxArray &ba, amrex::Geometry &geom, amrex::MultiFab *cons_in, amrex::iMultiFab *lmask, amrex::MultiFab *t_surf, amrex::Vector< amrex::MultiFab * > &lsm_input_ptrs, amrex::Vector< amrex::MultiFab * > &lsm_output_ptrs, amrex::MultiFab *qheating_rates, amrex::MultiFab *rad_fluxes, amrex::MultiFab *z_phys, amrex::MultiFab *lat, amrex::MultiFab *lon, const bool updated_lsm)=0
bool hasDatalog()
Definition: ERF_RadiationInterface.H:91
std::string datalogname
Definition: ERF_RadiationInterface.H:95
std::unique_ptr< std::fstream > datalog
Definition: ERF_RadiationInterface.H:94
void setupDataLog()
Definition: ERF_RadiationInterface.H:75
void setDataLogFrequency(const int nstep)
Definition: ERF_RadiationInterface.H:86
virtual void Init(const amrex::Geometry &geom, const amrex::BoxArray &ba, amrex::MultiFab *cons_in)=0
virtual bool is_nested_patch() const
Definition: ERF_RadiationInterface.H:69
int datalog_int
Definition: ERF_RadiationInterface.H:97
virtual ~IRadiation()=default
void setRecordDataInfo(const std::string &filename)
Definition: ERF_RadiationInterface.H:101
virtual void WriteDataLog(const double &time)=0
virtual amrex::Vector< std::string > get_lsm_input_varnames()
Definition: ERF_RadiationInterface.H:51
virtual amrex::Vector< std::string > get_lsm_output_varnames()
Definition: ERF_RadiationInterface.H:60