#include <iostream>
#include <vector>
#include "reverb.h"
#include "reverb_wrapper.h"
#include "dr_wav.h"


int main()
{
    unsigned int channels = 1, frame_size = 64, sample_rate = 48e3; double room_size = 0.3, wet_dry = 60.0, reverb_time = 1.9, pre_delay = 1.5;
    double high_damp_freq = 6890.0, high_ratio = 0.36, diffusion = 3.5, density = 1.3, hpf = 100.0, lpf = 11000.0;

    Reverb *reverb[1];
    reverb_wrapper_init((void **)reverb, channels, frame_size, sample_rate, room_size, wet_dry, reverb_time, pre_delay, high_damp_freq, 
                         high_ratio, diffusion, density, hpf, lpf);

    
    // 读入wav语音文件
    drwav_uint64 samples;
    float* pSampleDataIn = drwav_open_file_and_read_pcm_frames_f32("E:\\music\\Voice_print\\mengwanzhou.wav", &channels, &sample_rate, &samples, NULL);
    if (pSampleDataIn == NULL) {
        return -1;
    }

    int frame_count = samples / frame_size;
    samples = frame_count * frame_size;
    std::vector<float> outputData(samples);

    for (int i = 0; i < frame_count; i++) {
        float *src = pSampleDataIn + i * frame_size;
        float *dst = outputData.data() + i * frame_size;
        reverb_wrapper_process(*reverb, src, dst);
    }
    
    // 将处理后音频数据写入wav文件
    drwav_data_format format;
    format.container = drwav_container_riff;
    format.format = DR_WAVE_FORMAT_IEEE_FLOAT;
    format.channels = channels;
    format.sampleRate = sample_rate;
    format.bitsPerSample = 32;

    drwav wav;
    if (!drwav_init_file_write(&wav, "output.wav", &format, NULL)) {
        std::cerr << "Error: Cannot create output file" << std::endl;
        return -1;
    }

    drwav_uint64 framesWritten = drwav_write_pcm_frames(&wav, samples, outputData.data());
    if (framesWritten != samples) {
        std::cerr << "Error: Did not write all frames." << std::endl;
    }

    drwav_uninit(&wav);
    drwav_free(pSampleDataIn, NULL);

    std::cout << "Done." << std::endl;

    return 0;
}