OBJ exporter working (Python 3.0), but needs testing and fixing.
Current issues: - NURBS - needs API additions - "all scenes" export - cannot switch scene in bpy - normal calculation, disabled - duplis - need testing, only dupliverts tested - matrix problem - UI, 18 options currently don't fit into filesel panel, will do manual lay out once it's available - probably others... BPY: - made operator "execute" method required to avoid crash - added bpy.sys module which replicates old "sys" module API: - replaced create_*_mesh with a single create_mesh accepting type parameter - added Mesh.create_copy to create a copy of a mesh with 0 users Ran `dos2unix` on source/blender/python/SConscript
This commit is contained in:
File diff suppressed because it is too large
Load Diff
@@ -64,7 +64,7 @@ def write(filename, scene, ob, \
|
||||
raise Exception("Error, Select 1 active object")
|
||||
return
|
||||
|
||||
file = open(filename, 'wb')
|
||||
file = open(filename, 'w')
|
||||
|
||||
|
||||
#EXPORT_EDGES = Draw.Create(0)
|
||||
@@ -123,8 +123,8 @@ def write(filename, scene, ob, \
|
||||
mesh_verts = mesh.verts # save a lookup
|
||||
ply_verts = [] # list of dictionaries
|
||||
# vdict = {} # (index, normal, uv) -> new index
|
||||
vdict = [{} for i in xrange(len(mesh_verts))]
|
||||
ply_faces = [[] for f in xrange(len(mesh.faces))]
|
||||
vdict = [{} for i in range(len(mesh_verts))]
|
||||
ply_faces = [[] for f in range(len(mesh.faces))]
|
||||
vert_count = 0
|
||||
for i, f in enumerate(mesh.faces):
|
||||
|
||||
|
||||
@@ -863,19 +863,24 @@ def make_kf_obj_node(obj, name_to_id):
|
||||
"""
|
||||
|
||||
import BPyMessages
|
||||
def save_3ds(filename):
|
||||
def save_3ds(filename, context):
|
||||
'''Save the Blender scene to a 3ds file.'''
|
||||
# Time the export
|
||||
|
||||
if not filename.lower().endswith('.3ds'):
|
||||
filename += '.3ds'
|
||||
|
||||
if not BPyMessages.Warning_SaveOver(filename):
|
||||
return
|
||||
# XXX
|
||||
# if not BPyMessages.Warning_SaveOver(filename):
|
||||
# return
|
||||
|
||||
time1= Blender.sys.time()
|
||||
Blender.Window.WaitCursor(1)
|
||||
sce= bpy.data.scenes.active
|
||||
# XXX
|
||||
time1 = bpy.sys.time()
|
||||
# time1= Blender.sys.time()
|
||||
# Blender.Window.WaitCursor(1)
|
||||
|
||||
sce = context.scene
|
||||
# sce= bpy.data.scenes.active
|
||||
|
||||
# Initialize the main chunk (primary):
|
||||
primary = _3ds_chunk(PRIMARY)
|
||||
@@ -901,7 +906,8 @@ def save_3ds(filename):
|
||||
# each material is added once):
|
||||
materialDict = {}
|
||||
mesh_objects = []
|
||||
for ob in sce.objects.context:
|
||||
for ob in context.selected_objects:
|
||||
# for ob in sce.objects.context:
|
||||
for ob_derived, mat in getDerivedObjects(ob, False):
|
||||
data = getMeshFromObject(ob_derived, None, True, False, sce)
|
||||
if data:
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
@@ -32,7 +32,7 @@ class SCRIPT_HT_header(bpy.types.Header):
|
||||
|
||||
if context.area.show_menus:
|
||||
row = layout.row(align=True)
|
||||
row.itemM(context, "SCRIPT_MT_scripts")
|
||||
row.itemM("SCRIPT_MT_scripts")
|
||||
|
||||
class SCRIPT_MT_scripts(bpy.types.Menu):
|
||||
__space_type__ = "SCRIPTS_WINDOW"
|
||||
@@ -41,7 +41,7 @@ class SCRIPT_MT_scripts(bpy.types.Menu):
|
||||
def draw(self, context):
|
||||
layout = self.layout
|
||||
layout.column()
|
||||
layout.itemM(context, "SCRIPT_MT_export")
|
||||
layout.itemM("SCRIPT_MT_export")
|
||||
layout.itemO("SCRIPT_OT_reload_scripts")
|
||||
|
||||
class SCRIPT_MT_export(bpy.types.Menu):
|
||||
|
||||
@@ -36,6 +36,8 @@
|
||||
|
||||
#include "BKE_customdata.h"
|
||||
#include "BKE_DerivedMesh.h"
|
||||
#include "BKE_mesh.h"
|
||||
|
||||
#include "BLI_arithb.h"
|
||||
|
||||
#include "DNA_mesh_types.h"
|
||||
@@ -66,6 +68,14 @@ void rna_Mesh_transform(Mesh *me, float *mat)
|
||||
}
|
||||
}
|
||||
|
||||
Mesh *rna_Mesh_create_copy(Mesh *me)
|
||||
{
|
||||
Mesh *ret= copy_mesh(me);
|
||||
ret->id.us--;
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
#if 0
|
||||
/* extern struct EditVert *addvertlist(EditMesh *em, float *vec, struct EditVert *example); */
|
||||
|
||||
@@ -101,6 +111,11 @@ void RNA_api_mesh(StructRNA *srna)
|
||||
parm= RNA_def_float_matrix(func, "matrix", 16, NULL, 0.0f, 0.0f, "", "Matrix.", 0.0f, 0.0f);
|
||||
RNA_def_property_flag(parm, PROP_REQUIRED);
|
||||
|
||||
func= RNA_def_function(srna, "create_copy", "rna_Mesh_create_copy");
|
||||
RNA_def_function_ui_description(func, "Create a copy of this Mesh datablock.");
|
||||
parm= RNA_def_pointer(func, "mesh", "Mesh", "", "Mesh, remove it if it is only used for export.");
|
||||
RNA_def_function_return(func, parm);
|
||||
|
||||
/*
|
||||
func= RNA_def_function(srna, "add_geom", "rna_Mesh_add_geom");
|
||||
RNA_def_function_ui_description(func, "Add geometry data to mesh.");
|
||||
|
||||
@@ -36,6 +36,12 @@
|
||||
|
||||
#include "DNA_object_types.h"
|
||||
|
||||
/* parameter to rna_Object_create_mesh */
|
||||
typedef enum CreateMeshType {
|
||||
CREATE_MESH_PREVIEW = 0,
|
||||
CREATE_MESH_RENDER = 1
|
||||
} CreateMeshType;
|
||||
|
||||
#ifdef RNA_RUNTIME
|
||||
|
||||
#include "BKE_customdata.h"
|
||||
@@ -55,7 +61,7 @@
|
||||
#include "ED_mesh.h"
|
||||
|
||||
/* copied from init_render_mesh (render code) */
|
||||
static Mesh *create_mesh(Object *ob, bContext *C, ReportList *reports, int render_mesh)
|
||||
static Mesh *rna_Object_create_mesh(Object *ob, bContext *C, ReportList *reports, int type)
|
||||
{
|
||||
/* CustomDataMask mask = CD_MASK_BAREMESH|CD_MASK_MTFACE|CD_MASK_MCOL; */
|
||||
CustomDataMask mask = CD_MASK_MESH; /* this seems more suitable, exporter,
|
||||
@@ -72,7 +78,12 @@ static Mesh *create_mesh(Object *ob, bContext *C, ReportList *reports, int rende
|
||||
return NULL;
|
||||
}
|
||||
|
||||
dm= render_mesh ? mesh_create_derived_render(sce, ob, mask) : mesh_create_derived_view(sce, ob, mask);
|
||||
if (type == CREATE_MESH_PREVIEW) {
|
||||
dm= mesh_create_derived_view(sce, ob, mask);
|
||||
}
|
||||
else {
|
||||
dm= mesh_create_derived_render(sce, ob, mask);
|
||||
}
|
||||
|
||||
if(!dm) {
|
||||
/* TODO: report */
|
||||
@@ -87,16 +98,6 @@ static Mesh *create_mesh(Object *ob, bContext *C, ReportList *reports, int rende
|
||||
return me;
|
||||
}
|
||||
|
||||
static Mesh *rna_Object_create_render_mesh(Object *ob, bContext *C, ReportList *reports)
|
||||
{
|
||||
return create_mesh(ob, C, reports, 1);
|
||||
}
|
||||
|
||||
static Mesh *rna_Object_create_preview_mesh(Object *ob, bContext *C, ReportList *reports)
|
||||
{
|
||||
return create_mesh(ob, C, reports, 0);
|
||||
}
|
||||
|
||||
/* When no longer needed, duplilist should be freed with Object.free_duplilist */
|
||||
static void rna_Object_create_duplilist(Object *ob, bContext *C, ReportList *reports)
|
||||
{
|
||||
@@ -162,30 +163,17 @@ void RNA_api_object(StructRNA *srna)
|
||||
FunctionRNA *func;
|
||||
PropertyRNA *parm;
|
||||
|
||||
/* copied from rna_def_object */
|
||||
static EnumPropertyItem object_type_items[] = {
|
||||
{OB_EMPTY, "EMPTY", 0, "Empty", ""},
|
||||
{OB_MESH, "MESH", 0, "Mesh", ""},
|
||||
{OB_CURVE, "CURVE", 0, "Curve", ""},
|
||||
{OB_SURF, "SURFACE", 0, "Surface", ""},
|
||||
{OB_FONT, "TEXT", 0, "Text", ""},
|
||||
{OB_MBALL, "META", 0, "Meta", ""},
|
||||
{OB_LAMP, "LAMP", 0, "Lamp", ""},
|
||||
{OB_CAMERA, "CAMERA", 0, "Camera", ""},
|
||||
{OB_WAVE, "WAVE", 0, "Wave", ""},
|
||||
{OB_LATTICE, "LATTICE", 0, "Lattice", ""},
|
||||
{OB_ARMATURE, "ARMATURE", 0, "Armature", ""},
|
||||
{0, NULL, 0, NULL, NULL}};
|
||||
static EnumPropertyItem mesh_type_items[] = {
|
||||
{CREATE_MESH_PREVIEW, "PREVIEW", 0, "Preview", "Apply preview settings."},
|
||||
{CREATE_MESH_RENDER, "RENDER", 0, "Render", "Apply render settings."},
|
||||
{0, NULL, 0, NULL, NULL}
|
||||
};
|
||||
|
||||
func= RNA_def_function(srna, "create_render_mesh", "rna_Object_create_render_mesh");
|
||||
RNA_def_function_ui_description(func, "Create a Mesh datablock with all modifiers applied for rendering.");
|
||||
RNA_def_function_flag(func, FUNC_USE_CONTEXT|FUNC_USE_REPORTS);
|
||||
parm= RNA_def_pointer(func, "mesh", "Mesh", "", "Mesh created from object, remove it if it is only used for export.");
|
||||
RNA_def_function_return(func, parm);
|
||||
|
||||
func= RNA_def_function(srna, "create_preview_mesh", "rna_Object_create_preview_mesh");
|
||||
RNA_def_function_ui_description(func, "Create a Mesh datablock with all modifiers applied for preview.");
|
||||
func= RNA_def_function(srna, "create_mesh", "rna_Object_create_mesh");
|
||||
RNA_def_function_ui_description(func, "Create a Mesh datablock with all modifiers applied.");
|
||||
RNA_def_function_flag(func, FUNC_USE_CONTEXT|FUNC_USE_REPORTS);
|
||||
parm= RNA_def_enum(func, "type", mesh_type_items, 0, "", "Type of mesh settings to apply.");
|
||||
RNA_def_property_flag(parm, PROP_REQUIRED);
|
||||
parm= RNA_def_pointer(func, "mesh", "Mesh", "", "Mesh created from object, remove it if it is only used for export.");
|
||||
RNA_def_function_return(func, parm);
|
||||
|
||||
|
||||
@@ -19,6 +19,7 @@
|
||||
#include "bpy_rna.h"
|
||||
#include "bpy_operator.h"
|
||||
#include "bpy_ui.h"
|
||||
#include "bpy_sys.h"
|
||||
#include "bpy_util.h"
|
||||
|
||||
#include "DNA_anim_types.h"
|
||||
@@ -91,6 +92,7 @@ void BPY_update_modules( void )
|
||||
PyObject *mod= PyImport_ImportModuleLevel("bpy", NULL, NULL, NULL, 0);
|
||||
PyModule_AddObject( mod, "data", BPY_rna_module() );
|
||||
PyModule_AddObject( mod, "types", BPY_rna_types() );
|
||||
PyModule_AddObject( mod, "sys", BPY_sys_module() );
|
||||
}
|
||||
|
||||
/*****************************************************************************
|
||||
|
||||
@@ -331,7 +331,7 @@ PyObject *PYOP_wrap_add(PyObject *self, PyObject *py_class)
|
||||
{PYOP_ATTR_UINAME, 's', 0, BPY_CLASS_ATTR_OPTIONAL},
|
||||
{PYOP_ATTR_PROP, 'l', 0, BPY_CLASS_ATTR_OPTIONAL},
|
||||
{PYOP_ATTR_DESCRIPTION, 's', 0, BPY_CLASS_ATTR_NONE_OK},
|
||||
{"execute", 'f', 2, BPY_CLASS_ATTR_OPTIONAL},
|
||||
{"execute", 'f', 2, 0},
|
||||
{"invoke", 'f', 3, BPY_CLASS_ATTR_OPTIONAL},
|
||||
{"poll", 'f', 2, BPY_CLASS_ATTR_OPTIONAL},
|
||||
{NULL, 0, 0, 0}
|
||||
|
||||
@@ -1367,9 +1367,13 @@ PyObject *pyrna_param_to_py(PointerRNA *ptr, PropertyRNA *prop, void *data)
|
||||
newptr= *(PointerRNA*)data;
|
||||
}
|
||||
else {
|
||||
/* XXX this is missing the ID part! */
|
||||
if (RNA_struct_is_ID(type)) {
|
||||
RNA_id_pointer_create(*(void**)data, &newptr);
|
||||
}
|
||||
else {
|
||||
RNA_pointer_create(NULL, type, *(void**)data, &newptr);
|
||||
}
|
||||
}
|
||||
|
||||
if (newptr.data) {
|
||||
ret = pyrna_struct_CreatePyObject(&newptr);
|
||||
|
||||
460
source/blender/python/intern/bpy_sys.c
Normal file
460
source/blender/python/intern/bpy_sys.c
Normal file
@@ -0,0 +1,460 @@
|
||||
/*
|
||||
* $Id: Sys.c 17889 2008-12-16 11:26:55Z campbellbarton $
|
||||
*
|
||||
* ***** BEGIN GPL 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.
|
||||
*
|
||||
* 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
|
||||
*
|
||||
* ***** END GPL LICENSE BLOCK *****
|
||||
*/
|
||||
|
||||
#include "bpy_sys.h" /*This must come first*/
|
||||
#include "bpy_util.h"
|
||||
|
||||
#include "BKE_utildefines.h"
|
||||
#include "BKE_global.h"
|
||||
#include "BKE_context.h"
|
||||
|
||||
#include "BLI_blenlib.h"
|
||||
|
||||
#include "DNA_scene_types.h" /* G.scene-"r.cfra */
|
||||
|
||||
#include "PIL_time.h"
|
||||
/* #include "gen_utils.h" */
|
||||
|
||||
#ifdef WIN32
|
||||
#define DIRSEP '\\'
|
||||
#define DIRSEP_STR "\\"
|
||||
#else
|
||||
#define DIRSEP '/'
|
||||
#define DIRSEP_STR "/"
|
||||
#endif
|
||||
|
||||
|
||||
/*****************************************************************************/
|
||||
/* Python API function prototypes for the sys module. */
|
||||
/*****************************************************************************/
|
||||
static PyObject *M_sys_basename( PyObject * self, PyObject * value );
|
||||
static PyObject *M_sys_dirname( PyObject * self, PyObject * value );
|
||||
static PyObject *M_sys_join( PyObject * self, PyObject * args );
|
||||
static PyObject *M_sys_splitext( PyObject * self, PyObject * value );
|
||||
static PyObject *M_sys_makename( PyObject * self, PyObject * args,
|
||||
PyObject * kw );
|
||||
static PyObject *M_sys_exists( PyObject * self, PyObject * value );
|
||||
static PyObject *M_sys_time( PyObject * self );
|
||||
static PyObject *M_sys_sleep( PyObject * self, PyObject * args );
|
||||
static PyObject *M_sys_expandpath( PyObject *self, PyObject *value);
|
||||
static PyObject *M_sys_cleanpath( PyObject *self, PyObject *value);
|
||||
static PyObject *M_sys_relpath( PyObject *self, PyObject *args);
|
||||
|
||||
/*****************************************************************************/
|
||||
/* The following string definitions are used for documentation strings. */
|
||||
/* In Python these will be written to the console when doing a */
|
||||
/* Blender.sys.__doc__ */
|
||||
/*****************************************************************************/
|
||||
static char M_sys_doc[] = "The Blender.sys submodule\n\
|
||||
\n\
|
||||
This is a minimal system module to supply simple functionality available\n\
|
||||
in the default Python module os.";
|
||||
|
||||
static char M_sys_basename_doc[] =
|
||||
"(path) - Split 'path' in dir and filename.\n\
|
||||
Return the filename.";
|
||||
|
||||
static char M_sys_dirname_doc[] =
|
||||
"(path) - Split 'path' in dir and filename.\n\
|
||||
Return the dir.";
|
||||
|
||||
static char M_sys_join_doc[] =
|
||||
"(dir, file) - Join dir and file to form a full filename.\n\
|
||||
Return the filename.";
|
||||
|
||||
static char M_sys_splitext_doc[] =
|
||||
"(path) - Split 'path' in root and extension:\n\
|
||||
/this/that/file.ext -> ('/this/that/file','.ext').\n\
|
||||
Return the pair (root, extension).";
|
||||
|
||||
static char M_sys_makename_doc[] =
|
||||
"(path = Blender.Get('filename'), ext = \"\", strip = 0) -\n\
|
||||
Strip dir and extension from path, leaving only a name, then append 'ext'\n\
|
||||
to it (if given) and return the resulting string.\n\n\
|
||||
(path) - string: a pathname -- Blender.Get('filename') if 'path' isn't given;\n\
|
||||
(ext = \"\") - string: the extension to append.\n\
|
||||
(strip = 0) - int: strip dirname from 'path' if given and non-zero.\n\
|
||||
Ex: makename('/path/to/file/myfile.foo','-01.abc') returns 'myfile-01.abc'\n\
|
||||
Ex: makename(ext='.txt') returns 'untitled.txt' if Blender.Get('filename')\n\
|
||||
returns a path to the file 'untitled.blend'";
|
||||
|
||||
static char M_sys_time_doc[] =
|
||||
"() - Return a float representing time elapsed in seconds.\n\
|
||||
Each successive call is garanteed to return values greater than or\n\
|
||||
equal to the previous call.";
|
||||
|
||||
static char M_sys_sleep_doc[] =
|
||||
"(milliseconds = 10) - Sleep for the specified time.\n\
|
||||
(milliseconds = 10) - the amount of time in milliseconds to sleep.\n\
|
||||
This function can be necessary in tight 'get event' loops.";
|
||||
|
||||
static char M_sys_exists_doc[] =
|
||||
"(path) - Check if the given pathname exists.\n\
|
||||
The return value is as follows:\n\
|
||||
\t 0: path doesn't exist;\n\
|
||||
\t 1: path is an existing filename;\n\
|
||||
\t 2: path is an existing dirname;\n\
|
||||
\t-1: path exists but is neither a regular file nor a dir.";
|
||||
|
||||
static char M_sys_expandpath_doc[] =
|
||||
"(path) - Expand this Blender internal path to a proper file system path.\n\
|
||||
(path) - the string path to convert.\n\n\
|
||||
Note: internally Blender paths can contain two special character sequences:\n\
|
||||
- '//' (at start) for base path directory (the current .blend's dir path);\n\
|
||||
- '#' characters in the filename will be replaced by the frame number.\n\n\
|
||||
This function expands these to their actual content, returning a valid path.\n\
|
||||
If the special chars are not found in the given path, it is simply returned.";
|
||||
|
||||
static char M_sys_cleanpath_doc[] =
|
||||
"(path) - Removes parts of a path that are not needed paths such as '../foo/../bar/' and '//./././'";
|
||||
|
||||
static char M_sys_relpath_doc[] =
|
||||
"(path, start=\"//\") - Returns the path relative to the current blend file or start if spesified";
|
||||
|
||||
/*****************************************************************************/
|
||||
/* Python method structure definition for Blender.sys module: */
|
||||
/*****************************************************************************/
|
||||
struct PyMethodDef M_sys_methods[] = {
|
||||
{"basename", M_sys_basename, METH_O, M_sys_basename_doc},
|
||||
{"dirname", M_sys_dirname, METH_O, M_sys_dirname_doc},
|
||||
{"join", M_sys_join, METH_VARARGS, M_sys_join_doc},
|
||||
{"splitext", M_sys_splitext, METH_O, M_sys_splitext_doc},
|
||||
{"makename", ( PyCFunction ) M_sys_makename,
|
||||
METH_VARARGS | METH_KEYWORDS,
|
||||
M_sys_makename_doc},
|
||||
{"exists", M_sys_exists, METH_O, M_sys_exists_doc},
|
||||
{"sleep", M_sys_sleep, METH_VARARGS, M_sys_sleep_doc},
|
||||
{"time", ( PyCFunction ) M_sys_time, METH_NOARGS, M_sys_time_doc},
|
||||
{"expandpath", M_sys_expandpath, METH_O, M_sys_expandpath_doc},
|
||||
{"cleanpath", M_sys_cleanpath, METH_O, M_sys_cleanpath_doc},
|
||||
{"relpath", M_sys_relpath, METH_VARARGS, M_sys_relpath_doc},
|
||||
{NULL, NULL, 0, NULL}
|
||||
};
|
||||
|
||||
#if PY_VERSION_HEX >= 0x03000000
|
||||
static struct PyModuleDef sys_module = {
|
||||
PyModuleDef_HEAD_INIT,
|
||||
"bpysys",
|
||||
M_sys_doc,
|
||||
-1,/* multiple "initialization" just copies the module dict. */
|
||||
M_sys_methods,
|
||||
NULL, NULL, NULL, NULL
|
||||
};
|
||||
#endif
|
||||
|
||||
/* Module Functions */
|
||||
|
||||
PyObject *BPY_sys_module( void )
|
||||
{
|
||||
PyObject *submodule, *dict;
|
||||
|
||||
#if PY_VERSION_HEX >= 0x03000000
|
||||
submodule= PyModule_Create(&sys_module);
|
||||
#else /* Py2.x */
|
||||
submodule= Py_InitModule3( "bpysys", M_sys_methods, M_sys_doc );
|
||||
#endif
|
||||
|
||||
dict = PyModule_GetDict( submodule );
|
||||
|
||||
/* EXPP_dict_set_item_str( dict, "dirsep", PyString_FromString(DIRSEP_STR) ); */
|
||||
/* EXPP_dict_set_item_str( dict, "sep", PyString_FromString(DIRSEP_STR) ); */
|
||||
|
||||
return submodule;
|
||||
}
|
||||
|
||||
static PyObject *M_sys_basename( PyObject * self, PyObject * value )
|
||||
{
|
||||
char *name = _PyUnicode_AsString(value);
|
||||
char *p, basename[FILE_MAXDIR + FILE_MAXFILE];
|
||||
int n, len;
|
||||
|
||||
if( !name ) {
|
||||
return PyErr_Format( PyExc_TypeError, "expected string argument" );
|
||||
}
|
||||
|
||||
len = strlen( name );
|
||||
|
||||
#ifdef WIN32
|
||||
p = MAX2(strrchr( name, '/' ), strrchr( name, '\\' ));
|
||||
#else
|
||||
p = strrchr( name, DIRSEP );
|
||||
#endif
|
||||
|
||||
if( p ) {
|
||||
n = name + len - p - 1; /* - 1 because we don't want the sep */
|
||||
|
||||
if( n > FILE_MAXDIR + FILE_MAXFILE ) {
|
||||
return PyErr_Format( PyExc_RuntimeError, "path too long" );
|
||||
}
|
||||
|
||||
BLI_strncpy( basename, p + 1, n + 1 );
|
||||
return PyUnicode_FromString( basename );
|
||||
}
|
||||
|
||||
return PyUnicode_FromString( name );
|
||||
}
|
||||
|
||||
static PyObject *M_sys_dirname( PyObject * self, PyObject * value )
|
||||
{
|
||||
char *name = _PyUnicode_AsString(value);
|
||||
char *p, dirname[FILE_MAXDIR + FILE_MAXFILE];
|
||||
int n;
|
||||
|
||||
if( !name )
|
||||
return PyErr_Format( PyExc_TypeError, "expected string argument" );
|
||||
|
||||
#ifdef WIN32
|
||||
p = MAX2(strrchr( name, '/' ), strrchr( name, '\\' ));
|
||||
#else
|
||||
p = strrchr( name, DIRSEP );
|
||||
#endif
|
||||
|
||||
if( p ) {
|
||||
n = p - name;
|
||||
|
||||
if( n > FILE_MAXDIR + FILE_MAXFILE )
|
||||
return PyErr_Format( PyExc_RuntimeError, "path too long" );
|
||||
|
||||
BLI_strncpy( dirname, name, n + 1 );
|
||||
return PyUnicode_FromString( dirname );
|
||||
}
|
||||
|
||||
return PyUnicode_FromString( "." );
|
||||
}
|
||||
|
||||
static PyObject *M_sys_join( PyObject * self, PyObject * args )
|
||||
{
|
||||
char *name = NULL, *path = NULL;
|
||||
char filename[FILE_MAXDIR + FILE_MAXFILE];
|
||||
int pathlen = 0, namelen = 0;
|
||||
|
||||
if( !PyArg_ParseTuple( args, "ss:Blender.sys.join", &path, &name ) )
|
||||
return NULL;
|
||||
|
||||
pathlen = strlen( path ) + 1;
|
||||
namelen = strlen( name ) + 1; /* + 1 to account for '\0' for BLI_strncpy */
|
||||
|
||||
if( pathlen + namelen > FILE_MAXDIR + FILE_MAXFILE - 1 )
|
||||
return PyErr_Format( PyExc_RuntimeError, "filename is too long." );
|
||||
|
||||
BLI_strncpy( filename, path, pathlen );
|
||||
|
||||
if( filename[pathlen - 2] != DIRSEP ) {
|
||||
filename[pathlen - 1] = DIRSEP;
|
||||
pathlen += 1;
|
||||
}
|
||||
|
||||
BLI_strncpy( filename + pathlen - 1, name, namelen );
|
||||
|
||||
return PyUnicode_FromString( filename );
|
||||
}
|
||||
|
||||
static PyObject *M_sys_splitext( PyObject * self, PyObject * value )
|
||||
{
|
||||
char *name = _PyUnicode_AsString(value);
|
||||
char *dot, *p, path[FILE_MAXDIR + FILE_MAXFILE], ext[FILE_MAXDIR + FILE_MAXFILE];
|
||||
int n, len;
|
||||
|
||||
if( !name )
|
||||
return PyErr_Format( PyExc_TypeError, "expected string argument" );
|
||||
|
||||
len = strlen( name );
|
||||
dot = strrchr( name, '.' );
|
||||
|
||||
if( !dot )
|
||||
return Py_BuildValue( "ss", name, "" );
|
||||
|
||||
p = strrchr( name, DIRSEP );
|
||||
|
||||
if( p ) {
|
||||
if( p > dot )
|
||||
return Py_BuildValue( "ss", name, "" );
|
||||
}
|
||||
|
||||
n = name + len - dot;
|
||||
|
||||
/* loong extensions are supported -- foolish, but Python's os.path.splitext
|
||||
* supports them, so ... */
|
||||
|
||||
if( n >= FILE_MAXDIR + FILE_MAXFILE || ( len - n ) >= FILE_MAXDIR + FILE_MAXFILE )
|
||||
return PyErr_Format( PyExc_RuntimeError, "path too long" );
|
||||
|
||||
BLI_strncpy( ext, dot, n + 1 );
|
||||
BLI_strncpy( path, name, dot - name + 1 );
|
||||
|
||||
return Py_BuildValue( "ss", path, ext );
|
||||
}
|
||||
|
||||
static PyObject *M_sys_makename( PyObject * self, PyObject * args,
|
||||
PyObject * kw )
|
||||
{
|
||||
char *path = G.sce, *ext = NULL;
|
||||
int strip = 0;
|
||||
static char *kwlist[] = { "path", "ext", "strip", NULL };
|
||||
char *dot = NULL, *p = NULL, basename[FILE_MAXDIR + FILE_MAXFILE];
|
||||
int n, len, lenext = 0;
|
||||
|
||||
if( !PyArg_ParseTupleAndKeywords( args, kw, "|ssi:Blender.sys.makename", kwlist, &path, &ext, &strip ) )
|
||||
return NULL;
|
||||
|
||||
len = strlen( path ) + 1; /* + 1 to consider ending '\0' */
|
||||
if( ext )
|
||||
lenext = strlen( ext ) + 1;
|
||||
|
||||
if( ( len + lenext ) > FILE_MAXDIR + FILE_MAXFILE )
|
||||
return PyErr_Format( PyExc_RuntimeError, "path too long" );
|
||||
|
||||
p = strrchr( path, DIRSEP );
|
||||
|
||||
if( p && strip ) {
|
||||
n = path + len - p;
|
||||
BLI_strncpy( basename, p + 1, n ); /* + 1 to skip the sep */
|
||||
} else
|
||||
BLI_strncpy( basename, path, len );
|
||||
|
||||
dot = strrchr( basename, '.' );
|
||||
|
||||
/* now the extension: always remove the one in basename */
|
||||
if( dot || ext ) {
|
||||
if( !ext )
|
||||
basename[dot - basename] = '\0';
|
||||
else { /* if user gave an ext, append it */
|
||||
|
||||
if( dot )
|
||||
n = dot - basename;
|
||||
else
|
||||
n = strlen( basename );
|
||||
|
||||
BLI_strncpy( basename + n, ext, lenext );
|
||||
}
|
||||
}
|
||||
|
||||
return PyUnicode_FromString( basename );
|
||||
}
|
||||
|
||||
static PyObject *M_sys_time( PyObject * self )
|
||||
{
|
||||
return PyFloat_FromDouble( PIL_check_seconds_timer( ) );
|
||||
}
|
||||
|
||||
static PyObject *M_sys_sleep( PyObject * self, PyObject * args )
|
||||
{
|
||||
int millisecs = 10;
|
||||
|
||||
if( !PyArg_ParseTuple( args, "|i:Blender.sys.sleep", &millisecs ) )
|
||||
return NULL;
|
||||
|
||||
PIL_sleep_ms( millisecs );
|
||||
|
||||
Py_RETURN_NONE;
|
||||
}
|
||||
|
||||
static PyObject *M_sys_exists( PyObject * self, PyObject * value )
|
||||
{
|
||||
char *fname = _PyUnicode_AsString(value);
|
||||
|
||||
int mode = 0, i = -1;
|
||||
|
||||
if( !fname )
|
||||
return PyErr_Format( PyExc_TypeError, "expected string (pathname) argument" );
|
||||
|
||||
mode = BLI_exist(fname);
|
||||
|
||||
if( mode == 0 )
|
||||
i = 0;
|
||||
else if( S_ISREG( mode ) )
|
||||
i = 1;
|
||||
else if( S_ISDIR( mode ) )
|
||||
i = 2;
|
||||
/* i stays as -1 if path exists but is neither a regular file nor a dir */
|
||||
|
||||
return PyLong_FromLong(i);
|
||||
}
|
||||
|
||||
static PyObject *M_sys_expandpath( PyObject * self, PyObject * value )
|
||||
{
|
||||
char *path = _PyUnicode_AsString(value);
|
||||
char expanded[FILE_MAXDIR + FILE_MAXFILE];
|
||||
bContext *C = BPy_GetContext();
|
||||
Scene *scene = CTX_data_scene(C);
|
||||
|
||||
if (!path)
|
||||
return PyErr_Format( PyExc_TypeError, "expected string argument" );
|
||||
|
||||
BLI_strncpy(expanded, path, FILE_MAXDIR + FILE_MAXFILE);
|
||||
BLI_convertstringcode(expanded, G.sce);
|
||||
BLI_convertstringframe(expanded, scene->r.cfra);
|
||||
|
||||
return PyUnicode_FromString(expanded);
|
||||
}
|
||||
|
||||
static PyObject *M_sys_cleanpath( PyObject * self, PyObject * value )
|
||||
{
|
||||
char *path = _PyUnicode_AsString(value);
|
||||
char cleaned[FILE_MAXDIR + FILE_MAXFILE];
|
||||
int trailing_slash = 0, last;
|
||||
if (!path)
|
||||
return PyErr_Format( PyExc_TypeError, "expected string argument" );
|
||||
last = strlen(path)-1;
|
||||
if ((last >= 0) && ((path[last]=='/') || (path[last]=='\\'))) {
|
||||
trailing_slash = 1;
|
||||
}
|
||||
BLI_strncpy(cleaned, path, FILE_MAXDIR + FILE_MAXFILE);
|
||||
BLI_cleanup_file(NULL, cleaned);
|
||||
|
||||
if (trailing_slash) {
|
||||
BLI_add_slash(cleaned);
|
||||
}
|
||||
|
||||
return PyUnicode_FromString(cleaned);
|
||||
}
|
||||
|
||||
static PyObject *M_sys_relpath( PyObject * self, PyObject * args )
|
||||
{
|
||||
char *base = G.sce;
|
||||
char *path;
|
||||
char relpath[FILE_MAXDIR + FILE_MAXFILE];
|
||||
|
||||
if( !PyArg_ParseTuple( args, "s|s:Blender.sys.relpath", &path, &base ) )
|
||||
return NULL;
|
||||
|
||||
strncpy(relpath, path, sizeof(relpath));
|
||||
BLI_makestringcode(base, relpath);
|
||||
|
||||
return PyUnicode_FromString(relpath);
|
||||
}
|
||||
|
||||
#if 0
|
||||
|
||||
static PyObject *bpy_sys_get_blender_version()
|
||||
{
|
||||
return PyUnicode_FromString(G.version);
|
||||
}
|
||||
|
||||
#endif
|
||||
41
source/blender/python/intern/bpy_sys.h
Normal file
41
source/blender/python/intern/bpy_sys.h
Normal file
@@ -0,0 +1,41 @@
|
||||
/*
|
||||
* $Id: Sys.h 14444 2008-04-16 22:40:48Z hos $
|
||||
*
|
||||
* ***** BEGIN GPL 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.
|
||||
*
|
||||
* 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 LICENSE BLOCK *****
|
||||
*/
|
||||
|
||||
#ifndef BPY_SYS_H
|
||||
|
||||
/* #include <Python.h> */
|
||||
|
||||
/* PyObject *sys_Init( void ); */
|
||||
|
||||
#include <Python.h>
|
||||
|
||||
PyObject *BPY_sys_module( void );
|
||||
|
||||
|
||||
#endif /* BPY_SYS_H */
|
||||
Reference in New Issue
Block a user