35 lines
950 B
Python
35 lines
950 B
Python
|
from ctypes import *
|
||
|
|
from dftypes import *
|
||
|
|
import util
|
||
|
|
|
||
|
|
libdfhack.Buildings_GetCustomWorkshopType.argtypes = [ c_void_p, POINTER(CustomWorkshop) ]
|
||
|
|
|
||
|
|
class Buildings(object):
|
||
|
|
def __init__(self, ptr):
|
||
|
|
self._b_ptr = ptr
|
||
|
|
|
||
|
|
def start(self):
|
||
|
|
num = c_uint()
|
||
|
|
|
||
|
|
if libdfhack.Buildings_Start(self._b_ptr, byref(num)) > 0:
|
||
|
|
return int(num.value)
|
||
|
|
else:
|
||
|
|
return -1
|
||
|
|
|
||
|
|
def finish(self):
|
||
|
|
return libdfhack.Buildings_Finish(self._b_ptr) > 0
|
||
|
|
|
||
|
|
def read(self, index):
|
||
|
|
b = Building()
|
||
|
|
|
||
|
|
if libdfhack.Buildings_Read(self._b_ptr, c_uint(index), byref(b)) > 0:
|
||
|
|
return b
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
|
||
|
|
def read_custom_workshop_types(self):
|
||
|
|
return libdfhack.Buildings_ReadCustomWorkshopTypes(self._b_ptr)
|
||
|
|
|
||
|
|
def get_custom_workshop_type(self, custom_workshop):
|
||
|
|
return libdfhack.Buildings_GetCustomWorkshopType(self._b_ptr, byref(custom_workshop))
|