first commit

dfhack/python/DF_Maps.cpp

@@ -0,0 +1,458 @@
+/*
+www.sourceforge.net/projects/dfhack
+Copyright (c) 2009 Petr Mrázek (peterix), Kenneth Ferland (Impaler[WrG]), dorf, doomchild
+
+This software is provided 'as-is', without any express or implied
+warranty. In no event will the authors be held liable for any
+damages arising from the use of this software.
+
+Permission is granted to anyone to use this software for any
+purpose, including commercial applications, and to alter it and
+redistribute it freely, subject to the following restrictions:
+
+1. The origin of this software must not be misrepresented; you must
+not claim that you wrote the original software. If you use this
+software in a product, an acknowledgment in the product documentation
+would be appreciated but is not required.
+
+2. Altered source versions must be plainly marked as such, and
+must not be misrepresented as being the original software.
+
+3. This notice may not be removed or altered from any source
+distribution.
+*/
+
+#ifndef __DFMAPS__
+#define __DFMAPS__
+
+#include "Python.h"
+#include <vector>
+#include <string>
+
+using namespace std;
+
+#include "modules/Maps.h"
+#include "DF_Helpers.cpp"
+
+using namespace DFHack;
+
+static PyObject* BuildVein(DFHack::t_vein& v)
+{
+	PyObject* t_dict;
+	PyObject* temp;
+	
+	t_dict = PyDict_New();
+	
+	temp = PyInt_FromLong(v.vtable);
+	DICTADD(t_dict, "vtable", temp);
+	
+	temp = PyInt_FromLong(v.type);
+	DICTADD(t_dict, "type", temp);
+	
+	temp = PyInt_FromLong(v.flags);
+	DICTADD(t_dict, "flags", temp);
+	
+	temp = PyInt_FromLong(v.address_of);
+	DICTADD(t_dict, "address", temp);
+	
+	temp = PyList_New(16);
+	
+	for(int i = 0; i < 16; i++)
+		PyList_SET_ITEM(temp, i, PyInt_FromLong(v.assignment[i]);
+	
+	DICTADD(t_dict, "assignment", temp);
+	
+	return t_dict;
+}
+
+static PyObject* BuildFrozenLiquidVein(DFHack::t_frozenliquidvein& v)
+{
+	PyObject* t_dict;
+	PyObject *temp, *list;
+	
+	t_dict = PyDict_New();
+	
+	temp = PyInt_FromLong(v.vtable);
+	DICTADD(t_dict, "vtable", temp);
+	
+	temp = PyInt_FromLong(v.address_of);
+	DICTADD(t_dict, "address", temp);
+	
+	list = PyList_New(16);
+	
+	for(int i = 0; i < 16; i++)
+	{
+		temp = PyList_New(16);
+		
+		for(int j = 0; j < 16; j++)
+			PyList_SET_ITEM(temp, j, PyInt_FromLong(v.tiles[i][j]);
+		
+		PyList_SET_ITEM(list, i, temp);
+	}
+	
+	DICTADD(t_dict, "tiles", list);
+	
+	return t_dict;
+}
+
+static PyObject* BuildSpatterVein(DFHack::t_spattervein& v)
+{
+	PyObject* t_dict;
+	PyObject *temp, *list;
+	
+	t_dict = PyDict_New();
+	
+	temp = PyInt_FromLong(v.vtable);
+	DICTADD(t_dict, "vtable", temp);
+	
+	temp = PyInt_FromLong(v.address_of);
+	DICTADD(t_dict, "address", temp);
+	
+	temp = PyInt_FromLong(v.mat1);
+	DICTADD(t_dict, "mat1", temp);
+	
+	temp = PyInt_FromLong(v.unk1);
+	DICTADD(t_dict, "unk1", temp);
+	
+	temp = PyInt_FromLong(v.mat2);
+	DICTADD(t_dict, "mat2", temp);
+	
+	temp = PyInt_FromLong(v.mat3);
+	DICTADD(t_dict, "mat3", temp);
+	
+	list = PyList_New(16);
+	
+	for(int i = 0; i < 16; i++)
+	{
+		temp = PyList_New(16);
+		
+		for(int j = 0; j < 16; j++)
+			PyList_SET_ITEM(temp, j, PyInt_FromLong(v.intensity[i][j]);
+		
+		PyList_SET_ITEM(list, i, temp);
+	}
+	
+	DICTADD(t_dict, "intensity", list);
+	
+	return t_dict;
+}
+
+static PyObject* BuildTileTypes40d(DFHack::tiletypes40d& types)
+{
+	PyObject *list, *temp;
+	
+	list = PyList_New(16);
+	
+	for(int i = 0; i < 16; i++)
+	{
+		temp = PyList_New(16);
+		
+		for(int j = 0; j < 16; j++)
+			PyList_SET_ITEM(temp, j, PyInt_FromLong(types[i][j]));
+		
+		PyList_SET_ITEM(list, i, temp);
+	}
+	
+	return list;
+}
+
+static PyObject* BuildOccupancies40d(DFHack::occupancies40d& occ)
+{
+	PyObject *list, *temp;
+	
+	list = PyList_New(16);
+	
+	for(int i = 0; i < 16; i++)
+	{
+		temp = PyList_New(16);
+		
+		for(int j = 0; j < 16; j++)
+			PyList_SET_ITEM(temp, j, PyInt_FromLong(occ[i][j].whole));
+		
+		PyList_SET_ITEM(list, i, temp);
+	}
+	
+	return list;
+}
+
+static PyObject* BuildDesignations40d(DFHack::designations40d& des)
+{
+	PyObject *list, *temp;
+	
+	list = PyList_New(16);
+	
+	for(int i = 0; i < 16; i++)
+	{
+		temp = PyList_New(16);
+		
+		for(int j = 0; j < 16; j++)
+			PyList_SET_ITEM(temp, j, PyInt_FromLong(des[i][j].whole));
+		
+		PyList_SET_ITEM(list, i, temp);
+	}
+	
+	return list;
+}
+
+static PyObject* BuildBiomeIndices40d(DFHack::biome_indices40d& idx)
+{
+	PyObject *list;
+	
+	list = PyList_New(16);
+	
+	for(int i = 0; i < 16; i++)
+		PyList_SET_ITEM(list, i, PyInt_FromLong(idx[i]);
+	
+	return list;
+}
+
+static PyObject* BuildMapBlock40d(DFHack::mapblock40d& block)
+{
+	PyObject* t_dict;
+	PyObject* temp;
+	
+	t_dict = PyDict_New();
+	
+	temp = BuildTileTypes40d(block.tiletypes);
+	DICTADD(t_dict, "tiletypes", temp);
+	
+	temp = BuildDesignations40d(block.designation);
+	DICTADD(t_dict, "designation", temp);
+	
+	temp = BuildOccupancies40d(block.occupancy);
+	DICTADD(t_dict, "occupancy", temp);
+	
+	temp = BuildBiomeIdices40d(block.biome_indices);
+	DICTADD(t_dict, "biome_indices", temp);
+	
+	temp = PyInt_FromLong(block.origin);
+	DICTADD(t_dict, "origin", temp);
+	
+	temp = PyInt_FromLong(block.blockflags.whole);
+	DICTADD(t_dict, "_blockflags", temp);
+	
+	return t_dict;
+}
+
+struct DF_Map
+{
+	PyObject_HEAD
+	DFHack::Maps* m_Ptr;
+};
+
+static PyObject* DF_Map_new(PyTypeObject* type, PyObject* args, PyObject* kwds)
+{
+	DF_Map* self;
+	
+	self = (DF_Map*)type->tp_alloc(type, 0);
+	
+	if(self != NULL)
+		self->m_Ptr = NULL;
+	
+	return (PyObject*)self;
+}
+
+static int DF_Map_init(DF_Map* self, PyObject* args, PyObject* kwds)
+{
+	return 0;
+}
+
+static void DF_Map_dealloc(DF_Map* self)
+{
+	PySys_WriteStdout("map dealloc\n");
+	
+	if(self != NULL)
+	{
+		PySys_WriteStdout("map not NULL\n");
+		
+		if(self->m_Ptr != NULL)
+		{
+			PySys_WriteStdout("m_Ptr = %i\n", (int)self->m_Ptr);
+			
+			delete self->m_Ptr;
+			
+			PySys_WriteStdout("m_Ptr deleted\n");
+			
+			self->m_Ptr = NULL;
+		}
+		
+		self->ob_type->tp_free((PyObject*)self);
+	}
+	
+	PySys_WriteStdout("map dealloc done\n");
+}
+
+// Type methods
+
+static PyObject* DF_Map_Start(DF_Map* self, PyObject* args)
+{
+	if(self->m_Ptr != NULL)
+	{
+		if(self->m_Ptr->Start())
+			Py_RETURN_TRUE;
+		else
+			Py_RETURN_FALSE;
+	}
+	
+	Py_RETURN_NONE;
+}
+
+static PyObject* DF_Map_Finish(DF_Map* self, PyObject* args)
+{
+	if(self->m_Ptr != NULL)
+	{
+		if(self->m_Ptr->Finish())
+			Py_RETURN_TRUE;
+		else
+			Py_RETURN_FALSE;
+	}
+	
+	Py_RETURN_NONE;
+}
+
+static PyObject* DF_Map_IsValidBlock(DF_Map* self, PyObject* args)
+{
+	uint32_t x, y, z;
+	
+	if(self->m_Ptr != NULL)
+	{
+		if(!PyArg_ParseTuple(args, "III", &x, &y, &z))
+			return NULL;
+		
+		if(self->m_Ptr->isValidBlock(x, y, z))
+			Py_RETURN_TRUE;
+		else
+			Py_RETURN_FALSE;
+	}
+	
+	Py_RETURN_NONE;
+}
+
+static PyObject* DF_Map_ReadTileTypes(DF_Map* self, PyObject* args)
+{
+	uint32_t x, y, z;
+	
+	if(self->m_Ptr != NULL)
+	{
+		if(!PyArg_ParseTuple(args, "III", &x, &y, &z))
+			return NULL;
+		
+		tiletypes40d t_types;
+		
+		if(self->m_Ptr->ReadTileTypes(x, y, z, &t_types))
+			return BuildTileTypes40d(t_types);
+	}
+	
+	Py_RETURN_NONE;
+}
+
+static PyObject* DF_Map_ReadDesignations(DF_Map* self, PyObject* args)
+{
+	uint32_t x, y, z;
+	
+	if(self->m_Ptr != NULL)
+	{
+		if(!PyArg_ParseTuple(args, "III", &x, &y, &z))
+			return NULL;
+		
+		designations40d des;
+		
+		if(self->m_Ptr->ReadDesignations(x, y, z, &des))
+			return BuildDesignations40d(des);
+	}
+	
+	Py_RETURN_NONE;
+}
+
+static PyObject* DF_Map_ReadOccupancy(DF_Map* self, PyObject* args)
+{
+	uint32_t x, y, z;
+	
+	if(self->m_Ptr != NULL)
+	{
+		if(!PyArg_ParseTuple(args, "III", &x, &y, &z))
+			return NULL;
+		
+		occupancies40d occ;
+		
+		if(self->m_Ptr->ReadOccupancy(x, y, z, &occ))
+			return BuildOccupancies40d(occ);
+	}
+	
+	Py_RETURN_NONE;
+}
+
+static PyObject* DF_Map_ReadDirtyBit(DF_Map* self, PyObject* args)
+{
+	uint32_t x, y, z;
+	bool bit = false;
+	
+	if(self->m_Ptr != NULL)
+	{
+		if(!PyArg_ParseTuple(args, "III", &x, &y, &z))
+			return NULL;
+		
+		self->m_Ptr->ReadDirtyBit(x, y, z, &bit)
+		
+		if(bit)
+			Py_RETURN_TRUE;
+		else
+			Py_RETURN_FALSE;
+	}
+	
+	Py_RETURN_NONE;
+}
+
+static PyObject* DF_Map_ReadBlockFlags(DF_Map* self, PyObject* args)
+{
+	uint32_t x, y, z;
+	
+	if(self->m_Ptr != NULL)
+	{
+		if(!PyArg_ParseTuple(args, "III", &x, &y, &z))
+			return NULL;
+		
+		t_blockflags flags;
+		
+		if(self->m_Ptr->ReadBlockFlags(x, y, z, &flags))
+			return PyInt_FromLong(flags.whole);
+	}
+	
+	Py_RETURN_NONE;
+}
+
+static PyObject* DF_Map_ReadRegionOffsets(DF_Map* self, PyObject* args)
+{
+	uint32_t x, y, z;
+	
+	if(self->m_Ptr != NULL)
+	{
+		if(!PyArg_ParseTuple(args, "III", &x, &y, &z))
+			return NULL;
+		
+		biome_indices40d biome;
+		
+		if(self->m_Ptr->ReadRegionOffsets(x, y, z, &biome))
+			return BuildBiomeIndices40d(biome);
+	}
+	
+	Py_RETURN_NONE;
+}
+
+// Getter/Setter
+
+static PyObject* DF_Map_getSize(DF_Map* self, void* closure)
+{
+	uint32_t x, y, z;
+	
+	if(self->m_Ptr != NULL)
+	{
+		self->m_Ptr->getSize(x, y, z);
+		
+		return PyTuple_Pack(3, x, y, z);
+	}
+	
+	Py_RETURN_NONE;
+}
+
+#endif