mirror of
https://notabug.org/scuti/lib3ddevil1
synced 2024-11-29 02:02:04 +05:30
67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
import ctypes
|
|
import os, sys
|
|
# This is the folder containing the whole library.
|
|
sys.path.append(
|
|
os.path.abspath(
|
|
os.path.join(
|
|
os.path.dirname(__file__), "../../")))
|
|
from lib3ddevil1.bindings import libc
|
|
del os, sys
|
|
|
|
#--------------------------------------+
|
|
# Basic Struct
|
|
#--------------------------------------+
|
|
|
|
class PldHeader(ctypes.Structure):
|
|
_pack_ = 1
|
|
_fields_ = [
|
|
("numOffset", ctypes.c_int),
|
|
("offsets", ctypes.POINTER(ctypes.c_int))
|
|
]
|
|
|
|
class Devil1PLD_FN(ctypes.Structure):
|
|
_fields_ = [
|
|
("getheader" , ctypes.CFUNCTYPE(
|
|
ctypes.c_bool,
|
|
ctypes.POINTER(PldHeader),
|
|
ctypes.c_char_p)),
|
|
("sizeofsector", ctypes.CFUNCTYPE(
|
|
ctypes.c_int,
|
|
ctypes.POINTER(PldHeader),
|
|
ctypes.c_int)),
|
|
("printheader" , ctypes.CFUNCTYPE(None,
|
|
ctypes.POINTER(PldHeader)))
|
|
]
|
|
|
|
devil1pld = Devil1PLD_FN.in_dll(libc, "DEVIL1PLD")
|
|
del libc
|
|
|
|
#--------------------------------------+
|
|
# Pythonic Object
|
|
#--------------------------------------+
|
|
|
|
class pyPldHeader:
|
|
def __init__(self, filedata = None):
|
|
# Store C Struct in order to call C functions
|
|
self.cstruct = PldHeader()
|
|
if filedata:
|
|
if not devil1pld.getheader(ctypes.byref(self.cstruct), filedata):
|
|
raise RuntimeError("failed to get .pld header")
|
|
self.eof = len(filedata)
|
|
|
|
def show(self):
|
|
devil1pld.printheader(ctypes.byref(self.cstruct))
|
|
return
|
|
|
|
def getnumoffsets(self):
|
|
return self.cstruct.numOffsets
|
|
|
|
# return pythonic list of offsets
|
|
def getoffsets(self):
|
|
return self.cstruct.offsets[:self.cstruct.numOffset]
|
|
|
|
def sizeofsector(self, i):
|
|
ptr = ctypes.byref(self.cstruct)
|
|
return devil1pld.sizeofsector(ptr, i, self.eof)
|
|
|