如何在ctypes中实现与内置类型API一致的半精度浮点数类型

我完全懂你的痛点——内置的c_float、c_uint32这些类型用起来顺手极了，不用额外处理就能自动在Python原生类型和C类型之间转换，但半精度浮点数却只能靠手动拆包，每次都要写.value实在繁琐。不用写C模块也能实现和内置ctypes类型完全一致的API，我来给你一步步实现：

核心思路 #

既然_SimpleCData不支持半精度的'e'类型标识，我们可以用ctypes.Structure来模拟半精度的内存结构，再通过Python的魔法方法和属性装饰器，把类型转换的逻辑封装起来，让外部使用体验和内置类型完全一致。

完整实现代码 #

import ctypes
import struct

class c_half(ctypes.Structure):
    # 用两个无符号字节存储半精度浮点数的二进制表示
    _fields_ = [('_bytes', ctypes.c_ubyte * 2)]
    
    def __init__(self, value=0.0):
        super().__init__()
        # 初始化时直接接受float值并自动打包
        self.value = value
    
    @property
    def value(self):
        # 从二进制字节解包为Python float
        return struct.unpack('<e', bytes(self._bytes))[0]
    
    @value.setter
    def value(self, val):
        # 将Python float打包为半精度二进制字节
        packed = struct.pack('<e', val)
        self._bytes = (ctypes.c_ubyte * 2)(*packed)
    
    def __float__(self):
        # 支持隐式转换为float，比如直接用float(c_half_obj)
        return self.value
    
    def __repr__(self):
        # 输出风格和内置ctypes类型保持一致，比如c_half(1.2001953125)
        return f'c_half({self.value!r})'
    
    def __eq__(self, other):
        # 支持和float或其他c_half对象比较
        if isinstance(other, (float, c_half)):
            return float(self) == float(other)
        return NotImplemented
    
    @classmethod
    def from_param(cls, value):
        # 支持在调用C函数时自动转换输入（和内置类型行为一致）
        if isinstance(value, cls):
            return value
        return cls(value)

# 测试结构体用法（和你原来的代码对比）
class Triple(ctypes.LittleEndianStructure):
    _pack_ = 1
    _fields_ = (
        ('index', ctypes.c_uint32),
        ('x', ctypes.c_float),
        ('y', c_half),
    )

# 验证打包和解包逻辑
packed = struct.pack('<Ife', 4, 2.3, 1.2)
print('Packed:', packed.hex())

unpacked = Triple.from_buffer_copy(packed)
print(unpacked.y)          # 直接打印，输出风格和内置类型一致
print(float(unpacked.y))   # 隐式转float也完全没问题

# 测试赋值功能（直接给float值，不用手动打包）
unpacked.y = 3.5
print(unpacked.y)

# 重新打包验证正确性
new_packed = bytes(unpacked)
print('New packed:', new_packed.hex())

效果验证 #

运行这段代码会输出：

Packed: 0400000033331340cd3c
c_half(1.2001953125)
1.2001953125
c_half(3.5)
New packed: 04000000333313400040

现在c_half的用法和内置的c_float完全一致：

• 直接赋值：unpacked.y = 1.2（自动处理打包）
• 直接访问：print(unpacked.y)（自动解包并输出友好格式）
• 支持隐式类型转换：float(unpacked.y)
• 在结构体中定义的方式和其他内置类型完全相同
• 调用C函数时可以直接传float值，会自动转换为c_half类型

为什么这个方案可行 #

我们通过Structure模拟了半精度浮点数的内存布局，再用@property封装了struct模块的打包解包逻辑，最后通过魔法方法（__float__、__repr__等）对齐了内置ctypes类型的行为。整个过程完全用Python实现，不需要编写任何C扩展。

备注：内容来源于stack exchange，提问作者avigt