forked from bartvdbraak/blender
193 lines
4.8 KiB
Python
193 lines
4.8 KiB
Python
#!BPY
|
|
|
|
"""
|
|
Name: 'Clean Animation Curves'
|
|
Blender: 249
|
|
Group: 'Animation'
|
|
Tooltip: 'Remove unused keyframes for ipo curves'
|
|
"""
|
|
|
|
# ***** BEGIN GPL LICENSE BLOCK *****
|
|
#
|
|
# Copyright (C) 2008-2009: Blender Foundation
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; either version 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software Foundation,
|
|
# --------------------------------------------------------------------------
|
|
|
|
import bpy
|
|
from Blender import IpoCurve, Draw, Window
|
|
|
|
def clean_ipos(ipos):
|
|
eul = 0.001
|
|
|
|
def isflat(vec):
|
|
prev_y = vec[0][1]
|
|
mid_y = vec[1][1]
|
|
next_y = vec[2][1]
|
|
|
|
# flat status for prev and next
|
|
return abs(mid_y-prev_y) < eul, abs(mid_y-next_y) < eul
|
|
|
|
|
|
|
|
X=0
|
|
Y=1
|
|
PREV=0
|
|
MID=1
|
|
NEXT=2
|
|
|
|
LEFT = 0
|
|
RIGHT = 1
|
|
|
|
TOT = 0
|
|
TOTBEZ = 0
|
|
# for ipo in bpy.data.ipos:
|
|
for ipo in ipos:
|
|
if ipo.lib:
|
|
continue
|
|
# print ipo
|
|
for icu in ipo:
|
|
interp = icu.interpolation
|
|
extend = icu.extend
|
|
|
|
bezierPoints = icu.bezierPoints
|
|
bezierVecs = [bez.vec for bez in bezierPoints]
|
|
|
|
l = len(bezierPoints)
|
|
|
|
TOTBEZ += l
|
|
|
|
# our aim is to simplify this ipo as much as possible!
|
|
if interp == IpoCurve.InterpTypes.BEZIER or interp == interp == IpoCurve.InterpTypes.LINEAR:
|
|
#print "Not yet supported"
|
|
|
|
if interp == IpoCurve.InterpTypes.BEZIER:
|
|
flats = [isflat(bez) for bez in bezierVecs]
|
|
else:
|
|
# A bit of a waste but fake the locations for these so they will always be flats
|
|
# IS better then too much duplicate code.
|
|
flats = [(True, True)] * l
|
|
for v in bezierVecs:
|
|
v[PREV][Y] = v[NEXT][Y] = v[MID][Y]
|
|
|
|
|
|
# remove middle points
|
|
if l>2:
|
|
done_nothing = False
|
|
|
|
while not done_nothing and len(bezierVecs) > 2:
|
|
done_nothing = True
|
|
i = l-2
|
|
|
|
while i > 0:
|
|
#print i
|
|
#print i, len(bezierVecs)
|
|
if flats[i]==(True,True) and flats[i-1][RIGHT] and flats[i+1][LEFT]:
|
|
|
|
if abs(bezierVecs[i][MID][Y] - bezierVecs[i-1][MID][Y]) < eul and abs(bezierVecs[i][MID][Y] - bezierVecs[i+1][MID][Y]) < eul:
|
|
done_nothing = False
|
|
|
|
del flats[i]
|
|
del bezierVecs[i]
|
|
icu.delBezier(i)
|
|
TOT += 1
|
|
l-=1
|
|
i-=1
|
|
|
|
# remove endpoints
|
|
if extend == IpoCurve.ExtendTypes.CONST and len(bezierVecs) > 1:
|
|
#print l, len(bezierVecs)
|
|
# start
|
|
|
|
while l > 2 and (flats[0][RIGHT] and flats[1][LEFT] and (abs(bezierVecs[0][MID][Y] - bezierVecs[1][MID][Y]) < eul)):
|
|
print "\tremoving 1 point from start of the curve"
|
|
del flats[0]
|
|
del bezierVecs[0]
|
|
icu.delBezier(0)
|
|
TOT += 1
|
|
l-=1
|
|
|
|
|
|
# End
|
|
while l > 2 and flats[-2][RIGHT] and flats[-1][LEFT] and (abs(bezierVecs[-2][MID][Y] - bezierVecs[-1][MID][Y]) < eul):
|
|
print "\tremoving 1 point from end of the curve", l
|
|
del flats[l-1]
|
|
del bezierVecs[l-1]
|
|
icu.delBezier(l-1)
|
|
TOT += 1
|
|
l-=1
|
|
|
|
|
|
|
|
if l==2:
|
|
if isflat( bezierVecs[0] )[RIGHT] and isflat( bezierVecs[1] )[LEFT] and abs(bezierVecs[0][MID][Y] - bezierVecs[1][MID][Y]) < eul:
|
|
# remove the second point
|
|
print "\tremoving 1 point from 2 point bez curve"
|
|
# remove the second point
|
|
del flats[1]
|
|
del bezierVecs[1]
|
|
icu.delBezier(1)
|
|
TOT+=1
|
|
l-=1
|
|
|
|
# Change to linear for faster evaluation
|
|
'''
|
|
if l==1:
|
|
print 'Linear'
|
|
icu.interpolation = IpoCurve.InterpTypes.LINEAR
|
|
'''
|
|
|
|
|
|
|
|
|
|
if interp== IpoCurve.InterpTypes.CONST:
|
|
print "Not yet supported"
|
|
|
|
print 'total', TOT, TOTBEZ
|
|
return TOT, TOTBEZ
|
|
|
|
def main():
|
|
ret = Draw.PupMenu('Clean Selected Objects Ipos%t|Object IPO%x1|Object Action%x2|%l|All IPOs (be careful!)%x3')
|
|
|
|
sce = bpy.data.scenes.active
|
|
ipos = []
|
|
|
|
if ret == 3:
|
|
ipos.extend(list(bpy.data.ipos))
|
|
else:
|
|
for ob in sce.objects.context:
|
|
if ret == 1:
|
|
ipo = ob.ipo
|
|
if ipo:
|
|
ipos.append(ipo)
|
|
|
|
elif ret == 2:
|
|
action = ob.action
|
|
if action:
|
|
ipos.extend([ipo for ipo in action.getAllChannelIpos().values() if ipo])
|
|
|
|
|
|
|
|
if not ipos:
|
|
Draw.PupMenu('Error%t|No ipos found')
|
|
else:
|
|
total_removed, total = clean_ipos(ipos)
|
|
Draw.PupMenu('Done!%t|Removed ' + str(total_removed) + ' of ' + str(total) + ' points')
|
|
|
|
Window.RedrawAll()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|