1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
# SPDX-License-Identifier: Apache-2.0
# Copyright 2018-2021 The glTF-Blender-IO authors.
import bpy
def simulate_stash(obj, track_name, action, start_frame=None):
# Simulate stash :
# * add a track
# * add an action on track
# * lock & mute the track
if not obj.animation_data:
obj.animation_data_create()
tracks = obj.animation_data.nla_tracks
new_track = tracks.new(prev=None)
new_track.name = track_name
if start_frame is None:
start_frame = bpy.context.scene.frame_start
_strip = new_track.strips.new(action.name, start_frame, action)
new_track.lock = True
new_track.mute = True
def restore_animation_on_object(obj, anim_name):
if not getattr(obj, 'animation_data', None):
return
for track in obj.animation_data.nla_tracks:
if track.name != anim_name:
continue
if not track.strips:
continue
obj.animation_data.action = track.strips[0].action
return
obj.animation_data.action = None
def make_fcurve(action, co, data_path, index=0, group_name='', interpolation=None):
try:
fcurve = action.fcurves.new(data_path=data_path, index=index, action_group=group_name)
except:
# Some non valid files can have multiple target path
return None
fcurve.keyframe_points.add(len(co) // 2)
fcurve.keyframe_points.foreach_set('co', co)
# Setting interpolation
ipo = {
'CUBICSPLINE': 'BEZIER',
'LINEAR': 'LINEAR',
'STEP': 'CONSTANT',
}[interpolation or 'LINEAR']
ipo = bpy.types.Keyframe.bl_rna.properties['interpolation'].enum_items[ipo].value
fcurve.keyframe_points.foreach_set('interpolation', [ipo] * len(fcurve.keyframe_points))
# For CUBICSPLINE, also set the handle types to AUTO
if interpolation == 'CUBICSPLINE':
ty = bpy.types.Keyframe.bl_rna.properties['handle_left_type'].enum_items['AUTO'].value
fcurve.keyframe_points.foreach_set('handle_left_type', [ty] * len(fcurve.keyframe_points))
fcurve.keyframe_points.foreach_set('handle_right_type', [ty] * len(fcurve.keyframe_points))
fcurve.update() # force updating tangents (this may change when tangent will be managed)
return fcurve
|