#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.

void	serialize_checkpoint (io::CheckpointWriter &writer, const Eigen::MatrixXd &solution, const io::CheckpointMetadata &metadata) const override

void	deserialize_checkpoint (const io::CheckpointReader &reader, Eigen::MatrixXd &solution) override

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.

	VarForm (const io::ResourceIO &resources)

Public Member Functions inherited from polyfem::varform::VarForm
	VarForm (const io::ResourceIO &resources)

virtual	~VarForm ()=default

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

void	set_checkpoint_reader (const io::CheckpointReader &reader)

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

void	set_geometry (mesh::LoadedGeometry geometry, double loading_mesh_time=0)

void	solve (Eigen::MatrixXd &sol, const InitialConditionOverride *initial_condition_override=nullptr, const ForwardStepCallback &post_step={})
	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, const InitialConditionOverride *initial_condition_override, const ForwardStepCallback &post_step) override

void	build_rhs_assembler () override

Protected Member Functions inherited from polyfem::varform::VarForm
void	prepare ()
	Prepare all discretization and assembly data without running a solve.

std::string	resolve_output_path (const std::string &path) const

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

void	restore_checkpoint_integrator (const std::shared_ptr< time_integrator::ImplicitTimeIntegrator > &integrator, const std::string &group, double dt, const std::string &ordering, const FESpace &space, int block_size) const

void	write_checkpoint_ordering (io::CheckpointWriter &writer, const std::string &name, const FESpace &space) const

void	reorder_checkpoint_block (const io::CheckpointReader &reader, const std::string &name, const FESpace &space, int block_size, int row_offset, Eigen::MatrixXd &value) const

Eigen::VectorXi	checkpoint_ordering (const FESpace &space) const

void	validate_checkpoint_solution (const Eigen::MatrixXd &solution, int expected_rows) const

bool	checkpoint_reorder_enabled () const

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

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

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

void	build_fe_space (mesh::Mesh &mesh, const bool iso_parametric, const Eigen::VectorXi &disc_orders, const Eigen::VectorXi &disc_ordersq, 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)

virtual void	set_obstacle (mesh::Obstacle &&obstacle)

void	save_step_state (const double t0, const double dt, const int t, const Eigen::MatrixXd &solution, const time_integrator::ImplicitTimeIntegrator *time_integrator) const

void	ensure_output_sampler () const

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

bool	save_timestep_to_vtm (const double time, const int t, const double dt, const Eigen::MatrixXd &solution, paraviewo::VTMWriter &vtm, const std::string &block_prefix) 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

void	init_child_varform (VarForm &child, const std::string &formulation, const Units &units, const json &args, const std::string &out_path) 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 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	output_path

std::optional< std::reference_wrapper< const io::CheckpointReader > >	checkpoint_reader_

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

int	output_index_offset_ = 0

const io::ResourceIO &	resources_

Detailed Description

Definition at line 11 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 533 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 497 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, polyfem::varform::VarForm::resources_, rhs_, rhs_assembler_, 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 350 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 333 of file OperatorSplittingVarForm.cpp.

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 192 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:

◆ deserialize_checkpoint()

void polyfem::varform::OperatorSplittingVarForm::deserialize_checkpoint	(	const io::CheckpointReader &	reader,
		Eigen::MatrixXd &	solution
	)

overridevirtual

Reimplemented from polyfem::varform::VarForm.

Definition at line 36 of file OperatorSplittingVarForm.cpp.

References polyfem::varform::VarForm::deserialize_checkpoint(), polyfem::io::CheckpointReader::exists(), polyfem::log_and_throw_error(), polyfem::varform::VarForm::mesh_, polyfem::varform::FESpace::n_bases, pressure_space_, primary_ndof(), polyfem::io::CheckpointReader::read_matrix(), polyfem::varform::VarForm::reorder_checkpoint_block(), space_, stacked_ndof(), and polyfem::varform::VarForm::validate_checkpoint_solution().

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 212 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 89 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::resources_, 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 305 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 15 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 594 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 167 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 542 of file OperatorSplittingVarForm.cpp.

References pressure_block_size(), polyfem::varform::VarForm::problem, rhs_, rhs_assembler_, 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 323 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 318 of file OperatorSplittingVarForm.cpp.

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

Referenced by deserialize_checkpoint(), 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 65 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 138 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:

◆ serialize_checkpoint()

void polyfem::varform::OperatorSplittingVarForm::serialize_checkpoint	(	io::CheckpointWriter &	writer,
		const Eigen::MatrixXd &	solution,
		const io::CheckpointMetadata &	metadata
	)		const

overridevirtual

Reimplemented from polyfem::varform::VarForm.

Definition at line 22 of file OperatorSplittingVarForm.cpp.

References pressure_space_, polyfem::varform::VarForm::serialize_checkpoint(), space_, split_solution(), polyfem::varform::VarForm::write_checkpoint_ordering(), and polyfem::io::CheckpointWriter::write_matrix().

Here is the call graph for this function:

◆ solve_problem()

void polyfem::varform::OperatorSplittingVarForm::solve_problem	(	Eigen::MatrixXd &	sol,
		const InitialConditionOverride *	initial_condition_override,
		const ForwardStepCallback &	post_step
	)

overrideprotectedvirtual

Implements polyfem::varform::VarForm.

Definition at line 748 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 561 of file OperatorSplittingVarForm.cpp.

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

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

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