使用C语言getopt实现FFmpeg式冒号分隔命令行选项

我希望实现类似FFmpeg的命令行选项处理逻辑，具体支持两种调色板选项：

• 直接指定调色板：-p "255,0,0 0,255,0"
• 从配置文件加载调色板：-p:c blackwhite

这和FFmpeg处理编解码器选项的方式类似，比如：ffmpeg -i input.mp4 -c:v libx264 -c:a copy output.mp4

目前我用getopt实现的代码如下：

static struct option long_options[] = {
    {"palette", required_argument, 0, 'p'},
    // 其他选项...
};

int main(int argc, char** argv) {
    while ((c = getopt_long(argc, argv, "p:h", long_options, &option_index)) != -1) {
        switch (c) {
            case 'p': 
                if (optarg[0] == 'c' && optarg[1] == ':') {
                    // 无法正常工作 - optarg变为":c"
                    load_palette_from_config(optarg + 2);
                } else {
                    parse_numeric_palette(optarg);
                }
                break;
        }
    }
}

问题在于使用-p:c blackwhite时，getopt会把:c当作-p的参数，导致调色板加载失败。请问能不能用getopt实现这种FFmpeg风格的选项处理？

可以用getopt实现这种FFmpeg风格的选项处理，但需要调整getopt的使用逻辑，因为默认情况下getopt会把-p:c拆分为选项-p和参数:c，而非将:c识别为选项后缀。

核心思路是手动处理带后缀的选项格式，利用getopt的可选参数特性，配合手动提取后续参数来实现需求。

修改后的代码示例 #

#include <stdio.h>
#include <getopt.h>
#include <string.h>

void parse_numeric_palette(const char* arg) {
    printf("解析调色板：%s\n", arg);
}

void load_palette_from_config(const char* name) {
    printf("从配置加载调色板：%s\n", name);
}

int main(int argc, char** argv) {
    int c;
    int option_index = 0;
    static struct option long_options[] = {
        {"palette", required_argument, 0, 'p'},
        {"help", no_argument, 0, 'h'},
        {0, 0, 0, 0}
    };

    // 把选项字符串改为"p::h"，两个冒号表示-p的参数是可选的
    while ((c = getopt_long(argc, argv, "p::h", long_options, &option_index)) != -1) {
        switch (c) {
            case 'p':
                // 处理-p:c格式：optarg为":c"，需要取下一个参数作为配置名
                if (optarg != NULL && optarg[0] == ':') {
                    if (optind < argc) {
                        load_palette_from_config(argv[optind]);
                        optind++; // 手动跳过已处理的配置名参数
                    } else {
                        fprintf(stderr, "错误：-p:c 需要指定配置名\n");
                        return 1;
                    }
                } 
                // 处理直接指定调色板的情况
                else if (optarg != NULL) {
                    parse_numeric_palette(optarg);
                } 
                // 处理-p后无参数的错误情况
                else {
                    fprintf(stderr, "错误：-p 需要指定调色板或使用-p:c <配置名>\n");
                    return 1;
                }
                break;
            case 'h':
                printf("用法：\n");
                printf("  -p \"<RGB值列表>\" 直接指定调色板\n");
                printf("  -p:c <配置名> 从配置文件加载调色板\n");
                return 0;
            default:
                fprintf(stderr, "未知选项\n");
                return 1;
        }
    }

    return 0;
}

关键修改说明 #

1. 调整选项字符串：将原有的"p:h"改为"p::h"，双冒号表示-p的参数是可选的。这样输入-p:c时，getopt会把:c识别为-p的可选参数，而非将c当作单独选项。
2. 手动处理后缀与参数：当optarg以:开头时，判定为-p:c格式，此时从optind位置获取下一个参数作为配置名，处理完成后手动递增optind跳过该参数。
3. 增加错误处理：补充了参数缺失时的提示信息，提升命令行工具的健壮性。

测试示例 #

• 直接指定调色板：./program -p "255,0,0 0,255,0"，会调用parse_numeric_palette解析输入的RGB值。
• 从配置加载：./program -p:c blackwhite，会调用load_palette_from_config加载名为blackwhite的配置调色板。

内容的提问来源于stack exchange，提问作者N00B_141