blender/io_mspgl/export_animation.py

   1 import math
   2 import os
   3
   4 class AnimationExporter:
   5         def __init__(self):
   6                 self.export_all = False
   7                 self.collection = True
   8                 self.looping_threshold = 0.001
   9
  10         def export_to_file(self, context, out_fn):
  11                 if self.export_all:
  12                         actions = []
  13                         for o in context.selected_objects:
  14                                 if not o.animation_data:
  15                                         continue
  16
  17                                 for t in o.animation_data.nla_tracks:
  18                                         for s in t.strips:
  19                                                 if s.action and s.action not in actions:
  20                                                         actions.append(s.action)
  21
  22                         if not actions:
  23                                 raise Exception("No actions found")
  24
  25                         resources = {}
  26                         for a in actions:
  27                                 resources[a.name+".anim"] = self.export_animation(context, a)
  28
  29                         path, base = os.path.split(out_fn)
  30                         base, ext = os.path.splitext(base)
  31
  32                         if self.collection:
  33                                 from .datafile import Resource
  34                                 dummy = Resource("dummy", "dummy")
  35                                 dummy.references = list(sorted(resources.values(), key=lambda r: r.name))
  36                                 dummy.write_collection(os.path.join(path, base+".mdc"), exclude_self=True)
  37                         else:
  38                                 for r in resources.values():
  39                                         r.write_to_file(os.path.join(path, r.name))
  40                 else:
  41                         anim_data = context.active_object.animation_data
  42                         if not anim_data:
  43                                 raise Exception("Object {} has no animation data".format(context.active_object.name))
  44                         if not anim_data.action:
  45                                 raise Exception("Object {} has no active action".format(context.active_object.name))
  46
  47                         resource = self.export_animation(context, anim_data.action)
  48
  49                         resource.write_to_file(out_fn)
  50
  51         def export_animation(self, context, action):
  52                 from .animation import create_animation_from_action
  53                 anim = create_animation_from_action(context, action, looping_threshold=self.looping_threshold)
  54
  55                 from .datafile import Resource, Statement
  56                 resource = Resource(action.name+".anim", "animation")
  57
  58                 components = [(0, "location", "position"), (1, "rotation_euler", "euler"), (2, "scale", "scale")]
  59                 coords = "xyz"
  60                 prev_time = 0.0
  61                 for k in anim.keyframes:
  62                         if k.time>prev_time:
  63                                 resource.statements.append(Statement("interval", k.time-prev_time))
  64                                 prev_time = k.time
  65
  66                         st = Statement("control_keyframe" if k.control else "keyframe")
  67
  68                         transform = [0.0]*9
  69                         mask = 0
  70                         for c, i in k.curves:
  71                                 for j, dp, kw in components:
  72                                         if c.data_path==dp:
  73                                                 transform[j*3+c.array_index] = c.knots[i][1]
  74                                                 mask |= 1<<(j*3+c.array_index)
  75                                                 break
  76
  77                         if mask:
  78                                 ss = Statement("transform")
  79
  80                                 for i, dp, kw in components:
  81                                         v = transform[i*3:i*3+3]
  82                                         if i==1:
  83                                                 v = [c*180/math.pi for c in v]
  84
  85                                         m = 7<<(i*3)
  86                                         if (mask&m)==m:
  87                                                 ss.sub.append(Statement(kw, *v))
  88                                         else:
  89                                                 m &= m>>2
  90                                                 for j in range(3):
  91                                                         if mask&(m<<j):
  92                                                                 ss.sub.append(Statement("{}_{}".format(kw, coords[j]), v[j]))
  93
  94                                 st.sub.append(ss)
  95
  96                         resource.statements.append(st)
  97
  98                 resource.statements.append(Statement("looping", anim.looping))
  99
 100                 return resource
 101