ShapePrimitive.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstddef>
 
#include <opencv2/core.hpp>
#include <vtkCellArray.h>
#include <vtkCellData.h>
#include <vtkDoubleArray.h>
#include <vtkPointData.h>
#include <vtkPoints.h>
#include <vtkPolyData.h>
#include <vtkPolyDataReader.h>
#include <vtkSmartPointer.h>
 
#include "vc/core/types/ITKMesh.hpp"
#include "vc/core/types/OrderedPointSet.hpp"
#include "vc/core/types/PointSet.hpp"
#include "vc/core/types/SimpleMesh.hpp"
 
namespace volcart::shapes
{
class ShapePrimitive
{
public:
    ITKMesh::Pointer itkMesh();
 
    vtkSmartPointer<vtkPolyData> vtkMesh();
    PointSet<cv::Vec3d> unorderedPoints();
 
    PointSet<cv::Vec6d> unorderedPointNormal();
    std::vector<SimpleMesh::Vertex> getPoints() const { return points_; }
 
    std::vector<SimpleMesh::Cell> getCells() const { return cells_; }
    OrderedPointSet<cv::Vec3d> orderedPoints();
 
    OrderedPointSet<cv::Vec6d> orderedPointNormal();
 
    bool isOrdered() const { return ordered_; }
 
    std::size_t orderedWidth() const { return orderedWidth_; }
 
    std::size_t orderedHeight() const { return orderedHeight_; }
protected:
    ShapePrimitive();
 
    std::vector<SimpleMesh::Vertex> points_;
    std::vector<SimpleMesh::Cell> cells_;
 
    void addVertex_(double x, double y, double z);
 
    void addCell_(int v1, int v2, int v3);
 
    void updateNormal_(int vertID, double nx, double ny, double nz);
 
    bool ordered_{false};
    std::size_t orderedWidth_{0};
    std::size_t orderedHeight_{0};
};
}  // namespace volcart::shapes