[arrayfire] 241/408: Added C API for SIFT

[Ghislain Vaillant]

This is an automated email from the git hooks/post-receive script.

ghisvail-guest pushed a commit to branch debian/sid
in repository arrayfire.

commit 60546a18252fe423d22441d9eacc442cefd0dc3e
Author: Peter Andreas Entschev <peter at arrayfire.com>
Date:   Thu Aug 13 10:26:40 2015 -0400

    Added C API for SIFT
---
 src/api/c/sift.cpp | 90 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 90 insertions(+)

diff --git a/src/api/c/sift.cpp b/src/api/c/sift.cpp
new file mode 100644
index 0000000..352baf1
--- /dev/null
+++ b/src/api/c/sift.cpp
@@ -0,0 +1,90 @@
+/*******************************************************
+ * Copyright (c) 2015, ArrayFire
+ * All rights reserved.
+ *
+ * This file is distributed under 3-clause BSD license.
+ * The complete license agreement can be obtained at:
+ * http://arrayfire.com/licenses/BSD-3-Clause
+ ********************************************************/
+
+#include <af/dim4.hpp>
+#include <af/defines.h>
+#include <af/features.h>
+#include <af/vision.h>
+#include <handle.hpp>
+#include <err_common.hpp>
+#include <backend.hpp>
+#include <features.hpp>
+#include <sift.hpp>
+
+using af::dim4;
+using namespace detail;
+
+template<typename T, typename convAccT>
+static void sift(af_features& feat_, af_array& descriptors, const af_array& in, const unsigned n_layers,
+                 const float contrast_thr, const float edge_thr, const float init_sigma,
+                 const bool double_input, const float img_scale, const float feature_ratio)
+{
+    Array<float> x     = createEmptyArray<float>(dim4());
+    Array<float> y     = createEmptyArray<float>(dim4());
+    Array<float> score = createEmptyArray<float>(dim4());
+    Array<float> ori   = createEmptyArray<float>(dim4());
+    Array<float> size  = createEmptyArray<float>(dim4());
+    Array<float> desc  = createEmptyArray<float>(dim4());
+
+    af_features_t feat;
+
+    feat.n = sift<T, convAccT>(x, y, score, ori, size, desc, getArray<T>(in),
+                               n_layers, contrast_thr, edge_thr, init_sigma,
+                               double_input, img_scale, feature_ratio);
+
+    feat.x           = getHandle(x);
+    feat.y           = getHandle(y);
+    feat.score       = getHandle(score);
+    feat.orientation = getHandle(ori);
+    feat.size        = getHandle(size);
+
+    feat_ = getFeaturesHandle(feat);
+    descriptors = getHandle<float>(desc);
+}
+
+af_err af_sift(af_features* feat, af_array* desc, const af_array in, const unsigned n_layers,
+               const float contrast_thr, const float edge_thr, const float init_sigma,
+               const bool double_input, const float img_scale, const float feature_ratio)
+{
+    try {
+#ifdef AF_NONFREE
+        ArrayInfo info = getInfo(in);
+        af::dim4 dims  = info.dims();
+
+        ARG_ASSERT(2, (dims[0] >= 15 && dims[1] >= 15 && dims[2] == 1 && dims[3] == 1));
+        ARG_ASSERT(3, n_layers > 0);
+        ARG_ASSERT(4, contrast_thr > 0.0f);
+        ARG_ASSERT(5, edge_thr >= 1.0f);
+        ARG_ASSERT(6, init_sigma > 0.5f);
+        ARG_ASSERT(8, img_scale > 0.0f);
+        ARG_ASSERT(9, feature_ratio > 0.0f);
+
+        dim_t in_ndims = dims.ndims();
+        DIM_ASSERT(1, (in_ndims <= 3 && in_ndims >= 2));
+
+        af_array tmp_desc;
+        af_dtype type  = info.getType();
+        switch(type) {
+            case f32: sift<float , float >(*feat, tmp_desc, in, n_layers, contrast_thr,
+                                           edge_thr, init_sigma, double_input,
+                                           img_scale, feature_ratio); break;
+            case f64: sift<double, double>(*feat, tmp_desc, in, n_layers, contrast_thr,
+                                           edge_thr, init_sigma, double_input,
+                                           img_scale, feature_ratio); break;
+            default : TYPE_ERROR(1, type);
+        }
+        std::swap(*desc, tmp_desc);
+#else
+        AF_ERROR("ArrayFire was not built with nonfree support, SIFT disabled\n", AFF_ERR_NONFREE);
+#endif
+    }
+    CATCHALL;
+
+    return AF_SUCCESS;
+}

-- 
Alioth's /usr/local/bin/git-commit-notice on /srv/git.debian.org/git/debian-science/packages/arrayfire.git