Added Dnn::Layer, more Ruby-ish code, etc.

ext/opencv/dnn_layer.cpp

@@ -0,0 +1,63 @@
+#include "opencv2/dnn.hpp"
+
+#include "opencv.hpp"
+#include "error.hpp"
+
+namespace rubyopencv {
+  namespace Dnn {
+    namespace Layer {
+      VALUE rb_klass = Qnil;
+
+      void free_layer(void* ptr) {
+        delete (cv::dnn::Layer*)ptr;
+      }
+
+      size_t memsize_layer(const void* ptr) {
+        return sizeof(cv::dnn::Layer);
+      }
+
+      rb_data_type_t opencv_layer_type = {
+        "Dnn::Layer", { 0, free_layer, memsize_layer, }, 0, 0, 0
+      };
+
+      VALUE layer2obj(cv::dnn::Layer* ptr) {
+        return TypedData_Wrap_Struct(rb_klass, &opencv_layer_type, ptr);
+      }
+
+      cv::dnn::Layer* obj2layer(VALUE obj) {
+        cv::dnn::Layer* ptr = NULL;
+        TypedData_Get_Struct(obj, cv::dnn::Layer, &opencv_layer_type, ptr);
+        return ptr;
+      }
+
+      VALUE rb_allocate(VALUE klass) {
+        cv::dnn::Layer* ptr = NULL;
+        return TypedData_Wrap_Struct(klass, &opencv_layer_type, ptr);
+      }
+
+      VALUE rb_initialize(VALUE self) {
+        return self;
+      }
+
+      VALUE rb_name(VALUE self) {
+        cv::dnn::Layer* selfptr = obj2layer(self);
+        return rb_str_new_cstr(selfptr->name.c_str());
+      }
+
+      VALUE rb_type(VALUE self) {
+        cv::dnn::Layer* selfptr = obj2layer(self);
+        return rb_str_new_cstr(selfptr->type.c_str());
+      }
+
+      void init(VALUE rb_module) {
+        rb_klass = rb_define_class_under(rb_module, "Layer", rb_cData);
+        rb_define_alloc_func(rb_klass, rb_allocate);
+
+        rb_define_private_method(rb_klass, "initialize", RUBY_METHOD_FUNC(rb_initialize), 0);
+
+        rb_define_method(rb_klass, "name", RUBY_METHOD_FUNC(rb_name), 0);
+        rb_define_method(rb_klass, "type", RUBY_METHOD_FUNC(rb_type), 0);
+      }
+    }
+  }
+}