OutData.hpp

Go to the documentation of this file.

#pragma once
 
#include <polyfem/Common.hpp>
 
#include <polyfem/assembler/Problem.hpp>
 
#include <polyfem/basis/ElementBases.hpp>
 
#include <polyfem/mesh/Mesh.hpp>
 
#include <polyfem/solver/SolveData.hpp>
#include <polyfem/io/OutStatsData.hpp>
 
#include <paraviewo/ParaviewWriter.hpp>
#include <paraviewo/VTUWriter.hpp>
#include <paraviewo/HDF5VTUWriter.hpp>
 
#include <polyfem/utils/RefElementSampler.hpp>
 
#include <Eigen/Dense>
 
#include <ipc/potentials/normal_adhesion_potential.hpp>
#include <ipc/potentials/tangential_adhesion_potential.hpp>
 
namespace polyfem::legacy
{
    class State;
} // namespace polyfem::legacy
 
namespace polyfem::legacy::io
{
    using polyfem::io::OutRuntimeData;
    using polyfem::io::OutStatsData;
 
    class OutGeometryData
    {
    public:
        struct ExportOptions
        {
            std::vector<std::string> fields; // fields to export, empty means all
 
            bool volume;
            bool surface;
            bool wire;
            bool points;
            bool contact_forces;
            bool friction_forces;
            bool normal_adhesion_forces;
            bool tangential_adhesion_forces;
            bool forces;
            bool jacobian_validity;
 
            bool use_sampler;
            bool boundary_only;
            bool material_params;
            bool body_ids;
            bool sol_on_grid;
            bool velocity;
            bool acceleration;
            bool tensor_values;
            bool scalar_values;
            bool discretization_order;
            bool nodes;
 
            bool use_spline;
            bool reorder_output;
 
            bool use_hdf5;
 
            ExportOptions(const json &args,
                          const bool is_mesh_linear,
                          const bool mesh_has_prisms,
                          const bool is_problem_scalar);
 
            inline std::string file_extension() const
            {
                return use_hdf5 ? ".hdf" : ".vtu";
            }
 
            bool export_field(const std::string &field) const;
        };
 
        static void extract_boundary_mesh(
            const mesh::Mesh &mesh,
            const int n_bases,
            const std::vector<basis::ElementBases> &bases,
            const std::vector<mesh::LocalBoundary> &total_local_boundary,
            Eigen::MatrixXd &node_positions,
            Eigen::MatrixXi &boundary_edges,
            Eigen::MatrixXi &boundary_triangles,
            std::vector<Eigen::Triplet<double>> &displacement_map_entries);
 
        void init_sampler(const polyfem::mesh::Mesh &mesh, const double vismesh_rel_area);
 
        void build_grid(const polyfem::mesh::Mesh &mesh, const double spacing);
 
        void export_data(
            const State &state,
            const Eigen::MatrixXd &sol,
            const Eigen::MatrixXd &pressure,
            const bool is_time_dependent,
            const double tend_in,
            const double dt,
            const ExportOptions &opts,
            const std::string &vis_mesh_path,
            const std::string &nodes_path,
            const std::string &solution_path,
            const std::string &stress_path,
            const std::string &mises_path,
            const bool is_contact_enabled) const;
 
        void save_vtu(const std::string &path,
                      const State &state,
                      const Eigen::MatrixXd &sol,
                      const Eigen::MatrixXd &pressure,
                      const double t,
                      const double dt,
                      const ExportOptions &opts,
                      const bool is_contact_enabled) const;
 
        void save_volume(const std::string &path,
                         const State &state,
                         const Eigen::MatrixXd &sol,
                         const Eigen::MatrixXd &pressure,
                         const double t,
                         const double dt,
                         const ExportOptions &opts) const;
 
        void save_surface(const std::string &export_surface,
                          const State &state,
                          const Eigen::MatrixXd &sol,
                          const Eigen::MatrixXd &pressure,
                          const double t,
                          const double dt_in,
                          const ExportOptions &opts,
                          const bool is_contact_enabled) const;
 
        void save_contact_surface(
            const std::string &export_surface,
            const State &state,
            const Eigen::MatrixXd &sol,
            const Eigen::MatrixXd &pressure,
            const double t,
            const double dt_in,
            const ExportOptions &opts,
            const bool is_contact_enabled) const;
 
        void save_wire(const std::string &name,
                       const State &state,
                       const Eigen::MatrixXd &sol,
                       const double t,
                       const ExportOptions &opts) const;
 
        void save_points(
            const std::string &path,
            const State &state,
            const Eigen::MatrixXd &sol,
            const ExportOptions &opts) const;
 
        void save_pvd(const std::string &name, const std::function<std::string(int)> &vtu_names,
                      int time_steps, double t0, double dt, int skip_frame = 1) const;
 
    private:
        utils::RefElementSampler ref_element_sampler;
 
        Eigen::MatrixXd grid_points;
        Eigen::MatrixXi grid_points_to_elements;
        Eigen::MatrixXd grid_points_bc;
 
        void build_vis_boundary_mesh(
            const mesh::Mesh &mesh,
            const std::vector<basis::ElementBases> &bases,
            const std::vector<basis::ElementBases> &gbases,
            const std::vector<mesh::LocalBoundary> &total_local_boundary,
            const Eigen::MatrixXd &solution,
            const int problem_dim,
            Eigen::MatrixXd &boundary_vis_vertices,
            Eigen::MatrixXd &boundary_vis_local_vertices,
            Eigen::MatrixXi &boundary_vis_elements,
            Eigen::MatrixXi &boundary_vis_elements_ids,
            Eigen::MatrixXi &boundary_vis_primitive_ids,
            Eigen::MatrixXd &boundary_vis_normals,
            Eigen::MatrixXd &displaced_boundary_vis_normals) const;
 
        void build_vis_mesh(
            const mesh::Mesh &mesh,
            const Eigen::VectorXi &disc_orders,
            const std::vector<basis::ElementBases> &gbases,
            const std::map<int, Eigen::MatrixXd> &polys,
            const std::map<int, std::pair<Eigen::MatrixXd, Eigen::MatrixXi>> &polys_3d,
            const bool boundary_only,
            Eigen::MatrixXd &points,
            Eigen::MatrixXi &tets,
            Eigen::MatrixXi &el_id,
            Eigen::MatrixXd &discr) const;
 
        void build_high_order_vis_mesh(
            const mesh::Mesh &mesh,
            const Eigen::VectorXi &disc_orders,
            const Eigen::VectorXi &disc_ordersq,
            const std::vector<basis::ElementBases> &bases,
            Eigen::MatrixXd &points,
            std::vector<paraviewo::CellElement> &elements,
            Eigen::MatrixXi &el_id,
            Eigen::MatrixXd &discr) const;
 
        void save_volume_vector_field(
            const State &state,
            const Eigen::MatrixXd &points,
            const ExportOptions &opts,
            const std::string &name,
            const Eigen::VectorXd &field,
            paraviewo::ParaviewWriter &writer) const;
    };
 
    class EnergyCSVWriter
    {
    public:
        EnergyCSVWriter(const std::string &path, const solver::SolveData &solve_data);
        ~EnergyCSVWriter();
 
        void write(const int i, const Eigen::MatrixXd &sol);
 
    protected:
        std::ofstream file;
        const solver::SolveData &solve_data;
    };
 
    class RuntimeStatsCSVWriter
    {
    public:
        RuntimeStatsCSVWriter(const std::string &path, const State &state, const double t0, const double dt);
        ~RuntimeStatsCSVWriter();
 
        void write(const int t, const double forward, const double remeshing, const double global_relaxation, const Eigen::MatrixXd &sol);
 
    protected:
        std::ofstream file;
        const State &state;
        const double t0;
        const double dt;
        double total_forward_solve_time = 0;
        double total_remeshing_time = 0;
        double total_global_relaxation_time = 0;
    };
} // namespace polyfem::legacy::io