multinomial.c 文件参考

#include "../../../include/lammp/impl/ele_mul.h"
#include "../../../include/lammp/impl/inlines.h"
#include "../../../include/lammp/impl/lglg.h"
#include "../../../include/lammp/impl/longlong.h"
#include "../../../include/lammp/impl/mparam.h"
#include "../../../include/lammp/impl/prime_table.h"

multinomial.c 的引用(Include)关系图:

浏览源代码.

宏定义
#define	MULTINOMIAL_INT_LIMIT   (0xffffffff)
#define	MULTINOMIAL_SHORT_LIMIT   (0xffff)

函数
static uint	count_factors (fac_ptr fac, uint nfactors, uint n, const uintp r, uint m, uint p)
static mp_size_t	fac_size_bigger (uint n)
static mp_size_t	fac_size_lower (uint n)
	Copyright (C) 2026 HJimmyK(Jericho Knox)
static uint	factor_size_int (mp_size_t rn, uint n)
static ushort	factor_size_short (mp_size_t rn)
mp_size_t	lmmp_multinomial_ (mp_ptr restrict dst, mp_size_t rn, uint n, const uintp restrict r, uint m)
mp_size_t	lmmp_multinomial_size_ (const uintp r, uint m, ulong *restrict n)
static mp_size_t	lmmp_odd_multinomial_uint_ (mp_ptr restrict dst, mp_size_t rn, uint n, const uintp restrict r, uint m)
static mp_size_t	lmmp_odd_multinomial_ushort_ (mp_ptr restrict dst, mp_size_t rn, uint n, const uintp restrict r, uint m)

宏定义说明

MULTINOMIAL_INT_LIMIT

#define MULTINOMIAL_INT_LIMIT   (0xffffffff)

在文件 multinomial.c 第 153 行定义.

MULTINOMIAL_SHORT_LIMIT

#define MULTINOMIAL_SHORT_LIMIT   (0xffff)

在文件 multinomial.c 第 152 行定义.

函数说明

count_factors()

static uint count_factors ( fac_ptr  fac, uint  nfactors, uint  n, const uintp  r, uint  m, uint  p  )

inlinestatic

在文件 multinomial.c 第 57 行定义.

                                                                                                    {
    uint pn = n;
    uint e = 0;
    ulong inv = MP_ULONG_MAX / p + 1;
    while (pn > 0) {
        _udiv32by32_q_preinv(pn, pn, inv);
        e += pn;
    }
    for (uint i = 0; i < m; ++i) {
        pn = r[i];
        while (pn > 0) {
            _udiv32by32_q_preinv(pn, pn, inv);
            e -= pn;
        }
    }
    if (e > 0) {
        fac[nfactors].f = p;
        fac[nfactors++].j = e;
    }
    return nfactors;
}

引用了 _udiv32by32_q_preinv, MP_ULONG_MAX , 以及 n.

被这些函数引用 lmmp_odd_multinomial_uint_() , 以及 lmmp_odd_multinomial_ushort_().

这是这个函数的调用关系图:

fac_size_bigger()

static mp_size_t fac_size_bigger ( uint  n )

inlinestatic

在文件 multinomial.c 第 32 行定义.

                                                {
    if (n < 20) {
        return 64;
    } else {
        return log2_fac_ceil(n);
    }
}

引用了 log2_fac_ceil() , 以及 n.

被这些函数引用 lmmp_multinomial_size_().

函数调用图:

这是这个函数的调用关系图:

fac_size_lower()

static mp_size_t fac_size_lower ( uint  n )

inlinestatic

Copyright (C) 2026 HJimmyK(Jericho Knox)

This file is part of LAMMP.

LAMMP is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License (LGPL) as published by the Free Software Foundation; either version 3 of the License, or (at your option) any later version.

This program is distributed WITHOUT ANY WARRANTY.

See https://www.gnu.org/licenses/.

在文件 multinomial.c 第 24 行定义.

                                               {
    if (n < 20) {
        return 0;
    } else {
        return log2_fac_floor(n);
    }
}

引用了 log2_fac_floor() , 以及 n.

被这些函数引用 lmmp_multinomial_size_().

函数调用图:

这是这个函数的调用关系图:

factor_size_int()

static uint factor_size_int ( mp_size_t  rn, uint  n  )

inlinestatic

在文件 multinomial.c 第 79 行定义.

                                                         {
    /*
     使用类似组合数的思路来估计缓冲区大小。
    */
    ulong approx1 = rn * 8;
    lmmp_debug_assert(approx1 <= MP_UINT_MAX);
    ulong approx2 = lmmp_prime_size_(n);
    return approx1 < approx2 ? approx1 : approx2;
}

引用了 lmmp_debug_assert, lmmp_prime_size_(), MP_UINT_MAX , 以及 n.

被这些函数引用 lmmp_odd_multinomial_uint_().

函数调用图:

这是这个函数的调用关系图:

factor_size_short()

static ushort factor_size_short ( mp_size_t  rn )

inlinestatic

在文件 multinomial.c 第 89 行定义.

                                                     {
    ushort approx1 = rn * 8;
    lmmp_debug_assert(approx1 <= MP_USHORT_MAX);
    return approx1;
}

引用了 lmmp_debug_assert, MP_USHORT_MAX , 以及 n.

被这些函数引用 lmmp_odd_multinomial_ushort_().

这是这个函数的调用关系图:

lmmp_multinomial_()

mp_size_t lmmp_multinomial_	(	mp_ptr restrict	dst,
		mp_size_t	rn,
		uint	n,
		const uintp restrict	r,
		uint	m
	)

在文件 multinomial.c 第 155 行定义.

                                                                                                       {
    mp_size_t shl = n - lmmp_limb_popcnt_(n);
    for (uint j = 0; j < m; ++j) {
        shl += lmmp_limb_popcnt_(r[j]);
        shl -= r[j];
    }
    mp_size_t shw = shl / LIMB_BITS;
    shl %= LIMB_BITS;
    lmmp_zero(dst, shw);
    if (n <= MULTINOMIAL_SHORT_LIMIT) {
        rn = lmmp_odd_multinomial_ushort_(dst + shw, rn - shw, n, r, m);
    } else {
        rn = lmmp_odd_multinomial_uint_(dst + shw, rn - shw, n, r, m);
    }
 
    if (shl > 0) {
        dst[shw + rn] = lmmp_shl_(dst + shw, dst + shw, rn, shl);
        rn += shw + 1;
        rn -= dst[rn - 1] == 0 ? 1 : 0;
    } else {
        rn += shw;
    }
    return rn;
}

引用了 LIMB_BITS, lmmp_limb_popcnt_, lmmp_odd_multinomial_uint_(), lmmp_odd_multinomial_ushort_(), lmmp_shl_(), lmmp_zero, MULTINOMIAL_SHORT_LIMIT , 以及 n.

函数调用图:

lmmp_multinomial_size_()

mp_size_t lmmp_multinomial_size_	(	const uintp	r,
		uint	m,
		ulong *restrict	n
	)

在文件 multinomial.c 第 40 行定义.

                                                                           {
    ulong n_ret = 0;
    uint i = 0;
    for (; i < m; ++i) n_ret += r[i];
 
    lmmp_param_assert(n_ret <= MP_UINT_MAX);
    lmmp_param_assert(n_ret > 0);
    *n = n_ret;
    mp_size_t rn = fac_size_bigger(n_ret);
 
    for (i = 0; i < m; ++i) {
        rn -= fac_size_lower(r[i]);
    }
    rn = (rn + LIMB_BITS - 1) / LIMB_BITS;
    return rn + 2;
}

引用了 fac_size_bigger(), fac_size_lower(), LIMB_BITS, lmmp_param_assert, MP_UINT_MAX , 以及 n.

函数调用图:

lmmp_odd_multinomial_uint_()

static mp_size_t lmmp_odd_multinomial_uint_ ( mp_ptr restrict  dst, mp_size_t  rn, uint  n, const uintp restrict  r, uint  m  )

static

在文件 multinomial.c 第 129 行定义.

                                                                                                                       {
    lmmp_prime_int_table_init_(n);
    TEMP_B_DECL;
    uint nfactors = factor_size_int(rn, n);
    fac_ptr restrict fac = BALLOC_TYPE(nfactors, fac_t);
 
    nfactors = 0;
    prime_cache_t cache;
    lmmp_prime_cache_init_(&cache, n);
    while (cache.is_end == 0) {
        lmmp_prime_cache_next_(&cache);
        for (uint i = 0; i < cache.size; ++i) {
            nfactors = count_factors(fac, nfactors, n, r, m, cache.pp[i]);
        }
    }
    lmmp_prime_cache_free_(&cache);
 
    rn = lmmp_factors_mul_(dst, rn, fac, nfactors);
 
    TEMP_B_FREE;
    return rn;
}

引用了 BALLOC_TYPE, count_factors(), factor_size_int(), lmmp_factors_mul_(), lmmp_prime_cache_free_(), lmmp_prime_cache_init_(), lmmp_prime_cache_next_(), lmmp_prime_int_table_init_(), n, TEMP_B_DECL , 以及 TEMP_B_FREE.

被这些函数引用 lmmp_multinomial_().

函数调用图:

这是这个函数的调用关系图:

lmmp_odd_multinomial_ushort_()

static mp_size_t lmmp_odd_multinomial_ushort_ ( mp_ptr restrict  dst, mp_size_t  rn, uint  n, const uintp restrict  r, uint  m  )

static

在文件 multinomial.c 第 95 行定义.

  {
    if (n < ODD_FACTORIAL_SIZE) {
        lmmp_odd_nPr_ushort_(dst, rn, n, n);
        mp_limb_t t = 0;
        for (uint i = 0; i < m; ++i) {
            lmmp_odd_nPr_ushort_(&t, 1, r[i], r[i]);
            dst[0] /= t;
        }
        return 1;
    } else {
        TEMP_DECL;
        ushort primen = lmmp_prime_cnt16_(n);
        ushort nfactors = factor_size_short(rn);
        nfactors = primen < nfactors ? primen : nfactors;
        fac_ptr restrict fac = TALLOC_TYPE(nfactors, fac_t);
        nfactors = 0;
        for (ushort i = 1; i < primen; ++i) {
            ushort p = prime_short_table[i];
            nfactors = count_factors(fac, nfactors, n, r, m, p);
        }
 
        rn = lmmp_factors_mul_ushort_(dst, rn, fac, nfactors);
 
        TEMP_FREE;
        return rn;
    }
}

引用了 count_factors(), factor_size_short(), lmmp_factors_mul_ushort_(), lmmp_odd_nPr_ushort_(), lmmp_prime_cnt16_(), n, ODD_FACTORIAL_SIZE, prime_short_table, t, TALLOC_TYPE, TEMP_DECL , 以及 TEMP_FREE.

被这些函数引用 lmmp_multinomial_().

函数调用图:

这是这个函数的调用关系图: