Skip to content
Snippets Groups Projects
__init__.py 11.8 KiB
Newer Older
  • Learn to ignore specific revisions
  • # ##### 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
    #
    # ##### END GPL LICENSE BLOCK #####
    
    
    Campbell Barton's avatar
    Campbell Barton committed
    # <pep8-80 compliant>
    
    
    bl_info = {
        "name": "Wavefront OBJ format",
        "author": "Campbell Barton",
    
        "blender": (2, 58, 0),
    
        "location": "File > Import-Export",
    
        "description": "Import-Export OBJ, Import OBJ mesh, UV's, "
                       "materials and textures",
    
        "wiki_url": "http://wiki.blender.org/index.php/Extensions:2.6/Py/"
    
                    "Scripts/Import-Export/Wavefront_OBJ",
    
        "tracker_url": "",
        "support": 'OFFICIAL',
        "category": "Import-Export"}
    
    if "bpy" in locals():
        import imp
        if "import_obj" in locals():
            imp.reload(import_obj)
        if "export_obj" in locals():
            imp.reload(export_obj)
    
    
    import bpy
    
    Campbell Barton's avatar
    Campbell Barton committed
    from bpy.props import (BoolProperty,
                           FloatProperty,
                           StringProperty,
                           EnumProperty,
                           )
    
    from bpy_extras.io_utils import (ImportHelper,
                                     ExportHelper,
    
                                     path_reference_mode,
                                     axis_conversion,
                                     )
    
    
    
    class ImportOBJ(bpy.types.Operator, ImportHelper):
    
        """Load a Wavefront OBJ File"""
    
        bl_idname = "import_scene.obj"
        bl_label = "Import OBJ"
    
        bl_options = {'PRESET', 'UNDO'}
    
    Campbell Barton's avatar
    Campbell Barton committed
        filter_glob = StringProperty(
                default="*.obj;*.mtl",
                options={'HIDDEN'},
                )
    
    Campbell Barton's avatar
    Campbell Barton committed
        use_ngons = BoolProperty(
                name="NGons",
    
                description="Import faces with more than 4 verts as ngons",
    
    Campbell Barton's avatar
    Campbell Barton committed
                default=True,
                )
        use_edges = BoolProperty(
                name="Lines",
                description="Import lines and faces with 2 verts as edge",
                default=True,
                )
        use_smooth_groups = BoolProperty(
                name="Smooth Groups",
                description="Surround smooth groups by sharp edges",
                default=True,
                )
    
    Campbell Barton's avatar
    Campbell Barton committed
        use_split_objects = BoolProperty(
                name="Object",
                description="Import OBJ Objects into Blender Objects",
                default=True,
                )
        use_split_groups = BoolProperty(
                name="Group",
                description="Import OBJ Groups into Blender Objects",
                default=True,
                )
    
    Campbell Barton's avatar
    Campbell Barton committed
        use_groups_as_vgroups = BoolProperty(
                name="Poly Groups",
    
                description="Import OBJ groups as vertex groups",
    
    Campbell Barton's avatar
    Campbell Barton committed
                default=False,
                )
    
    Campbell Barton's avatar
    Campbell Barton committed
        use_image_search = BoolProperty(
                name="Image Search",
    
    Bastien Montagne's avatar
    Bastien Montagne committed
                description="Search subdirs for any associated images "
    
    Campbell Barton's avatar
    Campbell Barton committed
                default=True,
                )
    
                name="Split",
                items=(('ON', "Split", "Split geometry, omits unused verts"),
    
    Campbell Barton's avatar
    Campbell Barton committed
                       ('OFF', "Keep Vert Order", "Keep vertex order from file"),
    
    Campbell Barton's avatar
    Campbell Barton committed
        global_clamp_size = FloatProperty(
    
                name="Clamp Size",
                description="Clamp bounds under this value (zero to disable)",
    
    Campbell Barton's avatar
    Campbell Barton committed
                min=0.0, max=1000.0,
                soft_min=0.0, soft_max=1000.0,
                default=0.0,
                )
    
        axis_forward = EnumProperty(
    
                name="Forward",
                items=(('X', "X Forward", ""),
                       ('Y', "Y Forward", ""),
                       ('Z', "Z Forward", ""),
                       ('-X', "-X Forward", ""),
                       ('-Y', "-Y Forward", ""),
                       ('-Z', "-Z Forward", ""),
                       ),
                default='-Z',
                )
    
    
        axis_up = EnumProperty(
    
                name="Up",
                items=(('X', "X Up", ""),
                       ('Y', "Y Up", ""),
                       ('Z', "Z Up", ""),
                       ('-X', "-X Up", ""),
                       ('-Y', "-Y Up", ""),
                       ('-Z', "-Z Up", ""),
                       ),
                default='Y',
                )
    
    
        def execute(self, context):
            # print("Selected: " + context.active_object.name)
            from . import import_obj
    
    
            if self.split_mode == 'OFF':
                self.use_split_objects = False
                self.use_split_groups = False
            else:
                self.use_groups_as_vgroups = False
    
    
    Campbell Barton's avatar
    Campbell Barton committed
            keywords = self.as_keywords(ignore=("axis_forward",
                                                "axis_up",
                                                "filter_glob",
                                                "split_mode",
                                                ))
    
    Campbell Barton's avatar
    Campbell Barton committed
            global_matrix = axis_conversion(from_forward=self.axis_forward,
                                            from_up=self.axis_up,
                                            ).to_4x4()
    
            keywords["global_matrix"] = global_matrix
    
    
            if bpy.data.is_saved and context.user_preferences.filepaths.use_relative_paths:
                import os
                keywords["relpath"] = os.path.dirname((bpy.data.path_resolve("filepath", False).as_bytes()))
    
    
            return import_obj.load(self, context, **keywords)
    
    
        def draw(self, context):
            layout = self.layout
    
            row = layout.row(align=True)
            row.prop(self, "use_ngons")
            row.prop(self, "use_edges")
    
    
    
            box = layout.box()
            row = box.row()
            row.prop(self, "split_mode", expand=True)
    
            row = box.row()
            if self.split_mode == 'ON':
                row.label(text="Split by:")
                row.prop(self, "use_split_objects")
                row.prop(self, "use_split_groups")
            else:
                row.prop(self, "use_groups_as_vgroups")
    
            row = layout.split(percentage=0.67)
            row.prop(self, "global_clamp_size")
    
            layout.prop(self, "axis_forward")
            layout.prop(self, "axis_up")
    
    
            layout.prop(self, "use_image_search")
    
    
    
    class ExportOBJ(bpy.types.Operator, ExportHelper):
    
        """Save a Wavefront OBJ File"""
    
    
        bl_idname = "export_scene.obj"
        bl_label = 'Export OBJ'
        bl_options = {'PRESET'}
    
        filename_ext = ".obj"
    
    Campbell Barton's avatar
    Campbell Barton committed
        filter_glob = StringProperty(
                default="*.obj;*.mtl",
                options={'HIDDEN'},
                )
    
    Campbell Barton's avatar
    Campbell Barton committed
        use_selection = BoolProperty(
                name="Selection Only",
                description="Export selected objects only",
                default=False,
                )
        use_animation = BoolProperty(
                name="Animation",
    
                description="Write out an OBJ for each frame",
    
    Campbell Barton's avatar
    Campbell Barton committed
                default=False,
                )
    
        use_mesh_modifiers = BoolProperty(
    
    Campbell Barton's avatar
    Campbell Barton committed
                name="Apply Modifiers",
                description="Apply modifiers (preview resolution)",
                default=True,
                )
    
    Campbell Barton's avatar
    Campbell Barton committed
        use_edges = BoolProperty(
    
    Campbell Barton's avatar
    Campbell Barton committed
                description="",
                default=True,
                )
    
        use_smooth_groups = BoolProperty(
                name="Smooth Groups",
                description="Write sharp edges as smooth groups",
                default=False,
                )
    
        use_smooth_groups_bitflags = BoolProperty(
                name="Bitflag Smooth Groups",
                description="Same as 'Smooth Groups', but generate smooth groups IDs as bitflags "
                            "(produces at most 32 different smooth groups, usually much less)",
                default=False,
                )
    
    Campbell Barton's avatar
    Campbell Barton committed
        use_normals = BoolProperty(
    
    Campbell Barton's avatar
    Campbell Barton committed
                description="",
                default=False,
                )
        use_uvs = BoolProperty(
    
                name="Include UVs",
                description="Write out the active UV coordinates",
    
    Campbell Barton's avatar
    Campbell Barton committed
                default=True,
                )
        use_materials = BoolProperty(
    
                name="Write Materials",
                description="Write out the MTL file",
    
    Campbell Barton's avatar
    Campbell Barton committed
                default=True,
                )
        use_triangles = BoolProperty(
    
                name="Triangulate Faces",
                description="Convert all faces to triangles",
    
    Campbell Barton's avatar
    Campbell Barton committed
                default=False,
                )
        use_nurbs = BoolProperty(
    
                name="Write Nurbs",
    
                description="Write nurbs curves as OBJ nurbs rather than "
    
    Campbell Barton's avatar
    Campbell Barton committed
                default=False,
                )
    
        use_vertex_groups = BoolProperty(
                name="Polygroups",
                description="",
                default=False,
                )
    
    Campbell Barton's avatar
    Campbell Barton committed
        use_blen_objects = BoolProperty(
                name="Objects as OBJ Objects",
                description="",
                default=True,
                )
        group_by_object = BoolProperty(
                name="Objects as OBJ Groups ",
                description="",
                default=False,
                )
        group_by_material = BoolProperty(
                name="Material Groups",
                description="",
                default=False,
                )
        keep_vertex_order = BoolProperty(
                name="Keep Vertex Order",
                description="",
                default=False,
                )
    
        axis_forward = EnumProperty(
    
                name="Forward",
                items=(('X', "X Forward", ""),
                       ('Y', "Y Forward", ""),
                       ('Z', "Z Forward", ""),
                       ('-X', "-X Forward", ""),
                       ('-Y', "-Y Forward", ""),
                       ('-Z', "-Z Forward", ""),
                       ),
                default='-Z',
                )
    
        axis_up = EnumProperty(
    
                name="Up",
                items=(('X', "X Up", ""),
                       ('Y', "Y Up", ""),
                       ('Z', "Z Up", ""),
                       ('-X', "-X Up", ""),
                       ('-Y', "-Y Up", ""),
                       ('-Z', "-Z Up", ""),
                       ),
                default='Y',
                )
    
        global_scale = FloatProperty(
                name="Scale",
                min=0.01, max=1000.0,
                default=1.0,
                )
    
        def execute(self, context):
            from . import export_obj
    
    
            from mathutils import Matrix
    
    Campbell Barton's avatar
    Campbell Barton committed
            keywords = self.as_keywords(ignore=("axis_forward",
                                                "axis_up",
                                                "global_scale",
                                                "check_existing",
                                                "filter_glob",
                                                ))
    
            global_matrix = (Matrix.Scale(self.global_scale, 4) *
    
    Campbell Barton's avatar
    Campbell Barton committed
                             axis_conversion(to_forward=self.axis_forward,
                                             to_up=self.axis_up,
                                             ).to_4x4())
    
    
            keywords["global_matrix"] = global_matrix
            return export_obj.save(self, context, **keywords)
    
    
    
    def menu_func_import(self, context):
        self.layout.operator(ImportOBJ.bl_idname, text="Wavefront (.obj)")
    
    
    def menu_func_export(self, context):
        self.layout.operator(ExportOBJ.bl_idname, text="Wavefront (.obj)")
    
    
    def register():
    
        bpy.types.INFO_MT_file_import.append(menu_func_import)
        bpy.types.INFO_MT_file_export.append(menu_func_export)
    
    
    def unregister():
    
        bpy.types.INFO_MT_file_import.remove(menu_func_import)
        bpy.types.INFO_MT_file_export.remove(menu_func_export)
    
    if __name__ == "__main__":
        register()