This repository has been archived on 2023-10-09. You can view files and clone it. You cannot open issues or pull requests or push a commit.
Files
blender-archive/source/blender/python/api2_2x/Blender.c
Joseph Gilbert b89035906d Mathutils update
- also included is some fixes for preprocessor inclues and some clean up of the previous commit

-rewrite and bugfixes
  ----------------------------------
  Here's my changelog:
  -fixed Rand() so that it doesn't seed everytime and should generate better random numbers
  - changed a few error return types to something more appropriate
  - clean up of uninitialized variables & removal of unneccessary objects
  - NMesh returns wrapped vectors now
  - World returns wrapped matrices now
  - Object.getEuler() and Object.getBoundingBox() return Wrapped data when data is present
  - Object.getMatrix() returns wrapped data if it's worldspace, 'localspace' returns a new matrix
  - Vector, Euler, Mat, Quat, call all now internally wrap object without destroying internal datablocks
  - Removed memory allocation (unneeded) from all methods
  - Vector's resize methods are only applicable to new vectors not wrapped data.
  - Matrix(), Quat(), Euler(), Vector() now accepts ANY sequence list, including tuples, list, or a self object to copy - matrices accept multiple sequences
  - Fixed Slerp() so that it now works correctly values are clamped between 0 and 1
  - Euler.rotate does internal rotation now
  - Slice assignment now works better for all types
  - Vector * Vector and Quat * Quat are defined and return the DOT product
  - Mat * Vec and Vec * Mat are defined now
  - Moved #includes to .c file from headers. Also fixed prototypes in mathutils
  - Added new helper functions for incref'ing to genutils
  - Major cleanup of header files includes - include Mathutils.h for access to math types
  - matrix.toQuat() and .toEuler() now fixed take appropriate matrix sizes
  - Matrix() with no parameters now returns an identity matrix by default not a zero matrix
  - printf() now prints with 6 digits instead of 4
  - printf() now prints output with object descriptor
  - Matrices now support [x][y] assignment (e.g. matrix[x][y] = 5.4)
  - Matrix[index] = value now expectes a sequence not an integer. This will now set a ROW of the matrix through a sequence.  index cannot go above the row size of the matrix.
  - slice operations on matrices work with sequences now (rows of the matrix) example:  mymatrix[0:2] returns a list of 2 wrapped vectors with access to the matrix data.
  - slice assignment will no longer modify the data if the assignment operation fails
  - fixed error in matrix * scalar multiplication
  - euler.toMatrix(), toQuat() no longer causes "creep" from repeated use
  - Wrapped data will generate wrapped objects when toEuler(), toQuat(), toMatrix() is used
  - Quats can be created with angle/axis, axis/angle
  - 4x4 matrices can be multiplied by 3D vectors (by popular demand :))
  - vec *quat / quat * vec is now defined
  - vec.magnitude alias for vec.length
  - all self, internal methods return a pointer to self now so you can do print vector.internalmethod() or vector.internalmethod().nextmethod() (no more print matrix.inverse() returning 'none')
  - these methods have been deprecated (still functioning but suggested to use the corrected functionality):
    * CopyVec() - replaced by Vector() functionality
    * CopyMat() - replaced by Matrix() functionality
    * CopyQuat() - replace by Quaternion() functionality
    * CopyEuler() - replaced by Euler() functionality
    * RotateEuler() - replaced by Euler.rotate() funtionality
    * MatMultVec() - replaced by matrix * vector
    * VecMultMat() - replaced by vector * matrix
  -  New struct containers references to python object data or internally allocated blender data for wrapping
  * Explaination here:  math structs now function as a 'simple wrapper' or a 'py_object' - data that is created on the fly will now be a 'py_object' with its memory managed by python
  *    otherwise if the data is returned by blender's G.main then the math object is a 'simple wrapper' and data can be accessed directly from the struct just like other python objects.
2005-07-14 03:34:56 +00:00

809 lines
27 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): Michel Selten, Willian P. Germano, Joseph Gilbert,
* Campbell Barton
*
* ***** END GPL/BL DUAL LICENSE BLOCK *****
*/
#include <Python.h>
#include <stdio.h>
/* for open, close in Blender_Load */
#include <fcntl.h>
#ifndef WIN32
#include <unistd.h>
#else
#include <io.h>
#endif
#include "BKE_utildefines.h"
#include "BDR_editobject.h" /* exit_editmode() */
#include "BIF_usiblender.h"
#include "BLI_blenlib.h"
#include "BLO_writefile.h"
#include "BKE_exotic.h"
#include "BKE_global.h"
#include "BKE_packedFile.h"
#include "BKE_object.h"
#include "BKE_text.h"
#include "BPI_script.h"
#include "BSE_headerbuttons.h"
#include "DNA_ID.h"
#include "DNA_object_types.h"
#include "DNA_scene_types.h"
#include "DNA_screen_types.h" /* for SPACE_VIEW3D */
#include "DNA_space_types.h" /* for SPACE_VIEW3D */
#include "DNA_scriptlink_types.h"
#include "DNA_userdef_types.h"
#include "BKE_ipo.h"
#include "blendef.h"
#include "EXPP_interface.h" /* for bpy_gethome() */
#include "gen_utils.h"
#include "modules.h"
#include "constant.h"
#include "../BPY_extern.h" /* BPY_txt_do_python_Text */
#include "../BPY_menus.h" /* to update menus */
extern PyObject *bpy_registryDict; /* defined in ../BPY_interface.c */
/**********************************************************/
/* Python API function prototypes for the Blender module. */
/**********************************************************/
static PyObject *Blender_Set( PyObject * self, PyObject * args );
static PyObject *Blender_Get( PyObject * self, PyObject * args );
static PyObject *Blender_Redraw( PyObject * self, PyObject * args );
static PyObject *Blender_Quit( PyObject * self );
static PyObject *Blender_Load( PyObject * self, PyObject * args );
static PyObject *Blender_Save( PyObject * self, PyObject * args );
static PyObject *Blender_Run( PyObject * self, PyObject * args );
static PyObject *Blender_ShowHelp( PyObject * self, PyObject * args );
static PyObject *Blender_UpdateMenus( PyObject * self);
extern PyObject *Text3d_Init( void ); /* missing in some include */
/*****************************************************************************/
/* The following string definitions are used for documentation strings. */
/* In Python these will be written to the console when doing a */
/* Blender.__doc__ */
/*****************************************************************************/
static char Blender_Set_doc[] =
"(request, data) - Update settings in Blender\n\
\n\
(request) A string identifying the setting to change\n\
'curframe' - Sets the current frame using the number in data";
static char Blender_Get_doc[] = "(request) - Retrieve settings from Blender\n\
\n\
(request) A string indentifying the data to be returned\n\
'curframe' - Returns the current animation frame\n\
'curtime' - Returns the current animation time\n\
'staframe' - Returns the start frame of the animation\n\
'endframe' - Returns the end frame of the animation\n\
'filename' - Returns the name of the last file read or written\n\
'homedir' - Returns Blender's home dir\n\
'datadir' - Returns the dir where scripts can save their data, if available\n\
'scriptsdir' - Returns the main dir where scripts are kept, if available\n\
'uscriptsdir' - Returns the user defined dir for scripts, if available\n\
'version' - Returns the Blender version number";
static char Blender_Redraw_doc[] = "() - Redraw all 3D windows";
static char Blender_Quit_doc[] =
"() - Quit Blender. The current data is saved as 'quit.blend' before leaving.";
static char Blender_Load_doc[] = "(filename) - Load the given file.\n\
Supported formats:\n\
Blender, DXF, Inventor 1.0 ASCII, VRML 1.0 asc, STL, Videoscape, radiogour.\n\
\n\
Notes:\n\
1 - () - an empty argument loads the default .B.blend file;\n\
2 - if the substring '.B.blend' occurs inside 'filename', the default\n\
.B.blend file is loaded;\n\
3 - If a Blender file is loaded the script ends immediately.\n\
4 - The current data is always preserved as an autosave file, for safety;\n\
5 - This function only works if the script where it's executed is the\n\
only one running at the moment.";
static char Blender_Save_doc[] =
"(filename) - Save data to a file based on the filename's extension.\n\
Supported are: Blender's .blend and the builtin exporters:\n\
VRML 1.0 (.wrl), Videoscape (.obj), DXF (.dxf) and STL (.stl)\n\
(filename) - A filename with one of the supported extensions.\n\
Note 1: 'filename' should not contain the substring \".B.blend\" in it.\n\
Note 2: only .blend raises an error if file wasn't saved.\n\
\tYou can use Blender.sys.exists(filename) to make sure the file was saved\n\
\twhen writing to one of the other formats.";
static char Blender_Run_doc[] =
"(script) - Run the given Python script.\n\
(script) - the path to a file or the name of an available Blender Text.";
static char Blender_ShowHelp_doc[] =
"(script) - Show help for the given Python script.\n\
This will try to open the 'Scripts Help Browser' script, so to have\n\
any help displayed the passed 'script' must be properly documented\n\
with the expected strings (check API ref docs or any bundled script\n\
for examples).\n\n\
(script) - the filename of a script in the default or user defined\n\
scripts dir (no need to supply the full path name).";
static char Blender_UpdateMenus_doc[] =
"() - Update the menus where scripts are registered. Only needed for\n\
scripts that save other new scripts in the default or user defined folders.";
/*****************************************************************************/
/* Python method structure definition. */
/*****************************************************************************/
static struct PyMethodDef Blender_methods[] = {
{"Set", Blender_Set, METH_VARARGS, Blender_Set_doc},
{"Get", Blender_Get, METH_VARARGS, Blender_Get_doc},
{"Redraw", Blender_Redraw, METH_VARARGS, Blender_Redraw_doc},
{"Quit", ( PyCFunction ) Blender_Quit, METH_NOARGS, Blender_Quit_doc},
{"Load", Blender_Load, METH_VARARGS, Blender_Load_doc},
{"Save", Blender_Save, METH_VARARGS, Blender_Save_doc},
{"Run", Blender_Run, METH_VARARGS, Blender_Run_doc},
{"ShowHelp", Blender_ShowHelp, METH_VARARGS, Blender_ShowHelp_doc},
{"UpdateMenus", ( PyCFunction ) Blender_UpdateMenus, METH_NOARGS,
Blender_UpdateMenus_doc},
{NULL, NULL, 0, NULL}
};
/*****************************************************************************/
/* Global variables */
/*****************************************************************************/
PyObject *g_blenderdict;
/*****************************************************************************/
/* Function: Blender_Set */
/* Python equivalent: Blender.Set */
/*****************************************************************************/
static PyObject *Blender_Set( PyObject * self, PyObject * args )
{
char *name;
PyObject *arg;
int framenum;
if( !PyArg_ParseTuple( args, "sO", &name, &arg ) ) {
/* TODO: Do we need to generate a nice error message here? */
return ( NULL );
}
if( StringEqual( name, "curframe" ) ) {
if( !PyArg_Parse( arg, "i", &framenum ) ) {
/* TODO: Do we need to generate a nice error message here? */
return ( NULL );
}
G.scene->r.cfra = framenum;
update_for_newframe( );
} else {
return ( EXPP_ReturnPyObjError( PyExc_AttributeError,
"bad request identifier" ) );
}
return ( EXPP_incr_ret( Py_None ) );
}
/*****************************************************************************/
/* Function: Blender_Get */
/* Python equivalent: Blender.Get */
/*****************************************************************************/
static PyObject *Blender_Get( PyObject * self, PyObject * args )
{
PyObject *ret = NULL, *dict = NULL;
char *str = NULL;
if( !PyArg_ParseTuple( args, "s", &str ) )
return EXPP_ReturnPyObjError (PyExc_TypeError,
"expected string argument");
if( StringEqual( str, "curframe" ) )
ret = PyInt_FromLong( G.scene->r.cfra );
else if( StringEqual( str, "curtime" ) )
ret = PyFloat_FromDouble( frame_to_float( G.scene->r.cfra ) );
else if( StringEqual( str, "staframe" ) )
ret = PyInt_FromLong( G.scene->r.sfra );
else if( StringEqual( str, "endframe" ) )
ret = PyInt_FromLong( G.scene->r.efra );
else if( StringEqual( str, "filename" ) )
ret = PyString_FromString( G.sce );
else if( StringEqual( str, "homedir" ) ) {
char *hdir = bpy_gethome(0);
if( hdir && BLI_exists( hdir ))
ret = PyString_FromString( hdir );
else
ret = EXPP_incr_ret( Py_None );
}
else if( StringEqual( str, "datadir" ) ) {
char datadir[FILE_MAXDIR];
char *sdir = bpy_gethome(1);
if (sdir) {
BLI_make_file_string( "/", datadir, sdir, "bpydata" );
if( BLI_exists( datadir ) )
ret = PyString_FromString( datadir );
}
if (!ret) ret = EXPP_incr_ret( Py_None );
}
else if(StringEqual(str, "udatadir")) {
if (U.pythondir[0] != '\0') {
char upydir[FILE_MAXDIR];
BLI_strncpy(upydir, U.pythondir, FILE_MAXDIR);
BLI_convertstringcode(upydir, G.sce, 0);
if (BLI_exists(upydir)) {
char udatadir[FILE_MAXDIR];
BLI_make_file_string("/", udatadir, upydir, "bpydata");
if (BLI_exists(udatadir))
ret = PyString_FromString(udatadir);
}
}
if (!ret) ret = EXPP_incr_ret(Py_None);
}
else if( StringEqual( str, "scriptsdir" ) ) {
char *sdir = bpy_gethome(1);
if (sdir)
ret = PyString_FromString(sdir);
else
ret = EXPP_incr_ret( Py_None );
}
else if( StringEqual( str, "uscriptsdir" ) ) {
if (U.pythondir[0] != '\0') {
char upydir[FILE_MAXDIR];
BLI_strncpy(upydir, U.pythondir, FILE_MAXDIR);
BLI_convertstringcode(upydir, G.sce, 0);
if( BLI_exists( upydir ) )
ret = PyString_FromString( upydir );
}
if (!ret) ret = EXPP_incr_ret(Py_None);
}
/* USER PREFS: */
else if( StringEqual( str, "yfexportdir" ) ) {
if (U.yfexportdir[0] != '\0') {
char yfexportdir[FILE_MAXDIR];
BLI_strncpy(yfexportdir, U.yfexportdir, FILE_MAXDIR);
BLI_convertstringcode(yfexportdir, G.sce, 0);
if( BLI_exists( yfexportdir ) )
ret = PyString_FromString( yfexportdir );
}
if (!ret) ret = EXPP_incr_ret(Py_None);
}
/* fontsdir */
else if( StringEqual( str, "fontsdir" ) ) {
if (U.fontdir[0] != '\0') {
char fontdir[FILE_MAXDIR];
BLI_strncpy(fontdir, U.fontdir, FILE_MAXDIR);
BLI_convertstringcode(fontdir, G.sce, 0);
if( BLI_exists( fontdir ) )
ret = PyString_FromString( fontdir );
}
if (!ret) ret = EXPP_incr_ret(Py_None);
}
/* texturesdir */
else if( StringEqual( str, "texturesdir" ) ) {
if (U.textudir[0] != '\0') {
char textudir[FILE_MAXDIR];
BLI_strncpy(textudir, U.textudir, FILE_MAXDIR);
BLI_convertstringcode(textudir, G.sce, 0);
if( BLI_exists( textudir ) )
ret = PyString_FromString( textudir );
}
if (!ret) ret = EXPP_incr_ret(Py_None);
}
/* texpluginsdir */
else if( StringEqual( str, "texpluginsdir" ) ) {
if (U.plugtexdir[0] != '\0') {
char plugtexdir[FILE_MAXDIR];
BLI_strncpy(plugtexdir, U.plugtexdir, FILE_MAXDIR);
BLI_convertstringcode(plugtexdir, G.sce, 0);
if( BLI_exists( plugtexdir ) )
ret = PyString_FromString( plugtexdir );
}
if (!ret) ret = EXPP_incr_ret(Py_None);
}
/* seqpluginsdir */
else if( StringEqual( str, "seqpluginsdir" ) ) {
if (U.plugseqdir[0] != '\0') {
char plugseqdir[FILE_MAXDIR];
BLI_strncpy(plugseqdir, U.plugseqdir, FILE_MAXDIR);
BLI_convertstringcode(plugseqdir, G.sce, 0);
if( BLI_exists( plugseqdir ) )
ret = PyString_FromString( plugseqdir );
}
if (!ret) ret = EXPP_incr_ret(Py_None);
}
/* renderdir */
else if( StringEqual( str, "renderdir" ) ) {
if (U.renderdir[0] != '\0') {
char renderdir[FILE_MAXDIR];
BLI_strncpy(renderdir, U.renderdir, FILE_MAXDIR);
BLI_convertstringcode(renderdir, G.sce, 0);
if( BLI_exists( renderdir ) )
ret = PyString_FromString( renderdir );
}
if (!ret) ret = EXPP_incr_ret(Py_None);
}
/* soundsdir */
else if( StringEqual( str, "soundsdir" ) ) {
if (U.sounddir[0] != '\0') {
char sounddir[FILE_MAXDIR];
BLI_strncpy(sounddir, U.sounddir, FILE_MAXDIR);
BLI_convertstringcode(sounddir, G.sce, 0);
if( BLI_exists( sounddir ) )
ret = PyString_FromString( sounddir );
}
if (!ret) ret = EXPP_incr_ret(Py_None);
}
/* tempdir */
else if( StringEqual( str, "tempdir" ) ) {
if (U.tempdir[0] != '\0') {
char tempdir[FILE_MAXDIR];
BLI_strncpy(tempdir, U.tempdir, FILE_MAXDIR);
BLI_convertstringcode(tempdir, G.sce, 0);
if( BLI_exists( tempdir ) )
ret = PyString_FromString( tempdir );
}
if (!ret) ret = EXPP_incr_ret(Py_None);
}
/* According to the old file (opy_blender.c), the following if
statement is a quick hack and needs some clean up. */
else if( StringEqual( str, "vrmloptions" ) ) {
ret = PyDict_New( );
PyDict_SetItemString( dict, "twoside",
PyInt_FromLong( U.vrmlflag & USER_VRML_TWOSIDED ) );
PyDict_SetItemString( dict, "layers",
PyInt_FromLong( U.vrmlflag & USER_VRML_LAYERS ) );
PyDict_SetItemString( dict, "autoscale",
PyInt_FromLong( U.vrmlflag & USER_VRML_AUTOSCALE ) );
} /* End 'quick hack' part. */
else if(StringEqual( str, "version" ))
ret = PyInt_FromLong( G.version );
else
return EXPP_ReturnPyObjError( PyExc_AttributeError, "unknown attribute" );
if (ret) return ret;
else
return EXPP_ReturnPyObjError (PyExc_MemoryError,
"could not create pystring!");
}
/*****************************************************************************/
/* Function: Blender_Redraw */
/* Python equivalent: Blender.Redraw */
/*****************************************************************************/
static PyObject *Blender_Redraw( PyObject * self, PyObject * args )
{
int wintype = SPACE_VIEW3D;
if( !PyArg_ParseTuple( args, "|i", &wintype ) ) {
return EXPP_ReturnPyObjError( PyExc_TypeError,
"expected int argument (or nothing)" );
}
return M_Window_Redraw( self, Py_BuildValue( "(i)", wintype ) );
}
/*****************************************************************************/
/* Function: Blender_Quit */
/* Python equivalent: Blender.Quit */
/*****************************************************************************/
static PyObject *Blender_Quit( PyObject * self )
{
BIF_write_autosave( ); /* save the current data first */
exit_usiblender( ); /* renames last autosave to quit.blend */
Py_INCREF( Py_None );
return Py_None;
}
/**
* Blender.Load
* loads Blender's .blend, DXF, radiogour(?), STL, Videoscape,
* Inventor 1.0 ASCII, VRML 1.0 asc.
*/
static PyObject *Blender_Load( PyObject * self, PyObject * args )
{
char *fname = NULL;
int keep_oldfname = 0;
Script *script = NULL;
char str[32], name[FILE_MAXDIR];
int file, is_blend_file = 0;
if( !PyArg_ParseTuple( args, "|si", &fname, &keep_oldfname ) )
return EXPP_ReturnPyObjError( PyExc_TypeError,
"expected filename and optional int or nothing as arguments" );
if( fname ) {
if( strlen( fname ) > FILE_MAXDIR ) /* G.main->name's max length */
return EXPP_ReturnPyObjError( PyExc_AttributeError,
"filename too long!" );
else if( !BLI_exists( fname ) )
return EXPP_ReturnPyObjError( PyExc_AttributeError,
"requested file doesn't exist!" );
if( keep_oldfname )
BLI_strncpy( name, G.sce, FILE_MAXDIR );
}
/* We won't let a new .blend file be loaded if there are still other
* scripts running, since loading a new file will close and remove them. */
if( G.main->script.first != G.main->script.last )
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
"there are other scripts running at the Scripts win, close them first!" );
if( fname ) {
file = open( fname, O_BINARY | O_RDONLY );
if( file <= 0 ) {
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
"cannot open file!" );
} else {
read( file, str, 31 );
close( file );
if( strncmp( str, "BLEN", 4 ) == 0 )
is_blend_file = 1;
}
} else
is_blend_file = 1; /* no arg given means default: .B.blend */
if( is_blend_file ) {
int during_slink = during_scriptlink( );
/* when loading a .blend file from a scriptlink, the scriptlink pointer
* in BPY_do_pyscript becomes invalid during a loop. Inform it here.
* Also do not allow a nested scriptlink (called from inside another)
* to load .blend files, to avoid nasty problems. */
if( during_slink >= 1 ) {
if( during_slink == 1 )
disable_where_scriptlink( -1 );
else {
return EXPP_ReturnPyObjError
( PyExc_EnvironmentError,
"Blender.Load: cannot load .blend files from a nested scriptlink." );
}
}
/* trick: mark the script so that its script struct won't be freed after
* the script is executed (to avoid a double free warning on exit): */
script = G.main->script.first;
if( script )
script->flags |= SCRIPT_GUI;
BIF_write_autosave( ); /* for safety let's preserve the current data */
}
if( G.obedit )
exit_editmode( 1 );
/* for safety, any filename with .B.blend is considered the default one.
* It doesn't seem necessary to compare file attributes (like st_ino and
* st_dev, according to the glibc info pages) to find out if the given
* filename, that may have been given with a twisted misgiving path, is the
* default one for sure. Taking any .B.blend file as the default is good
* enough here. Note: the default file requires extra clean-up done by
* BIF_read_homefile: freeing the user theme data. */
if( !fname || ( strstr( fname, ".B.blend" ) && is_blend_file ) )
BIF_read_homefile( );
else
BIF_read_file( fname );
if( fname && keep_oldfname ) {
/*BLI_strncpy(G.main->name, name, FILE_MAXDIR); */
BLI_strncpy( G.sce, name, FILE_MAXDIR );
}
Py_INCREF( Py_None );
return Py_None;
}
static PyObject *Blender_Save( PyObject * self, PyObject * args )
{
char *fname = NULL;
int overwrite = 0, len = 0;
char *error = NULL;
Library *li;
if( !PyArg_ParseTuple( args, "s|i", &fname, &overwrite ) )
return EXPP_ReturnPyObjError( PyExc_TypeError,
"expected filename and optional int (overwrite flag) as arguments" );
for( li = G.main->library.first; li; li = li->id.next ) {
if( BLI_streq( li->name, fname ) ) {
return EXPP_ReturnPyObjError( PyExc_AttributeError,
"cannot overwrite used library" );
}
}
/* for safety, any filename with .B.blend is considered the default one
* and not accepted here. */
if( strstr( fname, ".B.blend" ) )
return EXPP_ReturnPyObjError( PyExc_AttributeError,
"filename can't contain the substring \".B.blend\" in it." );
len = strlen( fname );
if( len > FILE_MAXFILE )
return EXPP_ReturnPyObjError( PyExc_AttributeError,
"filename is too long!" );
else if( BLI_exists( fname ) && !overwrite )
return EXPP_ReturnPyObjError( PyExc_AttributeError,
"file already exists and overwrite flag was not given." );
disable_where_script( 1 ); /* to avoid error popups in the write_* functions */
if( BLI_testextensie( fname, ".blend" ) ) {
if( G.fileflags & G_AUTOPACK )
packAll( );
if( !BLO_write_file( fname, G.fileflags, &error ) ) {
disable_where_script( 0 );
return EXPP_ReturnPyObjError( PyExc_SystemError,
error );
}
} else if( BLI_testextensie( fname, ".dxf" ) )
write_dxf( fname );
else if( BLI_testextensie( fname, ".stl" ) )
write_stl( fname );
else if( BLI_testextensie( fname, ".wrl" ) )
write_vrml( fname );
else if( BLI_testextensie( fname, ".obj" ) )
write_videoscape( fname );
else {
disable_where_script( 0 );
return EXPP_ReturnPyObjError( PyExc_AttributeError,
"unknown file extension." );
}
disable_where_script( 0 );
Py_INCREF( Py_None );
return Py_None;
}
static PyObject *Blender_ShowHelp(PyObject *self, PyObject *args)
{
PyObject *script = NULL;
char hspath[FILE_MAXDIR + FILE_MAXFILE]; /* path to help_browser.py */
char *sdir = bpy_gethome(1);
PyObject *rkeyd = NULL, *arglist = NULL;
if (!PyArg_ParseTuple(args, "O!", &PyString_Type, &script))
return EXPP_ReturnPyObjError(PyExc_TypeError,
"expected a script filename as argument");
/* first we try to find the help_browser script */
if (sdir) BLI_make_file_string("/", hspath, sdir, "help_browser.py");
if (!sdir || (!BLI_exists(hspath) && (U.pythondir[0] != '\0'))) {
char upydir[FILE_MAXDIR];
BLI_strncpy(upydir, U.pythondir, FILE_MAXDIR);
BLI_convertstringcode(upydir, G.sce, 0);
BLI_make_file_string("/", hspath, upydir, "help_browser.py");
if (!BLI_exists(hspath))
return EXPP_ReturnPyObjError(PyExc_RuntimeError,
"can't find script help_browser.py");
}
/* now we store the passed script in the registry dict and call the
* help_browser to show help info for it */
rkeyd = PyDict_New();
if (!rkeyd)
return EXPP_ReturnPyObjError(PyExc_MemoryError,
"can't create py dictionary!");
PyDict_SetItemString(rkeyd, "script", script);
PyDict_SetItemString(bpy_registryDict, "__help_browser", rkeyd);
arglist = Py_BuildValue("(s)", hspath);
Blender_Run(self, arglist);
Py_DECREF(arglist);
return EXPP_incr_ret(Py_None);
}
static PyObject *Blender_Run(PyObject *self, PyObject *args)
{
char *fname = NULL;
Text *text = NULL;
int is_blender_text = 0;
Script *script = NULL;
if (!PyArg_ParseTuple(args, "s", &fname))
return EXPP_ReturnPyObjError(PyExc_TypeError,
"expected a filename or a Blender Text name as argument");
if (!BLI_exists(fname)) { /* if there's no such filename ... */
text = G.main->text.first; /* try an already existing Blender Text */
while (text) {
if (!strcmp(fname, text->id.name + 2)) break;
text = text->id.next;
}
if (!text) {
return EXPP_ReturnPyObjError(PyExc_AttributeError,
"no such file or Blender text");
}
else is_blender_text = 1; /* fn is already a Blender Text */
}
else {
text = add_text(fname);
if (!text) {
return EXPP_ReturnPyObjError(PyExc_RuntimeError,
"couldn't create Blender Text from given file");
}
}
/* (this is messy, check Draw.c's Method_Register and Window.c's file
* selector for more info)
* - caller script is the one that called this Blender_Run function;
* - called script is the argument to this function: fname;
* To mark scripts whose global dicts can't be freed right after
* the script execution (or better, 'first pass', since these scripts
* leave callbacks for gui or file/image selectors) we flag them. But to
* get a pointer to them we need to check which one is currently
* running (if none we're already at a spacescript). To make sure only
* the called script will have the SCRIPT_RUNNING flag on, we unset it
* for the caller script here: */
script = G.main->script.first;
while (script) {
if (script->flags & SCRIPT_RUNNING) break;
script = script->id.next;
}
if (script) script->flags &= ~SCRIPT_RUNNING; /* unset */
BPY_txt_do_python_Text(text); /* call new script */
if (script) script->flags |= SCRIPT_RUNNING; /* set */
if (!is_blender_text) free_libblock(&G.main->text, text);
return EXPP_incr_ret(Py_None);
}
static PyObject * Blender_UpdateMenus( PyObject * self )
{
BPyMenu_RemoveAllEntries();
if (BPyMenu_Init(1) == -1)
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
"invalid scripts dir");
Py_INCREF( Py_None );
return Py_None;
}
/*****************************************************************************/
/* Function: initBlender */
/*****************************************************************************/
void M_Blender_Init( void )
{
PyObject *module;
PyObject *dict, *smode, *SpaceHandlers;
g_blenderdict = NULL;
module = Py_InitModule3( "Blender", Blender_methods,
"The main Blender module" );
types_InitAll( ); /* set all our pytypes to &PyType_Type */
SpaceHandlers = M_constant_New();
if (SpaceHandlers) {
BPy_constant *d = (BPy_constant *)SpaceHandlers;
constant_insert(d,"VIEW3D_EVENT",PyInt_FromLong(SPACEHANDLER_VIEW3D_EVENT));
constant_insert(d,"VIEW3D_DRAW", PyInt_FromLong(SPACEHANDLER_VIEW3D_DRAW));
PyModule_AddObject(module, "SpaceHandlers", SpaceHandlers);
}
if (G.background)
smode = PyString_FromString("background");
else
smode = PyString_FromString("interactive");
dict = PyModule_GetDict( module );
g_blenderdict = dict;
PyDict_SetItemString( dict, "bylink", EXPP_incr_ret_False() );
PyDict_SetItemString( dict, "link", EXPP_incr_ret ( Py_None ) );
PyDict_SetItemString( dict, "event", PyString_FromString( "" ) );
PyDict_SetItemString( dict, "mode", smode );
PyDict_SetItemString( dict, "Types", Types_Init( ) );
PyDict_SetItemString( dict, "sys", sys_Init( ) );
PyDict_SetItemString( dict, "Registry", Registry_Init( ) );
PyDict_SetItemString( dict, "Scene", Scene_Init( ) );
PyDict_SetItemString( dict, "Object", Object_Init( ) );
PyDict_SetItemString( dict, "Material", Material_Init( ) );
PyDict_SetItemString( dict, "Camera", Camera_Init( ) );
PyDict_SetItemString( dict, "Lamp", Lamp_Init( ) );
PyDict_SetItemString( dict, "Lattice", Lattice_Init( ) );
PyDict_SetItemString( dict, "Curve", Curve_Init( ) );
PyDict_SetItemString( dict, "Armature", Armature_Init( ) );
PyDict_SetItemString( dict, "Ipo", Ipo_Init( ) );
PyDict_SetItemString( dict, "IpoCurve", IpoCurve_Init( ) );
PyDict_SetItemString( dict, "Metaball", Metaball_Init( ) );
PyDict_SetItemString( dict, "Image", Image_Init( ) );
PyDict_SetItemString( dict, "Window", Window_Init( ) );
PyDict_SetItemString( dict, "Draw", Draw_Init( ) );
PyDict_SetItemString( dict, "BGL", BGL_Init( ) );
PyDict_SetItemString( dict, "Effect", Effect_Init( ) );
PyDict_SetItemString( dict, "Text", Text_Init( ) );
PyDict_SetItemString( dict, "Text3d", Text3d_Init( ) );
PyDict_SetItemString( dict, "World", World_Init( ) );
PyDict_SetItemString( dict, "Texture", Texture_Init( ) );
PyDict_SetItemString( dict, "NMesh", NMesh_Init( ) );
PyDict_SetItemString( dict, "Noise", Noise_Init( ) );
PyDict_SetItemString( dict, "Mathutils", Mathutils_Init( ) );
PyDict_SetItemString( dict, "Library", Library_Init( ) );
PyDict_SetItemString( dict, "Sound", Sound_Init( ) );
PyDict_SetItemString( dict, "CurNurb", CurNurb_Init( ) );
PyDict_SetItemString( dict, "BezTriple", BezTriple_Init( ) );
PyModule_AddIntConstant( module, "TRUE", 1 );
PyModule_AddIntConstant( module, "FALSE", 0 );
}