githubmirror/FFmpeg

libavutil/tests/mathematics.c
175 строк8 KB
marcos ashton
tests/fate/libavutil: add FATE test for mathematics
05 апр 2026, 03:12
05 апр 2026, 03:12e18c8c5
/*
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include <inttypes.h>
#include <stdint.h>
#include <stdio.h>

#include "libavutil/avutil.h"
#include "libavutil/macros.h"
#include "libavutil/mathematics.h"
#include "libavutil/rational.h"

int main(void)
{
    int64_t last;

    /* av_gcd */
    printf("Testing av_gcd()\n");
    static const struct { int64_t a, b, expected; } gcd_tests[] = {
        {   0,   0,  0 },
        {   1,   0,  1 },
        {   0,   1,  1 },
        {   6,   4,  2 },
        {  12,   8,  4 },
        {  17,  13,  1 },
        { 100,  75, 25 },
        {  -6,   4,  2 },
        {   6,  -4,  2 },
    };
    for (int i = 0; i < FF_ARRAY_ELEMS(gcd_tests); i++)
        printf("gcd(%"PRId64", %"PRId64") = %"PRId64" %s\n",
               gcd_tests[i].a, gcd_tests[i].b,
               av_gcd(gcd_tests[i].a, gcd_tests[i].b),
               av_gcd(gcd_tests[i].a, gcd_tests[i].b) == gcd_tests[i].expected ? "OK" : "FAIL");

    /* av_rescale */
    printf("\nTesting av_rescale()\n");
    printf("rescale(6, 3, 2) = %"PRId64"\n", av_rescale(6, 3, 2));
    printf("rescale(0, 3, 2) = %"PRId64"\n", av_rescale(0, 3, 2));
    printf("rescale(1, 1, 1) = %"PRId64"\n", av_rescale(1, 1, 1));
    printf("rescale(-6, 3, 2) = %"PRId64"\n", av_rescale(-6, 3, 2));
    printf("rescale(90000, 1, 90000) = %"PRId64"\n", av_rescale(90000, 1, 90000));

    /* av_rescale_rnd with different rounding modes */
    printf("\nTesting av_rescale_rnd()\n");
    static const struct {
        int64_t a, b, c;
        enum AVRounding rnd;
        int64_t expected;
    } rnd_tests[] = {
        {  7,  1,  2, AV_ROUND_ZERO,      3 },
        {  7,  1,  2, AV_ROUND_INF,       4 },
        {  7,  1,  2, AV_ROUND_DOWN,      3 },
        {  7,  1,  2, AV_ROUND_UP,        4 },
        {  7,  1,  2, AV_ROUND_NEAR_INF,  4 },
        { -7,  1,  2, AV_ROUND_ZERO,     -3 },
        { -7,  1,  2, AV_ROUND_INF,      -4 },
        { -7,  1,  2, AV_ROUND_DOWN,     -4 },
        { -7,  1,  2, AV_ROUND_UP,       -3 },
        {  6,  1,  2, AV_ROUND_NEAR_INF,  3 },
    };
    for (int i = 0; i < FF_ARRAY_ELEMS(rnd_tests); i++) {
        int64_t r = av_rescale_rnd(rnd_tests[i].a, rnd_tests[i].b,
                                   rnd_tests[i].c, rnd_tests[i].rnd);
        printf("rescale_rnd(%"PRId64", %"PRId64", %"PRId64", %d) = %"PRId64" %s\n",
               rnd_tests[i].a, rnd_tests[i].b, rnd_tests[i].c,
               rnd_tests[i].rnd, r,
               r == rnd_tests[i].expected ? "OK" : "FAIL");
    }

    /* AV_ROUND_PASS_MINMAX */
    printf("\nTesting AV_ROUND_PASS_MINMAX\n");
    printf("INT64_MIN passthrough: %s\n",
           av_rescale_rnd(INT64_MIN, 1, 2,
                          AV_ROUND_UP | AV_ROUND_PASS_MINMAX) == INT64_MIN ? "OK" : "FAIL");
    printf("INT64_MAX passthrough: %s\n",
           av_rescale_rnd(INT64_MAX, 1, 2,
                          AV_ROUND_UP | AV_ROUND_PASS_MINMAX) == INT64_MAX ? "OK" : "FAIL");
    printf("normal with PASS_MINMAX: %"PRId64"\n",
           av_rescale_rnd(3, 1, 2, AV_ROUND_UP | AV_ROUND_PASS_MINMAX));

    /* large value rescale (exercises 128-bit multiply path) */
    printf("\nTesting large value rescale\n");
    printf("rescale(INT64_MAX/2, 2, 1) = %"PRId64"\n",
           av_rescale_rnd(INT64_MAX / 2, 2, 1, AV_ROUND_ZERO));
    printf("rescale(1000000007, 1000000009, 1000000007) = %"PRId64"\n",
           av_rescale(1000000007LL, 1000000009LL, 1000000007LL));
    /* b and c both > INT_MAX triggers 128-bit multiply */
    printf("rescale_rnd(10, INT_MAX+1, INT_MAX+1, ZERO) = %"PRId64"\n",
           av_rescale_rnd(10, (int64_t)INT32_MAX + 1, (int64_t)INT32_MAX + 1, AV_ROUND_ZERO));
    printf("rescale_rnd(7, 3000000000, 2000000000, NEAR_INF) = %"PRId64"\n",
           av_rescale_rnd(7, 3000000000LL, 2000000000LL, AV_ROUND_NEAR_INF));

    /* av_rescale_q */
    printf("\nTesting av_rescale_q()\n");
    printf("rescale_q(90000, 1/90000, 1/1000) = %"PRId64"\n",
           av_rescale_q(90000, (AVRational){1, 90000}, (AVRational){1, 1000}));
    printf("rescale_q(48000, 1/48000, 1/44100) = %"PRId64"\n",
           av_rescale_q(48000, (AVRational){1, 48000}, (AVRational){1, 44100}));

    /* av_compare_ts */
    printf("\nTesting av_compare_ts()\n");
    printf("compare(1, 1/1, 1, 1/1) = %d\n",
           av_compare_ts(1, (AVRational){1, 1}, 1, (AVRational){1, 1}));
    printf("compare(1, 1/1, 2, 1/1) = %d\n",
           av_compare_ts(1, (AVRational){1, 1}, 2, (AVRational){1, 1}));
    printf("compare(2, 1/1, 1, 1/1) = %d\n",
           av_compare_ts(2, (AVRational){1, 1}, 1, (AVRational){1, 1}));
    printf("compare(1, 1/1000, 1, 1/90000) = %d\n",
           av_compare_ts(1, (AVRational){1, 1000}, 1, (AVRational){1, 90000}));
    /* large values trigger rescale-based comparison path */
    printf("compare(INT64_MAX/2, 1/1, INT64_MAX/3, 1/1) = %d\n",
           av_compare_ts(INT64_MAX / 2, (AVRational){1, 1},
                         INT64_MAX / 3, (AVRational){1, 1}));

    /* av_compare_mod */
    printf("\nTesting av_compare_mod()\n");
    printf("compare_mod(3, 1, 16) = %"PRId64"\n", av_compare_mod(3, 1, 16));
    printf("compare_mod(1, 3, 16) = %"PRId64"\n", av_compare_mod(1, 3, 16));
    printf("compare_mod(5, 5, 16) = %"PRId64"\n", av_compare_mod(5, 5, 16));

    /* av_rescale_delta */
    printf("\nTesting av_rescale_delta()\n");
    last = AV_NOPTS_VALUE;
    for (int i = 0; i < 4; i++)
        printf("delta step %d: %"PRId64"\n", i,
               av_rescale_delta((AVRational){1, 48000}, i * 1024,
                                (AVRational){1, 48000}, 1024,
                                &last, (AVRational){1, 44100}));
    /* trigger clip-based path: use different in_tb and fs_tb */
    last = AV_NOPTS_VALUE;
    for (int i = 0; i < 4; i++)
        printf("delta clip %d: %"PRId64"\n", i,
               av_rescale_delta((AVRational){1, 44100}, i * 940,
                                (AVRational){1, 48000}, 1024,
                                &last, (AVRational){1, 90000}));

    /* av_add_stable */
    printf("\nTesting av_add_stable()\n");
    printf("add_stable(0, 1/1, 1/1000, 500) = %"PRId64"\n",
           av_add_stable((AVRational){1, 1}, 0, (AVRational){1, 1000}, 500));
    printf("add_stable(1000, 1/90000, 1/48000, 1024) = %"PRId64"\n",
           av_add_stable((AVRational){1, 90000}, 1000, (AVRational){1, 48000}, 1024));
    /* non-exact division path (m >= d, general case) */
    printf("add_stable(0, 1/48000, 1/90000, 90000) = %"PRId64"\n",
           av_add_stable((AVRational){1, 48000}, 0, (AVRational){1, 90000}, 90000));
    printf("add_stable(100, 1/1000, 1/90000, 3000) = %"PRId64"\n",
           av_add_stable((AVRational){1, 1000}, 100, (AVRational){1, 90000}, 3000));
    /* repeated addition: verify no rounding error accumulation */
    {
        int64_t ts = 0;
        for (int i = 0; i < 10000; i++)
            ts = av_add_stable((AVRational){1, 48000}, ts, (AVRational){1, 48000}, 1024);
        printf("add_stable 10000x1024 at 1/48000: %"PRId64" (expected %"PRId64") %s\n",
               ts, (int64_t)10000 * 1024,
               ts == (int64_t)10000 * 1024 ? "OK" : "FAIL");
    }

    return 0;
}