224 lines
7.3 KiB
Python
224 lines
7.3 KiB
Python
from pydfhack import *
|
|
from ctypes import *
|
|
|
|
class DFAPI(API):
|
|
def Read_Designations(self, x, y, z):
|
|
temp = API.Read_Designations(self, x, y, z)
|
|
|
|
d_list = []
|
|
|
|
for i in temp:
|
|
d = []
|
|
|
|
for j in i:
|
|
d.append(DesignationFlags(j))
|
|
|
|
d_list.append(d)
|
|
|
|
return d_list
|
|
def Write_Designations(self, x, y, z, d_list):
|
|
temp = []
|
|
|
|
for i in d_list:
|
|
t = []
|
|
|
|
for j in i:
|
|
t.append(j.whole)
|
|
|
|
temp.append(t)
|
|
|
|
API.Write_Designations(self, x, y, z, temp)
|
|
def Read_Occupancy(self, x, y, z):
|
|
temp = API.Read_Occupancy(self, x, y, z)
|
|
|
|
o_list = []
|
|
|
|
for i in temp:
|
|
o = []
|
|
|
|
for j in i:
|
|
o.append(OccupancyFlags(j))
|
|
|
|
o_list.append(o)
|
|
|
|
return o_list
|
|
|
|
class DesignationStruct(Structure):
|
|
_fields_ = [("flow_size", c_uint, 3),
|
|
("pile", c_uint, 1),
|
|
("dig", c_uint, 3),
|
|
("smooth", c_uint, 2),
|
|
("hidden", c_uint, 1),
|
|
("geolayer_index", c_uint, 4),
|
|
("light", c_uint, 1),
|
|
("subterranean", c_uint, 1),
|
|
("skyview", c_uint, 1),
|
|
("biome", c_uint, 4),
|
|
("liquid_type", c_uint, 1),
|
|
("water_table", c_uint, 1),
|
|
("rained", c_uint, 1),
|
|
("traffic", c_uint, 2),
|
|
("flow_forbid", c_uint, 1),
|
|
("liquid_static", c_uint, 1),
|
|
("moss", c_uint, 1),
|
|
("feature_present", c_uint, 1),
|
|
("liquid_character", c_uint, 2)]
|
|
|
|
class DesignationFlags(Union):
|
|
_fields_ = [("whole", c_uint, 32),
|
|
("bits", DesignationStruct)]
|
|
|
|
def __init__(self, initial = 0):
|
|
self.whole = initial
|
|
|
|
class OccupancyStruct(Strucure):
|
|
_fields_ = [("building", c_uint, 3),
|
|
("unit", c_uint, 1),
|
|
("unit_grounded", c_uint, 1),
|
|
("item", c_uint, 1),
|
|
("splatter", c_uint, 26)]
|
|
|
|
class OccupancyFlags(Union):
|
|
_fields_ = [("whole", c_uint, 32),
|
|
("bits", OccupancyStruct)]
|
|
|
|
def __init__(self, initial = 0):
|
|
self.whole = initial
|
|
|
|
class CreatureStruct1(Structure):
|
|
_fields_ = [("move_state", c_uint, 1),
|
|
("dead", c_uint, 1),
|
|
("has_mood", c_uint, 1),
|
|
("had_mood", c_uint, 1),
|
|
("marauder", c_uint, 1),
|
|
("drowning", c_uint, 1),
|
|
("merchant", c_uint, 1),
|
|
("forest", c_uint, 1),
|
|
("left", c_uint, 1),
|
|
("rider", c_uint, 1),
|
|
("incoming", c_uint, 1),
|
|
("diplomat", c_uint, 1),
|
|
("zombie", c_uint, 1),
|
|
("skeleton", c_uint, 1),
|
|
("can_swap", c_uint, 1),
|
|
("on_ground", c_uint, 1),
|
|
("projectile", c_uint, 1),
|
|
("active_invader", c_uint, 1),
|
|
("hidden_in_ambush", c_uint, 1),
|
|
("invader_origin", c_uint, 1),
|
|
("coward", c_uint, 1),
|
|
("hidden_ambusher", c_uint, 1),
|
|
("invades", c_uint, 1),
|
|
("check_flows", c_uint, 1),
|
|
("ridden", c_uint, 1),
|
|
("caged", c_uint, 1),
|
|
("tame", c_uint, 1),
|
|
("chained", c_uint, 1),
|
|
("royal_guard", c_uint, 1),
|
|
("fortress_guard", c_uint, 1),
|
|
("suppress_wield", c_uint, 1),
|
|
("important_historical_figure", c_uint, 1)]
|
|
|
|
class CreatureFlags1(Union):
|
|
_fields_ = [("whole", c_uint, 32),
|
|
("bits", CreatureStruct1)]
|
|
|
|
def __init__(self, initial = 0):
|
|
self.whole = initial
|
|
|
|
class CreatureStruct2(Structure):
|
|
_fields_ = [("swimming", c_uint, 1),
|
|
("sparring", c_uint, 1),
|
|
("no_notify", c_uint, 1),
|
|
("unused", c_uint, 1),
|
|
("calculated_nerves", c_uint, 1),
|
|
("calculated_bodyparts", c_uint, 1),
|
|
("important_historical_figure", c_uint, 1),
|
|
("killed", c_uint, 1),
|
|
("cleanup_1", c_uint, 1),
|
|
("cleanup_2", c_uint, 1),
|
|
("cleanup_3", c_uint, 1),
|
|
("for_trade", c_uint, 1),
|
|
("trade_resolved", c_uint, 1),
|
|
("has_breaks", c_uint, 1),
|
|
("gutted", c_uint, 1),
|
|
("circulatory_spray", c_uint, 1),
|
|
("locked_in_for_trading", c_uint, 1),
|
|
("slaughter", c_uint, 1),
|
|
("underworld", c_uint, 1),
|
|
("resident", c_uint, 1),
|
|
("cleanup_4", c_uint, 1),
|
|
("calculated_insulation", c_uint, 1),
|
|
("visitor_uninvited", c_uint, 1),
|
|
("visitor", c_uint, 1),
|
|
("calculated_inventory", c_uint, 1),
|
|
("vision_good", c_uint, 1),
|
|
("vision_damaged", c_uint, 1),
|
|
("vision_missing", c_uint, 1),
|
|
("breathing_good", c_uint, 1),
|
|
("breathing_problem", c_uint, 1),
|
|
("roaming_wilderness_population_source", c_uint, 1),
|
|
("roaming_wilderness_population_source_not_a_map_feature", c_uint, 1)]
|
|
|
|
class CreatureFlags2(Union):
|
|
_fields_ = [("whole", c_uint, 32),
|
|
("bits", CreatureStruct2)]
|
|
|
|
def __init__(self, initial = 0):
|
|
self.whole = initial
|
|
|
|
class ItemStruct(Structure):
|
|
_fields_ = [("on_ground", c_uint, 1),
|
|
("in_job", c_uint, 1),
|
|
("in_inventory", c_uint, 1),
|
|
("u_ngrd1", c_uint, 1),
|
|
("in_workshop", c_uint, 1),
|
|
("u_ngrd2", c_uint, 1),
|
|
("u_ngrd3", c_uint, 1),
|
|
("rotten", c_uint, 1),
|
|
("unk1", c_uint, 1),
|
|
("u_ngrd4", c_uint, 1),
|
|
("unk2", c_uint, 1),
|
|
("u_ngrd5", c_uint, 1),
|
|
("unk3", c_uint, 1),
|
|
("u_ngrd6", c_uint, 1),
|
|
("narrow", c_uint, 1),
|
|
("u_ngrd7", c_uint, 1),
|
|
("worn", c_uint, 1),
|
|
("unk4", c_uint, 1),
|
|
("u_ngrd8", c_uint, 1),
|
|
("forbid", c_uint, 1),
|
|
("unk5", c_uint, 1),
|
|
("dump", c_uint, 1),
|
|
("on_fire", c_uint, 1),
|
|
("melt", c_uint, 1),
|
|
("hidden", c_uint, 1),
|
|
("u_ngrd9", c_uint, 1),
|
|
("unk6", c_uint, 1),
|
|
("unk7", c_uint, 1),
|
|
("unk8", c_uint, 1),
|
|
("unk9", c_uint, 1),
|
|
("unk10", c_uint, 1),
|
|
("unk11", c_uint, 1)]
|
|
|
|
class ItemFlags(Union):
|
|
_fields_ = [("whole", c_uint, 32),
|
|
("bits", ItemStruct)]
|
|
|
|
def __init__(self, initial = 0):
|
|
self.whole = initial
|
|
|
|
dig_types = { "no" : 0,
|
|
"default" : 1,
|
|
"ud_stair" : 2,
|
|
"channel" : 3,
|
|
"ramp" : 4,
|
|
"d_stair" : 5,
|
|
"u_stair" : 6,
|
|
"whatever" : 7 }
|
|
|
|
traffic_types = { "normal" : 0,
|
|
"low" : 1,
|
|
"high" : 2,
|
|
"restricted" : 3 }
|