120 lines
3.4 KiB
Rust
120 lines
3.4 KiB
Rust
use crate::AnyhowResult;
|
|
use log::{info,warn};
|
|
use std::process::{Command, Stdio};
|
|
use subprocess::{Popen, PopenConfig, Redirection};
|
|
use std::fs::OpenOptions;
|
|
use std::path::Path;
|
|
|
|
/// This command is used to check tacotron for being a real model
|
|
#[derive(Clone)]
|
|
pub struct TacotronModelCheckCommand {
|
|
/// Where the Tacotron code lives
|
|
tacotron_root_code_directory: String,
|
|
|
|
/// eg. `source python/bin/activate`
|
|
virtual_env_activation_command: String,
|
|
|
|
tacotron_model_check_script_name: String,
|
|
}
|
|
|
|
impl TacotronModelCheckCommand {
|
|
pub fn new(
|
|
tacotron_root_code_directory: &str,
|
|
virtual_env_activation_command: &str,
|
|
tacotron_model_check_script_name: &str,
|
|
) -> Self {
|
|
Self {
|
|
tacotron_root_code_directory: tacotron_root_code_directory.to_string(),
|
|
virtual_env_activation_command: virtual_env_activation_command.to_string(),
|
|
tacotron_model_check_script_name: tacotron_model_check_script_name.to_string(),
|
|
}
|
|
}
|
|
|
|
pub fn execute<P: AsRef<Path>>(
|
|
&self,
|
|
synthesizer_checkpoint_path: P,
|
|
output_metadata_filename: P,
|
|
spawn_process: bool
|
|
) -> AnyhowResult<()> {
|
|
let mut command = String::new();
|
|
|
|
command.push_str("echo 'test'");
|
|
command.push_str(" && ");
|
|
command.push_str(&format!("cd {}", self.tacotron_root_code_directory));
|
|
command.push_str(" && ");
|
|
command.push_str(&self.virtual_env_activation_command);
|
|
command.push_str(" && ");
|
|
command.push_str("python ");
|
|
command.push_str(&self.tacotron_model_check_script_name);
|
|
command.push_str(" --synthesizer_checkpoint_path ");
|
|
command.push_str(&synthesizer_checkpoint_path.as_ref().display().to_string());
|
|
command.push_str(" --output_metadata_filename ");
|
|
command.push_str(&output_metadata_filename.as_ref().display().to_string());
|
|
|
|
info!("Command: {:?}", command);
|
|
|
|
let command_parts = [
|
|
"bash",
|
|
"-c",
|
|
&command
|
|
];
|
|
|
|
if spawn_process {
|
|
// NB: This forks and returns immediately.
|
|
//let _child_pid = command_builder.spawn()?;
|
|
|
|
let stdout_file = OpenOptions::new()
|
|
.read(true)
|
|
.write(true)
|
|
.create(true)
|
|
.truncate(true)
|
|
.open("/tmp/tacotron_upload_stdout.txt")?;
|
|
|
|
let stderr_file = OpenOptions::new()
|
|
.read(true)
|
|
.write(true)
|
|
.create(true)
|
|
.truncate(true)
|
|
.open("/tmp/tacotron_upload_stderr.txt")?;
|
|
|
|
let mut p = Popen::create(&command_parts, PopenConfig {
|
|
//stdout: Redirection::Pipe,
|
|
//stderr: Redirection::Pipe,
|
|
stdout: Redirection::File(stdout_file),
|
|
stderr: Redirection::File(stderr_file),
|
|
..Default::default()
|
|
})?;
|
|
|
|
info!("Pid : {:?}", p.pid());
|
|
|
|
p.detach();
|
|
|
|
} else {
|
|
// NB: This is a blocking call.
|
|
/*let output = command_builder.output()?;
|
|
|
|
info!("Output status: {}", output.status);
|
|
info!("Stdout: {:?}", String::from_utf8(output.stdout));
|
|
error!("Stderr: {:?}", String::from_utf8(output.stderr));
|
|
|
|
if !output.status.success() {
|
|
bail!("Bad error code: {:?}", output.status);
|
|
}*/
|
|
|
|
let mut p = Popen::create(&command_parts, PopenConfig {
|
|
//stdout: Redirection::Pipe,
|
|
//stderr: Redirection::Pipe,
|
|
..Default::default()
|
|
})?;
|
|
|
|
info!("Pid : {:?}", p.pid());
|
|
|
|
let exit_status = p.wait()?;
|
|
|
|
info!("Exit status: {:?}", exit_status);
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
}
|