first commit
doomchild
parent
2284fb32d2
commit
0d2554f829
@ -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
|
||||
Loading…
Reference in New Issue