webrtcvad 0.1.0 → 0.2.3

data/ext/webrtcvad/webrtcvad.c

@@ -0,0 +1,112 @@
+#include <ruby.h>
+#include "webrtc/common_audio/vad/include/webrtc_vad.h"
+
+struct vad_instance{
+  VadInst* inst;  
+  int rage;
+};
+
+void vad_free(struct vad_instance* ptr) {
+  WebRtcVad_Free(ptr->inst);
+}
+
+static VALUE rb_vad_create(VALUE klass, VALUE rb_rage) {
+  struct vad_instance* vad;
+  VALUE obj = Data_Make_Struct(klass, struct vad_instance, NULL, vad_free, vad);
+  VALUE argv[1];
+  argv[0] = rb_rage;
+  rb_obj_call_init(obj, 1, argv);
+  return obj;
+}
+
+static int __attribute__ ((unused)) get_aggressiveness(VALUE rb_rage) {
+  int rage = NUM2INT(rb_rage);
+  if (rage < 0 || rage > 3) {
+    rb_raise(rb_const_get(rb_cObject, rb_intern("ArgumentError")), "Invalid VAD aggressiveness value");
+  }
+  return rage;
+}
+
+static VALUE rb_vad_init(VALUE self, VALUE rb_rage) {
+  struct vad_instance* vad;
+  Data_Get_Struct(self, struct vad_instance, vad);
+  
+  vad->inst = WebRtcVad_Create();
+  vad->rage = get_aggressiveness(rb_rage);
+  
+  if (WebRtcVad_Init(vad->inst)) {
+    rb_raise(rb_eRuntimeError, "Cannot init VAD");
+  }
+  if (WebRtcVad_set_mode(vad->inst, vad->rage)) {
+    rb_raise(rb_eRuntimeError, "Cannot set VAD aggressiveness");
+  }
+  return self;
+}
+
+static VALUE rb_vad_set_aggressiveness(VALUE self, VALUE rb_rage) {
+  struct vad_instance* vad;
+  Data_Get_Struct(self, struct vad_instance, vad);
+  
+  int rage = get_aggressiveness(rb_rage);
+  if (WebRtcVad_set_mode(vad->inst, rage)) {
+    rb_raise(rb_eRuntimeError, "Cannot set VAD aggressiveness");
+  }
+  vad->rage = rage;
+  return rb_rage;
+}
+
+static VALUE rb_vad_get_aggressiveness(VALUE self) {
+  struct vad_instance* vad;
+  Data_Get_Struct(self, struct vad_instance, vad);
+  
+  return INT2NUM(vad->rage);
+}
+
+static VALUE rb_valid_rate_and_frame_length(VALUE self, VALUE rb_rate, VALUE rb_frame_len) {
+  int rate, frame_length;
+  rate = NUM2INT(rb_rate);
+  frame_length = NUM2INT(rb_frame_len);
+
+  if (WebRtcVad_ValidRateAndFrameLength(rate, frame_length)) {
+    return Qfalse;
+  } else {
+    return Qtrue;
+  }
+}
+
+static VALUE rb_vad_process(VALUE self, VALUE rb_sample_rate, VALUE rb_audio_frame, VALUE rb_offset, VALUE rb_frame_len) {
+  int sr = NUM2INT(rb_sample_rate);
+  int offset = NUM2INT(rb_offset);
+  int frame_length = NUM2INT(rb_frame_len);
+  int result;
+  int16_t *buf = (int16_t*)(StringValuePtr(rb_audio_frame) + offset);
+  
+  struct vad_instance* ptr;
+  Data_Get_Struct(self, struct vad_instance, ptr);
+  
+  result = WebRtcVad_Process(ptr->inst, sr, buf, frame_length);
+  switch (result) {
+  case 1:
+    return INT2NUM(1);
+  case 0:
+    return INT2NUM(0);
+  case -1:
+    break;
+  default:
+    rb_raise(rb_eRuntimeError, "Error while processing frame");
+  }
+  return Qnil;
+}
+
+void Init_webrtcvad(void) {
+  VALUE mod = rb_define_module("WebRTC");
+  VALUE cVad = rb_define_class_under(mod, "Vad", rb_cObject);
+
+  rb_define_singleton_method(cVad, "new", rb_vad_create, 1);
+  rb_define_method(cVad, "initialize", rb_vad_init, 1);
+  rb_define_method(cVad, "aggressiveness=", rb_vad_set_aggressiveness, 1);
+  rb_define_method(cVad, "aggressiveness", rb_vad_get_aggressiveness, 0);
+  rb_define_method(cVad, "process", rb_vad_process, 4);
+  rb_define_singleton_method(cVad, "valid_rate_and_frame_length?", rb_valid_rate_and_frame_length, 2);
+  rb_define_method(cVad, "valid_frame?", rb_valid_rate_and_frame_length, 2);
+}

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: webrtcvad
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.3
 platform: ruby
 authors:
 - Ivan Razuvaev
@@ -13,9 +13,43 @@ dependencies: []
 description: WebRTC Voice Activity Detection library ruby wrapper
 email: i@orlando-labs.com
 executables: []
-extensions: []
+extensions:
+- ext/webrtcvad/extconf.rb
 extra_rdoc_files: []
 files:
+- ext/webrtcvad/extconf.rb
+- ext/webrtcvad/webrtc/common_audio/signal_processing/division_operations.c
+- ext/webrtcvad/webrtc/common_audio/signal_processing/dot_product_with_scale.h
+- ext/webrtcvad/webrtc/common_audio/signal_processing/energy.c
+- ext/webrtcvad/webrtc/common_audio/signal_processing/get_scaling_square.c
+- ext/webrtcvad/webrtc/common_audio/signal_processing/include/signal_processing_library.h
+- ext/webrtcvad/webrtc/common_audio/signal_processing/include/spl_inl.h
+- ext/webrtcvad/webrtc/common_audio/signal_processing/resample_48khz.c
+- ext/webrtcvad/webrtc/common_audio/signal_processing/resample_by_2_internal.c
+- ext/webrtcvad/webrtc/common_audio/signal_processing/resample_by_2_internal.h
+- ext/webrtcvad/webrtc/common_audio/signal_processing/resample_fractional.c
+- ext/webrtcvad/webrtc/common_audio/third_party/spl_sqrt_floor/spl_sqrt_floor.c
+- ext/webrtcvad/webrtc/common_audio/third_party/spl_sqrt_floor/spl_sqrt_floor.h
+- ext/webrtcvad/webrtc/common_audio/third_party/spl_sqrt_floor/spl_sqrt_floor_mips.c
+- ext/webrtcvad/webrtc/common_audio/vad/include/webrtc_vad.h
+- ext/webrtcvad/webrtc/common_audio/vad/vad_core.c
+- ext/webrtcvad/webrtc/common_audio/vad/vad_core.h
+- ext/webrtcvad/webrtc/common_audio/vad/vad_filterbank.c
+- ext/webrtcvad/webrtc/common_audio/vad/vad_filterbank.h
+- ext/webrtcvad/webrtc/common_audio/vad/vad_gmm.c
+- ext/webrtcvad/webrtc/common_audio/vad/vad_gmm.h
+- ext/webrtcvad/webrtc/common_audio/vad/vad_sp.c
+- ext/webrtcvad/webrtc/common_audio/vad/vad_sp.h
+- ext/webrtcvad/webrtc/common_audio/vad/webrtc_vad.c
+- ext/webrtcvad/webrtc/rtc_base/checks.cc
+- ext/webrtcvad/webrtc/rtc_base/checks.h
+- ext/webrtcvad/webrtc/rtc_base/compile_assert_c.h
+- ext/webrtcvad/webrtc/rtc_base/numerics/safe_compare.h
+- ext/webrtcvad/webrtc/rtc_base/sanitizer.h
+- ext/webrtcvad/webrtc/rtc_base/system/inline.h
+- ext/webrtcvad/webrtc/rtc_base/system/rtc_export.h
+- ext/webrtcvad/webrtc/rtc_base/type_traits.h
+- ext/webrtcvad/webrtcvad.c
 - lib/webrtcvad.rb
 homepage: https://orlando-labs.com
 licenses:
@@ -27,7 +61,7 @@ require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
-  - - '='
+  - - ">="
     - !ruby/object:Gem::Version
       version: 2.4.0
 required_rubygems_version: !ruby/object:Gem::Requirement