Blender_VRML_cleanup_ribbon.py

# For license and additional information about this code, please refer to https://github.com/niaid/3Dmodel_scripts/blob/master/README.md
# This script runs in Blender: https://www.blender.org/

import os, sys, getopt

def usage():
    print ("")
    print ("Blender_VRML_cleanup.py: a Python script to be executed in Blender")
    print ("")
    print ("""Usage: blender --background --factory-startup --python Blender_VRML_cleanup.py -- <WRL_file>""")
    print ("")

try:
    import bpy
except ImportError:
    usage()
    sys.exit(2)

#print(sys.argv)

# Parse the command line arguments
try:
    opts, args = getopt.getopt(sys.argv[6:], "h", [ "help" ] )
except getopt.GetoptError as err:
    print (str(err))
    usage()
    sys.exit(2)

for o, a in opts:
    if o in ("-h", "--help"):
        usage()
        sys.exit()
    else:
        assert False, "unhandled options"

#print(args)

# open the file
if len(args) == 1:
    WRL_file = args[0]
    input_name, ext  = os.path.splitext(WRL_file)
    X3D_file = input_name + "-temp.x3d"
else:
    print("\nError: You must specify a single WRL file")
    usage()
    sys.exit(2)


os.chdir(".")


# CLEAN UP INITIAL VIEW
bpy.ops.object.select_by_type(extend=False, type='MESH')
bpy.ops.object.delete(use_global=False)
bpy.ops.object.select_by_type(extend=False, type='LAMP')
bpy.ops.object.delete(use_global=False)

# IMPORT THE MESH
print("Importing the VRML file\n")
bpy.ops.import_scene.x3d(filepath=WRL_file)

# DELETE THE OTHER IMPORTED OBJECTS
print("Deleting the imported lamps and curves\n")
bpy.ops.object.select_by_type(extend=False, type='LAMP')
bpy.ops.object.delete(use_global=False)
#bpy.ops.object.select_by_type(extend=False, type='CAMERA')
#bpy.ops.object.delete(use_global=False)
bpy.ops.object.select_by_type(extend=False, type='CURVE')
bpy.ops.object.delete(use_global=False)

# Convert meshes with a material to vertex colors, replace the material with the dummy one
# An X3D can have no more than one material per mesh
print("Converting materials to vertex colors\n")
def_mat = bpy.data.materials.new("Dummy")
def_mat.use_vertex_color_paint = True
def_mat.diffuse_color = def_mat.specular_color = [0.8, 0.8, 0.8]
for o in bpy.context.scene.objects:
    if o.type == 'MESH':
        mesh = o.data
        if len(mesh.vertex_colors) == 0 or len(mesh.vertex_colors[0].data) == 0:
            # No vertex colors already
            if len(mesh.materials) > 0:
                color = mesh.materials[0].diffuse_color
            else:
                color = (0.8, 0.8, 0.8)
            if len(mesh.vertex_colors) == 0:
                vc = mesh.vertex_colors.new()
            else:
                vc = mesh.vertex_colors[0] 
            for i in range(0, len(mesh.loops)):
                vc.data[i].color = color
        mesh.materials.clear()
        mesh.materials.append(def_mat)

# JOIN THE SUB-MESHES
print("Joining the sub-meshes\n")
#bpy.ops.object.select_pattern(pattern="Shape*", case_sensitive=False, extend=True)
bpy.ops.object.select_by_type(extend=False, type='MESH')
bpy.context.scene.objects.active = bpy.context.selected_objects[0]
bpy.ops.object.join()

# REMOVE DOUBLED VERTICES AND ADJUST NORMALS
print("Removing doubled vertices")
bpy.ops.object.mode_set(mode='EDIT')
bpy.ops.mesh.remove_doubles()
#bpy.ops.mesh.normals_make_consistent()
bpy.ops.object.mode_set(mode='OBJECT')

# Decimate
print("Decimating to ratio 0.75\n")
bpy.ops.object.modifier_add(type="DECIMATE")
bpy.context.object.modifiers['Decimate'].ratio = .75
bpy.ops.object.modifier_apply(apply_as='DATA', modifier="Decimate")

# REMOVE DOUBLED VERTICES AGAIN
print("Removing doubled vertices")
bpy.ops.object.mode_set(mode='EDIT')
bpy.ops.mesh.remove_doubles()
bpy.ops.object.mode_set(mode='OBJECT')

# CENTER AND COLOR
print("Moving object to center\n")
bpy.ops.object.origin_set(type='ORIGIN_GEOMETRY')
bpy.ops.object.location_clear()
bpy.ops.object.visual_transform_apply()
print("Setting the background and camera\n")
bpy.data.worlds['World'].horizon_color[0]=1
bpy.data.worlds['World'].horizon_color[1]=1
bpy.data.worlds['World'].horizon_color[2]=1
bpy.ops.view3d.camera_to_view_selected()
bpy.data.cameras['Camera'].clip_end = 1000000

# SAVE OUT X3D
print("Exporting cleaned X3D file")
bpy.ops.export_scene.x3d(filepath=X3D_file)

# SAVE .BLEND
print("Saving .blend file")
bpy.ops.wm.save_as_mainfile(filepath=input_name + ".blend", check_existing = False)

# QUIT
print("Quitting Blender")
bpy.ops.wm.quit_blender()
#sys.exit(0)