CBC_FLUDS.cc

Go to the documentation of this file.

#include "CBC_FLUDS.h"
 
#include "math/SpatialDiscretization/SpatialDiscretization.h"
#include "mesh/SweepUtilities/SPDS/SPDS.h"
#include "mesh/MeshContinuum/chi_meshcontinuum.h"
 
namespace lbs
{
 
CBC_FLUDS::CBC_FLUDS(size_t num_groups,
                     size_t num_angles,
                     const CBC_FLUDSCommonData& common_data,
                     std::vector<double>& local_psi_data,
                     const chi_math::UnknownManager& psi_uk_man,
                     const chi_math::SpatialDiscretization& sdm)
  : chi_mesh::sweep_management::FLUDS(
      num_groups, num_angles, common_data.GetSPDS()),
    common_data_(common_data),
    local_psi_data_(local_psi_data),
    psi_uk_man_(psi_uk_man),
    sdm_(sdm)
{
}
 
const chi_mesh::sweep_management::FLUDSCommonData& CBC_FLUDS::CommonData() const
{
  return common_data_;
}
 
const std::vector<double>& CBC_FLUDS::GetLocalUpwindDataBlock() const
{
  return local_psi_data_;
}
 
const double*
CBC_FLUDS::GetLocalCellUpwindPsi(const std::vector<double>& psi_data_block,
                             const chi_mesh::Cell& cell)
{
  const auto dof_map =
    sdm_.MapDOFLocal(cell, 0, psi_uk_man_, 0, 0);
 
  return &psi_data_block[dof_map];
}
 
const std::vector<double>&
CBC_FLUDS::GetNonLocalUpwindData(uint64_t cell_global_id,
                                 unsigned int face_id) const
{
  return deplocs_outgoing_messages_.at({cell_global_id, face_id});
}
 
const double*
CBC_FLUDS::GetNonLocalUpwindPsi(const std::vector<double>& psi_data,
                                unsigned int face_node_mapped,
                                unsigned int angle_set_index)
{
  const size_t dof_map =
    face_node_mapped * num_groups_and_angles_ + angle_set_index * num_groups_;
 
  return &psi_data[dof_map];
}
 
} // namespace lbs