|
PolyFEM
|
#include <ScalarVarForm.hpp>
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, 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 | build_rhs_assembler () override |
| void | prepare_initial_solution (Eigen::MatrixXd &solution) const |
| void | build_stiffness_mat (StiffnessMatrix &stiffness) |
| void | solve_problem (Eigen::MatrixXd &sol, const InitialConditionOverride *initial_condition_override, const ForwardStepCallback &post_step) override |
| 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_linear_system_with_constraints (const std::unique_ptr< polysolve::linear::Solver > &solver, StiffnessMatrix &A, Eigen::VectorXd &b, const bool compute_spectrum, const QuadratureOrders &boundary_samples, const double time, Eigen::MatrixXd &sol) |
| void | solve_static (Eigen::MatrixXd &sol, const ForwardStepCallback &post_step) |
| void | solve_transient (Eigen::MatrixXd &sol, const ForwardStepCallback &post_step) |
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 |
| 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 &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) |
| 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 |
| 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 |
| 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 |
Friends | |
| class | polyfem::test::VarFormTestAccess |
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) |
Definition at line 17 of file ScalarVarForm.hpp.
|
overrideprotectedvirtual |
Implements polyfem::varform::VarForm.
Definition at line 284 of file ScalarVarForm.cpp.
References polyfem::varform::VarForm::args, polyfem::io::OutRuntimeData::assembling_mass_mat_time, avg_mass_, polyfem::varform::FESpace::basis_list(), polyfem::varform::FESpace::geometry_basis_list(), polyfem::mesh::Mesh::is_volume(), polyfem::logger(), polyfem::utils::lump_matrix(), mass_, mass_ass_vals_cache_, 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, space_, polyfem::varform::VarForm::stats, and polyfem::varform::VarForm::timings.
|
overrideprotectedvirtual |
Implements polyfem::varform::VarForm.
Definition at line 253 of file ScalarVarForm.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_, primary_assembler_, polyfem::varform::VarForm::problem, rhs_, rhs_assembler_, polyfem::varform::VarForm::root_path, and polyfem::varform::VarForm::timings.
|
overrideprotectedvirtual |
Implements polyfem::varform::VarForm.
Definition at line 116 of file ScalarVarForm.cpp.
References polyfem::varform::VarForm::args, ass_vals_cache_, polyfem::varform::VarForm::assign_discr_orders(), polyfem::varform::FESpace::basis_list(), boundary_, polyfem::varform::VarFormBoundaryState::boundary_nodes, polyfem::varform::VarForm::build_fe_space(), polyfem::varform::VarFormBoundaryState::clear_boundary_conditions(), polyfem::io::OutStatsData::compute_mesh_size(), polyfem::io::OutStatsData::count_flipped_elements(), polyfem::assembler::Dirichlet, polyfem::varform::VarFormBoundaryState::dirichlet_nodes, polyfem::varform::VarFormBoundaryState::dirichlet_nodes_position, polyfem::varform::FESpace::geometry_basis_list(), polyfem::mesh::Mesh::get_node_id(), polyfem::assembler::AssemblyValsCache::init(), polyfem::assembler::AssemblyValsCache::init_empty(), polyfem::mesh::Mesh::is_volume(), polyfem::varform::VarFormBoundaryState::local_boundary, polyfem::varform::VarFormBoundaryState::local_neumann_boundary, polyfem::logger(), mass_ass_vals_cache_, mass_assembler_, polyfem::io::OutStatsData::mesh_size, polyfem::varform::FESpace::n_bases, polyfem::io::OutStatsData::n_flipped, polyfem::assembler::Neumann, polyfem::varform::VarFormBoundaryState::neumann_nodes, polyfem::varform::VarFormBoundaryState::neumann_nodes_position, polyfem::varform::VarFormBoundaryState::normalize_boundary_nodes(), polyfem::refinement::APriori::p_refine(), primary_assembler_, polyfem::varform::VarForm::problem, pure_mass_ass_vals_cache_, pure_mass_assembler_, polyfem::varform::VarForm::rebuild_node_positions(), space_, polyfem::varform::FESpace::space_in_node_to_node, polyfem::varform::VarForm::stats, polyfem::varform::VarFormBoundaryState::total_local_boundary, and polyfem::mesh::Mesh::update_nodes().
|
overrideprotectedvirtual |
Implements polyfem::varform::VarForm.
Definition at line 235 of file ScalarVarForm.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, polyfem::solver::SolveData::rhs_assembler, rhs_assembler_, solve_data_, and space_.
Referenced by assemble_rhs().
|
protected |
Definition at line 673 of file ScalarVarForm.cpp.
References polyfem::logger(), space_, and polyfem::varform::internal::write_matrix_market().
|
overridevirtual |
Get the error statistics of the variational formulation, for output purposes.
Implements polyfem::varform::VarForm.
Definition at line 404 of file ScalarVarForm.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, space_, and polyfem::varform::VarForm::stats.
|
overridevirtual |
Implements polyfem::varform::VarForm.
Definition at line 417 of file ScalarVarForm.cpp.
References polyfem::varform::VarForm::args, polyfem::varform::FESpace::basis_list(), polyfem::io::Evaluator::compute_stress_at_quadrature_points(), polyfem::varform::FESpace::disc_orders, polyfem::varform::FESpace::disc_ordersq, dt, polyfem::varform::VarForm::ensure_output_sampler(), polyfem::io::OutGeometryData::export_data(), polyfem::varform::VarForm::export_options(), polyfem::varform::FESpace::geometry_basis_list(), polyfem::logger(), polyfem::io::OutputSpace::mesh, polyfem::varform::VarForm::mesh_, polyfem::varform::FESpace::n_bases, polyfem::varform::VarForm::output_field_function(), polyfem::varform::VarForm::output_geometry_, output_space(), primary_assembler_, polyfem::varform::VarForm::problem, polyfem::varform::VarForm::resolve_output_path(), space_, polyfem::varform::FESpace::space_in_node_to_node, polyfem::utils::unflatten(), and polyfem::io::write_matrix().
|
overridevirtual |
Initialize the variational formulation with the given 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 66 of file ScalarVarForm.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(), mass_assembler_, primary_assembler_, polyfem::varform::VarForm::problem, pure_mass_assembler_, polyfem::varform::VarForm::root_path, t0, time_steps, and polyfem::varform::VarForm::units.
|
overrideprotectedvirtual |
Implements polyfem::varform::VarForm.
Definition at line 107 of file ScalarVarForm.cpp.
References mass_assembler_, primary_assembler_, polyfem::varform::VarForm::problem, pure_mass_assembler_, and polyfem::varform::VarForm::set_materials().
|
inlineoverridevirtual |
Get the name of the variational formulation.
Implements polyfem::varform::VarForm.
Definition at line 22 of file ScalarVarForm.hpp.
Referenced by polyfem::varform::DifferentiableScalarVarForm::name(), and save_json().
|
overridevirtual |
Get the output fields of the variational formulation, for output purposes.
| sample | Output sample |
| solution | Solution matrix |
| options | Output field options |
Implements polyfem::varform::VarForm.
Definition at line 500 of file ScalarVarForm.cpp.
References polyfem::varform::VarForm::args, polyfem::varform::FESpace::basis_list(), polyfem::utils::count_invalid(), 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::varform::FESpace::n_bases, polyfem::io::OutputSample::node_ids, polyfem::io::OutputField::Point, polyfem::io::OutputSample::points, polyfem::io::OutputSample::primitive_ids, polyfem::varform::VarForm::problem, space_, and polyfem::io::OutputSample::time.
|
overridevirtual |
Get the output space of the variational formulation, for output purposes.
Implements polyfem::varform::VarForm.
Definition at line 379 of file ScalarVarForm.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(), and polyfem::varform::DifferentiableScalarVarForm::save_vtu().
|
protected |
Definition at line 329 of file ScalarVarForm.cpp.
References polyfem::varform::VarForm::args, polyfem::varform::VarForm::problem, polyfem::varform::VarForm::read_initial_x_from_file(), polyfem::varform::VarForm::resolve_input_path(), rhs_, rhs_assembler_, space_, and polyfem::varform::FESpace::space_in_node_to_node.
|
overrideprotectedvirtual |
Implements polyfem::varform::VarForm.
Definition at line 43 of file ScalarVarForm.cpp.
References ass_vals_cache_, avg_mass_, boundary_, dt, polyfem::assembler::AssemblyValsCache::init_empty(), mass_, mass_ass_vals_cache_, mass_assembler_, primary_assembler_, pure_mass_, pure_mass_ass_vals_cache_, pure_mass_assembler_, polyfem::varform::FESpace::reset(), polyfem::varform::VarFormBoundaryState::reset(), polyfem::varform::VarForm::reset(), rhs_, rhs_assembler_, solve_data_, space_, t0, time_integrator, and time_steps.
|
overridevirtual |
Save the solution to a JSON file, for output purposes.
| solution | Solution matrix to save |
| out | Output stream to save the solution |
Implements polyfem::varform::VarForm.
Definition at line 350 of file ScalarVarForm.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(), primary_assembler_, polyfem::varform::VarForm::problem, polyfem::io::OutStatsData::save_json(), space_, polyfem::varform::VarForm::stats, and polyfem::varform::VarForm::timings.
|
protected |
Definition at line 695 of file ScalarVarForm.cpp.
References polyfem::logger(), space_, and x.
|
protected |
Definition at line 729 of file ScalarVarForm.cpp.
References entries, polyfem::solver::NLProblem::full_hessian_to_reduced_hessian(), polyfem::solver::NLProblem::full_to_reduced_grad(), polyfem::log_and_throw_error(), polyfem::logger(), polyfem::solver::NLProblem::reduced_size(), polyfem::solver::NLProblem::reduced_to_full(), space_, and weights.
|
overrideprotectedvirtual |
Implements polyfem::varform::VarForm.
Definition at line 937 of file ScalarVarForm.cpp.
References polyfem::varform::InitialConditionOverride::acceleration, polyfem::logger(), POLYFEM_SCOPED_TIMER, polyfem::varform::InitialConditionOverride::solution, space_, and polyfem::varform::InitialConditionOverride::velocity.
|
protected |
Definition at line 842 of file ScalarVarForm.cpp.
References polyfem::logger(), and space_.
|
protected |
Definition at line 880 of file ScalarVarForm.cpp.
References polyfem::time_integrator::ImplicitTimeIntegrator::construct_bdf_integrator(), polyfem::logger(), and space_.
|
friend |
Definition at line 19 of file ScalarVarForm.hpp.
|
protected |
Definition at line 47 of file ScalarVarForm.hpp.
Referenced by polyfem::varform::DifferentiableScalarVarForm::assembly_cache(), build_basis(), and reset().
|
protected |
Definition at line 53 of file ScalarVarForm.hpp.
Referenced by assemble_mass_mat(), and reset().
|
protected |
Definition at line 46 of file ScalarVarForm.hpp.
Referenced by polyfem::varform::DifferentiableScalarVarForm::boundary_state(), build_basis(), build_rhs_assembler(), output_space(), and reset().
|
protected |
Definition at line 63 of file ScalarVarForm.hpp.
Referenced by export_data(), init(), reset(), and polyfem::varform::DifferentiableScalarVarForm::save_vtu().
|
protected |
Definition at line 51 of file ScalarVarForm.hpp.
Referenced by assemble_mass_mat(), polyfem::varform::DifferentiableScalarVarForm::mass_matrix(), and reset().
|
protected |
Definition at line 48 of file ScalarVarForm.hpp.
Referenced by assemble_mass_mat(), build_basis(), build_rhs_assembler(), polyfem::varform::DifferentiableScalarVarForm::mass_assembly_cache(), and reset().
|
protected |
Definition at line 57 of file ScalarVarForm.hpp.
Referenced by assemble_mass_mat(), assemble_rhs(), build_basis(), init(), load_mesh(), polyfem::varform::DifferentiableScalarVarForm::mass_assembler(), and reset().
|
protected |
Definition at line 56 of file ScalarVarForm.hpp.
Referenced by assemble_rhs(), build_basis(), build_rhs_assembler(), export_data(), init(), load_mesh(), polyfem::varform::DifferentiableScalarVarForm::primary_assembler(), reset(), and save_json().
|
protected |
Definition at line 52 of file ScalarVarForm.hpp.
Referenced by reset().
|
protected |
Definition at line 49 of file ScalarVarForm.hpp.
Referenced by build_basis(), and reset().
|
protected |
Definition at line 58 of file ScalarVarForm.hpp.
Referenced by build_basis(), init(), load_mesh(), and reset().
|
protected |
Definition at line 54 of file ScalarVarForm.hpp.
Referenced by assemble_rhs(), prepare_initial_solution(), and reset().
|
protected |
Definition at line 50 of file ScalarVarForm.hpp.
Referenced by assemble_rhs(), build_rhs_assembler(), polyfem::varform::DifferentiableScalarVarForm::invalidate_after_geometry_update(), polyfem::varform::DifferentiableScalarVarForm::invalidate_after_parameter_update(), prepare_initial_solution(), and reset().
|
protected |
Definition at line 59 of file ScalarVarForm.hpp.
Referenced by build_rhs_assembler(), polyfem::varform::DifferentiableScalarVarForm::invalidate_after_geometry_update(), polyfem::varform::DifferentiableScalarVarForm::invalidate_after_parameter_update(), reset(), polyfem::varform::DifferentiableScalarVarForm::solve_data(), and polyfem::varform::DifferentiableScalarVarForm::solve_data().
|
protected |
Definition at line 45 of file ScalarVarForm.hpp.
Referenced by assemble_mass_mat(), build_basis(), build_rhs_assembler(), compute_errors(), export_data(), output_fields(), output_space(), prepare_initial_solution(), polyfem::varform::DifferentiableScalarVarForm::primary_space(), reset(), and save_json().
|
protected |
Definition at line 61 of file ScalarVarForm.hpp.
|
protected |
Definition at line 90 of file ScalarVarForm.hpp.
Referenced by polyfem::varform::DifferentiableScalarVarForm::invalidate_after_geometry_update(), polyfem::varform::DifferentiableScalarVarForm::invalidate_after_parameter_update(), and reset().
|
protected |
Definition at line 62 of file ScalarVarForm.hpp.