arith_seqprod.c 文件参考

#include "../../../include/lammp/impl/ele_mul.h"
#include "../../../include/lammp/impl/mparam.h"
#include "../../../include/lammp/impl/tmp_alloc.h"
#include "../../../include/lammp/impl/longlong.h"
#include "../../../include/lammp/lmmpn.h"
#include "../../../include/lammp/numth.h"

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

浏览源代码.

函数
static mp_size_t	_odd_nPr_ (mp_ptr restrict dst, mp_size_t rn, ulong n, ulong r)
static mp_size_t	_odd_pow_ (mp_ptr restrict dst, mp_size_t rn, uint base, ulong exp)
mp_size_t	lmmp_arith_seqprod_ (mp_ptr restrict dst, mp_size_t rn, uint x, uint n, uint m)
mp_size_t	lmmp_arith_seqprod_size_ (uint x, uint n, uint m)
	Copyright (C) 2026 HJimmyK(Jericho Knox)
static mp_size_t	pow_nPr_ (mp_ptr restrict dst, mp_size_t rn, uint x, uint n, uint m)

函数说明

_odd_nPr_()

static mp_size_t _odd_nPr_ ( mp_ptr restrict  dst, mp_size_t  rn, ulong  n, ulong  r  )

inlinestatic

在文件 arith_seqprod.c 第 47 行定义.

                                                                                       {
    if (n <= NPR_SHORT_LIMIT) {
        return lmmp_odd_nPr_ushort_(dst, rn, n, r);
    } else {
        return lmmp_odd_nPr_uint_(dst, rn, n, r);
    }
}

引用了 lmmp_odd_nPr_uint_(), lmmp_odd_nPr_ushort_(), n , 以及 NPR_SHORT_LIMIT.

被这些函数引用 pow_nPr_().

函数调用图:

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

_odd_pow_()

static mp_size_t _odd_pow_ ( mp_ptr restrict  dst, mp_size_t  rn, uint  base, ulong  exp  )

inlinestatic

在文件 arith_seqprod.c 第 36 行定义.

                                                                                           {
    if (base <= 0xf)
        return lmmp_u4_pow_1_(dst, rn, base, exp);
    else if (base <= MP_UCHAR_MAX)
        return lmmp_u8_pow_1_(dst, rn, base, exp);
    else if (base <= MP_USHORT_MAX)
        return lmmp_u16_pow_1_(dst, rn, base, exp);
    else
        return lmmp_u32_pow_1_(dst, rn, base, exp);
}

引用了 lmmp_u16_pow_1_(), lmmp_u32_pow_1_(), lmmp_u4_pow_1_(), lmmp_u8_pow_1_(), MP_UCHAR_MAX, MP_USHORT_MAX , 以及 n.

被这些函数引用 pow_nPr_().

函数调用图:

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

lmmp_arith_seqprod_()

mp_size_t lmmp_arith_seqprod_	(	mp_ptr restrict	dst,
		mp_size_t	rn,
		uint	x,
		uint	n,
		uint	m
	)

在文件 arith_seqprod.c 第 100 行定义.

                                                                                         {
    lmmp_param_assert(dst != NULL);
    lmmp_param_assert(rn >= 1);
    lmmp_param_assert(x >= 1);
    lmmp_param_assert(n >= 1);
    lmmp_param_assert(m > 1);
 
    if (x % m == 0) {
        return pow_nPr_(dst, rn, x, n, m);
    }
 
    mp_bitcnt_t bits = 0;
    while ((x & 1) == 0 && (m & 1) == 0) {
        x >>= 1;
        m >>= 1;
        bits++;
    }
    bits *= n + 1;
 
    TEMP_DECL;
    ulongp restrict limbs = TALLOC_TYPE((n + 1) / 2 + 1, ulong);
    mp_size_t limbn = 0;
    ulong t = 1, s, v;
    mp_bitcnt_t cnt;
    for (uint i = 0; i <= n; i++) {
        s = x + i * m;
        ctz_shr_u64(v, s, cnt);
        t *= v;
        bits += cnt;
        if (t > MP_UINT_MAX) {
            limbs[limbn++] = t;
            t = 1;
        }
    }
    ctz_shr_u64(v, t, cnt);
    bits += cnt;
    if (v != 1) {
        limbs[limbn++] = v;
    }
    mp_size_t shw = bits / LIMB_BITS;
    bits %= LIMB_BITS;
    mp_ptr restrict b = TALLOC_TYPE(limbn * 2, mp_limb_t);
    mp_size_t bn = lmmp_elem_mul_ulong_(b, limbs, limbn, b + limbn);
    lmmp_zero(dst, shw);
    if (bits > 0) {
        dst[shw + bn] = lmmp_shl_(dst + shw, b, bn, bits);
        rn = bn + shw + 1;
        rn -= dst[rn - 1] == 0 ? 1 : 0;
    } else {
        lmmp_copy(dst + shw, b, bn);
        rn = bn + shw;
    }
    TEMP_FREE;
    return rn;
}

引用了 bn, ctz_shr_u64, LIMB_BITS, lmmp_copy, lmmp_elem_mul_ulong_(), lmmp_param_assert, lmmp_shl_(), lmmp_zero, MP_UINT_MAX, n, pow_nPr_(), s, t, TALLOC_TYPE, TEMP_DECL , 以及 TEMP_FREE.

函数调用图:

lmmp_arith_seqprod_size_()

mp_size_t lmmp_arith_seqprod_size_	(	uint	x,
		uint	n,
		uint	m
	)

Copyright (C) 2026 HJimmyK(Jericho Knox)

计算等差数列乘积 x(x+m)...(x+n*m)的 limb 缓冲区长度

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/.

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

                                                           {
    lmmp_param_assert(x > 0);
    lmmp_param_assert(n > 0);
    lmmp_param_assert(m > 1);
    // x(x+m)(x+2m)...(x+nm) <= (x+m*n/2)^(n+1)
    ulong t = (x + ((ulong)m * n + 1) / 2);
    mp_size_t rn1 = lmmp_pow_1_size_(t, n + 1);
    // 共有n+1个数，每个数的位数最多为uint，一个limb最多可以容纳两个uint乘积
    mp_size_t rn2 = (n + 1) / 2 + 1;
    return LMMP_MIN(rn1, rn2);
}

引用了 LMMP_MIN, lmmp_param_assert, lmmp_pow_1_size_(), n , 以及 t.

函数调用图:

pow_nPr_()

static mp_size_t pow_nPr_ ( mp_ptr restrict  dst, mp_size_t  rn, uint  x, uint  n, uint  m  )

inlinestatic

在文件 arith_seqprod.c 第 61 行定义.

                                                                                            {
    TEMP_DECL;
    uint p = x / m;
 
    mp_bitcnt_t bits;
    mp_size_t tn = lmmp_nPr_size_(p + n, n + 1, &bits);
    tn -= bits / LIMB_BITS;
    mp_ptr restrict tp = TALLOC_TYPE(tn, mp_limb_t);
    tn = _odd_nPr_(tp, tn, p + n, n + 1);
 
    mp_size_t tz = lmmp_tailing_zeros_(m);
    m >>= tz;
    mp_size_t mpown = lmmp_pow_1_size_(m, n + 1);
    mp_ptr restrict mpow = TALLOC_TYPE(mpown, mp_limb_t);
    mpown = _odd_pow_(mpow, mpown, m, n + 1);
 
    bits += tz * (n + 1);
    mp_size_t shw = bits / LIMB_BITS;
    bits %= LIMB_BITS;
 
    lmmp_zero(dst, shw);
    if (tn >= mpown)
        lmmp_mul_(dst + shw, tp, tn, mpow, mpown);
    else
        lmmp_mul_(dst + shw, mpow, mpown, tp, tn);
 
    rn = tn + mpown;
    rn -= dst[shw + rn - 1] == 0 ? 1 : 0;
    if (bits > 0) {
        dst[shw + rn] = lmmp_shl_(dst + shw, dst + shw, rn, bits);
        rn += shw + 1;
        rn -= dst[rn - 1] == 0 ? 1 : 0;
    } else {
        rn += shw;
    }
    TEMP_FREE;
    return rn;
}

引用了 _odd_nPr_(), _odd_pow_(), LIMB_BITS, lmmp_mul_(), lmmp_nPr_size_(), lmmp_pow_1_size_(), lmmp_shl_(), lmmp_tailing_zeros_, lmmp_zero, n, TALLOC_TYPE, TEMP_DECL, TEMP_FREE , 以及 tp.

被这些函数引用 lmmp_arith_seqprod_().

函数调用图:

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