libraries/pyshapelib/shptreemodule.c

/* Copyright (c) 2001, 2002 by Intevation GmbH
 * Authors:
 * Bernhard Herzog <[email protected]>
 *
 * This program is free software under the GPL (>=v2)
 * Read the file COPYING coming with Thuban for details.
 */

/* Python wrapper for the shapelib SHPTree */

#include <Python.h>
#include <shapefil.h>

#include "pyshapelib_api.h"

PyShapeLibAPI * api;

typedef struct {
    PyObject_HEAD
    SHPTree * tree; 
} SHPTreeObject;

extern PyTypeObject SHPTreeType;

#define SHPTree_Check(v)                ((v)->ob_type == &SHPTreeType)

/* Create a new python wrapper object from a SHPTree pointer */
static PyObject *
SHPTreeObject_FromSHPTree(SHPTree* tree)
{
    SHPTreeObject * self = PyObject_NEW(SHPTreeObject, &SHPTreeType);
    if (!self)
        return NULL;

    self->tree = tree;

    return (PyObject *)self;
}

/* Deallocate the SHPTree wrapper. */
static void
shptree_dealloc(SHPTreeObject * self)
{
    api->SHPDestroyTree(self->tree);
    PyMem_DEL(self);
}

/* Return the repr of the wrapper */
static PyObject *
shptree_repr(SHPTreeObject * self)
{
    char buf[1000];
    sprintf(buf, "<SHPTree at %xul>", (unsigned long)self);
    return PyString_FromString(buf);
}

static PyObject *
shptree_find_shapes(SHPTreeObject * self, PyObject * args)
{
    double min[4] = {0, 0, 0, 0};
    double max[4] = {0, 0, 0, 0};
    int count, idx;
    int * ids;
    PyObject * list = NULL, *temp = NULL;

    if (!PyArg_ParseTuple(args, "(dd)(dd)", min + 0, min + 1,
                          max + 0, max + 1))
        return NULL;

    ids = api->SHPTreeFindLikelyShapes(self->tree, min, max, &count);

    list = PyList_New(count);
    if (!list)
        goto fail;

    /* Turn the returned array of indices into a python list of ints. */
    for (idx = 0; idx < count; idx++)
    {
        temp = PyInt_FromLong(ids[idx]);
        if (!temp)
            goto fail;

        if (PyList_SetItem(list, idx, temp) == -1)
        {
            /* temp's refcount has already be decreased. Set temp to
             * NULL so that the fail code doesn't do it again
             */
            temp = NULL;
            goto fail;
        }
    }

    free(ids);
    return list;
    
 fail:
    free(ids);
    Py_XDECREF(list);
    Py_XDECREF(temp);
    return NULL;
}


static struct PyMethodDef shptree_methods[] = {
    {"find_shapes",     (PyCFunction)shptree_find_shapes,       METH_VARARGS},
    {NULL,      NULL}
};

static PyObject *
shptree_getattr(PyObject * self, char * name)
{
    return Py_FindMethod(shptree_methods, self, name);
}


PyTypeObject SHPTreeType = {
        PyObject_HEAD_INIT(NULL)
        0,
        "SHPTree",
        sizeof(SHPTreeObject),
        0,
        (destructor)shptree_dealloc,    /*tp_dealloc*/
        (printfunc)NULL,                /*tp_print*/
        shptree_getattr,                /*tp_getattr*/
        0,                              /*tp_setattr*/
        0,                              /*tp_compare*/
        (reprfunc)shptree_repr,         /*tp_repr*/
        0,                              /*tp_as_number*/
        0,                              /*tp_as_sequence*/
        0,                              /*tp_as_mapping*/
        0,                              /*tp_hash*/
        0,                              /*tp_call*/
        0,                              /*tp_str*/
        0,                              /*tp_getattro*/
        0,                              /*tp_setattro*/
        0,                              /*tp_as_buffer*/
};


static PyObject *
shptree_from_shapefile(PyObject * self, PyObject * args)
{
    SHPTree * tree;
    SHPHandle handle;
    PyObject * cobject;
    int dimension, max_depth;

    if (!PyArg_ParseTuple(args, "O!ii", &PyCObject_Type, &cobject,
                          &dimension, &max_depth))
        return NULL;

    handle = PyCObject_AsVoidPtr(cobject);

    tree = api->SHPCreateTree(handle, dimension, max_depth, NULL, NULL);

    /* apparently SHPCreateTree doesn't do any error checking, so we
     * have to assume that tree is valid at this point. */
    return SHPTreeObject_FromSHPTree(tree);
}


static PyMethodDef module_functions[] = {
    {"SHPTree",         shptree_from_shapefile,         METH_VARARGS},
    { NULL, NULL }
};


void
initshptree()
{
    SHPTreeType.ob_type = &PyType_Type;

    Py_InitModule("shptree", module_functions);
    PYSHAPELIB_IMPORT_API(api);
}
1	/* Copyright (c) 2001, 2002 by Intevation GmbH
2	* Authors:
3	* Bernhard Herzog <[email protected]>
4	*
5	* This program is free software under the GPL (>=v2)
6	* Read the file COPYING coming with Thuban for details.
7	*/
8
9	/* Python wrapper for the shapelib SHPTree */
10
11	#include <Python.h>
12	#include <shapefil.h>
13
14	#include "pyshapelib_api.h"
15
16	PyShapeLibAPI * api;
17
18	typedef struct {
19	PyObject_HEAD
20	SHPTree * tree;
21	} SHPTreeObject;
22
23	extern PyTypeObject SHPTreeType;
24
25	#define SHPTree_Check(v) ((v)->ob_type == &SHPTreeType)
26
27	/* Create a new python wrapper object from a SHPTree pointer */
28	static PyObject *
29	SHPTreeObject_FromSHPTree(SHPTree* tree)
30	{
31	SHPTreeObject * self = PyObject_NEW(SHPTreeObject, &SHPTreeType);
32	if (!self)
33	return NULL;
34
35	self->tree = tree;
36
37	return (PyObject *)self;
38	}
39
40	/* Deallocate the SHPTree wrapper. */
41	static void
42	shptree_dealloc(SHPTreeObject * self)
43	{
44	api->SHPDestroyTree(self->tree);
45	PyMem_DEL(self);
46	}
47
48	/* Return the repr of the wrapper */
49	static PyObject *
50	shptree_repr(SHPTreeObject * self)
51	{
52	char buf[1000];
53	sprintf(buf, "<SHPTree at %xul>", (unsigned long)self);
54	return PyString_FromString(buf);
55	}
56
57	static PyObject *
58	shptree_find_shapes(SHPTreeObject * self, PyObject * args)
59	{
60	double min[4] = {0, 0, 0, 0};
61	double max[4] = {0, 0, 0, 0};
62	int count, idx;
63	int * ids;
64	PyObject * list = NULL, *temp = NULL;
65
66	if (!PyArg_ParseTuple(args, "(dd)(dd)", min + 0, min + 1,
67	max + 0, max + 1))
68	return NULL;
69
70	ids = api->SHPTreeFindLikelyShapes(self->tree, min, max, &count);
71
72	list = PyList_New(count);
73	if (!list)
74	goto fail;
75
76	/* Turn the returned array of indices into a python list of ints. */
77	for (idx = 0; idx < count; idx++)
78	{
79	temp = PyInt_FromLong(ids[idx]);
80	if (!temp)
81	goto fail;
82
83	if (PyList_SetItem(list, idx, temp) == -1)
84	{
85	/* temp's refcount has already be decreased. Set temp to
86	* NULL so that the fail code doesn't do it again
87	*/
88	temp = NULL;
89	goto fail;
90	}
91	}
92
93	free(ids);
94	return list;
95
96	fail:
97	free(ids);
98	Py_XDECREF(list);
99	Py_XDECREF(temp);
100	return NULL;
101	}
102
103
104	static struct PyMethodDef shptree_methods[] = {
105	{"find_shapes", (PyCFunction)shptree_find_shapes, METH_VARARGS},
106	{NULL, NULL}
107	};
108
109	static PyObject *
110	shptree_getattr(PyObject * self, char * name)
111	{
112	return Py_FindMethod(shptree_methods, self, name);
113	}
114
115
116	PyTypeObject SHPTreeType = {
117	PyObject_HEAD_INIT(NULL)
118	0,
119	"SHPTree",
120	sizeof(SHPTreeObject),
121	0,
122	(destructor)shptree_dealloc, /tp_dealloc/
123	(printfunc)NULL, /tp_print/
124	shptree_getattr, /tp_getattr/
125	0, /tp_setattr/
126	0, /tp_compare/
127	(reprfunc)shptree_repr, /tp_repr/
128	0, /tp_as_number/
129	0, /tp_as_sequence/
130	0, /tp_as_mapping/
131	0, /tp_hash/
132	0, /tp_call/
133	0, /tp_str/
134	0, /tp_getattro/
135	0, /tp_setattro/
136	0, /tp_as_buffer/
137	};
138
139
140
141	static PyObject *
142	shptree_from_shapefile(PyObject * self, PyObject * args)
143	{
144	SHPTree * tree;
145	SHPHandle handle;
146	PyObject * cobject;
147	int dimension, max_depth;
148
149	if (!PyArg_ParseTuple(args, "O!ii", &PyCObject_Type, &cobject,
150	&dimension, &max_depth))
151	return NULL;
152
153	handle = PyCObject_AsVoidPtr(cobject);
154
155	tree = api->SHPCreateTree(handle, dimension, max_depth, NULL, NULL);
156
157	/* apparently SHPCreateTree doesn't do any error checking, so we
158	* have to assume that tree is valid at this point. */
159	return SHPTreeObject_FromSHPTree(tree);
160	}
161
162
163	static PyMethodDef module_functions[] = {
164	{"SHPTree", shptree_from_shapefile, METH_VARARGS},
165	{ NULL, NULL }
166	};
167
168
169	void
170	initshptree()
171	{
172	SHPTreeType.ob_type = &PyType_Type;
173
174	Py_InitModule("shptree", module_functions);
175	PYSHAPELIB_IMPORT_API(api);
176	}
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision