在Python中模拟uint32_t?

时间:2013-10-07 19:33:00

标签: python

我试图将函数从C移植到Python并使其易于调试,我更喜欢它执行相同的CPU字大小限制操作,因此我可以比较中间结果。换句话说,我想要像:

a = UnsignedBoundedInt(32, 399999)
b = UnsignedBoundedInt(32, 399999)
print(a*b) # prints 1085410049 (159999200001 % 2**32)

实现这一目标的最佳方法是什么,以便所有操作(包括按位移位)都能像C一样工作?

2 个答案:

答案 0 :(得分:13)

您可以尝试使用ctypes.uint_32为您绑定结果:

>>> import ctypes
>>> print ctypes.c_uint32(399999 * 399999).value
1085410049

或者,您可以使用numpy的数据类型:

>>> import numpy as np
>>> a = np.uint32(399999)
>>> b = np.uint32(399999)
>>> a * b
__main__:1: RuntimeWarning: overflow encountered in uint_scalars
1085410049

答案 1 :(得分:1)

这是一个有趣的解决方案,虽然它只适用于Python 2:

class U32:
    """Emulates 32-bit unsigned int known from C programming language."""

    def __init__(self, num=0, base=None):
        """Creates the U32 object.

        Args:
            num: the integer/string to use as the initial state
            base: the base of the integer use if the num given was a string
        """
        if base is None:
            self.int_ = int(num) % 2**32
        else:
            self.int_ = int(num, base) % 2**32

    def __coerce__(self, ignored):
        return None

    def __str__(self):
        return "<U32 instance at 0x%x, int=%d>" % (id(self), self.int_)

    def __getattr__(self, attribute_name):
        print("getattr called, attribute_name=%s" % attribute_name)
        # you might want to take a look here:
        # https://stackoverflow.com/q/19611001/1091116
        r = getattr(self.int_, attribute_name)
        if callable(r):  # return a wrapper if integer's function was requested
            def f(*args, **kwargs):
                if args and isinstance(args[0], U32):
                    args = (args[0].int_, ) + args[1:]
                ret = r(*args, **kwargs)
                if ret is NotImplemented:
                    return ret
                if attribute_name in ['__str__', '__repr__', '__index__']:
                    return ret
                ret %= 2**32
                return U32(ret)
            return f
        return r

print(U32(4) / 2)
print(4 / U32(2))
print(U32(4) / U32(2))

对于Python 3兼容性,请查看here