offline-tts-zipvoice-frontend-test.cc
2.3 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
// sherpa-onnx/csrc/offline-tts-zipvoice-frontend-test.cc
//
// Copyright (c) 2025 Xiaomi Corporation
#include "sherpa-onnx/csrc/offline-tts-zipvoice-frontend.h"
#include "espeak-ng/speak_lib.h"
#include "gtest/gtest.h"
#include "phoneme_ids.hpp"
#include "phonemize.hpp"
#include "sherpa-onnx/csrc/file-utils.h"
#include "sherpa-onnx/csrc/macros.h"
namespace sherpa_onnx {
TEST(ZipVoiceFrontend, Case1) {
std::string data_dir = "../zipvoice/espeak-ng-data";
if (!FileExists(data_dir + "/en_dict")) {
SHERPA_ONNX_LOGE("%s/en_dict does not exist. Skipping test",
data_dir.c_str());
return;
}
if (!FileExists(data_dir + "/phontab")) {
SHERPA_ONNX_LOGE("%s/phontab does not exist. Skipping test",
data_dir.c_str());
return;
}
if (!FileExists(data_dir + "/phonindex")) {
SHERPA_ONNX_LOGE("%s/phonindex does not exist. Skipping test",
data_dir.c_str());
return;
}
if (!FileExists(data_dir + "/phondata")) {
SHERPA_ONNX_LOGE("%s/phondata does not exist. Skipping test",
data_dir.c_str());
return;
}
if (!FileExists(data_dir + "/intonations")) {
SHERPA_ONNX_LOGE("%s/intonations does not exist. Skipping test",
data_dir.c_str());
return;
}
std::string pinyin_dict = data_dir + "/../pinyin.dict";
if (!FileExists(pinyin_dict)) {
SHERPA_ONNX_LOGE("%s does not exist. Skipping test", pinyin_dict.c_str());
return;
}
std::string tokens_file = data_dir + "/../tokens.txt";
if (!FileExists(tokens_file)) {
SHERPA_ONNX_LOGE("%s does not exist. Skipping test", tokens_file.c_str());
return;
}
auto frontend = OfflineTtsZipvoiceFrontend(
tokens_file, data_dir, pinyin_dict,
OfflineTtsZipvoiceModelMetaData{.use_espeak = true, .use_pinyin = true},
true);
std::string text = "how are you doing?";
std::vector<sherpa_onnx::TokenIDs> ans =
frontend.ConvertTextToTokenIds(text, "en-us");
text = "这是第一句。这是第二句。";
ans = frontend.ConvertTextToTokenIds(text, "en-us");
text =
"这是第一句。这是第二句。<pin1><yin2>测试 [S1]and hello "
"world[S2]这是第三句。";
ans = frontend.ConvertTextToTokenIds(text, "en-us");
}
} // namespace sherpa_onnx