From: Erik Andresen Date: Sun, 21 Mar 2021 07:28:56 +0000 (+0100) Subject: asr_vosk: add comments X-Git-Url: https://defiant.homedns.org/gitweb/?p=ros_wild_thumper.git;a=commitdiff_plain;h=8e41425106e336ec1304320ef640e043abd26c55 asr_vosk: add comments --- diff --git a/scripts/asr_vosk.rb b/scripts/asr_vosk.rb index bc73949..b84a2f9 100755 --- a/scripts/asr_vosk.rb +++ b/scripts/asr_vosk.rb @@ -19,9 +19,10 @@ CONFIG = { class Speak def initialize(node) @logger = Logger.new(STDOUT) - @jsgf_enabled = false + @commands_enabled = false @publisher = node.advertise('asr_result', Std_msgs::String) + # Websocket handling EM.run do Signal.trap("INT") { send_eof } @ws = WebSocket::EventMachine::Client.connect(:uri => 'ws://192.168.36.4:2700') @@ -30,6 +31,7 @@ class Speak @ws.send '{"eof" : 1}' end + # Loop over all input data def run while true do data = ARGF.read(16000) @@ -69,29 +71,32 @@ class Speak @logger.debug "word=" + result["word"] end + # check for keywords first text = msg["text"] @logger.debug "text=" + msg["text"] if KEYWORDS.include? text keyword_detect(text) return end - if @jsgf_enabled + + # not a keyword, handle command if enabled + if @commands_enabled final_result(msg["text"]) end end end - # Enables/Disables the jsgf pipeline branch - def enable_jsgf(bEnable) - @jsgf_enabled = bEnable + # Enables/Disables the speech command + def enable_commands(bEnable) + @commands_enabled = bEnable end - # Result of jsgf pipeline branch + # Resulting speech command def final_result(hyp) @logger.info "final: " + hyp - enable_jsgf(false) + enable_commands(false) - # Publish pocketsphinx result as ros message + # Publish vosk result as ros message msg = Std_msgs::String.new msg.data = hyp @publisher.publish(msg) @@ -99,7 +104,7 @@ class Speak def keyword_detect(hyp) @logger.debug "Got keyword: " + hyp - enable_jsgf(true) + enable_commands(true) end end