blender-archive/source/blender/python/api2_2x/MTex.c

/* 
 * $Id$
 *
 * ***** 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): Alex Mole
 *
 * ***** END GPL/BL DUAL LICENSE BLOCK *****
*/

#include <BKE_main.h>
#include <BKE_global.h>
#include <BKE_object.h>
#include <BKE_library.h>
#include <BLI_blenlib.h>
#include <BKE_texture.h>
#include <BKE_utildefines.h>

#include "MTex.h"
#include "Texture.h"
#include "constant.h"
#include "gen_utils.h"
#include "modules.h"


/*****************************************************************************/
/* Python BPy_MTex methods declarations:                                     */
/*****************************************************************************/
static PyObject *MTex_setTex(BPy_MTex *self, PyObject *args);

/*****************************************************************************/
/* Python method structure definition for Blender.Texture.MTex module:       */
/*****************************************************************************/
struct PyMethodDef M_MTex_methods[] = {
    {NULL, NULL, 0, NULL}
};

/*****************************************************************************/
/* Python BPy_MTex methods table:                                            */
/*****************************************************************************/
static PyMethodDef BPy_MTex_methods[] = {
 /* name, method, flags, doc */
  {"setTex", (PyCFunction)MTex_setTex, METH_VARARGS,
                            "(i) - Set MTex Texture"},
  {NULL, NULL,0, NULL}
};

/*****************************************************************************/
/* Python MTex_Type callback function prototypes:                            */
/*****************************************************************************/
static void MTex_dealloc (BPy_MTex *self);
static int MTex_setAttr (BPy_MTex *self, char *name, PyObject *v);
static int MTex_compare (BPy_MTex *a, BPy_MTex *b);
static PyObject *MTex_getAttr (BPy_MTex *self, char *name);
static PyObject *MTex_repr (BPy_MTex *self);


/*****************************************************************************/
/* Python MTex_Type structure definition:                                    */
/*****************************************************************************/
PyTypeObject MTex_Type =
{
    PyObject_HEAD_INIT(NULL)
    0,                              /* ob_size */
    "Blender MTex",                 /* tp_name */
    sizeof (BPy_MTex),              /* tp_basicsize */
    0,                              /* tp_itemsize */
    /* methods */
    (destructor)MTex_dealloc,       /* tp_dealloc */
    0,                              /* tp_print */
    (getattrfunc)MTex_getAttr,      /* tp_getattr */
    (setattrfunc)MTex_setAttr,      /* tp_setattr */
    (cmpfunc)MTex_compare,          /* tp_compare */
    (reprfunc)MTex_repr,            /* tp_repr */
    0,                              /* tp_as_number */
    0,                              /* tp_as_sequence */
    0,                              /* tp_as_mapping */
    0,                              /* tp_as_hash */
    0,0,0,0,0,0,
    0,                              /* tp_doc */ 
    0,0,0,0,0,0,
    0,                              /* tp_methods */
    0,                              /* tp_members */
};


PyObject *MTex_Init (void)
{
    PyObject *submodule;
    
    MTex_Type.ob_type = &PyType_Type;

    submodule = Py_InitModule("Blender.Texture.MTex", M_MTex_methods);
    
    return submodule;
}

PyObject *MTex_CreatePyObject (MTex *mtex)
{
    BPy_MTex *pymtex;

    pymtex = (BPy_MTex *) PyObject_NEW (BPy_MTex, &MTex_Type);
    if (!pymtex)
        return EXPP_ReturnPyObjError (PyExc_MemoryError,
                                  "couldn't create BPy_MTex PyObject");

    pymtex->mtex = mtex;
    return (PyObject *) pymtex;
}

MTex *MTex_FromPyObject (PyObject *pyobj)
{
    return ((BPy_MTex *)pyobj)->mtex;
}


int MTex_CheckPyObject (PyObject *pyobj)
{
    return (pyobj->ob_type == &MTex_Type);
}


/*****************************************************************************/
/* Python BPy_MTex methods:                                                  */
/*****************************************************************************/

static PyObject *MTex_setTex(BPy_MTex *self, PyObject *args)
{
    BPy_Texture *pytex = NULL;
    if (!PyArg_ParseTuple(args, "O!", &Texture_Type, &pytex))
        return EXPP_ReturnPyObjError (PyExc_TypeError,
                                           "expected Texture argument");

    if (self->mtex->tex)
        self->mtex->tex->id.us--;
    
    self->mtex->tex = Texture_FromPyObject((PyObject*)pytex);

    Py_INCREF(Py_None);
    return Py_None;
}

static void MTex_dealloc (BPy_MTex *self)
{
    PyObject_DEL (self);
}

static PyObject *MTex_getAttr (BPy_MTex *self, char *name)
{
    if (STREQ(name, "tex"))
    {
        if (self->mtex->tex)
            return Texture_CreatePyObject (self->mtex->tex);
        else
        {
            Py_INCREF (Py_None);
            return Py_None;
        }        
    }
    else if (STREQ(name, "texco"))
        return PyInt_FromLong(self->mtex->texco);
    else if (STREQ(name, "mapto"))
        return PyInt_FromLong(self->mtex->mapto);
    
    else if (STREQ(name, "__members__"))
        return Py_BuildValue("[s,s,s]", "tex", "texco", "mapto");

    /* not an attribute, search the methods table */
    return Py_FindMethod(BPy_MTex_methods, (PyObject *)self, name);
}

static int MTex_setAttr (BPy_MTex *self, char *name, PyObject *value)
{
    PyObject *valtuple; 
    PyObject *error = NULL;

    /* Put "value" in a tuple, because we want to pass it to functions  *
     * that only accept PyTuples.                                       */
    valtuple = Py_BuildValue("(O)", value);
    if (!valtuple)
        return EXPP_ReturnIntError(PyExc_MemoryError,
                                "MTex_setAttr: couldn't create PyTuple");

    if (STREQ(name, "tex"))
        error = MTex_setTex(self, valtuple);
    else if (STREQ(name, "texco"))
    {
        if (PyInt_Check(value))
        {
            int texco = PyInt_AsLong(value);
            /* TODO: sanity-check this input! */
            self->mtex->texco = texco;
            Py_INCREF (Py_None); /* because we decref it below */
            error = Py_None;
        }            
    }
    else if (STREQ(name, "mapto"))
    {
        if (PyInt_Check(value))
        {
            int mapto = PyInt_AsLong(value);
            /* TODO: sanity-check this input! */
            self->mtex->mapto = mapto;
            Py_INCREF (Py_None); /* because we decref it below */
            error = Py_None;
        }            
    }
    
    else { 
        /* Error */
        Py_DECREF(valtuple);
        return EXPP_ReturnIntError (PyExc_KeyError, "attribute not found");
    }

    Py_DECREF (valtuple);

    if (error != Py_None) return -1;

    /* Py_None was INCREF'd by the set*() function, so we need to DECREF it */
    Py_DECREF (Py_None);
    
    return 0;
}

static int MTex_compare (BPy_MTex *a, BPy_MTex *b)
{
    return (a->mtex == b->mtex) ? 0 : -1;
}

static PyObject *MTex_repr (BPy_MTex *self)
{
    return PyString_FromFormat("[MTex]");
}