diff --git a/dfhack/python/pydfhack.cpp b/dfhack/python/pydfhack.cpp index 7fef20d62..da5e9690e 100644 --- a/dfhack/python/pydfhack.cpp +++ b/dfhack/python/pydfhack.cpp @@ -24,6 +24,7 @@ distribution. #include "Python.h" #include "DF_MemInfo.cpp" +#include "DF_Material.cpp" #include "DF_API.cpp" #ifndef PyMODINIT_FUNC @@ -66,14 +67,19 @@ PyMODINIT_FUNC initpydfhack(void) if(PyType_Ready(&DF_Position_type) < 0) return; + + if(PyType_Ready(&DF_Material_type) < 0) + return; module = Py_InitModule3("pydfhack", module_methods, "pydfhack extension module"); Py_INCREF(&DF_API_type); Py_INCREF(&DF_MemInfo_type); Py_INCREF(&DF_Position_type); + Py_INCREF(&DF_Material_type); PyModule_AddObject(module, "API", (PyObject*)&DF_API_type); PyModule_AddObject(module, "MemInfo", (PyObject*)&DF_MemInfo_type); PyModule_AddObject(module, "Position", (PyObject*)&DF_Position_type); + PyModule_AddObject(module, "Material", (PyObject*)&DF_Material_type); } \ No newline at end of file