1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
/*************************************************************************
* Copyright (C) 2009-2011 Tavian Barnes <tavianator@tavianator.com> *
* *
* This file is part of The Dimension Python Module. *
* *
* The Dimension Python Module is free software; you can redistribute it *
* and/ or modify it under the terms of the GNU Lesser General Public *
* License as published by the Free Software Foundation; either version *
* 3 of the License, or (at your option) any later version. *
* *
* The Dimension Python Module is distributed in the hope that it will *
* be useful, but WITHOUT ANY WARRANTY; without even the implied *
* warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See *
* the GNU Lesser General Public License for more details. *
* *
* You should have received a copy of the GNU Lesser General Public *
* License along with this program. If not, see *
* <http://www.gnu.org/licenses/>. *
*************************************************************************/
#include "dimension-python.h"
static int
dmnsn_py_Scene_init(dmnsn_py_Scene *self, PyObject *args, PyObject *kwds)
{
dmnsn_delete_scene(self->scene);
self->scene = dmnsn_new_scene();
static char *kwlist[] = { "canvas", "camera", "objects", NULL };
dmnsn_py_Canvas *canvas;
dmnsn_py_Camera *camera;
PyObject *objects;
if (!PyArg_ParseTupleAndKeywords(args, kwds, "O!O!O!", kwlist,
&dmnsn_py_CanvasType, &canvas,
&dmnsn_py_CameraType, &camera,
&PyList_Type, &objects))
return -1;
dmnsn_scene_set_canvas(self->scene, canvas->canvas);
if (!PyObject_CallMethod((PyObject *)camera, "initialize", ""))
return -1;
/* Account for image proportions in camera */
dmnsn_matrix stretch = dmnsn_scale_matrix(
dmnsn_new_vector(
(double)canvas->canvas->width/canvas->canvas->height,
1.0,
1.0
)
);
camera->camera->trans = dmnsn_matrix_mul(camera->camera->trans, stretch);
dmnsn_scene_set_camera(self->scene, camera->camera);
self->scene->background = dmnsn_white;
Py_ssize_t size = PyList_Size(objects);
for (Py_ssize_t i = 0; i < size; ++i) {
PyObject *obj = PyList_GetItem(objects, i);
if (!PyObject_TypeCheck(obj, &dmnsn_py_ObjectType)) {
PyErr_SetString(PyExc_TypeError, "Expected a list of objects");
return -1;
}
if (!PyObject_CallMethod(obj, "initialize", ""))
return -1;
dmnsn_py_Object *object = (dmnsn_py_Object *)obj;
dmnsn_scene_add_object(self->scene, object->object);
}
return 0;
}
static void
dmnsn_py_Scene_dealloc(dmnsn_py_Scene *self)
{
dmnsn_delete_scene(self->scene);
Py_TYPE(self)->tp_free((PyObject *)self);
}
static PyObject *
dmnsn_py_Scene_raytrace(dmnsn_py_Scene *self)
{
dmnsn_raytrace_scene(self->scene);
Py_INCREF(Py_None);
return Py_None;
}
static PyMethodDef dmnsn_py_Scene_methods[] = {
{ "raytrace", (PyCFunction)dmnsn_py_Scene_raytrace, METH_NOARGS,
"Raytrace a scene" },
{ NULL }
};
static PyGetSetDef dmnsn_py_Scene_getsetters[] = {
{ NULL }
};
PyTypeObject dmnsn_py_SceneType = {
PyVarObject_HEAD_INIT(NULL, 0)
.tp_name = "dimension.Scene",
.tp_basicsize = sizeof(dmnsn_py_Scene),
.tp_dealloc = (destructor)dmnsn_py_Scene_dealloc,
.tp_flags = Py_TPFLAGS_DEFAULT,
.tp_doc = "Dimension scene",
.tp_methods = dmnsn_py_Scene_methods,
.tp_getset = dmnsn_py_Scene_getsetters,
.tp_init = (initproc)dmnsn_py_Scene_init,
};
bool
dmnsn_py_init_SceneType(void)
{
dmnsn_py_SceneType.tp_new = PyType_GenericNew;
return PyType_Ready(&dmnsn_py_SceneType) >= 0;
}
|
