Function main

tests/test_geom_enc.cpp:13–142

};

int main(int argc, char** argv) {
    if (argc < 4) {
        fprintf(stderr, "Usage: %s <ref_dir> <prephase_ref_dir> <model_path>\n", argv[0]);
        return 1;
    }

    const std::string ref_dir = argv[1];
    const std::string prephase_ref_dir = argv[2];
    const std::string model_path = argv[3];
    const std::string cpp_root = ref_dir + "/cpp_out_geom";

    // Test cases matching the Python dump script
    std::vector<geom_case> cases;

    // Case 1: dummy prompt (no exemplars, just CLS)
    {
        geom_case c;
        c.id = "dummy_prompt";
        // No exemplars
        cases.push_back(std::move(c));
    }

    // Case 2: single positive box (cx=0.5, cy=0.5, w=0.3, h=0.3)
    // API uses XYXY format — convert from CxCyWH to XYXY
    {
        geom_case c;
        c.id = "single_box";
        float cx = 0.5f, cy = 0.5f, w = 0.3f, h = 0.3f;
        sam3_box box;
        box.x0 = cx - w * 0.5f;  // 0.35
        box.y0 = cy - h * 0.5f;  // 0.35
        box.x1 = cx + w * 0.5f;  // 0.65
        box.y1 = cy + h * 0.5f;  // 0.65
        c.params.pos_exemplars.push_back(box);
        cases.push_back(std::move(c));
    }

    // Case 3: two boxes (one positive, one negative)
    // Box1: cx=0.3, cy=0.4, w=0.2, h=0.25 (positive)
    // Box2: cx=0.7, cy=0.6, w=0.15, h=0.2 (negative)
    {
        geom_case c;
        c.id = "two_boxes";
        {
            float cx = 0.3f, cy = 0.4f, w = 0.2f, h = 0.25f;
            sam3_box box = {cx - w * 0.5f, cy - h * 0.5f, cx + w * 0.5f, cy + h * 0.5f};
            c.params.pos_exemplars.push_back(box);
        }
        {
            float cx = 0.7f, cy = 0.6f, w = 0.15f, h = 0.2f;
            sam3_box box = {cx - w * 0.5f, cy - h * 0.5f, cx + w * 0.5f, cy + h * 0.5f};
            c.params.neg_exemplars.push_back(box);
        }
        cases.push_back(std::move(c));
    }

    sam3_params mparams;
    mparams.model_path = model_path;