#include <ERF_SampleData.H>

Collaboration diagram for LineSampler:

[legend]

Public Member Functions
	LineSampler ()

amrex::Box	getIndexBox (const amrex::RealBox &real_box, const amrex::Geometry &geom)

void	write_coords (amrex::Vector< std::unique_ptr< amrex::MultiFab > > &z_phys_cc, amrex::Vector< amrex::Geometry > &geom)

void	get_sample_data (amrex::Vector< amrex::Geometry > &geom, amrex::Vector< amrex::Vector< amrex::MultiFab >> &vars_new, const amrex::Vector< amrex::MultiFab * > &tau13_lev={}, const amrex::Vector< amrex::MultiFab * > &tau23_lev={}, const amrex::Vector< amrex::MultiFab * > &hfx3_lev={}, const amrex::Vector< int > &prognostic_tke_available={})

void	write_sample_data (amrex::Vector< double > &time, amrex::Vector< int > &level_steps, amrex::Vector< amrex::IntVect > &ref_ratio, amrex::Vector< amrex::Geometry > &geom)

void	write_line_ascii (amrex::Vector< double > &time)

void	write_line_plotfile (amrex::Vector< double > &time, amrex::Vector< int > &level_steps, amrex::Vector< amrex::IntVect > &ref_ratio, amrex::Vector< amrex::Geometry > &geom)

Public Attributes
amrex::Vector< int >	m_dir

amrex::Vector< int >	m_lev

amrex::Vector< amrex::Box >	m_bnd_bx

amrex::Vector< amrex::RealBox >	m_bnd_rbx

amrex::Vector< amrex::MultiFab >	m_ls_mf

amrex::Vector< std::string >	m_name

bool	m_use_real_bx {false}

bool	m_write_ascii {false}

amrex::Vector< std::string >	m_varnames {"magvel","theta"}

amrex::Vector< std::unique_ptr< std::fstream > >	m_datastream

Constructor & Destructor Documentation

◆ LineSampler()

LineSampler::LineSampler ( )

inline

     {
         amrex::ParmParse pp("erf");
  
         bool has_line_idx  = pp.contains("sample_line_lo") || pp.contains("sample_line_hi");
         bool has_line_real = pp.contains("sample_line_lo_real") || pp.contains("sample_line_hi_real");
         if (has_line_idx && has_line_real) {
             amrex::Abort("Specify only one of erf.sample_line_lo/hi or erf.sample_line_lo_real/hi_real");
         }
  
         // Count number of lo and hi points define the line
         int n_line_lo  = pp.countval("sample_line_lo") / AMREX_SPACEDIM;
         int n_line_hi  = pp.countval("sample_line_hi") / AMREX_SPACEDIM;
         int n_line_lo_real = pp.countval("sample_line_lo_real") / AMREX_SPACEDIM;
         int n_line_hi_real = pp.countval("sample_line_hi_real") / AMREX_SPACEDIM;
         int n_line_dir = pp.countval("sample_line_dir");
         if (has_line_idx) {
             AMREX_ALWAYS_ASSERT( (n_line_lo==n_line_hi ) &&
                                  (n_line_lo==n_line_dir) );
         } else if (has_line_real) {
             AMREX_ALWAYS_ASSERT( (n_line_lo_real==n_line_hi_real ) &&
                                  (n_line_lo_real==n_line_dir) );
         }
  
         // Parse the data
         int nline = 0;
         if (has_line_idx) {
             nline = n_line_lo;
             if (nline > 0) {
                 // Parse lo
                 amrex::Vector<int> idx_lo; idx_lo.resize(nline*AMREX_SPACEDIM);
                 amrex::Vector<amrex::IntVect> iv_lo; iv_lo.resize(nline);
                 pp.queryarr("sample_line_lo",idx_lo,0,nline*AMREX_SPACEDIM);
                 for (int i(0); i < nline; i++) {
                     amrex::IntVect iv(idx_lo[AMREX_SPACEDIM*i+0],
                                       idx_lo[AMREX_SPACEDIM*i+1],
                                       idx_lo[AMREX_SPACEDIM*i+2]);
                     iv_lo[i] = iv;
                 }
  
                 // Parse hi
                 amrex::Vector<int> idx_hi; idx_hi.resize(nline*AMREX_SPACEDIM);
                 amrex::Vector<amrex::IntVect> iv_hi; iv_hi.resize(nline);
                 pp.queryarr("sample_line_hi",idx_hi,0,nline*AMREX_SPACEDIM);
                 for (int i(0); i < nline; i++) {
                     amrex::IntVect iv(idx_hi[AMREX_SPACEDIM*i+0],
                                       idx_hi[AMREX_SPACEDIM*i+1],
                                       idx_hi[AMREX_SPACEDIM*i+2]);
                     iv_hi[i] = iv;
                 }
  
                 // Construct vector of bounding boxes
                 m_bnd_bx.resize(nline);
                 for (int i = 0; i < nline; i++){
                     amrex::Box lbx(iv_lo[i],iv_hi[i]);
                     m_bnd_bx[i] = lbx;
                 }
             }
         } else if (has_line_real) {
             m_use_real_bx = true;
             nline = n_line_lo_real;
             if (nline > 0) {
                 // Parse lo
                 amrex::Vector<amrex::Real> real_lo; real_lo.resize(nline*AMREX_SPACEDIM);
                 amrex::Vector<amrex::Vector<amrex::Real>> rv_lo;
                 pp.queryarr("sample_line_lo_real",real_lo,0,nline*AMREX_SPACEDIM);
                 for (int i(0); i < nline; i++) {
                     amrex::Vector<amrex::Real> rv = {real_lo[AMREX_SPACEDIM*i+0],
                                                      real_lo[AMREX_SPACEDIM*i+1],
                                                      real_lo[AMREX_SPACEDIM*i+2]};
                     rv_lo.push_back(rv);
                 }
  
                 // Parse hi
                 amrex::Vector<amrex::Real> real_hi; real_hi.resize(nline*AMREX_SPACEDIM);
                 amrex::Vector<amrex::Vector<amrex::Real>> rv_hi;
                 pp.queryarr("sample_line_hi_real",real_hi,0,nline*AMREX_SPACEDIM);
                 for (int i(0); i < nline; i++) {
                     amrex::Vector<amrex::Real> rv = {real_hi[AMREX_SPACEDIM*i+0],
                                                      real_hi[AMREX_SPACEDIM*i+1],
                                                      real_hi[AMREX_SPACEDIM*i+2]};
                     rv_hi.push_back(rv);
                 }
  
                 // Construct vector of bounding real boxes
                 m_bnd_rbx.resize(nline);
                 for (int i = 0; i < nline; i++){
                     amrex::RealBox rbx(rv_lo[i].data(),rv_hi[i].data());
                     m_bnd_rbx[i] = rbx;
                 }
  
                 // Construct vector of bounding boxes
                 m_bnd_bx.resize(nline);
             }
         }
  
         if (nline > 0) {
             // Parse directionality
             m_dir.resize(n_line_dir);
             pp.queryarr("sample_line_dir",m_dir,0,n_line_dir);
  
             // Parse names
             std::string name_base = "plt_line_";
             m_name.resize(nline);
             int n_names = pp.countval("sample_line_name");
             if (n_names > 0) {
                 AMREX_ALWAYS_ASSERT( n_names==nline );
                 pp.queryarr("sample_line_name",m_name,0,n_names);
             } else {
                 for (int iline(0); iline<nline; ++iline) {
                     m_name[iline] = amrex::Concatenate(name_base, iline , 5);
                 }
             }
  
             // Allocate space for level indicator
             m_lev.resize(n_line_dir,0);
  
             // Allocate space for MF pointers
             m_ls_mf.resize(nline);
  
             // Get requested vars
             if (pp.countval("line_sampling_vars") > 0) {
                 m_varnames.clear();
                 amrex::Vector<std::string> requested_vars;
                 pp.queryarr("line_sampling_vars",requested_vars);
                 amrex::Print() << "Selected line sampling vars :";
                 if (containerHasElement(requested_vars, "density")) {
                     m_varnames.push_back("density");
                     amrex::Print() << " " << "density";
                 }
                 if (containerHasElement(requested_vars, "x_velocity")) {
                     m_varnames.push_back("x_velocity");
                     amrex::Print() << " " << "x_velocity";
                 }
                 if (containerHasElement(requested_vars, "y_velocity")) {
                     m_varnames.push_back("y_velocity");
                     amrex::Print() << " " << "y_velocity";
                 }
                 if (containerHasElement(requested_vars, "z_velocity")) {
                     m_varnames.push_back("z_velocity");
                     amrex::Print() << " " << "z_velocity";
                 }
                 if (containerHasElement(requested_vars, "magvel")) {
                     m_varnames.push_back("magvel");
                     amrex::Print() << " " << "magvel";
                 }
                 if (containerHasElement(requested_vars, "theta")) {
                     m_varnames.push_back("theta");
                     amrex::Print() << " " << "theta";
                 }
                 if (containerHasElement(requested_vars, "qv")) {
                     m_varnames.push_back("qv");
                     amrex::Print() << " " << "qv";
                 }
                 if (containerHasElement(requested_vars, "qc")) {
                     m_varnames.push_back("qc");
                     amrex::Print() << " " << "qc";
                 }
                 if (containerHasElement(requested_vars, "pressure")) {
                     m_varnames.push_back("pressure");
                     amrex::Print() << " " << "pressure";
                 }
                 if (containerHasElement(requested_vars, "sgs_tke")) {
                     m_varnames.push_back("sgs_tke");
                     amrex::Print() << " " << "sgs_tke";
                 }
                 if (containerHasElement(requested_vars, "sgs_tau13")) {
                     m_varnames.push_back("sgs_tau13");
                     amrex::Print() << " " << "sgs_tau13";
                 }
                 if (containerHasElement(requested_vars, "sgs_tau23")) {
                     m_varnames.push_back("sgs_tau23");
                     amrex::Print() << " " << "sgs_tau23";
                 }
                 if (containerHasElement(requested_vars, "sgs_hfx3")) {
                     m_varnames.push_back("sgs_hfx3");
                     amrex::Print() << " " << "sgs_hfx3";
                 }
                 amrex::Print() << std::endl;
             }
  
             // Write outputs to text files, one file per variable, with all
             // times appended to the same file
             pp.queryAdd("line_sampling_text_output",m_write_ascii);
             if (m_write_ascii && amrex::ParallelDescriptor::IOProcessor()) {
                 int nvar = static_cast<int>(m_varnames.size());
                 m_datastream.resize(nline * nvar);
                 int i = 0;
                 for (int iline(0); iline<nline; ++iline) {
                     for (int ivar(0); ivar<nvar; ++ivar) {
                         std::string filename = m_name[iline] + "." + m_varnames[ivar];
                         m_datastream[i] = std::make_unique<std::fstream>();
                         m_datastream[i]->open(filename.c_str(),std::ios::out|std::ios::app);
                         if (!m_datastream[i]->good()) {
                             amrex::FileOpenFailed(filename);
                         }
                         i++;
                     }
                 }
             }
         }
     }

Here is the call graph for this function:

Member Function Documentation

◆ get_sample_data()

void LineSampler::get_sample_data	(	amrex::Vector< amrex::Geometry > &	geom,
		amrex::Vector< amrex::Vector< amrex::MultiFab >> &	vars_new,
		const amrex::Vector< amrex::MultiFab * > &	tau13_lev = `{}`,
		const amrex::Vector< amrex::MultiFab * > &	tau23_lev = `{}`,
		const amrex::Vector< amrex::MultiFab * > &	hfx3_lev = `{}`,
		const amrex::Vector< int > &	prognostic_tke_available = `{}`
	)

inline

                                                                     {},
                      const amrex::Vector<amrex::MultiFab*>& tau23_lev = {},
                      const amrex::Vector<amrex::MultiFab*>& hfx3_lev = {},
                      const amrex::Vector<int>& prognostic_tke_available = {})
     {
         int nlev  = static_cast<int>(vars_new.size());
         int nline = static_cast<int>(m_bnd_bx.size());
         int ncomp = static_cast<int>(m_varnames.size());
  
         int qv_comp = -1;
  
         // Loop over each line
         for (int iline(0); iline<nline; ++iline) {
             int dir = m_dir[iline];
  
             // Search each level to get the finest data possible
             for (int ilev(nlev-1); ilev>=0; --ilev) {
                 amrex::Box bnd_bx = (m_use_real_bx)
                                   ? getIndexBox(m_bnd_rbx[iline], geom[ilev])
                                   : m_bnd_bx[iline];
                 amrex::IntVect cell = bnd_bx.smallEnd();
  
                 // Construct CC velocities
                 amrex::MultiFab mf_cc_vel;
                 auto ba = vars_new[ilev][Vars::cons].boxArray();
                 auto dm = vars_new[ilev][Vars::cons].DistributionMap();
                 mf_cc_vel.define(ba, dm, AMREX_SPACEDIM, amrex::IntVect(1,1,1));
                 average_face_to_cellcenter(mf_cc_vel,0,
                                            amrex::Array<const amrex::MultiFab*,3>{&vars_new[ilev][Vars::xvel],
                                                                                   &vars_new[ilev][Vars::yvel],
                                                                                   &vars_new[ilev][Vars::zvel]});
  
                 // Construct MultiFab holding requested variables
                 amrex::MultiFab mf_cc_data;
                 mf_cc_data.define(ba, dm, ncomp, 1);
  
                 int mf_comp = 0;
  
                 if (containerHasElement(m_varnames, "density")) {
                     amrex::MultiFab::Copy(mf_cc_data, vars_new[ilev][Vars::cons], Rho_comp, mf_comp, 1, 0);
                     mf_comp += 1;
                 }
  
                 if (containerHasElement(m_varnames, "x_velocity")) {
                     amrex::MultiFab::Copy(mf_cc_data, mf_cc_vel, 0, mf_comp, 1, 0);
                     mf_comp += 1;
                 }
                 if (containerHasElement(m_varnames, "y_velocity")) {
                     amrex::MultiFab::Copy(mf_cc_data, mf_cc_vel, 1, mf_comp, 1, 0);
                     mf_comp += 1;
                 }
                 if (containerHasElement(m_varnames, "z_velocity")) {
                     amrex::MultiFab::Copy(mf_cc_data, mf_cc_vel, 2, mf_comp, 1, 0);
                     mf_comp += 1;
                 }
  
                 if (containerHasElement(m_varnames, "magvel")) {
 #ifdef _OPENMP
 #pragma omp parallel if (amrex::Gpu::notInLaunchRegion())
 #endif
                     for (amrex::MFIter mfi(mf_cc_data, amrex::TilingIfNotGPU()); mfi.isValid(); ++mfi) {
                         const amrex::Box& tbx = mfi.tilebox();
                         auto const& dfab = mf_cc_data.array(mfi);
                         auto const& vfab = mf_cc_vel.array(mfi);
  
                         amrex::ParallelFor(tbx, [=] AMREX_GPU_DEVICE(int i, int j, int k) noexcept
                         {
                             dfab(i,j,k,mf_comp) = std::sqrt(vfab(i,j,k,0)*vfab(i,j,k,0)
                                                           + vfab(i,j,k,1)*vfab(i,j,k,1)
                                                           + vfab(i,j,k,2)*vfab(i,j,k,2)) ;
                         });
                     }
                     mf_comp += 1;
                 }
  
                 if (containerHasElement(m_varnames, "theta")) {
 #ifdef _OPENMP
 #pragma omp parallel if (amrex::Gpu::notInLaunchRegion())
 #endif
                     for (amrex::MFIter mfi(mf_cc_data, amrex::TilingIfNotGPU()); mfi.isValid(); ++mfi) {
                         const amrex::Box& tbx = mfi.tilebox();
                         auto const& dfab = mf_cc_data.array(mfi);
                         auto const& cfab = vars_new[ilev][Vars::cons].array(mfi);
  
                         amrex::ParallelFor(tbx, [=] AMREX_GPU_DEVICE(int i, int j, int k) noexcept
                         {
                             dfab(i,j,k,mf_comp) = cfab(i,j,k,RhoTheta_comp) / cfab(i,j,k,Rho_comp);
                         });
                     }
                     mf_comp += 1;
                 }
  
                 if (containerHasElement(m_varnames, "qv")) {
                     AMREX_ALWAYS_ASSERT_WITH_MESSAGE(vars_new[ilev][Vars::cons].nComp() > RhoQ1_comp,
                         "qv sampling requested but moisture components not present in state");
                     if (qv_comp >= 0) AMREX_ALWAYS_ASSERT(qv_comp == mf_comp);
                     qv_comp = mf_comp;
 #ifdef _OPENMP
 #pragma omp parallel if (amrex::Gpu::notInLaunchRegion())
 #endif
                     for (amrex::MFIter mfi(mf_cc_data, amrex::TilingIfNotGPU()); mfi.isValid(); ++mfi) {
                         const amrex::Box& tbx = mfi.tilebox();
                         auto const& dfab = mf_cc_data.array(mfi);
                         auto const& cfab = vars_new[ilev][Vars::cons].array(mfi);
  
                         amrex::ParallelFor(tbx, [=] AMREX_GPU_DEVICE(int i, int j, int k) noexcept
                         {
                             dfab(i,j,k,mf_comp) = cfab(i,j,k,RhoQ1_comp) / cfab(i,j,k,Rho_comp);
                         });
                     }
                     mf_comp += 1;
                 }
                 if (containerHasElement(m_varnames, "qc")) {
                     AMREX_ALWAYS_ASSERT_WITH_MESSAGE(vars_new[ilev][Vars::cons].nComp() > RhoQ2_comp,
                         "qc sampling requested but moisture components not present in state");
 #ifdef _OPENMP
 #pragma omp parallel if (amrex::Gpu::notInLaunchRegion())
 #endif
                     for (amrex::MFIter mfi(mf_cc_data, amrex::TilingIfNotGPU()); mfi.isValid(); ++mfi) {
                         const amrex::Box& tbx = mfi.tilebox();
                         auto const& dfab = mf_cc_data.array(mfi);
                         auto const& cfab = vars_new[ilev][Vars::cons].array(mfi);
  
                         amrex::ParallelFor(tbx, [=] AMREX_GPU_DEVICE(int i, int j, int k) noexcept
                         {
                             dfab(i,j,k,mf_comp) = cfab(i,j,k,RhoQ2_comp) / cfab(i,j,k,Rho_comp);
                         });
                     }
                     mf_comp += 1;
                 }
  
                 if (containerHasElement(m_varnames, "pressure")) {
                     if (vars_new[ilev][Vars::cons].nComp() > RhoQ1_comp) {
                         // moist pressure: use qv from dfab if already sampled, else compute inline
 #ifdef _OPENMP
 #pragma omp parallel if (amrex::Gpu::notInLaunchRegion())
 #endif
                         for (amrex::MFIter mfi(mf_cc_data, amrex::TilingIfNotGPU()); mfi.isValid(); ++mfi) {
                             const amrex::Box& tbx = mfi.tilebox();
                             auto const& dfab = mf_cc_data.array(mfi);
                             auto const& cfab = vars_new[ilev][Vars::cons].array(mfi);
  
                             amrex::ParallelFor(tbx, [=] AMREX_GPU_DEVICE(int i, int j, int k) noexcept
                             {
                                 amrex::Real qv_val = (qv_comp >= 0) ? dfab(i,j,k,qv_comp)
                                                                      : cfab(i,j,k,RhoQ1_comp) / cfab(i,j,k,Rho_comp);
                                 dfab(i,j,k,mf_comp) = getPgivenRTh(cfab(i,j,k,RhoTheta_comp), qv_val);
                             });
                         }
                     } else {
                         // dry pressure
 #ifdef _OPENMP
 #pragma omp parallel if (amrex::Gpu::notInLaunchRegion())
 #endif
                         for (amrex::MFIter mfi(mf_cc_data, amrex::TilingIfNotGPU()); mfi.isValid(); ++mfi) {
                             const amrex::Box& tbx = mfi.tilebox();
                             auto const& dfab = mf_cc_data.array(mfi);
                             auto const& cfab = vars_new[ilev][Vars::cons].array(mfi);
  
                             amrex::ParallelFor(tbx, [=] AMREX_GPU_DEVICE(int i, int j, int k) noexcept
                             {
                                 dfab(i,j,k,mf_comp) = getPgivenRTh(cfab(i,j,k,RhoTheta_comp));
                             });
                         }
                     }
                     mf_comp += 1;
                 }
  
                 if (containerHasElement(m_varnames, "sgs_tke")) {
                     const bool tke_available =
                         ilev < static_cast<int>(prognostic_tke_available.size()) &&
                         prognostic_tke_available[ilev] != 0;
                     if (!tke_available) {
                         mf_cc_data.setVal(amrex::Real(0.0), mf_comp, 1, 0);
                     } else {
 #ifdef _OPENMP
 #pragma omp parallel if (amrex::Gpu::notInLaunchRegion())
 #endif
                     for (amrex::MFIter mfi(mf_cc_data, amrex::TilingIfNotGPU()); mfi.isValid(); ++mfi) {
                         const amrex::Box& tbx = mfi.tilebox();
                         const auto& out = mf_cc_data.array(mfi);
                         const auto& state = vars_new[ilev][Vars::cons].const_array(mfi);
                         amrex::ParallelFor(tbx, [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept {
                             out(i,j,k,mf_comp) = state(i,j,k,RhoKE_comp) / state(i,j,k,Rho_comp);
                             });
                         }
                     }
                     mf_comp += 1;
                 }
  
                 if (containerHasElement(m_varnames, "sgs_tau13")) {
                     amrex::MultiFab* tau13 = ilev < static_cast<int>(tau13_lev.size())
                         ? tau13_lev[ilev] : nullptr;
                     if (tau13 != nullptr) {
 #ifdef _OPENMP
 #pragma omp parallel if (amrex::Gpu::notInLaunchRegion())
 #endif
                         for (amrex::MFIter mfi(mf_cc_data, amrex::TilingIfNotGPU()); mfi.isValid(); ++mfi) {
                             const amrex::Box& tbx = mfi.tilebox();
                             const auto& out = mf_cc_data.array(mfi);
                             const auto& tau = tau13->const_array(mfi);
                             amrex::ParallelFor(tbx, [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept {
                                 out(i,j,k,mf_comp) = amrex::Real(0.25) *
                                     (tau(i,j,k) + tau(i+1,j,k) +
                                      tau(i,j,k+1) + tau(i+1,j,k+1));
                             });
                         }
                     } else {
                         mf_cc_data.setVal(0.0, mf_comp, 1, 0);
                     }
                     mf_comp += 1;
                 }
  
                 if (containerHasElement(m_varnames, "sgs_tau23")) {
                     amrex::MultiFab* tau23 = ilev < static_cast<int>(tau23_lev.size())
                         ? tau23_lev[ilev] : nullptr;
                     if (tau23 != nullptr) {
 #ifdef _OPENMP
 #pragma omp parallel if (amrex::Gpu::notInLaunchRegion())
 #endif
                         for (amrex::MFIter mfi(mf_cc_data, amrex::TilingIfNotGPU()); mfi.isValid(); ++mfi) {
                             const amrex::Box& tbx = mfi.tilebox();
                             const auto& out = mf_cc_data.array(mfi);
                             const auto& tau = tau23->const_array(mfi);
                             amrex::ParallelFor(tbx, [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept {
                                 out(i,j,k,mf_comp) = amrex::Real(0.25) *
                                     (tau(i,j,k) + tau(i,j+1,k) +
                                      tau(i,j,k+1) + tau(i,j+1,k+1));
                             });
                         }
                     } else {
                         mf_cc_data.setVal(0.0, mf_comp, 1, 0);
                     }
                     mf_comp += 1;
                 }
  
                 if (containerHasElement(m_varnames, "sgs_hfx3")) {
                     amrex::MultiFab* hfx3 = ilev < static_cast<int>(hfx3_lev.size())
                         ? hfx3_lev[ilev] : nullptr;
                     if (hfx3 != nullptr) {
 #ifdef _OPENMP
 #pragma omp parallel if (amrex::Gpu::notInLaunchRegion())
 #endif
                         for (amrex::MFIter mfi(mf_cc_data, amrex::TilingIfNotGPU()); mfi.isValid(); ++mfi) {
                             const amrex::Box& tbx = mfi.tilebox();
                             const auto& out = mf_cc_data.array(mfi);
                             const auto& hfx = hfx3->const_array(mfi);
                             amrex::ParallelFor(tbx, [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept {
                                 out(i,j,k,mf_comp) = amrex::Real(0.5) * (hfx(i,j,k) + hfx(i,j,k+1));
                             });
                         }
                     } else {
                         mf_cc_data.setVal(0.0, mf_comp, 1, 0);
                     }
                     mf_comp += 1;
                 }
  
                 m_lev[iline] = ilev;
                 m_ls_mf[iline] = get_line_data(mf_cc_data, dir, cell, bnd_bx);
  
                 // We can stop if we got the entire line
                 auto min_bnd_bx = m_ls_mf[iline].boxArray().minimalBox();
                 if (bnd_bx == min_bnd_bx) { break; }
  
             } // ilev
  
             // Fill bnd_bx if use_rbx
             if (m_use_real_bx) {
                 m_bnd_bx[iline] = m_ls_mf[iline].boxArray().minimalBox();
             }
  
         }// iline
     }

◆ getIndexBox()

amrex::Box LineSampler::getIndexBox	(	const amrex::RealBox &	real_box,
		const amrex::Geometry &	geom
	)

inline

                                             {
         amrex::IntVect slice_lo, slice_hi;
  
         AMREX_D_TERM(slice_lo[0]=static_cast<int>(std::floor((real_box.lo(0) - geom.ProbLo(0))/geom.CellSize(0)));,
                      slice_lo[1]=static_cast<int>(std::floor((real_box.lo(1) - geom.ProbLo(1))/geom.CellSize(1)));,
                      slice_lo[2]=static_cast<int>(std::floor((real_box.lo(2) - geom.ProbLo(2))/geom.CellSize(2))););
  
         AMREX_D_TERM(slice_hi[0]=static_cast<int>(std::floor((real_box.hi(0) - geom.ProbLo(0))/geom.CellSize(0)));,
                      slice_hi[1]=static_cast<int>(std::floor((real_box.hi(1) - geom.ProbLo(1))/geom.CellSize(1)));,
                      slice_hi[2]=static_cast<int>(std::floor((real_box.hi(2) - geom.ProbLo(2))/geom.CellSize(2))););
  
         return amrex::Box(slice_lo, slice_hi) & geom.Domain();
     }

Referenced by write_coords().

Here is the caller graph for this function:

◆ write_coords()

void LineSampler::write_coords	(	amrex::Vector< std::unique_ptr< amrex::MultiFab > > &	z_phys_cc,
		amrex::Vector< amrex::Geometry > &	geom
	)

inline

     {
         if (!m_write_ascii) return;
  
         amrex::Print() << "Writing out line coordinates to text" << std::endl;
  
         for (int lev(0); lev < z_phys_cc.size(); ++lev) {
             // Write one text file per level
             std::ofstream outfile;
             if (amrex::ParallelDescriptor::IOProcessor()) {
                 std::string fname = amrex::Concatenate("plt_line_lev", lev, 1);
                 fname += ".zcc";
                 outfile.open(fname);
  
                 if (!outfile.is_open()) {
                     amrex::AllPrint() << "Could not open " << fname << std::endl;
                 }
             }
  
             // Loop over each line
             int nline = static_cast<int>(m_ls_mf.size());
             for (int iline(0); iline<nline; ++iline) {
                 int dir = m_dir[iline];
                 amrex::Box bnd_bx = (m_use_real_bx)
                                   ? getIndexBox(m_bnd_rbx[iline], geom[lev])
                                   : m_bnd_bx[iline];
                 amrex::IntVect first_cell = bnd_bx.smallEnd();
  
                 // Create multifab with "sampled" z_phys values
                 amrex::MultiFab line_coords_mf = get_line_data(
                     *z_phys_cc[lev], dir, first_cell, bnd_bx
                 );
  
                 // Convert multifab to vector
                 amrex::Gpu::HostVector<amrex::Real> vec = sumToLine(
                     line_coords_mf, 0, 1, bnd_bx, dir
                 );
  
                 // Append to file
                 if (amrex::ParallelDescriptor::IOProcessor()) {
                     for (const auto& zval : vec) {
                         outfile << " " << zval;
                     }
                     outfile << std::endl;
                 }
             } // line loop
  
             outfile.close();
         } // level loop
     }

Here is the call graph for this function:

◆ write_line_ascii()

void LineSampler::write_line_ascii ( amrex::Vector< double > & time )

inline

     {
         // same as definitions in ERF.H
         constexpr int datwidth = 14;
         constexpr int datprecision = 6;
         constexpr int timeprecision = 13;
  
         int nline = static_cast<int>(m_ls_mf.size());
         int nvar  = static_cast<int>(m_varnames.size());
         for (int iline(0); iline<nline; ++iline) {
             int dir = m_dir[iline];
             int lev = m_lev[iline];
             double m_time = time[lev];
             amrex::Box m_dom   = m_bnd_bx[iline];
  
             for (int ivar(0); ivar<nvar; ++ivar) {
                 // Convert multifab to vector
                 amrex::Gpu::HostVector<amrex::Real> vec = sumToLine(m_ls_mf[iline], ivar, 1, m_dom, dir);
  
                 if (amrex::ParallelDescriptor::IOProcessor()) {
                     int ifile = iline*nvar + ivar;
                     std::ostream& fs = *m_datastream[ifile];
                     fs << std::setw(datwidth) << std::setprecision(timeprecision) << m_time
                        << std::setw(datwidth) << std::setprecision(datprecision);
                     for (const auto& val : vec) {
                         fs << " " << val;
                     }
                     fs << std::endl;
                 }
             }
         }
     }

Referenced by write_sample_data().

Here is the caller graph for this function:

◆ write_line_plotfile()

void LineSampler::write_line_plotfile	(	amrex::Vector< double > &	time,
		amrex::Vector< int > &	level_steps,
		amrex::Vector< amrex::IntVect > &	ref_ratio,
		amrex::Vector< amrex::Geometry > &	geom
	)

inline

     {
         int nline = static_cast<int>(m_ls_mf.size());
         for (int iline(0); iline<nline; ++iline) {
             // Data members that can be used as-is
             int dir = m_dir[iline];
             int lev = m_lev[iline];
             double m_time = time[lev];
             amrex::Vector<int> m_level_steps = {level_steps[lev]};
             amrex::Vector<amrex::IntVect> m_ref_ratio  = {ref_ratio[lev]};
  
             // Create modified geometry object corresponding to the line
             auto plo = geom[lev].ProbLo();
             auto dx  = geom[lev].CellSize();
             amrex::Vector<amrex::Geometry> m_geom; m_geom.resize(1);
             amrex::Vector<int> is_per(AMREX_SPACEDIM,0);
             amrex::Box m_dom = m_bnd_bx[iline];
             amrex::RealBox m_rb;
             for (int d(0); d<AMREX_SPACEDIM; ++d) {
                 amrex::Real offset = (d==dir) ? 0 : myhalf;
                 amrex::Real lo = plo[d] + ( m_dom.smallEnd(d) - offset ) * dx[d];
                 amrex::Real hi = plo[d] + ( m_dom.bigEnd(d)   + offset ) * dx[d];
  
                 m_rb.setLo(d,lo);
                 m_rb.setHi(d,hi);
  
                 is_per[d] = geom[lev].isPeriodic(d);
             }
             m_geom[0].define(m_dom, &m_rb, geom[lev].Coord(), is_per.data());
  
             // Create plotfile name
             std::string name_line = m_name[iline];
             name_line += "_step_";
             std::string plotfilename = amrex::Concatenate(name_line, m_level_steps[0], 5);
  
             // Get the data
             amrex::Vector<const amrex::MultiFab*> mf = {&(m_ls_mf[iline])};
  
             // Write each line
             WriteMultiLevelPlotfile(plotfilename, 1, mf,
                                     m_varnames, m_geom, static_cast<amrex::Real>(m_time),
                                     m_level_steps, m_ref_ratio);
         }
     }

Referenced by write_sample_data().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ write_sample_data()

void LineSampler::write_sample_data	(	amrex::Vector< double > &	time,
		amrex::Vector< int > &	level_steps,
		amrex::Vector< amrex::IntVect > &	ref_ratio,
		amrex::Vector< amrex::Geometry > &	geom
	)

inline

     {
         if (m_write_ascii) {
             write_line_ascii(time);
         } else {
             write_line_plotfile(time, level_steps, ref_ratio, geom);
         }
     }