cfem_diffusion_solver.h

Go to the documentation of this file.

#ifndef CFEM_DIFFUSION_SOLVER_H
#define CFEM_DIFFUSION_SOLVER_H
 
#include "physics/SolverBase/chi_solver.h"
#include "math/PETScUtils/petsc_utils.h"
 
#include "cfem_diffusion_bndry.h"
#include "utils/chi_timer.h"
 
#include "console/chi_console.h"
 
#include <map>
 
#include "mesh/chi_mesh.h"
 
// forward declaration
namespace chi_mesh
{
class MeshContinuum; 
typedef std::shared_ptr<MeshContinuum> MeshContinuumPtr;
}
namespace chi_math
{
class SpatialDiscretization; 
typedef std::shared_ptr<SpatialDiscretization> SDMPtr ;
}
 
namespace cfem_diffusion
{
/** CFEM diffusion solver
 * 
*/
class Solver : public chi_physics::Solver
{
public:
  chi_mesh::MeshContinuumPtr grid_ptr_ = nullptr;
 
  chi_math::SDMPtr sdm_ptr_ = nullptr;
 
  size_t num_local_dofs_ = 0;
  size_t num_globl_dofs_ = 0;
 
  Vec            x_ = nullptr;            // approx solution
  Vec            b_ = nullptr;            // RHS
  Mat            A_ = nullptr;            // linear system matrix
 
  typedef std::pair<BoundaryType,std::vector<double>> BoundaryInfo;
  typedef std::map<std::string, BoundaryInfo> BoundaryPreferences;
  BoundaryPreferences     boundary_preferences_;
  std::map<uint64_t, Boundary>   boundaries_;
 
  explicit Solver(const std::string& in_solver_name);
  ~Solver() override;
 
  // void Initialize() override;
  void Initialize() override;
 
  void Execute() override;
 
  static double CallLua_iXYZFunction(lua_State* L,
                                     const std::string&,
                                     int,
                                     const chi_mesh::Vector3&);
 
  void UpdateFieldFunctions();
};
 
} // namespace cfem_diffusion
 
 
#endif //CFEM_DIFFUSION_SOLVER_H