Raku通过NativeCall调用C的rmultinom函数返回数组结果不符问题

以下是可能导致异常的核心问题及对应解决方案：

1. 结果数组未正确初始化 #

C版rmultinom函数不会自动初始化传入的结果数组，仅会在已有内存上写入数值。如果Raku中创建的结果数组未初始化，内存中残留的垃圾值会被当作计算结果输出，甚至出现负数。

修复方式：
创建结果数组时显式初始化为全0：

my CArray[int32] $count .= new: [0 xx $k]; # $k为类别数量

2. 数据类型不匹配 #

Windows平台下R的rmultinom中，count参数是int*（32位整数），prob是double*（64位浮点数）。如果Raku中使用了错误的Native类型（比如用int64代替int32），会导致内存读写错位，引发异常值。

修复方式：
严格匹配C端类型定义：

# 正确绑定函数类型
sub rmultinom(int32 $n, int32 $k, CArray[num64] $prob, CArray[int32] $count) is native('R.dll') { * }

3. 随机数环境未初始化 #

R的随机数生成依赖自身的全局状态，如果Raku中未初始化R的随机数环境，rmultinom可能生成异常的随机结果。

修复方式：
调用rmultinom前先初始化R的随机数生成器（需绑定R的set.seed或R_init_Rmath函数）：

sub R_init_Rmath() is native('R.dll') { * }
R_init_Rmath(); # 初始化R的数学库环境

完整示例代码 #

use NativeCall;

# 绑定R的核心函数
sub R_init_Rmath() is native('R.dll') { * }
sub rmultinom(int32 $n, int32 $k, CArray[num64] $prob, CArray[int32] $count) is native('R.dll') { * }

sub raku_rmultinom(int $size, @prob) {
    # 初始化R数学环境
    R_init_Rmath();
    
    my $k = @prob.elems;
    # 初始化结果数组为全0
    my CArray[int32] $count .= new: [0 xx $k];
    # 转换概率数组为C端兼容类型
    my CArray[num64] $prob-c .= new: @prob;
    
    rmultinom($size, $k, $prob-c, $count);
    
    return $count.list;
}

# 测试调用（10个类别，每个概率0.1）
my @result = raku_rmultinom(10, [0.1 xx 10]);
say @result;

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