Now viewer menu can be overriden with lambda function (for simple override), or virtual methods (for more complex situations).
74 lines
2.0 KiB
C++
Executable File
74 lines
2.0 KiB
C++
Executable File
#include "tutorial_shared_path.h"
|
|
#include <igl/opengl/glfw/Viewer.h>
|
|
#include <igl/opengl/glfw/imgui/ImGuiMenu.h>
|
|
#include <igl/opengl/glfw/imgui/ImGuiHelpers.h>
|
|
#include <string>
|
|
#include <map>
|
|
#include <iostream>
|
|
|
|
// Add custom data to a viewer mesh
|
|
struct MeshData
|
|
{
|
|
std::string name;
|
|
Eigen::RowVector3d color;
|
|
};
|
|
int last_colored_index = -1;
|
|
|
|
// Set colors of each mesh
|
|
void update_colors(igl::opengl::glfw::Viewer &viewer)
|
|
{
|
|
for (auto &data : viewer.data_list)
|
|
{
|
|
data.set_colors(data.attr<MeshData>().color);
|
|
}
|
|
viewer.data_list[viewer.selected_data_index].set_colors(Eigen::RowVector3d(0.9,0.1,0.1));
|
|
last_colored_index = viewer.selected_data_index;
|
|
}
|
|
|
|
int main(int argc, char * argv[])
|
|
{
|
|
igl::opengl::glfw::Viewer viewer;
|
|
auto names = {"cube.obj","sphere.obj","xcylinder.obj","ycylinder.obj","zcylinder.obj"};
|
|
for(const auto & name : names)
|
|
{
|
|
viewer.load_mesh_from_file(std::string(TUTORIAL_SHARED_PATH) + "/" + name);
|
|
viewer.data().attr<MeshData>().name = name;
|
|
viewer.data().attr<MeshData>().color = Eigen::RowVector3d::Random();
|
|
}
|
|
|
|
// Attach a custom menu
|
|
igl::opengl::glfw::imgui::ImGuiMenu menu;
|
|
viewer.plugins.push_back(&menu);
|
|
|
|
// Customize default menu
|
|
menu.draw_viewer_menu_func = [&]()
|
|
{
|
|
if (ImGui::Combo("Selected Mesh", (int *) &viewer.selected_data_index,
|
|
[&](int i) { return viewer.data_list[i].attr<MeshData>().name.c_str(); },
|
|
viewer.data_list.size())
|
|
|| last_colored_index != viewer.selected_data_index)
|
|
{
|
|
update_colors(viewer);
|
|
}
|
|
};
|
|
|
|
// Color each mesh differently
|
|
update_colors(viewer);
|
|
|
|
viewer.callback_key_down =
|
|
[&](igl::opengl::glfw::Viewer &, unsigned int key, int mod)
|
|
{
|
|
// Delete
|
|
if(key == '3')
|
|
{
|
|
viewer.erase_mesh(viewer.selected_data_index);
|
|
update_colors(viewer);
|
|
return true;
|
|
}
|
|
return false;
|
|
};
|
|
|
|
viewer.launch();
|
|
return EXIT_SUCCESS;
|
|
}
|