STLReader.h

Go to the documentation of this file.

// SPDX-License-Identifier: MIT
// Copyright (c) 2021-2025 MPM-Geomechanics Development Team
 
#ifndef STL_READER_H
#define STL_READER_H
 
#include <Eigen/Eigenvalues>
using namespace Eigen;
 
#include <string>
#include <vector>
 
struct Triangle {
    
    Vector3d normal; 
    Vector3d v1, v2, v3; 
 
    Vector3d getVertex1() const {
        return v1;
    }
 
    const Vector3d& getVertex2() const {
        return v2;
    }
 
    const Vector3d& getVertex3() const {
        return v3;
    }
 
    const Vector3d& getNormal() const {
        return normal;
    }
 
    Vector3d getCentroid() const {
        return (v1 + v2 + v3) / 3.0f;
    }
 
    double getArea() const {
        return 0.5f * ((v2 - v1).cross(v3 - v1)).norm();
    }
};
 
class STLReader {
 
public:
    bool read(const std::string& filename);
    
    const std::vector<Triangle>& getTriangles() const;
 
    void removeTrianglesOutsideLimits(const Vector3d& min, const Vector3d& max);
 
    void recalculateNormals();
 
    bool writeSTL(const std::string& output_filename) const;
    
    inline void STLResultsFlagSet(bool flag) { write_stl_results = flag; }
 
    inline bool STLResultsFlagGet() { return write_stl_results; }
 
    private:
    
    std::vector<Triangle> triangles;
 
    bool readASCII(std::ifstream& file);
    
    bool readBinary(std::ifstream& file);
 
    bool write_stl_results = false;
};
 
#endif