mirror of
https://github.com/HIllya51/LunaTranslator.git
synced 2025-01-13 15:43:53 +08:00
104 lines
3.3 KiB
C++
104 lines
3.3 KiB
C++
|
|
#include "ebyroid.h"
|
|
#include "api_adapter.h"
|
|
#include "ebyutil.h"
|
|
using ebyroid::Ebyroid;
|
|
#include "types.h"
|
|
|
|
int voiceroid2wmain(int argc, wchar_t *wargv[])
|
|
{
|
|
|
|
char **argv = new char *[argc];
|
|
for (int i = 0; i < argc; i++)
|
|
{
|
|
int length = WideCharToMultiByte(CP_ACP, 0, wargv[i], -1, NULL, 0, NULL, NULL);
|
|
argv[i] = new char[length];
|
|
WideCharToMultiByte(CP_ACP, 0, wargv[i], -1, argv[i], length, NULL, NULL);
|
|
}
|
|
HANDLE hPipe = CreateNamedPipeA(argv[3], PIPE_ACCESS_DUPLEX, PIPE_TYPE_MESSAGE | PIPE_READMODE_MESSAGE | PIPE_WAIT, PIPE_UNLIMITED_INSTANCES, 65535, 65535, NMPWAIT_WAIT_FOREVER, 0);
|
|
|
|
Ebyroid *ebyroid = nullptr;
|
|
|
|
std::string last;
|
|
float rate = -1;
|
|
auto handle = CreateFileMappingA(INVALID_HANDLE_VALUE, &allAccess, PAGE_EXECUTE_READWRITE, 0, 1024 * 1024 * 10, argv[5]);
|
|
|
|
auto mapview = (char *)MapViewOfFile(handle, FILE_MAP_ALL_ACCESS | FILE_MAP_EXECUTE, 0, 0, 1024 * 1024 * 10);
|
|
memset(mapview, 0, 1024 * 1024 * 10);
|
|
SetEvent(CreateEventA(&allAccess, FALSE, FALSE, argv[4]));
|
|
ConnectNamedPipe(hPipe, NULL);
|
|
int freq1;
|
|
char input_j[4096] = {0};
|
|
DWORD _;
|
|
while (true)
|
|
{
|
|
ZeroMemory(input_j, sizeof(input_j));
|
|
|
|
if (!ReadFile(hPipe, input_j, 4096, &_, NULL))
|
|
break;
|
|
std::string voice = (char *)input_j;
|
|
float _rate;
|
|
if (!ReadFile(hPipe, &_rate, 4, &_, NULL))
|
|
break;
|
|
float _pitch;
|
|
if (!ReadFile(hPipe, &_pitch, 4, &_, NULL))
|
|
break;
|
|
if (voice != last)
|
|
{
|
|
if (ebyroid)
|
|
delete ebyroid;
|
|
ebyroid = Ebyroid::Create(argv[1],
|
|
argv[2],
|
|
voice.c_str());
|
|
last = voice;
|
|
}
|
|
ebyroid->Setparam(2, _rate, _pitch); // 0.5-4, 0.5-2
|
|
ZeroMemory(input_j, sizeof(input_j));
|
|
if (!ReadFile(hPipe, input_j, 4096, &_, NULL))
|
|
break;
|
|
if (voice.find("_44") != voice.npos)
|
|
freq1 = 44100;
|
|
else
|
|
freq1 = 22050;
|
|
std::vector<char> output;
|
|
int result = ebyroid->Hiragana(input_j, output);
|
|
output.push_back(0);
|
|
std::vector<int16_t> binary;
|
|
result = ebyroid->Speech(output.data(), binary);
|
|
size_t output_size = binary.size() * 2;
|
|
int fsize = output_size + 44;
|
|
if (fsize > 1024 * 1024 * 10)
|
|
{
|
|
fsize = 0;
|
|
}
|
|
else
|
|
{
|
|
|
|
int ptr = 0;
|
|
memcpy(mapview, "RIFF", 4);
|
|
ptr += 4;
|
|
memcpy(mapview + ptr, &fsize, 4);
|
|
ptr += 4;
|
|
memcpy(mapview + ptr, "WAVEfmt ", 8);
|
|
ptr += 8;
|
|
memcpy(mapview + ptr, "\x10\x00\x00\x00\x01\x00\x01\x00", 8);
|
|
ptr += 8;
|
|
int freq = freq1;
|
|
memcpy(mapview + ptr, &freq, 4);
|
|
ptr += 4;
|
|
freq = freq * 2;
|
|
memcpy(mapview + ptr, &freq, 4);
|
|
ptr += 4;
|
|
memcpy(mapview + ptr, "\x02\x00\x10\x00", 4);
|
|
ptr += 4;
|
|
memcpy(mapview + ptr, "data", 4);
|
|
ptr += 4;
|
|
memcpy(mapview + ptr, &output_size, 4);
|
|
ptr += 4;
|
|
memcpy(mapview + ptr, binary.data(), output_size);
|
|
}
|
|
WriteFile(hPipe, &fsize, 4, &_, NULL);
|
|
}
|
|
return 0;
|
|
}
|