#include <OperatorSplittingVarForm.hpp>

Inheritance diagram for polyfem::varform::OperatorSplittingVarForm:

Collaboration diagram for polyfem::varform::OperatorSplittingVarForm:

Public Member Functions
std::string	name () const override
	Get the name of the variational formulation.

void	init (const std::string &formulation, const Units &units, const json &args, const std::string &out_path) override
	Initialize the variational formulation with the given parameters.

void	save_json (const Eigen::MatrixXd &solution, std::ostream &out) const override
	Save the solution to a JSON file, for output purposes.

void	export_data (const Eigen::MatrixXd &solution) const override

io::OutputSpace	output_space () const override
	Get the output space of the variational formulation, for output purposes.

io::OutStatsData	compute_errors (const Eigen::MatrixXd &solution) override
	Get the error statistics of the variational formulation, for output purposes.

std::vector< io::OutputField >	output_fields (const io::OutputSample &sample, const Eigen::MatrixXd &solution, const io::OutputFieldOptions &options) const override
	Get the output fields of the variational formulation, for output purposes.

Public Member Functions inherited from polyfem::varform::VarForm
virtual	~VarForm ()=default

void	set_args (const json &args)
	Reset the internal state of the variational formulation, e.g.

void	set_mesh (std::unique_ptr< mesh::Mesh > mesh, const double loading_mesh_time=0)
	Set the mesh for the variational formulation.

void	solve (Eigen::MatrixXd &sol)
	Solve the variational formulation and store the solution in the given matrix.

void	set_time_callback (const std::function< void(int, int, double, double)> &callback)

int	problem_dimension () const
	Get the problem dimension of the variational formulation, for output purposes.

virtual bool	is_contact_enabled () const
	Check if contact is enabled for the variational formulation, for output purposes.

const io::OutRuntimeData &	output_timings () const
	Get the runtime timings of the variational formulation, for output purposes.

void	save_json (const Eigen::MatrixXd &solution) const
	Save the solution to a JSON file, for output purposes.

Protected Member Functions
void	reset () override

void	load_mesh (const mesh::Mesh &mesh, const json &args) override

void	build_basis (mesh::Mesh &mesh, const bool iso_parametric, const json &args) override

void	assemble_rhs (const mesh::Mesh &mesh) override

void	assemble_mass_mat (const mesh::Mesh &mesh, const json &args) override

void	solve_problem (Eigen::MatrixXd &sol) override

void	build_rhs_assembler () override

Protected Member Functions inherited from polyfem::varform::VarForm
std::string	resolve_output_path (const std::string &path) const

std::string	resolve_input_path (const std::string &path, const bool only_if_exists=false) const

void	set_materials (assembler::Assembler &assembler, const int size) const

void	assign_discr_orders (const json &discr_order, const mesh::Mesh &mesh, Eigen::VectorXi &disc_orders)

void	assign_discr_orders (const json &discr_order, const int fe_space_id, const mesh::Mesh &mesh, Eigen::VectorXi &disc_orders)

void	prepare ()

QuadratureOrders	n_boundary_samples (const int discr_order, const int gdiscr_order) const

void	build_fe_space (mesh::Mesh &mesh, const bool iso_parametric, const Eigen::VectorXi &disc_orders, const std::string &basis_type, const std::string &poly_basis_type, const assembler::Assembler &space_assembler, const int value_dim, const int quadrature_order, const int mass_quadrature_order, const bool use_corner_quadrature, const int n_harmonic_samples, const int integral_constraints, FESpace &space, VarFormBoundaryState &boundary, std::shared_ptr< GeometryMapping > geometry=nullptr)

void	save_step_state (const double t0, const double dt, const int t, const time_integrator::ImplicitTimeIntegrator *time_integrator, const bool rest_mesh_written=false) const

void	ensure_output_sampler () const

void	save_restart_json (const double t0, const double dt, const int t, const bool rest_mesh_written) const

void	save_timestep (const double time, const int t, const double t0, const double dt, const Eigen::MatrixXd &solution) const

void	save_subsolve (const int i, const int t, const Eigen::MatrixXd &solution) const

int	output_file_index (const int t) const

void	notify_time_step (const int t, const int time_steps, const double t0, const double dt) const

io::OutGeometryData::ExportOptions	export_options (const io::OutputSpace &space) const

io::OutputFieldFunction	output_field_function (const Eigen::MatrixXd &solution, const io::OutGeometryData::ExportOptions &opts) const

Private Member Functions
int	primary_ndof () const

int	pressure_block_size () const

int	stacked_ndof () const

void	prepare_initial_solution (Eigen::MatrixXd &sol) const

void	split_solution (const Eigen::MatrixXd &stacked, Eigen::MatrixXd &primary, Eigen::MatrixXd &pressure) const

void	stack_solution (const Eigen::MatrixXd &primary, const Eigen::MatrixXd &pressure, Eigen::MatrixXd &stacked) const

Private Attributes
FESpace	space_

FESpace	pressure_space_

VarFormBoundaryState	boundary_

VarFormBoundaryState	pressure_boundary_

assembler::AssemblyValsCache	ass_vals_cache_

assembler::AssemblyValsCache	pressure_ass_vals_cache_

assembler::AssemblyValsCache	mass_ass_vals_cache_

std::shared_ptr< assembler::RhsAssembler >	rhs_assembler_

StiffnessMatrix	mass_

Eigen::MatrixXd	rhs_

std::shared_ptr< assembler::Assembler >	primary_assembler_ = nullptr

std::shared_ptr< assembler::Mass >	mass_assembler_ = nullptr

std::shared_ptr< assembler::MixedAssembler >	mixed_assembler_ = nullptr

std::shared_ptr< assembler::Assembler >	pressure_assembler_ = nullptr

bool	use_avg_pressure = true

double	avg_mass_ = 0

double	t0 = 0

int	time_steps = 0

double	dt = 0

Additional Inherited Members
Static Protected Member Functions inherited from polyfem::varform::VarForm
static bool	read_initial_x_from_file (const std::string &state_path, const std::string &x_name, const bool reorder, const Eigen::VectorXi &in_node_to_node, const int dim, Eigen::MatrixXd &x)

static void	rebuild_node_positions (const std::vector< basis::ElementBases > &bases, const std::vector< int > &node_ids, std::vector< RowVectorNd > &positions)

Protected Attributes inherited from polyfem::varform::VarForm
std::shared_ptr< assembler::Problem >	problem
	current problem, it contains rhs and bc

Units	units

json	args

io::OutStatsData	stats

io::OutRuntimeData	timings
	runtime statistics

std::string	root_path

std::string	output_path

std::unique_ptr< mesh::Mesh >	mesh_

std::function< void(int, int, double, double)>	time_callback

io::OutGeometryData	output_geometry_

bool	output_sampler_initialized_ = false

bool	prepared_ = false

Detailed Description

Definition at line 10 of file OperatorSplittingVarForm.hpp.

Member Function Documentation

◆ assemble_mass_mat()

void polyfem::varform::OperatorSplittingVarForm::assemble_mass_mat	(	const mesh::Mesh &	mesh,
		const json &	args
	)

overrideprotectedvirtual

Implements polyfem::varform::VarForm.

Definition at line 492 of file OperatorSplittingVarForm.cpp.

References polyfem::varform::VarForm::args, polyfem::io::OutRuntimeData::assembling_mass_mat_time, avg_mass_, mass_, and polyfem::varform::VarForm::timings.

◆ assemble_rhs()

void polyfem::varform::OperatorSplittingVarForm::assemble_rhs ( const mesh::Mesh & mesh )

overrideprotectedvirtual

Implements polyfem::varform::VarForm.

Definition at line 455 of file OperatorSplittingVarForm.cpp.

References polyfem::io::OutRuntimeData::assigning_rhs_time, polyfem::mesh::Mesh::bounding_box(), build_rhs_assembler(), polyfem::mesh::Mesh::is_volume(), polyfem::logger(), mass_assembler_, pressure_block_size(), primary_assembler_, polyfem::varform::VarForm::problem, rhs_, rhs_assembler_, polyfem::varform::VarForm::root_path, and polyfem::varform::VarForm::timings.

Here is the call graph for this function:

◆ build_basis()

void polyfem::varform::OperatorSplittingVarForm::build_basis	(	mesh::Mesh &	mesh,
		const bool	iso_parametric,
		const json &	args
	)

overrideprotectedvirtual

Implements polyfem::varform::VarForm.

Definition at line 310 of file OperatorSplittingVarForm.cpp.

Here is the call graph for this function:

◆ build_rhs_assembler()

void polyfem::varform::OperatorSplittingVarForm::build_rhs_assembler ( )

overrideprotectedvirtual

Implements polyfem::varform::VarForm.

Definition at line 293 of file OperatorSplittingVarForm.cpp.

References polyfem::varform::VarForm::args, polyfem::varform::FESpace::basis_list(), boundary_, polyfem::varform::VarFormBoundaryState::dirichlet_nodes, polyfem::varform::VarFormBoundaryState::dirichlet_nodes_position, polyfem::varform::FESpace::geometry_basis_list(), mass_ass_vals_cache_, polyfem::varform::VarForm::mesh_, polyfem::varform::FESpace::n_bases, polyfem::varform::VarFormBoundaryState::neumann_nodes, polyfem::varform::VarFormBoundaryState::neumann_nodes_position, primary_assembler_, polyfem::varform::VarForm::problem, rhs_assembler_, and space_.

Referenced by assemble_rhs(), and build_basis().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ compute_errors()

io::OutStatsData polyfem::varform::OperatorSplittingVarForm::compute_errors ( const Eigen::MatrixXd & solution )

overridevirtual

Get the error statistics of the variational formulation, for output purposes.

Returns: Error statistics

Implements polyfem::varform::VarForm.

Definition at line 152 of file OperatorSplittingVarForm.cpp.

References polyfem::varform::VarForm::args, polyfem::varform::FESpace::basis_list(), polyfem::io::OutStatsData::compute_errors(), dt, polyfem::varform::FESpace::geometry_basis_list(), polyfem::varform::VarForm::mesh_, polyfem::varform::FESpace::n_bases, polyfem::varform::VarForm::problem, space_, split_solution(), polyfem::varform::VarForm::stats, t0, and time_steps.

Here is the call graph for this function:

◆ export_data()

void polyfem::varform::OperatorSplittingVarForm::export_data ( const Eigen::MatrixXd & solution ) const

overridevirtual

Implements polyfem::varform::VarForm.

Definition at line 172 of file OperatorSplittingVarForm.cpp.

Here is the call graph for this function:

◆ init()

void polyfem::varform::OperatorSplittingVarForm::init	(	const std::string &	formulation,
		const Units &	units,
		const json &	args,
		const std::string &	out_path
	)

overridevirtual

Initialize the variational formulation with the given parameters.

Parameters

formulation	name of the variational formulation
units	unit system to use for the formulation
args	json input arguments, used to initialize the formulation
out_path	output path for the formulation, used to save intermediate data

Reimplemented from polyfem::varform::VarForm.

Definition at line 48 of file OperatorSplittingVarForm.cpp.

References polyfem::varform::VarForm::args, dt, polyfem::problem::ProblemFactory::factory(), polyfem::problem::ProblemFactory::get_problem(), polyfem::varform::VarForm::init(), polyfem::assembler::AssemblerUtils::make_assembler(), polyfem::assembler::AssemblerUtils::make_mixed_assembler(), mass_assembler_, mixed_assembler_, polyfem::assembler::AssemblerUtils::other_assembler_name(), pressure_assembler_, primary_assembler_, polyfem::varform::VarForm::problem, polyfem::varform::VarForm::root_path, t0, time_steps, and polyfem::varform::VarForm::units.

Here is the call graph for this function:

◆ load_mesh()

void polyfem::varform::OperatorSplittingVarForm::load_mesh	(	const mesh::Mesh &	mesh,
		const json &	args
	)

overrideprotectedvirtual

Implements polyfem::varform::VarForm.

Definition at line 265 of file OperatorSplittingVarForm.cpp.

References polyfem::varform::VarForm::args, polyfem::mesh::Mesh::dimension(), mass_assembler_, mixed_assembler_, pressure_assembler_, primary_assembler_, polyfem::varform::VarForm::problem, and polyfem::varform::VarForm::set_materials().

Here is the call graph for this function:

◆ name()

std::string polyfem::varform::OperatorSplittingVarForm::name ( ) const

inlineoverridevirtual

Get the name of the variational formulation.

Returns: Name of the variational formulation

Implements polyfem::varform::VarForm.

Definition at line 13 of file OperatorSplittingVarForm.hpp.

Referenced by save_json().

Here is the caller graph for this function:

◆ output_fields()

std::vector< io::OutputField > polyfem::varform::OperatorSplittingVarForm::output_fields	(	const io::OutputSample &	sample,
		const Eigen::MatrixXd &	solution,
		const io::OutputFieldOptions &	options
	)		const

overridevirtual

Get the output fields of the variational formulation, for output purposes.

Parameters

sample	Output sample
solution	Solution matrix
options	Output field options

Returns: Output fields

Implements polyfem::varform::VarForm.

Definition at line 561 of file OperatorSplittingVarForm.cpp.

References polyfem::varform::FESpace::basis_list(), polyfem::io::OutputSample::element_ids, polyfem::io::OutputFieldOptions::export_field(), polyfem::io::OutputFieldOptions::fields, polyfem::varform::FESpace::geometry_basis_list(), polyfem::io::Evaluator::interpolate_at_local_vals(), polyfem::io::OutputSample::local_points, polyfem::varform::VarForm::mesh_, polyfem::io::OutputSample::node_ids, polyfem::io::OutputField::Point, polyfem::io::OutputSample::points, polyfem::varform::VarForm::problem, space_, and split_solution().

Here is the call graph for this function:

◆ output_space()

io::OutputSpace polyfem::varform::OperatorSplittingVarForm::output_space ( ) const

overridevirtual

Get the output space of the variational formulation, for output purposes.

Returns: Output space

Implements polyfem::varform::VarForm.

Definition at line 127 of file OperatorSplittingVarForm.cpp.

References boundary_, polyfem::varform::VarFormBoundaryState::dirichlet_nodes, polyfem::varform::VarFormBoundaryState::dirichlet_nodes_position, polyfem::varform::FESpace::disc_orders, polyfem::varform::FESpace::disc_ordersq, polyfem::varform::FESpace::geometry_basis_list(), polyfem::varform::VarForm::mesh_, polyfem::varform::FESpace::polys, polyfem::varform::FESpace::polys_3d, space_, and polyfem::varform::VarFormBoundaryState::total_local_boundary.

Referenced by export_data().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ prepare_initial_solution()

void polyfem::varform::OperatorSplittingVarForm::prepare_initial_solution ( Eigen::MatrixXd & sol ) const

private

Definition at line 501 of file OperatorSplittingVarForm.cpp.

References polyfem::varform::VarForm::args, polyfem::varform::VarForm::mesh_, pressure_block_size(), polyfem::varform::VarForm::problem, polyfem::varform::VarForm::read_initial_x_from_file(), polyfem::varform::VarForm::resolve_input_path(), rhs_, rhs_assembler_, space_, polyfem::varform::FESpace::space_in_node_to_node, and stacked_ndof().

Here is the call graph for this function:

◆ pressure_block_size()

int polyfem::varform::OperatorSplittingVarForm::pressure_block_size ( ) const

private

Definition at line 283 of file OperatorSplittingVarForm.cpp.

References polyfem::varform::FESpace::n_bases, pressure_space_, and use_avg_pressure.

Referenced by assemble_rhs(), prepare_initial_solution(), and stacked_ndof().

Here is the caller graph for this function:

◆ primary_ndof()

int polyfem::varform::OperatorSplittingVarForm::primary_ndof ( ) const

private

Definition at line 278 of file OperatorSplittingVarForm.cpp.

References polyfem::varform::VarForm::mesh_, polyfem::varform::FESpace::n_bases, and space_.

Referenced by export_data(), save_json(), split_solution(), stack_solution(), and stacked_ndof().

Here is the caller graph for this function:

◆ reset()

void polyfem::varform::OperatorSplittingVarForm::reset ( )

overrideprotectedvirtual

Implements polyfem::varform::VarForm.

Definition at line 24 of file OperatorSplittingVarForm.cpp.

References ass_vals_cache_, avg_mass_, boundary_, dt, polyfem::assembler::AssemblyValsCache::init_empty(), mass_, mass_ass_vals_cache_, mass_assembler_, mixed_assembler_, pressure_ass_vals_cache_, pressure_assembler_, pressure_boundary_, pressure_space_, primary_assembler_, polyfem::varform::FESpace::reset(), polyfem::varform::VarFormBoundaryState::reset(), polyfem::varform::VarForm::reset(), rhs_, rhs_assembler_, space_, t0, time_steps, and use_avg_pressure.

Here is the call graph for this function:

◆ save_json()

void polyfem::varform::OperatorSplittingVarForm::save_json	(	const Eigen::MatrixXd &	solution,
		std::ostream &	out
	)		const

overridevirtual

Save the solution to a JSON file, for output purposes.

Parameters

solution	Solution matrix to save
out	Output stream to save the solution

Implements polyfem::varform::VarForm.

Definition at line 98 of file OperatorSplittingVarForm.cpp.

References polyfem::varform::VarForm::args, polyfem::varform::FESpace::disc_orders, polyfem::varform::FESpace::disc_ordersq, polyfem::varform::FESpace::is_iso_parametric(), polyfem::logger(), polyfem::varform::VarForm::mesh_, polyfem::varform::FESpace::n_bases, name(), pressure_space_, primary_assembler_, primary_ndof(), polyfem::varform::VarForm::problem, polyfem::io::OutStatsData::save_json(), space_, polyfem::varform::VarForm::stats, and polyfem::varform::VarForm::timings.

Here is the call graph for this function:

◆ solve_problem()

void polyfem::varform::OperatorSplittingVarForm::solve_problem ( Eigen::MatrixXd & sol )

overrideprotectedvirtual

Implements polyfem::varform::VarForm.

Definition at line 715 of file OperatorSplittingVarForm.cpp.

References polyfem::solver::OperatorSplittingSolver::advection(), polyfem::solver::OperatorSplittingSolver::advection_FLIP(), polyfem::assembler::Laplacian::assemble(), polyfem::solver::OperatorSplittingSolver::external_force(), polyfem::log_and_throw_error(), polyfem::logger(), polyfem::autogen::p_nodes_2d(), polyfem::autogen::p_nodes_3d(), polyfem::solver::OperatorSplittingSolver::projection(), polyfem::autogen::q_nodes_2d(), polyfem::autogen::q_nodes_3d(), polyfem::assembler::Assembler::set_size(), polyfem::solver::OperatorSplittingSolver::solve_diffusion_1st(), polyfem::solver::OperatorSplittingSolver::solve_pressure(), and space_.

Here is the call graph for this function:

◆ split_solution()

void polyfem::varform::OperatorSplittingVarForm::split_solution	(	const Eigen::MatrixXd &	stacked,
		Eigen::MatrixXd &	primary,
		Eigen::MatrixXd &	pressure
	)		const

private

Definition at line 528 of file OperatorSplittingVarForm.cpp.

References polyfem::varform::FESpace::n_bases, pressure_space_, and primary_ndof().

Referenced by compute_errors(), export_data(), and output_fields().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ stack_solution()

void polyfem::varform::OperatorSplittingVarForm::stack_solution	(	const Eigen::MatrixXd &	primary,
		const Eigen::MatrixXd &	pressure,
		Eigen::MatrixXd &	stacked
	)		const