archive/blender-archive
Archived
1
1
Fork 0
Code Pull Requests Packages Activity
This repository has been archived on 2023-10-09. You can view files and clone it, but cannot push or open issues or pull requests.
Files
b43473e4257c6ce958c7811b7e44cb085a4f3b07
blender-archive/source/blender/python/api2_2x/Lamp.c
Michel Selten ec669df6ee 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.
2003-04-08 19:54:14 +00:00

1282 lines
39 KiB
C
Raw Blame History

/*
*
* ***** 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);
}
View Git Blame Copy Permalink