SolveData.hpp

Go to the documentation of this file.

#pragma once
 
#include <polyfem/assembler/AssemblyValsCache.hpp>
#include <polyfem/assembler/RhsAssembler.hpp>
#include <polyfem/assembler/Assembler.hpp>
#include <polyfem/basis/ElementBases.hpp>
#include <polyfem/mesh/Obstacle.hpp>
#include <polyfem/mesh/LocalBoundary.hpp>
#include <polyfem/utils/JSONUtils.hpp>
#include <polyfem/solver/forms/lagrangian/BCLagrangianForm.hpp>
 
#include <ipc/collision_mesh.hpp>
#include <polyfem/solver/forms/ContactForm.hpp>
 
#include <Eigen/Core>
 
#include <memory>
#include <string>
#include <unordered_map>
 
#include <polyfem/solver/forms/ElasticForm.hpp>
 
namespace polyfem::time_integrator
{
    class ImplicitTimeIntegrator;
} // namespace polyfem::time_integrator
 
namespace polyfem::assembler
{
    class ViscousDamping;
    class MacroStrainValue;
    class PressureAssembler;
} // namespace polyfem::assembler
 
namespace polyfem::solver
{
    class NLProblem;
    class Form;
    class ContactForm;
    class PeriodicContactForm;
    class MacroStrainALForm;
    class FrictionForm;
    class BodyForm;
    class AugmentedLagrangianForm;
    class MacroStrainLagrangianForm;
    class MacroStrainALForm;
    class InertiaForm;
    class ElasticForm;
    class PressureForm;
    class NormalAdhesionForm;
    class TangentialAdhesionForm;
 
    class SolveData
    {
    public:
        std::vector<std::shared_ptr<Form>> init_forms(
            // General
            const Units &units,
            const int dim,
            const double t,
            const Eigen::VectorXi &in_node_to_node,
 
            // Elastic form
            const int n_bases,
            std::vector<basis::ElementBases> &bases,
            const std::vector<basis::ElementBases> &geom_bases,
            const assembler::Assembler &assembler,
            assembler::AssemblyValsCache &ass_vals_cache,
            const assembler::AssemblyValsCache &mass_ass_vals_cache,
            const double jacobian_threshold,
            const solver::ElementInversionCheck check_inversion,
            const unsigned conservative_max_iter,
 
            // Body form
            const int n_pressure_bases,
            const std::vector<int> &boundary_nodes,
            const std::vector<mesh::LocalBoundary> &local_boundary,
            const std::vector<mesh::LocalBoundary> &local_neumann_boundary,
            const QuadratureOrders &n_boundary_samples,
            const Eigen::MatrixXd &rhs,
            const Eigen::MatrixXd &sol,
            const assembler::Density &density,
 
            // Pressure form
            const std::vector<mesh::LocalBoundary> &local_pressure_boundary,
            const std::unordered_map<int, std::vector<mesh::LocalBoundary>> &local_pressure_cavity,
            const std::shared_ptr<assembler::PressureAssembler> pressure_assembler,
 
            // Inertia form
            const bool ignore_inertia,
            const StiffnessMatrix &mass,
            const std::shared_ptr<assembler::ViscousDamping> damping_assembler,
 
            // Lagged regularization form
            const double lagged_regularization_weight,
            const int lagged_regularization_iterations,
 
            // Constraint forms
            const size_t obstacle_ndof,
            const std::vector<std::string> &hard_constraint_files,
            const std::vector<json> &soft_constraint_files,
            const json &zero_mean,
 
            // Contact form
            const bool contact_enabled,
            const ipc::CollisionMesh &collision_mesh,
            const double dhat,
            const double avg_mass,
            const bool use_area_weighting,
            const bool use_improved_max_operator,
            const bool use_physical_barrier,
            const json &barrier_stiffness,
            const double initial_barrier_stiffness,
            const ipc::BroadPhaseMethod broad_phase,
            const double ccd_tolerance,
            const long ccd_max_iterations,
            const bool enable_shape_derivatives,
 
            // Smooth Contact Form
            const bool use_gcp_formulation,
            const double alpha_t,
            const double alpha_n,
            const bool use_adaptive_dhat,
            const double min_distance_ratio,
 
            // Normal Adhesion Form
            const bool adhesion_enabled,
            const double dhat_p,
            const double dhat_a,
            const double Y,
 
            // Tangential Adhesion Form
            const double tangential_adhesion_coefficient,
            const double epsa,
            const int tangential_adhesion_iterations,
 
            // Homogenization
            const assembler::MacroStrainValue &macro_strain_constraint,
 
            // Periodic contact
            const bool periodic_contact,
            const Eigen::VectorXi &tiled_to_single,
 
            // Friction form
            const double friction_coefficient,
            const double epsv,
            const int friction_iterations,
 
            // Rayleigh damping form
            const json &rayleigh_damping,
 
            // BC augmented-Lagrangian mass-metric lumping
            const BCLumpingMode al_lumping = BCLumpingMode::ROW_SUM,
 
            // Boundary-ID periodic constraints
            const mesh::Mesh *periodic_mesh = nullptr,
            const std::vector<mesh::LocalBoundary> *periodic_local_boundary = nullptr,
            const json &periodic_conditions = json::array(),
            const int fe_space_id = -1);
 
        void update_barrier_stiffness(const Eigen::VectorXd &x);
 
        void update_dt();
 
        std::vector<std::pair<std::string, std::shared_ptr<solver::Form>>> named_forms() const;
 
    public:
        std::shared_ptr<assembler::RhsAssembler> rhs_assembler;
        std::shared_ptr<assembler::PressureAssembler> pressure_assembler;
        std::shared_ptr<solver::NLProblem> nl_problem;
 
        std::vector<std::shared_ptr<solver::AugmentedLagrangianForm>> al_form;
        std::shared_ptr<solver::MacroStrainLagrangianForm> strain_al_lagr_form;
        std::shared_ptr<solver::BodyForm> body_form;
        std::shared_ptr<solver::ContactForm> contact_form;
        std::shared_ptr<solver::ElasticForm> damping_form;
        std::shared_ptr<solver::ElasticForm> elastic_form;
        std::shared_ptr<solver::FrictionForm> friction_form;
        std::shared_ptr<solver::InertiaForm> inertia_form;
        std::shared_ptr<solver::PressureForm> pressure_form;
        std::shared_ptr<solver::NormalAdhesionForm> normal_adhesion_form;
        std::shared_ptr<solver::TangentialAdhesionForm> tangential_adhesion_form;
 
        std::shared_ptr<solver::PeriodicContactForm> periodic_contact_form;
 
        std::shared_ptr<time_integrator::ImplicitTimeIntegrator> time_integrator;
    };
} // namespace polyfem::solver