Skip to content
Snippets Groups Projects
import_fbx.py 75.1 KiB
Newer Older
  • Learn to ignore specific revisions
  • Bastien Montagne's avatar
    Bastien Montagne committed
                kb.vertex_group = elem_name_utf8
    
    
    Bastien Montagne's avatar
    Bastien Montagne committed
    def blen_read_material(fbx_tmpl, fbx_obj, cycles_material_wrap_map, use_cycles):
        elem_name_utf8 = elem_name_ensure_class(fbx_obj, b'Material')
    
    
        ma = bpy.data.materials.new(name=elem_name_utf8)
    
        const_color_white = 1.0, 1.0, 1.0
    
    
        fbx_props = (elem_find_first(fbx_obj, b'Properties70'),
                     elem_find_first(fbx_tmpl, b'Properties70', fbx_elem_nil))
        assert(fbx_props[0] is not None)
    
    
        ma_diff = elem_props_get_color_rgb(fbx_props, b'DiffuseColor', const_color_white)
        ma_spec = elem_props_get_color_rgb(fbx_props, b'SpecularColor', const_color_white)
        ma_alpha = elem_props_get_number(fbx_props, b'Opacity', 1.0)
        ma_spec_intensity = ma.specular_intensity = elem_props_get_number(fbx_props, b'SpecularFactor', 0.25) * 2.0
        ma_spec_hardness = elem_props_get_number(fbx_props, b'Shininess', 9.6)
        ma_refl_factor = elem_props_get_number(fbx_props, b'ReflectionFactor', 0.0)
        ma_refl_color = elem_props_get_color_rgb(fbx_props, b'ReflectionColor', const_color_white)
    
        if use_cycles:
            from . import cycles_shader_compat
            # viewport color
            ma.diffuse_color = ma_diff
    
            ma_wrap = cycles_shader_compat.CyclesShaderWrapper(ma)
            ma_wrap.diffuse_color_set(ma_diff)
            ma_wrap.specular_color_set([c * ma_spec_intensity for c in ma_spec])
    
            ma_wrap.hardness_value_set(((ma_spec_hardness + 3.0) / 5.0) - 0.65)
    
            ma_wrap.alpha_value_set(ma_alpha)
            ma_wrap.reflect_factor_set(ma_refl_factor)
            ma_wrap.reflect_color_set(ma_refl_color)
    
            cycles_material_wrap_map[ma] = ma_wrap
        else:
            # TODO, number BumpFactor isnt used yet
            ma.diffuse_color = ma_diff
            ma.specular_color = ma_spec
            ma.alpha = ma_alpha
            ma.specular_intensity = ma_spec_intensity
            ma.specular_hardness = ma_spec_hardness * 5.10 + 1.0
    
            if ma_refl_factor != 0.0:
                ma.raytrace_mirror.use = True
                ma.raytrace_mirror.reflect_factor = ma_refl_factor
                ma.mirror_color = ma_refl_color
    
        return ma
    
    
    # -------
    # Texture
    
    
    def blen_read_texture(fbx_tmpl, fbx_obj, basedir, image_cache,
    
                          use_image_search):
        import os
        from bpy_extras import image_utils
    
    
    Bastien Montagne's avatar
    Bastien Montagne committed
        elem_name_utf8 = elem_name_ensure_class(fbx_obj, b'Texture')
    
    
        filepath = elem_find_first_string(fbx_obj, b'FileName')
        if os.sep == '/':
            filepath = filepath.replace('\\', '/')
        else:
            filepath = filepath.replace('/', '\\')
    
        image = image_cache.get(filepath)
        if image is not None:
            return image
    
        image = image_utils.load_image(
            filepath,
            dirname=basedir,
            place_holder=True,
            recursive=use_image_search,
            )
    
    
        image_cache[filepath] = image
    
        # name can be ../a/b/c
        image.name = os.path.basename(elem_name_utf8)
    
    def blen_read_camera(fbx_tmpl, fbx_obj, global_scale):
    
        # meters to inches
        M2I = 0.0393700787
    
    Bastien Montagne's avatar
    Bastien Montagne committed
        elem_name_utf8 = elem_name_ensure_class(fbx_obj, b'NodeAttribute')
    
        fbx_props = (elem_find_first(fbx_obj, b'Properties70'),
                     elem_find_first(fbx_tmpl, b'Properties70', fbx_elem_nil))
        assert(fbx_props[0] is not None)
    
    
        camera = bpy.data.cameras.new(name=elem_name_utf8)
    
    
        camera.lens = elem_props_get_number(fbx_props, b'FocalLength', 35.0)
        camera.sensor_width = elem_props_get_number(fbx_props, b'FilmWidth', 32.0 * M2I) / M2I
        camera.sensor_height = elem_props_get_number(fbx_props, b'FilmHeight', 32.0 * M2I) / M2I
    
        filmaspect = camera.sensor_width / camera.sensor_height
        # film offset
        camera.shift_x = elem_props_get_number(fbx_props, b'FilmOffsetX', 0.0) / (M2I * camera.sensor_width)
        camera.shift_y = elem_props_get_number(fbx_props, b'FilmOffsetY', 0.0) / (M2I * camera.sensor_height * filmaspect)
    
    
        camera.clip_start = elem_props_get_number(fbx_props, b'NearPlane', 0.01) * global_scale
        camera.clip_end = elem_props_get_number(fbx_props, b'FarPlane', 100.0) * global_scale
    
    def blen_read_light(fbx_tmpl, fbx_obj, global_scale):
    
    Bastien Montagne's avatar
    Bastien Montagne committed
        elem_name_utf8 = elem_name_ensure_class(fbx_obj, b'NodeAttribute')
    
        fbx_props = (elem_find_first(fbx_obj, b'Properties70'),
                     elem_find_first(fbx_tmpl, b'Properties70', fbx_elem_nil))
    
        # rare
        if fbx_props[0] is None:
            lamp = bpy.data.lamps.new(name=elem_name_utf8, type='POINT')
            return lamp
    
    
        light_type = {
            0: 'POINT',
            1: 'SUN',
            2: 'SPOT'}.get(elem_props_get_enum(fbx_props, b'LightType', 0), 'POINT')
    
        lamp = bpy.data.lamps.new(name=elem_name_utf8, type=light_type)
    
    
        if light_type == 'SPOT':
    
            spot_size = elem_props_get_number(fbx_props, b'OuterAngle', None)
            if spot_size is None:
                # Deprecated.
                spot_size = elem_props_get_number(fbx_props, b'Cone angle', 45.0)
            lamp.spot_size = math.radians(spot_size)
    
            spot_blend = elem_props_get_number(fbx_props, b'InnerAngle', None)
            if spot_blend is None:
                # Deprecated.
                spot_blend = elem_props_get_number(fbx_props, b'HotSpot', 45.0)
            lamp.spot_blend = 1.0 - (spot_blend / spot_size)
    
        lamp.color = elem_props_get_color_rgb(fbx_props, b'Color', (1.0, 1.0, 1.0))
    
        lamp.energy = elem_props_get_number(fbx_props, b'Intensity', 100.0) / 100.0
        lamp.distance = elem_props_get_number(fbx_props, b'DecayStart', 25.0) * global_scale
        lamp.shadow_method = ('RAY_SHADOW' if elem_props_get_bool(fbx_props, b'CastShadow', True) else 'NOSHADOW')
        lamp.shadow_color = elem_props_get_color_rgb(fbx_props, b'ShadowColor', (0.0, 0.0, 0.0))
    
    def is_ascii(filepath, size):
        with open(filepath, 'r', encoding="utf-8") as f:
            try:
                f.read(size)
                return True
            except UnicodeDecodeError:
                pass
    
        return False
    
    
    
    def load(operator, context, filepath="",
    
             use_manual_orientation=False,
             axis_forward='-Z',
             axis_up='Y',
             global_scale=1.0,
    
             use_image_search=False,
             use_alpha_decals=False,
             decal_offset=0.0):
    
        global fbx_elem_nil
        fbx_elem_nil = FBXElem('', (), (), ())
    
    
    Bastien Montagne's avatar
    Bastien Montagne committed
        import os, time
    
        from bpy_extras.io_utils import axis_conversion
        from mathutils import Matrix
    
    
        from .fbx_utils import RIGHT_HAND_AXES, FBX_FRAMERATES
    
    Bastien Montagne's avatar
    Bastien Montagne committed
        start_time = time.process_time()
    
    
        # detect ascii files
        if is_ascii(filepath, 24):
            operator.report({'ERROR'}, "ASCII FBX files are not supported %r" % filepath)
            return {'CANCELLED'}
    
    
        try:
            elem_root, version = parse_fbx.parse(filepath)
        except:
            import traceback
            traceback.print_exc()
    
            operator.report({'ERROR'}, "Couldn't open file %r" % filepath)
            return {'CANCELLED'}
    
        if version < 7100:
            operator.report({'ERROR'}, "Version %r unsupported, must be %r or later" % (version, 7100))
            return {'CANCELLED'}
    
    
        if bpy.ops.object.mode_set.poll():
            bpy.ops.object.mode_set(mode='OBJECT', toggle=False)
    
    
        # deselect all
        if bpy.ops.object.select_all.poll():
            bpy.ops.object.select_all(action='DESELECT')
    
        basedir = os.path.dirname(filepath)
    
        cycles_material_wrap_map = {}
        image_cache = {}
        if not use_cycles:
            texture_cache = {}
    
        # Tables: (FBX_byte_id -> [FBX_data, None or Blender_datablock])
        fbx_table_nodes = {}
    
    
        if use_alpha_decals:
            material_decals = set()
        else:
            material_decals = None
    
    
    
        #### Get some info from GlobalSettings.
    
        fbx_settings = elem_find_first(elem_root, b'GlobalSettings')
        fbx_settings_props = elem_find_first(fbx_settings, b'Properties70')
        if fbx_settings is None or fbx_settings_props is None:
            operator.report({'ERROR'}, "No 'GlobalSettings' found in file %r" % filepath)
            return {'CANCELLED'}
    
        # Compute global matrix and scale.
        if not use_manual_orientation:
            axis_forward = (elem_props_get_integer(fbx_settings_props, b'FrontAxis', 1),
                            elem_props_get_integer(fbx_settings_props, b'FrontAxisSign', 1))
            axis_up = (elem_props_get_integer(fbx_settings_props, b'UpAxis', 2),
                       elem_props_get_integer(fbx_settings_props, b'UpAxisSign', 1))
            axis_coord = (elem_props_get_integer(fbx_settings_props, b'CoordAxis', 0),
                          elem_props_get_integer(fbx_settings_props, b'CoordAxisSign', 1))
    
            axis_key = (axis_up, axis_forward, axis_coord)
            axis_up, axis_forward = {v: k for k, v in RIGHT_HAND_AXES.items()}.get(axis_key, ('Z', 'Y'))
    
            # FBX base unit seems to be the centimeter, while raw Blender Unit is equivalent to the meter...
            global_scale = elem_props_get_number(fbx_settings_props, b'UnitScaleFactor', 100.0) / 100.0
        global_matrix = (Matrix.Scale(global_scale, 4) *
                         axis_conversion(from_forward=axis_forward, from_up=axis_up).to_4x4())
    
        # Compute framerate settings.
        custom_fps = elem_props_get_number(fbx_settings_props, b'CustomFrameRate', 25.0)
        time_mode = elem_props_get_enum(fbx_settings_props, b'TimeMode')
        real_fps = {eid: val for val, eid in FBX_FRAMERATES[1:]}.get(time_mode, custom_fps)
        if real_fps < 0.0:
            real_fps = 25.0
        scene.render.fps = round(real_fps)
        scene.render.fps_base = scene.render.fps / real_fps
    
    
        #### And now, the "real" data.
    
    
        fbx_defs = elem_find_first(elem_root, b'Definitions')  # can be None
    
        fbx_nodes = elem_find_first(elem_root, b'Objects')
        fbx_connections = elem_find_first(elem_root, b'Connections')
    
        if fbx_nodes is None:
    
            operator.report({'ERROR'}, "No 'Objects' found in file %r" % filepath)
            return {'CANCELLED'}
    
        if fbx_connections is None:
    
            operator.report({'ERROR'}, "No 'Connections' found in file %r" % filepath)
            return {'CANCELLED'}
    
        # ----
        # First load property templates
        # Load 'PropertyTemplate' values.
        # Key is a tuple, (ObjectType, FBXNodeType)
        # eg, (b'Texture', b'KFbxFileTexture')
        #     (b'Geometry', b'KFbxMesh')
        fbx_templates = {}
    
        def _():
            if fbx_defs is not None:
                for fbx_def in fbx_defs.elems:
                    if fbx_def.id == b'ObjectType':
                        for fbx_subdef in fbx_def.elems:
                            if fbx_subdef.id == b'PropertyTemplate':
                                assert(fbx_def.props_type == b'S')
                                assert(fbx_subdef.props_type == b'S')
                                # (b'Texture', b'KFbxFileTexture') - eg.
                                key = fbx_def.props[0], fbx_subdef.props[0]
                                fbx_templates[key] = fbx_subdef
        _(); del _
    
        def fbx_template_get(key):
    
            ret = fbx_templates.get(key, fbx_elem_nil)
            if ret is None:
                # Newest FBX (7.4 and above) use no more 'K' in their type names...
                key = (key[0], key[1][1:])
                return fbx_templates.get(key, fbx_elem_nil)
            return ret
    
        def _():
            for fbx_obj in fbx_nodes.elems:
    
                # TODO, investigate what other items after first 3 may be
                assert(fbx_obj.props_type[:3] == b'LSS')
    
                fbx_uuid = elem_uuid(fbx_obj)
                fbx_table_nodes[fbx_uuid] = [fbx_obj, None]
        _(); del _
    
        # ----
    
    Bastien Montagne's avatar
    Bastien Montagne committed
        # http://download.autodesk.com/us/fbx/20112/FBX_SDK_HELP/index.html?url=
        #        WS73099cc142f487551fea285e1221e4f9ff8-7fda.htm,topicNumber=d0e6388
    
    
        fbx_connection_map = {}
        fbx_connection_map_reverse = {}
    
        def _():
            for fbx_link in fbx_connections.elems:
                c_type = fbx_link.props[0]
    
                if fbx_link.props_type[1:3] == b'LL':
                    c_src, c_dst = fbx_link.props[1:3]
                    fbx_connection_map.setdefault(c_src, []).append((c_dst, fbx_link))
                    fbx_connection_map_reverse.setdefault(c_dst, []).append((c_src, fbx_link))
    
            fbx_tmpl = fbx_template_get((b'Geometry', b'KFbxMesh'))
    
    
            for fbx_uuid, fbx_item in fbx_table_nodes.items():
                fbx_obj, blen_data = fbx_item
                if fbx_obj.id != b'Geometry':
                    continue
                if fbx_obj.props[-1] == b'Mesh':
                    assert(blen_data is None)
    
                    fbx_item[1] = blen_read_geom(fbx_tmpl, fbx_obj)
    
        _(); del _
    
        # ----
        # Load material data
        def _():
    
            fbx_tmpl = fbx_template_get((b'Material', b'KFbxSurfacePhong'))
            # b'KFbxSurfaceLambert'
    
    
            for fbx_uuid, fbx_item in fbx_table_nodes.items():
                fbx_obj, blen_data = fbx_item
                if fbx_obj.id != b'Material':
                    continue
                assert(blen_data is None)
    
                fbx_item[1] = blen_read_material(fbx_tmpl, fbx_obj,
    
                                                 cycles_material_wrap_map, use_cycles)
        _(); del _
    
        # ----
        # Load image data
        def _():
    
            fbx_tmpl = fbx_template_get((b'Texture', b'KFbxFileTexture'))
    
    
            for fbx_uuid, fbx_item in fbx_table_nodes.items():
                fbx_obj, blen_data = fbx_item
                if fbx_obj.id != b'Texture':
                    continue
    
                fbx_item[1] = blen_read_texture(fbx_tmpl, fbx_obj, basedir, image_cache,
    
        # ----
        # Load camera data
        def _():
    
            fbx_tmpl = fbx_template_get((b'NodeAttribute', b'KFbxCamera'))
    
    
            for fbx_uuid, fbx_item in fbx_table_nodes.items():
                fbx_obj, blen_data = fbx_item
                if fbx_obj.id != b'NodeAttribute':
                    continue
                if fbx_obj.props[-1] == b'Camera':
                    assert(blen_data is None)
    
                    fbx_item[1] = blen_read_camera(fbx_tmpl, fbx_obj, global_scale)
    
        _(); del _
    
        # ----
        # Load lamp data
        def _():
    
            fbx_tmpl = fbx_template_get((b'NodeAttribute', b'KFbxLight'))
    
    
            for fbx_uuid, fbx_item in fbx_table_nodes.items():
                fbx_obj, blen_data = fbx_item
                if fbx_obj.id != b'NodeAttribute':
                    continue
                if fbx_obj.props[-1] == b'Light':
                    assert(blen_data is None)
    
                    fbx_item[1] = blen_read_light(fbx_tmpl, fbx_obj, global_scale)
    
        # ----
        # Connections
        def connection_filter_ex(fbx_uuid, fbx_id, dct):
            return [(c_found[0], c_found[1], c_type)
                    for (c_uuid, c_type) in dct.get(fbx_uuid, ())
    
                    # 0 is used for the root node, which isnt in fbx_table_nodes
                    for c_found in (() if c_uuid is 0 else (fbx_table_nodes[c_uuid],))
    
                    if (fbx_id is None) or (c_found[0].id == fbx_id)]
    
    
        def connection_filter_forward(fbx_uuid, fbx_id):
            return connection_filter_ex(fbx_uuid, fbx_id, fbx_connection_map)
    
        def connection_filter_reverse(fbx_uuid, fbx_id):
            return connection_filter_ex(fbx_uuid, fbx_id, fbx_connection_map_reverse)
    
    
        # Armatures pre-processing!
        fbx_objects_ignore = set()
        fbx_objects_parent_ignore = set()
        # Arg! In some case, root bone is used as armature as well, in Blender we have to 'insert'
        # an armature object between them, so to handle possible parents of root bones we need a mapping
        # from root bone uuid to Blender's object...
        fbx_bones_to_fake_object = dict()
        armatures = []
        def _():
            nonlocal fbx_objects_ignore, fbx_objects_parent_ignore
            for a_uuid, a_item in fbx_table_nodes.items():
                root_bone = False
                fbx_adata, bl_adata = a_item = fbx_table_nodes.get(a_uuid, (None, None))
                if fbx_adata is None or fbx_adata.id != b'Model':
                    continue
                elif fbx_adata.props[2] != b'Null':
                    if fbx_adata.props[2] not in {b'LimbNode', b'Root'}:
                        continue
                    # In some cases, armatures have no root 'Null' object, we have to consider all root bones
                    # as armatures in this case. :/
                    root_bone = True
                    for p_uuid, p_ctype in fbx_connection_map.get(a_uuid, ()):
                        if p_ctype.props[0] != b'OO':
                            continue
                        fbx_pdata, bl_pdata = p_item = fbx_table_nodes.get(p_uuid, (None, None))
                        if (fbx_pdata and fbx_pdata.id == b'Model' and fbx_pdata.props[2] in {b'LimbNode', b'Root', b'Null'}):
                            # Not a root bone...
                            root_bone = False
                    if not root_bone:
                        continue
                    fbx_bones_to_fake_object[a_uuid] = None
    
                bones = {}
                todo_uuids = set() if root_bone else {a_uuid}
                init_uuids = {a_uuid} if root_bone else set()
                done_uuids = set()
                while todo_uuids or init_uuids:
                    if init_uuids:
                        p_uuid = None
                        uuids = [(uuid, None) for uuid in init_uuids]
                        init_uuids = None
                    else:
                        p_uuid = todo_uuids.pop()
                        uuids = fbx_connection_map_reverse.get(p_uuid, ())
                    # bone -> cluster -> skin -> mesh.
                    # XXX Note: only LimbNode for now (there are also Limb's :/ ).
                    for b_uuid, b_ctype in uuids:
                        if b_ctype and b_ctype.props[0] != b'OO':
                            continue
                        fbx_bdata, bl_bdata = b_item = fbx_table_nodes.get(b_uuid, (None, None))
                        if (fbx_bdata is None or fbx_bdata.id != b'Model' or
                            fbx_bdata.props[2] not in {b'LimbNode', b'Root'}):
                            continue
    
                        # Find bone's size.
                        size = 1.0
                        for t_uuid, t_ctype in fbx_connection_map_reverse.get(b_uuid, ()):
                            if t_ctype.props[0] != b'OO':
                                continue
                            fbx_tdata, _bl_tdata = fbx_table_nodes.get(t_uuid, (None, None))
                            if fbx_tdata is None or fbx_tdata.id != b'NodeAttribute' or fbx_tdata.props[2] != b'LimbNode':
                                continue
                            fbx_props = (elem_find_first(fbx_tdata, b'Properties70'),)
                            size = elem_props_get_number(fbx_props, b'Size', default=size)
                            break  # Only one bone data per bone!
    
                        clusters = []
                        for c_uuid, c_ctype in fbx_connection_map.get(b_uuid, ()):
                            if c_ctype.props[0] != b'OO':
                                continue
                            fbx_cdata, _bl_cdata = fbx_table_nodes.get(c_uuid, (None, None))
                            if fbx_cdata is None or fbx_cdata.id != b'Deformer' or fbx_cdata.props[2] != b'Cluster':
                                continue
                            meshes = set()
                            objects = []
                            for s_uuid, s_ctype in fbx_connection_map.get(c_uuid, ()):
                                if s_ctype.props[0] != b'OO':
                                    continue
                                fbx_sdata, _bl_sdata = fbx_table_nodes.get(s_uuid, (None, None))
                                if fbx_sdata is None or fbx_sdata.id != b'Deformer' or fbx_sdata.props[2] != b'Skin':
                                    continue
                                for m_uuid, m_ctype in fbx_connection_map.get(s_uuid, ()):
                                    if m_ctype.props[0] != b'OO':
                                        continue
                                    fbx_mdata, bl_mdata = fbx_table_nodes.get(m_uuid, (None, None))
                                    if fbx_mdata is None or fbx_mdata.id != b'Geometry' or fbx_mdata.props[2] != b'Mesh':
                                        continue
                                    # Blenmeshes are assumed already created at that time!
                                    assert(isinstance(bl_mdata, bpy.types.Mesh))
                                    # And we have to find all objects using this mesh!
                                    for o_uuid, o_ctype in fbx_connection_map.get(m_uuid, ()):
                                        if o_ctype.props[0] != b'OO':
                                            continue
                                        fbx_odata, bl_odata = o_item = fbx_table_nodes.get(o_uuid, (None, None))
                                        if fbx_odata is None or fbx_odata.id != b'Model' or fbx_odata.props[2] != b'Mesh':
                                            continue
                                        # bl_odata is still None, objects have not yet been created...
                                        objects.append(o_item)
                                    meshes.add(bl_mdata)
                                # Skin deformers are only here to connect clusters to meshes, for us, nothing else to do.
                            clusters.append((fbx_cdata, meshes, objects))
                        # For now, we assume there is only one cluster & skin per bone (at least for a given armature)!
                        # XXX This is not true, some apps export several clusters (kind of LoD), we only use first one!
                        # assert(len(clusters) <= 1)
                        bones[b_uuid] = (b_item, size, p_uuid if (p_uuid != a_uuid or root_bone) else None, clusters)
                        fbx_objects_parent_ignore.add(b_uuid)
                        done_uuids.add(p_uuid)
                        todo_uuids.add(b_uuid)
                if bones:
                    # in case we have no Null parent, rootbone will be a_item too...
                    armatures.append((a_item, bones))
                    fbx_objects_ignore.add(a_uuid)
            fbx_objects_ignore |= fbx_objects_parent_ignore
            # We need to handle parenting at object-level for rootbones-as-armature case :/
            fbx_objects_parent_ignore -= set(fbx_bones_to_fake_object.keys())
        _(); del _
    
    
            fbx_tmpl = fbx_template_get((b'Model', b'KFbxNode'))
    
    
            # Link objects, keep first, this also creates objects
    
            for fbx_uuid, fbx_item in fbx_table_nodes.items():
    
                if fbx_uuid in fbx_objects_ignore:
                    # armatures and bones, handled separately.
                    continue
    
                fbx_obj, blen_data = fbx_item
    
                if fbx_obj.id != b'Model' or fbx_obj.props[2] in {b'Root', b'LimbNode'}:
    
                # Create empty object or search for object data
                if fbx_obj.props[2] == b'Null':
                    fbx_lnk_item = None
                    ok = True
                else:
                    ok = False
    
                    for (fbx_lnk,
                         fbx_lnk_item,
                         fbx_lnk_type) in connection_filter_reverse(fbx_uuid, None):
    
    
                        if fbx_lnk_type.props[0] != b'OO':
                            continue
                        if not isinstance(fbx_lnk_item, bpy.types.ID):
                            continue
                        if isinstance(fbx_lnk_item, (bpy.types.Material, bpy.types.Image)):
                            continue
    
                        # Need to check why this happens, Bird_Leg.fbx
    
                        # This is basic object parenting, also used by "bones".
    
                        if isinstance(fbx_lnk_item, (bpy.types.Object)):
                            continue
    
                        ok = True
                        break
                if ok:
    
                    # create when linking since we need object data
    
                    obj = blen_read_object(fbx_tmpl, fbx_obj, fbx_lnk_item)
    
                    assert(fbx_item[1] is None)
                    fbx_item[1] = obj
    
    
                    # instance in scene
                    obj_base = scene.objects.link(obj)
                    obj_base.select = True
    
        # Now that we have objects...
    
        # I) We can handle shapes.
        blend_shape_channels = {}  # We do not need Shapes themselves, but keyblocks, for anim.
        def _():
            fbx_tmpl = fbx_template_get((b'Geometry', b'KFbxShape'))
    
            for s_uuid, s_item in fbx_table_nodes.items():
                fbx_sdata, bl_sdata = s_item = fbx_table_nodes.get(s_uuid, (None, None))
                if fbx_sdata is None or fbx_sdata.id != b'Geometry' or fbx_sdata.props[2] != b'Shape':
                    continue
    
                # shape -> blendshapechannel -> blendshape -> mesh.
                for bc_uuid, bc_ctype in fbx_connection_map.get(s_uuid, ()):
                    if bc_ctype.props[0] != b'OO':
                        continue
                    fbx_bcdata, _bl_bcdata = fbx_table_nodes.get(bc_uuid, (None, None))
                    if fbx_bcdata is None or fbx_bcdata.id != b'Deformer' or fbx_bcdata.props[2] != b'BlendShapeChannel':
                        continue
                    meshes = []
                    objects = []
                    for bs_uuid, bs_ctype in fbx_connection_map.get(bc_uuid, ()):
                        if bs_ctype.props[0] != b'OO':
                            continue
                        fbx_bsdata, _bl_bsdata = fbx_table_nodes.get(bs_uuid, (None, None))
                        if fbx_bsdata is None or fbx_bsdata.id != b'Deformer' or fbx_bsdata.props[2] != b'BlendShape':
                            continue
                        for m_uuid, m_ctype in fbx_connection_map.get(bs_uuid, ()):
                            if m_ctype.props[0] != b'OO':
                                continue
                            fbx_mdata, bl_mdata = fbx_table_nodes.get(m_uuid, (None, None))
                            if fbx_mdata is None or fbx_mdata.id != b'Geometry' or fbx_mdata.props[2] != b'Mesh':
                                continue
                            # Blenmeshes are assumed already created at that time!
                            assert(isinstance(bl_mdata, bpy.types.Mesh))
                            # And we have to find all objects using this mesh!
                            objects = []
                            for o_uuid, o_ctype in fbx_connection_map.get(m_uuid, ()):
                                if o_ctype.props[0] != b'OO':
                                    continue
                                fbx_odata, bl_odata = o_item = fbx_table_nodes.get(o_uuid, (None, None))
                                if fbx_odata is None or fbx_odata.id != b'Model' or fbx_odata.props[2] != b'Mesh':
                                    continue
                                # bl_odata is still None, objects have not yet been created...
                                objects.append(o_item)
                            meshes.append((bl_mdata, objects))
                        # BlendShape deformers are only here to connect BlendShapeChannels to meshes, nothing else to do.
    
                    # keyblocks is a list of tuples (mesh, keyblock) matching that shape/blendshapechannel, for animation.
                    keyblocks = blen_read_shape(fbx_tmpl, fbx_sdata, fbx_bcdata, meshes, scene, global_matrix)
                    blend_shape_channels[bc_uuid] = keyblocks
        _(); del _
    
    
        # II) We can finish armatures processing.
    
        def _():
            fbx_tmpl = fbx_template_get((b'Model', b'KFbxNode'))
    
    
            blen_read_armatures(fbx_tmpl, armatures, fbx_bones_to_fake_object, scene, global_matrix, arm_parents)
    
        def _():
    
            # Parent objects, after we created them...
    
            for fbx_uuid, fbx_item in fbx_table_nodes.items():
    
                if fbx_uuid in fbx_objects_parent_ignore:
                    # Ignore bones, but not armatures here!
                    continue
    
                fbx_obj, blen_data = fbx_item
                if fbx_obj.id != b'Model':
                    continue
    
                # Handle rootbone-as-armature case :/
                t_data = fbx_bones_to_fake_object.get(fbx_uuid)
                if t_data is not None:
                    blen_data = t_data
                elif blen_data is None:
    
                    continue  # no object loaded.. ignore
    
    
                for (fbx_lnk,
                     fbx_lnk_item,
                     fbx_lnk_type) in connection_filter_forward(fbx_uuid, b'Model'):
    
    
        _(); del _
    
        def _():
            if global_matrix is not None:
                # Apply global matrix last (after parenting)
                for fbx_uuid, fbx_item in fbx_table_nodes.items():
    
                    if fbx_uuid in fbx_objects_parent_ignore:
                        # Ignore bones, but not armatures here!
                        continue
    
                    fbx_obj, blen_data = fbx_item
                    if fbx_obj.id != b'Model':
                        continue
    
                    # Handle rootbone-as-armature case :/
                    t_data = fbx_bones_to_fake_object.get(fbx_uuid)
                    if t_data is not None:
                        blen_data = t_data
                    elif blen_data is None:
    
                        continue  # no object loaded.. ignore
    
    
                    if blen_data.parent is None:
                        blen_data.matrix_basis = global_matrix * blen_data.matrix_basis
    
    
                for (ob_arm, ob_me) in arm_parents:
                    # Rigged meshes are in global space in FBX...
                    ob_me.matrix_basis = global_matrix * ob_me.matrix_basis
                    # And reverse-apply armature transform, so that it gets valid parented (local) position!
                    ob_me.matrix_parent_inverse = ob_arm.matrix_basis.inverted()
    
        def _():
            # link Material's to Geometry (via Model's)
            for fbx_uuid, fbx_item in fbx_table_nodes.items():
                fbx_obj, blen_data = fbx_item
                if fbx_obj.id != b'Geometry':
                    continue
    
                mesh = fbx_table_nodes[fbx_uuid][1]
    
    
                # can happen in rare cases
                if mesh is None:
                    continue
    
    
                for (fbx_lnk,
                     fbx_lnk_item,
                     fbx_lnk_type) in connection_filter_forward(fbx_uuid, b'Model'):
    
    
                    # link materials
                    fbx_lnk_uuid = elem_uuid(fbx_lnk)
    
                    for (fbx_lnk_material,
                         material,
                         fbx_lnk_material_type) in connection_filter_reverse(fbx_lnk_uuid, b'Material'):
    
    
                        mesh.materials.append(material)
    
    
                # We have to validate mesh polygons' mat_idx, see T41015!
                # Some FBX seem to have an extra 'default' material which is not defined in FBX file.
    
                if mesh.validate_material_indices():
                    print("WARNING: mesh '%s' had invalid material indices, those were reset to first material" % mesh.name)
    
            fbx_tmpl = fbx_template_get((b'Material', b'KFbxSurfacePhong'))
            # b'KFbxSurfaceLambert'
    
    
            # textures that use this material
            def texture_bumpfac_get(fbx_obj):
    
                assert(fbx_obj.id == b'Material')
    
                fbx_props = (elem_find_first(fbx_obj, b'Properties70'),
                             elem_find_first(fbx_tmpl, b'Properties70', fbx_elem_nil))
                assert(fbx_props[0] is not None)
    
                # (x / 7.142) is only a guess, cycles usable range is (0.0 -> 0.5)
                return elem_props_get_number(fbx_props, b'BumpFactor', 2.5) / 7.142
    
            def texture_mapping_get(fbx_obj):
                assert(fbx_obj.id == b'Texture')
    
                fbx_props = (elem_find_first(fbx_obj, b'Properties70'),
                             elem_find_first(fbx_tmpl, b'Properties70', fbx_elem_nil))
                assert(fbx_props[0] is not None)
                return (elem_props_get_vector_3d(fbx_props, b'Translation', (0.0, 0.0, 0.0)),
                        elem_props_get_vector_3d(fbx_props, b'Rotation', (0.0, 0.0, 0.0)),
                        elem_props_get_vector_3d(fbx_props, b'Scaling', (1.0, 1.0, 1.0)),
    
                        (bool(elem_props_get_enum(fbx_props, b'WrapModeU', 0)),
                         bool(elem_props_get_enum(fbx_props, b'WrapModeV', 0))))
    
            if not use_cycles:
                # Simple function to make a new mtex and set defaults
    
                def material_mtex_new(material, image, tex_map):
    
                    tex = texture_cache.get(image)
                    if tex is None:
                        tex = bpy.data.textures.new(name=image.name, type='IMAGE')
                        tex.image = image
                        texture_cache[image] = tex
    
                    mtex = material.texture_slots.add()
                    mtex.texture = tex
                    mtex.texture_coords = 'UV'
                    mtex.use_map_color_diffuse = False
    
    
                    # No rotation here...
                    mtex.offset[:] = tex_map[0]
                    mtex.scale[:] = tex_map[2]
    
            for fbx_uuid, fbx_item in fbx_table_nodes.items():
                fbx_obj, blen_data = fbx_item
                if fbx_obj.id != b'Material':
                    continue
    
                material = fbx_table_nodes[fbx_uuid][1]
    
                for (fbx_lnk,
                     image,
                     fbx_lnk_type) in connection_filter_reverse(fbx_uuid, b'Texture'):
    
    
                    if use_cycles:
                        if fbx_lnk_type.props[0] == b'OP':
                            lnk_type = fbx_lnk_type.props[3]
    
                            ma_wrap = cycles_material_wrap_map[material]
    
    
                            # tx/rot/scale
                            tex_map = texture_mapping_get(fbx_lnk)
                            if (tex_map[0] == (0.0, 0.0, 0.0) and
    
    Bastien Montagne's avatar
    Bastien Montagne committed
                                    tex_map[1] == (0.0, 0.0, 0.0) and
                                    tex_map[2] == (1.0, 1.0, 1.0) and
                                    tex_map[3] == (False, False)):
    
                                use_mapping = False
                            else:
                                use_mapping = True
                                tex_map_kw = {
                                    "translation": tex_map[0],
    
                                    "rotation": [-i for i in tex_map[1]],
                                    "scale": [((1.0 / i) if i != 0.0 else 1.0) for i in tex_map[2]],
    
                            if lnk_type == b'DiffuseColor':
                                ma_wrap.diffuse_image_set(image)
    
                                if use_mapping:
                                    ma_wrap.diffuse_mapping_set(**tex_map_kw)
    
                            elif lnk_type == b'SpecularColor':
                                ma_wrap.specular_image_set(image)
    
                                if use_mapping:
                                    ma_wrap.specular_mapping_set(**tex_map_kw)
    
                            elif lnk_type == b'ReflectionColor':
                                ma_wrap.reflect_image_set(image)
    
                                if use_mapping:
                                    ma_wrap.reflect_mapping_set(**tex_map_kw)
    
                            elif lnk_type == b'TransparentColor':  # alpha
    
                                ma_wrap.alpha_image_set(image)
    
                                if use_mapping:
                                    ma_wrap.alpha_mapping_set(**tex_map_kw)
    
                                if use_alpha_decals:
                                    material_decals.add(material)
    
                            elif lnk_type == b'DiffuseFactor':
                                pass  # TODO
                            elif lnk_type == b'ShininessExponent':
                                ma_wrap.hardness_image_set(image)
    
                                if use_mapping:
                                    ma_wrap.hardness_mapping_set(**tex_map_kw)
                            elif lnk_type == b'NormalMap' or lnk_type == b'Bump':  # XXX, applications abuse bump!
    
                                ma_wrap.normal_image_set(image)
                                ma_wrap.normal_factor_set(texture_bumpfac_get(fbx_obj))
    
                                if use_mapping:
                                    ma_wrap.normal_mapping_set(**tex_map_kw)
                                """
    
                            elif lnk_type == b'Bump':
                                ma_wrap.bump_image_set(image)
                                ma_wrap.bump_factor_set(texture_bumpfac_get(fbx_obj))
    
                                if use_mapping:
                                    ma_wrap.bump_mapping_set(**tex_map_kw)
                                """
    
                            else:
                                print("WARNING: material link %r ignored" % lnk_type)
    
                            material_images.setdefault(material, {})[lnk_type] = image
    
                    else:
                        if fbx_lnk_type.props[0] == b'OP':
                            lnk_type = fbx_lnk_type.props[3]
    
    
                            # tx/rot/scale (rot is ignored here!).
                            tex_map = texture_mapping_get(fbx_lnk)
    
                            mtex = material_mtex_new(material, image, tex_map)
    
    
                            if lnk_type == b'DiffuseColor':
                                mtex.use_map_color_diffuse = True
                                mtex.blend_type = 'MULTIPLY'
                            elif lnk_type == b'SpecularColor':
                                mtex.use_map_color_spec = True
                                mtex.blend_type = 'MULTIPLY'
                            elif lnk_type == b'ReflectionColor':
                                mtex.use_map_raymir = True
    
                            elif lnk_type == b'TransparentColor':  # alpha
                                material.use_transparency = True
                                material.transparency_method = 'RAYTRACE'
                                material.alpha = 0.0
                                mtex.use_map_alpha = True
                                mtex.alpha_factor = 1.0
                                if use_alpha_decals:
                                    material_decals.add(material)
    
                            elif lnk_type == b'DiffuseFactor':
                                mtex.use_map_diffuse = True
                            elif lnk_type == b'ShininessExponent':
                                mtex.use_map_hardness = True
    
                            elif lnk_type == b'NormalMap' or lnk_type == b'Bump':  # XXX, applications abuse bump!
    
                                mtex.texture.use_normal_map = True  # not ideal!
    
                                mtex.use_map_normal = True
                                mtex.normal_factor = texture_bumpfac_get(fbx_obj)
    
                            elif lnk_type == b'Bump':
                                mtex.use_map_normal = True
                                mtex.normal_factor = texture_bumpfac_get(fbx_obj)
    
                            else:
                                print("WARNING: material link %r ignored" % lnk_type)
    
    
                            material_images.setdefault(material, {})[lnk_type] = image
    
            # Check if the diffuse image has an alpha channel,
            # if so, use the alpha channel.
    
            # Note: this could be made optional since images may have alpha but be entirely opaque
            for fbx_uuid, fbx_item in fbx_table_nodes.items():
                fbx_obj, blen_data = fbx_item
                if fbx_obj.id != b'Material':
                    continue
                material = fbx_table_nodes[fbx_uuid][1]
                image = material_images.get(material, {}).get(b'DiffuseColor')
                # do we have alpha?
                if image and image.depth == 32:
                    if use_alpha_decals:
                        material_decals.add(material)
    
                    if use_cycles:
                        ma_wrap = cycles_material_wrap_map[material]
                        if ma_wrap.node_bsdf_alpha.mute:
                            ma_wrap.alpha_image_set_from_diffuse()
                    else:
                        if not any((True for mtex in material.texture_slots if mtex and mtex.use_map_alpha)):
                            mtex = material_mtex_new(material, image)
    
                            material.use_transparency = True
                            material.transparency_method = 'RAYTRACE'
                            material.alpha = 0.0
                            mtex.use_map_alpha = True
                            mtex.alpha_factor = 1.0
    
    
                # propagate mapping from diffuse to all other channels which have none defined.
                if use_cycles:
                    ma_wrap = cycles_material_wrap_map[material]
                    ma_wrap.mapping_set_from_diffuse()
    
    
        def _():
            # Annoying workaround for cycles having no z-offset
            if material_decals and use_alpha_decals:
                for fbx_uuid, fbx_item in fbx_table_nodes.items():
                    fbx_obj, blen_data = fbx_item
                    if fbx_obj.id != b'Geometry':
                        continue
                    if fbx_obj.props[-1] == b'Mesh':
                        mesh = fbx_item[1]
    
                        if decal_offset != 0.0:
                            for material in mesh.materials:
                                if material in material_decals:
                                    for v in mesh.vertices:
                                        v.co += v.normal * decal_offset
                                    break
    
                        if use_cycles:
                            for obj in (obj for obj in bpy.data.objects if obj.data == mesh):
                                obj.cycles_visibility.shadow = False
                        else:
                            for material in mesh.materials:
                                if material in material_decals:
                                    # recieve but dont cast shadows
                                    material.use_raytrace = False
        _(); del _
    
    
    Bastien Montagne's avatar
    Bastien Montagne committed
        print('Import finished in %.4f sec.' % (time.process_time() - start_time))