lbsDO_sweepdata_00d_setswpchunk.cc

Go to the documentation of this file.

#include "lbs_discrete_ordinates_solver.h"
 
#include "SweepChunks/AAH_SweepChunk.h"
#include "SweepChunks/CBC_SweepChunk.h"
 
#include "chi_log_exceptions.h"
 
typedef chi_mesh::sweep_management::SweepChunk SweepChunk;
 
// ###################################################################
/**Sets up the sweek chunk for the given discretization method.*/
std::shared_ptr<SweepChunk>
lbs::DiscreteOrdinatesSolver::SetSweepChunk(LBSGroupset& groupset)
{
  if (sweep_type_ == "AAH")
  {
    auto sweep_chunk = std::make_shared<AAH_SweepChunk>(
      *grid_ptr_,                   // Spatial grid of cells
      *discretization_,             // Spatial discretization
      unit_cell_matrices_,          // Unit cell matrices
      cell_transport_views_,        // Cell transport views
      phi_new_local_,               // Destination phi
      psi_new_local_[groupset.id_], // Destination psi
      q_moments_local_,             // Source moments
      groupset,                     // Reference groupset
      matid_to_xs_map_,             // Material cross-sections
      num_moments_,
      max_cell_dof_count_);
 
    return sweep_chunk;
  }
  else if (sweep_type_ == "CBC")
  {
    auto sweep_chunk = std::make_shared<CBC_SweepChunk>(
      phi_new_local_, psi_new_local_[groupset.id_],
      *grid_ptr_,
      *discretization_,
      unit_cell_matrices_,
      cell_transport_views_,
      q_moments_local_,
      groupset,
      matid_to_xs_map_,
      num_moments_,
      max_cell_dof_count_);
 
    return sweep_chunk;
  }
  else
    ChiLogicalError("Unsupported sweep_type_ \"" + sweep_type_ + "\"");
}