// Headless render harness: drives the plugin's own decode engine and writes a WAV
// in the exact header format the reference CLI writes, so the two outputs can be
// compared byte for byte.
//
//   render_harness --input <file.eac3> --output <out.wav> [options]
//     --mode binaural|speaker     default binaural
//     --layout NAME               speaker layout (default 7.1)
//     --hrtf-source sofa|rosella  kind of model --hrtf points at (default sofa)
//     --hrtf PATH                 HRTF to render with; supplied by the caller and
//                                 never part of this repository.  Without it the
//                                 default location beside the executable is used.
//                                 Speaker layouts need no HRTF.
//     --gain-db X                 default 0
//     --tail S                    binaural tail seconds (default 5)
//     --ffmpeg PATH               default ffmpeg
//     --max-frames N              stop after N output frames (0 = all)
//
// It exists because the plugin's engine (src/joc_decode.*) has no foobar2000
// dependency: the very code that plays in foobar2000 can be run here and its
// output compared against the reference renderer, which is what the bit-exactness
// contract is about.  The WAV header mirrors the renderer's own writer: a simple
// IEEE-float header up to two channels, WAVE_FORMAT_EXTENSIBLE above that with a
// channel mask of zero.

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <vector>

#include "../src/joc_decode.h"

namespace {

struct Options {
    std::string input;
    std::string output;
    std::string mode = "binaural";
    std::string layout = "7.1";
    std::string hrtf;                  // SOFA file, or Rosella model file
    std::string hrtf_source = "sofa";  // sofa | rosella
    std::string ffmpeg = "ffmpeg";
    double gain_db = 0.0;
    double tail_seconds = 5.0;
    std::uint64_t max_frames = 0;
    std::uint64_t max_input_frames = 0;
};

void put_u16(std::string* out, unsigned value) {
    out->push_back(static_cast<char>(value & 0xFFu));
    out->push_back(static_cast<char>((value >> 8) & 0xFFu));
}

void put_u32(std::string* out, unsigned long long value) {
    for (int i = 0; i < 4; ++i) out->push_back(static_cast<char>((value >> (8 * i)) & 0xFFu));
}

// Same bytes the core's wav_writer produces.
std::string wav_header(unsigned channels, unsigned rate, std::uint64_t frames) {
    const unsigned bytes_per_sample = 4;
    const unsigned block_align = channels * bytes_per_sample;
    const std::uint64_t data_bytes = frames * block_align;

    std::string fmt;
    if (channels <= 2) {
        put_u16(&fmt, 3);  // WAVE_FORMAT_IEEE_FLOAT
        put_u16(&fmt, channels);
        put_u32(&fmt, rate);
        put_u32(&fmt, rate * block_align);
        put_u16(&fmt, block_align);
        put_u16(&fmt, 8 * bytes_per_sample);
    } else {
        put_u16(&fmt, 0xFFFE);  // WAVE_FORMAT_EXTENSIBLE
        put_u16(&fmt, channels);
        put_u32(&fmt, rate);
        put_u32(&fmt, rate * block_align);
        put_u16(&fmt, block_align);
        put_u16(&fmt, 8 * bytes_per_sample);
        put_u16(&fmt, 22);      // cbSize
        put_u16(&fmt, 8 * bytes_per_sample);
        put_u32(&fmt, 0);       // channel mask, as the core writes it
        static const unsigned char kFloatGuid[16] = {0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x10, 0x00,
                                                     0x80, 0x00, 0x00, 0xAA, 0x00, 0x38, 0x9B, 0x71};
        fmt.append(reinterpret_cast<const char*>(kFloatGuid), 16);
    }

    std::string header;
    header.append("RIFF", 4);
    put_u32(&header, 4u + 8u + fmt.size() + 8u + data_bytes);
    header.append("WAVE", 4);
    header.append("fmt ", 4);
    put_u32(&header, fmt.size());
    header.append(fmt);
    header.append("data", 4);
    put_u32(&header, data_bytes);
    return header;
}

bool parse(int argc, char** argv, Options* options) {
    for (int i = 1; i < argc; ++i) {
        const std::string arg = argv[i];
        auto next = [&]() -> std::string { return (i + 1 < argc) ? argv[++i] : std::string(); };
        if (arg == "--input") options->input = next();
        else if (arg == "--output") options->output = next();
        else if (arg == "--mode") options->mode = next();
        else if (arg == "--layout") options->layout = next();
        else if (arg == "--hrtf") options->hrtf = next();
        else if (arg == "--hrtf-source") options->hrtf_source = next();
        else if (arg == "--ffmpeg") options->ffmpeg = next();
        else if (arg == "--gain-db") options->gain_db = std::atof(next().c_str());
        else if (arg == "--tail") options->tail_seconds = std::atof(next().c_str());
        else if (arg == "--max-frames") options->max_frames = std::strtoull(next().c_str(), nullptr, 10);
        else if (arg == "--max-input-frames") options->max_input_frames = std::strtoull(next().c_str(), nullptr, 10);
        else if (arg == "--help" || arg == "-h") return false;
        else {
            std::fprintf(stderr, "render_harness: unknown argument %s\n", arg.c_str());
            return false;
        }
    }
    return !options->input.empty() && !options->output.empty();
}

}  // namespace

int main(int argc, char** argv) {
    Options options;
    if (!parse(argc, argv, &options)) {
        std::fprintf(stderr,
                     "usage: render_harness --input <eac3> --output <wav> [--mode binaural|speaker]\n"
                     "       [--layout NAME] [--hrtf-source sofa|rosella] [--hrtf PATH]\n"
                     "       [--gain-db X] [--tail S] [--ffmpeg path] [--max-frames N]\n"
                     "       [--max-input-frames N]\n");
        return 2;
    }

    joc_decode::Settings settings;
    settings.output = (options.mode == "speaker") ? joc_decode::Output::kSpeaker
                                                  : joc_decode::Output::kBinaural;
    settings.speaker_layout = options.layout;
    settings.hrtf_source = (options.hrtf_source == "rosella") ? joc_decode::HrtfSource::kRosella
                                                             : joc_decode::HrtfSource::kSofa;
    settings.hrtf_file = options.hrtf;
    settings.gain_db = options.gain_db;
    settings.tail_seconds = options.tail_seconds;
    settings.ffmpeg_path = options.ffmpeg;
    settings.input_frame_limit = options.max_input_frames;

    joc_decode::Engine engine;
    std::string error;
    if (!engine.start(options.input, settings, &error)) {
        std::fprintf(stderr, "render_harness: engine start failed: %s\n", error.c_str());
        return 1;
    }
    const unsigned channels = engine.channels();
    if (channels == 0) {
        std::fprintf(stderr, "render_harness: engine reported zero channels\n");
        return 1;
    }

    std::FILE* file = std::fopen(options.output.c_str(), "wb");
    if (file == nullptr) {
        std::fprintf(stderr, "render_harness: cannot write %s\n", options.output.c_str());
        return 1;
    }
    // The header carries the length, so write a placeholder and come back to it.
    const std::string header = wav_header(channels, 48000, 0);
    std::fwrite(header.data(), 1, header.size(), file);

    constexpr std::size_t kChunk = 4096;
    std::vector<float> buffer(kChunk * channels);
    std::uint64_t frames_written = 0;
    double peak = 0.0;
    for (;;) {
        std::size_t want = kChunk;
        if (options.max_frames != 0) {
            if (frames_written >= options.max_frames) break;
            const std::uint64_t left = options.max_frames - frames_written;
            if (left < want) want = static_cast<std::size_t>(left);
        }
        const std::size_t frames = engine.read(buffer.data(), want, &error);
        if (frames == 0) {
            if (!error.empty()) {
                std::fprintf(stderr, "render_harness: read failed: %s\n", error.c_str());
                std::fclose(file);
                return 1;
            }
            break;
        }
        for (std::size_t i = 0; i < frames * channels; ++i) {
            const double value = buffer[i] < 0.0f ? -static_cast<double>(buffer[i])
                                                  : static_cast<double>(buffer[i]);
            if (value > peak) peak = value;
        }
        std::fwrite(buffer.data(), sizeof(float), frames * channels, file);
        frames_written += frames;
    }
    engine.stop();

    const std::string final_header = wav_header(channels, 48000, frames_written);
    std::fseek(file, 0, SEEK_SET);
    std::fwrite(final_header.data(), 1, final_header.size(), file);
    std::fclose(file);

    std::printf("render_harness: %s -> %s\n", options.input.c_str(), options.output.c_str());
    std::printf("  channels=%u frames=%llu samples_per_channel=%llu peak=%.9f\n", channels,
                static_cast<unsigned long long>(frames_written),
                static_cast<unsigned long long>(frames_written), peak);
    return 0;
}