python binding material opt

Author: Huangzizhou

cmake/recipes/polyfem.cmake

@@ -17,4 +17,4 @@ message(STATUS "Third-party: creating target 'polyfem::polyfem'")
 # FetchContent_MakeAvailable(polyfem)
 
 include(CPM)
-CPMAddPackage("gh:polyfem/polyfem#71b67e6416c59f498589ddc1633c11e6c246b392")
+CPMAddPackage("gh:polyfem/polyfem#c18c17c538dd6abc5a7f1b4ed502b7e9bb3a7f51")

src/differentiable/adjoint.cpp

@@ -25,6 +25,19 @@ void define_adjoint(py::module_ &m)
       },
       py::arg("solver"));
 
+  m.def(
+      "elastic_material_derivative",
+      [](State &state) {
+        Eigen::VectorXd term;
+        if (state.problem->is_time_dependent())
+          AdjointTools::dJ_material_transient_adjoint_term(state, state.get_adjoint_mat(1), state.get_adjoint_mat(0), term);
+        else
+          AdjointTools::dJ_material_static_adjoint_term(state, state.diff_cached.u(0), state.get_adjoint_mat(0), term);
+
+        return utils::unflatten(term, state.bases.size());
+      },
+      py::arg("solver"));
+
   m.def(
       "initial_velocity_derivative",
       [](State &state) {

src/differentiable/objective.cpp

@@ -37,6 +37,4 @@ void define_objective(py::module_ &m)
 
     m.def("create_objective", &AdjointOptUtils::create_simple_form,
         py::arg("obj_type"), py::arg("param_type"), py::arg("solver"), py::arg("parameters"));
-
-    // obj = std::make_shared<StressNormForm>(var2sim, *(states[args["state"]]), args);
 }

src/state/state.cpp

@@ -600,7 +600,19 @@ void define_solver(py::module_ &m)
             }
           },
           "set initial displacement for one body", py::arg("body_id"),
-          py::arg("displacement"));
+          py::arg("displacement"))
+      .def(
+          "set_per_element_material",
+          [](State &self, const Eigen::VectorXd &lambda, const Eigen::VectorXd &mu) {
+            if (self.bases.size() == 0)
+              log_and_throw_adjoint_error("Build basis first!");
+
+            assert(lambda.size() == self.bases.size());
+            assert(mu.size() == self.bases.size());
+            self.assembler->update_lame_params(lambda, mu);
+          },
+          "set per-element Lame parameters", py::arg("lambda"),
+          py::arg("mu"));
 }
 
 void define_solve(py::module_ &m)