ProjectStore.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612
  1. // Include own headers
  2. #include "ProjectStore.hpp"
  3. // Include modules
  4. #include "MainWindow.hpp"
  5. #include "StringBasics.hpp"
  6. #include "TrackPointRenderer.hpp"
  7. #include "PlatformSupport.hpp"
  8. #include "STLImport.hpp"
  9. #include "MeshTools.hpp"
  10. // Include dependencies
  11. #include <typeinfo>
  12. #include <iostream>
  13. #include <filesystem>
  14. #define META_NAMESPACE "tk-ar-tracking"
  15. ProjectStore::ProjectStore() {
  16. _projectLoaded = false;
  17. _projectModified = false;
  18. load3mfLib();
  19. }
  20. ProjectStore::~ProjectStore() {
  21. }
  22. void ProjectStore::loadMesh(std::string meshFile) {
  23. if (meshFile == "") {
  24. return;
  25. }
  26. if (StringBasics::endsWithCaseInsensitive(meshFile, ".STL")) {
  27. _projectLoaded = true;
  28. _projectModified = false;
  29. // Read STL file
  30. std::vector<Lib3MF::sPosition> verticesBuffer;
  31. std::vector<Lib3MF::sTriangle> triangleBuffer;
  32. STLImport::readSTL(meshFile, &verticesBuffer, &triangleBuffer);
  33. Lib3MF::PMeshObject baseMesh = _project->AddMeshObject();
  34. baseMesh->SetGeometry(verticesBuffer, triangleBuffer);
  35. render3MFMesh();
  36. MainWindow::getInstance()->renderView(Edit);
  37. } else if (StringBasics::endsWithCaseInsensitive(meshFile, ".3MF")) {
  38. _projectLoaded = true;
  39. _projectModified = false;
  40. // Read 3MF file
  41. Lib3MF::PReader reader = _project->QueryReader("3mf");
  42. reader->ReadFromFile(meshFile);
  43. render3MFMesh();
  44. MainWindow::getInstance()->renderView(Edit);
  45. } else {
  46. MainWindow::getInstance()->showErrorMessage("Unsupported file type. Please use .stl or .3mf files.", "Error opening file.");
  47. }
  48. }
  49. bool ProjectStore::loadProject(std::string projectFile) {
  50. if (projectFile == "") {
  51. return false;
  52. }
  53. if (!_projectLoaded) {
  54. Lib3MF::PReader reader = _project->QueryReader("3mf");
  55. reader->ReadFromFile(projectFile);
  56. _projectLoaded = true;
  57. _projectModified = false;
  58. _projectFile = projectFile;
  59. loadMetaData();
  60. return true;
  61. }
  62. return false;
  63. }
  64. bool ProjectStore::saveProject() {
  65. if (_projectFile != "") {
  66. return saveProject(_projectFile);
  67. }
  68. return false;
  69. }
  70. bool ProjectStore::saveProject(std::string path) {
  71. if (path == "") {
  72. return false;
  73. }
  74. updateMetaData();
  75. Lib3MF::PWriter writer = _project->QueryWriter("3mf");
  76. writer->WriteToFile(path);
  77. _projectFile = path;
  78. _projectModified = false;
  79. return true;
  80. }
  81. bool ProjectStore::exportProject(std::string path, ExportSettings settings) {
  82. if (path == "") {
  83. return false;
  84. }
  85. OpenScadRenderer* renderer = new OpenScadRenderer();
  86. // Base for rendering mesh
  87. Lib3MF::PWriter writer = _project->QueryWriter("3mf");
  88. writer->WriteToFile(std::filesystem::temp_directory_path().u8string() + fileDelimiter + "trackpointapp_export.3mf");
  89. // Export file
  90. Lib3MF::PModel exportModel = _wrapper->CreateModel();
  91. Lib3MF::PMetaDataGroup metaData = exportModel->GetMetaDataGroup();
  92. if (settings.OptiTrack) {
  93. renderer->renderOptiTrack(_optiTrackPoints);
  94. Lib3MF::PModel optiTrackModel = _wrapper->CreateModel();
  95. Lib3MF::PReader reader = optiTrackModel->QueryReader("3mf");
  96. reader->ReadFromFile(std::filesystem::temp_directory_path().u8string() + fileDelimiter + "trackpointapp_render_optitrack.3mf");
  97. Lib3MF::PMeshObjectIterator meshIterator = optiTrackModel->GetMeshObjects();
  98. if (meshIterator->Count() != 1) {
  99. return false;
  100. }
  101. meshIterator->MoveNext();
  102. Lib3MF::PMeshObject renderedMesh = meshIterator->GetCurrentMeshObject();
  103. Lib3MF::PMeshObject exportMesh = exportModel->AddMeshObject();
  104. exportMesh->SetName("optitrack");
  105. std::vector<Lib3MF::sPosition> verticesBuffer;
  106. std::vector<Lib3MF::sTriangle> triangleBuffer;
  107. renderedMesh->GetVertices(verticesBuffer);
  108. renderedMesh->GetTriangleIndices(triangleBuffer);
  109. exportMesh->SetGeometry(verticesBuffer, triangleBuffer);
  110. json trackpointData = json::array();
  111. for (OptiTrackPoint* point: _optiTrackPoints) {
  112. osg::Vec3 trackPoint = point->getTrackPoint();
  113. osg::Vec3 trackNormal = point->getNormal();
  114. json pointData = {
  115. {"point", {trackPoint.x(), trackPoint.y(), trackPoint.z()}},
  116. {"normal", {trackNormal.x(), trackNormal.y(), trackNormal.z()}}
  117. };
  118. trackpointData.push_back(pointData);
  119. }
  120. Lib3MF::PMetaDataGroup optiMetaData = exportMesh->GetMetaDataGroup();
  121. optiMetaData->AddMetaData(META_NAMESPACE, "trackpoints-optitrack", trackpointData.dump(), "string", true);
  122. exportModel->AddBuildItem(exportMesh.get(), _wrapper->GetIdentityTransform());
  123. }
  124. if (settings.EMFTrack) {
  125. renderer->renderEMFTrack(_emfTrackPoints);
  126. Lib3MF::PModel emfTrackModel = _wrapper->CreateModel();
  127. Lib3MF::PReader reader = emfTrackModel->QueryReader("3mf");
  128. reader->ReadFromFile(std::filesystem::temp_directory_path().u8string() + fileDelimiter + "trackpointapp_render_emftrack.3mf");
  129. Lib3MF::PMeshObjectIterator meshIterator = emfTrackModel->GetMeshObjects();
  130. if (meshIterator->Count() != 1) {
  131. return false;
  132. }
  133. meshIterator->MoveNext();
  134. Lib3MF::PMeshObject renderedMesh = meshIterator->GetCurrentMeshObject();
  135. Lib3MF::PMeshObject exportMesh = exportModel->AddMeshObject();
  136. exportMesh->SetName("emftrack");
  137. std::vector<Lib3MF::sPosition> verticesBuffer;
  138. std::vector<Lib3MF::sTriangle> triangleBuffer;
  139. renderedMesh->GetVertices(verticesBuffer);
  140. renderedMesh->GetTriangleIndices(triangleBuffer);
  141. exportMesh->SetGeometry(verticesBuffer, triangleBuffer);
  142. json trackpointData = json::array();
  143. for (EMFTrackPoint* point: _emfTrackPoints) {
  144. osg::Vec3 moveToMid = point->getNormal().operator*(-(point->getDepth() / 2));
  145. osg::Vec3 trackPoint = point->getTrackPoint().operator+(moveToMid);
  146. osg::Vec3 trackNormal = point->getNormal();
  147. json pointData = {
  148. {"point", {trackPoint.x(), trackPoint.y(), trackPoint.z()}},
  149. {"normal", {trackNormal.x(), trackNormal.y(), trackNormal.z()}},
  150. {"rotation", point->getNormalRotation()}
  151. };
  152. trackpointData.push_back(pointData);
  153. }
  154. Lib3MF::PMetaDataGroup emfMetaData = exportMesh->GetMetaDataGroup();
  155. emfMetaData->AddMetaData(META_NAMESPACE, "trackpoints-emftrack", trackpointData.dump(), "string", true);
  156. exportModel->AddBuildItem(exportMesh.get(), _wrapper->GetIdentityTransform());
  157. }
  158. if (settings.SteamVRTrack) {
  159. renderer->renderSteamVRTrack(_steamVrTrackPoints);
  160. Lib3MF::PModel steamVrTrackModel = _wrapper->CreateModel();
  161. Lib3MF::PReader reader = steamVrTrackModel->QueryReader("3mf");
  162. reader->ReadFromFile(std::filesystem::temp_directory_path().u8string() + fileDelimiter + "trackpointapp_render_steamvrtrack.3mf");
  163. Lib3MF::PMeshObjectIterator meshIterator = steamVrTrackModel->GetMeshObjects();
  164. if (meshIterator->Count() != 1) {
  165. return false;
  166. }
  167. meshIterator->MoveNext();
  168. Lib3MF::PMeshObject renderedMesh = meshIterator->GetCurrentMeshObject();
  169. Lib3MF::PMeshObject exportMesh = exportModel->AddMeshObject();
  170. exportMesh->SetName("steamvrtrack");
  171. std::vector<Lib3MF::sPosition> verticesBuffer;
  172. std::vector<Lib3MF::sTriangle> triangleBuffer;
  173. renderedMesh->GetVertices(verticesBuffer);
  174. renderedMesh->GetTriangleIndices(triangleBuffer);
  175. exportMesh->SetGeometry(verticesBuffer, triangleBuffer);
  176. json trackpointData = json::array();
  177. for (SteamVRTrackPoint* point: _steamVrTrackPoints) {
  178. osg::Vec3 trackPoint = point->getTrackPoint();
  179. osg::Vec3 trackNormal = point->getNormal();
  180. json pointData = {
  181. {"point", {trackPoint.x(), trackPoint.y(), trackPoint.z()}},
  182. {"normal", {trackNormal.x(), trackNormal.y(), trackNormal.z()}}
  183. };
  184. trackpointData.push_back(pointData);
  185. }
  186. Lib3MF::PMetaDataGroup steamVrMetaData = exportMesh->GetMetaDataGroup();
  187. steamVrMetaData->AddMetaData(META_NAMESPACE, "trackpoints-steamvrtrack", trackpointData.dump(), "string", true);
  188. exportModel->AddBuildItem(exportMesh.get(), _wrapper->GetIdentityTransform());
  189. }
  190. delete renderer;
  191. // Export action point metadata
  192. json actionPointData;
  193. for (ActionPoint* point: _actionPoints) {
  194. osg::Vec3 translation = point->getTranslation();
  195. osg::Vec3 normal = point->getNormal();
  196. actionPointData[point->getIdentifier()] = {
  197. {"point", {translation.x(), translation.y(), translation.z()}},
  198. {"normal", {normal.x(), normal.y(), normal.z()}}
  199. };
  200. }
  201. metaData->AddMetaData(META_NAMESPACE, "trackpoints-actionpoints", actionPointData.dump(), "string", true);
  202. Lib3MF::PWriter exportWriter = exportModel->QueryWriter("3mf");
  203. exportWriter->WriteToFile(path);
  204. return true;
  205. }
  206. bool ProjectStore::isProjectOpen() {
  207. return _projectLoaded;
  208. }
  209. void ProjectStore::closeProject() {
  210. _projectLoaded = false;
  211. reset();
  212. }
  213. bool ProjectStore::isModified() {
  214. return _projectModified;
  215. }
  216. void ProjectStore::projectModified() {
  217. _projectModified = true;
  218. }
  219. TrackPoint* ProjectStore::getTrackPointById(int id, ActiveTrackingSystem activeTrackingSystem) {
  220. switch(activeTrackingSystem) {
  221. case OptiTrack: {
  222. return _optiTrackPoints[id];
  223. };
  224. case EMFTrack: {
  225. return _emfTrackPoints[id];
  226. };
  227. case SteamVRTrack: {
  228. return _steamVrTrackPoints[id];
  229. };
  230. case ActionPoints: {
  231. return _actionPoints[id];
  232. };
  233. }
  234. }
  235. void ProjectStore::addTrackPoint(osg::Vec3 point, osg::Vec3 normal, ActiveTrackingSystem activeTrackingSystem) {
  236. switch(activeTrackingSystem) {
  237. case OptiTrack: {
  238. OptiTrackPoint* optiTrackPoint = new OptiTrackPoint(point, normal, _normalModifier, _normalRotation, _compensation, _optiTrackSettings.length, _optiTrackSettings.radius);
  239. _optiTrackPoints.push_back(optiTrackPoint);
  240. break;
  241. }
  242. case EMFTrack: {
  243. normal = normal.operator*(-1.0f);
  244. EMFTrackPoint* emfTrackPoint = new EMFTrackPoint(point, normal, _normalModifier, _normalRotation, _compensation, _emfTrackSettings.width, _emfTrackSettings.height, _emfTrackSettings.depth);
  245. _emfTrackPoints.push_back(emfTrackPoint);
  246. break;
  247. }
  248. case SteamVRTrack: {
  249. SteamVRTrackPoint* steamVrTrackPoint = new SteamVRTrackPoint(point, normal, _normalModifier, _normalRotation, _compensation, _steamVrTrackSettings.length);
  250. _steamVrTrackPoints.push_back(steamVrTrackPoint);
  251. break;
  252. }
  253. case ActionPoints: {
  254. ActionPoint* actionPoint = new ActionPoint(point, normal, _normalModifier, _normalRotation, _compensation, _actionPointSettings.identifier);
  255. _actionPoints.push_back(actionPoint);
  256. break;
  257. }
  258. }
  259. projectModified();
  260. MainWindow::getInstance()->getEditWiget()->updateTrackpointCount();
  261. }
  262. int ProjectStore::getCount(ActiveTrackingSystem activeTrackingSystem) {
  263. switch(activeTrackingSystem) {
  264. case OptiTrack: {
  265. return _optiTrackPoints.size();
  266. };
  267. case EMFTrack: {
  268. return _emfTrackPoints.size();
  269. };
  270. case SteamVRTrack: {
  271. return _steamVrTrackPoints.size();
  272. };
  273. case ActionPoints: {
  274. return _actionPoints.size();
  275. };
  276. }
  277. }
  278. void ProjectStore::removeTrackPoint(int id, ActiveTrackingSystem activeTrackingSystem) {
  279. switch(activeTrackingSystem) {
  280. case OptiTrack: {
  281. _optiTrackPoints.erase(_optiTrackPoints.begin() + id);
  282. break;
  283. }
  284. case EMFTrack: {
  285. _emfTrackPoints.erase(_emfTrackPoints.begin() + id);
  286. break;
  287. }
  288. case SteamVRTrack: {
  289. _steamVrTrackPoints.erase(_steamVrTrackPoints.begin() + id);
  290. break;
  291. }
  292. case ActionPoints: {
  293. _actionPoints.erase(_actionPoints.begin() + id);
  294. break;
  295. }
  296. }
  297. projectModified();
  298. MainWindow::getInstance()->getEditWiget()->updateTrackpointCount();
  299. }
  300. void ProjectStore::updateNormalModifier(osg::Vec3 modifier) {
  301. _normalModifier = modifier;
  302. }
  303. osg::Vec3 ProjectStore::getNormalModifier() {
  304. return _normalModifier;
  305. }
  306. void ProjectStore::updateNormalRotation(float normalRotation) {
  307. _normalRotation = normalRotation;
  308. }
  309. float ProjectStore::getNormalRotation() {
  310. return _normalRotation;
  311. }
  312. void ProjectStore::updateCompensation(bool compensation) {
  313. _compensation = compensation;
  314. }
  315. bool ProjectStore::getCompensation() {
  316. return _compensation;
  317. }
  318. std::vector<OptiTrackPoint*> ProjectStore::getOptiTrackPoints() {
  319. return _optiTrackPoints;
  320. }
  321. void ProjectStore::updateOptiTrackSettings(OptiTrackSettings optiTrackSettings) {
  322. _optiTrackSettings = optiTrackSettings;
  323. }
  324. OptiTrackSettings ProjectStore::getOptiTrackSettings() {
  325. return _optiTrackSettings;
  326. }
  327. std::vector<EMFTrackPoint*> ProjectStore::getEMFTrackPoints() {
  328. return _emfTrackPoints;
  329. }
  330. void ProjectStore::updateEMFTrackSettings(EMFTrackSettings emfTrackSettings) {
  331. _emfTrackSettings = emfTrackSettings;
  332. }
  333. EMFTrackSettings ProjectStore::getEMFTrackSettings() {
  334. return _emfTrackSettings;
  335. }
  336. std::vector<SteamVRTrackPoint*> ProjectStore::getSteamVRTrackPoints() {
  337. return _steamVrTrackPoints;
  338. }
  339. void ProjectStore::updateSteamVRTrackSettings(SteamVRTrackSettings steamVrTrackSettings) {
  340. _steamVrTrackSettings = steamVrTrackSettings;
  341. }
  342. SteamVRTrackSettings ProjectStore::getSteamVRTrackSettings() {
  343. return _steamVrTrackSettings;
  344. }
  345. std::vector<ActionPoint*> ProjectStore::getActionPoints() {
  346. return _actionPoints;
  347. }
  348. void ProjectStore::updateActionPointSettings(ActionPointSettings actionPointSettings) {
  349. _actionPointSettings = actionPointSettings;
  350. }
  351. ActionPointSettings ProjectStore::getActionPointSettings() {
  352. return _actionPointSettings;
  353. }
  354. unsigned int ProjectStore::actionPointIdentifierInUse(std::string candidate, int current) {
  355. unsigned int count = 0;
  356. int i = 0;
  357. for (ActionPoint* actionPoint: _actionPoints) {
  358. if (i != current && candidate.compare(actionPoint->getIdentifier()) == 0) {
  359. count++;
  360. }
  361. i++;
  362. }
  363. return count;
  364. }
  365. void ProjectStore::load3mfLib() {
  366. _wrapper = Lib3MF::CWrapper::loadLibrary();
  367. _project = _wrapper->CreateModel();
  368. }
  369. void ProjectStore::reset() {
  370. _project = _wrapper->CreateModel();
  371. _optiTrackPoints.clear();
  372. _steamVrTrackPoints.clear();
  373. _actionPoints.clear();
  374. _optiTrackSettings = OptiTrackSettings {OPTITRACK_DEFAULT_LENGTH, OPTITRACK_DEFAULT_RADIUS};
  375. _steamVrTrackSettings = SteamVRTrackSettings {STEAMVR_DEFAULT_LENGTH};
  376. _actionPointSettings = ActionPointSettings {ACTIONPOINT_DEFAULT_IDENFIFIER};
  377. _normalModifier = osg::Vec3(0.0f, 0.0f, 0.0f);
  378. }
  379. void ProjectStore::render3MFMesh() {
  380. // Get meshes
  381. Lib3MF::PMeshObjectIterator meshIterator = _project->GetMeshObjects();
  382. // Our use case supports just a single mesh per project
  383. if (meshIterator->Count() != 1) {
  384. MainWindow::getInstance()->showErrorMessage("The input file contains more than one mesh. Currently, we don't support this case, so the first mesh will be used.", "Input file error.");
  385. }
  386. meshIterator->MoveNext();
  387. Lib3MF::PMeshObject mesh = meshIterator->GetCurrentMeshObject();
  388. // Load vertices and triangles and render them
  389. std::vector<Lib3MF::sPosition> verticesBuffer;
  390. mesh->GetVertices(verticesBuffer);
  391. std::vector<Lib3MF::sTriangle> triangleBuffer;
  392. mesh->GetTriangleIndices(triangleBuffer);
  393. // Create osg style arrays
  394. osg::ref_ptr<osg::Vec3Array> vertices = new osg::Vec3Array;
  395. osg::ref_ptr<osg::Vec3Array> normals = new osg::Vec3Array;
  396. // Convert data to osg format and calculate vertex normals
  397. MeshTools::calculateNormals(verticesBuffer, triangleBuffer, vertices, normals);
  398. // Call renderer
  399. MainWindow* mainWindow = MainWindow::getInstance();
  400. mainWindow->getOsgWidget()->renderBaseMesh(vertices, normals);
  401. }
  402. void ProjectStore::updateMetaData() {
  403. Lib3MF::PMetaDataGroup metaData = _project->GetMetaDataGroup();
  404. try {
  405. Lib3MF::PMetaData versionInformation = metaData->GetMetaDataByKey(META_NAMESPACE, "format");
  406. } catch (Lib3MF::ELib3MFException &e) {
  407. metaData->AddMetaData(META_NAMESPACE, "format", "1.0.0", "string", true);
  408. }
  409. json optiTrackData = json::array();
  410. for (OptiTrackPoint* optiTrackPoint: _optiTrackPoints) {
  411. optiTrackData.push_back({
  412. {"point", osgVecToStdVec(optiTrackPoint->getTranslation())},
  413. {"normal", osgVecToStdVec(optiTrackPoint->getNormal())},
  414. {"normalModifier", osgVecToStdVec(optiTrackPoint->getNormalModifier())},
  415. {"normalRotation", optiTrackPoint->getNormalRotation()},
  416. {"compensation", optiTrackPoint->getCompensation()},
  417. {"length", optiTrackPoint->getLength()},
  418. {"radius", optiTrackPoint->getRadius()}
  419. });
  420. }
  421. try {
  422. Lib3MF::PMetaData optiTrackPoints = metaData->GetMetaDataByKey(META_NAMESPACE, "optitrack");
  423. optiTrackPoints->SetValue(optiTrackData.dump());
  424. } catch (Lib3MF::ELib3MFException &e) {
  425. metaData->AddMetaData(META_NAMESPACE, "optitrack", optiTrackData.dump(), "string", true);
  426. }
  427. json emfTrackData = json::array();
  428. for (EMFTrackPoint* emfTrackPoint: _emfTrackPoints) {
  429. emfTrackData.push_back({
  430. {"point", osgVecToStdVec(emfTrackPoint->getTranslation())},
  431. {"normal", osgVecToStdVec(emfTrackPoint->getNormal())},
  432. {"normalModifier", osgVecToStdVec(emfTrackPoint->getNormalModifier())},
  433. {"normalRotation", emfTrackPoint->getNormalRotation()},
  434. {"compensation", emfTrackPoint->getCompensation()},
  435. {"width", emfTrackPoint->getWidth()},
  436. {"height", emfTrackPoint->getHeight()},
  437. {"depth", emfTrackPoint->getDepth()}
  438. });
  439. }
  440. try {
  441. Lib3MF::PMetaData emfTrackPoints = metaData->GetMetaDataByKey(META_NAMESPACE, "emftrack");
  442. emfTrackPoints->SetValue(emfTrackData.dump());
  443. } catch (Lib3MF::ELib3MFException &e) {
  444. metaData->AddMetaData(META_NAMESPACE, "emftrack", emfTrackData.dump(), "string", true);
  445. }
  446. json steamVrTrackData = json::array();
  447. for (SteamVRTrackPoint* steamVrTrackPoint: _steamVrTrackPoints) {
  448. steamVrTrackData.push_back({
  449. {"point", osgVecToStdVec(steamVrTrackPoint->getTranslation())},
  450. {"normal", osgVecToStdVec(steamVrTrackPoint->getNormal())},
  451. {"normalModifier", osgVecToStdVec(steamVrTrackPoint->getNormalModifier())},
  452. {"normalRotation", steamVrTrackPoint->getNormalRotation()},
  453. {"compensation", steamVrTrackPoint->getCompensation()},
  454. {"length", steamVrTrackPoint->getLength()}
  455. });
  456. }
  457. try {
  458. Lib3MF::PMetaData steamVrTrackPoints = metaData->GetMetaDataByKey(META_NAMESPACE, "steamvrtrack");
  459. steamVrTrackPoints->SetValue(steamVrTrackData.dump());
  460. } catch (Lib3MF::ELib3MFException &e) {
  461. metaData->AddMetaData(META_NAMESPACE, "steamvrtrack", steamVrTrackData.dump(), "string", true);
  462. }
  463. json actionPointData = json::array();
  464. for (ActionPoint* actionPoint: _actionPoints) {
  465. actionPointData.push_back({
  466. {"point", osgVecToStdVec(actionPoint->getTranslation())},
  467. {"normal", osgVecToStdVec(actionPoint->getNormal())},
  468. {"normalModifier", osgVecToStdVec(actionPoint->getNormalModifier())},
  469. {"normalRotation", actionPoint->getNormalRotation()},
  470. {"compensation", actionPoint->getCompensation()},
  471. {"identifier", actionPoint->getIdentifier()}
  472. });
  473. }
  474. try {
  475. Lib3MF::PMetaData actionPoints = metaData->GetMetaDataByKey(META_NAMESPACE, "actionpoints");
  476. actionPoints->SetValue(actionPointData.dump());
  477. } catch (Lib3MF::ELib3MFException &e) {
  478. metaData->AddMetaData(META_NAMESPACE, "actionpoints", actionPointData.dump(), "string", true);
  479. }
  480. }
  481. void ProjectStore::loadMetaData() {
  482. Lib3MF::PMetaDataGroup metaData = _project->GetMetaDataGroup();
  483. try {
  484. Lib3MF::PMetaData versionInformation = metaData->GetMetaDataByKey(META_NAMESPACE, "format");
  485. } catch (Lib3MF::ELib3MFException &e) {
  486. MainWindow::getInstance()->showErrorMessage("This is not a valid TrackpointApp project.", "Error opening project file.");
  487. }
  488. Lib3MF::PMetaData optiTrackString;
  489. try {
  490. optiTrackString = metaData->GetMetaDataByKey(META_NAMESPACE, "optitrack");
  491. auto optiTrackData = json::parse(optiTrackString->GetValue());
  492. _optiTrackPoints.clear();
  493. for (const auto pointData: optiTrackData) {
  494. osg::Vec3f point = osg::Vec3f(pointData["point"][0], pointData["point"][1], pointData["point"][2]);
  495. osg::Vec3f normal = osg::Vec3f(pointData["normal"][0], pointData["normal"][1], pointData["normal"][2]);
  496. osg::Vec3f normalModifier = osg::Vec3f(pointData["normalModifier"][0], pointData["normalModifier"][1], pointData["normalModifier"][2]);
  497. OptiTrackPoint* optiTrackPoint = new OptiTrackPoint(point, normal, normalModifier, static_cast<float>(pointData["normalRotation"]), static_cast<bool>(pointData["compensation"]), static_cast<double>(pointData["length"]), static_cast<double>(pointData["radius"]));
  498. _optiTrackPoints.push_back(optiTrackPoint);
  499. }
  500. } catch (Lib3MF::ELib3MFException &e) {
  501. MainWindow::getInstance()->showErrorMessage("Unable to load OptiTrack trackpoints.", "Error opening project file.");
  502. }
  503. Lib3MF::PMetaData emfTrackString;
  504. try {
  505. emfTrackString = metaData->GetMetaDataByKey(META_NAMESPACE, "emftrack");
  506. auto emfTrackData = json::parse(emfTrackString->GetValue());
  507. _emfTrackPoints.clear();
  508. for (const auto pointData: emfTrackData) {
  509. osg::Vec3f point = osg::Vec3f(pointData["point"][0], pointData["point"][1], pointData["point"][2]);
  510. osg::Vec3f normal = osg::Vec3f(pointData["normal"][0], pointData["normal"][1], pointData["normal"][2]);
  511. osg::Vec3f normalModifier = osg::Vec3f(pointData["normalModifier"][0], pointData["normalModifier"][1], pointData["normalModifier"][2]);
  512. EMFTrackPoint* emfTrackPoint = new EMFTrackPoint(point, normal, normalModifier, static_cast<float>(pointData["normalRotation"]), static_cast<bool>(pointData["compensation"]), static_cast<double>(pointData["width"]), static_cast<double>(pointData["height"]), static_cast<double>(pointData["depth"]));
  513. _emfTrackPoints.push_back(emfTrackPoint);
  514. }
  515. } catch (Lib3MF::ELib3MFException &e) {
  516. MainWindow::getInstance()->showErrorMessage("Unable to load EMFTrack trackpoints.", "Error opening project file.");
  517. }
  518. Lib3MF::PMetaData steamVrTrackString;
  519. try {
  520. steamVrTrackString = metaData->GetMetaDataByKey(META_NAMESPACE, "steamvrtrack");
  521. auto steamVrData = json::parse(steamVrTrackString->GetValue());
  522. _steamVrTrackPoints.clear();
  523. for (const auto pointData: steamVrData) {
  524. osg::Vec3f point = osg::Vec3f(pointData["point"][0], pointData["point"][1], pointData["point"][2]);
  525. osg::Vec3f normal = osg::Vec3f(pointData["normal"][0], pointData["normal"][1], pointData["normal"][2]);
  526. osg::Vec3f normalModifier = osg::Vec3f(pointData["normalModifier"][0], pointData["normalModifier"][1], pointData["normalModifier"][2]);
  527. SteamVRTrackPoint* steamVrTrackPoint = new SteamVRTrackPoint(point, normal, normalModifier, static_cast<float>(pointData["normalRotation"]), static_cast<bool>(pointData["compensation"]), static_cast<double>(pointData["length"]));
  528. _steamVrTrackPoints.push_back(steamVrTrackPoint);
  529. }
  530. } catch (Lib3MF::ELib3MFException &e) {
  531. MainWindow::getInstance()->showErrorMessage("Unable to load SteamVR trackpoints.", "Error opening project file.");
  532. }
  533. Lib3MF::PMetaData actionPointString;
  534. try {
  535. actionPointString = metaData->GetMetaDataByKey(META_NAMESPACE, "actionpoints");
  536. auto actionPointData = json::parse(actionPointString->GetValue());
  537. _actionPoints.clear();
  538. for (const auto pointData: actionPointData) {
  539. osg::Vec3f point = osg::Vec3f(pointData["point"][0], pointData["point"][1], pointData["point"][2]);
  540. osg::Vec3f normal = osg::Vec3f(pointData["normal"][0], pointData["normal"][1], pointData["normal"][2]);
  541. osg::Vec3f normalModifier = osg::Vec3f(pointData["normalModifier"][0], pointData["normalModifier"][1], pointData["normalModifier"][2]);
  542. ActionPoint* actionPoint = new ActionPoint(point, normal, normalModifier, static_cast<float>(pointData["normalRotation"]), static_cast<bool>(pointData["compensation"]), pointData["identifier"]);
  543. _actionPoints.push_back(actionPoint);
  544. }
  545. } catch (Lib3MF::ELib3MFException &e) {
  546. MainWindow::getInstance()->showErrorMessage("Unable to load ActionPoints.", "Error opening project file.");
  547. }
  548. render3MFMesh();
  549. MainWindow::getInstance()->renderView(Edit);
  550. MainWindow::getInstance()->getOsgWidget()->getPointRenderer()->render(MainWindow::getInstance()->getEditWiget()->getSelectedTrackingSystem());
  551. MainWindow::getInstance()->getEditWiget()->updateTrackpointCount();
  552. }
  553. std::vector<float> ProjectStore::osgVecToStdVec(osg::Vec3f input) {
  554. std::vector<float> vector;
  555. vector.push_back(input.x());
  556. vector.push_back(input.y());
  557. vector.push_back(input.z());
  558. return vector;
  559. }
  560. osg::Vec3f ProjectStore::stdVecToOsgVec(std::vector<float> input) {
  561. return osg::Vec3f(input[0], input[1], input[2]);
  562. }