model.h

Go to the documentation of this file.

#ifndef MODEL_H
#define MODEL_H

#include <string>

#include "vector3d.h"
#include "cell.h"
#include "material.h"

class Model
{
  private:
    std::string filename;

    bool isSTL;

    std::vector<Vector3D> vertices;

    std::vector<Material> materials;

    std::vector<Cell> cells;

    // Parsing functions

    void parseVertex(std::string line);

    void parseMaterial(std::string line);

    void parseCell(std::string line);

    // Misc functions
    std::vector<std::string> splitString(std::string line);

    bool isExtension(const std::string &str, const std::string &suffix);

  public:
    Model() = default;
    ~Model() = default;
    // Loads model from file
    Model(std::string filename);

    // Accessors

    std::string getFilename();

    bool getIsSTL();

    std::vector<Material> getMaterials();

    std::vector<Vector3D> getVertices();

    std::vector<Cell> getCells();

    int getMaterialCount();

    int getVertexCount();

    int getCellCount();

    std::string getCellList();

    Vector3D getCentre();

    // Misc functions

    void copyToFile(std::string filename);

    void saveToFile(std::string filename);
};

#endif /* MODEL_H */