基于LibAV的PTS转帧号辅助函数代码正确性咨询

我正在使用LibAV开发一款MP4解复用器，需要一个将PTS转换为帧号的辅助函数。我编写了如下代码，该代码在我的场景（视频帧率为25fps，时间刻度为25000，每次PTS递增40）下可得到正确的frameNumber，但不确定该代码在LibAV体系中对任意视频输入是否通用。我曾尝试使用LibAV的不同时间缩放函数，但未能得到正确结果，特此咨询该代码是否为标准实现方式。

static double get_frame_rate(AVStream* stream)
{
    if (stream->r_frame_rate.num > 0 && stream->r_frame_rate.den > 0)
    {
        return av_q2d(stream->r_frame_rate);
    }
    else if (stream->avg_frame_rate.num > 0 && stream->avg_frame_rate.den > 0)
    {
        return av_q2d(stream->avg_frame_rate);
    }
    else
    {
        printf("无法确定帧率\n");
        return 0;
    }
}


int64_t timestamp_to_frame(Demuxer* demuxer, int64_t pts)
{
    AVStream* stream = demuxer->fmtc->streams[demuxer->iVideoStream];
    if (!stream || pts < 0)
    {
        fprintf(stderr, "无效的流或PTS。\n");
        return -1;
    }

    // 获取帧率（FPS）
    const double fps = get_frame_rate(stream);//此处返回25.0

    // 流时间基准（例如我的场景中为1/25000）
    const AVRational timeBase = stream->time_base;

    // 计算单帧持续时间（秒）
    const double frameDurationInSeconds = 1.0 / fps;

    const int64_t ptsIncrement = (int64_t)((frameDurationInSeconds * timeBase.den) * frameDurationInSeconds + 0.5);
    // 通过将PTS除以增量计算帧号
    const int64_t frameNumber = pts / ptsIncrement;

    return frameNumber;
}

现有代码的问题 #

你的代码在特定固定帧率场景下能工作，但存在两个核心问题，导致它无法适配所有视频输入：

• 浮点数精度丢失：将帧率转换为double类型计算，在处理高时长视频或非整数帧率（如29.97fps、59.94fps）时，会累积精度误差，导致帧号计算偏差。
• PTS增量计算错误：代码中ptsIncrement的公式(frameDurationInSeconds * timeBase.den) * frameDurationInSeconds是错误的，它实际计算的是单帧持续时间的平方乘以时间基准分母，而非单帧对应的PTS增量。你当前场景下得到正确结果只是巧合。

标准实现方式 #

LibAV的设计核心是用整数有理数运算处理时间，避免浮点数误差。正确的PTS转帧号逻辑应该基于时间基准的整数转换，而非浮点数计算。以下是通用实现：

1. 修正帧率获取（返回有理数而非浮点数） #

static AVRational get_frame_rate_rational(AVStream* stream)
{
    // 优先使用r_frame_rate（编码时的标称帧率）
    if (stream->r_frame_rate.num > 0 && stream->r_frame_rate.den > 0)
    {
        return stream->r_frame_rate;
    }
    // fallback到avg_frame_rate（统计平均帧率）
    else if (stream->avg_frame_rate.num > 0 && stream->avg_frame_rate.den > 0)
    {
        return stream->avg_frame_rate;
    }
    else
    {
        fprintf(stderr, "无法确定帧率\n");
        return (AVRational){0, 1};
    }
}

2. 正确的PTS转帧号函数 #

int64_t timestamp_to_frame(Demuxer* demuxer, int64_t pts)
{
    AVStream* stream = demuxer->fmtc->streams[demuxer->iVideoStream];
    if (!stream || pts < 0)
    {
        fprintf(stderr, "无效的流或PTS。\n");
        return -1;
    }

    AVRational fps_rational = get_frame_rate_rational(stream);
    if (fps_rational.num == 0 || fps_rational.den == 0)
    {
        return -1;
    }

    // 使用LibAV内置的整数时间转换函数av_rescale_q
    // 逻辑：将PTS从流的时间基准，转换为"帧"作为单位的基准（1帧 = 1/fps 秒）
    int64_t frameNumber = av_rescale_q(pts, stream->time_base, av_inv_q(fps_rational));

    return frameNumber;
}

关键说明 #

• av_rescale_q的作用：该函数负责在不同时间基准之间做整数缩放，避免浮点数运算的精度丢失，是LibAV处理时间转换的标准方式。
• 可变帧率（VFR）视频的局限性：如果视频是可变帧率，帧号只能是近似值，因为每帧的实际持续时间不同。这种情况下，若需要精确帧号，必须逐帧跟踪PTS并计数，无法通过单一公式计算。
• 帧率优先级：r_frame_rate是编码时的标称帧率，优先使用；avg_frame_rate是统计得到的平均帧率，适合无法获取标称帧率的场景。

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