ProjectStore.cpp 28 KB

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