nolan/tts-rs
1
0
Fork 0
mirror of https://github.com/ndarilek/tts-rs.git synced 2024-11-22 15:19:36 +00:00
Code Issues Projects Releases Wiki Activity

Remove InMemoryRandomAccessStream in WinRt::speak

Browse Source
This commit is contained in:
Bear-03 2022-07-23 14:03:45 +02:00
parent 87cf05f78e
commit 91a0f03f1a
No known key found for this signature in database
GPG Key ID: 3D1DC5AFDA57B32E
1 changed files with 18 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
src/backends/winrt.rs
View File

@ -9,14 +9,15 @@ use lazy_static::lazy_static;
use log::{info, trace}; use log::{info, trace};
use unic_langid::LanguageIdentifier; use unic_langid::LanguageIdentifier;
use windows::{ use windows::{
core::HSTRING,
Foundation::TypedEventHandler, Foundation::TypedEventHandler,
Media::{ Media::{
Core::MediaSource, Core::MediaSource,
Playback::{MediaPlayer, MediaPlayerAudioCategory}, Playback::{MediaPlayer, MediaPlayerAudioCategory},
SpeechSynthesis::{SpeechSynthesizer, VoiceGender, VoiceInformation}, SpeechSynthesis::{
SpeechSynthesisStream, SpeechSynthesizer, VoiceGender, VoiceInformation,
},
}, },
Storage::Streams::{DataReader, DataWriter, InMemoryRandomAccessStream}, Storage::Streams::DataReader,
}; };
use crate::{Backend, BackendId, Error, Features, Gender, UtteranceId, Voice, CALLBACKS}; use crate::{Backend, BackendId, Error, Features, Gender, UtteranceId, Voice, CALLBACKS};
@ -142,24 +143,18 @@ impl WinRt {
}) })
} }
fn synthesize_with_content_type(&mut self, text: &str) -> Result<(Vec<u8>, HSTRING), Error> { fn create_synthesis_stream(&mut self, text: &str) -> Result<SpeechSynthesisStream, Error> {
self.synth.Options()?.SetSpeakingRate(self.rate.into())?; self.synth.Options()?.SetSpeakingRate(self.rate.into())?;
self.synth.Options()?.SetAudioPitch(self.pitch.into())?; self.synth.Options()?.SetAudioPitch(self.pitch.into())?;
self.synth.Options()?.SetAudioVolume(self.volume.into())?; self.synth.Options()?.SetAudioVolume(self.volume.into())?;
self.synth.SetVoice(&self.voice)?; self.synth.SetVoice(&self.voice)?;
let synth_stream = self let stream = self
.synth .synth
.SynthesizeTextToStreamAsync(&text.into())? .SynthesizeTextToStreamAsync(&text.into())?
.get()?; .get()?;
let size = synth_stream.Size()?; Ok(stream)
let data_reader = DataReader::CreateDataReader(&synth_stream.GetInputStreamAt(0)?)?;
let mut bytes = vec![0; size as usize];
data_reader.LoadAsync(size as u32)?.get()?;
data_reader.ReadBytes(&mut bytes)?;
Ok((bytes, synth_stream.ContentType()?))
} }
} }
@ -213,16 +208,10 @@ impl Backend for WinRt {
} }
} }
if no_utterances { if no_utterances {
let (bytes, content_type) = self.synthesize_with_content_type(text)?; let stream = self.create_synthesis_stream(text)?;
let stream = InMemoryRandomAccessStream::new()?; let media_source = MediaSource::CreateFromStream(&stream, &stream.ContentType()?)?;
let data_writer = DataWriter::CreateDataWriter(&stream)?; self.player.SetSource(&media_source)?;
data_writer.WriteBytes(&bytes)?;
data_writer.StoreAsync()?;
data_writer.FlushAsync()?;
let source = MediaSource::CreateFromStream(&stream, &content_type)?;
self.player.SetSource(&source)?;
self.player.Play()?; self.player.Play()?;
let mut callbacks = CALLBACKS.lock().unwrap(); let mut callbacks = CALLBACKS.lock().unwrap();
@ -235,7 +224,14 @@ impl Backend for WinRt {
} }
fn synthesize(&mut self, text: &str) -> Result<Vec<u8>, Error> { fn synthesize(&mut self, text: &str) -> Result<Vec<u8>, Error> {
let (bytes, _) = self.synthesize_with_content_type(text)?; let stream = self.create_synthesis_stream(text)?;
let size = stream.Size()?;
let data_reader = DataReader::CreateDataReader(&stream.GetInputStreamAt(0)?)?;
let mut bytes = vec![0; size as usize];
data_reader.LoadAsync(size as u32)?;
data_reader.ReadBytes(&mut bytes)?;
Ok(bytes) Ok(bytes)
} }