io_import_ripdump.py

bl_info = {
    "name": "Import Ninja Ripper/Dx Ripper RIPDUMP",
    "category": "Import-Export",
    "author": "Chris Barrett",
    "version": (2, 0, 1),
    "blender": (2, 74, 0),
    "location": "File > Import > Ninja Ripper RipDump",
    "description": "Imports one directory's frameNNN.mesh.txt RIPDUMP 1.1 contents. "
                   "One object is created per .mesh.txt.",
    "warning": "",
}

# Using blender/scripts/addons/io_import_images_as_planes.py for reference.


import bpy
#import mathutils
#from mathutils import Vector

import re
import glob
import os


def create_new_object(vert_coords, faces, object_name='RippedMesh'):
    edges = []

    mesh = bpy.data.meshes.new(name=object_name)
    mesh.from_pydata(vert_coords, edges, faces)
    #mesh.validate(verbose=True)
    mesh.update(calc_edges=True)

    obj = bpy.data.objects.new(object_name, mesh)
    obj.location = bpy.context.scene.cursor_location
    bpy.context.scene.objects.link(obj)
    bpy.context.scene.objects.active = obj
    obj.select = True


def import_ninja_ripdump(filename, object_name):
    f = open(filename, mode='rt')
    l = f.readline()
    if 'RIPDUMP 1.1' not in l:
        print('Expected first line to contain "RIPDUMP 1.1", got', l, 'instead.')
        return

    vert_pattern = re.compile('([0-9]{6}):.* ([-0-9.]+) .* ([-0-9.]+) .* ([-0-9.]+) .* ([-0-9.]+) .* ([-0-9.]+) .* ([-0-9.]+) .* ([-0-9.]+) .* ([-0-9.]+) .* ([-0-9.]+) .* ([-0-9.]+)$')
    face_pattern = re.compile('([0-9]{6}):.* (\d+) (\d+) (\d+)$')

    vert_coords = []
    #vert_norms = []
    #vert_tex_coords = []
    faces = []

    for l in f.readlines():
        # Try to read vertex
        m = vert_pattern.match(l)
        if m:
            vert_coords.append((
                float(m.group(2)),
                float(m.group(3)),
                float(m.group(4))
            ))
            '''
            vert_norms.append((
                float(m.group(5)),
                float(m.group(6)),
                float(m.group(7))
            ))
            '''
            '''
            vert_tex_coords.append((
                float(m.group(8)),
                float(m.group(9))
            ))
            '''
            continue
        # Try to read face
        m = face_pattern.match(l)
        if m:
            faces.append((
                int(m.group(2)),
                int(m.group(3)),
                int(m.group(4))
            ))
            continue
    # end for each line in file

    create_new_object(vert_coords, faces, object_name)
    return


#==============================================================================
# Blender Operator class
#==============================================================================

class IMPORT_OT_ripdump(bpy.types.Operator):
    """Create objects from Nina Ripper / DX Ripper ripdumps on disk"""
    bl_idname = 'import.ninja_ripdump'
    bl_label = 'Import Ripdumps as Objects'
    bl_options = {'REGISTER', 'UNDO'}

    # -----------
    # Properties assigned by the file selection window.
    #filepath = None
    #filename = None
    directory = bpy.props.StringProperty(maxlen=1024, subtype='FILE_PATH', options={'HIDDEN', 'SKIP_SAVE'})
    files = bpy.props.CollectionProperty(type=bpy.types.OperatorFileListElement, options={'HIDDEN', 'SKIP_SAVE'})

    # invoke is called when the user picks our Import menu entry.
    def invoke(self, context, event):
        context.window_manager.fileselect_add(self)
        return {'RUNNING_MODAL'}

    # execute is called when the user is done using the modal file-select window.
    def execute(self, context):
        dir = self.directory
        for file in self.files:
            filestr = str(file.name)
            if filestr.startswith("mesh") and filestr.endswith(".rip.txt"):
                import_ninja_ripdump(
                    filename=dir + filestr,
                    object_name='rip' + filestr[4:-8]
                )
            else:
                print('Ignored non-"meshNNNN.rip.txt" file', filestr)
        return {'FINISHED'}


#==============================================================================
# Register plugin with Blender
#==============================================================================

def import_ripdump_button(self, context):
    self.layout.operator(IMPORT_OT_ripdump.bl_idname,
                         text="Ninja/DX Ripdump")

def register():
    bpy.utils.register_module(__name__)
    bpy.types.INFO_MT_file_import.append(import_ripdump_button)

def unregister():
    bpy.utils.unregister_module(__name__)
    bpy.types.INFO_MT_file_import.remove(import_ripdump_button)

if __name__ == '__main__':
    register()