123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511 |
- // Include own headers
- #include "ProjectStore.hpp"
- // Include modules
- #include "MainWindow.hpp"
- #include "StringBasics.hpp"
- #include "TrackPointRenderer.hpp"
- #include "PlatformSupport.hpp"
- #include "STLImport.hpp"
- #include "MeshTools.hpp"
- // Include dependencies
- #include <typeinfo>
- #include <iostream>
- #include <filesystem>
- #define META_NAMESPACE "tk-ar-tracking"
- ProjectStore::ProjectStore() {
- _projectLoaded = false;
- _projectModified = false;
- load3mfLib();
- }
- ProjectStore::~ProjectStore() {
- }
- void ProjectStore::loadMesh(std::string meshFile) {
- if (meshFile == "") {
- return;
- }
- if (StringBasics::endsWithCaseInsensitive(meshFile, ".STL")) {
- _projectLoaded = true;
- _projectModified = false;
- // Read STL file
- std::vector<Lib3MF::sPosition> verticesBuffer;
- std::vector<Lib3MF::sTriangle> triangleBuffer;
- STLImport::readSTL(meshFile, &verticesBuffer, &triangleBuffer);
- Lib3MF::PMeshObject baseMesh = _project->AddMeshObject();
- baseMesh->SetGeometry(verticesBuffer, triangleBuffer);
- render3MFMesh();
- MainWindow::getInstance()->renderView(Edit);
- } else if (StringBasics::endsWithCaseInsensitive(meshFile, ".3MF")) {
- _projectLoaded = true;
- _projectModified = false;
- // Read 3MF file
- Lib3MF::PReader reader = _project->QueryReader("3mf");
- reader->ReadFromFile(meshFile);
- render3MFMesh();
- MainWindow::getInstance()->renderView(Edit);
- } else {
- // TODO: Show error popup
- printf("Unsupported file type.\n");
- }
- }
- bool ProjectStore::loadProject(std::string projectFile) {
- if (projectFile == "") {
- return false;
- }
- if (!_projectLoaded) {
- Lib3MF::PReader reader = _project->QueryReader("3mf");
- reader->ReadFromFile(projectFile);
- _projectLoaded = true;
- _projectModified = false;
- _projectFile = projectFile;
- loadMetaData();
- return true;
- }
- return false;
- }
- bool ProjectStore::saveProject() {
- if (_projectFile != "") {
- return saveProject(_projectFile);
- }
- return false;
- }
- bool ProjectStore::saveProject(std::string path) {
- if (path == "") {
- return false;
- }
- updateMetaData();
- Lib3MF::PWriter writer = _project->QueryWriter("3mf");
- writer->WriteToFile(path);
- _projectFile = path;
- _projectModified = false;
- return true;
- }
- bool ProjectStore::exportProject(std::string path, ExportSettings settings) {
- if (path == "") {
- return false;
- }
- OpenScadRenderer* renderer = new OpenScadRenderer();
- // Base for rendering mesh
- Lib3MF::PWriter writer = _project->QueryWriter("3mf");
- writer->WriteToFile(std::filesystem::temp_directory_path().u8string() + fileDelimiter + "trackpointapp_export.3mf");
- // Export file
- Lib3MF::PModel exportModel = _wrapper->CreateModel();
- Lib3MF::PMetaDataGroup metaData = exportModel->GetMetaDataGroup();
- if (settings.OptiTrack) {
- renderer->renderOptiTrack(_optiTrackPoints);
- Lib3MF::PModel optiTrackModel = _wrapper->CreateModel();
- Lib3MF::PReader reader = optiTrackModel->QueryReader("3mf");
- reader->ReadFromFile(std::filesystem::temp_directory_path().u8string() + fileDelimiter + "trackpointapp_render_optitrack.3mf");
- Lib3MF::PMeshObjectIterator meshIterator = optiTrackModel->GetMeshObjects();
- if (meshIterator->Count() != 1) {
- return false;
- }
- meshIterator->MoveNext();
- Lib3MF::PMeshObject renderedMesh = meshIterator->GetCurrentMeshObject();
- Lib3MF::PMeshObject exportMesh = exportModel->AddMeshObject();
- exportMesh->SetName("optitrack");
- std::vector<Lib3MF::sPosition> verticesBuffer;
- std::vector<Lib3MF::sTriangle> triangleBuffer;
- renderedMesh->GetVertices(verticesBuffer);
- renderedMesh->GetTriangleIndices(triangleBuffer);
- exportMesh->SetGeometry(verticesBuffer, triangleBuffer);
- json trackpointData = json::array();
- for (OptiTrackPoint* point: _optiTrackPoints) {
- osg::Vec3 trackPoint = point->getTrackPoint();
- osg::Vec3 trackNormal = point->getNormal();
- json pointData = {
- {"point", {trackPoint.x(), trackPoint.y(), trackPoint.z()}},
- {"normal", {trackNormal.x(), trackNormal.y(), trackNormal.z()}}
- };
- trackpointData.push_back(pointData);
- }
- Lib3MF::PMetaDataGroup optiMetaData = exportMesh->GetMetaDataGroup();
- optiMetaData->AddMetaData(META_NAMESPACE, "trackpoints-optitrack", trackpointData.dump(), "string", true);
- exportModel->AddBuildItem(exportMesh.get(), _wrapper->GetIdentityTransform());
- }
- if (settings.EMFTrack) {
- }
- if (settings.SteamVRTrack) {
- renderer->renderSteamVRTrack(_steamVrTrackPoints);
- Lib3MF::PModel steamVrTrackModel = _wrapper->CreateModel();
- Lib3MF::PReader reader = steamVrTrackModel->QueryReader("3mf");
- reader->ReadFromFile(std::filesystem::temp_directory_path().u8string() + fileDelimiter + "trackpointapp_render_steamvrtrack.3mf");
- Lib3MF::PMeshObjectIterator meshIterator = steamVrTrackModel->GetMeshObjects();
- if (meshIterator->Count() != 1) {
- return false;
- }
- meshIterator->MoveNext();
- Lib3MF::PMeshObject renderedMesh = meshIterator->GetCurrentMeshObject();
- Lib3MF::PMeshObject exportMesh = exportModel->AddMeshObject();
- exportMesh->SetName("steamvrtrack");
- std::vector<Lib3MF::sPosition> verticesBuffer;
- std::vector<Lib3MF::sTriangle> triangleBuffer;
- renderedMesh->GetVertices(verticesBuffer);
- renderedMesh->GetTriangleIndices(triangleBuffer);
- exportMesh->SetGeometry(verticesBuffer, triangleBuffer);
- json trackpointData = json::array();
- for (SteamVRTrackPoint* point: _steamVrTrackPoints) {
- osg::Vec3 trackPoint = point->getTrackPoint();
- osg::Vec3 trackNormal = point->getNormal();
- json pointData = {
- {"point", {trackPoint.x(), trackPoint.y(), trackPoint.z()}},
- {"normal", {trackNormal.x(), trackNormal.y(), trackNormal.z()}}
- };
- trackpointData.push_back(pointData);
- }
- Lib3MF::PMetaDataGroup steamVrMetaData = exportMesh->GetMetaDataGroup();
- steamVrMetaData->AddMetaData(META_NAMESPACE, "trackpoints-steamvrtrack", trackpointData.dump(), "string", true);
- exportModel->AddBuildItem(exportMesh.get(), _wrapper->GetIdentityTransform());
- }
- delete renderer;
- // Export action point metadata
- json actionPointData;
- for (ActionPoint* point: _actionPoints) {
- osg::Vec3 translation = point->getTranslation();
- osg::Vec3 normal = point->getNormal();
- actionPointData[point->getIdentifier()] = {
- {"point", {translation.x(), translation.y(), translation.z()}},
- {"normal", {normal.x(), normal.y(), normal.z()}}
- };
- }
- metaData->AddMetaData(META_NAMESPACE, "trackpoints-actionpoints", actionPointData.dump(), "string", true);
- Lib3MF::PWriter exportWriter = exportModel->QueryWriter("3mf");
- exportWriter->WriteToFile(path);
- return true;
- }
- bool ProjectStore::isProjectOpen() {
- return _projectLoaded;
- }
- void ProjectStore::closeProject() {
- _projectLoaded = false;
- reset();
- }
- bool ProjectStore::isModified() {
- return _projectModified;
- }
- void ProjectStore::projectModified() {
- _projectModified = true;
- }
- TrackPoint* ProjectStore::getTrackPointById(int id, ActiveTrackingSystem activeTrackingSystem) {
- switch(activeTrackingSystem) {
- case OptiTrack: {
- return _optiTrackPoints[id];
- };
- case EMFTrack: {
- break;
- };
- case SteamVRTrack: {
- return _steamVrTrackPoints[id];
- };
- case ActionPoints: {
- return _actionPoints[id];
- };
- }
- }
- void ProjectStore::addTrackPoint(osg::Vec3 point, osg::Vec3 normal, ActiveTrackingSystem activeTrackingSystem) {
- switch(activeTrackingSystem) {
- case OptiTrack: {
- OptiTrackPoint* optiTrackPoint = new OptiTrackPoint(point, normal, _normalModifier, _optiTrackSettings.length, _optiTrackSettings.radius);
- _optiTrackPoints.push_back(optiTrackPoint);
- break;
- }
- case EMFTrack: {
- break;
- }
- case SteamVRTrack: {
- SteamVRTrackPoint* steamVrTrackPoint = new SteamVRTrackPoint(point, normal, _normalModifier, _steamVrTrackSettings.length);
- _steamVrTrackPoints.push_back(steamVrTrackPoint);
- break;
- }
- case ActionPoints: {
- ActionPoint* actionPoint = new ActionPoint(point, normal, _normalModifier, _actionPointSettings.identifier);
- _actionPoints.push_back(actionPoint);
- break;
- }
- }
- projectModified();
- MainWindow::getInstance()->getEditWiget()->updateTrackpointCount();
- }
- int ProjectStore::getCount(ActiveTrackingSystem activeTrackingSystem) {
- switch(activeTrackingSystem) {
- case OptiTrack: {
- return _optiTrackPoints.size();
- };
- case EMFTrack: {
- break;
- };
- case SteamVRTrack: {
- return _steamVrTrackPoints.size();
- };
- case ActionPoints: {
- return _actionPoints.size();
- };
- }
- }
- void ProjectStore::removeTrackPoint(int id, ActiveTrackingSystem activeTrackingSystem) {
- switch(activeTrackingSystem) {
- case OptiTrack: {
- _optiTrackPoints.erase(_optiTrackPoints.begin() + id);
- break;
- }
- case EMFTrack: {
- break;
- }
- case SteamVRTrack: {
- _steamVrTrackPoints.erase(_steamVrTrackPoints.begin() + id);
- break;
- }
- case ActionPoints: {
- _actionPoints.erase(_actionPoints.begin() + id);
- break;
- }
- }
- projectModified();
- MainWindow::getInstance()->getEditWiget()->updateTrackpointCount();
- }
- void ProjectStore::updateNormalModifier(osg::Vec3 modifier) {
- _normalModifier = modifier;
- }
- osg::Vec3 ProjectStore::getNormalModifier() {
- return _normalModifier;
- }
- std::vector<OptiTrackPoint*> ProjectStore::getOptiTrackPoints() {
- return _optiTrackPoints;
- }
- void ProjectStore::updateOptiTrackSettings(OptiTrackSettings optiTrackSettings) {
- _optiTrackSettings = optiTrackSettings;
- }
- OptiTrackSettings ProjectStore::getOptiTrackSettings() {
- return _optiTrackSettings;
- }
- std::vector<SteamVRTrackPoint*> ProjectStore::getSteamVRTrackPoints() {
- return _steamVrTrackPoints;
- }
- void ProjectStore::updateSteamVRTrackSettings(SteamVRTrackSettings steamVrTrackSettings) {
- _steamVrTrackSettings = steamVrTrackSettings;
- }
- SteamVRTrackSettings ProjectStore::getSteamVRTrackSettings() {
- return _steamVrTrackSettings;
- }
- std::vector<ActionPoint*> ProjectStore::getActionPoints() {
- return _actionPoints;
- }
- void ProjectStore::updateActionPointSettings(ActionPointSettings actionPointSettings) {
- _actionPointSettings = actionPointSettings;
- }
- ActionPointSettings ProjectStore::getActionPointSettings() {
- return _actionPointSettings;
- }
- unsigned int ProjectStore::actionPointIdentifierInUse(std::string candidate, int current) {
- unsigned int count = 0;
- int i = 0;
- for (ActionPoint* actionPoint: _actionPoints) {
- if (i != current && candidate.compare(actionPoint->getIdentifier()) == 0) {
- count++;
- }
- i++;
- }
- return count;
- }
- void ProjectStore::load3mfLib() {
- _wrapper = Lib3MF::CWrapper::loadLibrary();
- _project = _wrapper->CreateModel();
- }
- void ProjectStore::reset() {
- _project = _wrapper->CreateModel();
- _optiTrackPoints.clear();
- _steamVrTrackPoints.clear();
- _actionPoints.clear();
- _optiTrackSettings = OptiTrackSettings {OPTITRACK_DEFAULT_LENGTH, OPTITRACK_DEFAULT_RADIUS};
- _steamVrTrackSettings = SteamVRTrackSettings {STEAMVR_DEFAULT_LENGTH};
- _actionPointSettings = ActionPointSettings {ACTIONPOINT_DEFAULT_IDENFIFIER};
- _normalModifier = osg::Vec3(0.0f, 0.0f, 0.0f);
- }
- void ProjectStore::render3MFMesh() {
- // Get meshes
- Lib3MF::PMeshObjectIterator meshIterator = _project->GetMeshObjects();
- // Our use case supports just a single mesh per project
- if (meshIterator->Count() != 1) {
- // TODO: Show error popup
- printf("Not 1 mesh: %llu\n", meshIterator->Count());
- return;
- }
- meshIterator->MoveNext();
- Lib3MF::PMeshObject mesh = meshIterator->GetCurrentMeshObject();
- // Load vertices and triangles and render them
- std::vector<Lib3MF::sPosition> verticesBuffer;
- mesh->GetVertices(verticesBuffer);
- std::vector<Lib3MF::sTriangle> triangleBuffer;
- mesh->GetTriangleIndices(triangleBuffer);
- // Create osg style arrays
- osg::ref_ptr<osg::Vec3Array> vertices = new osg::Vec3Array;
- osg::ref_ptr<osg::Vec3Array> normals = new osg::Vec3Array;
- // Convert data to osg format and calculate vertex normals
- MeshTools::calculateNormals(verticesBuffer, triangleBuffer, vertices, normals);
- // Call renderer
- MainWindow* mainWindow = MainWindow::getInstance();
- mainWindow->getOsgWidget()->renderBaseMesh(vertices, normals);
- }
- void ProjectStore::updateMetaData() {
- Lib3MF::PMetaDataGroup metaData = _project->GetMetaDataGroup();
- try {
- Lib3MF::PMetaData versionInformation = metaData->GetMetaDataByKey(META_NAMESPACE, "format");
- } catch (Lib3MF::ELib3MFException &e) {
- metaData->AddMetaData(META_NAMESPACE, "format", "1.0.0", "string", true);
- }
- json optiTrackData = json::array();
- for (OptiTrackPoint* optiTrackPoint: _optiTrackPoints) {
- optiTrackData.push_back({
- {"point", osgVecToStdVec(optiTrackPoint->getTranslation())},
- {"normal", osgVecToStdVec(optiTrackPoint->getNormal())},
- {"normalModifier", osgVecToStdVec(optiTrackPoint->getNormalModifier())},
- {"length", optiTrackPoint->getLength()},
- {"radius", optiTrackPoint->getRadius()}
- });
- }
- try {
- Lib3MF::PMetaData optiTrackPoints = metaData->GetMetaDataByKey(META_NAMESPACE, "optitrack");
- optiTrackPoints->SetValue(optiTrackData.dump());
- } catch (Lib3MF::ELib3MFException &e) {
- metaData->AddMetaData(META_NAMESPACE, "optitrack", optiTrackData.dump(), "string", true);
- }
- json steamVrTrackData = json::array();
- for (SteamVRTrackPoint* steamVrTrackPoint: _steamVrTrackPoints) {
- steamVrTrackData.push_back({
- {"point", osgVecToStdVec(steamVrTrackPoint->getTranslation())},
- {"normal", osgVecToStdVec(steamVrTrackPoint->getNormal())},
- {"normalModifier", osgVecToStdVec(steamVrTrackPoint->getNormalModifier())},
- {"length", steamVrTrackPoint->getLength()}
- });
- }
- try {
- Lib3MF::PMetaData steamVrTrackPoints = metaData->GetMetaDataByKey(META_NAMESPACE, "steamvrtrack");
- steamVrTrackPoints->SetValue(steamVrTrackData.dump());
- } catch (Lib3MF::ELib3MFException &e) {
- metaData->AddMetaData(META_NAMESPACE, "steamvrtrack", steamVrTrackData.dump(), "string", true);
- }
- json actionPointData = json::array();
- for (ActionPoint* actionPoint: _actionPoints) {
- actionPointData.push_back({
- {"point", osgVecToStdVec(actionPoint->getTranslation())},
- {"normal", osgVecToStdVec(actionPoint->getNormal())},
- {"normalModifier", osgVecToStdVec(actionPoint->getNormalModifier())},
- {"identifier", actionPoint->getIdentifier()}
- });
- }
- try {
- Lib3MF::PMetaData actionPoints = metaData->GetMetaDataByKey(META_NAMESPACE, "actionpoints");
- actionPoints->SetValue(actionPointData.dump());
- } catch (Lib3MF::ELib3MFException &e) {
- metaData->AddMetaData(META_NAMESPACE, "actionpoints", actionPointData.dump(), "string", true);
- }
- }
- void ProjectStore::loadMetaData() {
- Lib3MF::PMetaDataGroup metaData = _project->GetMetaDataGroup();
- try {
- Lib3MF::PMetaData versionInformation = metaData->GetMetaDataByKey(META_NAMESPACE, "format");
- } catch (Lib3MF::ELib3MFException &e) {
- // TODO: Alert not a TrackpointApp poject
- }
- Lib3MF::PMetaData optiTrackString;
- try {
- optiTrackString = metaData->GetMetaDataByKey(META_NAMESPACE, "optitrack");
- auto optiTrackData = json::parse(optiTrackString->GetValue());
- _optiTrackPoints.clear();
- for (const auto pointData: optiTrackData) {
- osg::Vec3f point = osg::Vec3f(pointData["point"][0], pointData["point"][1], pointData["point"][2]);
- osg::Vec3f normal = osg::Vec3f(pointData["normal"][0], pointData["normal"][1], pointData["normal"][2]);
- osg::Vec3f normalModifier = osg::Vec3f(pointData["normalModifier"][0], pointData["normalModifier"][1], pointData["normalModifier"][2]);
- OptiTrackPoint* optiTrackPoint = new OptiTrackPoint(point, normal, normalModifier, static_cast<double>(pointData["length"]), static_cast<double>(pointData["radius"]));
- _optiTrackPoints.push_back(optiTrackPoint);
- }
- } catch (Lib3MF::ELib3MFException &e) {
- // TODO: Something is wrong with the file
- }
- Lib3MF::PMetaData steamVrTrackString;
- try {
- steamVrTrackString = metaData->GetMetaDataByKey(META_NAMESPACE, "steamvrtrack");
- auto steamVrData = json::parse(steamVrTrackString->GetValue());
- _steamVrTrackPoints.clear();
- for (const auto pointData: steamVrData) {
- osg::Vec3f point = osg::Vec3f(pointData["point"][0], pointData["point"][1], pointData["point"][2]);
- osg::Vec3f normal = osg::Vec3f(pointData["normal"][0], pointData["normal"][1], pointData["normal"][2]);
- osg::Vec3f normalModifier = osg::Vec3f(pointData["normalModifier"][0], pointData["normalModifier"][1], pointData["normalModifier"][2]);
- SteamVRTrackPoint* steamVrTrackPoint = new SteamVRTrackPoint(point, normal, normalModifier, static_cast<double>(pointData["length"]));
- _steamVrTrackPoints.push_back(steamVrTrackPoint);
- }
- } catch (Lib3MF::ELib3MFException &e) {
- // TODO: Something is wrong with the file
- }
- Lib3MF::PMetaData actionPointString;
- try {
- actionPointString = metaData->GetMetaDataByKey(META_NAMESPACE, "actionpoints");
- auto actionPointData = json::parse(actionPointString->GetValue());
- _actionPoints.clear();
- for (const auto pointData: actionPointData) {
- osg::Vec3f point = osg::Vec3f(pointData["point"][0], pointData["point"][1], pointData["point"][2]);
- osg::Vec3f normal = osg::Vec3f(pointData["normal"][0], pointData["normal"][1], pointData["normal"][2]);
- osg::Vec3f normalModifier = osg::Vec3f(pointData["normalModifier"][0], pointData["normalModifier"][1], pointData["normalModifier"][2]);
- ActionPoint* actionPoint = new ActionPoint(point, normal, normalModifier, pointData["identifier"]);
- _actionPoints.push_back(actionPoint);
- }
- } catch (Lib3MF::ELib3MFException &e) {
- // TODO: Something is wrong with the file
- }
- render3MFMesh();
- MainWindow::getInstance()->renderView(Edit);
- MainWindow::getInstance()->getOsgWidget()->getPointRenderer()->render(MainWindow::getInstance()->getEditWiget()->getSelectedTrackingSystem());
- MainWindow::getInstance()->getEditWiget()->updateTrackpointCount();
- }
- std::vector<float> ProjectStore::osgVecToStdVec(osg::Vec3f input) {
- std::vector<float> vector;
- vector.push_back(input.x());
- vector.push_back(input.y());
- vector.push_back(input.z());
- return vector;
- }
- osg::Vec3f ProjectStore::stdVecToOsgVec(std::vector<float> input) {
- return osg::Vec3f(input[0], input[1], input[2]);
- }
|