// 演示与对拍工具：读一张原始像素转储，推理并打印框。
//
//   单张: ./film_predict <model.xml> <image.bin> <宽> <高> <通道> [conf] [overlap]
//   批量: ./film_predict <model.xml> --manifest <清单> [conf] [overlap]
//
// 输入是连续的 HWC uint8 缓冲（用 scripts/dump_raw.py 从图片生成）。产线里通常直接
// 拿相机帧的指针，不需要这一步。
//
// 清单每行 `路径 宽 高 通道`，模型只加载一次——逐张起进程的话加载耗时会盖过推理。
#include "film_detector.hpp"

#include <cstdio>
#include <fstream>
#include <iostream>
#include <vector>

namespace {

bool read_pixels(const std::string& path, size_t expected, std::vector<unsigned char>& out) {
    std::ifstream file(path, std::ios::binary);
    if (!file) { std::cerr << "打不开 " << path << "\n"; return false; }
    out.assign(std::istreambuf_iterator<char>(file), std::istreambuf_iterator<char>());
    if (out.size() != expected) {
        std::cerr << path << " 的缓冲是 " << out.size() << " 字节，应为 " << expected << "\n";
        return false;
    }
    return true;
}

void run_one(const film::Detector& detector, const film::ImageView& image, float conf,
             int overlap) {
    const auto boxes = (image.width == detector.input_size() &&
                        image.height == detector.input_size())
        ? detector.infer_tile(image, conf)
        : detector.infer(image, conf, 0.5f, overlap);
    std::printf("%zu\n", boxes.size());
    for (const auto& b : boxes) {
        std::printf("%d %.6f %.6f %.6f %.6f %.4f\n",
                    b.class_id, b.x1, b.y1, b.x2, b.y2, b.score);
    }
}

}  // namespace

int main(int argc, char** argv) {
    if (argc < 4) {
        std::cerr << "用法:\n  " << argv[0] << " <model.xml> <image.bin> <宽> <高> <通道>"
                  << " [conf] [overlap]\n  " << argv[0]
                  << " <model.xml> --manifest <清单> [conf] [overlap]\n";
        return 1;
    }
    const std::string model_path = argv[1];
    const bool batch = std::string(argv[2]) == "--manifest";
    const float conf = batch ? (argc > 4 ? std::stof(argv[4]) : 0.25f)
                             : (argc > 6 ? std::stof(argv[6]) : 0.25f);
    const int overlap = batch ? (argc > 5 ? std::stoi(argv[5]) : 32)
                              : (argc > 7 ? std::stoi(argv[7]) : 32);

    film::Detector detector(model_path);
    std::vector<unsigned char> pixels;

    if (batch) {
        std::ifstream manifest(argv[3]);
        if (!manifest) { std::cerr << "打不开清单 " << argv[3] << "\n"; return 1; }
        std::string path;
        int width = 0, height = 0, channels = 0;
        while (manifest >> path >> width >> height >> channels) {
            if (!read_pixels(path, static_cast<size_t>(width) * height * channels, pixels)) {
                return 1;
            }
            run_one(detector, film::ImageView{pixels.data(), width, height, channels},
                    conf, overlap);
        }
        return 0;
    }

    if (argc < 6) { std::cerr << "单张模式需要 宽 高 通道\n"; return 1; }
    const int width = std::stoi(argv[3]), height = std::stoi(argv[4]);
    const int channels = std::stoi(argv[5]);
    if (!read_pixels(argv[2], static_cast<size_t>(width) * height * channels, pixels)) return 1;
    run_one(detector, film::ImageView{pixels.data(), width, height, channels}, conf, overlap);
    return 0;
}