BuildFromJson.cpp

Go to the documentation of this file.

#include <polyfem/optimization/BuildFromJson.hpp>
 
#include <polyfem/varforms/diff/DifferentiableVarForm.hpp>
#include <polyfem/State.hpp>
#include <polyfem/Common.hpp>
 
#include <polyfem/mesh/MeshLoader.hpp>
 
#include <polyfem/utils/Logger.hpp>
#include <polyfem/utils/StringUtils.hpp>
#include <polyfem/utils/JSONUtils.hpp>
 
#include <polyfem/optimization/NodeSelectionUtils.hpp>
 
#include <polyfem/optimization/forms/AMIPSForm.hpp>
#include <polyfem/optimization/forms/AdjointForm.hpp>
#include <polyfem/optimization/forms/BarrierForms.hpp>
#include <polyfem/optimization/forms/CompositeForms.hpp>
#include <polyfem/optimization/forms/ParametrizedProductForm.hpp>
#include <polyfem/optimization/forms/SmoothingForms.hpp>
#include <polyfem/optimization/forms/SpatialIntegralForms.hpp>
#include <polyfem/optimization/forms/SumCompositeForm.hpp>
#include <polyfem/optimization/forms/SurfaceTractionForms.hpp>
#include <polyfem/optimization/forms/TargetForms.hpp>
#include <polyfem/optimization/forms/TransientForm.hpp>
 
#include <polyfem/optimization/var2sims/VariableToSimulation.hpp>
#include <polyfem/optimization/var2sims/VariableToSimulationGroup.hpp>
#include <polyfem/optimization/var2sims/ShapeVariableToSimulation.hpp>
#include <polyfem/optimization/var2sims/ElasticVariableToSimulation.hpp>
#include <polyfem/optimization/var2sims/FrictionVariableToSimulation.hpp>
#include <polyfem/optimization/var2sims/DampingVariableToSimulation.hpp>
#include <polyfem/optimization/var2sims/InitialConditionVariableToSimulation.hpp>
#include <polyfem/optimization/var2sims/DirichletBoundaryVariableToSimulation.hpp>
#include <polyfem/optimization/var2sims/DirichletNodesVariableToSimulation.hpp>
#include <polyfem/optimization/var2sims/PressureBoundaryVariableToSimulation.hpp>
#include <polyfem/optimization/var2sims/PeriodicShapeVariableToSimulation.hpp>
#ifdef POLYFEM_WITH_INFLATOR
#include <polyfem/optimization/var2sims/InflatedPeriodicShapeVariableToSimulation.hpp>
#endif
 
#include <polyfem/optimization/parametrization/Parametrization.hpp>
#include <polyfem/optimization/parametrization/Parametrizations.hpp>
#include <polyfem/optimization/parametrization/SplineParametrizations.hpp>
 
#include <Eigen/Core>
#include <spdlog/fmt/fmt.h>
 
#include <string>
#include <memory>
#include <cassert>
#include <cstddef>
#include <fstream>
#include <set>
#include <utility>
#include <vector>
 
namespace polyfem::from_json
{
    namespace
    {
        Eigen::VectorXi eigen_vector_xi_from_json(const json &j)
        {
            auto tmp = j.get<std::vector<int>>();
            Eigen::VectorXi out = Eigen::Map<Eigen::VectorXi>(tmp.data(), tmp.size());
            return out;
        }
 
        Eigen::VectorXi eigen_vector_xi_from_file(const std::string &path, const io::ResourceIO &resources)
        {
            const Eigen::MatrixXi mat = resources.read_int_matrix(path);
            return mat.reshaped();
        }
 
        Eigen::VectorXi parse_active_geometry_nodes(const json &j, const varform::DifferentiableVarForm &varform, const io::ResourceIO &resources)
        {
            if (j.is_array())
            {
                return eigen_vector_xi_from_json(j);
            }
            if (j.is_string())
            {
                return eigen_vector_xi_from_file(j.get<std::string>(), resources);
            }
 
            // Advanced selection.
            std::string type = j["type"].get<std::string>();
            auto selection = j["selection"].get<std::vector<int>>();
            if (type == "interior")
            {
                return select_interior_nodes(varform, selection);
            }
            else if (type == "boundary")
            {
                return select_boundary_nodes(varform, selection);
            }
            else if (type == "boundary_excluding_surface")
            {
                return select_boundary_nodes_excluding_surfaces(varform, selection);
            }
            else
            {
                log_and_throw_adjoint_error("Unknown advanced active geometry selection type name {}.", type);
            }
        }
 
    } // namespace
 
    std::shared_ptr<varform::DifferentiableVarForm> build_differentiable_varform(
        const json &args,
        const io::ResourceIO &resources,
        const size_t max_threads)
    {
        json in_args = args;
        in_args["solver"]["max_threads"] = max_threads;
 
        auto state = std::make_shared<State>();
        state->init(in_args, resources, true, true);
        state->load_mesh();
        auto differentiable_varform =
            std::dynamic_pointer_cast<varform::DifferentiableVarForm>(state->variational_formulation);
        if (!differentiable_varform)
        {
            log_and_throw_adjoint_error(
                "Variational formulation {} does not support differentiable/adjoint optimization.",
                state->variational_formulation->name());
        }
        differentiable_varform->prepare();
        return std::shared_ptr<varform::DifferentiableVarForm>(
            std::move(state), differentiable_varform.get());
    }
 
    std::shared_ptr<solver::Parametrization> build_parametrization(
        const json &args,
        const std::vector<std::shared_ptr<varform::DifferentiableVarForm>> &varforms,
        const std::vector<int> &variable_sizes)
    {
        using namespace polyfem::solver;
 
        std::shared_ptr<Parametrization> map;
        const std::string type = args["type"];
        if (type == "per-body-to-per-elem")
        {
            map = std::make_shared<PerBody2PerElem>(varforms[args["state"]]->get_mesh());
        }
        else if (type == "per-body-to-per-node")
        {
            const auto &varform = varforms[args["state"]];
            map = std::make_shared<PerBody2PerNode>(varform->get_mesh(),
                                                    varform->primary_space().basis_list(),
                                                    varform->primary_space().n_bases);
        }
        else if (type == "E-nu-to-lambda-mu")
        {
            map = std::make_shared<ENu2LambdaMu>(args["is_volume"]);
        }
        else if (type == "slice")
        {
            if (args["from"] != -1 || args["to"] != -1)
            {
                map = std::make_shared<SliceMap>(args["from"], args["to"], args["last"]);
            }
            else if (args["parameter_index"] != -1)
            {
                int idx = args["parameter_index"].get<int>();
                int from, to, last;
                int cumulative = 0;
                for (int i = 0; i < variable_sizes.size(); ++i)
                {
                    if (i == idx)
                    {
                        from = cumulative;
                        to = from + variable_sizes[i];
                    }
                    cumulative += variable_sizes[i];
                }
                last = cumulative;
                map = std::make_shared<SliceMap>(from, to, last);
            }
            else
            {
                log_and_throw_adjoint_error("Incorrect spec for SliceMap!");
            }
        }
        else if (type == "exp")
        {
            map = std::make_shared<ExponentialMap>(args["from"], args["to"]);
        }
        else if (type == "scale")
        {
            map = std::make_shared<Scaling>(args["value"]);
        }
        else if (type == "power")
        {
            map = std::make_shared<PowerMap>(args["power"]);
        }
        else if (type == "append-values")
        {
            Eigen::VectorXd vals = args["values"];
            map = std::make_shared<InsertConstantMap>(vals, args["start"]);
        }
        else if (type == "append-const")
        {
            map = std::make_shared<InsertConstantMap>(args["size"], args["value"], args["start"]);
        }
        else if (type == "linear-filter")
        {
            map = std::make_shared<LinearFilter>(varforms[args["state"]]->get_mesh(), args["radius"]);
        }
        else if (type == "bounded-biharmonic-weights")
        {
            map = std::make_shared<BoundedBiharmonicWeights2Dto3D>(
                args["num_control_vertices"], args["num_vertices"],
                *varforms[args["state"]], args["allow_rotations"]);
        }
        else if (type == "scalar-velocity-parametrization")
        {
            map = std::make_shared<ScalarVelocityParametrization>(args["start_val"], args["dt"]);
        }
        else
        {
            log_and_throw_adjoint_error("Unkown parametrization!");
        }
 
        return map;
    }
 
    std::shared_ptr<solver::VariableToSimulation> build_variable_to_simulation(
        const json &args,
        const std::vector<std::shared_ptr<varform::DifferentiableVarForm>> &varforms,
        const std::vector<std::shared_ptr<DiffCache>> &diff_caches,
        const std::vector<int> &variable_sizes,
        const std::vector<std::reference_wrapper<const io::ResourceIO>> &resources)
    {
        using namespace polyfem::solver;
 
        // Collect relevant varforms from the state-index JSON field.
        std::vector<std::shared_ptr<varform::DifferentiableVarForm>> relevant_varforms;
        std::vector<std::shared_ptr<DiffCache>> rel_diff_caches;
        int resource_id;
        if (args["state"].is_array())
        {
            resource_id = args["state"][0].get<int>();
            for (int i : args["state"])
            {
                relevant_varforms.push_back(varforms[i]);
                rel_diff_caches.push_back(diff_caches[i]);
            }
        }
        else
        {
            const int varform_id = args["state"];
            resource_id = varform_id;
            relevant_varforms.push_back(varforms[varform_id]);
            rel_diff_caches.push_back(diff_caches[varform_id]);
        }
 
        // Build all parametrizations.
        std::vector<std::shared_ptr<Parametrization>> map_list;
        for (const auto &arg : args["composition"])
        {
            map_list.push_back(build_parametrization(arg, varforms, variable_sizes));
        }
        CompositeParametrization compo{std::move(map_list)};
 
        // Build VariableToSimulation.
        std::shared_ptr<VariableToSimulation> var2sim;
        std::string var2sim_type = args["type"];
        if (var2sim_type == "shape")
        {
            Eigen::VectorXi active_dimensions = eigen_vector_xi_from_json(args["active_dimensions"]);
            Eigen::VectorXi active_nodes = parse_active_geometry_nodes(args["active_geometry_nodes"], *relevant_varforms[0], resources.at(resource_id).get());
 
            var2sim = std::make_shared<ShapeVariableToSimulation>(
                std::move(relevant_varforms),
                std::move(rel_diff_caches),
                std::move(compo),
                std::move(active_dimensions),
                std::move(active_nodes));
        }
        else if (var2sim_type == "elastic")
        {
            var2sim = std::make_shared<ElasticVariableToSimulation>(
                std::move(relevant_varforms),
                std::move(rel_diff_caches),
                std::move(compo));
        }
        else if (var2sim_type == "friction")
        {
            var2sim = std::make_shared<FrictionVariableToSimulation>(
                std::move(relevant_varforms),
                std::move(rel_diff_caches),
                std::move(compo));
        }
        else if (var2sim_type == "damping")
        {
            var2sim = std::make_shared<DampingVariableToSimulation>(
                std::move(relevant_varforms),
                std::move(rel_diff_caches),
                std::move(compo));
        }
        else if (var2sim_type == "initial")
        {
            Eigen::VectorXi active_dofs = eigen_vector_xi_from_json(args["active_dofs"]);
            var2sim = std::make_shared<InitialConditionVariableToSimulation>(
                std::move(relevant_varforms),
                std::move(rel_diff_caches),
                std::move(compo),
                std::move(active_dofs));
        }
        else if (var2sim_type == "dirichlet-boundary")
        {
            Eigen::VectorXi active_boundary_ids = eigen_vector_xi_from_json(args["active_boundary_ids"]);
            Eigen::VectorXi active_time_slices = eigen_vector_xi_from_json(args["active_time_slices"]);
            var2sim = std::make_shared<DirichletBoundaryVariableToSimulation>(
                std::move(relevant_varforms),
                std::move(rel_diff_caches),
                std::move(compo),
                std::move(active_boundary_ids),
                std::move(active_time_slices));
        }
        else if (var2sim_type == "dirichlet-nodes")
        {
            Eigen::VectorXi active_nodes = parse_active_geometry_nodes(args["active_geometry_nodes"], *relevant_varforms[0], resources.at(resource_id).get());
            var2sim = std::make_shared<DirichletNodesVariableToSimulation>(
                std::move(relevant_varforms),
                std::move(rel_diff_caches),
                std::move(compo),
                std::move(active_nodes));
        }
        else if (var2sim_type == "pressure")
        {
            Eigen::VectorXi active_boundary_ids = eigen_vector_xi_from_json(args["active_boundary_ids"]);
            Eigen::VectorXi active_time_slices = eigen_vector_xi_from_json(args["active_time_slices"]);
            var2sim = std::make_shared<PressureBoundaryVariableToSimulation>(
                std::move(relevant_varforms),
                std::move(rel_diff_caches),
                std::move(compo),
                std::move(active_boundary_ids),
                std::move(active_time_slices));
        }
        else if (var2sim_type == "inflated-periodic-shape")
        {
#ifdef POLYFEM_WITH_INFLATOR
            var2sim = std::make_shared<InflatedPeriodicShapeVariableToSimulation>(
                std::move(relevant_varforms),
                std::move(rel_diff_caches),
                std::move(compo), args["inflation"], resources.at(resource_id).get());
#else
            log_and_throw_adjoint_error("Inflated periodic shape requires POLYFEM_WITH_INFLATOR=ON.");
#endif
        }
        else if (var2sim_type == "periodic-shape")
        {
            var2sim = std::make_shared<PeriodicShapeVariableToSimulation>(
                std::move(relevant_varforms),
                std::move(rel_diff_caches),
                std::move(compo));
        }
        else
        {
            log_and_throw_adjoint_error("Unknown variable to simulation name {}.", var2sim_type);
        }
 
        return var2sim;
    }
 
    solver::VariableToSimulationGroup build_variable_to_simulation_group(
        const json &args,
        const std::vector<std::shared_ptr<varform::DifferentiableVarForm>> &varforms,
        const std::vector<std::shared_ptr<DiffCache>> &diff_caches,
        const std::vector<int> &variable_sizes,
        const std::vector<std::reference_wrapper<const io::ResourceIO>> &resources)
    {
        solver::VariableToSimulationGroup v2s_group;
        for (const auto &arg : args)
        {
            v2s_group.data.push_back(
                build_variable_to_simulation(arg, varforms, diff_caches, variable_sizes, resources));
        }
        return v2s_group;
    }
 
    std::shared_ptr<solver::AdjointForm> build_form(
        const json &args,
        const solver::VariableToSimulationGroup &var2sim,
        const std::vector<std::shared_ptr<varform::DifferentiableVarForm>> &varforms,
        const std::vector<std::shared_ptr<DiffCache>> &diff_caches,
        const std::vector<std::reference_wrapper<const io::ResourceIO>> &resources)
    {
        using namespace polyfem::solver;
 
        std::shared_ptr<AdjointForm> obj;
        if (args.is_array())
        {
            std::vector<std::shared_ptr<AdjointForm>> forms;
            for (const auto &arg : args)
            {
                forms.push_back(build_form(arg, var2sim, varforms, diff_caches, resources));
            }
 
            obj = std::make_shared<SumCompositeForm>(var2sim, forms);
        }
        else
        {
            const std::string type = args["type"];
            if (type == "transient_integral")
            {
                std::shared_ptr<StaticForm> static_obj =
                    std::dynamic_pointer_cast<StaticForm>(build_form(args["static_objective"], var2sim, varforms, diff_caches, resources));
                if (!static_obj)
                {
                    log_and_throw_adjoint_error("Transient integral objective must have a static objective!");
                }
                const auto &varform = varforms[args["state"]];
                obj = std::make_shared<TransientForm>(
                    var2sim, varform->get_args()["time"]["time_steps"], varform->get_args()["time"]["dt"],
                    args["integral_type"], args["steps"].get<std::vector<int>>(),
                    static_obj);
            }
            else if (type == "proxy_transient_integral")
            {
                std::shared_ptr<StaticForm> static_obj =
                    std::dynamic_pointer_cast<StaticForm>(build_form(args["static_objective"], var2sim, varforms, diff_caches, resources));
                if (!static_obj)
                {
                    log_and_throw_adjoint_error("Transient integral objective must have a static objective!");
                }
                if (args["steps"].size() == 0)
                {
                    log_and_throw_adjoint_error("ProxyTransientForm requires non-empty \"steps\"!");
                }
                const auto &varform = varforms[args["state"]];
                obj = std::make_shared<ProxyTransientForm>(
                    var2sim, varform->get_args()["time"]["time_steps"], varform->get_args()["time"]["dt"],
                    args["integral_type"], args["steps"].get<std::vector<int>>(),
                    static_obj);
            }
            else if (type == "power")
            {
                std::shared_ptr<AdjointForm> obj_aux =
                    build_form(args["objective"], var2sim, varforms, diff_caches, resources);
                obj = std::make_shared<PowerForm>(obj_aux, args["power"]);
            }
            else if (type == "divide")
            {
                std::shared_ptr<AdjointForm> obj1 =
                    build_form(args["objective"][0], var2sim, varforms, diff_caches, resources);
                std::shared_ptr<AdjointForm> obj2 =
                    build_form(args["objective"][1], var2sim, varforms, diff_caches, resources);
                std::vector<std::shared_ptr<AdjointForm>> objs({obj1, obj2});
                obj = std::make_shared<DivideForm>(objs);
            }
            else if (type == "plus-const")
            {
                obj = std::make_shared<PlusConstCompositeForm>(
                    build_form(args["objective"], var2sim, varforms, diff_caches, resources), args["value"]);
            }
            else if (type == "log")
            {
                obj = std::make_shared<LogCompositeForm>(
                    build_form(args["objective"], var2sim, varforms, diff_caches, resources));
            }
            else if (type == "compliance")
            {
                obj = std::make_shared<ComplianceForm>(var2sim, varforms[args["state"]], diff_caches[args["state"]],
                                                       args);
            }
            else if (type == "acceleration")
            {
                obj = std::make_shared<AccelerationForm>(var2sim,
                                                         varforms[args["state"]], diff_caches[args["state"]], args);
            }
            else if (type == "kinetic")
            {
                obj = std::make_shared<AccelerationForm>(var2sim,
                                                         varforms[args["state"]], diff_caches[args["state"]], args);
            }
            else if (type == "target")
            {
                std::shared_ptr<TargetForm> tmp =
                    std::make_shared<TargetForm>(var2sim, varforms[args["state"]], diff_caches[args["state"]], args, resources.at(args["state"].get<int>()).get());
                auto reference_cached =
                    args["reference_cached_body_ids"].get<std::vector<int>>();
                tmp->set_reference(
                    varforms[args["target_state"]],
                    diff_caches[args["target_state"]],
                    std::set(reference_cached.begin(), reference_cached.end()));
                obj = tmp;
            }
            else if (type == "displacement-target")
            {
                std::shared_ptr<TargetForm> tmp =
                    std::make_shared<TargetForm>(var2sim, varforms[args["state"]], diff_caches[args["state"]], args, resources.at(args["state"].get<int>()).get());
 
                Eigen::VectorXd target_displacement;
                target_displacement.setZero(varforms[args["state"]]->get_mesh().dimension());
                if (target_displacement.size() != args["target_displacement"].size())
                {
                    log_and_throw_error("Target displacement shape must match the dimension of the simulation");
                }
                for (int i = 0; i < target_displacement.size(); ++i)
                {
                    target_displacement(i) = args["target_displacement"][i].get<double>();
                }
                if (args["active_dimension"].size() > 0)
                {
                    if (target_displacement.size() != args["active_dimension"].size())
                    {
                        log_and_throw_error("Active dimension shape must match the dimension of the simulation");
                    }
                    std::vector<bool> active_dimension_mask(args["active_dimension"].size());
                    for (int i = 0; i < args["active_dimension"].size(); ++i)
                    {
                        active_dimension_mask[i] = args["active_dimension"][i].get<bool>();
                    }
                    tmp->set_active_dimension(active_dimension_mask);
                }
                tmp->set_reference(target_displacement);
                obj = tmp;
            }
            else if (type == "center-target")
            {
                obj = std::make_shared<BarycenterTargetForm>(
                    var2sim, args, varforms[args["state"]], diff_caches[args["state"]], varforms[args["target_state"]], diff_caches[args["target_state"]]);
            }
            else if (type == "sdf-target")
            {
                std::shared_ptr<SDFTargetForm> tmp = std::make_shared<SDFTargetForm>(
                    var2sim, varforms[args["state"]], diff_caches[args["state"]], args);
                double delta = args["delta"].get<double>();
                if (!varforms[args["state"]]->get_mesh().is_volume())
                {
                    int dim = 2;
                    Eigen::MatrixXd control_points(args["control_points"].size(), dim);
                    for (int i = 0; i < control_points.rows(); ++i)
                    {
                        for (int j = 0; j < control_points.cols(); ++j)
                        {
                            control_points(i, j) = args["control_points"][i][j].get<double>();
                        }
                    }
                    Eigen::VectorXd knots(args["knots"].size());
                    for (int i = 0; i < knots.size(); ++i)
                    {
                        knots(i) = args["knots"][i].get<double>();
                    }
                    tmp->set_bspline_target(control_points, knots, delta);
                }
                else
                {
                    int dim = 3;
                    Eigen::MatrixXd control_points_grid(args["control_points_grid"].size(), dim);
                    for (int i = 0; i < control_points_grid.rows(); ++i)
                    {
                        for (int j = 0; j < control_points_grid.cols(); ++j)
                        {
                            control_points_grid(i, j) = args["control_points_grid"][i][j].get<double>();
                        }
                    }
                    Eigen::VectorXd knots_u(args["knots_u"].size());
                    for (int i = 0; i < knots_u.size(); ++i)
                    {
                        knots_u(i) = args["knots_u"][i].get<double>();
                    }
                    Eigen::VectorXd knots_v(args["knots_v"].size());
                    for (int i = 0; i < knots_v.size(); ++i)
                    {
                        knots_v(i) = args["knots_v"][i].get<double>();
                    }
                    tmp->set_bspline_target(control_points_grid, knots_u, knots_v, delta);
                }
 
                obj = tmp;
            }
            else if (type == "mesh-target")
            {
                std::shared_ptr<MeshTargetForm> tmp = std::make_shared<MeshTargetForm>(
                    var2sim, varforms[args["state"]], diff_caches[args["state"]], args);
                double delta = args["delta"].get<double>();
 
                const auto surface = mesh::MeshLoader(resources.at(args["state"].get<int>()).get())
                                         .load_surface(args["mesh_path"].get<std::string>());
                tmp->set_surface_mesh_target(surface.vertices, surface.faces, delta);
                obj = tmp;
            }
            else if (type == "function-target")
            {
                std::shared_ptr<TargetForm> tmp =
                    std::make_shared<TargetForm>(var2sim, varforms[args["state"]], diff_caches[args["state"]], args, resources.at(args["state"].get<int>()).get());
                tmp->set_reference(args["target_function"], args["target_function_gradient"]);
                obj = tmp;
            }
            else if (type == "node-target")
            {
                obj = std::make_shared<NodeTargetForm>(varforms[args["state"]], diff_caches[args["state"]], var2sim, args, resources.at(args["state"].get<int>()).get());
            }
            else if (type == "min-dist-target")
            {
                obj = std::make_shared<MinTargetDistForm>(
                    var2sim, args["steps"], args["target"], args, varforms[args["state"]], diff_caches[args["state"]]);
            }
            else if (type == "position")
            {
                obj = std::make_shared<PositionForm>(var2sim, varforms[args["state"]], diff_caches[args["state"]], args);
            }
            else if (type == "homo_disp_grad")
            {
                obj = std::make_shared<HomogenizedDispGradForm>(var2sim, varforms[args["state"]], diff_caches[args["state"]], args);
            }
            else if (type == "stress")
            {
                obj = std::make_shared<StressForm>(var2sim, varforms[args["state"]], diff_caches[args["state"]], args);
            }
            else if (type == "stress_norm")
            {
                obj = std::make_shared<StressNormForm>(var2sim, varforms[args["state"]], diff_caches[args["state"]], args);
            }
            else if (type == "dirichlet_energy")
            {
                obj = std::make_shared<DirichletEnergyForm>(var2sim, varforms[args["state"]], diff_caches[args["state"]], args);
            }
            else if (type == "elastic_energy")
            {
                obj = std::make_shared<ElasticEnergyForm>(var2sim, varforms[args["state"]], diff_caches[args["state"]], args);
            }
            else if (type == "quadratic_contact_force_norm")
            {
                obj = std::make_shared<ProxyContactForceForm>(
                    var2sim, varforms[args["state"]], diff_caches[args["state"]], args["dhat"], true, args);
            }
            else if (type == "log_contact_force_norm")
            {
                obj = std::make_shared<ProxyContactForceForm>(
                    var2sim, varforms[args["state"]], diff_caches[args["state"]], args["dhat"], false, args);
            }
            else if (type == "max_stress")
            {
                obj = std::make_shared<MaxStressForm>(var2sim, varforms[args["state"]], diff_caches[args["state"]], args);
            }
            // "smooth_contact_force_norm" is the pre-GCP-rename spelling, kept
            // so existing objective JSONs keep working.
            else if (type == "gcp_force_norm" || type == "smooth_contact_force_norm")
            {
                // assert(varforms[args["state"]]->args["contact"]["use_gcp_formulation"]);
                obj = std::make_shared<GCPContactForceForm>(var2sim, varforms[args["state"]], diff_caches[args["state"]], args);
            }
            else if (type == "volume")
            {
                obj = std::make_shared<VolumeForm>(var2sim, varforms[args["state"]], diff_caches[args["state"]], args);
            }
            else if (type == "soft_constraint")
            {
                std::vector<std::shared_ptr<AdjointForm>> forms({build_form(args["objective"], var2sim, varforms, diff_caches, resources)});
                Eigen::VectorXd bounds = args["soft_bound"];
                obj = std::make_shared<InequalityConstraintForm>(forms, bounds, args["power"]);
            }
            else if (type == "min_jacobian")
            {
                obj = std::make_shared<MinJacobianForm>(var2sim, varforms[args["state"]]);
            }
            else if (type == "AMIPS")
            {
                obj = std::make_shared<AMIPSForm>(
                    var2sim, varforms[args["state"]], resources.at(args["state"].get<int>()).get());
            }
            else if (type == "boundary_smoothing")
            {
                if (args["surface_selection"].is_array())
                {
                    obj = std::make_shared<BoundarySmoothingForm>(
                        var2sim, varforms[args["state"]], args["scale_invariant"],
                        args["power"], args["surface_selection"].get<std::vector<int>>(),
                        args["dimensions"].get<std::vector<int>>());
                }
                else
                {
                    obj = std::make_shared<BoundarySmoothingForm>(
                        var2sim, varforms[args["state"]], args["scale_invariant"],
                        args["power"],
                        std::vector<int>{args["surface_selection"].get<int>()},
                        args["dimensions"].get<std::vector<int>>());
                }
            }
            else if (type == "elastic_material_smoothing")
            {
                obj = std::make_shared<ElasticMaterialSmoothingForm>(
                    var2sim, varforms[args["state"]],
                    args["volume_selection"].get<std::vector<int>>());
            }
            else if (type == "collision_barrier")
            {
                obj = std::make_shared<CollisionBarrierForm>(
                    var2sim, varforms[args["state"]], resources.at(args["state"].get<int>()).get(), args["dhat"]);
            }
            else if (type == "layer_thickness")
            {
                obj = std::make_shared<LayerThicknessForm>(
                    var2sim, varforms[args["state"]], resources.at(args["state"].get<int>()).get(),
                    args["boundary_ids"].get<std::vector<int>>(), args["dhat"]);
            }
            else if (type == "layer_thickness_log")
            {
                obj = std::make_shared<LayerThicknessForm>(
                    var2sim, varforms[args["state"]], resources.at(args["state"].get<int>()).get(),
                    args["boundary_ids"].get<std::vector<int>>(), args["dhat"], true,
                    args["dmin"]);
            }
            else if (type == "deformed_collision_barrier")
            {
                obj = std::make_shared<DeformedCollisionBarrierForm>(
                    var2sim, varforms[args["state"]], diff_caches[args["state"]], resources.at(args["state"].get<int>()).get(), args["dhat"]);
            }
            else if (type == "parametrized_product")
            {
                std::vector<std::shared_ptr<Parametrization>> map_list;
                for (const auto &arg : args["parametrization"])
                {
                    map_list.push_back(build_parametrization(arg, varforms, {}));
                }
                obj = std::make_shared<ParametrizedProductForm>(
                    CompositeParametrization(std::move(map_list)));
            }
            else
            {
                log_and_throw_adjoint_error("Objective not implemented!");
            }
 
            obj->set_weight(args["weight"]);
            if (args["print_energy"].get<std::string>() != "")
            {
                obj->enable_energy_print(args["print_energy"]);
            }
        }
 
        return obj;
    }
 
} // namespace polyfem::from_json