#include <opencv2/dnn.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/imgcodecs.hpp>
#include <iostream>
#include <string>
#include <vector>

static std::string argVal(int argc, char** argv, const std::string& key, const std::string& def)
{
    for (int i = 1; i + 1 < argc; ++i)
        if (key == argv[i]) return argv[i + 1];
    return def;
}

int main(int argc, char** argv)
{
    std::string model  = argVal(argc, argv, "--model",  "east_text_detection_2026jul.onnx");
    std::string image  = argVal(argc, argv, "--image",  "example_outputs/input_image.png");
    std::string output = argVal(argc, argv, "--output", "example_outputs/output_image.png");

    cv::Mat img = cv::imread(image);
    if (img.empty())
    {
        std::cerr << "could not read image: " << image << std::endl;
        return 1;
    }

    cv::dnn::TextDetectionModel_EAST east(model);
    east.setConfidenceThreshold(0.5f).setNMSThreshold(0.4f);
    east.setInputParams(1.0, cv::Size(320, 320), cv::Scalar(123.68, 116.78, 103.94), true, false);

    std::vector<cv::RotatedRect> boxes;
    east.detectTextRectangles(img, boxes);
    std::cout << "detections " << boxes.size() << std::endl;

    cv::Mat out = img.clone();
    for (const cv::RotatedRect& box : boxes)
    {
        cv::Mat pts;
        cv::boxPoints(box, pts);
        std::vector<cv::Point> poly(4);
        for (int i = 0; i < 4; ++i)
            poly[i] = cv::Point(cvRound(pts.at<float>(i, 0)), cvRound(pts.at<float>(i, 1)));
        cv::polylines(out, poly, true, cv::Scalar(0, 255, 0), 2);
    }
    cv::imwrite(output, out);
    std::cout << "wrote " << output << std::endl;
    return 0;
}