Viewer.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935
  1. // This file is part of libigl, a simple c++ geometry processing library.
  2. //
  3. // Copyright (C) 2014 Daniele Panozzo <daniele.panozzo@gmail.com>
  4. //
  5. // This Source Code Form is subject to the terms of the Mozilla Public License
  6. // v. 2.0. If a copy of the MPL was not distributed with this file, You can
  7. // obtain one at http://mozilla.org/MPL/2.0/.
  8. #include "Viewer.h"
  9. #include <chrono>
  10. #include <thread>
  11. #include <Eigen/LU>
  12. #include "../gl.h"
  13. #include <GLFW/glfw3.h>
  14. #include <cmath>
  15. #include <cstdio>
  16. #include <sstream>
  17. #include <iomanip>
  18. #include <iostream>
  19. #include <fstream>
  20. #include <algorithm>
  21. #include <limits>
  22. #include <cassert>
  23. #include <igl/project.h>
  24. #include <igl/get_seconds.h>
  25. #include <igl/readOBJ.h>
  26. #include <igl/readOFF.h>
  27. #include <igl/adjacency_list.h>
  28. #include <igl/writeOBJ.h>
  29. #include <igl/writeOFF.h>
  30. #include <igl/massmatrix.h>
  31. #include <igl/file_dialog_open.h>
  32. #include <igl/file_dialog_save.h>
  33. #include <igl/quat_mult.h>
  34. #include <igl/axis_angle_to_quat.h>
  35. #include <igl/trackball.h>
  36. #include <igl/two_axis_valuator_fixed_up.h>
  37. #include <igl/snap_to_canonical_view_quat.h>
  38. #include <igl/unproject.h>
  39. #include <igl/serialize.h>
  40. // Internal global variables used for glfw event handling
  41. static igl::opengl::glfw::Viewer * __viewer;
  42. static double highdpi = 1;
  43. static double scroll_x = 0;
  44. static double scroll_y = 0;
  45. static void glfw_mouse_press(GLFWwindow* window, int button, int action, int modifier)
  46. {
  47. igl::opengl::glfw::Viewer::MouseButton mb;
  48. if (button == GLFW_MOUSE_BUTTON_1)
  49. mb = igl::opengl::glfw::Viewer::MouseButton::Left;
  50. else if (button == GLFW_MOUSE_BUTTON_2)
  51. mb = igl::opengl::glfw::Viewer::MouseButton::Right;
  52. else //if (button == GLFW_MOUSE_BUTTON_3)
  53. mb = igl::opengl::glfw::Viewer::MouseButton::Middle;
  54. if (action == GLFW_PRESS)
  55. __viewer->mouse_down(mb,modifier);
  56. else
  57. __viewer->mouse_up(mb,modifier);
  58. }
  59. static void glfw_error_callback(int error, const char* description)
  60. {
  61. fputs(description, stderr);
  62. }
  63. static void glfw_char_mods_callback(GLFWwindow* window, unsigned int codepoint, int modifier)
  64. {
  65. __viewer->key_pressed(codepoint, modifier);
  66. }
  67. static void glfw_key_callback(GLFWwindow* window, int key, int scancode, int action, int modifier)
  68. {
  69. if (key == GLFW_KEY_ESCAPE && action == GLFW_PRESS)
  70. glfwSetWindowShouldClose(window, GL_TRUE);
  71. if (action == GLFW_PRESS)
  72. __viewer->key_down(key, modifier);
  73. else if(action == GLFW_RELEASE)
  74. __viewer->key_up(key, modifier);
  75. }
  76. static void glfw_window_size(GLFWwindow* window, int width, int height)
  77. {
  78. int w = width*highdpi;
  79. int h = height*highdpi;
  80. __viewer->post_resize(w, h);
  81. }
  82. static void glfw_mouse_move(GLFWwindow* window, double x, double y)
  83. {
  84. __viewer->mouse_move(x*highdpi, y*highdpi);
  85. }
  86. static void glfw_mouse_scroll(GLFWwindow* window, double x, double y)
  87. {
  88. using namespace std;
  89. scroll_x += x;
  90. scroll_y += y;
  91. __viewer->mouse_scroll(y);
  92. }
  93. static void glfw_drop_callback(GLFWwindow *window,int count,const char **filenames)
  94. {
  95. }
  96. namespace igl
  97. {
  98. namespace opengl
  99. {
  100. namespace glfw
  101. {
  102. IGL_INLINE int Viewer::launch(bool resizable,bool fullscreen)
  103. {
  104. // TODO return values are being ignored...
  105. launch_init(resizable,fullscreen);
  106. launch_rendering(true);
  107. launch_shut();
  108. return EXIT_SUCCESS;
  109. }
  110. IGL_INLINE int Viewer::launch_init(bool resizable,bool fullscreen)
  111. {
  112. glfwSetErrorCallback(glfw_error_callback);
  113. if (!glfwInit())
  114. {
  115. return EXIT_FAILURE;
  116. }
  117. glfwWindowHint(GLFW_SAMPLES, 8);
  118. glfwWindowHint(GLFW_CONTEXT_VERSION_MAJOR, 3);
  119. glfwWindowHint(GLFW_CONTEXT_VERSION_MINOR, 3);
  120. #ifdef __APPLE__
  121. glfwWindowHint(GLFW_OPENGL_PROFILE, GLFW_OPENGL_CORE_PROFILE);
  122. glfwWindowHint(GLFW_OPENGL_FORWARD_COMPAT, GL_TRUE);
  123. #endif
  124. if(fullscreen)
  125. {
  126. GLFWmonitor *monitor = glfwGetPrimaryMonitor();
  127. const GLFWvidmode *mode = glfwGetVideoMode(monitor);
  128. window = glfwCreateWindow(mode->width,mode->height,"libigl viewer",monitor,nullptr);
  129. }
  130. else
  131. {
  132. if (core.viewport.tail<2>().any()) {
  133. window = glfwCreateWindow(core.viewport(2),core.viewport(3),"libigl viewer",nullptr,nullptr);
  134. } else {
  135. window = glfwCreateWindow(1280,800,"libigl viewer",nullptr,nullptr);
  136. }
  137. }
  138. if (!window)
  139. {
  140. glfwTerminate();
  141. return EXIT_FAILURE;
  142. }
  143. glfwMakeContextCurrent(window);
  144. // Load OpenGL and its extensions
  145. if (!gladLoadGLLoader((GLADloadproc) glfwGetProcAddress))
  146. {
  147. printf("Failed to load OpenGL and its extensions\n");
  148. return(-1);
  149. }
  150. printf("OpenGL Version %d.%d loaded\n", GLVersion.major, GLVersion.minor);
  151. #if defined(DEBUG) || defined(_DEBUG)
  152. int major, minor, rev;
  153. major = glfwGetWindowAttrib(window, GLFW_CONTEXT_VERSION_MAJOR);
  154. minor = glfwGetWindowAttrib(window, GLFW_CONTEXT_VERSION_MINOR);
  155. rev = glfwGetWindowAttrib(window, GLFW_CONTEXT_REVISION);
  156. printf("OpenGL version received: %d.%d.%d\n", major, minor, rev);
  157. printf("Supported OpenGL is %s\n", (const char*)glGetString(GL_VERSION));
  158. printf("Supported GLSL is %s\n", (const char*)glGetString(GL_SHADING_LANGUAGE_VERSION));
  159. #endif
  160. glfwSetInputMode(window,GLFW_CURSOR,GLFW_CURSOR_NORMAL);
  161. // Initialize FormScreen
  162. __viewer = this;
  163. // Register callbacks
  164. glfwSetKeyCallback(window, glfw_key_callback);
  165. glfwSetCursorPosCallback(window,glfw_mouse_move);
  166. glfwSetWindowSizeCallback(window,glfw_window_size);
  167. glfwSetMouseButtonCallback(window,glfw_mouse_press);
  168. glfwSetScrollCallback(window,glfw_mouse_scroll);
  169. glfwSetCharModsCallback(window,glfw_char_mods_callback);
  170. glfwSetDropCallback(window,glfw_drop_callback);
  171. // Handle retina displays (windows and mac)
  172. int width, height;
  173. glfwGetFramebufferSize(window, &width, &height);
  174. int width_window, height_window;
  175. glfwGetWindowSize(window, &width_window, &height_window);
  176. highdpi = width/width_window;
  177. glfw_window_size(window,width_window,height_window);
  178. //opengl.init();
  179. core.align_camera_center(data().V,data().F);
  180. // Initialize IGL viewer
  181. init();
  182. return EXIT_SUCCESS;
  183. }
  184. IGL_INLINE bool Viewer::launch_rendering(bool loop)
  185. {
  186. // glfwMakeContextCurrent(window);
  187. // Rendering loop
  188. const int num_extra_frames = 5;
  189. int frame_counter = 0;
  190. while (!glfwWindowShouldClose(window))
  191. {
  192. double tic = get_seconds();
  193. draw();
  194. glfwSwapBuffers(window);
  195. if(core.is_animating || frame_counter++ < num_extra_frames)
  196. {
  197. glfwPollEvents();
  198. // In microseconds
  199. double duration = 1000000.*(get_seconds()-tic);
  200. const double min_duration = 1000000./core.animation_max_fps;
  201. if(duration<min_duration)
  202. {
  203. std::this_thread::sleep_for(std::chrono::microseconds((int)(min_duration-duration)));
  204. }
  205. }
  206. else
  207. {
  208. glfwWaitEvents();
  209. frame_counter = 0;
  210. }
  211. if (!loop)
  212. return !glfwWindowShouldClose(window);
  213. }
  214. return EXIT_SUCCESS;
  215. }
  216. IGL_INLINE void Viewer::launch_shut()
  217. {
  218. for(auto & data : data_list)
  219. {
  220. data.meshgl.free();
  221. }
  222. core.shut();
  223. shutdown_plugins();
  224. glfwDestroyWindow(window);
  225. glfwTerminate();
  226. return;
  227. }
  228. IGL_INLINE void Viewer::init()
  229. {
  230. core.init();
  231. if (callback_init)
  232. if (callback_init(*this))
  233. return;
  234. init_plugins();
  235. }
  236. IGL_INLINE void Viewer::init_plugins()
  237. {
  238. // Init all plugins
  239. for (unsigned int i = 0; i<plugins.size(); ++i)
  240. {
  241. plugins[i]->init(this);
  242. }
  243. }
  244. IGL_INLINE void Viewer::shutdown_plugins()
  245. {
  246. for (unsigned int i = 0; i<plugins.size(); ++i)
  247. {
  248. plugins[i]->shutdown();
  249. }
  250. }
  251. IGL_INLINE Viewer::Viewer():
  252. data_list(1),
  253. selected_data_index(0)
  254. {
  255. window = nullptr;
  256. // Temporary variables initialization
  257. down = false;
  258. hack_never_moved = true;
  259. scroll_position = 0.0f;
  260. // Per face
  261. data().set_face_based(false);
  262. // C-style callbacks
  263. callback_init = nullptr;
  264. callback_pre_draw = nullptr;
  265. callback_post_draw = nullptr;
  266. callback_mouse_down = nullptr;
  267. callback_mouse_up = nullptr;
  268. callback_mouse_move = nullptr;
  269. callback_mouse_scroll = nullptr;
  270. callback_key_down = nullptr;
  271. callback_key_up = nullptr;
  272. callback_init_data = nullptr;
  273. callback_pre_draw_data = nullptr;
  274. callback_post_draw_data = nullptr;
  275. callback_mouse_down_data = nullptr;
  276. callback_mouse_up_data = nullptr;
  277. callback_mouse_move_data = nullptr;
  278. callback_mouse_scroll_data = nullptr;
  279. callback_key_down_data = nullptr;
  280. callback_key_up_data = nullptr;
  281. #ifndef IGL_VIEWER_VIEWER_QUIET
  282. const std::string usage(R"(igl::opengl::glfw::Viewer usage:
  283. [drag] Rotate scene
  284. A,a Toggle animation (tight draw loop)
  285. F,f Toggle face based
  286. I,i Toggle invert normals
  287. L,l Toggle wireframe
  288. O,o Toggle orthographic/perspective projection
  289. T,t Toggle filled faces
  290. Z Snap to canonical view
  291. [,] Toggle between rotation control types (trackball, two-axis
  292. valuator with fixed up, 2D mode with no rotation))
  293. <,> Toggle between models
  294. ; Toggle vertex labels
  295. : Toggle face labels)"
  296. );
  297. std::cout<<usage<<std::endl;
  298. #endif
  299. }
  300. IGL_INLINE Viewer::~Viewer()
  301. {
  302. }
  303. IGL_INLINE bool Viewer::load_mesh_from_file(
  304. const std::string & mesh_file_name_string)
  305. {
  306. // first try to load it with a plugin
  307. for (unsigned int i = 0; i<plugins.size(); ++i)
  308. {
  309. if (plugins[i]->load(mesh_file_name_string))
  310. {
  311. return true;
  312. }
  313. }
  314. // Create new data slot and set to selected
  315. if(!(data().F.rows() == 0 && data().V.rows() == 0))
  316. {
  317. append_mesh();
  318. }
  319. data().clear();
  320. size_t last_dot = mesh_file_name_string.rfind('.');
  321. if (last_dot == std::string::npos)
  322. {
  323. std::cerr<<"Error: No file extension found in "<<
  324. mesh_file_name_string<<std::endl;
  325. return false;
  326. }
  327. std::string extension = mesh_file_name_string.substr(last_dot+1);
  328. if (extension == "off" || extension =="OFF")
  329. {
  330. Eigen::MatrixXd V;
  331. Eigen::MatrixXi F;
  332. if (!igl::readOFF(mesh_file_name_string, V, F))
  333. return false;
  334. data().set_mesh(V,F);
  335. }
  336. else if (extension == "obj" || extension =="OBJ")
  337. {
  338. Eigen::MatrixXd corner_normals;
  339. Eigen::MatrixXi fNormIndices;
  340. Eigen::MatrixXd UV_V;
  341. Eigen::MatrixXi UV_F;
  342. Eigen::MatrixXd V;
  343. Eigen::MatrixXi F;
  344. if (!(
  345. igl::readOBJ(
  346. mesh_file_name_string,
  347. V, UV_V, corner_normals, F, UV_F, fNormIndices)))
  348. {
  349. return false;
  350. }
  351. data().set_mesh(V,F);
  352. data().set_uv(UV_V,UV_F);
  353. }
  354. else
  355. {
  356. // unrecognized file type
  357. printf("Error: %s is not a recognized file type.\n",extension.c_str());
  358. return false;
  359. }
  360. data().compute_normals();
  361. data().uniform_colors(Eigen::Vector3d(51.0/255.0,43.0/255.0,33.3/255.0),
  362. Eigen::Vector3d(255.0/255.0,228.0/255.0,58.0/255.0),
  363. Eigen::Vector3d(255.0/255.0,235.0/255.0,80.0/255.0));
  364. if (data().V_uv.rows() == 0)
  365. {
  366. data().grid_texture();
  367. }
  368. core.align_camera_center(data().V,data().F);
  369. for (unsigned int i = 0; i<plugins.size(); ++i)
  370. if (plugins[i]->post_load())
  371. return true;
  372. return true;
  373. }
  374. IGL_INLINE bool Viewer::save_mesh_to_file(
  375. const std::string & mesh_file_name_string)
  376. {
  377. // first try to load it with a plugin
  378. for (unsigned int i = 0; i<plugins.size(); ++i)
  379. if (plugins[i]->save(mesh_file_name_string))
  380. return true;
  381. size_t last_dot = mesh_file_name_string.rfind('.');
  382. if (last_dot == std::string::npos)
  383. {
  384. // No file type determined
  385. std::cerr<<"Error: No file extension found in "<<
  386. mesh_file_name_string<<std::endl;
  387. return false;
  388. }
  389. std::string extension = mesh_file_name_string.substr(last_dot+1);
  390. if (extension == "off" || extension =="OFF")
  391. {
  392. return igl::writeOFF(
  393. mesh_file_name_string,data().V,data().F);
  394. }
  395. else if (extension == "obj" || extension =="OBJ")
  396. {
  397. Eigen::MatrixXd corner_normals;
  398. Eigen::MatrixXi fNormIndices;
  399. Eigen::MatrixXd UV_V;
  400. Eigen::MatrixXi UV_F;
  401. return igl::writeOBJ(mesh_file_name_string,
  402. data().V,
  403. data().F,
  404. corner_normals, fNormIndices, UV_V, UV_F);
  405. }
  406. else
  407. {
  408. // unrecognized file type
  409. printf("Error: %s is not a recognized file type.\n",extension.c_str());
  410. return false;
  411. }
  412. return true;
  413. }
  414. IGL_INLINE bool Viewer::key_pressed(unsigned int unicode_key,int modifiers)
  415. {
  416. if (callback_key_pressed)
  417. if (callback_key_pressed(*this,unicode_key,modifiers))
  418. return true;
  419. for (unsigned int i = 0; i<plugins.size(); ++i)
  420. {
  421. if (plugins[i]->key_pressed(unicode_key, modifiers))
  422. {
  423. return true;
  424. }
  425. }
  426. switch(unicode_key)
  427. {
  428. case 'A':
  429. case 'a':
  430. {
  431. core.is_animating = !core.is_animating;
  432. return true;
  433. }
  434. case 'F':
  435. case 'f':
  436. {
  437. data().set_face_based(!data().face_based);
  438. return true;
  439. }
  440. case 'I':
  441. case 'i':
  442. {
  443. data().dirty |= MeshGL::DIRTY_NORMAL;
  444. data().invert_normals = !data().invert_normals;
  445. return true;
  446. }
  447. case 'L':
  448. case 'l':
  449. {
  450. data().show_lines = !data().show_lines;
  451. return true;
  452. }
  453. case 'O':
  454. case 'o':
  455. {
  456. core.orthographic = !core.orthographic;
  457. return true;
  458. }
  459. case 'T':
  460. case 't':
  461. {
  462. data().show_faces = !data().show_faces;
  463. return true;
  464. }
  465. case 'Z':
  466. {
  467. snap_to_canonical_quaternion();
  468. return true;
  469. }
  470. case '[':
  471. case ']':
  472. {
  473. if(core.rotation_type == ViewerCore::ROTATION_TYPE_TRACKBALL)
  474. core.set_rotation_type(ViewerCore::ROTATION_TYPE_TWO_AXIS_VALUATOR_FIXED_UP);
  475. else
  476. core.set_rotation_type(ViewerCore::ROTATION_TYPE_TRACKBALL);
  477. return true;
  478. }
  479. case '<':
  480. case '>':
  481. {
  482. selected_data_index =
  483. (selected_data_index + data_list.size() + (unicode_key=='>'?1:-1))%data_list.size();
  484. return true;
  485. }
  486. case ';':
  487. data().show_vertid = !data().show_vertid;
  488. return true;
  489. case ':':
  490. data().show_faceid = !data().show_faceid;
  491. return true;
  492. default: break;//do nothing
  493. }
  494. return false;
  495. }
  496. IGL_INLINE bool Viewer::key_down(int key,int modifiers)
  497. {
  498. if (callback_key_down)
  499. if (callback_key_down(*this,key,modifiers))
  500. return true;
  501. for (unsigned int i = 0; i<plugins.size(); ++i)
  502. if (plugins[i]->key_down(key, modifiers))
  503. return true;
  504. return false;
  505. }
  506. IGL_INLINE bool Viewer::key_up(int key,int modifiers)
  507. {
  508. if (callback_key_up)
  509. if (callback_key_up(*this,key,modifiers))
  510. return true;
  511. for (unsigned int i = 0; i<plugins.size(); ++i)
  512. if (plugins[i]->key_up(key, modifiers))
  513. return true;
  514. return false;
  515. }
  516. IGL_INLINE bool Viewer::mouse_down(MouseButton button,int modifier)
  517. {
  518. // Remember mouse location at down even if used by callback/plugin
  519. down_mouse_x = current_mouse_x;
  520. down_mouse_y = current_mouse_y;
  521. if (callback_mouse_down)
  522. if (callback_mouse_down(*this,static_cast<int>(button),modifier))
  523. return true;
  524. for (unsigned int i = 0; i<plugins.size(); ++i)
  525. if(plugins[i]->mouse_down(static_cast<int>(button),modifier))
  526. return true;
  527. down = true;
  528. down_translation = core.model_translation;
  529. // Initialization code for the trackball
  530. Eigen::RowVector3d center;
  531. if (data().V.rows() == 0)
  532. {
  533. center << 0,0,0;
  534. }else
  535. {
  536. center = data().V.colwise().sum()/data().V.rows();
  537. }
  538. Eigen::Vector3f coord =
  539. igl::project(
  540. Eigen::Vector3f(center(0),center(1),center(2)),
  541. (core.view * core.model).eval(),
  542. core.proj,
  543. core.viewport);
  544. down_mouse_z = coord[2];
  545. down_rotation = core.trackball_angle;
  546. mouse_mode = MouseMode::Rotation;
  547. switch (button)
  548. {
  549. case MouseButton::Left:
  550. if (core.rotation_type == ViewerCore::ROTATION_TYPE_NO_ROTATION) {
  551. mouse_mode = MouseMode::Translation;
  552. } else {
  553. mouse_mode = MouseMode::Rotation;
  554. }
  555. break;
  556. case MouseButton::Right:
  557. mouse_mode = MouseMode::Translation;
  558. break;
  559. default:
  560. mouse_mode = MouseMode::None;
  561. break;
  562. }
  563. return true;
  564. }
  565. IGL_INLINE bool Viewer::mouse_up(MouseButton button,int modifier)
  566. {
  567. down = false;
  568. if (callback_mouse_up)
  569. if (callback_mouse_up(*this,static_cast<int>(button),modifier))
  570. return true;
  571. for (unsigned int i = 0; i<plugins.size(); ++i)
  572. if(plugins[i]->mouse_up(static_cast<int>(button),modifier))
  573. return true;
  574. mouse_mode = MouseMode::None;
  575. return true;
  576. }
  577. IGL_INLINE bool Viewer::mouse_move(int mouse_x,int mouse_y)
  578. {
  579. if(hack_never_moved)
  580. {
  581. down_mouse_x = mouse_x;
  582. down_mouse_y = mouse_y;
  583. hack_never_moved = false;
  584. }
  585. current_mouse_x = mouse_x;
  586. current_mouse_y = mouse_y;
  587. if (callback_mouse_move)
  588. if (callback_mouse_move(*this,mouse_x,mouse_y))
  589. return true;
  590. for (unsigned int i = 0; i<plugins.size(); ++i)
  591. if (plugins[i]->mouse_move(mouse_x, mouse_y))
  592. return true;
  593. if (down)
  594. {
  595. switch (mouse_mode)
  596. {
  597. case MouseMode::Rotation:
  598. {
  599. switch(core.rotation_type)
  600. {
  601. default:
  602. assert(false && "Unknown rotation type");
  603. case ViewerCore::ROTATION_TYPE_NO_ROTATION:
  604. break;
  605. case ViewerCore::ROTATION_TYPE_TRACKBALL:
  606. igl::trackball(
  607. core.viewport(2),
  608. core.viewport(3),
  609. 2.0f,
  610. down_rotation,
  611. down_mouse_x,
  612. down_mouse_y,
  613. mouse_x,
  614. mouse_y,
  615. core.trackball_angle);
  616. break;
  617. case ViewerCore::ROTATION_TYPE_TWO_AXIS_VALUATOR_FIXED_UP:
  618. igl::two_axis_valuator_fixed_up(
  619. core.viewport(2),core.viewport(3),
  620. 2.0,
  621. down_rotation,
  622. down_mouse_x, down_mouse_y, mouse_x, mouse_y,
  623. core.trackball_angle);
  624. break;
  625. }
  626. //Eigen::Vector4f snapq = core.trackball_angle;
  627. break;
  628. }
  629. case MouseMode::Translation:
  630. {
  631. //translation
  632. Eigen::Vector3f pos1 = igl::unproject(Eigen::Vector3f(mouse_x, core.viewport[3] - mouse_y, down_mouse_z), (core.view * core.model).eval(), core.proj, core.viewport);
  633. Eigen::Vector3f pos0 = igl::unproject(Eigen::Vector3f(down_mouse_x, core.viewport[3] - down_mouse_y, down_mouse_z), (core.view * core.model).eval(), core.proj, core.viewport);
  634. Eigen::Vector3f diff = pos1 - pos0;
  635. core.model_translation = down_translation + Eigen::Vector3f(diff[0],diff[1],diff[2]);
  636. break;
  637. }
  638. case MouseMode::Zoom:
  639. {
  640. float delta = 0.001f * (mouse_x - down_mouse_x + mouse_y - down_mouse_y);
  641. core.camera_zoom *= 1 + delta;
  642. down_mouse_x = mouse_x;
  643. down_mouse_y = mouse_y;
  644. break;
  645. }
  646. default:
  647. break;
  648. }
  649. }
  650. return true;
  651. }
  652. IGL_INLINE bool Viewer::mouse_scroll(float delta_y)
  653. {
  654. scroll_position += delta_y;
  655. if (callback_mouse_scroll)
  656. if (callback_mouse_scroll(*this,delta_y))
  657. return true;
  658. for (unsigned int i = 0; i<plugins.size(); ++i)
  659. if (plugins[i]->mouse_scroll(delta_y))
  660. return true;
  661. // Only zoom if there's actually a change
  662. if(delta_y != 0)
  663. {
  664. float mult = (1.0+((delta_y>0)?1.:-1.)*0.05);
  665. const float min_zoom = 0.1f;
  666. core.camera_zoom = (core.camera_zoom * mult > min_zoom ? core.camera_zoom * mult : min_zoom);
  667. }
  668. return true;
  669. }
  670. IGL_INLINE bool Viewer::load_scene()
  671. {
  672. std::string fname = igl::file_dialog_open();
  673. if(fname.length() == 0)
  674. return false;
  675. return load_scene(fname);
  676. }
  677. IGL_INLINE bool Viewer::load_scene(std::string fname)
  678. {
  679. igl::deserialize(core,"Core",fname.c_str());
  680. igl::deserialize(data(),"Data",fname.c_str());
  681. return true;
  682. }
  683. IGL_INLINE bool Viewer::save_scene()
  684. {
  685. std::string fname = igl::file_dialog_save();
  686. if (fname.length() == 0)
  687. return false;
  688. return save_scene(fname);
  689. }
  690. IGL_INLINE bool Viewer::save_scene(std::string fname)
  691. {
  692. igl::serialize(core,"Core",fname.c_str(),true);
  693. igl::serialize(data(),"Data",fname.c_str());
  694. return true;
  695. }
  696. IGL_INLINE void Viewer::draw()
  697. {
  698. using namespace std;
  699. using namespace Eigen;
  700. int width, height;
  701. glfwGetFramebufferSize(window, &width, &height);
  702. int width_window, height_window;
  703. glfwGetWindowSize(window, &width_window, &height_window);
  704. auto highdpi_tmp = width/width_window;
  705. if(fabs(highdpi_tmp-highdpi)>1e-8)
  706. {
  707. post_resize(width, height);
  708. highdpi=highdpi_tmp;
  709. }
  710. core.clear_framebuffers();
  711. if (callback_pre_draw)
  712. {
  713. if (callback_pre_draw(*this))
  714. {
  715. return;
  716. }
  717. }
  718. for (unsigned int i = 0; i<plugins.size(); ++i)
  719. {
  720. if (plugins[i]->pre_draw())
  721. {
  722. return;
  723. }
  724. }
  725. for(int i = 0;i<data_list.size();i++)
  726. {
  727. core.draw(data_list[i]);
  728. }
  729. if (callback_post_draw)
  730. {
  731. if (callback_post_draw(*this))
  732. {
  733. return;
  734. }
  735. }
  736. for (unsigned int i = 0; i<plugins.size(); ++i)
  737. {
  738. if (plugins[i]->post_draw())
  739. {
  740. break;
  741. }
  742. }
  743. }
  744. IGL_INLINE void Viewer::resize(int w,int h)
  745. {
  746. if (window) {
  747. glfwSetWindowSize(window, w/highdpi, h/highdpi);
  748. }
  749. post_resize(w, h);
  750. }
  751. IGL_INLINE void Viewer::post_resize(int w,int h)
  752. {
  753. core.viewport = Eigen::Vector4f(0,0,w,h);
  754. for (unsigned int i = 0; i<plugins.size(); ++i)
  755. {
  756. plugins[i]->post_resize(w, h);
  757. }
  758. }
  759. IGL_INLINE void Viewer::snap_to_canonical_quaternion()
  760. {
  761. Eigen::Quaternionf snapq = this->core.trackball_angle;
  762. igl::snap_to_canonical_view_quat(snapq,1.0f,this->core.trackball_angle);
  763. }
  764. IGL_INLINE void Viewer::open_dialog_load_mesh()
  765. {
  766. std::string fname = igl::file_dialog_open();
  767. if (fname.length() == 0)
  768. return;
  769. this->load_mesh_from_file(fname.c_str());
  770. }
  771. IGL_INLINE void Viewer::open_dialog_save_mesh()
  772. {
  773. std::string fname = igl::file_dialog_save();
  774. if(fname.length() == 0)
  775. return;
  776. this->save_mesh_to_file(fname.c_str());
  777. }
  778. IGL_INLINE ViewerData& Viewer::data()
  779. {
  780. assert(!data_list.empty() && "data_list should never be empty");
  781. assert(
  782. (selected_data_index >= 0 && selected_data_index < data_list.size()) &&
  783. "selected_data_index should be in bounds");
  784. return data_list[selected_data_index];
  785. }
  786. IGL_INLINE size_t Viewer::append_mesh()
  787. {
  788. assert(data_list.size() >= 1);
  789. data_list.emplace_back();
  790. selected_data_index = data_list.size()-1;
  791. return data_list.size();
  792. }
  793. IGL_INLINE bool Viewer::erase_mesh(const size_t index)
  794. {
  795. assert((index >= 0 && index < data_list.size()) && "index should be in bounds");
  796. assert(data_list.size() >= 1);
  797. if(data_list.size() == 1)
  798. {
  799. // Cannot remove last mesh
  800. return false;
  801. }
  802. data_list[index].meshgl.free();
  803. data_list.erase(data_list.begin() + index);
  804. if(selected_data_index >= index && selected_data_index>0)
  805. {
  806. selected_data_index--;
  807. }
  808. return true;
  809. }
  810. } // end namespace
  811. } // end namespace
  812. }