Intorduced PixelDataDim structure

Author: krzysg

src/data_structures/Mesh/PixelData.hpp

@@ -29,6 +29,39 @@
 #include "misc/CudaMemory.cuh"
 #endif
 
+struct PixelDataDim {
+    size_t y;
+    size_t x;
+    size_t z;
+
+    PixelDataDim(size_t y, size_t x, size_t z) : y(y), x(x), z(z) {}
+
+    size_t size() const { return y * x * z; }
+
+    PixelDataDim operator+(const PixelDataDim &rhs) const { return {y + rhs.y, x + rhs.x, z + rhs.z}; }
+    PixelDataDim operator-(const PixelDataDim &rhs) const { return {y - rhs.y, x - rhs.x, z - rhs.z}; }
+
+    template<typename INTEGER>
+    constexpr typename std::enable_if<std::is_integral<INTEGER>::value, PixelDataDim>::type
+    operator+(INTEGER i) const { return {y + i, x + i, z + i}; }
+
+    template<typename INTEGER>
+    constexpr typename std::enable_if<std::is_integral<INTEGER>::value, PixelDataDim>::type
+    operator-(INTEGER i) const { return *this + (-i); }
+
+    friend bool operator==(const PixelDataDim& lhs, const PixelDataDim& rhs) {
+        return (lhs.x == rhs.x) && (lhs.y == rhs.y) && (lhs.z == rhs.z);
+    }
+    friend bool operator!=(const PixelDataDim& lhs, const PixelDataDim& rhs) {
+        return !(lhs == rhs);
+    }
+
+    friend std::ostream &operator<<(std::ostream &os, const PixelDataDim &obj) {
+        os << "{" << obj.y << ", " << obj.x << ", " << obj.z << "}";
+        return os;
+    }
+};
+
 template <typename T>
 class ArrayWrapper
 {
@@ -438,6 +471,13 @@ public :
         if (aShouldCopyData) std::copy(aMesh.mesh.begin(), aMesh.mesh.end(), mesh.begin());
     }
 
+    /**
+     * Returns dimensions of PixelData
+     */
+    PixelDataDim getDimension() const {
+        return {static_cast<size_t>(y_num), static_cast<size_t>(x_num), static_cast<size_t>(z_num)};
+    }
+
     /**
      * Creates copy of this mesh converting each element to new type
      * @tparam U new type of mesh

test/MeshDataTest.cpp

@@ -24,6 +24,64 @@ namespace {
         VectorData<MESH_TYPE> m;
     };
 
+    TEST(MeshDataSimpleTest, PixelDataDimTest) {
+        // size provided
+        {
+            PixelData<int> md(10, 20, 30);
+            auto d = md.getDimension();
+
+            ASSERT_EQ(d.y, 10);
+            ASSERT_EQ(d.x, 20);
+            ASSERT_EQ(d.z, 30);
+            ASSERT_EQ(d.size(), 10*20*30);
+        }
+        { // adding int to all dims
+
+            PixelDataDim x = {1,2,3};
+            auto d = x + 1;
+            ASSERT_EQ(d.y, 2);
+            ASSERT_EQ(d.x, 3);
+            ASSERT_EQ(d.z, 4);
+        }
+        { // subtract int from all dims
+
+            const PixelDataDim x = {1,2,3};
+            auto d = x - 1;
+            ASSERT_EQ(d.y, 0);
+            ASSERT_EQ(d.x, 1);
+            ASSERT_EQ(d.z, 2);
+        }
+        { // adding another PixelDataDim
+
+            PixelDataDim x = {1,2,3};
+            const PixelDataDim y = {5, 10, 15};
+            auto d = x + y;
+            ASSERT_EQ(d.y, 6);
+            ASSERT_EQ(d.x, 12);
+            ASSERT_EQ(d.z, 18);
+        }
+        { // subtract another PixelDataDim
+
+            const PixelDataDim x = {5, 10, 15};
+            PixelDataDim y = {1, 2, 3};
+            auto d = x - y;
+            ASSERT_EQ(d.y, 4);
+            ASSERT_EQ(d.x, 8);
+            ASSERT_EQ(d.z, 12);
+        }
+        { // compare two PixelDataDim structures
+            const PixelDataDim x = {2, 3, 5};
+            const PixelDataDim y = {2, 3, 5};
+            const PixelDataDim z = {3, 4, 5};
+
+            ASSERT_TRUE(x == y);
+            ASSERT_FALSE(x != y);
+
+            ASSERT_FALSE(x == z);
+            ASSERT_TRUE(x != z);
+        }
+    }
+
     TEST_F(VectorDataTest, InitTest) {
         // Check initialize and resize and size are working correctly