mirror of
https://codeberg.org/vcbferreira/NuFI_deal.ii
synced 2026-08-12 22:43:17 +02:00
109 lines
3.1 KiB
C++
109 lines
3.1 KiB
C++
#ifndef NUFI_FIELDS_H_
|
|
#define NUFI_FIELDS_H_
|
|
|
|
#include "nufi/grids.h"
|
|
#include "nufi/parameters.h"
|
|
|
|
#include <cmath>
|
|
#include <cstddef>
|
|
#include <deal.II/base/function.h>
|
|
#include <deal.II/base/point.h>
|
|
#include <vector>
|
|
|
|
using namespace dealii;
|
|
|
|
inline std::vector<double> make_x_eval(size_t Nx) {
|
|
std::vector<double> x_eval_E;
|
|
|
|
double dx = (Parameters::X_DOMAIN_RIGHT - Parameters::X_DOMAIN_LEFT) / Nx;
|
|
|
|
for (unsigned int i = 0; i < Nx; ++i)
|
|
x_eval_E.push_back(Parameters::X_DOMAIN_LEFT + (i + 0.5) * dx);
|
|
|
|
return x_eval_E;
|
|
}
|
|
|
|
inline void reset_x_eval(std::vector<double> &x_vals) {
|
|
const size_t Nx = x_vals.size();
|
|
const double dx = Parameters::LX / Nx;
|
|
for (size_t i = 0; i < Nx; ++i)
|
|
x_vals[i] = Parameters::X_DOMAIN_LEFT + i * dx;
|
|
};
|
|
|
|
inline double f0(const double x, const double v,
|
|
const size_t f0_type = Parameters::f0_TYPE) {
|
|
const double eps = Parameters::EPS;
|
|
const double k = Parameters::WAVE_NR;
|
|
|
|
double result;
|
|
|
|
switch (f0_type) {
|
|
case 0:
|
|
|
|
const double prefactor =
|
|
Parameters::F0_FACTOR * (1.0 + eps * std::cos(k * x));
|
|
const double gaussian = v * v * std::exp(-0.5 * v * v);
|
|
result = prefactor * gaussian;
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
// wrapper for eval_point() { VectorTools::point_values() }
|
|
inline std::vector<double> eval(std::vector<double> &X,
|
|
const GridStructure<1> &grid,
|
|
const Vector<double> &solution) noexcept {
|
|
AssertThrow(grid.dof_handler->n_dofs() == solution.size(),
|
|
ExcMessage("@ eval(...) grid's number of DoFs doesn't correspond "
|
|
"to solution's size"));
|
|
size_t x_size = X.size();
|
|
std::vector<Point<1>> Points(x_size);
|
|
|
|
for (size_t i = 0; i < x_size; ++i) {
|
|
X[i] = X[i] - Parameters::X_DOMAIN_LEFT;
|
|
X[i] = X[i] - Parameters::LX * std::floor(X[i] * Parameters::LX_INV);
|
|
|
|
Points[i][0] = X[i];
|
|
}
|
|
|
|
return grid.eval_vector_grad(solution, Points);
|
|
}
|
|
|
|
inline double integral_space_vector(const GridStructure<1> &grid,
|
|
const Vector<double> &solution,
|
|
size_t Nx = Parameters::PLOT_NX) {
|
|
double integral = 0.0;
|
|
std::vector<double> x_eval = make_x_eval(Nx);
|
|
const double dx = Parameters::LX / Nx;
|
|
|
|
std::vector<double> tmp = eval(x_eval, grid, solution);
|
|
for (size_t i = 0; i < Nx; ++i)
|
|
integral += tmp[i];
|
|
return integral * dx;
|
|
}
|
|
|
|
inline double integral_space_vector_squared(const GridStructure<1> &grid,
|
|
const Vector<double> &solution,
|
|
size_t Nx = Parameters::PLOT_NX) {
|
|
double integral = 0.0;
|
|
std::vector<double> x_eval = make_x_eval(Nx);
|
|
const double dx = Parameters::LX / Nx;
|
|
|
|
std::vector<double> tmp = eval(x_eval, grid, solution);
|
|
for (size_t i = 0; i < Nx; ++i)
|
|
integral += tmp[i] * tmp[i];
|
|
return integral * dx;
|
|
}
|
|
|
|
inline std::vector<double>
|
|
Point_vector_to_double_vector(const std::vector<Point<1>> &Points) {
|
|
const size_t n_points = Points.size();
|
|
std::vector<double> vector(n_points);
|
|
|
|
for (size_t i = 0; i < n_points; ++i)
|
|
vector[i] = Points[i][0];
|
|
|
|
return vector;
|
|
}
|
|
#endif // NUFI_FIELDS_H_
|