dfhack/library/python/pydfhack/materials.py

from ctypes import *
import dftypes
from dftypes import libdfhack, Matgloss, CreatureType, DescriptorColor, MatglossOther

libdfhack.Materials_getInorganic.restype = POINTER(Matgloss)
libdfhack.Materials_getOrganic.restype = POINTER(Matgloss)
libdfhack.Materials_getTree.restype = POINTER(Matgloss)
libdfhack.Materials_getPlant.restype = POINTER(Matgloss)
libdfhack.Materials_getRace.restype = POINTER(Matgloss)
libdfhack.Materials_getRaceEx.restype = POINTER(CreatureType)
libdfhack.Materials_getColor.restype = POINTER(DescriptorColor)
libdfhack.Materials_getOther.restype = POINTER(MatglossOther)
libdfhack.Materials_getAllDesc.restype = POINTER(Matgloss)

class Materials(object):
    def __init__(self, ptr):
        self._mat_ptr = ptr

        self.inorganic = None
        self.organic = None
        self.tree = None
        self.plant = None
        self.race = None
        self.race_ex = None
        self.color = None
        self.other = None
    
    def _get_inorganic(self):
        inorganic = libdfhack.Materials_getInorganic(self._mat_ptr)
        
        if id(inorganic) in dftypes.pointer_dict:
            self.inorganic = dftypes.pointer_dict[id(inorganic)][1]
            del dftypes.pointer_dict[id(inorganic)]

    def _get_organic(self):
        organic = libdfhack.Materials_getOrganic(self._mat_ptr)
        
        if id(organic) in dftypes.pointer_dict:
            self.organic = dftypes.pointer_dict[id(organic)][1]
            del dftypes.pointer_dict[id(inorganic)]

    def _get_tree(self):
        tree = libdfhack.Materials_getTree(self._mat_ptr)
        
        if id(tree) in dftypes.pointer_dict:
            self.tree = dftypes.pointer_dict[id(tree)][1]
            del dftypes.pointer_dict[id(tree)]

    def _get_plant(self):
        plant = libdfhack.Materials_getPlant(self._mat_ptr)
        
        if id(plant) in dftypes.pointer_dict:
            self.tree = dftypes.pointer_dict[id(tree)][1]
            del dftypes.pointer_dict[id(tree)]

    def _get_race(self):
        race = libdfhack.Materials_getRace(self._mat_ptr)
        
        if id(race) in dftypes.pointer_dict:
            self.race = dftypes.pointer_dict[id(race)][1]
            del dftypes.pointer_dict[id(race)]

    def _get_race_ex(self):
        race_ex = libdfhack.Materials_getRaceEx(self._mat_ptr)
        
        if id(race_ex) in dftypes.pointer_dict:
            self.race_ex = dftypes.pointer_dict[id(race_ex)][1]
            del dftypes.pointer_dict[id(race_ex)]

    def _get_color(self):
        color = libdfhack.Materials_getColor(self._mat_ptr)
        
        if id(color) in dftypes.pointer_dict:
            self.color = dftypes.pointer_dict[id(color)][1]
            del dftypes.pointer_dict[id(color)]

    def _get_other(self):
        other = libdfhack.Materials_getOther(self._mat_ptr)
        
        if id(other) in dftypes.pointer_dict:
            self.other = dftypes.pointer_dict[id(other)][1]
            del dftypes.pointer_dict[id(other)]
    
    def _get_all(self):
        self._get_inorganic()
        self._get_organic()
        self._get_tree()
        self._get_plant()
        self._get_race()
        self._get_race_ex()
        self._get_color()
        self._get_other()
    def _clear_all(self):
        self.inorganic = None
        self.organic = None
        self.tree = None
        self.plant = None
        self.race = None
        self.race_ex = None
        self.color = None
        self.other = None

    def read_inorganic(self):
        result = libdfhack.Materials_ReadInorganicMaterials(self._mat_ptr) > 0
        
        if result == True:
            self._get_inorganic()
        else:
            self.inorganic = None
        
        return result

    def read_organic(self):
        result = libdfhack.Materials_ReadOrganicMaterials(self._mat_ptr) > 0
        
        if result == True:
            self._get_organic()
        else:
            self.organic = None
        
        return result

    def read_tree(self):
        result = libdfhack.Materials_ReadWoodMaterials(self._mat_ptr) > 0
        
        if result == True:
            self._get_tree()
        else:
            self.tree = None
        
        return result

    def read_plant(self):
        result = libdfhack.Materials_ReadPlantMaterials(self._mat_ptr) > 0
        
        if result == True:
            self._get_plant()
        else:
            self.plant = None
        
        return result

    def read_creature_types(self):
        result = libdfhack.Materials_ReadCreatureTypes(self._mat_ptr) > 0
        
        if result == True:
            self._get_race()
        else:
            self.race = None
        
        return result

    def read_creature_types_ex(self):
        result = libdfhack.Materials_ReadCreatureTypesEx(self._mat_ptr) > 0
        
        if result == True:
            self._get_race_ex()
        else:
            self.race_ex = None
        
        return result

    def read_descriptor_colors(self):
        result = libdfhack.Materials_ReadDescriptorColors(self._mat_ptr) > 0
        
        if result == True:
            self._get_color()
        else:
            self.color = None
        
        return result

    def read_others(self):
        result = libdfhack.Materials_ReadOthers(self._mat_ptr) > 0
        
        if result == True:
            self._get_other()
        else:
            self.other = None
        
        return result

    def read_all(self):
        result = libdfhack.Materials_ReadAllMaterials(self._mat_ptr) > 0
        
        if result == True:
            self._get_all()
        else:
            self._clear_all()
        
        return result
    
    def get_type(self, material):
        return libdfhack.Materials_getType(self._mat_ptr, byref(material))

    def get_description(self, material):
        return libdfhack.Materials_getDescription(self._mat_ptr, byref(material))