| /* |
| * Copyright 2016 Google Inc. All Rights Reserved. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include "cobalt/speech/speech_recognizer.h" |
| |
| #include "base/bind.h" |
| #include "base/rand_util.h" |
| #include "base/string_number_conversions.h" |
| #include "base/string_util.h" |
| #include "base/utf_string_conversions.h" |
| #include "cobalt/deprecated/platform_delegate.h" |
| #include "cobalt/loader/fetcher_factory.h" |
| #include "cobalt/network/network_module.h" |
| #include "cobalt/speech/google_streaming_api.pb.h" |
| #include "cobalt/speech/mic.h" |
| #include "cobalt/speech/speech_recognition_error.h" |
| #include "net/base/escape.h" |
| #include "net/url_request/url_fetcher.h" |
| |
| namespace cobalt { |
| namespace speech { |
| |
| namespace { |
| const char kBaseStreamURL[] = |
| "https://www.google.com/speech-api/full-duplex/v1"; |
| const char kUp[] = "up"; |
| const char kDown[] = "down"; |
| const char kClient[] = "com.speech.tv"; |
| |
| GURL AppendPath(const GURL& url, const std::string& value) { |
| std::string path(url.path()); |
| |
| if (!path.empty()) path += "/"; |
| |
| path += net::EscapePath(value); |
| GURL::Replacements replacements; |
| replacements.SetPathStr(path); |
| return url.ReplaceComponents(replacements); |
| } |
| |
| GURL AppendQueryParameter(const GURL& url, const std::string& new_query, |
| const std::string& value) { |
| std::string query(url.query()); |
| |
| if (!query.empty()) query += "&"; |
| |
| query += net::EscapeQueryParamValue(new_query, true); |
| |
| if (!value.empty()) { |
| query += "=" + net::EscapeQueryParamValue(value, true); |
| } |
| |
| GURL::Replacements replacements; |
| replacements.SetQueryStr(query); |
| return url.ReplaceComponents(replacements); |
| } |
| |
| SpeechRecognitionResultList::SpeechRecognitionResults |
| ProcessProtoSuccessResults(const proto::SpeechRecognitionEvent& event) { |
| DCHECK_EQ(event.status(), proto::SpeechRecognitionEvent::STATUS_SUCCESS); |
| |
| SpeechRecognitionResultList::SpeechRecognitionResults results; |
| for (int i = 0; i < event.result_size(); ++i) { |
| SpeechRecognitionResult::SpeechRecognitionAlternatives alternatives; |
| const proto::SpeechRecognitionResult& kProtoResult = event.result(i); |
| for (int j = 0; j < kProtoResult.alternative_size(); ++j) { |
| const proto::SpeechRecognitionAlternative& kAlternative = |
| kProtoResult.alternative(j); |
| scoped_refptr<SpeechRecognitionAlternative> alternative( |
| new SpeechRecognitionAlternative(kAlternative.transcript(), |
| kAlternative.confidence())); |
| alternatives.push_back(alternative); |
| } |
| bool final = kProtoResult.has_final() && kProtoResult.final(); |
| scoped_refptr<SpeechRecognitionResult> recognition_result( |
| new SpeechRecognitionResult(alternatives, final)); |
| results.push_back(recognition_result); |
| } |
| return results; |
| } |
| |
| // TODO: Feed error messages when creating SpeechRecognitionError. |
| void ProcessAndFireErrorEvent( |
| const proto::SpeechRecognitionEvent& event, |
| const SpeechRecognizer::EventCallback& event_callback) { |
| scoped_refptr<dom::Event> error_event; |
| switch (event.status()) { |
| case proto::SpeechRecognitionEvent::STATUS_SUCCESS: |
| NOTREACHED(); |
| return; |
| case proto::SpeechRecognitionEvent::STATUS_NO_SPEECH: |
| error_event = |
| new SpeechRecognitionError(SpeechRecognitionError::kNoSpeech, ""); |
| break; |
| case proto::SpeechRecognitionEvent::STATUS_ABORTED: |
| error_event = |
| new SpeechRecognitionError(SpeechRecognitionError::kAborted, ""); |
| break; |
| case proto::SpeechRecognitionEvent::STATUS_AUDIO_CAPTURE: |
| error_event = |
| new SpeechRecognitionError(SpeechRecognitionError::kAudioCapture, ""); |
| break; |
| case proto::SpeechRecognitionEvent::STATUS_NETWORK: |
| error_event = |
| new SpeechRecognitionError(SpeechRecognitionError::kNetwork, ""); |
| break; |
| case proto::SpeechRecognitionEvent::STATUS_NOT_ALLOWED: |
| error_event = |
| new SpeechRecognitionError(SpeechRecognitionError::kNotAllowed, ""); |
| break; |
| case proto::SpeechRecognitionEvent::STATUS_SERVICE_NOT_ALLOWED: |
| error_event = new SpeechRecognitionError( |
| SpeechRecognitionError::kServiceNotAllowed, ""); |
| break; |
| case proto::SpeechRecognitionEvent::STATUS_BAD_GRAMMAR: |
| error_event = |
| new SpeechRecognitionError(SpeechRecognitionError::kBadGrammar, ""); |
| break; |
| case proto::SpeechRecognitionEvent::STATUS_LANGUAGE_NOT_SUPPORTED: |
| error_event = new SpeechRecognitionError( |
| SpeechRecognitionError::kLanguageNotSupported, ""); |
| break; |
| } |
| |
| DCHECK(error_event); |
| event_callback.Run(error_event); |
| } |
| |
| } // namespace |
| |
| SpeechRecognizer::SpeechRecognizer(network::NetworkModule* network_module, |
| const EventCallback& event_callback) |
| : network_module_(network_module), |
| thread_("speech_recognizer"), |
| started_(false), |
| event_callback_(event_callback) { |
| thread_.StartWithOptions(base::Thread::Options(MessageLoop::TYPE_IO, 0)); |
| } |
| |
| SpeechRecognizer::~SpeechRecognizer() { |
| Stop(); |
| } |
| |
| void SpeechRecognizer::Start(const SpeechRecognitionConfig& config, |
| int sample_rate) { |
| // Called by the speech recognition manager thread. |
| thread_.message_loop()->PostTask( |
| FROM_HERE, base::Bind(&SpeechRecognizer::StartInternal, |
| base::Unretained(this), config, sample_rate)); |
| } |
| |
| void SpeechRecognizer::Stop() { |
| // Called by the speech recognition manager thread. |
| thread_.message_loop()->PostTask( |
| FROM_HERE, |
| base::Bind(&SpeechRecognizer::StopInternal, base::Unretained(this))); |
| } |
| |
| void SpeechRecognizer::RecognizeAudio(scoped_ptr<ShellAudioBus> audio_bus, |
| bool is_last_chunk) { |
| // Called by the speech recognition manager thread. |
| thread_.message_loop()->PostTask( |
| FROM_HERE, base::Bind(&SpeechRecognizer::UploadAudioDataInternal, |
| base::Unretained(this), base::Passed(&audio_bus), |
| is_last_chunk)); |
| } |
| |
| void SpeechRecognizer::OnURLFetchDownloadData( |
| const net::URLFetcher* source, scoped_ptr<std::string> download_data) { |
| DCHECK_EQ(thread_.message_loop(), MessageLoop::current()); |
| |
| if (source == downstream_fetcher_.get()) { |
| chunked_byte_buffer_.Append(*download_data); |
| while (chunked_byte_buffer_.HasChunks()) { |
| scoped_ptr<std::vector<uint8_t> > chunk = |
| chunked_byte_buffer_.PopChunk().Pass(); |
| |
| proto::SpeechRecognitionEvent event; |
| if (!event.ParseFromString(std::string(chunk->begin(), chunk->end()))) { |
| DLOG(WARNING) << "Parse proto string error."; |
| return; |
| } |
| |
| if (event.status() == proto::SpeechRecognitionEvent::STATUS_SUCCESS) { |
| ProcessAndFireSuccessEvent(ProcessProtoSuccessResults(event)); |
| } else { |
| ProcessAndFireErrorEvent(event, event_callback_); |
| } |
| } |
| } |
| } |
| |
| void SpeechRecognizer::OnURLFetchComplete(const net::URLFetcher* source) { |
| DCHECK_EQ(thread_.message_loop(), MessageLoop::current()); |
| UNREFERENCED_PARAMETER(source); |
| started_ = false; |
| } |
| |
| void SpeechRecognizer::StartInternal(const SpeechRecognitionConfig& config, |
| int sample_rate) { |
| DCHECK_EQ(thread_.message_loop(), MessageLoop::current()); |
| |
| if (started_) { |
| // Recognizer is already started. |
| return; |
| } |
| started_ = true; |
| |
| encoder_.reset(new AudioEncoderFlac(sample_rate)); |
| |
| // Required for streaming on both up and down connections. |
| std::string pair = base::Uint64ToString(base::RandUint64()); |
| |
| // Set up down stream first. |
| GURL down_url(kBaseStreamURL); |
| down_url = AppendPath(down_url, kDown); |
| down_url = AppendQueryParameter(down_url, "pair", pair); |
| // Use protobuffer as the output format. |
| down_url = AppendQueryParameter(down_url, "output", "pb"); |
| |
| downstream_fetcher_.reset( |
| net::URLFetcher::Create(down_url, net::URLFetcher::GET, this)); |
| downstream_fetcher_->SetRequestContext( |
| network_module_->url_request_context_getter()); |
| downstream_fetcher_->Start(); |
| |
| // Up stream. |
| GURL up_url(kBaseStreamURL); |
| up_url = AppendPath(up_url, kUp); |
| up_url = AppendQueryParameter(up_url, "client", kClient); |
| up_url = AppendQueryParameter(up_url, "pair", pair); |
| up_url = AppendQueryParameter(up_url, "output", "pb"); |
| up_url = AppendQueryParameter(up_url, "key", GetSpeechAPIKey()); |
| |
| // Language is required. If no language is specified, use the system language. |
| if (!config.lang.empty()) { |
| up_url = AppendQueryParameter(up_url, "lang", config.lang); |
| } else { |
| up_url = AppendQueryParameter( |
| up_url, "lang", |
| cobalt::deprecated::PlatformDelegate::Get()->GetSystemLanguage()); |
| } |
| |
| if (config.max_alternatives) { |
| up_url = AppendQueryParameter(up_url, "maxAlternatives", |
| base::UintToString(config.max_alternatives)); |
| } |
| |
| if (config.continuous) { |
| up_url = AppendQueryParameter(up_url, "continuous", ""); |
| } |
| if (config.interim_results) { |
| up_url = AppendQueryParameter(up_url, "interim", ""); |
| } |
| |
| upstream_fetcher_.reset( |
| net::URLFetcher::Create(up_url, net::URLFetcher::POST, this)); |
| upstream_fetcher_->SetRequestContext( |
| network_module_->url_request_context_getter()); |
| upstream_fetcher_->SetChunkedUpload(encoder_->GetMimeType()); |
| upstream_fetcher_->Start(); |
| } |
| |
| void SpeechRecognizer::StopInternal() { |
| DCHECK_EQ(thread_.message_loop(), MessageLoop::current()); |
| |
| if (!started_) { |
| // Recognizer is not started. |
| return; |
| } |
| started_ = false; |
| |
| upstream_fetcher_.reset(); |
| downstream_fetcher_.reset(); |
| encoder_.reset(); |
| |
| // Clear the final results. |
| final_results_.clear(); |
| // Clear any remaining audio data. |
| chunked_byte_buffer_.Clear(); |
| } |
| |
| void SpeechRecognizer::UploadAudioDataInternal( |
| scoped_ptr<ShellAudioBus> audio_bus, bool is_last_chunk) { |
| DCHECK_EQ(thread_.message_loop(), MessageLoop::current()); |
| DCHECK(audio_bus); |
| |
| std::string encoded_audio_data; |
| if (encoder_) { |
| encoder_->Encode(audio_bus.get()); |
| if (is_last_chunk) { |
| encoder_->Finish(); |
| } |
| encoded_audio_data = encoder_->GetAndClearAvailableEncodedData(); |
| } |
| |
| if (upstream_fetcher_ && !encoded_audio_data.empty()) { |
| upstream_fetcher_->AppendChunkToUpload(encoded_audio_data, is_last_chunk); |
| } |
| } |
| |
| void SpeechRecognizer::ProcessAndFireSuccessEvent( |
| const SpeechRecognitionResults& new_results) { |
| DCHECK_EQ(thread_.message_loop(), MessageLoop::current()); |
| |
| SpeechRecognitionResults success_results; |
| size_t total_size = final_results_.size() + new_results.size(); |
| success_results.reserve(total_size); |
| success_results = final_results_; |
| success_results.insert(success_results.end(), new_results.begin(), |
| new_results.end()); |
| |
| size_t result_index = final_results_.size(); |
| // Update final results list. |
| for (size_t i = 0; i < new_results.size(); ++i) { |
| if (new_results[i]->is_final()) { |
| final_results_.push_back(new_results[i]); |
| } |
| } |
| |
| scoped_refptr<SpeechRecognitionResultList> recognition_list( |
| new SpeechRecognitionResultList(success_results)); |
| scoped_refptr<SpeechRecognitionEvent> recognition_event( |
| new SpeechRecognitionEvent(SpeechRecognitionEvent::kResult, |
| static_cast<uint32>(result_index), |
| recognition_list)); |
| event_callback_.Run(recognition_event); |
| } |
| |
| } // namespace speech |
| } // namespace cobalt |