164 lines
5.2 KiB
Python
164 lines
5.2 KiB
Python
from ctypes import *
|
|
from pydftypes import libdfhack
|
|
from util import _uintify
|
|
|
|
libdfhack.Maps_getSize.argtypes = [ c_void_p, uint_ptr, uint_ptr, uint_ptr ]
|
|
libdfhack.Maps_ReadTileTypes.argtypes = [ c_void_p, c_uint, c_uint, c_uint, POINTER(TileTypes40d) ]
|
|
libdfhack.Maps_WriteTileTypes.argtypes = [ c_void_p, c_uint, c_uint, c_uint, POINTER(TileTypes40d) ]
|
|
libdfhack.Maps_ReadDesignations.argtypes = [ c_void_p, c_uint, c_uint, c_uint, POINTER(Designations40d) ]
|
|
libdfhack.Maps_WriteDesignations.argtypes = [ c_void_p, c_uint, c_uint, c_uint, POINTER(Designations40d) ]
|
|
libdfhack.Maps_ReadTemperatures.argtypes = [ c_void_p, c_uint, c_uint, c_uint, POINTER(Temperatures) ]
|
|
libdfhack.Maps_WriteTemperatures.argtypes = [ c_void_p, c_uint, c_uint, c_uint, POINTER(Temperatures) ]
|
|
libdfhack.Maps_ReadOccupancy.argtypes = [ c_void_p, c_uint, c_uint, c_uint, POINTER(Occupancies40d) ]
|
|
libdfhack.Maps_WriteOccupancy.argtypes = [ c_void_p, c_uint, c_uint, c_uint, POINTER(Occupancies40d) ]
|
|
libdfhack.Maps_ReadRegionOffsets.argtypes = [ c_void_p, c_uint, c_uint, c_uint, POINTER(BiomeIndices40d) ]
|
|
|
|
class Maps(object):
|
|
def __init__(self, ptr):
|
|
self._map_ptr = ptr
|
|
|
|
def start(self):
|
|
return libdfhack.Maps_Start(self._map_ptr) > 0
|
|
|
|
def finish(self):
|
|
return libdfhack.Maps_Finish(self._map_ptr) > 0
|
|
|
|
def is_valid_block(self, x, y, z):
|
|
return libdfhack.Maps_isValidBlock(self._map_ptr, *_uintify(x, y, z)) > 0
|
|
|
|
def read_tile_types(self, x, y, z):
|
|
tt = TileTypes40d()
|
|
|
|
ux, uy, uz = _uintify(x, y, z)
|
|
|
|
if libdfhack.Maps_ReadTileTypes(self._map_ptr, ux, uy, uz, tt) > 0:
|
|
return tt
|
|
else:
|
|
return None
|
|
|
|
def write_tile_types(self, x, y, z, tt):
|
|
ux, uy, uz = _uintify(x, y, z)
|
|
|
|
return libdfhack.Maps_WriteTileTypes(self._map_ptr, ux, uy, uz, tt) > 0
|
|
|
|
def read_designations(self, x, y, z):
|
|
d = Designations40d()
|
|
|
|
ux, uy, uz = _uintify(x, y, z)
|
|
|
|
if libdfhack.Maps_ReadDesignations(self._map_ptr, ux, uy, uz, d) > 0:
|
|
return d
|
|
else:
|
|
return None
|
|
|
|
def write_designations(self, x, y, z, d):
|
|
ux, uy, uz = _uintify(x, y, z)
|
|
|
|
return libdfhack.Maps_WriteDesignations(self._map_ptr, ux, uy, uz, d) > 0
|
|
|
|
def read_temperatures(self, x, y, z):
|
|
t = Temperatures()
|
|
|
|
ux, uy, uz = _uintify(x, y, z)
|
|
|
|
if libdfhack.Maps_ReadDesignations(self._map_ptr, ux, uy, uz, t) > 0:
|
|
return t
|
|
else:
|
|
return None
|
|
|
|
def write_temperatures(self, x, y, z, t):
|
|
ux, uy, uz = _uintify(x, y, z)
|
|
|
|
return libdfhack.Maps_WriteDesignations(self._map_ptr, ux, uy, uz, t) > 0
|
|
|
|
def read_occupancy(self, x, y, z):
|
|
o = Occupancies40d()
|
|
|
|
ux, uy, uz = _uintify(x, y, z)
|
|
|
|
if libdfhack.Maps_ReadDesignations(self._map_ptr, ux, uy, uz, o) > 0:
|
|
return o
|
|
else:
|
|
return None
|
|
|
|
def write_designations(self, x, y, z, o):
|
|
ux, uy, uz = _uintify(x, y, z)
|
|
|
|
return libdfhack.Maps_WriteDesignations(self._map_ptr, ux, uy, uz, o) > 0
|
|
|
|
def read_dirty_bit(self, x, y, z):
|
|
bit = c_int(0)
|
|
|
|
ux, uy, uz = _uintify(x, y, z)
|
|
|
|
if libdfhack.Maps_ReadDirtyBit(self._map_ptr, ux, uy, uz, byref(bit)) > 0:
|
|
if bit > 0:
|
|
return True
|
|
else:
|
|
return False
|
|
else:
|
|
return None
|
|
|
|
def write_dirty_bit(self, x, y, z, dirty):
|
|
ux, uy, uz = _uintify(x, y, z)
|
|
|
|
return libdfhack.Maps_WriteDirtyBit(self._map_ptr, ux, uy, uz, c_int(dirty)) > 0
|
|
|
|
def read_features(self, x, y, z):
|
|
lf = c_short()
|
|
gf = c_short()
|
|
|
|
ux, uy, uz = _uintify(x, y, z)
|
|
|
|
libdfhack.Maps_ReadFeatures(self._map_ptr, ux, uy, uz, byref(lf), byref(fg))
|
|
|
|
return (lf, gf)
|
|
|
|
def write_local_feature(self, x, y, z, local_feature = -1):
|
|
ux, uy, uz = _uintify(x, y, z)
|
|
|
|
return libdfhack.Maps_WriteLocalFeature(self._map_ptr, ux, uy, uz, c_short(local_feature)) > 0
|
|
|
|
def write_global_feature(self, x, y, z, global_feature = -1):
|
|
ux, uy, uz = _uintify(x, y, z)
|
|
|
|
return libdfhack.Maps_WriteGlobalFeature(self._map_ptr, ux, uy, uz, c_short(global_feature)) > 0
|
|
|
|
def read_block_flags(self, x, y, z):
|
|
bf = BlockFlags()
|
|
|
|
ux, uy, uz = _uintify(x, y, z)
|
|
|
|
if libdfhack.Maps_ReadBlockFlags(self._map_ptr, ux, uy, uz, byref(bf)) > 0:
|
|
return bf
|
|
else:
|
|
return None
|
|
|
|
def write_block_flags(self, x, y, z, block_flags):
|
|
ux, uy, uz = _uintify(x, y, z)
|
|
|
|
return libdfhack.Maps_WriteBlockFlags(self._map_ptr, ux, uy, uz, block_flags) > 0
|
|
|
|
def read_region_offsets(self, x, y, z):
|
|
bi = BiomeIndices40d()
|
|
|
|
ux, uy, uz = _uintify(x, y, z)
|
|
|
|
if libdfhack.Maps_ReadRegionOffsets(self._map_ptr, ux, uy, uz, byref(bi)) > 0:
|
|
return bi
|
|
else:
|
|
return None
|
|
|
|
@property
|
|
def size(self):
|
|
x = c_uint()
|
|
y = c_uint()
|
|
z = c_uint()
|
|
|
|
retval = libdfhack.Maps_getSize(self._map_ptr, byref(x), byref(y), byref(z))
|
|
|
|
if retval > 0:
|
|
return (int(x.value), int(y.value), int(z.value))
|
|
else:
|
|
return (-1, -1, -1)
|