123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144 |
- // Include own headers
- #include "ProjectStore.hpp"
- // Include modules
- #include "MainWindow.hpp"
- #include "StringBasics.hpp"
- // Include dependencies
- #include <typeinfo>
- #include <iostream>
- ProjectStore::ProjectStore() {
- projectLoaded = 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);
- // 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);
- render3MFMesh(verticesBuffer, triangleBuffer);
- MainWindow* mainWindow = MainWindow::getInstance();
- mainWindow->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;
- return true;
- }
- return false;
- }
- bool ProjectStore::saveProject() {
- if (_projectFile != "") {
- Lib3MF::PWriter writer = _project->QueryWriter("3mf");
- writer->WriteToFile(_projectFile);
- return true;
- }
- return false;
- }
- bool ProjectStore::saveProject(std::string path) {
- Lib3MF::PWriter writer = _project->QueryWriter("3mf");
- writer->WriteToFile(path);
- return true;
- }
- bool ProjectStore::exportProject(std::string path, ExportSettings settings) {
- return false; // TODO
- }
- void ProjectStore::updateNormalModifier(osg::Vec3 modifier) {
- _normalModifier = modifier;
- }
- osg::Vec3 ProjectStore::getNormalModifier() {
- return _normalModifier;
- }
- std::vector<OptiTrackPoint*> ProjectStore::getOptiTrackPoints() {
- return _optiTrackPoints;
- }
- void ProjectStore::addOptiTrackPoint(osg::Vec3 point, osg::Vec3 normal) {
- OptiTrackPoint* optiTrackPoint = new OptiTrackPoint(point, normal, _normalModifier, _optiTrackSettings.length, _optiTrackSettings.radius);
- _optiTrackPoints.push_back(optiTrackPoint);
- }
- void ProjectStore::removeOptiTrackPoint() {
- // TODO
- }
- void ProjectStore::updateOptiTrackSettings(OptiTrackSettings optiTrackSettings) {
- _optiTrackSettings = optiTrackSettings;
- }
- OptiTrackSettings ProjectStore::getOptiTrackSettings() {
- return _optiTrackSettings;
- }
- void ProjectStore::load3mfLib() {
- _wrapper = Lib3MF::CWrapper::loadLibrary();
- _project = _wrapper->CreateModel();
- }
- void ProjectStore::render3MFMesh(const std::vector<Lib3MF::sPosition> verticesBuffer, const std::vector<Lib3MF::sTriangle> 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::exportMetaData() {
- }
|