Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

VoiceModel::get_all_modelsをC APIに移動 #649

Merged
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 1 addition & 5 deletions Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@ async_zip = { version = "0.0.11", features = ["full"] }
clap = { version = "4.0.10", features = ["derive"] }
easy-ext = "1.0.1"
fs-err = { version = "2.9.0", features = ["tokio"] }
futures = "0.3.26"
itertools = "0.10.5"
once_cell = "1.18.0"
regex = "1.10.0"
Expand All @@ -33,11 +34,6 @@ voicevox_core = { path = "crates/voicevox_core" }
tokio = { version = "1.25.0", features = ["rt", "rt-multi-thread", "macros", "sync"] }
derive-getters = "0.2.0"

# FIXME: iOS対応のpull request(https://github.com/wesleywiser/process_path/pull/16)がマージされる見込みが無いため
[workspace.dependencies.process_path]
git = "https://github.com/VOICEVOX/process_path.git"
rev = "de226a26e8e18edbdb1d6f986afe37bbbf35fbf4"

[workspace.package]
version = "0.0.0"
edition = "2021"
Expand Down
3 changes: 1 addition & 2 deletions crates/voicevox_core/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -18,12 +18,11 @@ derive_more = "0.99.17"
duplicate = "1.0.0"
easy-ext.workspace = true
fs-err.workspace = true
futures = "0.3.26"
futures.workspace = true
indexmap = { version = "2.0.0", features = ["serde"] }
itertools.workspace = true
nanoid = "0.4.0"
once_cell.workspace = true
process_path.workspace = true
regex.workspace = true
serde.workspace = true
serde_json.workspace = true
Expand Down
36 changes: 2 additions & 34 deletions crates/voicevox_core/src/voice_model.rs
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
use async_zip::{read::fs::ZipFileReader, ZipEntry};
use futures::future::{join3, join_all};
use futures::future::join3;
use serde::{de::DeserializeOwned, Deserialize};

use super::*;
use std::{
collections::{BTreeMap, HashMap},
env, io,
io,
path::{Path, PathBuf},
};

Expand Down Expand Up @@ -76,38 +76,6 @@ impl VoiceModel {
})
}

// FIXME: C APIに移動する
/// # Panics
///
/// 失敗したらパニックする
pub async fn get_all_models() -> Vec<Self> {
let root_dir = if let Some(root_dir) = env::var_os(Self::ROOT_DIR_ENV_NAME) {
root_dir.into()
} else {
process_path::get_dylib_path()
.or_else(process_path::get_executable_path)
.unwrap()
.parent()
.unwrap_or_else(|| "".as_ref())
.join("model")
};

let vvm_paths = root_dir
.read_dir()
.and_then(|entries| entries.collect::<std::result::Result<Vec<_>, _>>())
.unwrap_or_else(|e| panic!("{}が読めませんでした: {e}", root_dir.display()))
.into_iter()
.filter(|entry| entry.path().extension().map_or(false, |ext| ext == "vvm"))
.map(|entry| Self::from_path(entry.path()));

join_all(vvm_paths)
.await
.into_iter()
.collect::<std::result::Result<_, _>>()
.unwrap()
}
const ROOT_DIR_ENV_NAME: &str = "VV_MODELS_ROOT_DIR";

/// モデル内のすべてのスタイルに対するモデル内IDを取得する。
///
/// モデル内IDのマッピングが存在しない場合はそのままスタイルIDを返す。
Expand Down
7 changes: 6 additions & 1 deletion crates/voicevox_core_c_api/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ anstyle-query = "1.0.0"
colorchoice = "1.0.0"
cstr = "0.2.11"
derive-getters.workspace = true
futures.workspace = true
itertools.workspace = true
libc = "0.2.134"
once_cell.workspace = true
Expand All @@ -37,6 +38,11 @@ version = "0.4.23"
default-features = false
features = ["clock"]

# FIXME: iOS対応のpull request(https://github.com/wesleywiser/process_path/pull/16)がマージされる見込みが無いため
[dependencies.process_path]
git = "https://github.com/VOICEVOX/process_path.git"
rev = "de226a26e8e18edbdb1d6f986afe37bbbf35fbf4"

[dev-dependencies]
anyhow.workspace = true
assert_cmd = { version = "2.0.8", features = ["color-auto"] }
Expand All @@ -48,7 +54,6 @@ libloading = "0.7.3"
libtest-mimic = "0.6.0"
ndarray = "0.15.6"
ndarray-stats = "0.5.1"
process_path.workspace = true
regex.workspace = true
serde.workspace = true
serde_with = "3.3.0"
Expand Down
50 changes: 43 additions & 7 deletions crates/voicevox_core_c_api/src/compatible_engine.rs
Original file line number Diff line number Diff line change
Expand Up @@ -20,13 +20,14 @@ macro_rules! ensure_initialized {
static ERROR_MESSAGE: Lazy<Mutex<String>> = Lazy::new(|| Mutex::new(String::new()));

struct VoiceModelSet {
all_vvms: Vec<VoiceModel>,
all_metas_json: CString,
style_model_map: BTreeMap<StyleId, VoiceModelId>,
model_map: BTreeMap<VoiceModelId, VoiceModel>,
}

static VOICE_MODEL_SET: Lazy<Mutex<VoiceModelSet>> = Lazy::new(|| {
let all_vvms = RUNTIME.block_on(VoiceModel::get_all_models());
static VOICE_MODEL_SET: Lazy<VoiceModelSet> = Lazy::new(|| {
let all_vvms = RUNTIME.block_on(get_all_models());
Comment on lines -28 to +30
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ここはもとからMutexである必要が無かったので、これを機に外す。

let model_map: BTreeMap<_, _> = all_vvms
.iter()
.map(|vvm| (vvm.id().clone(), vvm.clone()))
Expand All @@ -41,15 +42,50 @@ static VOICE_MODEL_SET: Lazy<Mutex<VoiceModelSet>> = Lazy::new(|| {
}
}

Mutex::new(VoiceModelSet {
return VoiceModelSet {
all_metas_json: CString::new(serde_json::to_string(&metas).unwrap()).unwrap(),
all_vvms,
style_model_map,
model_map,
})
};

/// # Panics
///
/// 失敗したらパニックする
async fn get_all_models() -> Vec<VoiceModel> {
let root_dir = if let Some(root_dir) = env::var_os(ROOT_DIR_ENV_NAME) {
root_dir.into()
} else {
process_path::get_dylib_path()
.or_else(process_path::get_executable_path)
.unwrap()
.parent()
.unwrap_or_else(|| "".as_ref())
.join("model")
};

let vvm_paths = root_dir
.read_dir()
.and_then(|entries| entries.collect::<std::result::Result<Vec<_>, _>>())
.unwrap_or_else(|e| panic!("{}が読めませんでした: {e}", root_dir.display()))
.into_iter()
.filter(|entry| entry.path().extension().map_or(false, |ext| ext == "vvm"))
.map(|entry| VoiceModel::from_path(entry.path()));

futures::future::join_all(vvm_paths)
.await
.into_iter()
.collect::<std::result::Result<_, _>>()
.unwrap()
}

const ROOT_DIR_ENV_NAME: &str = "VV_MODELS_ROOT_DIR";
});

fn voice_model_set() -> MutexGuard<'static, VoiceModelSet> {
VOICE_MODEL_SET.lock().unwrap()
// FIXME: この関数を消して直接`VOICE_MODEL_SET`を参照するか、あるいは`once_cell::sync::OnceCell`
// 経由でエラーをエンジンに伝達するようにする
fn voice_model_set() -> &'static VoiceModelSet {
&VOICE_MODEL_SET
}

static SYNTHESIZER: Lazy<Mutex<Option<voicevox_core::Synthesizer>>> =
Expand Down Expand Up @@ -83,7 +119,7 @@ pub extern "C" fn initialize(use_gpu: bool, cpu_num_threads: c_int, load_all_mod
.await?;

if load_all_models {
for model in &VoiceModel::get_all_models().await {
for model in &voice_model_set().all_vvms {
synthesizer.load_voice_model(model).await?;
}
}
Expand Down
Loading