code refactoring

1 files changed, 46 insertions, 0 deletions

diff --git a/src/o3d/obj3d.cpp b/src/o3d/obj3d.cpp
new file mode 100644
index 0000000..46f4db6
--- /dev/null
+++ b/src/o3d/obj3d.cpp
@@ -0,0 +1,46 @@
+#include "o3d/obj3d.h"
+#include <vector>
+#include <array>
+#include "o3d/vertex.h"
+#include "o3d/tri_vertex.h"
+
+Object3D::TriangleVertex3Iterator::TriangleVertex3Iterator(const Object3D* obj, int face_ind) : obj{obj}, face_ind{face_ind} {
+}
+
+Object3D::TriangleVertex3Iterator& Object3D::TriangleVertex3Iterator::operator++() {
+    face_ind++;
+    return *this;
+}
+
+Object3D::TriangleVertex3Iterator Object3D::TriangleVertex3Iterator::operator++(int) {
+    TriangleVertex3Iterator retval = *this;
+    ++(*this);
+    return retval;
+}
+
+bool Object3D::TriangleVertex3Iterator::operator==(TriangleVertex3Iterator other) const {
+    return obj == other.obj && face_ind == other.face_ind;
+}
+
+bool Object3D::TriangleVertex3Iterator::operator!=(TriangleVertex3Iterator other) const {
+    return !(*this == other);
+}
+
+Object3D::TriangleVertex3Iterator::reference Object3D::TriangleVertex3Iterator::operator*() const {
+    return {
+        obj->pts[obj->faces[face_ind][0]],
+        obj->pts[obj->faces[face_ind][1]],
+        obj->pts[obj->faces[face_ind][2]]
+    };
+}
+
+Object3D::Object3D(std::vector<Vertex3> pts, std::vector<std::array<int, 3>> faces) : pts{pts}, faces{faces} {
+}
+
+Object3D::TriangleVertex3Iterator Object3D::begin() {
+    return Object3D::TriangleVertex3Iterator{this};
+}
+
+Object3D::TriangleVertex3Iterator Object3D::end() {
+    return Object3D::TriangleVertex3Iterator{this, static_cast<int>(faces.size())};
+}