Added operators for dim4

shehzan10 · shehzan10 · commit 133999607fd4 · 2015-03-12T16:51:22.000-04:00
diff --git a/include/af/dim4.hpp b/include/af/dim4.hpp
@@ -52,6 +52,10 @@ class AFAPI dim4
     const   dim_type* get() const   { return dims; }
 };
 
+dim4 operator+(const dim4& first, const dim4& second);
+dim4 operator-(const dim4& first, const dim4& second);
+dim4 operator*(const dim4& first, const dim4& second);
+
 static inline
 std::ostream&
 operator<<(std::ostream& ostr, const dim4& dims)
diff --git a/src/api/c/dim4.cpp b/src/api/c/dim4.cpp
@@ -142,6 +142,34 @@ dim4::operator-=(const dim4 &other)
     return *this;
 }
 
+dim4 operator+(const dim4& first, const dim4& second)
+{
+    dim4 dims;
+    for(unsigned i = 0; i < 4; i++) {
+        dims[i] = first[i] + second[i];
+    }
+    return dims;
+}
+
+dim4 operator-(const dim4& first, const dim4& second)
+{
+    dim4 dims;
+    for(unsigned i = 0; i < 4; i++) {
+        dims[i] = first[i] - second[i];
+    }
+    return dims;
+}
+
+dim4 operator*(const dim4& first, const dim4& second)
+{
+    dim4 dims;
+    for(unsigned i = 0; i < 4; i++) {
+        dims[i] = first[i] * second[i];
+    }
+    return dims;
+}
+
+
 bool
 isEnd(const af_seq &seq)    { return (seq.end <= -1); }
 
