ProjectStore.cpp 24 KB

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