#include <IncompressibleElasticVarForm.hpp>

Inheritance diagram for polyfem::varform::IncompressibleElasticVarForm:

Collaboration diagram for polyfem::varform::IncompressibleElasticVarForm:

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.

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.

Public Member Functions inherited from polyfem::varform::ElasticVarForm
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.

	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.

Private 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

int	primary_ndof () 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	build_stiffness_mat (StiffnessMatrix &stiffness)

void	solve_linear_system (const std::unique_ptr< polysolve::linear::Solver > &solver, StiffnessMatrix &A, Eigen::VectorXd &b, const bool compute_spectrum, Eigen::MatrixXd &sol)

void	solve_static_linear (Eigen::MatrixXd &sol)

void	solve_transient_linear (Eigen::MatrixXd &sol)

void	build_rhs_assembler () override

Private Attributes
FESpace	pressure_space_

int	displacement_space_id_ = -1

int	pressure_space_id_ = -1

VarFormBoundaryState	pressure_boundary_

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

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

assembler::AssemblyValsCache	pressure_ass_vals_cache_

std::shared_ptr< time_integrator::ImplicitTimeIntegrator >	time_integrator

Additional Inherited Members
Protected Member Functions inherited from polyfem::varform::ElasticVarForm
void	build_elastic_basis (mesh::Mesh &mesh, const bool iso_parametric, const json &args, const int fe_space_id)

void	initial_solution (Eigen::MatrixXd &solution, const InitialConditionOverride *override=nullptr, const std::string &state_prefix="") const

void	initial_velocity (Eigen::MatrixXd &velocity, const InitialConditionOverride *override=nullptr, const std::string &state_prefix="") const

void	initial_acceleration (Eigen::MatrixXd &acceleration, const InitialConditionOverride *override=nullptr, const std::string &state_prefix="") const

QuadratureOrders	elastic_boundary_samples () const

std::vector< int >	elastic_primitive_to_node () const

std::vector< int >	elastic_node_to_primitive () const

void	build_mesh_matrices (Eigen::MatrixXd &V, Eigen::MatrixXi &F) const

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

std::vector< io::OutputField >	elastic_output_fields (const io::OutputSample &sample, const Eigen::MatrixXd &solution, const io::OutputFieldOptions &options, const mesh::Obstacle obstacle, const time_integrator::ImplicitTimeIntegrator time_integrator, const std::vector< std::pair< std::string, std::shared_ptr< solver::Form > > > &named_forms, const solver::Form elastic_form, const solver::ContactForm contact_form=nullptr) const

void	append_primary_output_fields (std::vector< io::OutputField > &fields, const io::OutputSample &sample, const Eigen::MatrixXd &solution, const io::OutputFieldOptions &options, const mesh::Obstacle *obstacle=nullptr) const

Eigen::MatrixXd	displaced_output_normals (const io::OutputSample &sample, const Eigen::MatrixXd &solution) const

virtual int	n_obstacle_vertices () const

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

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::ElasticVarForm
FESpace	space_

VarFormBoundaryState	boundary_

assembler::AssemblyValsCache	ass_vals_cache_

assembler::AssemblyValsCache	mass_ass_vals_cache_

assembler::AssemblyValsCache	pure_mass_ass_vals_cache_

std::shared_ptr< assembler::RhsAssembler >	rhs_assembler_

StiffnessMatrix	mass_

StiffnessMatrix	pure_mass_

double	avg_mass_ = 0

Eigen::MatrixXd	rhs_

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

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

std::shared_ptr< assembler::HRZMass >	pure_mass_assembler_ = nullptr

double	t0 = 0

int	time_steps = 0

double	dt = 0

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 15 of file IncompressibleElasticVarForm.hpp.

Member Function Documentation

◆ assemble_mass_mat()

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

overrideprivatevirtual

Reimplemented from polyfem::varform::ElasticVarForm.

Definition at line 273 of file IncompressibleElasticVarForm.cpp.

References polyfem::varform::VarForm::args, polyfem::io::OutRuntimeData::assembling_mass_mat_time, polyfem::varform::ElasticVarForm::avg_mass_, polyfem::varform::FESpace::basis_list(), polyfem::varform::FESpace::geometry_basis_list(), polyfem::mesh::Mesh::is_volume(), polyfem::logger(), polyfem::utils::lump_matrix(), polyfem::varform::ElasticVarForm::mass_, polyfem::varform::ElasticVarForm::mass_ass_vals_cache_, polyfem::varform::ElasticVarForm::mass_assembler_, polyfem::io::OutStatsData::mat_size, polyfem::varform::FESpace::n_bases, polyfem::io::OutStatsData::nn_zero, polyfem::io::OutStatsData::num_dofs, polyfem::varform::VarForm::problem, polyfem::varform::ElasticVarForm::space_, polyfem::varform::VarForm::stats, and polyfem::varform::VarForm::timings.

Here is the call graph for this function:

◆ assemble_rhs()

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

overrideprivatevirtual

Reimplemented from polyfem::varform::ElasticVarForm.

Definition at line 265 of file IncompressibleElasticVarForm.cpp.

References polyfem::varform::ElasticVarForm::assemble_rhs(), polyfem::varform::FESpace::n_bases, pressure_space_, and polyfem::varform::ElasticVarForm::rhs_.

Here is the call graph for this function:

◆ build_basis()

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

overrideprivatevirtual

Reimplemented from polyfem::varform::ElasticVarForm.

Definition at line 185 of file IncompressibleElasticVarForm.cpp.

Here is the call graph for this function:

◆ build_rhs_assembler()

void polyfem::varform::IncompressibleElasticVarForm::build_rhs_assembler ( )

overrideprivatevirtual

Reimplemented from polyfem::varform::ElasticVarForm.

Definition at line 168 of file IncompressibleElasticVarForm.cpp.

Referenced by build_basis().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ build_stiffness_mat()

void polyfem::varform::IncompressibleElasticVarForm::build_stiffness_mat ( StiffnessMatrix & stiffness )

private

Definition at line 331 of file IncompressibleElasticVarForm.cpp.

Referenced by solve_static_linear(), and solve_transient_linear().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ compute_errors()

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

overridevirtual

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

Returns: Error statistics

Reimplemented from polyfem::varform::ElasticVarForm.

Definition at line 134 of file IncompressibleElasticVarForm.cpp.

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

Here is the call graph for this function:

◆ deserialize_checkpoint()

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

overridevirtual

Reimplemented from polyfem::varform::ElasticVarForm.

Definition at line 34 of file IncompressibleElasticVarForm.cpp.

References polyfem::varform::VarForm::deserialize_checkpoint(), polyfem::varform::VarForm::mesh_, pressure_space_, primary_ndof(), polyfem::varform::VarForm::reorder_checkpoint_block(), polyfem::varform::ElasticVarForm::space_, stacked_ndof(), and polyfem::varform::VarForm::validate_checkpoint_solution().

Here is the call graph for this function:

◆ init()

void polyfem::varform::IncompressibleElasticVarForm::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::ElasticVarForm.

Definition at line 61 of file IncompressibleElasticVarForm.cpp.

References polyfem::varform::VarForm::args, displacement_space_id_, polyfem::varform::ElasticVarForm::init(), polyfem::log_and_throw_error(), polyfem::assembler::AssemblerUtils::make_assembler(), polyfem::assembler::AssemblerUtils::make_mixed_assembler(), mixed_assembler_, polyfem::assembler::AssemblerUtils::other_assembler_name(), pressure_assembler_, pressure_space_id_, polyfem::varform::ElasticVarForm::primary_assembler_, and polyfem::varform::VarForm::units.

Here is the call graph for this function:

◆ load_mesh()

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

overrideprivatevirtual

Reimplemented from polyfem::varform::ElasticVarForm.

Definition at line 149 of file IncompressibleElasticVarForm.cpp.

References polyfem::varform::VarForm::args, polyfem::mesh::Mesh::dimension(), polyfem::varform::ElasticVarForm::load_mesh(), mixed_assembler_, pressure_assembler_, and polyfem::varform::VarForm::set_materials().

Here is the call graph for this function:

◆ name()

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

inlineoverridevirtual

Get the name of the variational formulation.

Returns: Name of the variational formulation

Implements polyfem::varform::VarForm.

Definition at line 19 of file IncompressibleElasticVarForm.hpp.

Referenced by save_json().

Here is the caller graph for this function:

◆ output_fields()

std::vector< io::OutputField > polyfem::varform::IncompressibleElasticVarForm::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 476 of file IncompressibleElasticVarForm.cpp.

References polyfem::varform::FESpace::basis_list(), polyfem::varform::ElasticVarForm::elastic_output_fields(), polyfem::io::OutputFieldOptions::export_field(), polyfem::io::OutputFieldOptions::fields, polyfem::varform::FESpace::geometry_basis_list(), polyfem::varform::VarForm::mesh_, polyfem::io::OutputField::Point, pressure_space_, polyfem::varform::internal::sample_scalar_field(), polyfem::varform::ElasticVarForm::space_, split_solution(), and time_integrator.

Here is the call graph for this function:

◆ prepare_initial_solution()

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

private

Definition at line 305 of file IncompressibleElasticVarForm.cpp.

References polyfem::varform::ElasticVarForm::initial_solution(), polyfem::varform::FESpace::n_bases, pressure_space_, and stacked_ndof().

Referenced by solve_problem().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ primary_ndof()

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

private

Definition at line 158 of file IncompressibleElasticVarForm.cpp.

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

Referenced by deserialize_checkpoint(), save_json(), solve_linear_system(), solve_transient_linear(), split_solution(), and stacked_ndof().

Here is the caller graph for this function:

◆ reset()

void polyfem::varform::IncompressibleElasticVarForm::reset ( )

overrideprivatevirtual

Reimplemented from polyfem::varform::ElasticVarForm.

Definition at line 48 of file IncompressibleElasticVarForm.cpp.

References displacement_space_id_, polyfem::assembler::AssemblyValsCache::init_empty(), mixed_assembler_, pressure_ass_vals_cache_, pressure_assembler_, pressure_boundary_, pressure_space_, pressure_space_id_, polyfem::varform::FESpace::reset(), polyfem::varform::VarFormBoundaryState::reset(), polyfem::varform::ElasticVarForm::reset(), and time_integrator.

Here is the call graph for this function:

◆ save_json()

void polyfem::varform::IncompressibleElasticVarForm::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

Reimplemented from polyfem::varform::ElasticVarForm.

Definition at line 105 of file IncompressibleElasticVarForm.cpp.

Here is the call graph for this function:

◆ serialize_checkpoint()

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

overridevirtual

Reimplemented from polyfem::varform::ElasticVarForm.

Definition at line 24 of file IncompressibleElasticVarForm.cpp.

References pressure_space_, polyfem::varform::VarForm::serialize_checkpoint(), polyfem::varform::ElasticVarForm::space_, and polyfem::varform::VarForm::write_checkpoint_ordering().

Here is the call graph for this function:

◆ solve_linear_system()

void polyfem::varform::IncompressibleElasticVarForm::solve_linear_system	(	const std::unique_ptr< polysolve::linear::Solver > &	solver,
		StiffnessMatrix &	A,
		Eigen::VectorXd &	b,
		const bool	compute_spectrum,
		Eigen::MatrixXd &	sol
	)

private

Definition at line 355 of file IncompressibleElasticVarForm.cpp.

References polyfem::varform::VarForm::args, polyfem::varform::ElasticVarForm::boundary_, polyfem::varform::VarFormBoundaryState::boundary_nodes, primary_ndof(), polyfem::io::OutStatsData::solver_info, polyfem::io::OutStatsData::spectrum, polyfem::varform::VarForm::stats, and x.

Referenced by solve_static_linear(), and solve_transient_linear().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ solve_problem()

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

overrideprivatevirtual

Implements polyfem::varform::VarForm.

Definition at line 451 of file IncompressibleElasticVarForm.cpp.

References polyfem::logger(), prepare_initial_solution(), polyfem::varform::ElasticVarForm::primary_assembler_, polyfem::varform::VarForm::problem, solve_static_linear(), solve_transient_linear(), polyfem::io::OutRuntimeData::solving_time, polyfem::io::OutStatsData::spectrum, polyfem::varform::VarForm::stats, time_integrator, and polyfem::varform::VarForm::timings.

Here is the call graph for this function:

◆ solve_static_linear()

void polyfem::varform::IncompressibleElasticVarForm::solve_static_linear ( Eigen::MatrixXd & sol )

private

Definition at line 378 of file IncompressibleElasticVarForm.cpp.

References polyfem::varform::VarForm::args, polyfem::varform::ElasticVarForm::boundary_, polyfem::varform::VarFormBoundaryState::boundary_nodes, build_stiffness_mat(), polyfem::varform::ElasticVarForm::elastic_boundary_samples(), polyfem::varform::VarFormBoundaryState::local_boundary, polyfem::varform::VarFormBoundaryState::local_neumann_boundary, polyfem::logger(), polyfem::varform::ElasticVarForm::rhs_, polyfem::varform::ElasticVarForm::rhs_assembler_, and solve_linear_system().

Referenced by solve_problem().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ solve_transient_linear()

void polyfem::varform::IncompressibleElasticVarForm::solve_transient_linear ( Eigen::MatrixXd & sol )

private

Definition at line 389 of file IncompressibleElasticVarForm.cpp.

Referenced by solve_problem().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ split_solution()

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