mirror of
https://codeberg.org/vcbferreira/NuFI_deal.ii
synced 2026-08-12 22:43:17 +02:00
117 lines
3.2 KiB
C++
117 lines
3.2 KiB
C++
#ifndef GRIDS_H
|
|
#define GRIDS_H
|
|
|
|
#include "nufi/cells.h"
|
|
#include <deal.II/dofs/dof_handler.h>
|
|
#include <deal.II/fe/fe_q.h>
|
|
#include <deal.II/fe/mapping_q.h>
|
|
#include <deal.II/grid/tria.h>
|
|
#include <deal.II/matrix_free/fe_point_evaluation.h>
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
using namespace dealii;
|
|
|
|
template <int dim> class PoissonProblem;
|
|
|
|
template <int dim> struct GridStructure {
|
|
//==//==//
|
|
// Vars //
|
|
//==//==//
|
|
std::unique_ptr<Triangulation<dim>> triangulation;
|
|
std::unique_ptr<DoFHandler<dim>> dof_handler;
|
|
std::unique_ptr<MappingQ<dim>> mapping;
|
|
std::unique_ptr<FE_Q<dim>> fe;
|
|
|
|
CellLocator<dim> locator;
|
|
|
|
unsigned int grid_version;
|
|
|
|
// === // === //
|
|
// Evaluator //
|
|
// === // === //
|
|
std::vector<double>
|
|
eval_vector_grad(const Vector<double> &solution,
|
|
const std::vector<Point<dim>> &points) const {
|
|
|
|
std::vector<double> values(points.size());
|
|
#pragma omp parallel
|
|
{
|
|
std::vector<double> local_solution_buffer(fe->n_dofs_per_cell());
|
|
FEPointEvaluation<dim, dim> evaluator(*mapping, *fe, update_gradients);
|
|
#pragma omp for
|
|
for (unsigned int p = 0; p < points.size(); ++p) {
|
|
|
|
const auto cell_location = locator.locate(points[p]);
|
|
|
|
cell_location.info->cell->get_dof_values(solution,
|
|
local_solution_buffer.begin(),
|
|
local_solution_buffer.end());
|
|
|
|
evaluator.reinit(
|
|
cell_location.info->cell,
|
|
ArrayView<const Point<dim>>(&cell_location.reference_point, 1));
|
|
|
|
evaluator.evaluate(local_solution_buffer, EvaluationFlags::gradients);
|
|
|
|
values[p] = evaluator.get_gradient(0)[0];
|
|
}
|
|
}
|
|
return values;
|
|
}
|
|
};
|
|
|
|
template <int dim>
|
|
GridStructure<dim> make_grid_snapshot(const PoissonProblem<dim> &poisson) {
|
|
GridStructure<dim> grid;
|
|
grid.grid_version = 0;
|
|
|
|
grid.triangulation = std::make_unique<Triangulation<dim>>();
|
|
grid.triangulation->copy_triangulation(poisson.get_triangulation());
|
|
|
|
grid.mapping = std::make_unique<MappingQ<dim>>(poisson.get_mapping());
|
|
|
|
grid.dof_handler = std::make_unique<DoFHandler<dim>>(*grid.triangulation);
|
|
grid.dof_handler->distribute_dofs(poisson.get_dof_handler().get_fe());
|
|
|
|
grid.locator.rebuild(*grid.dof_handler, *grid.triangulation);
|
|
|
|
grid.fe = std::make_unique<FE_Q<dim>>(poisson.get_fe());
|
|
|
|
return grid;
|
|
}
|
|
|
|
template <int dim> struct SolutionSnapshot {
|
|
unsigned int grid_version;
|
|
Vector<double> solution;
|
|
};
|
|
|
|
template <int dim>
|
|
inline void update_grid_versions(std::vector<GridStructure<dim>> &grid_versions,
|
|
PoissonProblem<dim> &poisson) {
|
|
auto grid = make_grid_snapshot(poisson);
|
|
|
|
if (!grid_versions.empty())
|
|
grid.grid_version = grid_versions.back().grid_version + 1;
|
|
|
|
grid_versions.push_back(std::move(grid));
|
|
}
|
|
|
|
template <int dim>
|
|
inline void
|
|
update_solution_history(std::vector<SolutionSnapshot<dim>> &solution_history,
|
|
PoissonProblem<dim> &poisson,
|
|
unsigned int current_grid_version) {
|
|
|
|
SolutionSnapshot<dim> snapshot;
|
|
|
|
snapshot.grid_version = current_grid_version;
|
|
|
|
Vector<double> solution = poisson.get_solution();
|
|
snapshot.solution = solution;
|
|
|
|
solution_history.push_back(std::move(snapshot));
|
|
}
|
|
|
|
#endif // !GRIDS_H
|