mirror of
https://github.com/SickGear/SickGear.git
synced 2024-11-16 18:05:04 +00:00
65 lines
2 KiB
Python
65 lines
2 KiB
Python
|
from hachoir.field import (FieldError,
|
||
|
|
RawBits, RawBytes,
|
||
|
|
PaddingBits, PaddingBytes,
|
||
|
|
NullBits, NullBytes,
|
||
|
|
GenericString, GenericInteger)
|
||
|
|
from hachoir.stream import FileOutputStream
|
||
|
|
|
||
|
|
|
||
|
|
def createRawField(parent, size, name="raw[]", description=None):
|
||
|
|
if size <= 0:
|
||
|
|
raise FieldError("Unable to create raw field of %s bits" % size)
|
||
|
|
if (size % 8) == 0:
|
||
|
|
return RawBytes(parent, name, size // 8, description)
|
||
|
|
else:
|
||
|
|
return RawBits(parent, name, size, description)
|
||
|
|
|
||
|
|
|
||
|
|
def createPaddingField(parent, nbits, name="padding[]", description=None):
|
||
|
|
if nbits <= 0:
|
||
|
|
raise FieldError("Unable to create padding of %s bits" % nbits)
|
||
|
|
if (nbits % 8) == 0:
|
||
|
|
return PaddingBytes(parent, name, nbits // 8, description)
|
||
|
|
else:
|
||
|
|
return PaddingBits(parent, name, nbits, description)
|
||
|
|
|
||
|
|
|
||
|
|
def createNullField(parent, nbits, name="padding[]", description=None):
|
||
|
|
if nbits <= 0:
|
||
|
|
raise FieldError("Unable to create null padding of %s bits" % nbits)
|
||
|
|
if (nbits % 8) == 0:
|
||
|
|
return NullBytes(parent, name, nbits // 8, description)
|
||
|
|
else:
|
||
|
|
return NullBits(parent, name, nbits, description)
|
||
|
|
|
||
|
|
|
||
|
|
def isString(field):
|
||
|
|
return issubclass(field.__class__, GenericString)
|
||
|
|
|
||
|
|
|
||
|
|
def isInteger(field):
|
||
|
|
return issubclass(field.__class__, GenericInteger)
|
||
|
|
|
||
|
|
|
||
|
|
def writeIntoFile(fieldset, filename):
|
||
|
|
output = FileOutputStream(filename)
|
||
|
|
with output:
|
||
|
|
fieldset.writeInto(output)
|
||
|
|
|
||
|
|
|
||
|
|
def createOrphanField(fieldset, address, field_cls, *args, **kw):
|
||
|
|
"""
|
||
|
|
Create an orphan field at specified address:
|
||
|
|
field_cls(fieldset, *args, **kw)
|
||
|
|
|
||
|
|
The field uses the fieldset properties but it isn't added to the
|
||
|
|
field set.
|
||
|
|
"""
|
||
|
|
save_size = fieldset._current_size
|
||
|
|
try:
|
||
|
|
fieldset._current_size = address
|
||
|
|
field = field_cls(fieldset, *args, **kw)
|
||
|
|
finally:
|
||
|
|
fieldset._current_size = save_size
|
||
|
|
return field
|