diff options
author | Michel Selten <michel@mselten.demon.nl> | 2003-04-08 23:54:14 +0400 |
---|---|---|
committer | Michel Selten <michel@mselten.demon.nl> | 2003-04-08 23:54:14 +0400 |
commit | ec669df6eec3188938d9d84dd4666eac8b5aa1d1 (patch) | |
tree | 39985e13620956eb4e9d708b2d880ed8b03d0bd6 /source/blender/python/api2_2x/Lamp.c | |
parent | 0850182872fef788d8cf128f3a5901dca142fa8f (diff) |
The following updates have been contributed by Willian P. Germano:
* Implemented BPY_end_python function.
* Implemented error handling. This results in rerunning a script after an
error has occurred. No need to restart blender anymore.
* Camera module supports dir()
* variable assignment now calls the Python equivalent function - this has
type checking and should be safer now.
* Implemented the Lamp module. Used the Camera module as a template.
* Implemented the Image module.
* Added EXPP_ClampFloat and EXPP_intError functions to gen_utils.[ch]
* Implemented 'constant' object.
Diffstat (limited to 'source/blender/python/api2_2x/Lamp.c')
-rw-r--r-- | source/blender/python/api2_2x/Lamp.c | 1281 |
1 files changed, 1281 insertions, 0 deletions
diff --git a/source/blender/python/api2_2x/Lamp.c b/source/blender/python/api2_2x/Lamp.c new file mode 100644 index 00000000000..7247b411ef1 --- /dev/null +++ b/source/blender/python/api2_2x/Lamp.c @@ -0,0 +1,1281 @@ +/* + * + * ***** BEGIN GPL/BL DUAL LICENSE BLOCK ***** + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. The Blender + * Foundation also sells licenses for use in proprietary software under + * the Blender License. See http://www.blender.org/BL/ for information + * about this. + * + * This program 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 General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software Foundation, + * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. + * + * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV. + * All rights reserved. + * + * This is a new part of Blender. + * + * Contributor(s): Willian P. Germano + * + * ***** END GPL/BL DUAL LICENSE BLOCK ***** +*/ + +#include "Lamp.h" + +/*****************************************************************************/ +/* Function: M_Lamp_New */ +/* Python equivalent: Blender.Lamp.New */ +/*****************************************************************************/ +static PyObject *M_Lamp_New(PyObject *self, PyObject *args, PyObject *keywords) +{ + char *type_str = "Lamp"; + char *name_str = "Data"; + static char *kwlist[] = {"type_str", "name_str", NULL}; + C_Lamp *lamp; + PyObject *type, *name; + int type_int; + char buf[21]; + + printf ("In Lamp_New()\n"); + + if (!PyArg_ParseTupleAndKeywords(args, keywords, "|ss", kwlist, + &type_str, &name_str)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected string(s) or empty argument")); + + if (!strcmp (type_str, "Lamp")) + type_int = EXPP_LAMP_TYPE_LAMP; + else if (!strcmp (type_str, "Sun")) + type_int = EXPP_LAMP_TYPE_SUN; + else if (!strcmp (type_str, "Spot")) + type_int = EXPP_LAMP_TYPE_SPOT; + else if (!strcmp (type_str, "Hemi")) + type_int = EXPP_LAMP_TYPE_HEMI; + else + return (PythonReturnErrorObject (PyExc_AttributeError, + "unknown lamp type")); + + lamp = (C_Lamp *)LampCreatePyObject(NULL); + + if (lamp == NULL) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create Lamp Data object")); + + type = PyInt_FromLong(type_int); + if (type) LampSetAttr(lamp, "type", type); + else { + Py_DECREF((PyObject *)lamp); + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create Python string")); + } + + if (strcmp(name_str, "Data") == 0) + return (PyObject *)lamp; + + PyOS_snprintf(buf, sizeof(buf), "%s", name_str); + name = PyString_FromString(buf); + if (name) LampSetAttr(lamp, "name", name); + else { + Py_DECREF((PyObject *)lamp); + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create Python string")); + } + + return (PyObject *)lamp; +} + +/*****************************************************************************/ +/* Function: M_Lamp_Get */ +/* Python equivalent: Blender.Lamp.Get */ +/*****************************************************************************/ +static PyObject *M_Lamp_Get(PyObject *self, PyObject *args) +{ + char *name; + Lamp *lamp_iter; + C_Lamp *wanted_lamp; + + printf ("In Lamp_Get()\n"); + if (!PyArg_ParseTuple(args, "s", &name)) + { + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected string argument")); + } + + /* Use the name to search for the lamp requested. */ + wanted_lamp = NULL; + lamp_iter = G.main->lamp.first; + + while ((lamp_iter) && (wanted_lamp == NULL)) { + + if (strcmp (name, GetIdName (&(lamp_iter->id))) == 0) + wanted_lamp = (C_Lamp *)LampCreatePyObject(lamp_iter); + + lamp_iter = lamp_iter->id.next; + } + + if (wanted_lamp == NULL) { + /* No lamp exists with the name specified in the argument name. */ + char error_msg[64]; + PyOS_snprintf(error_msg, sizeof(error_msg), + "Lamp \"%s\" not found", name); + return (PythonReturnErrorObject (PyExc_NameError, error_msg)); + } + wanted_lamp->linked = 1; /* TRUE: linked to a Blender Lamp Object */ + return ((PyObject*)wanted_lamp); +} + +/*****************************************************************************/ +/* Function: M_Lamp_Init */ +/*****************************************************************************/ +PyObject *M_Lamp_Init (void) +{ + PyObject *module; + + printf ("In M_Lamp_Init()\n"); + + module = Py_InitModule3("Lamp", M_Lamp_methods, M_Lamp_doc); + + return (module); +} + +/*****************************************************************************/ +/* Python C_Lamp methods: */ +/*****************************************************************************/ +static PyObject *Lamp_getName(C_Lamp *self) +{ + PyObject *attr; + attr = PyDict_GetItemString(self->dict, "name"); + if (attr) { + Py_INCREF(attr); + return attr; + } + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't get Lamp.name attribute")); +} + +static PyObject *Lamp_getType(C_Lamp *self) +{ + PyObject *attr; + attr = PyDict_GetItemString(self->dict, "type"); + if (attr) { + Py_INCREF(attr); + return attr; + } + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't get Lamp.type attribute")); +} + +static PyObject *Lamp_getMode(C_Lamp *self) +{/* XXX improve this, add the constants to the Lamp dict */ + PyObject *attr; + attr = PyDict_GetItemString(self->dict, "mode"); + if (attr) { + Py_INCREF(attr); + return attr; + } + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't get Lamp.mode attribute")); +} + +static PyObject *Lamp_getSamples(C_Lamp *self) +{ + PyObject *attr; + attr = PyDict_GetItemString(self->dict, "samples"); + if (attr) { + Py_INCREF(attr); + return attr; + } + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't get Lamp.samples attribute")); +} + +static PyObject *Lamp_getBufferSize(C_Lamp *self) +{ + PyObject *attr; + attr = PyDict_GetItemString(self->dict, "bufferSize"); + if (attr) { + Py_INCREF(attr); + return attr; + } + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't get Lamp.bufferSize attribute")); +} + +static PyObject *Lamp_getHaloStep(C_Lamp *self) +{ + PyObject *attr; + attr = PyDict_GetItemString(self->dict, "haloStep"); + if (attr) { + Py_INCREF(attr); + return attr; + } + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't get Lamp.haloStep attribute")); +} + +static PyObject *Lamp_getEnergy(C_Lamp *self) +{ + PyObject *attr; + attr = PyDict_GetItemString(self->dict, "energy"); + if (attr) { + Py_INCREF(attr); + return attr; + } + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't get Lamp.energy attribute")); +} + +static PyObject *Lamp_getDist(C_Lamp *self) +{ + PyObject *attr; + attr = PyDict_GetItemString(self->dict, "dist"); + if (attr) { + Py_INCREF(attr); + return attr; + } + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't get Lamp.dist attribute")); +} + +static PyObject *Lamp_getSpotSize(C_Lamp *self) +{ + PyObject *attr; + attr = PyDict_GetItemString(self->dict, "spotSize"); + if (attr) { + Py_INCREF(attr); + return attr; + } + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't get Lamp.spotSize attribute")); +} + +static PyObject *Lamp_getSpotBlend(C_Lamp *self) +{ + PyObject *attr; + attr = PyDict_GetItemString(self->dict, "spotBlend"); + if (attr) { + Py_INCREF(attr); + return attr; + } + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't get Lamp.spotBlend attribute")); +} + +static PyObject *Lamp_getClipStart(C_Lamp *self) +{ + PyObject *attr; + attr = PyDict_GetItemString(self->dict, "clipStart"); + if (attr) { + Py_INCREF(attr); + return attr; + } + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't get Lamp.clipStart attribute")); +} + +static PyObject *Lamp_getClipEnd(C_Lamp *self) +{ + PyObject *attr; + attr = PyDict_GetItemString(self->dict, "clipEnd"); + if (attr) { + Py_INCREF(attr); + return attr; + } + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't get Lamp.clipEnd attribute")); +} + +static PyObject *Lamp_getBias(C_Lamp *self) +{ + PyObject *attr; + attr = PyDict_GetItemString(self->dict, "bias"); + if (attr) { + Py_INCREF(attr); + return attr; + } + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't get Lamp.bias attribute")); +} + +static PyObject *Lamp_getSoftness(C_Lamp *self) +{ + PyObject *attr; + attr = PyDict_GetItemString(self->dict, "softness"); + if (attr) { + Py_INCREF(attr); + return attr; + } + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't get Lamp.softness attribute")); +} + +static PyObject *Lamp_getHaloInt(C_Lamp *self) +{ + PyObject *attr; + attr = PyDict_GetItemString(self->dict, "haloInt"); + if (attr) { + Py_INCREF(attr); + return attr; + } + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't get Lamp.haloInt attribute")); +} + +static PyObject *Lamp_rename(C_Lamp *self, PyObject *args) +{ + char *name_str; + char buf[21]; + PyObject *name; + + if (!PyArg_ParseTuple(args, "s", &name_str)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected string argument")); + + PyOS_snprintf(buf, sizeof(buf), "%s", name_str); + + if (self->linked) { /* update the Blender Lamp, too */ + ID *tmp_id = &self->lamp->id; + rename_id(tmp_id, buf); + PyOS_snprintf(buf, sizeof(buf), "%s", tmp_id->name+2);/* may have changed */ + } + + name = PyString_FromString(buf); + + if (!name) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyString Object")); + + if (PyDict_SetItemString(self->dict, "name", name) != 0) { + Py_DECREF(name); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't set Lamp.name attribute")); + } + + Py_INCREF(Py_None); + return Py_None; +} + +static PyObject *Lamp_setType(C_Lamp *self, PyObject *args) +{ + short value; + char *type_str; + PyObject *type; + + if (!PyArg_ParseTuple(args, "s", &type_str)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected one string argument")); + + if (strcmp (type_str, "Lamp") == 0) + value = EXPP_LAMP_TYPE_LAMP; + else if (strcmp (type_str, "Sun") == 0) + value = EXPP_LAMP_TYPE_SUN; + else if (strcmp (type_str, "Spot") == 0) + value = EXPP_LAMP_TYPE_SPOT; + else if (strcmp (type_str, "Hemi") == 0) + value = EXPP_LAMP_TYPE_HEMI; + else + return (PythonReturnErrorObject (PyExc_AttributeError, + "unknown lamp type")); + + type = PyInt_FromLong(value); + if (!type) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyInt Object")); + + if (PyDict_SetItemString(self->dict, "type", type) != 0) { + Py_DECREF(type); + return (PythonReturnErrorObject (PyExc_AttributeError, + "couldn't set Lamp.type attribute")); + } + + if (self->linked) /* update the Blender Lamp, too */ + self->lamp->type = value; + + Py_INCREF(Py_None); + return Py_None; +} + +/* This one is 'private'. It is not really a method, just a helper function for + * when script writers use Lamp.type = t instead of Lamp.setType(t), since in + * the first case t shoud be an int and in the second it should be a string. So + * while the method setType expects a string ('persp' or 'ortho') or an empty + * argument, this function should receive an int (0 or 1). */ +static PyObject *Lamp_setIntType(C_Lamp *self, PyObject *args) +{ + short value; + PyObject *type; + + if (!PyArg_ParseTuple(args, "h", &value)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected int argument in [0,3]")); + + if (value >= 0 && value <= 3) + type = PyInt_FromLong(value); + else + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected int argument in [0,3]")); + + if (!type) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyInt Object")); + + if (PyDict_SetItemString(self->dict, "type", type) != 0) { + Py_DECREF(type); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "could not set Lamp.type attribute")); + } + + if (self->linked) /* update the Blender Lamp, too */ + self->lamp->type = value; + + Py_INCREF(Py_None); + return Py_None; +} + +static PyObject *Lamp_setMode(C_Lamp *self, PyObject *args) +{/* Quad, Sphere, Shadows, Halo, Layer, Negative, OnlyShadow, Square */ + char *m[8] = {NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL}; + short i, flag = 0; + PyObject *mode; + + if (!PyArg_ParseTuple(args, "|ssssssss", &m[0], &m[1], &m[2], + &m[3], &m[4], &m[5], &m[6], &m[7])) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected string argument(s)")); + + for (i = 0; i < 8; i++) { + if (m[i] == NULL) break; + if (strcmp(m[i], "Shadows") == 0) + flag |= EXPP_LAMP_MODE_SHADOWS; + else if (strcmp(m[i], "Halo") == 0) + flag |= EXPP_LAMP_MODE_HALO; + else if (strcmp(m[i], "Layer") == 0) + flag |= EXPP_LAMP_MODE_LAYER; + else if (strcmp(m[i], "Quad") == 0) + flag |= EXPP_LAMP_MODE_QUAD; + else if (strcmp(m[i], "Negative") == 0) + flag |= EXPP_LAMP_MODE_NEGATIVE; + else if (strcmp(m[i], "OnlyShadow") == 0) + flag |= EXPP_LAMP_MODE_ONLYSHADOW; + else if (strcmp(m[i], "Sphere") == 0) + flag |= EXPP_LAMP_MODE_SPHERE; + else if (strcmp(m[i], "Square") == 0) + flag |= EXPP_LAMP_MODE_SQUARE; + else + return (PythonReturnErrorObject (PyExc_AttributeError, + "unknown lamp flag argument")); + } + + mode = PyInt_FromLong(flag); + if (!mode) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyInt Object")); + + if (PyDict_SetItemString(self->dict, "mode", mode) != 0) { + Py_DECREF(mode); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't set Lamp.mode attribute")); + } + + if (self->linked) /* update the Blender Lamp, too */ + self->lamp->mode = flag; + + Py_INCREF(Py_None); + return Py_None; +} + +/* Another helper function, for the same reason. + * (See comment before Lamp_setIntType above). */ +static PyObject *Lamp_setIntMode(C_Lamp *self, PyObject *args) +{ + short value; + PyObject *mode; + + if (!PyArg_ParseTuple(args, "h", &value)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected int argument")); + +/* well, with so many flag bits, we just accept any short int, no checking */ + mode = PyInt_FromLong(value); + + if (!mode) + return (PythonReturnErrorObject (PyExc_AttributeError, + "couldn't create PyInt object")); + + if (PyDict_SetItemString(self->dict, "mode", mode) != 0) { + Py_DECREF(mode); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "could not set Lamp.mode attribute")); + } + + if (self->linked) /* update the Blender Lamp, too */ + self->lamp->mode = value; + + Py_INCREF(Py_None); + return Py_None; +} + +static PyObject *Lamp_setSamples(C_Lamp *self, PyObject *args) +{ + short value; + PyObject *samples; + + if (!PyArg_ParseTuple(args, "h", &value)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected int argument in [1,16]")); + + if (value >= EXPP_LAMP_SAMPLES_MIN && + value <= EXPP_LAMP_SAMPLES_MAX) + samples = PyInt_FromLong(value); + else + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected int argument in [1,16]")); + + if (!samples) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyInt Object")); + + if (PyDict_SetItemString(self->dict, "samples", samples) != 0) { + Py_DECREF(samples); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "could not set Lamp.samples attribute")); + } + + if (self->linked) /* update the Blender Lamp, too */ + self->lamp->samp = value; + + Py_INCREF(Py_None); + return Py_None; +} + +static PyObject *Lamp_setBufferSize(C_Lamp *self, PyObject *args) +{ + short value; + PyObject *bufferSize; + + if (!PyArg_ParseTuple(args, "h", &value)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected int argument, any of [512, 768, 1024, 1536, 2560]")); + + switch (value) { + case 512: + case 768: + case 1024: + case 1536: + case 2560: + bufferSize = PyInt_FromLong(value); + break; + default: + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected int argument, any of [512, 768, 1024, 1536, 2560]")); + } + + if (!bufferSize) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyInt Object")); + + if (PyDict_SetItemString(self->dict, "bufferSize", bufferSize) != 0) { + Py_DECREF(bufferSize); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "could not set Lamp.bufferSize attribute")); + } + + if (self->linked) /* update the Blender Lamp, too */ + self->lamp->bufsize = value; + + Py_INCREF(Py_None); + return Py_None; +} + +static PyObject *Lamp_setHaloStep(C_Lamp *self, PyObject *args) +{ + short value; + PyObject *haloStep; + + if (!PyArg_ParseTuple(args, "h", &value)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected int argument in [0,12]")); + + if (value >= EXPP_LAMP_HALOSTEP_MIN && + value <= EXPP_LAMP_HALOSTEP_MAX) + haloStep = PyInt_FromLong(value); + else + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected int argument in [0,12]")); + + if (!haloStep) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyInt Object")); + + if (PyDict_SetItemString(self->dict, "haloStep", haloStep) != 0) { + Py_DECREF(haloStep); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't set Lamp.haloStep attribute")); + } + + if (self->linked) /* update the Blender Lamp, too */ + self->lamp->shadhalostep = value; + + Py_INCREF(Py_None); + return Py_None; +} + +static PyObject *Lamp_setColorComponent(C_Lamp *self, char *key, PyObject *args) +{ + float value; + PyObject *component; + + if (!PyArg_ParseTuple(args, "f", &value)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected float argument in [0.0, 1.0]")); + + value = EXPP_ClampFloat (value, 0.0, 1.0); + component = PyFloat_FromDouble(value); + + if (!component) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyFloat Object")); + + if (PyDict_SetItemString(self->dict, key, component) != 0) { + Py_DECREF(component); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't set Lamp color component attribute")); + } + + if (self->linked) { /* update the Blender Lamp, too */ + if (!strcmp(key, "R")) + self->lamp->r = value; + else if (!strcmp(key, "G")) + self->lamp->g = value; + else if (!strcmp(key, "B")) + self->lamp->b = value; + } + + Py_INCREF(Py_None); + return Py_None; +} + +static PyObject *Lamp_setEnergy(C_Lamp *self, PyObject *args) +{ + float value; + PyObject *energy; + + if (!PyArg_ParseTuple(args, "f", &value)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected float argument")); + + value = EXPP_ClampFloat (value, EXPP_LAMP_ENERGY_MIN, EXPP_LAMP_ENERGY_MAX); + energy = PyFloat_FromDouble(value); + + if (!energy) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyFloat Object")); + + if (PyDict_SetItemString(self->dict, "energy", energy) != 0) { + Py_DECREF(energy); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't set Lamp.energy attribute")); + } + + if (self->linked) /* update the Blender Lamp, too */ + self->lamp->energy = value; + + Py_INCREF(Py_None); + return Py_None; +} + +static PyObject *Lamp_setDist(C_Lamp *self, PyObject *args) +{ + float value; + PyObject *dist; + + if (!PyArg_ParseTuple(args, "f", &value)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected float argument")); + + value = EXPP_ClampFloat (value, EXPP_LAMP_DIST_MIN, EXPP_LAMP_DIST_MAX); + dist = PyFloat_FromDouble(value); + + if (!dist) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyFloat Object")); + + if (PyDict_SetItemString(self->dict, "dist", dist) != 0) { + Py_DECREF(dist); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't set Lamp.dist attribute")); + } + + if (self->linked) /* update the Blender Lamp, too */ + self->lamp->dist = value; + + Py_INCREF(Py_None); + return Py_None; +} + +static PyObject *Lamp_setSpotSize(C_Lamp *self, PyObject *args) +{ + float value; + PyObject *spotSize; + + if (!PyArg_ParseTuple(args, "f", &value)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected float argument")); + + value = EXPP_ClampFloat (value, EXPP_LAMP_SPOTSIZE_MIN, EXPP_LAMP_SPOTSIZE_MAX); + spotSize = PyFloat_FromDouble(value); + + if (!spotSize) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyFloat Object")); + + if (PyDict_SetItemString(self->dict, "spotSize", spotSize) != 0) { + Py_DECREF(spotSize); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't set Lamp.spotSize attribute")); + } + + if (self->linked) /* update the Blender Lamp, too */ + self->lamp->spotsize = value; + + Py_INCREF(Py_None); + return Py_None; +} + +static PyObject *Lamp_setSpotBlend(C_Lamp *self, PyObject *args) +{ + float value; + PyObject *spotBlend; + + if (!PyArg_ParseTuple(args, "f", &value)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected float argument")); + + value = EXPP_ClampFloat (value, EXPP_LAMP_SPOTBLEND_MIN, + EXPP_LAMP_SPOTBLEND_MAX); + spotBlend = PyFloat_FromDouble(value); + + if (!spotBlend) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyFloat Object")); + + if (PyDict_SetItemString(self->dict, "spotBlend", spotBlend) != 0) { + Py_DECREF(spotBlend); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't set Lamp.spotBlend attribute")); + } + + if (self->linked) /* update the Blender Lamp, too */ + self->lamp->spotblend = value; + + Py_INCREF(Py_None); + return Py_None; +} + +static PyObject *Lamp_setClipStart(C_Lamp *self, PyObject *args) +{ + float value; + PyObject *clipStart; + + if (!PyArg_ParseTuple(args, "f", &value)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected a float number as argument")); + + value = EXPP_ClampFloat (value, EXPP_LAMP_CLIPSTART_MIN, + EXPP_LAMP_CLIPSTART_MAX); + + clipStart = PyFloat_FromDouble(value); + + if (!clipStart) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyFloat Object")); + + if (PyDict_SetItemString(self->dict, "clipStart", clipStart) != 0) { + Py_DECREF(clipStart); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't set Lamp.clipStart attribute")); + } + + if (self->linked) /* update the Blender Lamp, too */ + self->lamp->clipsta = value; + + Py_INCREF(Py_None); + return Py_None; +} + +static PyObject *Lamp_setClipEnd(C_Lamp *self, PyObject *args) +{ + float value; + PyObject *clipEnd; + + if (!PyArg_ParseTuple(args, "f", &value)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected a float number as argument")); + + value = EXPP_ClampFloat (value, EXPP_LAMP_CLIPEND_MIN, + EXPP_LAMP_CLIPEND_MAX); + + clipEnd = PyFloat_FromDouble(value); + + if (!clipEnd) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyFloat Object")); + + if (PyDict_SetItemString(self->dict, "clipEnd", clipEnd) != 0) { + Py_DECREF(clipEnd); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't set Lamp.clipEnd attribute")); + } + + if (self->linked) /* update the Blender Lamp, too */ + self->lamp->clipend = value; + + Py_INCREF(Py_None); + return Py_None; +} + +static PyObject *Lamp_setBias(C_Lamp *self, PyObject *args) +{ + float value; + PyObject *bias; + + if (!PyArg_ParseTuple(args, "f", &value)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected a float number as argument")); + + value = EXPP_ClampFloat (value, EXPP_LAMP_BIAS_MIN, EXPP_LAMP_BIAS_MAX); + + bias = PyFloat_FromDouble(value); + + if (!bias) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyFloat Object")); + + if (PyDict_SetItemString(self->dict, "bias", bias) != 0) { + Py_DECREF(bias); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't set Lamp.bias attribute")); + } + + if (self->linked) /* update the Blender Lamp, too */ + self->lamp->bias = value; + + Py_INCREF(Py_None); + return Py_None; +} + +static PyObject *Lamp_setSoftness(C_Lamp *self, PyObject *args) +{ + float value; + PyObject *softness; + + if (!PyArg_ParseTuple(args, "f", &value)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected a float number as argument")); + + value = EXPP_ClampFloat (value, EXPP_LAMP_SOFTNESS_MIN, + EXPP_LAMP_SOFTNESS_MAX); + + softness = PyFloat_FromDouble(value); + + if (!softness) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyFloat Object")); + + if (PyDict_SetItemString(self->dict, "softness", softness) != 0) { + Py_DECREF(softness); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't set Lamp.softness attribute")); + } + + if (self->linked) /* update the Blender Lamp, too */ + self->lamp->soft = value; + + Py_INCREF(Py_None); + return Py_None; +} + +static PyObject *Lamp_setHaloInt(C_Lamp *self, PyObject *args) +{ + float value; + PyObject *haloInt; + + if (!PyArg_ParseTuple(args, "f", &value)) + return (PythonReturnErrorObject (PyExc_AttributeError, + "expected a float number as argument")); + + value = EXPP_ClampFloat (value, EXPP_LAMP_HALOINT_MIN, + EXPP_LAMP_HALOINT_MAX); + + haloInt = PyFloat_FromDouble(value); + + if (!haloInt) + return (PythonReturnErrorObject (PyExc_MemoryError, + "couldn't create PyFloat Object")); + + if (PyDict_SetItemString(self->dict, "haloInt", haloInt) != 0) { + Py_DECREF(haloInt); + return (PythonReturnErrorObject (PyExc_RuntimeError, + "couldn't set Lamp.haloInt attribute")); + } + + if (self->linked) /* update the Blender Lamp, too */ + self->lamp->haint = value; + + Py_INCREF(Py_None); + return Py_None; +} + +/*****************************************************************************/ +/* Function: LampCreatePyObject */ +/* Description: This function will create a new C_Lamp. If the Lamp */ +/* struct passed to it is not NULL, it'll use its attributes. */ +/*****************************************************************************/ +PyObject *LampCreatePyObject (Lamp *blenderLamp) +{ + PyObject *name, *type, *mode, *R, *G, *B, *energy, *dist; + PyObject *spotSize, *spotBlend, *clipStart, *clipEnd; + PyObject *bias, *softness, *samples, *bufferSize; + PyObject *haloInt, *haloStep; + PyObject *Types, *Lamp, *Sun, *Spot, *Hemi; + PyObject *Modes, *Shadows, *Halo, *Layer, *Quad; + PyObject *Negative, *OnlyShadow, *Sphere, *Square; + C_Lamp *lamp; + + printf ("In LampCreatePyObject\n"); + + lamp = (C_Lamp *)PyObject_NEW(C_Lamp, &Lamp_Type); + + if (lamp == NULL) + return NULL; + + lamp->linked = 0; + lamp->dict = PyDict_New(); + + if (lamp->dict == NULL) { + Py_DECREF((PyObject *)lamp); + return NULL; + } + + if (blenderLamp == NULL) { /* Not linked to a Lamp Object yet */ + name = PyString_FromString("Data"); + type = PyInt_FromLong(EXPP_LAMP_TYPE); + mode = PyInt_FromLong(EXPP_LAMP_MODE); + samples = PyInt_FromLong(EXPP_LAMP_SAMPLES); + bufferSize = PyInt_FromLong(EXPP_LAMP_BUFFERSIZE); + haloStep = PyInt_FromLong(EXPP_LAMP_HALOSTEP); + R = PyFloat_FromDouble(1.0); + G = PyFloat_FromDouble(1.0); + B = PyFloat_FromDouble(1.0); + energy = PyFloat_FromDouble(EXPP_LAMP_ENERGY); + dist = PyFloat_FromDouble(EXPP_LAMP_DIST); + spotSize = PyFloat_FromDouble(EXPP_LAMP_SPOTSIZE); + spotBlend = PyFloat_FromDouble(EXPP_LAMP_SPOTBLEND); + clipStart = PyFloat_FromDouble(EXPP_LAMP_CLIPSTART); + clipEnd = PyFloat_FromDouble(EXPP_LAMP_CLIPEND); + bias = PyFloat_FromDouble(EXPP_LAMP_BIAS); + softness = PyFloat_FromDouble(EXPP_LAMP_SOFTNESS); + haloInt = PyFloat_FromDouble(EXPP_LAMP_HALOINT); + } + else { /* Lamp Object available, get its attributes directly */ + name = PyString_FromString(blenderLamp->id.name+2); + type = PyInt_FromLong(blenderLamp->type); + mode = PyInt_FromLong(blenderLamp->mode); + samples = PyInt_FromLong(blenderLamp->samp); + bufferSize = PyInt_FromLong(blenderLamp->bufsize); + haloStep = PyInt_FromLong(blenderLamp->shadhalostep); + R = PyFloat_FromDouble(blenderLamp->r); + G = PyFloat_FromDouble(blenderLamp->g); + B = PyFloat_FromDouble(blenderLamp->b); + energy = PyFloat_FromDouble(blenderLamp->energy); + dist = PyFloat_FromDouble(blenderLamp->dist); + spotSize = PyFloat_FromDouble(blenderLamp->spotsize); + spotBlend = PyFloat_FromDouble(blenderLamp->spotblend); + clipStart = PyFloat_FromDouble(blenderLamp->clipsta); + clipEnd = PyFloat_FromDouble(blenderLamp->clipend); + bias = PyFloat_FromDouble(blenderLamp->bias); + softness = PyFloat_FromDouble(blenderLamp->soft); + haloInt = PyFloat_FromDouble(blenderLamp->haint); + /*there's shadspotsize, too ... plus others, none in 2.25*/ + } + + Types = constant_New(); + Lamp = PyInt_FromLong(EXPP_LAMP_TYPE_LAMP); + Sun = PyInt_FromLong(EXPP_LAMP_TYPE_SUN); + Spot = PyInt_FromLong(EXPP_LAMP_TYPE_SPOT); + Hemi = PyInt_FromLong(EXPP_LAMP_TYPE_HEMI); + + Modes = constant_New(); + Shadows = PyInt_FromLong(EXPP_LAMP_MODE_SHADOWS); + Halo = PyInt_FromLong(EXPP_LAMP_MODE_HALO); + Layer = PyInt_FromLong(EXPP_LAMP_MODE_LAYER); + Quad = PyInt_FromLong(EXPP_LAMP_MODE_QUAD); + Negative = PyInt_FromLong(EXPP_LAMP_MODE_NEGATIVE); + OnlyShadow = PyInt_FromLong(EXPP_LAMP_MODE_ONLYSHADOW); + Sphere = PyInt_FromLong(EXPP_LAMP_MODE_SPHERE); + Square = PyInt_FromLong(EXPP_LAMP_MODE_SQUARE); + + if (name == NULL || type == NULL || mode == NULL || + samples == NULL || bufferSize == NULL || haloStep == NULL || + R == NULL || G == NULL || B == NULL || energy == NULL || + dist == NULL || spotSize == NULL || spotBlend == NULL || + clipStart == NULL || clipEnd == NULL || bias == NULL || + softness == NULL || haloInt == NULL || Types == NULL || + Lamp == NULL || Sun == NULL || Spot == NULL || Hemi == NULL || + Modes == NULL || Shadows == NULL || Halo == NULL || + Layer == NULL || Quad == NULL || Negative == NULL || + OnlyShadow == NULL || Sphere == NULL || Square == NULL) /* ack! */ + goto fail; + + if ((PyDict_SetItemString(lamp->dict, "name", name) != 0) || + (PyDict_SetItemString(lamp->dict, "type", type) != 0) || + (PyDict_SetItemString(lamp->dict, "mode", mode) != 0) || + (PyDict_SetItemString(lamp->dict, "samples", samples) != 0) || + (PyDict_SetItemString(lamp->dict, "bufferSize", bufferSize) != 0) || + (PyDict_SetItemString(lamp->dict, "R", R) != 0) || + (PyDict_SetItemString(lamp->dict, "G", G) != 0) || + (PyDict_SetItemString(lamp->dict, "B", B) != 0) || + (PyDict_SetItemString(lamp->dict, "energy", energy) != 0) || + (PyDict_SetItemString(lamp->dict, "dist", dist) != 0) || + (PyDict_SetItemString(lamp->dict, "spotSize", spotSize) != 0) || + (PyDict_SetItemString(lamp->dict, "spotBlend", spotBlend) != 0) || + (PyDict_SetItemString(lamp->dict, "clipStart", clipStart) != 0) || + (PyDict_SetItemString(lamp->dict, "clipEnd", clipEnd) != 0) || + (PyDict_SetItemString(lamp->dict, "bias", bias) != 0) || + (PyDict_SetItemString(lamp->dict, "softness", softness) != 0) || + (PyDict_SetItemString(lamp->dict, "haloInt", haloInt) != 0) || + (PyDict_SetItemString(lamp->dict, "haloStep", haloStep) != 0) || + (PyDict_SetItemString(lamp->dict, "Types", Types) != 0) || + (PyDict_SetItemString(lamp->dict, "Modes", Modes) != 0) || + (PyDict_SetItemString(lamp->dict, "__members__", + PyDict_Keys(lamp->dict)) != 0)) + goto fail; + + if ((PyDict_SetItemString(((C_constant *)Types)->dict, + "Lamp", Lamp) != 0) || + (PyDict_SetItemString(((C_constant *)Types)->dict, + "Sun", Sun) != 0) || + (PyDict_SetItemString(((C_constant *)Types)->dict, + "Spot", Spot) != 0) || + (PyDict_SetItemString(((C_constant *)Types)->dict, + "Hemi", Hemi) != 0) || + (PyDict_SetItemString(((C_constant *)Modes)->dict, + "Shadows", Shadows) != 0) || + (PyDict_SetItemString(((C_constant *)Modes)->dict, + "Halo", Halo) != 0) || + (PyDict_SetItemString(((C_constant *)Modes)->dict, + "Layer", Layer) != 0) || + (PyDict_SetItemString(((C_constant *)Modes)->dict, + "Quad", Quad) != 0) || + (PyDict_SetItemString(((C_constant *)Modes)->dict, + "Negative", Negative) != 0) || + (PyDict_SetItemString(((C_constant *)Modes)->dict, + "OnlyShadow", OnlyShadow) != 0) || + (PyDict_SetItemString(((C_constant *)Modes)->dict, + "Sphere", Sphere) != 0) || + (PyDict_SetItemString(((C_constant *)Modes)->dict, + "Square", Square) != 0)) + goto fail; + + lamp->lamp = blenderLamp; /* it's NULL when creating only lamp "data" */ + return ((PyObject*)lamp); + +fail: + Py_XDECREF(name); + Py_XDECREF(type); + Py_XDECREF(mode); + Py_XDECREF(samples); + Py_XDECREF(bufferSize); + Py_XDECREF(R); + Py_XDECREF(G); + Py_XDECREF(B); + Py_XDECREF(energy); + Py_XDECREF(dist); + Py_XDECREF(spotSize); + Py_XDECREF(spotBlend); + Py_XDECREF(clipStart); + Py_XDECREF(clipEnd); + Py_XDECREF(bias); + Py_XDECREF(softness); + Py_XDECREF(haloInt); + Py_XDECREF(haloStep); + Py_XDECREF(Types); + Py_XDECREF(Lamp); + Py_XDECREF(Sun); + Py_XDECREF(Spot); + Py_XDECREF(Hemi); + Py_XDECREF(Modes); + Py_XDECREF(Shadows); + Py_XDECREF(Halo); + Py_XDECREF(Layer); + Py_XDECREF(Quad); + Py_XDECREF(Negative); + Py_XDECREF(OnlyShadow); + Py_XDECREF(Sphere); + Py_XDECREF(Square); + Py_DECREF(lamp->dict); + Py_DECREF((PyObject *)lamp); + return NULL; +} + +/*****************************************************************************/ +/* Function: LampDeAlloc */ +/* Description: This is a callback function for the C_Lamp type. It is */ +/* the destructor function. */ +/*****************************************************************************/ +static void LampDeAlloc (C_Lamp *self) +{ + Py_DECREF(self->dict); + PyObject_DEL (self); +} + +/*****************************************************************************/ +/* Function: LampGetAttr */ +/* Description: This is a callback function for the C_Lamp type. It is */ +/* the function that accesses C_Lamp member variables and */ +/* methods. */ +/*****************************************************************************/ +static PyObject* LampGetAttr (C_Lamp *lamp, char *name) +{/* first try the attributes dictionary */ + if (lamp->dict) { + PyObject *v = PyDict_GetItemString(lamp->dict, name); + if (v) { + Py_INCREF(v); /* was a borrowed ref */ + return v; + } + } +/* not an attribute, search the methods table */ + return Py_FindMethod(C_Lamp_methods, (PyObject *)lamp, name); +} + +/*****************************************************************************/ +/* Function: LampSetAttr */ +/* Description: This is a callback function for the C_Lamp type. It is the */ +/* function that changes Lamp Data members values. If this */ +/* data is linked to a Blender Lamp, it also gets updated. */ +/*****************************************************************************/ +static int LampSetAttr (C_Lamp *self, char *name, PyObject *value) +{ + PyObject *valtuple; + PyObject *error = NULL; + + if (self->dict == NULL) return -1; + +/* We're playing a trick on the Python API users here. Even if they use + * Lamp.member = val instead of Lamp.setMember(value), we end up using the + * function anyway, since it already has error checking, clamps to the right + * interval and updates the Blender Lamp structure when necessary. */ + + valtuple = PyTuple_New(1); /* the set* functions expect a tuple */ + + if (!valtuple) + return EXPP_intError(PyExc_MemoryError, + "LampSetAttr: couldn't create tuple"); + + if (PyTuple_SetItem(valtuple, 0, value) != 0) { + Py_DECREF(value); /* PyTuple_SetItem incref's value even when it fails */ + Py_DECREF(valtuple); + return EXPP_intError(PyExc_MemoryError, + "LampSetAttr: couldn't fill tuple"); + } + + if (strcmp (name, "name") == 0) + error = Lamp_rename (self, valtuple); + else if (strcmp (name, "type") == 0) + error = Lamp_setIntType (self, valtuple); /* special case */ + else if (strcmp (name, "mode") == 0) + error = Lamp_setIntMode (self, valtuple); /* special case */ + else if (strcmp (name, "samples") == 0) + error = Lamp_setSamples (self, valtuple); + else if (strcmp (name, "bufferSize") == 0) + error = Lamp_setBufferSize (self, valtuple); + else if (strcmp (name, "haloStep") == 0) + error = Lamp_setHaloStep (self, valtuple); + else if (strcmp (name, "R") == 0) + error = Lamp_setColorComponent (self, "R", valtuple); + else if (strcmp (name, "G") == 0) + error = Lamp_setColorComponent (self, "G", valtuple); + else if (strcmp (name, "B") == 0) + error = Lamp_setColorComponent (self, "B", valtuple); + else if (strcmp (name, "energy") == 0) + error = Lamp_setEnergy (self, valtuple); + else if (strcmp (name, "dist") == 0) + error = Lamp_setDist (self, valtuple); + else if (strcmp (name, "spotSize") == 0) + error = Lamp_setSpotSize (self, valtuple); + else if (strcmp (name, "spotBlend") == 0) + error = Lamp_setSpotBlend (self, valtuple); + else if (strcmp (name, "clipStart") == 0) + error = Lamp_setClipStart (self, valtuple); + else if (strcmp (name, "clipEnd") == 0) + error = Lamp_setClipEnd (self, valtuple); + else if (strcmp (name, "bias") == 0) + error = Lamp_setBias (self, valtuple); + else if (strcmp (name, "softness") == 0) + error = Lamp_setSoftness (self, valtuple); + else if (strcmp (name, "haloInt") == 0) + error = Lamp_setHaloInt (self, valtuple); + else { /* Error: no such member in the Lamp Data structure */ + Py_DECREF(value); + Py_DECREF(valtuple); + return (EXPP_intError (PyExc_AttributeError, + "attribute not found")); + } + + if (error == Py_None) return 0; /* normal exit */ + + Py_DECREF(value); + Py_DECREF(valtuple); + + return -1; +} + +/*****************************************************************************/ +/* Function: LampPrint */ +/* Description: This is a callback function for the C_Lamp type. It */ +/* builds a meaninful string to 'print' lamp objects. */ +/*****************************************************************************/ +static int LampPrint(C_Lamp *self, FILE *fp, int flags) +{ + char *lstate = "unlinked"; + char *name; + + if (self->linked) + lstate = "linked"; + + name = PyString_AsString(Lamp_getName(self)); + + fprintf(fp, "[Lamp \"%s\" (%s)]", name, lstate); + + return 0; +} + +/*****************************************************************************/ +/* Function: LampRepr */ +/* Description: This is a callback function for the C_Lamp type. It */ +/* builds a meaninful string to represent lamp objects. */ +/*****************************************************************************/ +static PyObject *LampRepr (C_Lamp *self) +{ + char buf[64]; + char *lstate = "unlinked"; + char *name; + + if (self->linked) + lstate = "linked"; + + name = PyString_AsString(Lamp_getName(self)); + + PyOS_snprintf(buf, sizeof(buf), "[Lamp \"%s\" (%s)]", name, lstate); + + return PyString_FromString(buf); +} |