pihsi 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d949dd5f7e6d5f382c6c74a44f3678fbfe0a8fa8
-  data.tar.gz: ed8b316d1c8ffb11f954a57a0389d787079acd70
+  metadata.gz: 39a878f44496fcecc39a526f4452d31df5dc2fdf
+  data.tar.gz: 64e75af38da0c529812822f718efaed69c6e4db5
 SHA512:
-  metadata.gz: 5b09454ee9df43e8aac32714dcaf0a9c11fa7a56f5ace89730cd817fdba4115a4cbca1199ce4ef04e30f3f82ae6727bb7ebf037be84a0d02a6fb3c8ec0a2c7b9
-  data.tar.gz: 31ccc59f39331163b8fbeb290e7ee2db0e2ded8e40f9b77a5046610cbbdb5f79d34d40b5cf3fdc660aca5842699896d6e43a2b7ba48db23cb9db293b6174a694
+  metadata.gz: 6b7c9e706f8130b30cae6cb064fd7ab0ad019371a781867c05499412f09a5718b8a41512839d807f9e86c69a78f9382d2b9c901d4dbca6ef431cfaf93658d8ca
+  data.tar.gz: 5d53020195f0b5e8791ae2c60aef44e4be810f8141c0cbb0ec5a4220025b8b16568d41b237f45025e4ebc83ca33a663ed42f1d56b99a83ed384b31c38bdf1e52

data/README.md

@@ -32,7 +32,7 @@ Initialize Pihsi::SpeechRecognizer with proper [hmm, lm and dict](http://cmusphi
 recognizer = Pihsi::SpeechRecognizer.new hmm: 'xxx', lm: 'yyy', dict: 'zzz'
 ```
 
-Recognize a string read from your audio file:
+Recognize a string read from your audio file or an object respond to #read (IO object):
 
 ```ruby
 recognizer.recognize data

data/ext/pocket_sphinx/pocket_sphinx.c

@@ -7,20 +7,10 @@ typedef struct ps {
 
 static VALUE rb_eUtteranceError;
 
-/* Converts raw audio data into text.
- *
- * @param data [String] the raw audio data
- * @return [String, nil] the transcribed text or nil
- */
-VALUE recognize(VALUE self, VALUE data) {
+static VALUE decode(ps_decoder_t *ps, VALUE data) {
     char const *hyp, *uttid;
     int rv;
     int32 score;
-    ps_decoder_t *ps;
-    PocketSphinx *pocketSphinx;
-
-    Data_Get_Struct(self, PocketSphinx, pocketSphinx);
-    ps = pocketSphinx -> decoder;
 
     rv = ps_start_utt(ps, "goforward");
 
@@ -42,6 +32,26 @@ VALUE recognize(VALUE self, VALUE data) {
     }
 }
 
+/* Converts raw audio data into text.
+ *
+ * @param data [String, #read] the raw audio data or its io object
+ * @return [String, nil] the transcribed text or nil
+ */
+VALUE recognize(VALUE self, VALUE data) {
+    VALUE string;
+    PocketSphinx *pocketSphinx;
+
+    Data_Get_Struct(self, PocketSphinx, pocketSphinx);
+    if (rb_funcall(data, rb_intern("respond_to?"), 1, rb_str_new2("read")) == Qtrue) {
+        string = rb_funcall(data, rb_intern("read"), 0);
+    } else if (rb_obj_is_kind_of(data, rb_cString)) {
+        string = data;
+    } else {
+        rb_raise(rb_eArgError, "data can only be a string or an IO object");
+    }
+    return decode(pocketSphinx -> decoder, string);
+}
+
 static void deallocate(void *ps) {
     PocketSphinx *pocketSphinx = ps;
     ps_free(pocketSphinx -> decoder);

data/lib/pihsi/version.rb

@@ -1,3 +1,3 @@
 module Pihsi
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/spec/speech_recognizer_spec.rb

@@ -9,5 +9,17 @@ RSpec.describe Pihsi::SpeechRecognizer do
 
       it { should eq "go forward ten meters" }
     end
+
+    context 'when data is an IO object of goforward.raw' do
+      let(:data) { File.open('spec/fixtures/goforward.raw') }
+
+      it { should eq "go forward ten meters" }
+    end
+
+    context 'when data is invalid' do
+      let(:data) { 1 }
+
+      specify { expect { subject }.to raise_error(ArgumentError) }
+    end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pihsi
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Zhi-Qiang Lei
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-09-07 00:00:00.000000000 Z
+date: 2014-09-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler