cflin
2 years ago
5 changed files with 393 additions and 190 deletions
@ -0,0 +1,233 @@ |
|||||
|
//
|
||||
|
// Created by cflin on 4/14/23.
|
||||
|
//
|
||||
|
#include <igl/file_dialog_save.h> |
||||
|
#include "StaticSimGUI.h" |
||||
|
#include "Util.h" |
||||
|
namespace ssim{ |
||||
|
|
||||
|
void StaticSimGUI::menu_draw_viewer_menu() { |
||||
|
// Draw parent menu content
|
||||
|
// menu_.draw_viewer_menu();
|
||||
|
|
||||
|
auto io_menu = [&]() { |
||||
|
if (ImGui::Button("加载文件##IO", ImVec2(-1, 0))) { |
||||
|
#ifndef DEBUG_SSIM |
||||
|
std::string f_json = igl::file_dialog_open(); |
||||
|
if (f_json != "" ) { |
||||
|
reload_json(f_json); |
||||
|
gui_ctrl_.is_initialized = true; |
||||
|
gui_ctrl_.is_loaded_json = true; |
||||
|
} |
||||
|
#else |
||||
|
load(CMAKE_SOURCE_DIR "/sim-test/rigid-test/rocker-arm/config.json"); |
||||
|
#endif |
||||
|
} |
||||
|
|
||||
|
|
||||
|
|
||||
|
|
||||
|
}; |
||||
|
auto model_menu = [&]() { |
||||
|
|
||||
|
if (ImGui::Checkbox("网格可视化", &gui_ctrl_.is_visible_mesh)) { |
||||
|
gui_ctrl_.is_modified = true; |
||||
|
} |
||||
|
|
||||
|
|
||||
|
// --------------------------------------------------------------------
|
||||
|
if (ImGui::CollapsingHeader("边界条件", ImGuiTreeNodeFlags_DefaultOpen)) { |
||||
|
|
||||
|
if (ImGui::Checkbox("边界可视化", &gui_ctrl_.is_visible_BC)) { |
||||
|
gui_ctrl_.is_modified = true; |
||||
|
} |
||||
|
|
||||
|
if (ImGui::CollapsingHeader( |
||||
|
"Dirichlet边界", ImGuiTreeNodeFlags_Leaf)) { |
||||
|
// --------------------------------------------------------------------
|
||||
|
|
||||
|
for (int i = 0; i < sp_StaticSim_->DirichletBCs.size(); ++i) { |
||||
|
// each boundary
|
||||
|
// ImGui::Checkbox("受力", &m_state.m_solve_collisions);
|
||||
|
auto &i_DBC = sp_StaticSim_->DirichletBCs[i]; |
||||
|
std::vector<float> v_min_point = {(float) i_DBC.relMinBBox.x(), |
||||
|
(float) i_DBC.relMinBBox.y(), |
||||
|
(float) i_DBC.relMinBBox.z()}; |
||||
|
std::vector<float> v_max_point = {(float) i_DBC.relMaxBBox.x(), |
||||
|
(float) i_DBC.relMaxBBox.y(), |
||||
|
(float) i_DBC.relMaxBBox.z()}; |
||||
|
if (ImGui::InputFloat3(("Dirichlet最小值 " + std::to_string(i + 1)).c_str(), |
||||
|
v_min_point.data()) || |
||||
|
ImGui::InputFloat3(("Dirichlet最大值 " + std::to_string(i + 1)).c_str(), |
||||
|
v_max_point.data())) { |
||||
|
gui_ctrl_.is_modified = true; |
||||
|
i_DBC.relMinBBox = {v_min_point[0], v_min_point[1], v_min_point[2]}; |
||||
|
i_DBC.relMaxBBox = {v_max_point[0], v_max_point[1], v_max_point[2]}; |
||||
|
} |
||||
|
} |
||||
|
if (gui_ctrl_.is_modified) { |
||||
|
sp_StaticSim_->updateBC(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
if (ImGui::CollapsingHeader( |
||||
|
"Neumann边界", ImGuiTreeNodeFlags_Leaf)) { |
||||
|
// --------------------------------------------------------------------
|
||||
|
|
||||
|
for (int i = 0; i < sp_StaticSim_->NeumannBCs.size(); ++i) { |
||||
|
auto &i_NBC = sp_StaticSim_->NeumannBCs[i]; |
||||
|
std::vector<float> v_min_point = {(float) i_NBC.relMinBBox.x(), |
||||
|
(float) i_NBC.relMinBBox.y(), |
||||
|
(float) i_NBC.relMinBBox.z()}; |
||||
|
std::vector<float> v_max_point = {(float) i_NBC.relMaxBBox.x(), |
||||
|
(float) i_NBC.relMaxBBox.y(), |
||||
|
(float) i_NBC.relMaxBBox.z()}; |
||||
|
std::vector<float> v_force = {(float) i_NBC.force.x(), (float) i_NBC.force.y(), |
||||
|
(float) i_NBC.force.z()}; |
||||
|
if ( |
||||
|
ImGui::InputFloat3(("Neumann最小值 " + std::to_string(i + 1)).c_str(), |
||||
|
v_min_point.data()) || |
||||
|
ImGui::InputFloat3(("Neumann最大值 " + std::to_string(i + 1)).c_str(), |
||||
|
v_max_point.data()) || |
||||
|
ImGui::InputFloat3(("Neumann力 " + std::to_string(i + 1)).c_str(), |
||||
|
v_force.data())) { |
||||
|
gui_ctrl_.is_modified = true; |
||||
|
i_NBC.relMinBBox = {v_min_point[0], v_min_point[1], v_min_point[2]}; |
||||
|
i_NBC.relMaxBBox = {v_max_point[0], v_max_point[1], v_max_point[2]}; |
||||
|
i_NBC.force = {v_force[0], v_force[1], v_force[2]}; |
||||
|
|
||||
|
} |
||||
|
} |
||||
|
if (gui_ctrl_.is_modified) { |
||||
|
sp_StaticSim_->updateBC(); |
||||
|
} |
||||
|
|
||||
|
} |
||||
|
|
||||
|
} |
||||
|
|
||||
|
// --------------------------------------------------------------------
|
||||
|
if (ImGui::CollapsingHeader("材料设置", ImGuiTreeNodeFlags_DefaultOpen)) { |
||||
|
|
||||
|
ImGui::InputFloat("杨氏模量", &sp_StaticSim_->get_material_property().Youngs_Modulus, 1.0f, 1e10f, |
||||
|
"%.3f"); |
||||
|
ImGui::InputFloat("泊松比", &sp_StaticSim_->get_material_property().Poisson_ratio, 0.01f, 1.0f, |
||||
|
"%.3f"); |
||||
|
ImGui::InputFloat("密度", &sp_StaticSim_->get_material_property().density, 1.0f, 1000.0f, "%.3f"); |
||||
|
} |
||||
|
|
||||
|
if (ImGui::Button(("原始模型"), ImVec2(-1, 0))) { |
||||
|
gui_ctrl_.is_single_color = true; |
||||
|
gui_ctrl_.is_modified = true; |
||||
|
} |
||||
|
|
||||
|
if (ImGui::Button("开始计算", ImVec2(-1, 0))) { |
||||
|
sp_StaticSim_->simulation(); |
||||
|
gui_ctrl_.is_modified = true; |
||||
|
gui_ctrl_.is_solved = true; |
||||
|
gui_ctrl_.is_single_color = false; |
||||
|
gui_ctrl_.is_visible_BC = false; |
||||
|
|
||||
|
} |
||||
|
}; |
||||
|
auto result_menu = [&]() { |
||||
|
ImGui::TextWrapped("柔顺度 (compliance): %.3e", |
||||
|
sp_StaticSim_->EvaluateTarget(ssim::SimTargetOption::COMPLIANCE)(0, 0)); |
||||
|
static const ImVec2 button_size(200,40); |
||||
|
static bool arr_extract[SimTargetOption::Target::ENUM_SIZE]={false}; |
||||
|
int idx=0; |
||||
|
if (ImGui::Button("位移范数",button_size)) { |
||||
|
gui_ctrl_.set_target(ssim::SimTargetOption::Target::U_NORM); |
||||
|
} |
||||
|
ImGui::SameLine(); |
||||
|
ImGui::Checkbox("导出位移范数",arr_extract+idx); |
||||
|
++idx; |
||||
|
|
||||
|
if (ImGui::Button("位移-X", button_size)) { |
||||
|
gui_ctrl_.set_target(ssim::SimTargetOption::Target::UX); |
||||
|
} |
||||
|
ImGui::SameLine(); |
||||
|
ImGui::Checkbox("导出位移-X",arr_extract+idx); |
||||
|
++idx; |
||||
|
|
||||
|
if (ImGui::Button("位移-Y", button_size)) { |
||||
|
gui_ctrl_.set_target(ssim::SimTargetOption::Target::UY); |
||||
|
} |
||||
|
ImGui::SameLine(); |
||||
|
ImGui::Checkbox("导出位移-Y",arr_extract+idx); |
||||
|
++idx; |
||||
|
|
||||
|
if (ImGui::Button("位移-Z", button_size)) { |
||||
|
gui_ctrl_.set_target(ssim::SimTargetOption::Target::UZ); |
||||
|
} |
||||
|
ImGui::SameLine(); |
||||
|
ImGui::Checkbox("导出位移-Z",arr_extract+idx); |
||||
|
++idx; |
||||
|
|
||||
|
if (ImGui::Button("应力范数", button_size)) { |
||||
|
gui_ctrl_.set_target(ssim::SimTargetOption::Target::S_NORM); |
||||
|
} |
||||
|
ImGui::SameLine(); |
||||
|
ImGui::Checkbox("导出应力范数",arr_extract+idx); |
||||
|
++idx; |
||||
|
|
||||
|
if (ImGui::Button("冯氏应力", button_size)) { |
||||
|
gui_ctrl_.set_target(ssim::SimTargetOption::Target::S_VON_Mises); |
||||
|
} |
||||
|
ImGui::SameLine(); |
||||
|
ImGui::Checkbox("导出冯氏应力",arr_extract+idx); |
||||
|
++idx; |
||||
|
|
||||
|
if (ImGui::Button("应力-X", button_size)) { |
||||
|
gui_ctrl_.set_target(ssim::SimTargetOption::Target::SX); |
||||
|
} |
||||
|
ImGui::SameLine(); |
||||
|
ImGui::Checkbox("导出应力-X",arr_extract+idx); |
||||
|
++idx; |
||||
|
|
||||
|
if (ImGui::Button("应力-Y", button_size)) { |
||||
|
gui_ctrl_.set_target(ssim::SimTargetOption::Target::SY); |
||||
|
} |
||||
|
ImGui::SameLine(); |
||||
|
ImGui::Checkbox("导出应力-Y",arr_extract+idx); |
||||
|
++idx; |
||||
|
|
||||
|
if (ImGui::Button("应力-Z", button_size)) { |
||||
|
gui_ctrl_.set_target(ssim::SimTargetOption::Target::SZ); |
||||
|
} |
||||
|
ImGui::SameLine(); |
||||
|
ImGui::Checkbox("导出应力-Z",arr_extract+idx); |
||||
|
++idx; |
||||
|
|
||||
|
static bool is_extract_coord= false; |
||||
|
if(ImGui::Button("导出结果文件",button_size)){ |
||||
|
std::string extract_dir=ssim::directory_dialog_save(); |
||||
|
spdlog::info("extract to: {}",extract_dir); |
||||
|
std::vector<bool> v_is_extract(arr_extract,arr_extract+SimTargetOption::Target::ENUM_SIZE); |
||||
|
sp_StaticSim_->ExtractTarget(extract_dir,v_is_extract,is_extract_coord); |
||||
|
} |
||||
|
ImGui::SameLine(); |
||||
|
ImGui::Checkbox("导出坐标",&is_extract_coord); |
||||
|
|
||||
|
}; |
||||
|
|
||||
|
|
||||
|
if (ImGui::CollapsingHeader("载入模型", ImGuiTreeNodeFlags_DefaultOpen)) { |
||||
|
io_menu(); |
||||
|
} |
||||
|
if (!gui_ctrl_.is_initialized && gui_ctrl_.is_loaded_json) { |
||||
|
if (ImGui::CollapsingHeader("模型设置", ImGuiTreeNodeFlags_DefaultOpen)) { |
||||
|
model_menu(); |
||||
|
} |
||||
|
|
||||
|
if (gui_ctrl_.is_solved) { |
||||
|
if (ImGui::CollapsingHeader("仿真结果", ImGuiTreeNodeFlags_DefaultOpen)) { |
||||
|
result_menu(); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
} |
||||
|
|
||||
|
|
||||
|
} |
||||
@ -0,0 +1,116 @@ |
|||||
|
//
|
||||
|
// Created by cflin on 4/14/23.
|
||||
|
//
|
||||
|
|
||||
|
#ifndef EXAMPLE_UTIL_H |
||||
|
#define EXAMPLE_UTIL_H |
||||
|
#include <string> |
||||
|
#include <cstdio> |
||||
|
#include <cstring> |
||||
|
|
||||
|
#ifdef _WIN32 |
||||
|
#include <windows.h> |
||||
|
#include <Commdlg.h> |
||||
|
#endif |
||||
|
namespace ssim { |
||||
|
|
||||
|
inline std::string directory_dialog_save(){ |
||||
|
const int FILE_DIALOG_MAX_BUFFER = 1024; |
||||
|
char buffer[FILE_DIALOG_MAX_BUFFER]; |
||||
|
buffer[0] = '\0'; |
||||
|
buffer[FILE_DIALOG_MAX_BUFFER - 1] = 'x'; // Initialize last character with a char != '\0'
|
||||
|
|
||||
|
#ifdef __APPLE__ |
||||
|
// For apple use applescript hack
|
||||
|
// There is currently a bug in Applescript that strips extensions off
|
||||
|
// of chosen existing files in the "choose file name" dialog
|
||||
|
// I'm assuming that will be fixed soon
|
||||
|
FILE * output = popen( |
||||
|
"osascript -e \"" |
||||
|
" tell application \\\"System Events\\\"\n" |
||||
|
" activate\n" |
||||
|
" set existing_file to choose file name\n" |
||||
|
" end tell\n" |
||||
|
" set existing_file_path to (POSIX path of (existing_file))\n" |
||||
|
"\" 2>/dev/null | tr -d '\n' ","r"); |
||||
|
if (output) |
||||
|
{ |
||||
|
auto ret = fgets(buffer, FILE_DIALOG_MAX_BUFFER, output); |
||||
|
if (ret == NULL || ferror(output)) |
||||
|
{ |
||||
|
// I/O error
|
||||
|
buffer[0] = '\0'; |
||||
|
} |
||||
|
if (buffer[FILE_DIALOG_MAX_BUFFER - 1] == '\0') |
||||
|
{ |
||||
|
// File name too long, buffer has been filled, so we return empty string instead
|
||||
|
buffer[0] = '\0'; |
||||
|
} |
||||
|
} |
||||
|
#elif defined _WIN32 |
||||
|
|
||||
|
// Use native windows file dialog box
|
||||
|
// (code contributed by Tino Weinkauf)
|
||||
|
|
||||
|
OPENFILENAME ofn; // common dialog box structure
|
||||
|
char szFile[260]; // buffer for file name
|
||||
|
|
||||
|
// Initialize OPENFILENAME
|
||||
|
ZeroMemory(&ofn, sizeof(ofn)); |
||||
|
ofn.lStructSize = sizeof(ofn); |
||||
|
ofn.hwndOwner = NULL;//hwnd;
|
||||
|
ofn.lpstrFile = szDir; |
||||
|
// Set lpstrFile[0] to '\0' so that GetOpenFileName does not
|
||||
|
// use the contents of szFile to initialize itself.
|
||||
|
ofn.lpstrFile[0] = '\0'; |
||||
|
ofn.nMaxFile = sizeof(szDir); |
||||
|
ofn.lpstrFilter = ""; |
||||
|
ofn.nFilterIndex = 1; |
||||
|
ofn.lpstrFileTitle = NULL; |
||||
|
ofn.nMaxFileTitle = 0; |
||||
|
ofn.lpstrInitialDir = NULL; |
||||
|
ofn.Flags = OFN_PATHMUSTEXIST; |
||||
|
|
||||
|
// Display the Open dialog box.
|
||||
|
int pos = 0; |
||||
|
if (GetSaveFileName(&ofn)==TRUE) |
||||
|
{ |
||||
|
while(ofn.lpstrFile[pos] != '\0') |
||||
|
{ |
||||
|
buffer[pos] = (char)ofn.lpstrFile[pos]; |
||||
|
pos++; |
||||
|
} |
||||
|
buffer[pos] = 0; |
||||
|
} |
||||
|
|
||||
|
#else |
||||
|
// For every other machine type use zenity
|
||||
|
FILE * output = popen("/usr/bin/zenity --directory --file-selection --save","r"); |
||||
|
if (output) |
||||
|
{ |
||||
|
auto ret = fgets(buffer, FILE_DIALOG_MAX_BUFFER, output); |
||||
|
if (ret == NULL || ferror(output)) |
||||
|
{ |
||||
|
// I/O error
|
||||
|
buffer[0] = '\0'; |
||||
|
} |
||||
|
if (buffer[FILE_DIALOG_MAX_BUFFER - 1] == '\0') |
||||
|
{ |
||||
|
// File name too long, buffer has been filled, so we return empty string instead
|
||||
|
buffer[0] = '\0'; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// Replace last '\n' by '\0'
|
||||
|
if(strlen(buffer) > 0) |
||||
|
{ |
||||
|
buffer[strlen(buffer)-1] = '\0'; |
||||
|
} |
||||
|
|
||||
|
#endif |
||||
|
return std::string(buffer); |
||||
|
} |
||||
|
|
||||
|
} // ssim
|
||||
|
|
||||
|
#endif //EXAMPLE_UTIL_H
|
||||
Loading…
Reference in new issue