Implemented hephaestus equation system functionality into thermo mech example

examples/CMakeLists.txt

@@ -3,4 +3,5 @@ add_subdirectory(team7)
 add_subdirectory(complex_team7)
 add_subdirectory(closed_coil)
 add_subdirectory(open_coil)
-add_subdirectory(magnetostatic)
+add_subdirectory(magnetostatic)
+add_subdirectory(thermal_expansion)

examples/thermal_expansion/CMakeLists.txt

@@ -0,0 +1,23 @@
+file(GLOB_RECURSE example_src "*.cpp")
+
+file(GLOB_RECURSE test_src_files "${PROJECT_SOURCE_DIR}/src/*.h" "${PROJECT_SOURCE_DIR}/src/*.hpp" "${PROJECT_SOURCE_DIR}/src/*.cpp")
+
+set (${PROJECT_NAME}_INCLUDE_DIRS "")
+foreach (_srcFile ${test_src_files})
+    get_filename_component(_dir ${_srcFile} PATH)
+    list (APPEND ${PROJECT_NAME}_INCLUDE_DIRS ${_dir})
+endforeach()
+list (REMOVE_DUPLICATES ${PROJECT_NAME}_INCLUDE_DIRS)
+
+
+add_executable(thermalexpansion ${example_src})
+add_compile_options(thermalexpansion ${BUILD_TYPE_COMPILER_FLAGS})
+target_include_directories(thermalexpansion PUBLIC ${MFEM_COMMON_INCLUDES} ${MFEM_INCLUDE_DIRS})
+target_include_directories(thermalexpansion PUBLIC ${${PROJECT_NAME}_INCLUDE_DIRS})
+target_include_directories(thermalexpansion PUBLIC ${PROJECT_SOURCE_DIR}/data/)
+
+set_property(TARGET thermalexpansion PROPERTY CXX_STANDARD 17)
+
+target_link_libraries(thermalexpansion ${GTEST_LIBRARY} ${GTEST_MAIN_LIBRARY} pthread)
+target_link_libraries(thermalexpansion ${CMAKE_LIBRARY_OUTPUT_DIRECTORY}/lib${PROJECT_NAME}.so)
+target_link_libraries(thermalexpansion ${MFEM_LIBRARIES} ${MFEM_COMMON_LIBRARY} -lrt)

examples/thermal_expansion/Main.cpp

@@ -0,0 +1,120 @@
+#include "hephaestus.hpp"
+
+const char *DATA_DIR = "../../data/";
+
+hephaestus::Coefficients defineCoefficients(){
+
+  hephaestus::Coefficients coefficients;
+
+  coefficients.scalars.Register("thermal_expansion_coef",
+                                new mfem::ConstantCoefficient(1e-04),
+                                true);
+
+  coefficients.scalars.Register("thermal_conductivity",
+                                new mfem::ConstantCoefficient(300),
+                                true);
+
+  coefficients.scalars.Register("lame_param",
+                                new mfem::ConstantCoefficient(10),
+                                true);
+
+  coefficients.scalars.Register("shear_modulus",
+                                new mfem::ConstantCoefficient(10),
+                                true);                        
+
+  coefficients.scalars.Register("stress_free_temp",
+                                new mfem::ConstantCoefficient(250),
+                                true);
+
+  coefficients.scalars.Register("zero",
+                                new mfem::ConstantCoefficient(0.0),
+                                true);
+
+
+  return coefficients;
+}
+
+hephaestus::Outputs defineOutputs() {
+  // std::map<std::string, mfem::DataCollection *> data_collections;
+  // data_collections["ParaViewDataCollection"] =
+  //     new mfem::ParaViewDataCollection("ThermalExpansionExample");
+  hephaestus::Outputs outputs;
+  outputs.Register("Paraview", new mfem::ParaViewDataCollection("ThermalExpansionExample"), true);
+  return outputs;
+}
+
+
+hephaestus::BCMap defineBoundaries() {
+
+  hephaestus::BCMap boundaries;
+  mfem::ConstantCoefficient *cold = new mfem::ConstantCoefficient(300.00);
+  mfem::ConstantCoefficient *hot = new mfem::ConstantCoefficient(500.00);
+  // mfem::VectorConstantCoefficient *zero = new mfem::VectorConstantCoefficient(mfem::Vector({0, 0, 0}));
+  mfem::Array<int> therm_bound_one_arr(1);
+  mfem::Array<int> therm_bound_two_arr(1);
+  mfem::Array<int> fixed_disp_arr(1);
+
+  therm_bound_one_arr = 1;
+  therm_bound_two_arr = 2;
+  fixed_disp_arr = 1;
+
+  boundaries.Register("thermal_boundary_one", new hephaestus::DirichletBC("temperature", therm_bound_one_arr, cold), true);
+  boundaries.Register("thermal_boundary_two", new hephaestus::DirichletBC("temperature", therm_bound_two_arr, hot), true);
+  boundaries.Register("fixed_displacement", new hephaestus::VectorDirichletBC("displacement", fixed_disp_arr,  new mfem::VectorConstantCoefficient(mfem::Vector({0, 0, 0})), nullptr, hephaestus::VectorDirichletBC::APPLY_TYPE::STANDARD), true);
+
+  return boundaries;
+}
+
+
+int main(int argc, char *argv[]) {
+  mfem::OptionsParser args(argc, argv);
+  args.AddOption(&DATA_DIR, "-dataDir", "--data_directory",
+                 "Directory storing input data for tests.");
+  args.Parse();
+  MPI_Init(&argc, &argv);
+
+  // Create Formulation
+  hephaestus::SteadyStateProblemBuilder *problem_builder = new hephaestus::ThermalExpansionFormulation();
+  // Set Mesh
+  mfem::Mesh mesh((std::string(DATA_DIR) + std::string("./simple_cube.g")).c_str(), 1,
+                  1);
+  std::shared_ptr<mfem::ParMesh> pmesh =
+      std::make_shared<mfem::ParMesh>(mfem::ParMesh(MPI_COMM_WORLD, mesh));
+  problem_builder->SetMesh(pmesh);
+  problem_builder->AddFESpace(std::string("H1"), std::string("H1_3D_P1"));
+  problem_builder->AddFESpace(std::string("H1_3"), std::string("H1_3D_P1"), 3, mfem::Ordering::byVDIM);
+
+  problem_builder->AddGridFunction(std::string("temperature"),
+                                   std::string("H1"));
+  problem_builder->AddGridFunction(std::string("displacement"),
+                                   std::string("H1_3"));
+
+  hephaestus::Coefficients coefficients = defineCoefficients();
+  problem_builder->SetCoefficients(coefficients);
+
+  hephaestus::Outputs outputs = defineOutputs();
+  problem_builder->SetOutputs(outputs);
+
+  hephaestus::BCMap boundaries = defineBoundaries();
+  problem_builder->SetBoundaryConditions(boundaries);
+
+  hephaestus::InputParameters solver_options;
+  solver_options.SetParam("Tolerance", float(1.0e-5));
+  solver_options.SetParam("MaxIter", (unsigned int)1000);
+  solver_options.SetParam("PrintLevel", 0);
+  problem_builder->SetSolverOptions(solver_options);
+
+  hephaestus::ProblemBuildSequencer sequencer(problem_builder);
+  sequencer.ConstructEquationSystemProblem();
+  std::unique_ptr<hephaestus::SteadyStateProblem> problem =
+      problem_builder->ReturnProblem();
+  hephaestus::InputParameters exec_params;
+  exec_params.SetParam("Problem", problem.get());
+  hephaestus::SteadyExecutioner *executioner =
+      new hephaestus::SteadyExecutioner(exec_params);
+
+  mfem::out << "Created executioner";
+  executioner->Execute();
+
+  MPI_Finalize();
+}