102 lines
3.5 KiB
C++
102 lines
3.5 KiB
C++
// framed_test.cpp — рендер входа через SpectralProcessor (real mask chain).
|
|
// Usage: framed_test <input.wav> <output.wav> [fc[,q[,sens]] ...]
|
|
#include "spectral.hpp"
|
|
#include <cstdio>
|
|
#include <cstdlib>
|
|
#include <vector>
|
|
#include <cmath>
|
|
#include <cstring>
|
|
|
|
static bool load_wav(const char* path, std::vector<float>& out, int& sr) {
|
|
FILE* f = fopen(path, "rb");
|
|
if (!f) return false;
|
|
char hdr[44];
|
|
if (fread(hdr, 1, 44, f) != 44) return false;
|
|
sr = *(int*)(hdr + 24);
|
|
int ch = *(short*)(hdr + 22);
|
|
int bits = *(short*)(hdr + 34);
|
|
int data = *(int*)(hdr + 40);
|
|
int n = data / (ch * (bits / 8));
|
|
std::vector<short> raw(n * ch);
|
|
fread(raw.data(), 2, n * ch, f);
|
|
fclose(f);
|
|
out.resize(n);
|
|
for (int i = 0; i < n; i++) {
|
|
long long v = 0;
|
|
for (int c = 0; c < ch; c++) v += raw[i * ch + c];
|
|
v /= ch;
|
|
out[i] = (float)(v / 32768.0);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static bool save_wav(const char* path, const std::vector<float>& x, int sr) {
|
|
FILE* f = fopen(path, "wb");
|
|
if (!f) return false;
|
|
int data = (int)(x.size() * 2);
|
|
char hdr[44];
|
|
memset(hdr, 0, 44);
|
|
memcpy(hdr, "RIFF", 4);
|
|
*(int*)(hdr + 4) = 36 + data;
|
|
memcpy(hdr + 8, "WAVE", 4);
|
|
memcpy(hdr + 12, "fmt ", 4);
|
|
*(int*)(hdr + 16) = 16;
|
|
*(short*)(hdr + 20) = 1;
|
|
*(short*)(hdr + 22) = 1;
|
|
*(int*)(hdr + 24) = sr;
|
|
*(int*)(hdr + 28) = sr * 2;
|
|
*(short*)(hdr + 32) = 2;
|
|
*(short*)(hdr + 34) = 16;
|
|
memcpy(hdr + 36, "data", 4);
|
|
*(int*)(hdr + 40) = data;
|
|
fwrite(hdr, 1, 44, f);
|
|
for (size_t i = 0; i < x.size(); i++) {
|
|
short v = (short)(std::max(-1.0f, std::min(1.0f, x[i])) * 32767.0f);
|
|
fwrite(&v, 2, 1, f);
|
|
}
|
|
fclose(f);
|
|
return true;
|
|
}
|
|
|
|
int main(int argc, char** argv) {
|
|
if (argc < 3) { fprintf(stderr, "usage: %s in.wav out.wav [fc,q,sens] ...\n", argv[0]); return 1; }
|
|
std::vector<float> x;
|
|
int sr;
|
|
if (!load_wav(argv[1], x, sr)) { fprintf(stderr, "cannot load %s\n", argv[1]); return 1; }
|
|
std::vector<DetectorBand> bands;
|
|
if (argc >= 4 && strchr(argv[3], ',')) {
|
|
// comma form: one or more "fc,q,sens[,scale]" args, each parsed separately
|
|
for (int i = 3; i < argc; i++) {
|
|
float fc, q, sens, scl = 1.0f;
|
|
if (sscanf(argv[i], "%f,%f,%f,%f", &fc, &q, &sens, &scl) < 3) continue;
|
|
DetectorBand b; b.fc = fc; b.q = q; b.sens = sens; b.level_scale = scl;
|
|
bands.push_back(b);
|
|
}
|
|
} else {
|
|
for (int i = 3; i + 2 < argc; i += 3) {
|
|
DetectorBand b;
|
|
b.fc = (float)atof(argv[i]);
|
|
b.q = (float)atof(argv[i + 1]);
|
|
b.sens = (float)atof(argv[i + 2]);
|
|
if (i + 3 < argc) b.level_scale = (float)atof(argv[i + 3]);
|
|
bands.push_back(b);
|
|
}
|
|
}
|
|
if (bands.empty()) bands.push_back({1000.0f, 1.0f, 12.0f});
|
|
|
|
SpectralProcessor sp(2048, 512);
|
|
sp.setDetectorParams(bands);
|
|
std::vector<float> y(x.size());
|
|
const size_t BLK = 1 << 16;
|
|
std::vector<float> inb(BLK), outb(BLK);
|
|
for (size_t s = 0; s < x.size(); s += BLK) {
|
|
size_t n = std::min(BLK, x.size() - s);
|
|
memcpy(inb.data(), x.data() + s, n * sizeof(float));
|
|
for (size_t i = n; i < BLK; i++) inb[i] = 0.0f;
|
|
sp.processBlock(inb.data(), outb.data(), BLK, 1);
|
|
memcpy(y.data() + s, outb.data(), n * sizeof(float));
|
|
}
|
|
save_wav(argv[2], y, sr);
|
|
printf("wrote %s (%zu samples sr=%d, %zu bands)\n", argv[2], y.size(), sr, bands.size());
|
|
return 0;
|
|
} |