Campbell Barton
3fe793f2b5
- Modified pack so it can repack too, rather the n raising an error. editobject - Dont need to recalc data when hiding and unhiding. rather do what layers do and re-sort the scene.
1394 lines
42 KiB
C
1394 lines
42 KiB
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): Willian P. Germano, Campbell Barton, Joilnen B. Leite,
|
|
* Austin Benesh
|
|
*
|
|
* ***** END GPL/BL DUAL LICENSE BLOCK *****
|
|
*/
|
|
#include "Image.h" /*This must come first */
|
|
|
|
#include "BDR_drawmesh.h" /* free_realtime_image */
|
|
#include "BKE_main.h"
|
|
#include "BKE_global.h"
|
|
#include "BKE_library.h"
|
|
#include "BKE_image.h"
|
|
#include "BKE_idprop.h"
|
|
#include "BIF_drawimage.h"
|
|
#include "BLI_blenlib.h"
|
|
#include "DNA_space_types.h" /* FILE_MAXDIR = 160 */
|
|
#include "IMB_imbuf_types.h" /* for the IB_rect define */
|
|
#include "BIF_gl.h"
|
|
#include "gen_utils.h"
|
|
#include "gen_library.h"
|
|
#include "BKE_packedFile.h"
|
|
#include "DNA_packedFile_types.h"
|
|
#include "BKE_icons.h"
|
|
#include "IMB_imbuf.h"
|
|
#include "IDProp.h"
|
|
|
|
/* used so we can get G.scene->r.cfra for getting the
|
|
current image frame, some images change frame if they are a sequence */
|
|
#include "DNA_scene_types.h"
|
|
|
|
/*****************************************************************************/
|
|
/* Python BPy_Image defaults: */
|
|
/*****************************************************************************/
|
|
#define EXPP_IMAGE_REP 1
|
|
#define EXPP_IMAGE_REP_MIN 1
|
|
#define EXPP_IMAGE_REP_MAX 16
|
|
|
|
|
|
enum img_consts {
|
|
EXPP_IMAGE_ATTR_XREP = 0,
|
|
EXPP_IMAGE_ATTR_YREP,
|
|
EXPP_IMAGE_ATTR_START,
|
|
EXPP_IMAGE_ATTR_END,
|
|
EXPP_IMAGE_ATTR_SPEED,
|
|
EXPP_IMAGE_ATTR_BINDCODE,
|
|
EXPP_IMAGE_ATTR_SOURCE,
|
|
};
|
|
|
|
/************************/
|
|
/*** The Image Module ***/
|
|
/************************/
|
|
|
|
/*****************************************************************************/
|
|
/* Python API function prototypes for the Image module. */
|
|
/*****************************************************************************/
|
|
static PyObject *M_Image_New( PyObject * self, PyObject * args );
|
|
static PyObject *M_Image_Get( PyObject * self, PyObject * args );
|
|
static PyObject *M_Image_GetCurrent( PyObject * self );
|
|
static PyObject *M_Image_Load( PyObject * self, PyObject * args );
|
|
|
|
|
|
/*****************************************************************************/
|
|
/* Python BPy_Image methods declarations: */
|
|
/*****************************************************************************/
|
|
static PyObject *Image_getFilename( BPy_Image * self );
|
|
static PyObject *Image_getSize( BPy_Image * self );
|
|
static PyObject *Image_getDepth( BPy_Image * self );
|
|
static PyObject *Image_getXRep( BPy_Image * self );
|
|
static PyObject *Image_getYRep( BPy_Image * self );
|
|
static PyObject *Image_getBindCode( BPy_Image * self );
|
|
static PyObject *Image_getStart( BPy_Image * self );
|
|
static PyObject *Image_getEnd( BPy_Image * self );
|
|
static PyObject *Image_getSpeed( BPy_Image * self );
|
|
static int Image_setFilename( BPy_Image * self, PyObject * args );
|
|
static PyObject *Image_oldsetFilename( BPy_Image * self, PyObject * args );
|
|
static PyObject *Image_setXRep( BPy_Image * self, PyObject * args );
|
|
static PyObject *Image_setYRep( BPy_Image * self, PyObject * args );
|
|
static PyObject *Image_setStart( BPy_Image * self, PyObject * args );
|
|
static PyObject *Image_setEnd( BPy_Image * self, PyObject * args );
|
|
static PyObject *Image_setSpeed( BPy_Image * self, PyObject * args );
|
|
static PyObject *Image_reload( BPy_Image * self );
|
|
static PyObject *Image_glLoad( BPy_Image * self );
|
|
static PyObject *Image_glFree( BPy_Image * self );
|
|
static PyObject *Image_getPixelF( BPy_Image * self, PyObject * args );
|
|
static PyObject *Image_getPixelI( BPy_Image * self, PyObject * args );
|
|
static PyObject *Image_setPixelF( BPy_Image * self, PyObject * args );
|
|
static PyObject *Image_setPixelI( BPy_Image * self, PyObject * args );
|
|
static PyObject *Image_getMaxXY( BPy_Image * self );
|
|
static PyObject *Image_getMinXY( BPy_Image * self );
|
|
static PyObject *Image_save( BPy_Image * self );
|
|
static PyObject *Image_unpack( BPy_Image * self, PyObject * args );
|
|
static PyObject *Image_pack( BPy_Image * self );
|
|
static PyObject *Image_makeCurrent( BPy_Image * self );
|
|
|
|
|
|
/*****************************************************************************/
|
|
/* Python BPy_Image methods table: */
|
|
/*****************************************************************************/
|
|
static PyMethodDef BPy_Image_methods[] = {
|
|
/* name, method, flags, doc */
|
|
{"getPixelF", ( PyCFunction ) Image_getPixelF, METH_VARARGS,
|
|
"(int, int) - Get pixel color as floats 0.0-1.0 returns [r,g,b,a]"},
|
|
{"getPixelI", ( PyCFunction ) Image_getPixelI, METH_VARARGS,
|
|
"(int, int) - Get pixel color as ints 0-255 returns [r,g,b,a]"},
|
|
{"setPixelF", ( PyCFunction ) Image_setPixelF, METH_VARARGS,
|
|
"(int, int, [f r,f g,f b,f a]) - Set pixel color using floats 0.0-1.0"},
|
|
{"setPixelI", ( PyCFunction ) Image_setPixelI, METH_VARARGS,
|
|
"(int, int, [i r, i g, i b, i a]) - Set pixel color using ints 0-255"},
|
|
{"getMaxXY", ( PyCFunction ) Image_getMaxXY, METH_NOARGS,
|
|
"() - Get maximum x & y coordinates of current image as [x, y]"},
|
|
{"getMinXY", ( PyCFunction ) Image_getMinXY, METH_NOARGS,
|
|
"() - Get minimun x & y coordinates of image as [x, y]"},
|
|
{"getName", ( PyCFunction ) GenericLib_getName, METH_NOARGS,
|
|
"() - Return Image object name"},
|
|
{"getFilename", ( PyCFunction ) Image_getFilename, METH_NOARGS,
|
|
"() - Return Image object filename"},
|
|
{"getSize", ( PyCFunction ) Image_getSize, METH_NOARGS,
|
|
"() - Return Image object [width, height] dimension in pixels"},
|
|
{"getDepth", ( PyCFunction ) Image_getDepth, METH_NOARGS,
|
|
"() - Return Image object pixel depth"},
|
|
{"getXRep", ( PyCFunction ) Image_getXRep, METH_NOARGS,
|
|
"() - Return Image object x repetition value"},
|
|
{"getYRep", ( PyCFunction ) Image_getYRep, METH_NOARGS,
|
|
"() - Return Image object y repetition value"},
|
|
{"getStart", ( PyCFunction ) Image_getStart, METH_NOARGS,
|
|
"() - Return Image object start frame."},
|
|
{"getEnd", ( PyCFunction ) Image_getEnd, METH_NOARGS,
|
|
"() - Return Image object end frame."},
|
|
{"getSpeed", ( PyCFunction ) Image_getSpeed, METH_NOARGS,
|
|
"() - Return Image object speed (fps)."},
|
|
{"getBindCode", ( PyCFunction ) Image_getBindCode, METH_NOARGS,
|
|
"() - Return Image object's bind code value"},
|
|
{"reload", ( PyCFunction ) Image_reload, METH_NOARGS,
|
|
"() - Reload the image from the filesystem"},
|
|
{"glLoad", ( PyCFunction ) Image_glLoad, METH_NOARGS,
|
|
"() - Load the image data in OpenGL texture memory.\n\
|
|
The bindcode (int) is returned."},
|
|
{"glFree", ( PyCFunction ) Image_glFree, METH_NOARGS,
|
|
"() - Free the image data from OpenGL texture memory only,\n\
|
|
see also image.glLoad()."},
|
|
{"setName", ( PyCFunction ) GenericLib_setName_with_method, METH_VARARGS,
|
|
"(str) - Change Image object name"},
|
|
{"setFilename", ( PyCFunction ) Image_oldsetFilename, METH_VARARGS,
|
|
"(str) - Change Image file name"},
|
|
{"setXRep", ( PyCFunction ) Image_setXRep, METH_VARARGS,
|
|
"(int) - Change Image object x repetition value"},
|
|
{"setYRep", ( PyCFunction ) Image_setYRep, METH_VARARGS,
|
|
"(int) - Change Image object y repetition value"},
|
|
{"setStart", ( PyCFunction ) Image_setStart, METH_VARARGS,
|
|
"(int) - Change Image object animation start value"},
|
|
{"setEnd", ( PyCFunction ) Image_setEnd, METH_VARARGS,
|
|
"(int) - Change Image object animation end value"},
|
|
{"setSpeed", ( PyCFunction ) Image_setSpeed, METH_VARARGS,
|
|
"(int) - Change Image object animation speed (fps)"},
|
|
{"save", ( PyCFunction ) Image_save, METH_NOARGS,
|
|
"() - Write image buffer to file"},
|
|
{"unpack", ( PyCFunction ) Image_unpack, METH_VARARGS,
|
|
"(int) - Unpack image. Uses the values defined in Blender.UnpackModes."},
|
|
{"pack", ( PyCFunction ) Image_pack, METH_NOARGS,
|
|
"() - Pack the image"},
|
|
{"makeCurrent", ( PyCFunction ) Image_makeCurrent, METH_NOARGS,
|
|
"() - Make this the currently displayed image"},
|
|
{NULL, NULL, 0, NULL}
|
|
};
|
|
|
|
|
|
/*****************************************************************************/
|
|
/* The following string definitions are used for documentation strings. */
|
|
/* In Python these will be written to the console when doing a */
|
|
/* Blender.Image.__doc__ */
|
|
/*****************************************************************************/
|
|
static char M_Image_doc[] = "The Blender Image module\n\n";
|
|
|
|
static char M_Image_New_doc[] =
|
|
"() - return a new Image object";
|
|
|
|
static char M_Image_Get_doc[] =
|
|
"(name) - return the image with the name 'name', \
|
|
returns None if not found.\n If 'name' is not specified, \
|
|
it returns a list of all images in the\ncurrent scene.";
|
|
|
|
static char M_Image_GetCurrent_doc[] =
|
|
"() - return the current image, from last active the uv/image view, \
|
|
returns None no image is in the view.\n";
|
|
|
|
static char M_Image_Load_doc[] =
|
|
"(filename) - return image from file filename as Image Object, \
|
|
returns None if not found.\n";
|
|
|
|
/*****************************************************************************/
|
|
/* Python method structure definition for Blender.Image module: */
|
|
/*****************************************************************************/
|
|
struct PyMethodDef M_Image_methods[] = {
|
|
{"New", M_Image_New, METH_VARARGS, M_Image_New_doc},
|
|
{"Get", M_Image_Get, METH_VARARGS, M_Image_Get_doc},
|
|
{"GetCurrent", ( PyCFunction ) M_Image_GetCurrent, METH_NOARGS, M_Image_GetCurrent_doc},
|
|
{"get", M_Image_Get, METH_VARARGS, M_Image_Get_doc},
|
|
{"Load", M_Image_Load, METH_VARARGS, M_Image_Load_doc},
|
|
{NULL, NULL, 0, NULL}
|
|
};
|
|
|
|
|
|
/*****************************************************************************/
|
|
/* Function: M_Image_New */
|
|
/* Python equivalent: Blender.Image.New */
|
|
/*****************************************************************************/
|
|
static PyObject *M_Image_New( PyObject * self, PyObject * args)
|
|
{
|
|
int width, height, depth;
|
|
char *name;
|
|
Image *image;
|
|
if( !PyArg_ParseTuple( args, "siii", &name, &width, &height, &depth ) )
|
|
return ( EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected 1 string and 3 ints" ) );
|
|
if (width > 5000 || height > 5000 || width < 1 || height < 1)
|
|
return ( EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"Image width and height must be between 1 and 5000" ) );
|
|
image = BKE_add_image_size(width, height, name, 0);
|
|
if( !image )
|
|
return ( EXPP_ReturnPyObjError( PyExc_MemoryError,
|
|
"couldn't create PyObject Image_Type" ) );
|
|
|
|
/* reset usage count, since BKE_add_image_size() incremented it */
|
|
/* image->id.us--; */
|
|
/* Strange, new images have a user count of one???, otherwise it messes up */
|
|
|
|
return Image_CreatePyObject( image );
|
|
}
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
/* Function: M_Image_Get */
|
|
/* Python equivalent: Blender.Image.Get */
|
|
/* Description: Receives a string and returns the image object */
|
|
/* whose name matches the string. If no argument is */
|
|
/* passed in, a list of all image names in the */
|
|
/* current scene is returned. */
|
|
/*****************************************************************************/
|
|
static PyObject *M_Image_Get( PyObject * self, PyObject * args )
|
|
{
|
|
char *name = NULL;
|
|
Image *img_iter;
|
|
|
|
if( !PyArg_ParseTuple( args, "|s", &name ) )
|
|
return ( EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected string argument (or nothing)" ) );
|
|
|
|
img_iter = G.main->image.first;
|
|
|
|
if( name ) { /* (name) - Search image by name */
|
|
|
|
BPy_Image *wanted_image = NULL;
|
|
|
|
while( ( img_iter ) && ( wanted_image == NULL ) ) {
|
|
if( strcmp( name, img_iter->id.name + 2 ) == 0 ) {
|
|
wanted_image = ( BPy_Image * )
|
|
PyObject_NEW( BPy_Image, &Image_Type );
|
|
if( wanted_image )
|
|
wanted_image->image = img_iter;
|
|
}
|
|
img_iter = img_iter->id.next;
|
|
}
|
|
|
|
if( wanted_image == NULL ) { /* Requested image doesn't exist */
|
|
char error_msg[64];
|
|
PyOS_snprintf( error_msg, sizeof( error_msg ),
|
|
"Image \"%s\" not found", name );
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_NameError, error_msg ) );
|
|
}
|
|
|
|
return ( PyObject * ) wanted_image;
|
|
}
|
|
|
|
else { /* () - return a list of all images in the scene */
|
|
int index = 0;
|
|
PyObject *img_list, *pyobj;
|
|
|
|
img_list = PyList_New( BLI_countlist( &( G.main->image ) ) );
|
|
|
|
if( img_list == NULL )
|
|
return ( EXPP_ReturnPyObjError( PyExc_MemoryError,
|
|
"couldn't create PyList" ) );
|
|
|
|
while( img_iter ) {
|
|
pyobj = Image_CreatePyObject( img_iter );
|
|
|
|
PyList_SET_ITEM( img_list, index, pyobj );
|
|
|
|
img_iter = img_iter->id.next;
|
|
index++;
|
|
}
|
|
|
|
return ( img_list );
|
|
}
|
|
}
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
/* Function: M_Image_GetCurrent*/
|
|
/* Python equivalent: Blender.Image.GetCurrent */
|
|
/* Description: Returns the active current (G.sima) */
|
|
/* This will be the image last under the mouse cursor */
|
|
/* None if there is no Image. */
|
|
/*****************************************************************************/
|
|
static PyObject *M_Image_GetCurrent( PyObject * self )
|
|
{
|
|
if (!G.sima || !G.sima->image) {
|
|
Py_RETURN_NONE;
|
|
}
|
|
what_image( G.sima ); /* make sure image data exists */
|
|
return Image_CreatePyObject( G.sima->image );
|
|
}
|
|
|
|
|
|
/*****************************************************************************/
|
|
/* Function: M_Image_Load */
|
|
/* Python equivalent: Blender.Image.Load */
|
|
/* Description: Receives a string and returns the image object */
|
|
/* whose filename matches the string. */
|
|
/*****************************************************************************/
|
|
static PyObject *M_Image_Load( PyObject * self, PyObject * args )
|
|
{
|
|
char *fname;
|
|
Image *img_ptr;
|
|
BPy_Image *image;
|
|
|
|
if( !PyArg_ParseTuple( args, "s", &fname ) )
|
|
return ( EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected string argument" ) );
|
|
|
|
image = ( BPy_Image * ) PyObject_NEW( BPy_Image, &Image_Type );
|
|
|
|
if( !image )
|
|
return ( EXPP_ReturnPyObjError( PyExc_MemoryError,
|
|
"couldn't create PyObject Image_Type" ) );
|
|
|
|
img_ptr = BKE_add_image_file( fname );
|
|
if( !img_ptr )
|
|
return ( EXPP_ReturnPyObjError( PyExc_IOError,
|
|
"couldn't load image" ) );
|
|
|
|
/* force a load the image buffers*/
|
|
BKE_image_get_ibuf(img_ptr, NULL);
|
|
|
|
image->image = img_ptr;
|
|
|
|
return ( PyObject * ) image;
|
|
}
|
|
|
|
|
|
/**
|
|
* getPixelF( x, y )
|
|
* returns float list of pixel colors in rgba order.
|
|
* returned values are floats normalized to 0.0 - 1.0.
|
|
* blender images are all 4x8 bit at the moment apr-2005
|
|
*/
|
|
|
|
static PyObject *Image_getPixelF( BPy_Image * self, PyObject * args )
|
|
{
|
|
|
|
PyObject *attr = PyList_New(4);
|
|
ImBuf *ibuf= BKE_image_get_ibuf(self->image, NULL);
|
|
char *pixel; /* image data */
|
|
int index; /* offset into image data */
|
|
int x = 0;
|
|
int y = 0;
|
|
int pixel_size = 4; /* each pixel is 4 x 8-bits packed in unsigned int */
|
|
int i;
|
|
|
|
if (!attr)
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't allocate memory for color list" );
|
|
|
|
if( !PyArg_ParseTuple( args, "ii", &x, &y ) )
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected 2 integers" );
|
|
|
|
if( !ibuf || !ibuf->rect ) /* loading didn't work */
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't load image data in Blender" );
|
|
|
|
if( ibuf->type == 1 ) /* bitplane image */
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"unsupported bitplane image format" );
|
|
|
|
if( x > ( ibuf->x - 1 )
|
|
|| y > ( ibuf->y - 1 )
|
|
|| x < ibuf->xorig || y < ibuf->yorig )
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"x or y is out of range" );
|
|
|
|
/*
|
|
assumption: from looking at source, skipx is often not set,
|
|
so we calc ourselves
|
|
*/
|
|
|
|
index = ( x + y * ibuf->x ) * pixel_size;
|
|
|
|
pixel = ( char * ) ibuf->rect;
|
|
for (i=0; i<4; i++) {
|
|
PyList_SetItem( attr, i, PyFloat_FromDouble( ( ( double ) pixel[index+i] ) / 255.0 ));
|
|
}
|
|
return attr;
|
|
}
|
|
|
|
|
|
/**
|
|
* getPixelI( x, y )
|
|
* returns integer list of pixel colors in rgba order.
|
|
* returned values are ints normalized to 0-255.
|
|
* blender images are all 4x8 bit at the moment apr-2005
|
|
*/
|
|
|
|
static PyObject *Image_getPixelI( BPy_Image * self, PyObject * args )
|
|
{
|
|
PyObject *attr = PyList_New(4);
|
|
ImBuf *ibuf= BKE_image_get_ibuf(self->image, NULL);
|
|
char *pixel; /* image data */
|
|
int index; /* offset into image data */
|
|
int x = 0;
|
|
int y = 0;
|
|
int pixel_size = 4; /* each pixel is 4 x 8-bits packed in unsigned int */
|
|
int i;
|
|
|
|
if (!attr)
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't allocate memory for color list" );
|
|
|
|
if( !PyArg_ParseTuple( args, "ii", &x, &y ) )
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected 2 integers" );
|
|
|
|
if( !ibuf || !ibuf->rect ) /* didn't work */
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't load image data in Blender" );
|
|
|
|
if( ibuf->type == 1 ) /* bitplane image */
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"unsupported bitplane image format" );
|
|
|
|
if( x > ( ibuf->x - 1 )
|
|
|| y > ( ibuf->y - 1 )
|
|
|| x < ibuf->xorig || y < ibuf->yorig )
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"x or y is out of range" );
|
|
|
|
/*
|
|
assumption: from looking at source, skipx is often not set,
|
|
so we calc ourselves
|
|
*/
|
|
|
|
index = ( x + y * ibuf->x ) * pixel_size;
|
|
pixel = ( char * ) ibuf->rect;
|
|
|
|
for (i=0; i<4; i++) {
|
|
PyList_SetItem( attr, i, PyInt_FromLong( pixel[index+i] ));
|
|
}
|
|
return attr;
|
|
}
|
|
|
|
|
|
/* set pixel as floats */
|
|
|
|
static PyObject *Image_setPixelF( BPy_Image * self, PyObject * args )
|
|
{
|
|
ImBuf *ibuf= BKE_image_get_ibuf(self->image, NULL);
|
|
char *pixel; /* image data */
|
|
int index; /* offset into image data */
|
|
int x = 0;
|
|
int y = 0;
|
|
int a = 0;
|
|
int pixel_size = 4; /* each pixel is 4 x 8-bits packed in unsigned int */
|
|
float p[4];
|
|
|
|
if( !PyArg_ParseTuple
|
|
( args, "ii(ffff)", &x, &y, &p[0], &p[1], &p[2], &p[3] ) )
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected 2 integers and an array of 4 floats" );
|
|
|
|
if( !ibuf || !ibuf->rect ) /* didn't work */
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't load image data in Blender" );
|
|
|
|
if( ibuf->type == 1 ) /* bitplane image */
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"unsupported bitplane image format" );
|
|
|
|
if( x > ( ibuf->x - 1 )
|
|
|| y > ( ibuf->y - 1 )
|
|
|| x < ibuf->xorig || y < ibuf->yorig )
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"x or y is out of ruange" );
|
|
|
|
for( a = 0; a < 4; a++ ) {
|
|
if( p[a] > 1.0 || p[a] < 0.0 )
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"r, g, b, or a is out of range" );
|
|
}
|
|
|
|
|
|
/*
|
|
assumption: from looking at source, skipx is often not set,
|
|
so we calc ourselves
|
|
*/
|
|
|
|
index = ( x + y * ibuf->x ) * pixel_size;
|
|
|
|
pixel = ( char * ) ibuf->rect;
|
|
|
|
pixel[index] = ( char ) ( p[0] * 255.0 );
|
|
pixel[index + 1] = ( char ) ( p[1] * 255.0 );
|
|
pixel[index + 2] = ( char ) ( p[2] * 255.0 );
|
|
pixel[index + 3] = ( char ) ( p[3] * 255.0 );
|
|
|
|
ibuf->userflags |= IB_BITMAPDIRTY;
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
|
|
/* set pixel as ints */
|
|
|
|
static PyObject *Image_setPixelI( BPy_Image * self, PyObject * args )
|
|
{
|
|
ImBuf *ibuf= BKE_image_get_ibuf(self->image, NULL);
|
|
char *pixel; /* image data */
|
|
int index; /* offset into image data */
|
|
int x = 0;
|
|
int y = 0;
|
|
int a = 0;
|
|
int pixel_size = 4; /* each pixel is 4 x 8-bits packed in unsigned int */
|
|
int p[4];
|
|
|
|
if( !PyArg_ParseTuple
|
|
( args, "ii(iiii)", &x, &y, &p[0], &p[1], &p[2], &p[3] ) )
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected 2 integers and an list of 4 ints" );
|
|
|
|
if( !ibuf || !ibuf->rect ) /* didn't work */
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't load image data in Blender" );
|
|
|
|
if( ibuf->type == 1 ) /* bitplane image */
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"unsupported bitplane image format" );
|
|
|
|
if( x > ( ibuf->x - 1 )
|
|
|| y > ( ibuf->y - 1 )
|
|
|| x < ibuf->xorig || y < ibuf->yorig )
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"x or y is out of range" );
|
|
|
|
for( a = 0; a < 4; a++ ) {
|
|
if( p[a] > 255 || p[a] < 0 )
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"r, g, b, or a is out of range" );
|
|
}
|
|
|
|
/*
|
|
assumption: from looking at source, skipx is often not set,
|
|
so we calc ourselves
|
|
*/
|
|
|
|
index = ( x + y * ibuf->x ) * pixel_size;
|
|
|
|
pixel = ( char * ) ibuf->rect;
|
|
|
|
pixel[index] = ( char ) p[0];
|
|
pixel[index + 1] = ( char ) p[1];
|
|
pixel[index + 2] = ( char ) p[2];
|
|
pixel[index + 3] = ( char ) p[3];
|
|
|
|
ibuf->userflags |= IB_BITMAPDIRTY;
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
|
|
/* get max extent of image */
|
|
|
|
static PyObject *Image_getMaxXY( BPy_Image * self )
|
|
{
|
|
ImBuf *ibuf= BKE_image_get_ibuf(self->image, NULL);
|
|
PyObject *attr;
|
|
|
|
if( !ibuf || !ibuf->rect ) /* didn't work */
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't load image data in Blender" );
|
|
|
|
attr = Py_BuildValue( "[i,i]", ibuf->x, ibuf->y );
|
|
|
|
if( attr )
|
|
return attr;
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"could not determine max x or y" );
|
|
}
|
|
|
|
|
|
/* get min extent of image */
|
|
|
|
static PyObject *Image_getMinXY( BPy_Image * self )
|
|
{
|
|
ImBuf *ibuf= BKE_image_get_ibuf(self->image, NULL);
|
|
PyObject *attr;
|
|
|
|
if( !ibuf || !ibuf->rect ) /* didn't work */
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't load image data in Blender" );
|
|
|
|
attr = Py_BuildValue( "[i,i]", ibuf->xorig,
|
|
ibuf->yorig );
|
|
|
|
if( attr )
|
|
return attr;
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"could not determine min x or y" );
|
|
}
|
|
|
|
/* unpack image */
|
|
|
|
static PyObject *Image_unpack( BPy_Image * self, PyObject * args )
|
|
{
|
|
Image *image = self->image;
|
|
int mode;
|
|
|
|
/*get the absolute path */
|
|
if( !PyArg_ParseTuple( args, "i", &mode ) )
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected 1 integer" );
|
|
|
|
if (image->packedfile==NULL)
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"image not packed" );
|
|
|
|
if (unpackImage(image, mode) == RET_ERROR)
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"error unpacking image" );
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
/* pack image */
|
|
|
|
static PyObject *Image_pack( BPy_Image * self )
|
|
{
|
|
Image *image = self->image;
|
|
ImBuf *ibuf = BKE_image_get_ibuf(image, NULL);
|
|
|
|
if( !ibuf || !ibuf->rect ) /* didn't work */
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't load image data in Blender" );
|
|
|
|
if (image->packedfile ) { /* RePack? */
|
|
if (ibuf->userflags & IB_BITMAPDIRTY)
|
|
BKE_image_memorypack(image);
|
|
} else { /* Pack for the first time */
|
|
if (ibuf && (ibuf->userflags & IB_BITMAPDIRTY))
|
|
BKE_image_memorypack(image);
|
|
else
|
|
image->packedfile = newPackedFile(image->name);
|
|
}
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
|
|
static PyObject *Image_makeCurrent( BPy_Image * self )
|
|
{
|
|
#if 0 /* add back in when bpy becomes "official" */
|
|
static char warning = 1;
|
|
if( warning ) {
|
|
printf("image.makeCurrent() deprecated!\n\t use 'bpy.images.active = image instead'\n");
|
|
--warning;
|
|
}
|
|
#endif
|
|
|
|
if (!G.sima)
|
|
Py_RETURN_FALSE;
|
|
|
|
G.sima->image= self->image;
|
|
Py_RETURN_TRUE;
|
|
}
|
|
|
|
|
|
/* save image to file */
|
|
|
|
static PyObject *Image_save( BPy_Image * self )
|
|
{
|
|
ImBuf *ibuf= BKE_image_get_ibuf(self->image, NULL);
|
|
|
|
if(!ibuf || !IMB_saveiff( ibuf, self->image->name, ibuf->flags ) )
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"could not save image" );
|
|
|
|
Py_RETURN_NONE; /* normal return, image saved */
|
|
}
|
|
|
|
static PyObject *M_Image_SourceDict( void )
|
|
{
|
|
PyObject *Dict = PyConstant_New( );
|
|
if( Dict ) {
|
|
BPy_constant *d = ( BPy_constant * ) Dict;
|
|
PyConstant_Insert(d, "STILL", PyInt_FromLong(IMA_SRC_FILE));
|
|
PyConstant_Insert(d, "MOVIE", PyInt_FromLong(IMA_SRC_MOVIE));
|
|
PyConstant_Insert(d, "SEQUENCE", PyInt_FromLong(IMA_SRC_SEQUENCE));
|
|
PyConstant_Insert(d, "GENERATED", PyInt_FromLong(IMA_SRC_GENERATED));
|
|
}
|
|
return Dict;
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: Image_Init */
|
|
/*****************************************************************************/
|
|
PyObject *Image_Init( void )
|
|
{
|
|
PyObject *submodule;
|
|
PyObject *Sources = M_Image_SourceDict( );
|
|
|
|
if( PyType_Ready( &Image_Type ) < 0 )
|
|
return NULL;
|
|
|
|
submodule =
|
|
Py_InitModule3( "Blender.Image", M_Image_methods,
|
|
M_Image_doc );
|
|
|
|
if( Sources )
|
|
PyModule_AddObject( submodule, "Sources", Sources );
|
|
|
|
return submodule;
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Python Image_Type callback function prototypes: */
|
|
/*****************************************************************************/
|
|
static int Image_compare( BPy_Image * a, BPy_Image * b );
|
|
static PyObject *Image_repr( BPy_Image * self );
|
|
|
|
/*****************************************************************************/
|
|
/* Function: Image_CreatePyObject */
|
|
/* Description: This function will create a new BPy_Image from an existing */
|
|
/* Blender image structure. */
|
|
/*****************************************************************************/
|
|
PyObject *Image_CreatePyObject( Image * image )
|
|
{
|
|
BPy_Image *py_img;
|
|
py_img = ( BPy_Image * ) PyObject_NEW( BPy_Image, &Image_Type );
|
|
|
|
if( !py_img )
|
|
return EXPP_ReturnPyObjError( PyExc_MemoryError,
|
|
"couldn't create BPy_Image object" );
|
|
|
|
py_img->image = image;
|
|
return ( PyObject * ) py_img;
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: Image_FromPyObject */
|
|
/* Description: Returns the Blender Image associated with this object */
|
|
/*****************************************************************************/
|
|
Image *Image_FromPyObject( PyObject * pyobj )
|
|
{
|
|
return ( ( BPy_Image * ) pyobj )->image;
|
|
}
|
|
|
|
static PyObject *Image_getFilename( BPy_Image * self )
|
|
{
|
|
PyObject *attr = PyString_FromString( self->image->name );
|
|
|
|
if( attr )
|
|
return attr;
|
|
|
|
return ( EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't get Image.filename attribute" ) );
|
|
}
|
|
|
|
static PyObject *Image_getSize( BPy_Image * self )
|
|
{
|
|
ImBuf *ibuf= BKE_image_get_ibuf(self->image, NULL);
|
|
PyObject *attr = PyList_New(2);
|
|
|
|
if( !ibuf ) /* didn't work */
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't load image data in Blender" );
|
|
if( !attr )
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't get Image.size attribute" );
|
|
|
|
PyList_SetItem( attr, 0, PyInt_FromLong(ibuf->x));
|
|
PyList_SetItem( attr, 1, PyInt_FromLong(ibuf->y));
|
|
return attr;
|
|
}
|
|
|
|
static PyObject *Image_getDepth( BPy_Image * self )
|
|
{
|
|
ImBuf *ibuf= BKE_image_get_ibuf(self->image, NULL);
|
|
PyObject *attr;
|
|
|
|
if( !ibuf ) /* didn't work */
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't load image data in Blender" );
|
|
|
|
attr = Py_BuildValue( "h", ibuf->depth );
|
|
|
|
if( attr )
|
|
return attr;
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't get Image.depth attribute" );
|
|
}
|
|
|
|
|
|
static PyObject *Image_getXRep( BPy_Image * self )
|
|
{
|
|
PyObject *attr = PyInt_FromLong( self->image->xrep );
|
|
|
|
if( attr )
|
|
return attr;
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't get Image.xrep attribute" );
|
|
}
|
|
|
|
static PyObject *Image_getYRep( BPy_Image * self )
|
|
{
|
|
PyObject *attr = PyInt_FromLong( self->image->yrep );
|
|
|
|
if( attr )
|
|
return attr;
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't get Image.yrep attribute" );
|
|
}
|
|
|
|
static PyObject *Image_getStart( BPy_Image * self )
|
|
{
|
|
PyObject *attr = PyInt_FromLong( self->image->twsta );
|
|
|
|
if( attr )
|
|
return attr;
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't get Image.start attribute" );
|
|
}
|
|
|
|
static PyObject *Image_getEnd( BPy_Image * self )
|
|
{
|
|
PyObject *attr = PyInt_FromLong( self->image->twend );
|
|
|
|
if( attr )
|
|
return attr;
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't get Image.end attribute" );
|
|
}
|
|
|
|
static PyObject *Image_getSpeed( BPy_Image * self )
|
|
{
|
|
PyObject *attr = PyInt_FromLong( self->image->animspeed );
|
|
|
|
if( attr )
|
|
return attr;
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't get Image.speed attribute" );
|
|
}
|
|
|
|
static PyObject *Image_getBindCode( BPy_Image * self )
|
|
{
|
|
PyObject *attr = PyLong_FromUnsignedLong( self->image->bindcode );
|
|
|
|
if( attr )
|
|
return attr;
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't get Image.bindcode attribute" );
|
|
}
|
|
|
|
static PyObject *Image_reload( BPy_Image * self )
|
|
{
|
|
Image *image = self->image;
|
|
|
|
BKE_image_signal(image, NULL, IMA_SIGNAL_RELOAD);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *Image_glFree( BPy_Image * self )
|
|
{
|
|
Image *image = self->image;
|
|
|
|
free_realtime_image( image );
|
|
/* remove the nocollect flag, image is available for garbage collection again */
|
|
image->flag &= ~IMA_NOCOLLECT;
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *Image_glLoad( BPy_Image * self )
|
|
{
|
|
Image *image = self->image;
|
|
unsigned int *bind = &image->bindcode;
|
|
|
|
if( *bind == 0 ) {
|
|
ImBuf *ibuf= BKE_image_get_ibuf(self->image, NULL);
|
|
|
|
if( !ibuf ) /* didn't work */
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't load image data in Blender" );
|
|
|
|
glGenTextures( 1, ( GLuint * ) bind );
|
|
glBindTexture( GL_TEXTURE_2D, *bind );
|
|
|
|
gluBuild2DMipmaps( GL_TEXTURE_2D, GL_RGBA, ibuf->x,
|
|
ibuf->y, GL_RGBA, GL_UNSIGNED_BYTE,
|
|
ibuf->rect );
|
|
glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER,
|
|
GL_LINEAR_MIPMAP_NEAREST );
|
|
glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER,
|
|
GL_LINEAR );
|
|
glTexEnvi( GL_TEXTURE_ENV, GL_TEXTURE_ENV_MODE, GL_MODULATE );
|
|
|
|
glTexImage2D( GL_TEXTURE_2D, 0, GL_RGBA, ibuf->x,
|
|
ibuf->y, 0, GL_RGBA, GL_UNSIGNED_BYTE,
|
|
ibuf->rect );
|
|
|
|
/* raise the nocollect flag,
|
|
image is not available for garbage collection
|
|
(python GL might use it directly)
|
|
*/
|
|
image->flag |= IMA_NOCOLLECT;
|
|
}
|
|
|
|
return PyLong_FromUnsignedLong( image->bindcode );
|
|
}
|
|
|
|
static int Image_setFilename( BPy_Image * self, PyObject * value )
|
|
{
|
|
char *name;
|
|
|
|
name = PyString_AsString(value);
|
|
|
|
if( !name )
|
|
return ( EXPP_ReturnIntError( PyExc_TypeError,
|
|
"expected a string argument" ) );
|
|
|
|
/* max len is FILE_MAXDIR == 160, FILE_MAXFILE == 80 chars like done in DNA_image_types.h */
|
|
if( strlen(name) >= FILE_MAXDIR + FILE_MAXFILE )
|
|
return ( EXPP_ReturnIntError( PyExc_TypeError,
|
|
"string argument is limited to 240 chars at most" ) );
|
|
|
|
strcpy( self->image->name, name );
|
|
return 0;
|
|
}
|
|
|
|
static PyObject *Image_oldsetFilename( BPy_Image * self, PyObject * args )
|
|
{
|
|
return EXPP_setterWrapper( (void *)self, args, (setter)Image_setFilename );
|
|
}
|
|
|
|
static PyObject *Image_setXRep( BPy_Image * self, PyObject * args )
|
|
{
|
|
short value;
|
|
|
|
if( !PyArg_ParseTuple( args, "h", &value ) )
|
|
return ( EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected int argument in [1,16]" ) );
|
|
|
|
if( value >= EXPP_IMAGE_REP_MIN && value <= EXPP_IMAGE_REP_MAX )
|
|
self->image->xrep = value;
|
|
else
|
|
return ( EXPP_ReturnPyObjError( PyExc_ValueError,
|
|
"expected int argument in [1,16]" ) );
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *Image_setYRep( BPy_Image * self, PyObject * args )
|
|
{
|
|
short value;
|
|
|
|
if( !PyArg_ParseTuple( args, "h", &value ) )
|
|
return ( EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected int argument in [1,16]" ) );
|
|
|
|
if( value >= EXPP_IMAGE_REP_MIN && value <= EXPP_IMAGE_REP_MAX )
|
|
self->image->yrep = value;
|
|
else
|
|
return ( EXPP_ReturnPyObjError( PyExc_ValueError,
|
|
"expected int argument in [1,16]" ) );
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
|
|
static PyObject *Image_setStart( BPy_Image * self, PyObject * args )
|
|
{
|
|
short value;
|
|
|
|
if( !PyArg_ParseTuple( args, "h", &value ) )
|
|
return ( EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected int argument in [0,128]" ) );
|
|
|
|
if( value >= 0 && value <= 128 )
|
|
self->image->twsta = value;
|
|
else
|
|
return ( EXPP_ReturnPyObjError( PyExc_ValueError,
|
|
"expected int argument in [0,128]" ) );
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
|
|
static PyObject *Image_setEnd( BPy_Image * self, PyObject * args )
|
|
{
|
|
short value;
|
|
|
|
if( !PyArg_ParseTuple( args, "h", &value ) )
|
|
return ( EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected int argument in [0,128]" ) );
|
|
|
|
if( value >= 0 && value <= 128 )
|
|
self->image->twend = value;
|
|
else
|
|
return ( EXPP_ReturnPyObjError( PyExc_ValueError,
|
|
"expected int argument in [0,128]" ) );
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *Image_setSpeed( BPy_Image * self, PyObject * args )
|
|
{
|
|
short value;
|
|
|
|
if( !PyArg_ParseTuple( args, "h", &value ) )
|
|
return ( EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected int argument in [0,128]" ) );
|
|
|
|
if( value >= 1 && value <= 100 )
|
|
self->image->animspeed = value;
|
|
else
|
|
return ( EXPP_ReturnPyObjError( PyExc_ValueError,
|
|
"expected int argument in [0,128]" ) );
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
|
|
/*****************************************************************************/
|
|
/* Function: Image_compare */
|
|
/* Description: This is a callback function for the BPy_Image type. It */
|
|
/* compares two Image_Type objects. Only the "==" and "!=" */
|
|
/* comparisons are meaninful. Returns 0 for equality and -1 if */
|
|
/* they don't point to the same Blender Image struct. */
|
|
/* In Python it becomes 1 if they are equal, 0 otherwise. */
|
|
/*****************************************************************************/
|
|
static int Image_compare( BPy_Image * a, BPy_Image * b )
|
|
{
|
|
return ( a->image == b->image ) ? 0 : -1;
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: Image_repr */
|
|
/* Description: This is a callback function for the BPy_Image type. It */
|
|
/* builds a meaninful string to represent image objects. */
|
|
/*****************************************************************************/
|
|
static PyObject *Image_repr( BPy_Image * self )
|
|
{
|
|
return PyString_FromFormat( "[Image \"%s\"]",
|
|
self->image->id.name + 2 );
|
|
}
|
|
|
|
static PyObject *Image_getPacked(BPy_Image *self, void *closure)
|
|
{
|
|
if (self->image->packedfile)
|
|
Py_RETURN_TRUE;
|
|
else
|
|
Py_RETURN_FALSE;
|
|
}
|
|
|
|
static PyObject *Image_hasData(BPy_Image *self, void *closure)
|
|
{
|
|
if (self->image->ibufs.first)
|
|
Py_RETURN_TRUE;
|
|
else
|
|
Py_RETURN_FALSE;
|
|
}
|
|
|
|
static PyObject *Image_getFlag(BPy_Image *self, void *flag)
|
|
{
|
|
if (self->image->flag & (int)flag)
|
|
Py_RETURN_TRUE;
|
|
else
|
|
Py_RETURN_FALSE;
|
|
|
|
}
|
|
|
|
static PyObject *Image_getFlagTpage(BPy_Image *self, void *flag)
|
|
{
|
|
if (self->image->tpageflag & (int)flag)
|
|
Py_RETURN_TRUE;
|
|
else
|
|
Py_RETURN_FALSE;
|
|
|
|
}
|
|
|
|
static int Image_setSource( BPy_Image *self, PyObject *args)
|
|
{
|
|
PyObject* integer = PyNumber_Int( args );
|
|
short value;
|
|
|
|
if( !integer )
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
"expected integer argument" );
|
|
|
|
value = ( short )PyInt_AS_LONG( integer );
|
|
Py_DECREF( integer );
|
|
|
|
if( value < IMA_SRC_FILE || value > IMA_SRC_GENERATED )
|
|
return EXPP_ReturnIntError( PyExc_ValueError,
|
|
"expected integer argument in range 1-4" );
|
|
|
|
self->image->source = value;
|
|
return 0;
|
|
}
|
|
|
|
static int Image_setFlag(BPy_Image *self, PyObject *value, void *flag)
|
|
{
|
|
if ( PyObject_IsTrue(value) )
|
|
self->image->flag |= (int)flag;
|
|
else
|
|
self->image->flag &= ~(int)flag;
|
|
return 0;
|
|
}
|
|
|
|
static int Image_setFlagTpage(BPy_Image *self, PyObject *value, void *flag)
|
|
{
|
|
if ( PyObject_IsTrue(value) )
|
|
self->image->tpageflag |= (int)flag;
|
|
else
|
|
self->image->tpageflag &= ~(int)flag;
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* get integer attributes
|
|
*/
|
|
static PyObject *getIntAttr( BPy_Image *self, void *type )
|
|
{
|
|
PyObject *attr = NULL;
|
|
int param;
|
|
struct Image *image = self->image;
|
|
|
|
switch( (int)type ) {
|
|
case EXPP_IMAGE_ATTR_XREP:
|
|
param = image->xrep;
|
|
break;
|
|
case EXPP_IMAGE_ATTR_YREP:
|
|
param = image->xrep;
|
|
break;
|
|
case EXPP_IMAGE_ATTR_START:
|
|
param = image->twsta;
|
|
break;
|
|
case EXPP_IMAGE_ATTR_END:
|
|
param = image->twend;
|
|
break;
|
|
case EXPP_IMAGE_ATTR_SPEED:
|
|
param = image->animspeed;
|
|
break;
|
|
case EXPP_IMAGE_ATTR_BINDCODE:
|
|
param = image->bindcode;
|
|
break;
|
|
case EXPP_IMAGE_ATTR_SOURCE:
|
|
param = image->source;
|
|
break;
|
|
default:
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"undefined type in getIntAttr" );
|
|
}
|
|
|
|
attr = PyInt_FromLong( param );
|
|
|
|
if( attr )
|
|
return attr;
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_MemoryError,
|
|
"PyInt_FromLong() failed!" );
|
|
}
|
|
|
|
|
|
/*
|
|
* set integer attributes which require clamping
|
|
*/
|
|
|
|
static int setIntAttrClamp( BPy_Image *self, PyObject *value, void *type )
|
|
{
|
|
void *param;
|
|
struct Image *image = self->image;
|
|
int min, max, size;
|
|
|
|
switch( (int)type ) {
|
|
case EXPP_IMAGE_ATTR_XREP:
|
|
min = EXPP_IMAGE_REP_MIN;
|
|
max = EXPP_IMAGE_REP_MAX;
|
|
size = 'h';
|
|
param = (void *)&image->xrep;
|
|
break;
|
|
case EXPP_IMAGE_ATTR_YREP:
|
|
min = EXPP_IMAGE_REP_MIN;
|
|
max = EXPP_IMAGE_REP_MAX;
|
|
size = 'h';
|
|
param = (void *)&image->yrep;
|
|
break;
|
|
case EXPP_IMAGE_ATTR_START:
|
|
min = 0;
|
|
max = 128;
|
|
size = 'h';
|
|
param = (void *)&image->twsta;
|
|
break;
|
|
case EXPP_IMAGE_ATTR_END:
|
|
min = 0;
|
|
max = 128;
|
|
size = 'h';
|
|
param = (void *)&image->twend;
|
|
break;
|
|
case EXPP_IMAGE_ATTR_SPEED:
|
|
min = 0;
|
|
max = 100;
|
|
size = 'h';
|
|
param = (void *)&image->animspeed;
|
|
break;
|
|
default:
|
|
return EXPP_ReturnIntError( PyExc_RuntimeError,
|
|
"undefined type in setIntAttrClamp");
|
|
}
|
|
return EXPP_setIValueClamped( value, param, min, max, size );
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Python attributes get/set structure: */
|
|
/*****************************************************************************/
|
|
static PyGetSetDef BPy_Image_getseters[] = {
|
|
GENERIC_LIB_GETSETATTR,
|
|
{"filename", (getter)Image_getFilename, (setter)Image_setFilename,
|
|
"image path", NULL},
|
|
/* readonly */
|
|
{"depth", (getter)Image_getDepth, (setter)NULL,
|
|
"image depth", NULL},
|
|
{"size", (getter)Image_getSize, (setter)NULL,
|
|
"image size", NULL},
|
|
{"packed", (getter)Image_getPacked, (setter)NULL,
|
|
"image packed state", NULL },
|
|
{"has_data", (getter)Image_hasData, (setter)NULL,
|
|
"is image data loaded", NULL },
|
|
/* ints */
|
|
{"xrep", (getter)getIntAttr, (setter)setIntAttrClamp,
|
|
"image xrep", (void *)EXPP_IMAGE_ATTR_XREP },
|
|
{"yrep", (getter)getIntAttr, (setter)setIntAttrClamp,
|
|
"image yrep", (void *)EXPP_IMAGE_ATTR_YREP },
|
|
{"start", (getter)getIntAttr, (setter)setIntAttrClamp,
|
|
"image start frame", (void *)EXPP_IMAGE_ATTR_START },
|
|
{"end", (getter)getIntAttr, (setter)setIntAttrClamp,
|
|
"image end frame", (void *)EXPP_IMAGE_ATTR_END },
|
|
{"speed", (getter)getIntAttr, (setter)setIntAttrClamp,
|
|
"image end frame", (void *)EXPP_IMAGE_ATTR_SPEED },
|
|
{"bindcode", (getter)getIntAttr, (setter)NULL,
|
|
"openGL bindcode", (void *)EXPP_IMAGE_ATTR_BINDCODE },
|
|
{"source", (getter)getIntAttr, (setter)Image_setSource,
|
|
"image source type", (void *)EXPP_IMAGE_ATTR_SOURCE },
|
|
/* flags */
|
|
{"fields", (getter)Image_getFlag, (setter)Image_setFlag,
|
|
"image fields toggle", (void *)IMA_FIELDS },
|
|
{"fields_odd", (getter)Image_getFlag, (setter)Image_setFlag,
|
|
"image odd fields toggle", (void *)IMA_STD_FIELD },
|
|
{"antialias", (getter)Image_getFlag, (setter)Image_setFlag,
|
|
"image antialiasing toggle", (void *)IMA_ANTIALI },
|
|
{"reflect", (getter)Image_getFlag, (setter)Image_setFlag,
|
|
"image reflect toggle", (void *)IMA_REFLECT },
|
|
{"clampX", (getter)Image_getFlagTpage, (setter)Image_setFlagTpage,
|
|
"disable tiling on the X axis", (void *)IMA_CLAMP_U },
|
|
{"clampY", (getter)Image_getFlagTpage, (setter)Image_setFlagTpage,
|
|
"disable tiling on the Y axis", (void *)IMA_CLAMP_V },
|
|
{NULL,NULL,NULL,NULL,NULL} /* Sentinel */
|
|
};
|
|
|
|
|
|
/*****************************************************************************/
|
|
/* Python Image_Type structure definition: */
|
|
/*****************************************************************************/
|
|
PyTypeObject Image_Type = {
|
|
PyObject_HEAD_INIT( NULL ) /* required macro. ( no comma needed ) */
|
|
0, /* ob_size */
|
|
"Blender Image", /* tp_name */
|
|
sizeof( BPy_Image ), /* tp_basicsize */
|
|
0, /* tp_itemsize */
|
|
/* methods */
|
|
NULL, /* tp_dealloc */
|
|
0, /* tp_print */
|
|
NULL, /* tp_getattr */
|
|
NULL, /* tp_setattr */
|
|
( cmpfunc ) Image_compare, /* tp_compare */
|
|
( reprfunc ) Image_repr, /* tp_repr */
|
|
|
|
/* Method suites for standard classes */
|
|
|
|
NULL, /* PyNumberMethods *tp_as_number; */
|
|
NULL, /* PySequenceMethods *tp_as_sequence; */
|
|
NULL, /* PyMappingMethods *tp_as_mapping; */
|
|
|
|
/* More standard operations (here for binary compatibility) */
|
|
|
|
( hashfunc ) GenericLib_hash, /* hashfunc tp_hash; */
|
|
NULL, /* ternaryfunc tp_call; */
|
|
NULL, /* reprfunc tp_str; */
|
|
NULL, /* getattrofunc tp_getattro; */
|
|
NULL, /* setattrofunc tp_setattro; */
|
|
|
|
/* Functions to access object as input/output buffer */
|
|
NULL, /* PyBufferProcs *tp_as_buffer; */
|
|
|
|
/*** Flags to define presence of optional/expanded features ***/
|
|
Py_TPFLAGS_DEFAULT, /* long tp_flags; */
|
|
|
|
NULL, /* char *tp_doc; Documentation string */
|
|
/*** Assigned meaning in release 2.0 ***/
|
|
/* call function for all accessible objects */
|
|
NULL, /* traverseproc tp_traverse; */
|
|
|
|
/* delete references to contained objects */
|
|
NULL, /* inquiry tp_clear; */
|
|
|
|
/*** Assigned meaning in release 2.1 ***/
|
|
/*** rich comparisons ***/
|
|
NULL, /* richcmpfunc tp_richcompare; */
|
|
|
|
/*** weak reference enabler ***/
|
|
0, /* long tp_weaklistoffset; */
|
|
|
|
/*** Added in release 2.2 ***/
|
|
/* Iterators */
|
|
NULL, /* getiterfunc tp_iter; */
|
|
NULL, /* iternextfunc tp_iternext; */
|
|
|
|
/*** Attribute descriptor and subclassing stuff ***/
|
|
BPy_Image_methods, /* struct PyMethodDef *tp_methods; */
|
|
NULL, /* struct PyMemberDef *tp_members; */
|
|
BPy_Image_getseters, /* struct PyGetSetDef *tp_getset; */
|
|
NULL, /* struct _typeobject *tp_base; */
|
|
NULL, /* PyObject *tp_dict; */
|
|
NULL, /* descrgetfunc tp_descr_get; */
|
|
NULL, /* descrsetfunc tp_descr_set; */
|
|
0, /* long tp_dictoffset; */
|
|
NULL, /* initproc tp_init; */
|
|
NULL, /* allocfunc tp_alloc; */
|
|
NULL, /* newfunc tp_new; */
|
|
/* Low-level free-memory routine */
|
|
NULL, /* freefunc tp_free; */
|
|
/* For PyObject_IS_GC */
|
|
NULL, /* inquiry tp_is_gc; */
|
|
NULL, /* PyObject *tp_bases; */
|
|
/* method resolution order */
|
|
NULL, /* PyObject *tp_mro; */
|
|
NULL, /* PyObject *tp_cache; */
|
|
NULL, /* PyObject *tp_subclasses; */
|
|
NULL, /* PyObject *tp_weaklist; */
|
|
NULL
|
|
};
|