export matrix was wrong, cleanup import, more memory efficient and import the rest...
[blender.git] / release / scripts / import_mdd.py
1 #!BPY
2
3  #"""
4  #Name: 'Load MDD to Mesh RVKs'
5  #Blender: 242
6  #Group: 'Animation'
7  #Tooltip: 'baked vertex animation to active mesh object.'
8  #"""
9 __author__ = "Bill L.Nieuwendorp"
10 __bpydoc__ = """\
11 This script Imports Lightwaves MotionDesigner format.
12
13 The .mdd format has become quite a popular Pipeline format<br>
14 for moving animations from package to package.
15 """
16 # mdd importer  
17 #
18 # Warning if the vertex order or vertex count differs from the
19 # origonal model the mdd was Baked out from their will be Strange
20 # behavior
21
22 #
23 #vertex animation to ShapeKeys with ipo  and gives the frame a value of 1.0 
24 #A modifier to read mdd files would be Ideal but thats for another day :)
25 #
26 #Please send any fixes,updates,bugs to Slow67_at_Gmail.com
27 #Bill Niewuendorp
28
29 try:
30         from struct import unpack
31 except:
32         unpack = None
33
34 import Blender
35 from Blender import Mesh, Object, Scene
36 import BPyMessages
37
38 def mdd_import(filepath, ob, PREF_IPONAME, PREF_START_FRAME, PREF_JUMP):
39         
40         print '\n\nimporting mdd "%s"' % filepath
41         
42         Blender.Window.DrawProgressBar (0.0, "Importing mdd ...")
43         Blender.Window.EditMode(0)
44         Blender.Window.WaitCursor(1)
45         
46         file = open(filepath, 'rb')
47         frames, points = unpack(">2i", file.read(8))
48         time = unpack((">%df" % frames), file.read(frames * 4))
49         
50         print '\tpoints:%d frames:%d' % (points,frames)
51
52         scn = Scene.GetCurrent()
53         ctx = scn.getRenderingContext()
54         Blender.Set("curframe", PREF_START_FRAME)
55         me = ob.getData(mesh=1)
56         
57         def UpdateMesh(me,fr):
58                 for v in me.verts:
59                         # 12 is the size of 3 floats
60                         x,y,z= unpack('>3f', file.read(12))
61                         v.co[:] = x,z,y
62                 me.update()
63         
64         Blender.Window.DrawProgressBar (0.4, "4 Importing mdd ...")
65         
66         
67         curfr = ctx.currentFrame()
68         print'\twriting mdd data...'
69         for i in xrange(frames):
70                 Blender.Set("curframe", i+PREF_START_FRAME)
71                 if len(me.verts) > 1 and (curfr >= PREF_START_FRAME) and (curfr <= PREF_START_FRAME+frames):
72                         UpdateMesh(me, i)
73                         ob.insertShapeKey()
74         
75         Blender.Window.DrawProgressBar (0.5, "5 Importing mdd ...")
76         
77         key= me.key
78         
79         # Add the key of its not there
80         if not key:
81                 me.insertKey(1, 'relative')
82                 key= me.key
83         
84         key.ipo = Blender.Ipo.New('Key', PREF_IPONAME)
85         ipo = key.ipo
86         # block = key.getBlocks() # not used.
87         all_keys = ipo.curveConsts
88
89         for i in xrange(PREF_JUMP+1, len(all_keys), PREF_JUMP):
90                 curve = ipo.getCurve(i)
91                 if curve == None:
92                         curve = ipo.addCurve(all_keys[i])
93                 
94                 curve.append((PREF_START_FRAME+i-1,1))
95                 curve.append((PREF_START_FRAME+i- PREF_JUMP -1,0))
96                 curve.append((PREF_START_FRAME+i+ PREF_JUMP-1,0))
97                 curve.setInterpolation('Linear')
98                 curve.recalc()
99         
100         print 'done'
101         Blender.Window.WaitCursor(0)
102         Blender.Window.DrawProgressBar (1.0, '')
103
104
105 def mdd_import_ui(filepath):
106         
107         if BPyMessages.Error_NoFile(filepath):
108                 return
109                 
110         scn= Scene.GetCurrent()
111         ob_act= scn.objects.active
112         
113         if ob_act == None or ob_act.type != 'Mesh':
114                 BPyMessages.Error_NoMeshActive()
115                 return
116         
117         PREF_IPONAME = Blender.Draw.Create(filepath.split('/')[-1].split('\\')[-1].split('.')[0])
118         PREF_START_FRAME = Blender.Draw.Create(1)
119         PREF_JUMP = Blender.Draw.Create(1)
120         
121         block = [\
122         ("Ipo Name: ", PREF_IPONAME, 0, 30, "Ipo name for the new shape key"),\
123         ("Start Frame: ", PREF_START_FRAME, 1, 3000, "Start frame for the animation"),\
124         ("Key Skip: ", PREF_JUMP, 1, 100, "KeyReduction, Skip every Nth Frame")\
125         ]
126
127         if not Blender.Draw.PupBlock("Import MDD", block):
128                 return
129         orig_frame = Blender.Get('curframe')
130         mdd_import(filepath, ob_act, PREF_IPONAME.val, PREF_START_FRAME.val, PREF_JUMP.val)
131         Blender.Set('curframe', orig_frame)
132
133 if __name__ == '__main__':
134         if not unpack:
135                 Draw.PupMenu('Error%t|This script requires a full python install')
136         
137         Blender.Window.FileSelector(mdd_import_ui, 'IMPORT MDD', '*.mdd')