microPython的源码解析之 objcomplex.c

MicroPython 是一种适用于微控制器和其他受限环境的 Python 编程语言的实现。它旨在提供与 Python 3 语言的紧密兼容，同时考虑到内存和计算资源的限制。MicroPython 库是这门语言的核心组成部分，提供了一系列的模块和函数，使得开发者能够在硬件上执行各种任务。
下面将通过系列文章,逐一解读microPython,以便让读者了解掌握microPython的整个核心逻辑.,以便读者可以透过这个Python的最小内核,掌握Python解析器的核心实现逻辑,学习世界上最优秀的源码设计之一.

microPython Python最小内核源码解析

#include <stdlib.h> // 标准库，提供内存分配、随机数等函数
#include <stdio.h>  // 标准输入输出库
#include <assert.h> // 断言库，用于程序调试

#include "py/parsenum.h" // MicroPython的数字解析模块
#include "py/runtime.h"  // MicroPython的运行时环境

#if MICROPY_PY_BUILTINS_COMPLEX // 如果MicroPython支持内置的复数类型

#include <math.h> // 数学库，提供数学函数
#include "py/formatfloat.h" // MicroPython的浮点数格式化模块

// 定义复数对象的结构体
typedef struct _mp_obj_complex_t {
   
    mp_obj_base_t base; // 基础对象结构体
    mp_float_t real; // 复数的实部
    mp_float_t imag; // 复数的虚部
} mp_obj_complex_t;

// 复数对象的打印函数
static void complex_print(const mp_print_t *print, mp_obj_t o_in, mp_print_kind_t kind) {
   
    (void)kind; // 打印时不需要关心打印类型
    mp_obj_complex_t *o = MP_OBJ_TO_PTR(o_in); // 将输入的对象指针转换为复数对象指针
    // 根据浮点数实现的不同，选择不同长度的缓冲区和精度
    #if MICROPY_FLOAT_IMPL == MICROPY_FLOAT_IMPL_FLOAT
    char buf[16];
    #if MICROPY_OBJ_REPR == MICROPY_OBJ_REPR_C
    const int precision = 6;
    #else
    const int precision = 7;
    #endif
    #else
    char buf[32];
    const int precision = 16;
    #endif
    // 如果实部为0，只打印虚部
    if (o->real == 0) {
   
        mp_format_float(o->imag, buf, sizeof(buf), 'g', precision, '\0'); // 格式化虚部
        mp_printf(print, "%sj", buf); // 打印虚部和'j'
    } else {
   
        mp_format_float(o->real, buf, sizeof(buf), 'g', precision, '\0'); // 格式化实部
        mp_printf(print, "(%s", buf); // 打印实部和'('
        // 如果虚部大于等于0或为NaN，打印'+'
        if (o->imag >= 0 || isnan(o->imag)) {
   
            mp_print_str(print, "+");
        }
        mp_format_float(o->imag, buf, sizeof(buf), 'g', precision, '\0'); // 格式化虚部
        mp_printf(print, "%sj)", buf); // 打印虚部和')'
    }
}

// 创建新的复数对象
static mp_obj_t complex_make_new(const mp_obj_type_t *type_in, size_t n_args, size_t n_kw, const mp_obj_t *args) {
   
    (void)type_in; // 类型参数未使用
    mp_arg_check_num(n_args, n_kw, 0, 2, false); // 检查参数数量，0到2个位置参数，不接受关键字参数

    // 根据参数数量进行不同的处理
    switch (n_args) {
   
        case 0:
            return mp_obj_new_complex(0, 0); // 如果没有参数，创建一个实部和虚部都为0的复数对象

        case 1:
            // 如果第一个参数是字符串，解析它
            if (mp_obj_is_str(args[0])) {
   
                size_t l;
                const char *s = mp_obj_str_get_data(args[0], &l);
                return mp_parse_num_complex(s, l, NULL);
            } else if (mp_obj_is_type(args[0], &mp_type_complex)) {
   
                // 如果参数已经是复数类型，直接返回
                return args[0];
            } else {
   
                mp_float_t real, imag;
                mp_obj_get_complex(args[0], &real, &imag); // 获取复数的实部和虚部
                return mp_obj_new_complex(real, imag); // 创建新的复数对象
            }

        case 2:
        default: {
   
            mp_float_t real, imag;
            // 如果第一个参数是复数类型，获取它的实部和虚部
            if (mp_obj_is_type(args[0], &mp_type_complex)) {
   
                mp_obj_complex_get(args[0]<