MicroPython模块中带参数类方法的添加方法咨询

嘿，我来帮你理清这个问题——在MicroPython里给本地类添加带额外参数的方法其实有清晰的套路，你之前的困惑主要是没搞清楚不同函数对象宏的用法和函数签名的对应关系。下面我一步步给你拆解：

核心原则：方法的第一个参数永远是self #

不管你用哪种宏定义方法，你的C函数第一个参数必须是mp_obj_t self（也就是类的实例对象），后续才是你需要的额外参数。不同的宏只是用来适配**参数总数（包括self）**的不同情况。

情况1：固定数量的额外参数 #

如果你的方法需要接收固定个数的额外参数，比如1个或2个，那就用MP_DEFINE_CONST_FUN_OBJ_N系列宏，这里的N代表函数接收的参数总数（包括self）。

举个例子，假设你要给类加一个接收1个整数参数的方法：

1. 先写C函数实现：

// self是类实例，arg1是额外传入的参数
static mp_obj_t my_class_double(mp_obj_t self_in, mp_obj_t arg1_in) {
    // 如果需要操作类内部的数据，先把self转成你的类结构体指针
    // my_class_obj_t *self = MP_OBJ_TO_PTR(self_in);
    
    // 把传入的参数转成整数（根据你的需求转换类型）
    mp_int_t num = mp_obj_get_int(arg1_in);
    
    // 这里写你的业务逻辑，比如返回输入值的两倍
    return mp_obj_new_int(num * 2);
}

2. 用对应数量的宏定义函数对象：
   因为这个方法接收2个参数（self + 1个额外参数），所以用MP_DEFINE_CONST_FUN_OBJ_2：

STATIC MP_DEFINE_CONST_FUN_OBJ_2(my_class_double_obj, my_class_double);

3. 把方法添加到类的本地字典中：

STATIC const mp_rom_map_elem_t my_class_locals_dict_table[] = {
    { MP_ROM_QSTR(MP_QSTR_double), MP_ROM_PTR(&my_class_double_obj) },
    // 其他方法或属性...
};
STATIC MP_DEFINE_CONST_DICT(my_class_locals_dict, my_class_locals_dict_table);

情况2：可变数量的额外参数 #

如果你的方法需要接收不确定数量的参数，那就用MP_DEFINE_CONST_FUN_OBJ_VAR宏。这个宏的第一个参数是最少需要的参数个数（包括self），函数签名会多两个参数：size_t n_args（实际传入的总参数数）和mp_obj_t *args（参数数组，args[0]就是self）。

比如写一个可以接收任意数量整数并求和的方法：

1. 函数实现：

static mp_obj_t my_class_sum(mp_obj_t self_in, size_t n_args, mp_obj_t *args) {
    mp_int_t total = 0;
    // 从args[1]开始遍历，因为args[0]是self
    for (size_t i = 1; i < n_args; i++) {
        total += mp_obj_get_int(args[i]);
    }
    return mp_obj_new_int(total);
}

2. 定义函数对象：
   这里最少需要1个参数（仅self，允许不传额外参数），所以宏的第一个参数写1：

STATIC MP_DEFINE_CONST_FUN_OBJ_VAR(my_class_sum_obj, 1, my_class_sum);

3. 同样把它加到类的本地字典里即可。

你之前踩的坑解释 #

• 用MP_DEFINE_CONST_FUN_OBJ_1报错：这个宏对应的函数只能接收1个参数（仅self），但你写的函数有2个参数（self + 额外参数），参数数量不匹配，所以报错。
• 对MP_DEFINE_CONST_FUN_OBJ_VAR的误解：宏的第一个int参数是最少参数个数（不是函数的第一个参数类型！），函数本身的第一个参数依然是mp_obj_t self，完全符合本地方法的要求。

额外提示：关键字参数 #

如果你的方法需要支持关键字参数，可以用MP_DEFINE_CONST_FUN_OBJ_KW宏，不过这个稍复杂，需要处理关键字参数的解析，等你把固定和可变参数的用法摸熟了再尝试也不迟。

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