Add copyMakeBorder

VisionLabs · Apr 14, 2018 · b4e516a · b4e516a
1 parent 9763873
commit b4e516a
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 0 deletions.
diff --git a/cv/core/init.lua b/cv/core/init.lua
@@ -8,6 +8,10 @@ ffi.cdef[[
 int getNumThreads();
 
 void setNumThreads(int nthreads);
+
+struct TensorWrapper copyMakeBorder(struct TensorWrapper src, struct TensorWrapper dst, int top, 
+                                    int bottom, int left, int right, int borderType,
+                                    struct ScalarWrapper value);
 ]]
 
 local C = ffi.load(cv.libPath('core'))
@@ -25,4 +29,18 @@ function cv.setNumThreads(t)
     return C.setNumThreads(cv.argcheck(t, argRules))
 end
 
+function cv.copyMakeBorder(t)
+	local argRules = {
+		{"src", required = true, operator = cv.wrap_tensor},
+		{"dst", default = nil, operator = cv.wrap_tensor},
+		{"top", required = true},
+		{"bottom", required = true},
+		{"left", required = true},
+		{"right", required = true},
+		{"borderType", required = true},
+		{"value", default = {0,0,0,0} , operator = cv.Scalar}
+	}
+	return cv.unwrap_tensors(C.copyMakeBorder(cv.argcheck(t, argRules)))
+end
+
 return cv
diff --git a/include/core.hpp b/include/core.hpp
@@ -7,4 +7,7 @@ int getNumThreads();
 
 void setNumThreads(int nthreads);
 
+struct TensorWrapper copyMakeBorder(struct TensorWrapper src, struct TensorWrapper dst, int top, 
+                                    int bottom, int left, int right, int borderType,
+                                    struct ScalarWrapper value);
 }
diff --git a/src/core.cpp b/src/core.cpp
@@ -13,4 +13,13 @@ void setNumThreads(int nthreads)
     cv::setNumThreads(nthreads);
 }
 
+struct TensorWrapper copyMakeBorder(struct TensorWrapper src, struct TensorWrapper dst, int top, 
+                                    int bottom, int left, int right, int borderType,
+                                    struct ScalarWrapper value)
+{
+    MatT dstMat = dst.toMatT();
+    cv::copyMakeBorder(src.toMat(), dstMat, top, bottom, left, right, borderType, value);
+    return TensorWrapper(dstMat);
+}
+
 } // extern "C"