83 lines
2.8 KiB
C++
83 lines
2.8 KiB
C++
#include "whisper.h"
|
||
#include "common.h"
|
||
|
||
#define MINIAUDIO_IMPLEMENTATION
|
||
#include "miniaudio.h"
|
||
|
||
#include <vector>
|
||
#include <cstdio>
|
||
#include <string>
|
||
|
||
// 音频回调：将采集到的数据存入 buffer
|
||
void data_callback(ma_device* pDevice, void* pOutput, const void* pInput, ma_uint32 frameCount) {
|
||
std::vector<float>* pBuffer = (std::vector<float>*)pDevice->pUserData;
|
||
const float* pInputFloat = (const float*)pInput;
|
||
if (pInputFloat == NULL) return;
|
||
|
||
pBuffer->insert(pBuffer->end(), pInputFloat, pInputFloat + frameCount);
|
||
// 保持 buffer 在最近 10 秒以内，防止内存溢出
|
||
if (pBuffer->size() > 16000 * 10) {
|
||
pBuffer->erase(pBuffer->begin(), pBuffer->begin() + (pBuffer->size() - 16000 * 10));
|
||
}
|
||
}
|
||
|
||
int main(int argc, char** argv) {
|
||
if (argc < 2) {
|
||
fprintf(stderr, "Usage: %s <model_path>\n", argv[0]);
|
||
return 1;
|
||
}
|
||
|
||
const char* model_path = argv[1];
|
||
|
||
// 1. 初始化 Whisper
|
||
struct whisper_context_params cparams = whisper_context_default_params();
|
||
cparams.use_gpu = true; // 你的 4050 显卡
|
||
struct whisper_context* ctx = whisper_init_from_file_with_params(model_path, cparams);
|
||
if (!ctx) return 1;
|
||
|
||
// 2. 初始化 Miniaudio
|
||
std::vector<float> audio_buffer;
|
||
ma_device_config deviceConfig = ma_device_config_init(ma_device_type_capture);
|
||
deviceConfig.capture.format = ma_format_f32; // Whisper 需要 float32
|
||
deviceConfig.capture.channels = 1; // 单声道
|
||
deviceConfig.sampleRate = 16000; // Whisper 硬指标 16kHz
|
||
deviceConfig.dataCallback = data_callback;
|
||
deviceConfig.pUserData = &audio_buffer;
|
||
|
||
ma_device device;
|
||
if (ma_device_init(NULL, &deviceConfig, &device) != MA_SUCCESS) {
|
||
fprintf(stderr, "Failed to open capture device.\n");
|
||
return -2;
|
||
}
|
||
|
||
ma_device_start(&device);
|
||
printf("🎤 录音中... 请说话 (按回车键进行单次识别，Ctrl+C 退出)\n");
|
||
|
||
while (true) {
|
||
getchar(); // 等待用户敲回车触发识别
|
||
|
||
printf("正在识别...\n");
|
||
|
||
whisper_full_params wparams = whisper_full_default_params(WHISPER_SAMPLING_GREEDY);
|
||
wparams.language = "zh";
|
||
wparams.n_threads = 12;
|
||
wparams.print_progress = false;
|
||
|
||
if (whisper_full(ctx, wparams, audio_buffer.data(), audio_buffer.size()) != 0) {
|
||
fprintf(stderr, "识别失败\n");
|
||
continue;
|
||
}
|
||
|
||
const int n_segments = whisper_full_n_segments(ctx);
|
||
for (int i = 0; i < n_segments; ++i) {
|
||
const char* text = whisper_full_get_segment_text(ctx, i);
|
||
printf("📝: %s\n", text);
|
||
}
|
||
audio_buffer.clear(); // 清空，准备下一轮
|
||
}
|
||
|
||
ma_device_uninit(&device);
|
||
whisper_free(ctx);
|
||
return 0;
|
||
}
|