Initial commit of glTF 2.0 importer/exporter

Official Khronos Group Blender glTF 2.0 importer and exporter.

glTF specification: https://github.com/KhronosGroup/glTF
The upstream repository can be found here:
https://github.com/KhronosGroup/glTF-Blender-IO

Reviewed By: Bastien, Campbell

Differential Revision: https://developer.blender.org/D3929

1 files changed, 131 insertions, 0 deletions

diff --git a/io_scene_gltf2/blender/exp/gltf2_blender_gather_animation_channels.py b/io_scene_gltf2/blender/exp/gltf2_blender_gather_animation_channels.py
new file mode 100755
index 00000000..808c970d
--- /dev/null
+++ b/io_scene_gltf2/blender/exp/gltf2_blender_gather_animation_channels.py
@@ -0,0 +1,131 @@
+# Copyright 2018 The glTF-Blender-IO authors.
+#
+# 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.
+
+import bpy
+import typing
+
+from ..com.gltf2_blender_data_path import get_target_object_path, get_target_property_name
+from io_scene_gltf2.io.com import gltf2_io
+from io_scene_gltf2.io.com import gltf2_io_debug
+from io_scene_gltf2.blender.exp.gltf2_blender_gather_cache import cached
+from io_scene_gltf2.blender.exp import gltf2_blender_gather_animation_samplers
+from io_scene_gltf2.blender.exp import gltf2_blender_gather_animation_channel_target
+
+
+@cached
+def gather_animation_channels(blender_action: bpy.types.Action,
+                              blender_object: bpy.types.Object,
+                              export_settings
+                              ) -> typing.List[gltf2_io.AnimationChannel]:
+    channels = []
+
+    for channel_group in __get_channel_groups(blender_action, blender_object):
+        channel = __gather_animation_channel(channel_group, blender_object, export_settings)
+        if channel is not None:
+            channels.append(channel)
+
+    return channels
+
+
+def __gather_animation_channel(channels: typing.Tuple[bpy.types.FCurve],
+                               blender_object: bpy.types.Object,
+                               export_settings
+                               ) -> typing.Union[gltf2_io.AnimationChannel, None]:
+    if not __filter_animation_channel(channels, blender_object, export_settings):
+        return None
+
+    return gltf2_io.AnimationChannel(
+        extensions=__gather_extensions(channels, blender_object, export_settings),
+        extras=__gather_extras(channels, blender_object, export_settings),
+        sampler=__gather_sampler(channels, blender_object, export_settings),
+        target=__gather_target(channels, blender_object, export_settings)
+    )
+
+
+def __filter_animation_channel(channels: typing.Tuple[bpy.types.FCurve],
+                               blender_object: bpy.types.Object,
+                               export_settings
+                               ) -> bool:
+    return True
+
+
+def __gather_extensions(channels: typing.Tuple[bpy.types.FCurve],
+                        blender_object: bpy.types.Object,
+                        export_settings
+                        ) -> typing.Any:
+    return None
+
+
+def __gather_extras(channels: typing.Tuple[bpy.types.FCurve],
+                    blender_object: bpy.types.Object,
+                    export_settings
+                    ) -> typing.Any:
+    return None
+
+
+def __gather_sampler(channels: typing.Tuple[bpy.types.FCurve],
+                     blender_object: bpy.types.Object,
+                     export_settings
+                     ) -> gltf2_io.AnimationSampler:
+    return gltf2_blender_gather_animation_samplers.gather_animation_sampler(
+        channels,
+        blender_object,
+        export_settings
+    )
+
+
+def __gather_target(channels: typing.Tuple[bpy.types.FCurve],
+                    blender_object: bpy.types.Object,
+                    export_settings
+                    ) -> gltf2_io.AnimationChannelTarget:
+    return gltf2_blender_gather_animation_channel_target.gather_animation_channel_target(
+        channels, blender_object, export_settings)
+
+
+def __get_channel_groups(blender_action: bpy.types.Action, blender_object: bpy.types.Object):
+    targets = {}
+    for fcurve in blender_action.fcurves:
+        target_property = get_target_property_name(fcurve.data_path)
+        object_path = get_target_object_path(fcurve.data_path)
+
+        # find the object affected by this action
+        if not object_path:
+            target = blender_object
+        else:
+            try:
+                target = blender_object.path_resolve(object_path)
+            except ValueError:
+                # if the object is a mesh and the action target path can not be resolved, we know that this is a morph
+                # animation.
+                if blender_object.type == "MESH":
+                    # if you need the specific shape key for some reason, this is it:
+                    # shape_key = blender_object.data.shape_keys.path_resolve(object_path)
+                    target = blender_object.data.shape_keys
+                else:
+                    gltf2_io_debug.print_console("WARNING", "Can not export animations with target {}".format(object_path))
+                    continue
+
+        # group channels by target object and affected property of the target
+        target_properties = targets.get(target, {})
+        channels = target_properties.get(target_property, [])
+        channels.append(fcurve)
+        target_properties[target_property] = channels
+        targets[target] = target_properties
+
+    groups = []
+    for p in targets.values():
+        groups += list(p.values())
+
+    return map(tuple, groups)
+