EngineX-Iluvatar/enginex_bi_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

c-api add wave write to buffer. (#1962)

Browse Source 
Co-authored-by: jian.chen03 <jian.chen03@transwarp.io>
This commit is contained in:
cjsdurj
2025-03-10 17:21:23 +08:00
committed by GitHub
parent 6e261ed63f
commit b87fce9a7f
5 changed files with 48 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
sherpa-onnx/csrc/wave-writer.cc
View File

@@ -4,6 +4,7 @@
#include "sherpa-onnx/csrc/wave-writer.h"
#include <cstring>
#include <fstream>
#include <string>
#include <vector>
@@ -35,8 +36,12 @@ struct WaveHeader {
} // namespace
bool WriteWave(const std::string &filename, int32_t sampling_rate,
const float *samples, int32_t n) {
int64_t WaveFileSize(int32_t n_samples) {
return sizeof(WaveHeader) + n_samples * sizeof(int16_t);
}
void WriteWave(char *buffer, int32_t sampling_rate, const float *samples,
int32_t n) {
WaveHeader header{};
header.chunk_id = 0x46464952; // FFIR
header.format = 0x45564157; // EVAW
@@ -61,21 +66,26 @@ bool WriteWave(const std::string &filename, int32_t sampling_rate,
samples_int16[i] = samples[i] * 32676;
}
memcpy(buffer, &header, sizeof(WaveHeader));
memcpy(buffer + sizeof(WaveHeader), samples_int16.data(),
n * sizeof(int16_t));
}
bool WriteWave(const std::string &filename, int32_t sampling_rate,
const float *samples, int32_t n) {
std::string buffer;
buffer.resize(WaveFileSize(n));
WriteWave(buffer.data(), sampling_rate, samples, n);
std::ofstream os(filename, std::ios::binary);
if (!os) {
SHERPA_ONNX_LOGE("Failed to create %s", filename.c_str());
return false;
}
os.write(reinterpret_cast<const char *>(&header), sizeof(header));
os.write(reinterpret_cast<const char *>(samples_int16.data()),
samples_int16.size() * sizeof(int16_t));
os << buffer;
if (!os) {
SHERPA_ONNX_LOGE("Write %s failed", filename.c_str());
return false;
}
return true;
}