123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344 |
- // Include own headers
- #include "ProjectStore.hpp"
- // Include modules
- #include "MainWindow.hpp"
- #include "StringBasics.hpp"
- #include "TrackPointRenderer.hpp"
- #include "PlatformSupport.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 (StringBasics::endsWithCaseInsensitive(meshFile, ".STL")) {
- printf("Currently unsupported.\n");
- } else if (StringBasics::endsWithCaseInsensitive(meshFile, ".3MF")) {
- _projectLoaded = true;
- // 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 (!_projectLoaded) {
- Lib3MF::PReader reader = _project->QueryReader("3mf");
- reader->ReadFromFile(projectFile);
- _projectLoaded = true;
- _projectModified = false;
- loadMetaData();
- return true;
- }
- return false;
- }
- bool ProjectStore::saveProject() {
- if (_projectFile != "") {
- return saveProject(_projectFile);
- }
- return false;
- }
- bool ProjectStore::saveProject(std::string path) {
- updateMetaData();
- Lib3MF::PWriter writer = _project->QueryWriter("3mf");
- writer->WriteToFile(path);
- _projectFile = path;
- return true;
- }
- bool ProjectStore::exportProject(std::string path, ExportSettings settings) {
- 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();
- std::vector<Lib3MF::sPosition> verticesBuffer;
- std::vector<Lib3MF::sTriangle> triangleBuffer;
- renderedMesh->GetVertices(verticesBuffer);
- renderedMesh->GetTriangleIndices(triangleBuffer);
- exportMesh->SetGeometry(verticesBuffer, triangleBuffer);
- std::vector<std::vector<float>> pointsList;
- for (OptiTrackPoint* point: _optiTrackPoints) {
- std::vector<float> pointData;
- osg::Vec3 trackPoint = point->getTrackPoint();
- pointData.push_back(trackPoint.x());
- pointData.push_back(trackPoint.y());
- pointData.push_back(trackPoint.z());
- pointsList.push_back(pointData);
- }
- json trackpointData = pointsList;
- metaData->AddMetaData("tk-ar-tracking", "trackpoints-optitrack", trackpointData.dump(), "string", true);
- exportModel->AddBuildItem(exportMesh.get(), _wrapper->GetIdentityTransform());
- }
- if (settings.EMFTrack) {
- }
- if (settings.SteamVRTrack) {
- }
- delete renderer;
- Lib3MF::PWriter exportWriter = exportModel->QueryWriter("3mf");
- exportWriter->WriteToFile(path);
- return true;
- }
- TrackPoint* ProjectStore::getTrackPointById(int id, ActiveTrackingSystem activeTrackingSystem) {
- switch(activeTrackingSystem) {
- case OptiTrack: {
- return _optiTrackPoints[id];
- };
- case EMFTrack: {
- break;
- };
- case SteamVRTrack: {
- break;
- };
- case ActionPoints: {
- break;
- };
- }
- }
- 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);
- MainWindow::getInstance()->getEditWiget()->updateTrackpointCount();
- break;
- }
- case EMFTrack: {
- break;
- }
- case SteamVRTrack: {
- SteamVRTrackPoint* steamVrTrackPoint = new SteamVRTrackPoint(point, normal, _normalModifier, _steamVrTrackSettings.length);
- _steamVrTrackPoints.push_back(steamVrTrackPoint);
- MainWindow::getInstance()->getEditWiget()->updateTrackpointCount();
- break;
- }
- case ActionPoints: {
- break;
- }
- }
- }
- int ProjectStore::getCount(ActiveTrackingSystem activeTrackingSystem) {
- switch(activeTrackingSystem) {
- case OptiTrack: {
- return _optiTrackPoints.size();
- };
- case EMFTrack: {
- break;
- };
- case SteamVRTrack: {
- break;
- };
- case ActionPoints: {
- break;
- };
- }
- }
- void ProjectStore::removeTrackPoint(int id, ActiveTrackingSystem activeTrackingSystem) {
- switch(activeTrackingSystem) {
- case OptiTrack: {
- _optiTrackPoints.erase(_optiTrackPoints.begin() + id);
- break;
- }
- case EMFTrack: {
- break;
- }
- case SteamVRTrack: {
- break;
- }
- case ActionPoints: {
- break;
- }
- }
- 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<TrackPoint*> ProjectStore::getActionPoints() {
- return _actionPoints;
- }
- void ProjectStore::load3mfLib() {
- _wrapper = Lib3MF::CWrapper::loadLibrary();
- _project = _wrapper->CreateModel();
- }
- 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
- for (const Lib3MF::sTriangle triangle: triangleBuffer) {
- // Create osg vectors
- osg::Vec3 vertex[3] = {};
- for (char i = 0; i < 3; i++) {
- vertex[i] = osg::Vec3(verticesBuffer[triangle.m_Indices[i]].m_Coordinates[0], verticesBuffer[triangle.m_Indices[i]].m_Coordinates[1], verticesBuffer[triangle.m_Indices[i]].m_Coordinates[2]);
- }
- // Calculate normal
- osg::Vec3 edgeOne = vertex[1].operator-(vertex[0]);
- osg::Vec3 edgeTwo = vertex[2].operator-(vertex[0]);
- osg::Vec3 normal = edgeOne.operator^(edgeTwo);
- normal.normalize();
- // Store them
- for (int i = 0; i < 3; i++) {
- vertices->push_back(vertex[i]);
- normals->push_back(normal);
- }
- }
- // 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);
- }
- }
- 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");
- } catch (Lib3MF::ELib3MFException &e) {
- // TODO: Something is wrong with the file
- }
- 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);
- }
- 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]);
- }
|