add gateway method

clEsperanto · Nov 25, 2020 · 8914e65 · 8914e65
1 parent 9c27287
commit 8914e65
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/include/CLE.h b/include/CLE.h
@@ -49,6 +49,8 @@ class CLE
     void AddImageAndScalar(Buffer&, Buffer&, float);
     void AddImagesWeighted(Buffer&, Buffer&, Buffer&, float, float);
     void AddImages(Buffer&, Buffer&, Buffer&);
+    void DilateSphere(Buffer&, Buffer&);
+    void ErodeSphere(Buffer&, Buffer&);
     void GreaterThanConstant(Buffer&, Buffer&, float);
     void SmallerOrEqualConstant(Buffer&, Buffer&, float);
     void MaximumZProjection(Buffer&, Buffer&);

diff --git a/src/CLE.cpp b/src/CLE.cpp
@@ -50,6 +50,22 @@ void CLE::AddImages(Buffer& src1, Buffer& src2, Buffer& dst)
     this->AddImagesWeighted(src1, src2, dst, 1, 1);
 }
 
+void CLE::DilateSphere(Buffer& src, Buffer& dst)
+{
+    DilateSphereKernel kernel(this->gpu);
+    kernel.SetInput(src);
+    kernel.SetOutput(dst);
+    kernel.Execute();
+}
+
+void CLE::ErodeSphere(Buffer& src, Buffer& dst)
+{
+    ErodeSphereKernel kernel(this->gpu);
+    kernel.SetInput(src);
+    kernel.SetOutput(dst);
+    kernel.Execute();
+}
+
 void CLE::GreaterThanConstant(Buffer& src, Buffer& dst, float scalar)
 {
     GreaterConstantKernel kernel(this->gpu);