You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
3.4 KiB
72 lines
3.4 KiB
import ctypes, functools, sys
|
|
from typing import TYPE_CHECKING
|
|
|
|
def _do_ioctl(__idir, __base, __nr, __struct, __fd, **kwargs):
|
|
import tinygrad.runtime.support.hcq as hcq, fcntl
|
|
ioctl = __fd.ioctl if isinstance(__fd, hcq.FileIOInterface) else functools.partial(fcntl.ioctl, __fd)
|
|
if (rc:=ioctl((__idir<<30)|(ctypes.sizeof(out:=__struct(**kwargs))<<16)|(__base<<8)|__nr, out)): raise RuntimeError(f"ioctl returned {rc}")
|
|
return out
|
|
|
|
def _IO(base, nr): return functools.partial(_do_ioctl, 0, ord(base) if isinstance(base, str) else base, nr, None)
|
|
def _IOW(base, nr, typ): return functools.partial(_do_ioctl, 1, ord(base) if isinstance(base, str) else base, nr, typ)
|
|
def _IOR(base, nr, typ): return functools.partial(_do_ioctl, 2, ord(base) if isinstance(base, str) else base, nr, typ)
|
|
def _IOWR(base, nr, typ): return functools.partial(_do_ioctl, 3, ord(base) if isinstance(base, str) else base, nr, typ)
|
|
|
|
def CEnum(typ: type[ctypes._SimpleCData]):
|
|
class _CEnum(typ): # type: ignore
|
|
_val_to_name_: dict[int,str] = {}
|
|
|
|
@classmethod
|
|
def from_param(cls, val): return val if isinstance(val, cls) else cls(val)
|
|
@classmethod
|
|
def get(cls, val, default="unknown"): return cls._val_to_name_.get(val.value if isinstance(val, cls) else val, default)
|
|
@classmethod
|
|
def items(cls): return cls._val_to_name_.items()
|
|
@classmethod
|
|
def define(cls, name, val):
|
|
cls._val_to_name_[val] = name
|
|
return val
|
|
|
|
def __eq__(self, other): return self.value == other
|
|
def __repr__(self): return self.get(self) if self.value in self.__class__._val_to_name_ else str(self.value)
|
|
|
|
return _CEnum
|
|
|
|
# supports gcc (C11) __attribute__((packed))
|
|
if TYPE_CHECKING: Struct = ctypes.Structure
|
|
else:
|
|
class MetaStruct(type(ctypes.Structure)):
|
|
def __new__(mcs, name, bases, dct):
|
|
fields = dct.pop("_fields_", None)
|
|
cls = super().__new__(mcs, name, bases, dct)
|
|
if dct.get("_packed_", False) and fields is not None: mcs._build(cls, fields)
|
|
return cls
|
|
|
|
def __setattr__(cls, k, v):
|
|
# NB: _fields_ must be set after _packed_ because PyCStructType_setattro marks _fields_ as final.
|
|
if k == "_fields_" and getattr(cls, "_packed_", False): type(cls)._build(cls, v)
|
|
elif k == "_packed_" and hasattr(cls, "_fields_"): type(cls)._build(cls, cls._fields_)
|
|
else: super().__setattr__(k, v)
|
|
|
|
@staticmethod
|
|
def _build(cls, fields):
|
|
o = 0
|
|
for n,t,b in [(f[0], f[1], f[2] if len(f) == 3 else 0) for f in fields]:
|
|
if b == 0: o = (o + 7) & ~7
|
|
m = (1 << (sz:=ctypes.sizeof(t)*8 if b == 0 else b)) - 1
|
|
def _s(self,v,m,s,b): self._data[:] = ((int.from_bytes(self._data,sys.byteorder)&~(m<<s))|((v&m)<<s)).to_bytes(len(self._data), sys.byteorder)
|
|
setattr(cls, n, property(functools.partial(lambda self,m,s:(int.from_bytes(self._data,sys.byteorder)>>s)&m,m=m,s=o),
|
|
functools.partial(_s,m=m,s=o,b=b)))
|
|
o += sz
|
|
|
|
type(ctypes.Structure).__setattr__(cls, '_fields_', [('_data', ctypes.c_ubyte * ((o + 7) // 8))])
|
|
type(ctypes.Structure).__setattr__(cls, '_packed_', True)
|
|
setattr(cls, '_packed_fields_', fields)
|
|
|
|
class Struct(ctypes.Structure, metaclass=MetaStruct):
|
|
def __init__(self, *args, **kwargs):
|
|
if hasattr(self, '_packed_fields_'):
|
|
for f,v in zip(self._packed_fields_, args): setattr(self, f[0], v)
|
|
for k,v in kwargs.items(): setattr(self, k, v)
|
|
else: super().__init__(*args, **kwargs)
|
|
|
|
|