mirror of
https://github.com/Sneed-Group/Poodletooth-iLand
synced 2024-12-26 21:22:27 -06:00
918 lines
40 KiB
Python
918 lines
40 KiB
Python
"""
|
|
Defines Graph Editor
|
|
"""
|
|
import os,wx
|
|
import math
|
|
import cPickle as pickle
|
|
from PaletteTreeCtrl import *
|
|
import ObjectGlobals as OG
|
|
import AnimGlobals as AG
|
|
from wx.lib.embeddedimage import PyEmbeddedImage
|
|
|
|
property = [
|
|
"translateX",
|
|
"translateY",
|
|
"translateZ"
|
|
]
|
|
|
|
#----------------------------------------------------------------------
|
|
ZoomIn = PyEmbeddedImage(
|
|
"iVBORw0KGgoAAAANSUhEUgAAABwAAAAcCAIAAAD9b0jDAAAAA3NCSVQICAjb4U/gAAACgklE"
|
|
"QVRIid3VW0vjQBQA4DlJMziZlCKCSB/9Bf4a3wRtNU6SHyXK6pay2nVbQbasskXFO15SqmwL"
|
|
"LtYnn7Zg20zSyT50wZikXsCHxXk9Z74ZZuacgebtLXrvIb27+PFQjDEADIoqiiLL8htQAMAY"
|
|
"Hx4dPTw8SHEuIaRWqzWbzUQi8SoUAIaGhgqFgn15SSn1Iwmqqh4fH5c2NpLJpO8/xhORzEeR"
|
|
"EJLL5VqtViaT8TxPCBFMoJTu7e1tb28zxpLJpOu6L6D9PS6vrHQ6nWw267puVNzZ2alUKqZh"
|
|
"UE0LivFoX/y0vOxynpmZ4ZxHxZ+Vyu7urmmaqqqGxBhUkiSM8eLiIgDMTE87EVHTtB9bW4cH"
|
|
"B5ZpEkKiYhjt3/XCwoKiKFNTU47jRPf4vVw+PTmxLAtjHCs+QQEAAJaWljRNm5yc7Ha7wQtF"
|
|
"CGGMy+WybduWZSmK4nlerIiCTwoAOOfX19cTExMSgPDDT0iSJNu2x8fHU6nUM+ITVAhBCLFM"
|
|
"83MuV63VKCGhVM65ruv1RqNYLFJKX4UihFzXTafTc7Oz+XzerlZVVQ1GhRCyLJuGcXV1VSqV"
|
|
"QtGBKEKo6zhjY2Pzur66tnZxcRGa2ev1AMA0zXqj8W3wfmPK1HGc0dFRNj//dX397Ows6iKE"
|
|
"DMZ+39wUCgVKabTXxNe+4zgjIyMGY8VS6eT0NOr6vs8Yu7u7+7K6SggJuQMbCud8eHjYYGxz"
|
|
"c3N/f19RlJDb6/V0Xb+/v8/n829ofZzzVCqlz82dn5//abUk6UmyEEIIkc1k2u32r3o9uCq8"
|
|
"+PHJsuz7vud5IfTffABZljnnwejLnb9/grEiQih2vVd9J36kup4fH+w3/S/QvwejQg8ibHgo"
|
|
"AAAAAElFTkSuQmCC")
|
|
|
|
#----------------------------------------------------------------------
|
|
ZoomOut = PyEmbeddedImage(
|
|
"iVBORw0KGgoAAAANSUhEUgAAABwAAAAcCAIAAAD9b0jDAAAAA3NCSVQICAjb4U/gAAACmUlE"
|
|
"QVRIieXVz0sbQRQH8NmdH8smh3jwEIIlB+tNEW1NVps0WxPBv0BQXKQx4N2/xR2xTYQ9iIHc"
|
|
"/A+shyqevHiQVBGLeqgoxM1kNjHTQ8DayWb14KX0Hfe9+fDd4bGr/Ly4AK9d6quL/y6qKIqm"
|
|
"aQHThBBVfT7HnwlFUTjn3/f3MUKKoviMqurh4WG9XkcIvRQVQoRCodPT0+1yWdO0bhchxD2v"
|
|
"WCxyzjHGL0IBAA8PD9biIgDAcRxN06Q39Tzvk2kahmHbNmMswP3rmBCi0WhYlkUI2dzcxBhL"
|
|
"ruu6mUwmnU6v2bbrur1c+dY77sLCQjgcLpVKGKFuN5VKmaZpU1qr1Qghz6OP7vz8fF9f39di"
|
|
"Efq5H6amcrkcpfTu7q7b9d8PIQRjbG5urr+//8vGBoSw200mErOzs3R9/fb2VnLh6uqqrwsA"
|
|
"aLVao6OjV9fX3/b2RkZGJLfZbMbj8UgkUqlU3g4NhcNhIURQ0k612+12u/1ufPxHtcoY6157"
|
|
"z/OGh4fv7++vLi8hhI/Pg9ZY1/Xz83PHcVZWViKRSLPZfNpFCHmeRynNZrNjY2ONRuOx1TOp"
|
|
"rutnZ2eO41iWNTg4KIkYY875mm2/n5iYnp5+KvZMqut6tVrd2tpaWlqKx+OMMUmsM0YpnZqc"
|
|
"NE3TdV3puA+q6/rJycl2ufw5n38zMCClwBjXXdem9GMmk06lukUfNBQKHR8fVyqVwvJyLBaT"
|
|
"REJIrVbr3KNhGL6ijGKMj46OdnZ2CoVCNBqVRAjhr5ubUqmUm5kxksleIpA+fYyxg4ODfD4f"
|
|
"jUY559IohHB3dzebzSYTiQARAKA8/fEJIVRVhRC2Wi3faSEEIUTahKCknbAAgF4iAEBV1WdF"
|
|
"Ge1kCZgO7vZEX6X+c/Q3Qy8zmmCLWqIAAAAASUVORK5CYII=")
|
|
|
|
#----------------------------------------------------------------------
|
|
OneTangent = PyEmbeddedImage(
|
|
"iVBORw0KGgoAAAANSUhEUgAAABwAAAAcCAIAAAD9b0jDAAAAA3NCSVQICAjb4U/gAAACKUlE"
|
|
"QVRIia2Wz2saURDH5/3Yt6aVsE3SomArGO0phxCSU6A2EFgSklP+qPwrXkOQkBCoaG8eQrBQ"
|
|
"KRYVU1tabUij3V1337weAtZI1FX3e9qF2c/OzJv9zpKbRgMGRAhRSsF8oo9uOCeUMk0LDEo1"
|
|
"rVUuF46Pv2SzhDEgJBhoLZer5/OViwu32yWUjnlsvHj/SrluPJ222m3GORMC5ugsGTwoyjkT"
|
|
"4nuxeFetvj06ko4zG/RRjeh5rmVFNjeZrt8UCnxhIQAoAIBSnm0n9vb+1Gq/y2Wm60FAAUAp"
|
|
"UGr18LCRz9vtNuH8iZipoQAopW4Yb3Z2vmazgDjtJIyMlo7zIpUyksnq2RkTIhgoAHiWFdve"
|
|
"RsTmx4IuNP/ZToiUvV7CNNt3natb61b57cKEMIXIOeu+N0uof5IaEl8f7+R3K1SG8sKc0FqF"
|
|
"oPTjCZOhEiDF0Vyk8da3+occC4UCgAKAAhBe73X6Xa/T+VEs8klcv0eKCtD1kgcHP6+v7xsN"
|
|
"OtZzp5hqhUh1PbG/Xz0/l5ZFGAsACgDous+j0cjWVuX0lI428qmdWNr2q/X10NJS/fJyVHNn"
|
|
"sXfPtuO7u39brV+lkgiHmRBDKZOhbepThDHpOJ8zmZ7jLMZiq6YJhPSXxYyLSEkZMgx9ebl8"
|
|
"clLKZO6bTTbgkFN7ZV+e47xcW4tubIQjkWcrKyjl/zpmK/9BlDH0vIfZUogBZAoAKCWhVCk1"
|
|
"tHrnggLAk/9I/wC0eekA1bLbIQAAAABJRU5ErkJggg==")
|
|
|
|
#----------------------------------------------------------------------
|
|
TwoTangents = PyEmbeddedImage(
|
|
"iVBORw0KGgoAAAANSUhEUgAAABwAAAAcCAIAAAD9b0jDAAAAA3NCSVQICAjb4U/gAAACL0lE"
|
|
"QVRIie3UzU8TQRQA8DczO1vatRgh1YhWD/WD+kG91FgOXLhx4+CpiYnE9L8g/gVwJHhp4lUP"
|
|
"pPHgQRMOGiEaDabV1LRAtdjUAG23m3W/Znc8YEo/trRNOGjCO+7M++3MezODdopFOO7Axy6e"
|
|
"oC0oAqAEY4QGyCcE4faVtaII7dQUy7b7dDEheq3GdB0R4o76KH2+kZ158uzxyzcO56iXS0Rx"
|
|
"N5tdnZ9fW1w0FaXZPUQxRl/Le7+q8mruh2paQsem2kLweCq5XGVrq7K5adTruGk+ahx+AeOS"
|
|
"rLz4ktctdtYvPYje0i3Gu22cUr1a3VheHg6FhoPBC9EoBwDO21EOQDH2igIALL39NOrz3r8z"
|
|
"rppWp3jQmQ8LC5enp4NTU0zTmGG0/PJwKgBzHEU3VcN6eHfie7X++ltBEmknSoaG0snkSDg8"
|
|
"Njmp12ptIrieU4dzAEjEIu8KPz8Wy14qNI9SScqvrHCA67Ozlqq6F8f1K3McL6WPYpFUOre9"
|
|
"L3uEv50VfL7S+vpuJjMxN8cMo1HEvlAAMG37nF+KR28+fZ/e/62JBAuiWC8U8qlUJJFAhHDH"
|
|
"6ZZ71LnRLXY1MDJzI5Rc+6wwR5PlTDIZjsd9gYBjuTSwEajneypR4dV2qXxq1F/dG5fLodg9"
|
|
"U9OOThGOHgYAg7FrF8fKKqjng/4rl+xeYl+ozeEMsNsiwmCftnnXQg6EcgABICxyAGAcut2x"
|
|
"wdADt+uFdYv/9OU/Qf9d9A9dhOgRU8MmewAAAABJRU5ErkJggg==")
|
|
|
|
#----------------------------------------------------------------------
|
|
|
|
class GraphEditorWindow(wx.Window):
|
|
"""
|
|
This is the main graph editor window.
|
|
"""
|
|
def __init__(self, parent, windowSize, property, xRange, yRange, curFrame, object):
|
|
wx.Window.__init__(self, parent, size = windowSize, style = wx.SUNKEN_BORDER)
|
|
|
|
self._mainDialog = wx.GetTopLevelParent(self)
|
|
self.w,self.h = self.GetClientSize()
|
|
|
|
self.zoom = float(2)
|
|
self._mouseIn = False
|
|
self._selectRec = False
|
|
self._selectHandler = False
|
|
self._OneTangent = True
|
|
|
|
self.object = object
|
|
self.curFrame = curFrame
|
|
self.property = property
|
|
|
|
self.zeroPos = (float(0), self.h/float(2))
|
|
self.zero = 0
|
|
self.unitWidth = self.w/float(xRange)
|
|
self.unitHeight = self.h/float(yRange)
|
|
|
|
self.generateInfo()
|
|
self.InitBuffer()
|
|
|
|
self.Bind(wx.EVT_PAINT, self.OnPaint)
|
|
self.Bind(wx.EVT_SIZE, self.OnSize)
|
|
self.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
|
|
self.Bind(wx.EVT_LEFT_UP, self.OnLeftUp)
|
|
self.Bind(wx.EVT_MIDDLE_DOWN, self.OnMiddleDown)
|
|
self.Bind(wx.EVT_MIDDLE_UP, self.OnMiddleUp)
|
|
self.Bind(wx.EVT_MOTION, self.OnMotion)
|
|
|
|
def refresh(self):
|
|
self._mouseIn = False
|
|
self._selectRec = False
|
|
self._selectHandler = False
|
|
self.generateInfo()
|
|
|
|
def generateInfo(self):
|
|
#generate the information for animation curve generation
|
|
self.X = []
|
|
self.Y = []
|
|
self.Z = []
|
|
|
|
if self._mainDialog.editor.animMgr.keyFramesInfo != {}:
|
|
self.keyFramesInfo = self._mainDialog.editor.animMgr.keyFramesInfo
|
|
for key in self.keyFramesInfo.keys():
|
|
if key == (self.object[OG.OBJ_UID], 'X'):
|
|
for i in range(len(self.keyFramesInfo[key])):
|
|
item = self.keyFramesInfo[key][i]
|
|
handler = self.generateHandler(item)
|
|
self.X.append([key, i, handler[0], handler[1], handler[2], handler[3], handler[4]])
|
|
if key == (self.object[OG.OBJ_UID], 'Y'):
|
|
for i in range(len(self.keyFramesInfo[key])):
|
|
item = self.keyFramesInfo[key][i]
|
|
handler = self.generateHandler(item)
|
|
self.Y.append([key, i, handler[0], handler[1], handler[2], handler[3], handler[4]])
|
|
if key == (self.object[OG.OBJ_UID], 'Z'):
|
|
for i in range(len(self.keyFramesInfo[key])):
|
|
item = self.keyFramesInfo[key][i]
|
|
handler = self.generateHandler(item)
|
|
self.Z.append([key, i, handler[0], handler[1], handler[2], handler[3], handler[4]])
|
|
|
|
def generateHandler(self, item):
|
|
#generate the position for the control handler
|
|
x1 = self.zeroPos[0] + float(item[AG.FRAME])*self.unitWidth
|
|
y1 = self.zeroPos[1] - float(item[AG.VALUE])*self.unitHeight
|
|
|
|
t1x = item[AG.INSLOPE][0]*self.unitWidth
|
|
t1y = item[AG.INSLOPE][1]*self.unitHeight
|
|
|
|
t2x = item[AG.OUTSLOPE][0]*self.unitWidth
|
|
t2y = item[AG.OUTSLOPE][1]*self.unitHeight
|
|
|
|
tanA = t1y/t1x
|
|
temp1 = float(1)/(tanA**2+1)
|
|
if t1x <0 :
|
|
cosA = -math.sqrt(abs(temp1))
|
|
if t1x >=0:
|
|
cosA = math.sqrt(abs(temp1))
|
|
temp2 = (tanA**2)*temp1
|
|
if t1y <0 :
|
|
sinA = -math.sqrt(abs(temp2))
|
|
if t1y >=0:
|
|
sinA = math.sqrt(abs(temp2))
|
|
|
|
x2 = x1-float(self.unitWidth*self.zoom)*cosA
|
|
y2 = y1+float(self.unitWidth*self.zoom)*sinA
|
|
|
|
tanA = t2y/t2x
|
|
temp1 = float(1)/(tanA**2+1)
|
|
if t2x <0 :
|
|
cosA = -math.sqrt(abs(temp1))
|
|
if t2x >=0:
|
|
cosA = math.sqrt(abs(temp1))
|
|
temp2 = (tanA**2)*temp1
|
|
if t2y <0 :
|
|
sinA = -math.sqrt(abs(temp2))
|
|
if t2y >=0:
|
|
sinA = math.sqrt(abs(temp2))
|
|
|
|
x3 = x1+float(self.unitWidth*self.zoom)*cosA
|
|
y3 = y1-float(self.unitWidth*self.zoom)*sinA
|
|
|
|
return [[(x1,y1),0],[(x2,y2),0],[(x3,y3),0],[t1x,t1y],[t2x,t2y]]
|
|
|
|
def InitBuffer(self):
|
|
self.buffer = wx.EmptyBitmap(self.w, self.h)
|
|
dc = wx.BufferedDC(wx.ClientDC(self), self.buffer)
|
|
self.DrawXCoord(dc)
|
|
self.DrawYCoord(dc)
|
|
self.DrawFrame(dc)
|
|
self.DrawCurve(dc)
|
|
self.DrawSelectRec(dc)
|
|
|
|
def SetGraphEditorData(self, property, curFrame = 1):
|
|
self.curFrame = curFrame
|
|
self.property = property
|
|
|
|
self.InitBuffer()
|
|
|
|
def OnPaint(self, evt):
|
|
dc = wx.BufferedPaintDC(self, self.buffer)
|
|
|
|
def DrawXCoord(self,dc):
|
|
dc.SetBackground(wx.Brush(wx.Colour(200, 200, 200)))
|
|
dc.Clear()
|
|
|
|
dc.SetPen(wx.BLACK_PEN)
|
|
dc.SetBrush(wx.BLACK_BRUSH)
|
|
dc.SetFont(wx.Font(8, wx.SWISS, wx.NORMAL, wx.NORMAL))
|
|
|
|
dc.DrawLine(self.zeroPos[0], float(0), self.zeroPos[0], self.h)
|
|
st = str(self.zero)
|
|
self.tw,self.th = dc.GetTextExtent(st)
|
|
dc.DrawText(st, self.zeroPos[0]+1.0, self.h-self.th-0.5)
|
|
|
|
dc.SetPen(wx.Pen(wx.Colour(150, 150, 150)))
|
|
dc.SetBrush(wx.Brush(wx.Colour(150, 150, 150)))
|
|
|
|
if self.unitWidth >= 25:
|
|
posPos = self.zeroPos[0]+self.unitWidth
|
|
posNum = self.zero + 1
|
|
while posPos <= self.w:
|
|
dc.DrawLine(posPos, float(0), posPos, self.h)
|
|
st = str(posNum)
|
|
self.drawXNumber(dc, st, posPos)
|
|
posPos += self.unitWidth
|
|
posNum += 1
|
|
|
|
negPos = self.zeroPos[0]-self.unitWidth
|
|
negNum = self.zero - 1
|
|
while negPos >= float(0):
|
|
dc.DrawLine(negPos, float(0), negPos, self.h)
|
|
st = str(negNum)
|
|
self.drawXNumber(dc, st, negPos)
|
|
negPos -= self.unitWidth
|
|
posNum -= 1
|
|
|
|
elif self.unitWidth >= 10 and self.unitWidth <= 25:
|
|
posPos = self.zeroPos[0]+self.unitWidth*float(2)
|
|
posNum = self.zero + 2
|
|
while posPos <= self.w:
|
|
dc.DrawLine(posPos, float(0), posPos, self.h)
|
|
st = str(posNum)
|
|
self.drawXNumber(dc, st, posPos)
|
|
posPos += self.unitWidth*float(2)
|
|
posNum += 2
|
|
|
|
negPos = self.zeroPos[0]-self.unitWidth*float(2)
|
|
negNum = self.zero - 2
|
|
while negPos >= float(0):
|
|
dc.DrawLine(negPos, float(0), negPos, self.h)
|
|
st = str(negNum)
|
|
self.drawXNumber(dc, st, negPos)
|
|
negPos -= self.unitWidth*float(2)
|
|
posNum -= 2
|
|
|
|
elif self.unitWidth >= 2 and self.unitWidth <= 10:
|
|
posPos = self.zeroPos[0]+self.unitWidth*float(5)
|
|
posNum = self.zero + 5
|
|
while posPos <= self.w:
|
|
dc.DrawLine(posPos, float(0), posPos, self.h)
|
|
st = str(posNum)
|
|
self.drawXNumber(dc, st, posPos)
|
|
posPos += self.unitWidth*float(5)
|
|
posNum += 5
|
|
|
|
negPos = self.zeroPos[0]-self.unitWidth*float(5)
|
|
negNum = self.zero - 5
|
|
while negPos >= float(0):
|
|
dc.DrawLine(negPos, float(0), negPos, self.h)
|
|
st = str(negNum)
|
|
self.drawXNumber(dc, st, negPos)
|
|
negPos -= self.unitWidth*float(5)
|
|
posNum -= 5
|
|
|
|
def DrawYCoord(self,dc):
|
|
dc.SetPen(wx.BLACK_PEN)
|
|
dc.SetBrush(wx.BLACK_BRUSH)
|
|
dc.SetFont(wx.Font(8, wx.SWISS, wx.NORMAL, wx.NORMAL))
|
|
|
|
dc.DrawLine(float(0), self.zeroPos[1], self.w, self.zeroPos[1])
|
|
st = str(self.zero)
|
|
dc.DrawText(st, 5.0, self.zeroPos[1]-1.0)
|
|
|
|
dc.SetPen(wx.Pen(wx.Colour(150, 150, 150)))
|
|
dc.SetBrush(wx.Brush(wx.Colour(150, 150, 150)))
|
|
dc.SetLogicalFunction(wx.AND)
|
|
|
|
posPos = self.zeroPos[1]-self.unitHeight*float(5)
|
|
posNum = self.zero + 5
|
|
while posPos >= float(0):
|
|
dc.DrawLine(float(0), posPos, self.w, posPos)
|
|
st = str(posNum)
|
|
self.drawYNumber(dc, st, posPos)
|
|
posPos -= self.unitHeight*float(5)
|
|
posNum += 5
|
|
|
|
negPos = self.zeroPos[1]+self.unitHeight*float(5)
|
|
negNum = self.zero - 5
|
|
while negPos <= self.h:
|
|
dc.DrawLine(float(0), negPos, self.w, negPos)
|
|
st = str(negNum)
|
|
self.drawYNumber(dc, st, negPos)
|
|
negPos += self.unitHeight*float(5)
|
|
negNum -= 5
|
|
|
|
def drawXNumber(self, dc, st, pos):
|
|
oldPen, oldBrush, oldMode = dc.GetPen(), dc.GetBrush(), dc.GetLogicalFunction()
|
|
|
|
dc.SetPen(wx.BLACK_PEN)
|
|
dc.SetBrush(wx.BLACK_BRUSH)
|
|
dc.DrawText(st, pos+1.0, self.h-self.th-0.5)
|
|
|
|
dc.SetPen(oldPen)
|
|
dc.SetBrush(oldBrush)
|
|
dc.SetLogicalFunction(oldMode)
|
|
|
|
def drawYNumber(self, dc, st, pos):
|
|
oldPen, oldBrush, oldMode = dc.GetPen(), dc.GetBrush(), dc.GetLogicalFunction()
|
|
|
|
dc.SetPen(wx.BLACK_PEN)
|
|
dc.SetBrush(wx.BLACK_BRUSH)
|
|
dc.DrawText(st, 5.0, pos-1.0)
|
|
|
|
dc.SetPen(oldPen)
|
|
dc.SetBrush(oldBrush)
|
|
dc.SetLogicalFunction(oldMode)
|
|
|
|
def DrawFrame(self, dc):
|
|
if self._mainDialog.editor.mode == self._mainDialog.editor.ANIM_MODE:
|
|
curFramePos = self.zeroPos[0]+self.curFrame*self.unitWidth
|
|
dc.SetPen(wx.Pen("red"))
|
|
dc.SetBrush(wx.Brush("red"))
|
|
dc.DrawLine(curFramePos, float(0), curFramePos, self.h)
|
|
else:
|
|
pass
|
|
|
|
def drawX(self, dc):
|
|
dc.SetPen(wx.Pen("red"))
|
|
dc.SetBrush(wx.Brush("red"))
|
|
self.drawSingleCurve(self.X, dc)
|
|
self.drawKeys(self.X, dc)
|
|
self.drawHandler(self.X, dc)
|
|
|
|
def drawY(self, dc):
|
|
dc.SetPen(wx.Pen("green"))
|
|
dc.SetBrush(wx.Brush("green"))
|
|
self.drawSingleCurve(self.Y, dc)
|
|
self.drawKeys(self.Y, dc)
|
|
self.drawHandler(self.Y, dc)
|
|
|
|
def drawZ(self, dc):
|
|
dc.SetPen(wx.Pen("blue"))
|
|
dc.SetBrush(wx.Brush("blue"))
|
|
self.drawSingleCurve(self.Z, dc)
|
|
self.drawKeys(self.Z, dc)
|
|
self.drawHandler(self.Z, dc)
|
|
|
|
def DrawCurve(self, dc):
|
|
if self.property == self._mainDialog.namestr:
|
|
self.drawX(dc)
|
|
self.drawY(dc)
|
|
self.drawZ(dc)
|
|
return
|
|
if self.property == property[AG.X]:
|
|
self.drawX(dc)
|
|
return
|
|
if self.property == property[AG.Y]:
|
|
self.drawY(dc)
|
|
return
|
|
if self.property == property[AG.Z]:
|
|
self.drawZ(dc)
|
|
return
|
|
|
|
def drawSingleCurve(self, list, dc):
|
|
if len(list) == 1:
|
|
dc.DrawPoint(list[0][AG.KEYFRAME][AG.LOCAL_VALUE][0], list[0][AG.KEYFRAME][AG.LOCAL_VALUE][1])
|
|
return
|
|
|
|
if len(list) == 2:
|
|
dc.DrawLine(list[0][AG.KEYFRAME][AG.LOCAL_VALUE][0], list[0][AG.KEYFRAME][AG.LOCAL_VALUE][1], list[1][AG.KEYFRAME][AG.LOCAL_VALUE][0], list[1][AG.KEYFRAME][AG.LOCAL_VALUE][1])
|
|
return
|
|
|
|
if len(list)>=3 :
|
|
for i in range(len(list)-1):
|
|
x1 = list[i][AG.KEYFRAME][AG.LOCAL_VALUE][0]
|
|
y1 = list[i][AG.KEYFRAME][AG.LOCAL_VALUE][1]
|
|
|
|
x4 = list[i+1][AG.KEYFRAME][AG.LOCAL_VALUE][0]
|
|
y4 = list[i+1][AG.KEYFRAME][AG.LOCAL_VALUE][1]
|
|
|
|
t1x = list[i][AG.OUT_SLOPE][0]
|
|
t1y = list[i][AG.OUT_SLOPE][1]
|
|
|
|
t2x = list[i+1][AG.IN_SLOPE][0]
|
|
t2y = list[i+1][AG.IN_SLOPE][1]
|
|
|
|
x2 = x1 + (x4 - x1) / float(3);
|
|
scale1 = (x2 - x1) / t1x;
|
|
y2 = y1 - t1y * scale1;
|
|
|
|
|
|
x3 = x4 - (x4 - x1) / float(3);
|
|
scale2 = (x4 - x3) / t2x;
|
|
y3 = y4 + t2y * scale2;
|
|
|
|
ax = - float(1) * x1 + float(3) * x2 - float(3) * x3 + float(1) * x4;
|
|
bx = float(3) * x1 - float(6) * x2 + float(3) * x3 + float(0) * x4;
|
|
cx = - float(3) * x1 + float(3) * x2 + float(0) * x3 + float(0) * x4;
|
|
dx = float(1) * x1 + float(0) * x2 - float(0) * x3 + float(0) * x4;
|
|
|
|
ay = - float(1) * y1 + float(3) * y2 - float(3) * y3 + float(1) * y4;
|
|
by = float(3) * y1 - float(6) * y2 + float(3) * y3 + float(0) * y4;
|
|
cy = - float(3) * y1 + float(3) * y2 + float(0) * y3 + float(0) * y4;
|
|
dy = float(1) * y1 + float(0) * y2 - float(0) * y3 + float(0) * y4;
|
|
|
|
preX = x1
|
|
preY = y1
|
|
t = 0.001
|
|
|
|
while t<=float(1):
|
|
x = ax * t*t*t + bx * t*t + cx * t + dx;
|
|
y = ay * t*t*t + by * t*t + cy * t + dy;
|
|
|
|
curX = x
|
|
curY = y
|
|
|
|
dc.DrawLine(preX, preY, curX, curY)
|
|
|
|
preX = curX
|
|
preY = curY
|
|
|
|
t += 0.001
|
|
|
|
def drawKeys(self, list, dc):
|
|
for i in range(len(list)):
|
|
pointX = list[i][AG.KEYFRAME][AG.LOCAL_VALUE][0]
|
|
pointY = list[i][AG.KEYFRAME][AG.LOCAL_VALUE][1]
|
|
|
|
if list[i][AG.KEYFRAME][AG.SELECT] == 0:
|
|
dc.SetPen(wx.Pen("black",3))
|
|
dc.SetBrush(wx.Brush("black"))
|
|
dc.DrawCircle(pointX, pointY, 2)
|
|
|
|
if list[i][AG.KEYFRAME][AG.SELECT] == 1:
|
|
dc.SetPen(wx.Pen("cyan",3))
|
|
dc.SetBrush(wx.Brush("cyan"))
|
|
dc.DrawCircle(pointX, pointY, 2)
|
|
|
|
def drawHandler(self, list, dc):
|
|
for i in range(len(list)):
|
|
if list[i][AG.KEYFRAME][AG.SELECT] == 1:
|
|
X1 = list[i][AG.KEYFRAME][AG.LOCAL_VALUE][0]
|
|
Y1 = list[i][AG.KEYFRAME][AG.LOCAL_VALUE][1]
|
|
if self._OneTangent == True:
|
|
for j in range(3,5):
|
|
X = list[i][j][AG.LOCAL_VALUE][0]
|
|
Y = list[i][j][AG.LOCAL_VALUE][1]
|
|
if list[i][j][AG.SELECT] == 1:
|
|
dc.SetPen(wx.Pen("cyan",3))
|
|
dc.SetBrush(wx.Brush("cyan"))
|
|
dc.DrawCircle(X, Y, 2)
|
|
|
|
dc.SetPen(wx.Pen("cyan",1))
|
|
dc.DrawLine(X1, Y1, X, Y)
|
|
|
|
if list[i][j][AG.SELECT] == 0:
|
|
dc.SetPen(wx.Pen("brown",3))
|
|
dc.SetBrush(wx.Brush("brown"))
|
|
dc.DrawCircle(X, Y, 2)
|
|
|
|
dc.SetPen(wx.Pen("brown",1))
|
|
dc.DrawLine(X1, Y1, X, Y)
|
|
|
|
if self._OneTangent == False:
|
|
if list[i][AG.IN_TANGENT][AG.SELECT] == 1:
|
|
X = list[i][AG.IN_TANGENT][AG.LOCAL_VALUE][0]
|
|
Y = list[i][AG.IN_TANGENT][AG.LOCAL_VALUE][1]
|
|
dc.SetPen(wx.Pen("cyan",3))
|
|
dc.SetBrush(wx.Brush("cyan"))
|
|
dc.DrawCircle(X, Y, 2)
|
|
|
|
dc.SetPen(wx.Pen("cyan",1))
|
|
dc.DrawLine(X1, Y1, X, Y)
|
|
|
|
if list[i][AG.IN_TANGENT][AG.SELECT] == 0:
|
|
X = list[i][AG.IN_TANGENT][AG.LOCAL_VALUE][0]
|
|
Y = list[i][AG.IN_TANGENT][AG.LOCAL_VALUE][1]
|
|
dc.SetPen(wx.Pen("navy",3))
|
|
dc.SetBrush(wx.Brush("navy"))
|
|
dc.DrawCircle(X, Y, 2)
|
|
|
|
dc.SetPen(wx.Pen("navy",1))
|
|
dc.DrawLine(X1, Y1, X, Y)
|
|
|
|
if list[i][AG.OUT_TANGENT][AG.SELECT] == 1:
|
|
X = list[i][AG.OUT_TANGENT][AG.LOCAL_VALUE][0]
|
|
Y = list[i][AG.OUT_TANGENT][AG.LOCAL_VALUE][1]
|
|
dc.SetPen(wx.Pen("cyan",3))
|
|
dc.SetBrush(wx.Brush("cyan"))
|
|
dc.DrawCircle(X, Y, 2)
|
|
|
|
dc.SetPen(wx.Pen("cyan",1))
|
|
dc.DrawLine(X1, Y1, X, Y)
|
|
|
|
if list[i][AG.OUT_TANGENT][AG.SELECT] == 0:
|
|
X = list[i][AG.OUT_TANGENT][AG.LOCAL_VALUE][0]
|
|
Y = list[i][AG.OUT_TANGENT][AG.LOCAL_VALUE][1]
|
|
dc.SetPen(wx.Pen("brown",3))
|
|
dc.SetBrush(wx.Brush("brown"))
|
|
dc.DrawCircle(X, Y, 2)
|
|
|
|
dc.SetPen(wx.Pen("brown",1))
|
|
dc.DrawLine(X1, Y1, X, Y)
|
|
|
|
def DrawSelectRec(self, dc):
|
|
if self._selectRec == True:
|
|
dc.SetPen(wx.Pen("navy", 1))
|
|
dc.SetBrush(wx.Brush("navy"))
|
|
## dc.SetLogicalFunction(wx.AND)
|
|
dc.DrawLine(self.pos[0], self.pos[1], self.pos[0], self.newPos[1])
|
|
dc.DrawLine(self.pos[0], self.pos[1], self.newPos[0], self.pos[1])
|
|
dc.DrawLine(self.newPos[0], self.newPos[1], self.pos[0], self.newPos[1])
|
|
dc.DrawLine(self.newPos[0], self.newPos[1], self.newPos[0], self.pos[1])
|
|
|
|
def OnSize(self,evt):
|
|
self.InitBuffer()
|
|
|
|
def OnLeftDown(self,evt):
|
|
point = (evt.GetX(), evt.GetY())
|
|
|
|
if point[1]>= float(0) and point[1]<= float(self.h):
|
|
if point[0]>= float(0) and point[0]<= float(self.w):
|
|
self._mouseIn = True
|
|
|
|
if self._mouseIn:
|
|
self.CaptureMouse()
|
|
self.pos = point
|
|
|
|
def OnLeftUp(self,evt):
|
|
if self.GetCapture():
|
|
self.ReleaseMouse()
|
|
self._mouseIn = False
|
|
self._selectRec = False
|
|
self.setSelection()
|
|
self.SetGraphEditorData(self.property, self.curFrame)
|
|
|
|
def OnMiddleDown(self,evt):
|
|
point = (evt.GetX(), evt.GetY())
|
|
|
|
if point[1]>= float(0) and point[1]<= float(self.h):
|
|
if point[0]>= float(0) and point[0]<= float(self.w):
|
|
self._mouseIn = True
|
|
|
|
if self._mouseIn:
|
|
self.CaptureMouse()
|
|
self.midPos = point
|
|
|
|
def OnMiddleUp(self, evt):
|
|
if self.GetCapture():
|
|
self.ReleaseMouse()
|
|
|
|
def OnMotion(self,evt):
|
|
self._mouseIn = False
|
|
if evt.Dragging() and evt.LeftIsDown():
|
|
self.newPos = (evt.GetX(), evt.GetY())
|
|
if self.newPos[1]>= float(0) and self.newPos[1]<= float(self.h):
|
|
if self.newPos[0]>= float(0) and self.newPos[0]<= float(self.w):
|
|
self._mouseIn = True
|
|
|
|
if self._mouseIn:
|
|
if self.newPos == self.pos:
|
|
evt.Skip()
|
|
self._mouseIn = False
|
|
else:
|
|
self._selectRec = True
|
|
self.SetGraphEditorData(self.property, self.curFrame)
|
|
|
|
if evt.Dragging() and evt.MiddleIsDown():
|
|
self.newMidPos = (evt.GetX(), evt.GetY())
|
|
if self.newMidPos[1]>= float(0) and self.newMidPos[1]<= float(self.h):
|
|
if self.newMidPos[0]>= float(0) and self.newMidPos[0]<= float(self.w):
|
|
self._mouseIn = True
|
|
|
|
if self._mouseIn:
|
|
if self.newMidPos == self.midPos:
|
|
evt.Skip()
|
|
self._mouseIn = False
|
|
else:
|
|
self.recalculateSlope()
|
|
self.onAnimation()
|
|
self.midPos = self.newMidPos
|
|
|
|
evt.Skip()
|
|
self._mouseIn = False
|
|
self._selectRec = False
|
|
|
|
def setExistKey(self, list):
|
|
flag = False
|
|
for i in range(len(list)):
|
|
if list[i][AG.KEYFRAME][AG.SELECT] == 1:
|
|
inside = self.inside(self.pos, self.newPos, (list[i][AG.KEYFRAME][AG.LOCAL_VALUE][0], list[i][AG.KEYFRAME][AG.LOCAL_VALUE][1]))
|
|
if inside == True:
|
|
list[i][AG.KEYFRAME][AG.SELECT] = 0
|
|
if inside == False:
|
|
find = False
|
|
for j in range(3,5):
|
|
inside = self.inside(self.pos, self.newPos, (list[i][j][AG.LOCAL_VALUE][0], list[i][j][AG.LOCAL_VALUE][1]))
|
|
if inside == False:
|
|
list[i][j][AG.SELECT] = 0
|
|
if inside == True:
|
|
list[i][j][AG.SELECT] = 1
|
|
find = True
|
|
flag = True
|
|
if find == False:
|
|
list[i][AG.KEYFRAME][AG.SELECT] == 0
|
|
|
|
return flag
|
|
|
|
def setNewKey(self, list):
|
|
for i in range(len(list)):
|
|
inside = self.inside(self.pos, self.newPos, (list[i][2][0][0], list[i][2][0][1]))
|
|
if inside == True:
|
|
list[i][AG.KEYFRAME][AG.SELECT] = 1
|
|
if inside == False:
|
|
list[i][AG.KEYFRAME][AG.SELECT] = 0
|
|
|
|
def setSelection(self):
|
|
if self.property == self._mainDialog.namestr:
|
|
self.setSelectionBase(self.X)
|
|
self.setSelectionBase(self.Y)
|
|
self.setSelectionBase(self.Z)
|
|
return
|
|
if self.property == property[AG.X]:
|
|
self.setSelectionBase(self.X)
|
|
return
|
|
if self.property == property[AG.Y]:
|
|
self.setSelectionBase(self.Y)
|
|
return
|
|
if self.property == property[AG.Z]:
|
|
self.setSelectionBase(self.Z)
|
|
return
|
|
|
|
def setSelectionBase(self, list):
|
|
self.setExistKey(list)
|
|
if self.setExistKey(list) == True:
|
|
return
|
|
else:
|
|
self.setNewKey(list)
|
|
|
|
def inside(self, point0, point1, point):
|
|
if point0[0]<=point1[0] and point0[1]<=point1[1]:
|
|
if point0[0]<point[0] and point[0]<point1[0] and point0[1]<point[1] and point[1]<point1[1]:
|
|
return True
|
|
else:
|
|
return False
|
|
elif point1[0]<=point0[0] and point0[1]<=point1[1]:
|
|
if point1[0]<point[0] and point[0]<point0[0] and point0[1]<point[1] and point[1]<point1[1]:
|
|
return True
|
|
else:
|
|
return False
|
|
elif point0[0]<=point1[0] and point1[1]<=point0[1]:
|
|
if point0[0]<point[0] and point[0]<point1[0] and point1[1]<point[1] and point[1]<point0[1]:
|
|
return True
|
|
else:
|
|
return False
|
|
elif point1[0]<=point0[0] and point1[1]<=point0[1]:
|
|
if point1[0]<point[0] and point[0]<point0[0] and point1[1]<point[1] and point[1]<point0[1]:
|
|
return True
|
|
else:
|
|
return False
|
|
else:
|
|
return False
|
|
|
|
def recalculateSlope(self):
|
|
if self.property == self._mainDialog.namestr:
|
|
self.recalculateSlopeBase(self.X)
|
|
self.recalculateSlopeBase(self.Y)
|
|
self.recalculateSlopeBase(self.Z)
|
|
return
|
|
if self.property == property[AG.X]:
|
|
self.recalculateSlopeBase(self.X)
|
|
return
|
|
if self.property == property[AG.Y]:
|
|
self.recalculateSlopeBase(self.Y)
|
|
return
|
|
if self.property == property[AG.Z]:
|
|
self.recalculateSlopeBase(self.Z)
|
|
return
|
|
|
|
def recalculateSlopeBase(self, list):
|
|
#recalculate the tangent slope
|
|
moveX = self.newMidPos[0]-self.midPos[0]
|
|
moveY = self.newMidPos[1]-self.midPos[1]
|
|
|
|
for i in range(len(list)):
|
|
if list[i][AG.KEYFRAME][AG.SELECT] == 1:
|
|
if list[i][AG.IN_TANGENT][AG.SELECT] == 1:
|
|
newPointX = list[i][AG.IN_TANGENT][AG.LOCAL_VALUE][0] + moveX
|
|
newPointY = list[i][AG.IN_TANGENT][AG.LOCAL_VALUE][1] + moveY
|
|
|
|
newSlope = [list[i][AG.KEYFRAME][AG.LOCAL_VALUE][0] - newPointX , newPointY - list[i][AG.KEYFRAME][AG.LOCAL_VALUE][1]]
|
|
|
|
temp0 = self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]][AG.INSLOPE][0]
|
|
temp1 = self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]][AG.INSLOPE][1]
|
|
self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]][AG.INSLOPE][0] = newSlope[0]/self.unitWidth
|
|
self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]][AG.INSLOPE][1] = newSlope[1]/self.unitHeight
|
|
handler = self.generateHandler(self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]])
|
|
if handler[1][0][0]>= list[i][AG.KEYFRAME][AG.LOCAL_VALUE][0]:
|
|
self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]][AG.INSLOPE][0] = temp0
|
|
self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]][AG.INSLOPE][1] = temp1
|
|
return
|
|
if handler[1][0][0] < list[i][AG.KEYFRAME][AG.LOCAL_VALUE][0]:
|
|
if self._OneTangent == False:
|
|
list[i][AG.IN_TANGENT][0] = handler[1][0]
|
|
list[i][AG.IN_SLOPE][0] = handler[3][0]
|
|
list[i][AG.IN_SLOPE][1] = handler[3][1]
|
|
|
|
if self._OneTangent == True:
|
|
self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]][AG.OUTSLOPE][0] = newSlope[0]/self.unitWidth
|
|
self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]][AG.OUTSLOPE][1] = newSlope[1]/self.unitHeight
|
|
handler = self.generateHandler(self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]])
|
|
list[i][AG.IN_TANGENT][0] = handler[1][0]
|
|
list[i][AG.OUT_TANGENT][0] = handler[2][0]
|
|
list[i][AG.IN_SLOPE][0] = handler[3][0]
|
|
list[i][AG.IN_SLOPE][1] = handler[3][1]
|
|
list[i][AG.OUT_SLOPE][0] = handler[4][0]
|
|
list[i][AG.OUT_SLOPE][1] = handler[4][1]
|
|
|
|
self.SetGraphEditorData(self.property, self.curFrame)
|
|
|
|
if list[i][AG.OUT_TANGENT][AG.SELECT] == 1:
|
|
newPointX = list[i][AG.OUT_TANGENT][AG.LOCAL_VALUE][0] + moveX
|
|
newPointY = list[i][AG.OUT_TANGENT][AG.LOCAL_VALUE][1] + moveY
|
|
|
|
newSlope = [newPointX - list[i][AG.KEYFRAME][AG.LOCAL_VALUE][0] , list[i][AG.KEYFRAME][AG.LOCAL_VALUE][1] - newPointY]
|
|
|
|
temp0 = self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]][AG.OUTSLOPE][0]
|
|
temp1 = self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]][AG.OUTSLOPE][1]
|
|
self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]][AG.OUTSLOPE][0] = newSlope[0]/self.unitWidth
|
|
self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]][AG.OUTSLOPE][1] = newSlope[1]/self.unitHeight
|
|
|
|
handler = self.generateHandler(self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]])
|
|
if handler[2][0][0] <= list[i][AG.KEYFRAME][AG.LOCAL_VALUE][0]:
|
|
self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]][AG.OUTSLOPE][0] = temp0
|
|
self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]][AG.OUTSLOPE][1] = temp1
|
|
return
|
|
if handler[2][0][0] > list[i][AG.KEYFRAME][AG.LOCAL_VALUE][0]:
|
|
if self._OneTangent == False:
|
|
list[i][AG.OUT_TANGENT][0] = handler[2][0]
|
|
list[i][AG.OUT_SLOPE][0] = handler[4][0]
|
|
list[i][AG.OUT_SLOPE][1] = handler[4][1]
|
|
|
|
if self._OneTangent == True:
|
|
self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]][AG.INSLOPE][0] = newSlope[0]/self.unitWidth
|
|
self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]][AG.INSLOPE][1] = newSlope[1]/self.unitHeight
|
|
handler = self.generateHandler(self._mainDialog.editor.animMgr.keyFramesInfo[list[i][AG.KEY]][list[i][AG.I]])
|
|
list[i][AG.IN_TANGENT][0] = handler[1][0]
|
|
list[i][AG.OUT_TANGENT][0] = handler[2][0]
|
|
list[i][AG.IN_SLOPE][0] = handler[3][0]
|
|
list[i][AG.IN_SLOPE][1] = handler[3][1]
|
|
list[i][AG.OUT_SLOPE][0] = handler[4][0]
|
|
list[i][AG.OUT_SLOPE][1] = handler[4][1]
|
|
|
|
self.SetGraphEditorData(self.property, self.curFrame)
|
|
|
|
def selectHandler(self):
|
|
self._selectHandler = False
|
|
|
|
def onAnimation(self):
|
|
if self._mainDialog.editor.mode == self._mainDialog.editor.ANIM_MODE:
|
|
self._mainDialog.editor.ui.animUI.OnAnimation(self._mainDialog.editor.ui.animUI.curFrame)
|
|
else:
|
|
pass
|
|
|
|
|
|
class GraphEditorUI(wx.Dialog):
|
|
"""
|
|
This is the graph editor main class implementation.
|
|
"""
|
|
def __init__(self, parent, editor, object):
|
|
wx.Dialog.__init__(self, parent, id=wx.ID_ANY, title="Graph Editor",
|
|
pos=wx.DefaultPosition, size=(735, 535))
|
|
|
|
self.editor = editor
|
|
self.editor.GRAPH_EDITOR = True
|
|
self.object = object
|
|
self.xRange = 24+1
|
|
self.yRange = 50
|
|
if self.editor.mode == self.editor.ANIM_MODE:
|
|
self.curFrame = self.editor.ui.animUI.curFrame
|
|
self.curFrame = 1
|
|
|
|
self.mainPanel1 = wx.Panel(self, -1)
|
|
|
|
bmpZoomIn = ZoomIn.GetBitmap()
|
|
bmpZoomOut = ZoomOut.GetBitmap()
|
|
bmpOneTangent = OneTangent.GetBitmap()
|
|
bmpTwoTangents = TwoTangents.GetBitmap()
|
|
|
|
self.buttonZoomIn = wx.BitmapButton(self.mainPanel1, -1, bmpZoomIn, size = (30,30),style = wx.BU_AUTODRAW)
|
|
self.buttonZoomOut = wx.BitmapButton(self.mainPanel1, -1, bmpZoomOut, size = (30,30),style = wx.BU_AUTODRAW)
|
|
self.buttonOneTangent = wx.BitmapButton(self.mainPanel1, -1, bmpOneTangent, size = (30,30),style = wx.BU_AUTODRAW)
|
|
self.buttonTwoTangents = wx.BitmapButton(self.mainPanel1, -1, bmpTwoTangents, size = (30,30),style = wx.BU_AUTODRAW)
|
|
|
|
self.mainPanel2 = wx.Panel(self, -1)
|
|
|
|
self.tree = self.tree = wx.TreeCtrl(self.mainPanel2, id=-1, pos=wx.DefaultPosition,size=wx.Size(200, 450), style=wx.TR_MULTIPLE|wx.TR_DEFAULT_STYLE,validator=wx.DefaultValidator, name="treeCtrl")
|
|
self.namestr = "%s"%(object[OG.OBJ_DEF].name)
|
|
self.root = self.tree.AddRoot(self.namestr)
|
|
self.AddTreeNodes(self.root, property)
|
|
self.tree.Expand(self.root)
|
|
self.tree.SelectItem(self.root,select=True)
|
|
self.str = self.tree.GetItemText(self.root)
|
|
|
|
self.graphEditorWindow =GraphEditorWindow(self.mainPanel2, wx.Size(500, 450), str(object[OG.OBJ_DEF].name), self.xRange, self.yRange, self.curFrame, self.object)
|
|
|
|
self.SetProperties()
|
|
self.DoLayout()
|
|
|
|
self.Bind(wx.EVT_BUTTON, self.OnZoomIn, self.buttonZoomIn)
|
|
self.Bind(wx.EVT_BUTTON, self.OnZoomOut, self.buttonZoomOut)
|
|
self.Bind(wx.EVT_BUTTON, self.OnOneTangent, self.buttonOneTangent)
|
|
self.Bind(wx.EVT_BUTTON, self.OnTwoTangents, self.buttonTwoTangents)
|
|
|
|
self.Bind(wx.EVT_CLOSE, self.OnExit)
|
|
self.tree.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnSelChanged)
|
|
|
|
def SetProperties(self):
|
|
pass
|
|
|
|
def DoLayout(self):
|
|
dialogSizer = wx.BoxSizer(wx.VERTICAL)
|
|
mainSizer1 = wx.FlexGridSizer(1, 4, 0, 0)
|
|
mainSizer2 = wx.FlexGridSizer(1, 2, 0, 0)
|
|
|
|
mainSizer1.Add(self.buttonOneTangent, 0, wx.ALIGN_CENTER_VERTICAL|wx.LEFT, 570)
|
|
mainSizer1.Add(self.buttonTwoTangents, 0, wx.ALIGN_CENTER_VERTICAL|wx.LEFT, 10)
|
|
mainSizer1.Add(self.buttonZoomIn, 0, wx.ALIGN_CENTER_VERTICAL|wx.LEFT, 10)
|
|
mainSizer1.Add(self.buttonZoomOut, 0, wx.ALIGN_CENTER_VERTICAL|wx.LEFT|wx.RIGHT, 10)
|
|
|
|
mainSizer2.Add(self.tree, 0, wx.ALIGN_CENTER_VERTICAL|wx.LEFT|wx.RIGHT, 10)
|
|
mainSizer2.Add(self.graphEditorWindow, 0, wx.ALIGN_CENTER_VERTICAL|wx.RIGHT, 10)
|
|
|
|
self.mainPanel1.SetSizerAndFit(mainSizer1)
|
|
self.mainPanel2.SetSizer(mainSizer2)
|
|
|
|
dialogSizer.Add(self.mainPanel2, 0, wx.ALIGN_CENTER_VERTICAL|wx.TOP, 10)
|
|
dialogSizer.Add(self.mainPanel1, 0, wx.ALIGN_CENTER_VERTICAL|wx.TOP, 5)
|
|
|
|
self.SetSizer(dialogSizer)
|
|
self.Layout()
|
|
|
|
self.dialogSizer = dialogSizer
|
|
|
|
def AddTreeNodes(self, parentItem, items):
|
|
for item in items:
|
|
if type(item) == str:
|
|
self.tree.AppendItem(parentItem, item)
|
|
|
|
def OnSelChanged(self, evt):
|
|
item = evt.GetItem()
|
|
if item:
|
|
self.str = self.tree.GetItemText(item)
|
|
self.graphEditorWindow.refresh()
|
|
self.graphEditorWindow.SetGraphEditorData(self.str, self.curFrame)
|
|
|
|
def OnZoomIn(self,evt):
|
|
self.graphEditorWindow.zoom = self.graphEditorWindow.zoom/float(1.2)
|
|
self.graphEditorWindow.unitWidth = self.graphEditorWindow.unitWidth*float(1.2)
|
|
self.graphEditorWindow.unitHeight = self.graphEditorWindow.unitHeight*float(1.2)
|
|
self.graphEditorWindow.generateInfo()
|
|
self.graphEditorWindow.SetGraphEditorData(self.str, self.curFrame)
|
|
|
|
def OnZoomOut(self,evt):
|
|
self.graphEditorWindow.zoom = self.graphEditorWindow.zoom*float(1.2)
|
|
self.graphEditorWindow.unitWidth = self.graphEditorWindow.unitWidth/float(1.2)
|
|
self.graphEditorWindow.unitHeight = self.graphEditorWindow.unitHeight/float(1.2)
|
|
self.graphEditorWindow.generateInfo()
|
|
self.graphEditorWindow.SetGraphEditorData(self.str, self.curFrame)
|
|
|
|
def OnOneTangent(self,evt):
|
|
self.graphEditorWindow._OneTangent = True
|
|
self.graphEditorWindow.SetGraphEditorData(self.str, self.curFrame)
|
|
|
|
def OnTwoTangents(self,evt):
|
|
self.graphEditorWindow._OneTangent = False
|
|
self.graphEditorWindow.SetGraphEditorData(self.str, self.curFrame)
|
|
|
|
def curFrameChange(self):
|
|
if self.editor.mode == self.editor.ANIM_MODE:
|
|
self.curFrame = self.editor.ui.animUI.curFrame
|
|
self.graphEditorWindow.SetGraphEditorData(self.str, self.curFrame)
|
|
else:
|
|
pass
|
|
|
|
def OnExit(self,evt):
|
|
self.Destroy()
|
|
self.editor.ui.graphEditorMenuItem.Check(False)
|
|
self.object = None
|
|
self.editor.GRAPH_EDITOR = False
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|