Function main

examples/benchmark.cpp:447–582

// ── Main ────────────────────────────────────────────────────────────────────

int main(int argc, char ** argv) {
    std::string models_dir = "models/";
    std::string video_path = "data/test_video.mp4";
    float       px         = 315.0f;
    float       py         = 250.0f;
    int         n_frames        = 10;
    int         n_threads       = 4;
    int         encode_img_size = 0;
    bool        cpu_only        = false;
    bool        gpu_only        = false;
    std::string filter;

    for (int i = 1; i < argc; i++) {
        std::string arg = argv[i];
        if      (arg == "--models-dir" && i + 1 < argc) { models_dir = argv[++i]; }
        else if (arg == "--video"      && i + 1 < argc) { video_path = argv[++i]; }
        else if (arg == "--point-x"    && i + 1 < argc) { px = (float)atof(argv[++i]); }
        else if (arg == "--point-y"    && i + 1 < argc) { py = (float)atof(argv[++i]); }
        else if (arg == "--n-frames"   && i + 1 < argc) { n_frames = atoi(argv[++i]); }
        else if (arg == "--n-threads"  && i + 1 < argc) { n_threads = atoi(argv[++i]); }
        else if (arg == "--encode-img-size" && i + 1 < argc) { encode_img_size = atoi(argv[++i]); }
        else if (arg == "--cpu-only")  { cpu_only = true; }
        else if (arg == "--gpu-only")  { gpu_only = true; }
        else if (arg == "--filter"     && i + 1 < argc) { filter = argv[++i]; }
        else if (arg == "--help" || arg == "-h") {
            fprintf(stderr,
                "Usage: %s [options]\n"
                "  --models-dir <path>   Models directory       (default: models/)\n"
                "  --video <path>        Video file             (default: data/test_video.mp4)\n"
                "  --point-x <f>         Click X                (default: 315.0)\n"
                "  --point-y <f>         Click Y                (default: 250.0)\n"
                "  --n-frames <n>        Frames to track        (default: 10)\n"
                "  --n-threads <n>       CPU threads            (default: 4)\n"
                "  --cpu-only            Skip Metal runs\n"
                "  --gpu-only            Skip CPU runs\n"
                "  --filter <substr>     Filter model filenames\n",
                argv[0]);
            return 0;
        } else {
            fprintf(stderr, "Unknown argument: %s\n", arg.c_str());
            return 1;
        }
    }

    if (n_frames < 2) {
        fprintf(stderr, "ERROR: --n-frames must be >= 2\n");
        return 1;
    }

    // Discover models
    auto entries = discover_models(models_dir, filter);
    if (entries.empty()) {
        fprintf(stderr, "ERROR: no .ggml files found in '%s'", models_dir.c_str());
        if (!filter.empty()) fprintf(stderr, " (filter: '%s')", filter.c_str());
        fprintf(stderr, "\n");
        return 1;
    }