run_blender_script.py

Go to the documentation of this file.

#! /usr/bin/env python

# Copyright (C) 2014 Mikael Arguedas
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# run_blender_script.py
# Authors: Mikael Arguedas [mikael.arguedas@gmail.com]

# This script is a launcher for every blender script available in naoqi_tools
# It defines a set of parameter and format them to sned the right bash commands
# to Launch blender with the chosen script and the right arguments

from __future__ import print_function
import os
import argparse
import subprocess

parser = argparse.ArgumentParser(usage='Export meshes and convert them')
parser.add_argument('-s', '--scriptfile', default='io_export_collision.py',
                    help='name of the blender script to Launch')
parser.add_argument('-b', '--blenderdir', default='/usr/bin',
                    help='location of your blender directory')
parser.add_argument('-i', '--inputmeshdir', default=None,
                    help='directory where collada meshes are located')
parser.add_argument('-o', '--outputdir', default=None,
                    help='directory to export the meshes to')
parser.add_argument('-r', '--ratio', default=0.1,
                    help='float value, ratio used to decimate meshes',
                    type=float)
parser.add_argument('-t', '--threshold', default=50,
                    help='integer: minimum number of vertices for decimation',
                    type=int)
parser.add_argument('--scale', default=1,
                    help='scale to resize the collada meshes to', type=float)
parser.add_argument('-n', '--outputfilename', default=None,
                    help='name of the output file (for material file)')
parser.add_argument('-f', '--blenderfile', default=None,
                    help='path of the blender file to process')

args = parser.parse_args()

clean = False


def check_output_dir(out):
    if not os.path.isdir(out):
        print('creating the output folder because it does not exist')
        os.makedirs(out)
    return out

# Create path to script
script_path = subprocess.check_output('rospack find naoqi_tools',
                                      stderr=subprocess.STDOUT,
                                      shell=True)[:-1]
script_path = os.path.join(script_path, 'scripts', 'blender', args.scriptfile)
# Check script existence
if(not os.path.isfile(script_path)):
    print("script doesn't exist\nExiting now")
    exit(1)

# Check if args.outputdir is valid
if args.scriptfile != 'io_export_visual.py':
    if args.outputdir is None:
        print('\nno valid output directory: using ' + str(args.inputmeshdir) +
              ' as destination folder')
        output_dir = args.inputmeshdir
    else:
        output_dir = check_output_dir(args.outputdir)

    # Check existence of the input directory
    if args.inputmeshdir is None or not os.path.isdir(args.inputmeshdir):
        print('Invalid mesh folder provided\nExiting now')
        exit(1)
else:
    if args.outputdir is None:
        output_dir = os.path.dirname(args.blenderfile)
        print('\nno valid output directory: using ' +
              output_dir + ' as destination folder')
    else:
        output_dir = check_output_dir(args.outputdir)

# Set the parameters and bash command for each script
if(args.scriptfile == 'normalize_meshes.py'):
    cmd = (script_path + ' -i ' + args.inputmeshdir + ' -s ' +
           str(args.scale) + ' -o ' + output_dir)

elif(args.scriptfile == 'io_export_collision.py'):
    cmd = (args.blenderdir + '/blender --background -P ' + script_path +
           ' -- ' + args.inputmeshdir + ' ' + str(args.ratio) + ' ' +
           str(args.threshold) + ' ' + output_dir)

elif(args.scriptfile == 'io_export_visual.py'):
    if(args.blenderfile is None or not os.path.isfile(args.blenderfile)):
        print('invalid blender file provided\nExiting now')
        exit(1)
    if(args.outputfilename is None):
        basename = os.path.basename(args.blenderfile)
        print('no name specified for output material file. unsing: ' +
              basename)
    else:
        basename = os.path.basename(args.outputfilename)
    if(basename.rfind('.') != -1):
        basename = basename[0:basename.rfind('.')]
    print('exporting material to ' + basename + '.material')
    cmd = (os.path.join(args.blenderdir, 'blender') + ' ' + args.blenderfile +
           ' -P ' + script_path + ' -- ' + output_dir +
           ' ' + basename)

elif(args.scriptfile == 'io_export_ogre.py'):
    cmd = ((os.path.join(args.blenderdir, 'blender')) + ' -P ' + script_path +
           ' -- ' + args.inputmeshdir + ' ' + output_dir)
    clean = True

os.system(cmd)

# If ogre exporter called
# Remove files left behind by the OGRE exporter
if clean is True:
    file_list = sorted(os.listdir(output_dir))
    for file in file_list:
        if file.endswith('.mesh.xml'):
            print('removing ' + file)
            os.remove(os.path.join(output_dir, file))