Cura/gui/util/openglGui.py

   1 from __future__ import division
   2 __copyright__ = "Copyright (C) 2013 David Braam - Released under terms of the AGPLv3 License"
   3
   4 import wx
   5 import traceback
   6 import sys
   7 import os
   8 import time
   9
  10 from wx import glcanvas
  11 import OpenGL
  12 OpenGL.ERROR_CHECKING = False
  13 from OpenGL.GL import *
  14
  15 from Cura.util import version
  16 from Cura.gui.util import openglHelpers
  17
  18 class animation(object):
  19         def __init__(self, gui, start, end, runTime):
  20                 self._start = start
  21                 self._end = end
  22                 self._startTime = time.time()
  23                 self._runTime = runTime
  24                 gui._animationList.append(self)
  25
  26         def isDone(self):
  27                 return time.time() > self._startTime + self._runTime
  28
  29         def getPosition(self):
  30                 if self.isDone():
  31                         return self._end
  32                 f = (time.time() - self._startTime) / self._runTime
  33                 ts = f*f
  34                 tc = f*f*f
  35                 #f = 6*tc*ts + -15*ts*ts + 10*tc
  36                 f = tc + -3*ts + 3*f
  37                 return self._start + (self._end - self._start) * f
  38
  39 class glGuiControl(object):
  40         def __init__(self, parent, pos):
  41                 self._parent = parent
  42                 self._base = parent._base
  43                 self._pos = pos
  44                 self._size = (0,0, 1, 1)
  45                 self._parent.add(self)
  46
  47         def setSize(self, x, y, w, h):
  48                 self._size = (x, y, w, h)
  49
  50         def getSize(self):
  51                 return self._size
  52
  53         def getMinSize(self):
  54                 return 1, 1
  55
  56         def updateLayout(self):
  57                 pass
  58
  59         def focusNext(self):
  60                 for n in xrange(self._parent._glGuiControlList.index(self) + 1, len(self._parent._glGuiControlList)):
  61                         if self._parent._glGuiControlList[n].setFocus():
  62                                 return
  63                 for n in xrange(0, self._parent._glGuiControlList.index(self)):
  64                         if self._parent._glGuiControlList[n].setFocus():
  65                                 return
  66
  67         def focusPrevious(self):
  68                 for n in xrange(self._parent._glGuiControlList.index(self) -1, -1, -1):
  69                         if self._parent._glGuiControlList[n].setFocus():
  70                                 return
  71                 for n in xrange(len(self._parent._glGuiControlList) - 1, self._parent._glGuiControlList.index(self), -1):
  72                         if self._parent._glGuiControlList[n].setFocus():
  73                                 return
  74
  75         def setFocus(self):
  76                 return False
  77
  78         def hasFocus(self):
  79                 return self._base._focus == self
  80
  81         def OnMouseUp(self, x, y):
  82                 pass
  83
  84         def OnKeyChar(self, key):
  85                 pass
  86
  87 class glGuiContainer(glGuiControl):
  88         def __init__(self, parent, pos):
  89                 self._glGuiControlList = []
  90                 glGuiLayoutButtons(self)
  91                 super(glGuiContainer, self).__init__(parent, pos)
  92
  93         def add(self, ctrl):
  94                 self._glGuiControlList.append(ctrl)
  95                 self.updateLayout()
  96
  97         def OnMouseDown(self, x, y, button):
  98                 for ctrl in self._glGuiControlList:
  99                         if ctrl.OnMouseDown(x, y, button):
 100                                 return True
 101                 return False
 102
 103         def OnMouseUp(self, x, y):
 104                 for ctrl in self._glGuiControlList:
 105                         if ctrl.OnMouseUp(x, y):
 106                                 return True
 107                 return False
 108
 109         def OnMouseMotion(self, x, y):
 110                 handled = False
 111                 for ctrl in self._glGuiControlList:
 112                         if ctrl.OnMouseMotion(x, y):
 113                                 handled = True
 114                 return handled
 115
 116         def draw(self):
 117                 for ctrl in self._glGuiControlList:
 118                         ctrl.draw()
 119
 120         def updateLayout(self):
 121                 self._layout.update()
 122                 for ctrl in self._glGuiControlList:
 123                         ctrl.updateLayout()
 124
 125 class glGuiPanel(glcanvas.GLCanvas):
 126         def __init__(self, parent):
 127                 attribList = (glcanvas.WX_GL_RGBA, glcanvas.WX_GL_DOUBLEBUFFER, glcanvas.WX_GL_DEPTH_SIZE, 24, glcanvas.WX_GL_STENCIL_SIZE, 8, 0)
 128                 glcanvas.GLCanvas.__init__(self, parent, style=wx.WANTS_CHARS, attribList = attribList)
 129                 self._base = self
 130                 self._focus = None
 131                 self._container = None
 132                 self._container = glGuiContainer(self, (0,0))
 133                 self._shownError = False
 134
 135                 self._context = glcanvas.GLContext(self)
 136                 self._glButtonsTexture = None
 137                 self._glRobotTexture = None
 138                 self._buttonSize = 64
 139
 140                 self._animationList = []
 141                 self.glReleaseList = []
 142                 self._refreshQueued = False
 143                 self._idleCalled = False
 144
 145                 wx.EVT_PAINT(self, self._OnGuiPaint)
 146                 wx.EVT_SIZE(self, self._OnSize)
 147                 wx.EVT_ERASE_BACKGROUND(self, self._OnEraseBackground)
 148                 wx.EVT_LEFT_DOWN(self, self._OnGuiMouseDown)
 149                 wx.EVT_LEFT_DCLICK(self, self._OnGuiMouseDown)
 150                 wx.EVT_LEFT_UP(self, self._OnGuiMouseUp)
 151                 wx.EVT_RIGHT_DOWN(self, self._OnGuiMouseDown)
 152                 wx.EVT_RIGHT_DCLICK(self, self._OnGuiMouseDown)
 153                 wx.EVT_RIGHT_UP(self, self._OnGuiMouseUp)
 154                 wx.EVT_MIDDLE_DOWN(self, self._OnGuiMouseDown)
 155                 wx.EVT_MIDDLE_DCLICK(self, self._OnGuiMouseDown)
 156                 wx.EVT_MIDDLE_UP(self, self._OnGuiMouseUp)
 157                 wx.EVT_MOTION(self, self._OnGuiMouseMotion)
 158                 wx.EVT_CHAR(self, self._OnGuiKeyChar)
 159                 wx.EVT_KILL_FOCUS(self, self.OnFocusLost)
 160                 wx.EVT_IDLE(self, self._OnIdle)
 161
 162         def _OnIdle(self, e):
 163                 self._idleCalled = True
 164                 if len(self._animationList) > 0 or self._refreshQueued:
 165                         self._refreshQueued = False
 166                         for anim in self._animationList:
 167                                 if anim.isDone():
 168                                         self._animationList.remove(anim)
 169                         self.Refresh()
 170
 171         def _OnGuiKeyChar(self, e):
 172                 if self._focus is not None:
 173                         self._focus.OnKeyChar(e.GetKeyCode())
 174                         self.Refresh()
 175                 else:
 176                         self.OnKeyChar(e.GetKeyCode())
 177
 178         def OnFocusLost(self, e):
 179                 self._focus = None
 180                 self.Refresh()
 181
 182         def _OnGuiMouseDown(self,e):
 183                 self.SetFocus()
 184                 if self._container.OnMouseDown(e.GetX(), e.GetY(), e.GetButton()):
 185                         self.Refresh()
 186                         return
 187                 self.OnMouseDown(e)
 188
 189         def _OnGuiMouseUp(self, e):
 190                 if self._container.OnMouseUp(e.GetX(), e.GetY()):
 191                         self.Refresh()
 192                         return
 193                 self.OnMouseUp(e)
 194
 195         def _OnGuiMouseMotion(self,e):
 196                 self.Refresh()
 197                 if not self._container.OnMouseMotion(e.GetX(), e.GetY()):
 198                         self.OnMouseMotion(e)
 199
 200         def _OnGuiPaint(self, e):
 201                 self._idleCalled = False
 202                 h = self.GetSize().GetHeight()
 203                 w = self.GetSize().GetWidth()
 204                 oldButtonSize = self._buttonSize
 205                 if h / 3 < w / 4:
 206                         w = h * 4 / 3
 207                 if w < 64 * 8:
 208                         self._buttonSize = 32
 209                 elif w < 64 * 10:
 210                         self._buttonSize = 48
 211                 elif w < 64 * 15:
 212                         self._buttonSize = 64
 213                 elif w < 64 * 20:
 214                         self._buttonSize = 80
 215                 else:
 216                         self._buttonSize = 96
 217                 if self._buttonSize != oldButtonSize:
 218                         self._container.updateLayout()
 219
 220                 dc = wx.PaintDC(self)
 221                 try:
 222                         self.SetCurrent(self._context)
 223                         for obj in self.glReleaseList:
 224                                 obj.release()
 225                         del self.glReleaseList[:]
 226                         renderStartTime = time.time()
 227                         self.OnPaint(e)
 228                         self._drawGui()
 229                         glFlush()
 230                         if version.isDevVersion():
 231                                 renderTime = time.time() - renderStartTime
 232                                 if renderTime == 0:
 233                                         renderTime = 0.001
 234                                 glLoadIdentity()
 235                                 glTranslate(10, self.GetSize().GetHeight() - 30, -1)
 236                                 glColor4f(0.2,0.2,0.2,0.5)
 237                                 openglHelpers.glDrawStringLeft("fps:%d" % (1 / renderTime))
 238                         self.SwapBuffers()
 239                 except:
 240                         # When an exception happens, catch it and show a message box. If the exception is not caught the draw function bugs out.
 241                         # Only show this exception once so we do not overload the user with popups.
 242                         errStr = _("An error has occurred during the 3D view drawing.")
 243                         tb = traceback.extract_tb(sys.exc_info()[2])
 244                         errStr += "\n%s: '%s'" % (str(sys.exc_info()[0].__name__), str(sys.exc_info()[1]))
 245                         for n in xrange(len(tb)-1, -1, -1):
 246                                 locationInfo = tb[n]
 247                                 errStr += "\n @ %s:%s:%d" % (os.path.basename(locationInfo[0]), locationInfo[2], locationInfo[1])
 248                         if not self._shownError:
 249                                 traceback.print_exc()
 250                                 wx.CallAfter(wx.MessageBox, errStr, _("3D window error"), wx.OK | wx.ICON_EXCLAMATION)
 251                                 self._shownError = True
 252
 253         def _drawGui(self):
 254                 if self._glButtonsTexture is None:
 255                         self._glButtonsTexture = openglHelpers.loadGLTexture('glButtons.png')
 256                         self._glRobotTexture = openglHelpers.loadGLTexture('UltimakerRobot.png')
 257
 258                 glDisable(GL_DEPTH_TEST)
 259                 glEnable(GL_BLEND)
 260                 glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA)
 261                 glDisable(GL_LIGHTING)
 262                 glColor4ub(255,255,255,255)
 263
 264                 glMatrixMode(GL_PROJECTION)
 265                 glLoadIdentity()
 266                 size = self.GetSize()
 267                 glOrtho(0, size.GetWidth()-1, size.GetHeight()-1, 0, -1000.0, 1000.0)
 268                 glMatrixMode(GL_MODELVIEW)
 269                 glLoadIdentity()
 270
 271                 self._container.draw()
 272
 273                 # glBindTexture(GL_TEXTURE_2D, self._glRobotTexture)
 274                 # glEnable(GL_TEXTURE_2D)
 275                 # glPushMatrix()
 276                 # glColor4f(1,1,1,1)
 277                 # glTranslate(size.GetWidth(),size.GetHeight(),0)
 278                 # s = self._buttonSize * 1
 279                 # glScale(s,s,s)
 280                 # glTranslate(-1.2,-0.2,0)
 281                 # glBegin(GL_QUADS)
 282                 # glTexCoord2f(1, 0)
 283                 # glVertex2f(0,-1)
 284                 # glTexCoord2f(0, 0)
 285                 # glVertex2f(-1,-1)
 286                 # glTexCoord2f(0, 1)
 287                 # glVertex2f(-1, 0)
 288                 # glTexCoord2f(1, 1)
 289                 # glVertex2f(0, 0)
 290                 # glEnd()
 291                 # glDisable(GL_TEXTURE_2D)
 292                 # glPopMatrix()
 293
 294         def _OnEraseBackground(self,event):
 295                 #Workaround for windows background redraw flicker.
 296                 pass
 297
 298         def _OnSize(self,e):
 299                 self._container.setSize(0, 0, self.GetSize().GetWidth(), self.GetSize().GetHeight())
 300                 self._container.updateLayout()
 301                 self.Refresh()
 302
 303         def OnMouseDown(self,e):
 304                 pass
 305         def OnMouseUp(self,e):
 306                 pass
 307         def OnMouseMotion(self, e):
 308                 pass
 309         def OnKeyChar(self, keyCode):
 310                 pass
 311         def OnPaint(self, e):
 312                 pass
 313         def OnKeyChar(self, keycode):
 314                 pass
 315
 316         def QueueRefresh(self):
 317                 wx.CallAfter(self._queueRefresh)
 318
 319         def _queueRefresh(self):
 320                 if self._idleCalled:
 321                         wx.CallAfter(self.Refresh)
 322                 else:
 323                         self._refreshQueued = True
 324
 325         def add(self, ctrl):
 326                 if self._container is not None:
 327                         self._container.add(ctrl)
 328
 329 class glGuiLayoutButtons(object):
 330         def __init__(self, parent):
 331                 self._parent = parent
 332                 self._parent._layout = self
 333
 334         def update(self):
 335                 bs = self._parent._base._buttonSize
 336                 x0, y0, w, h = self._parent.getSize()
 337                 gridSize = bs * 1.0
 338                 for ctrl in self._parent._glGuiControlList:
 339                         pos = ctrl._pos
 340                         if pos[0] < 0:
 341                                 x = w + pos[0] * gridSize - bs * 0.2
 342                         else:
 343                                 x = pos[0] * gridSize + bs * 0.2
 344                         if pos[1] < 0:
 345                                 y = h + pos[1] * gridSize * 1.2 - bs * 0.0
 346                         else:
 347                                 y = pos[1] * gridSize * 1.2 + bs * 0.2
 348                         ctrl.setSize(x, y, gridSize, gridSize)
 349
 350         def getLayoutSize(self):
 351                 _, _, w, h = self._parent.getSize()
 352                 return w, h
 353
 354 class glGuiLayoutGrid(object):
 355         def __init__(self, parent):
 356                 self._parent = parent
 357                 self._parent._layout = self
 358                 self._size = 0,0
 359                 self._alignBottom = True
 360
 361         def update(self):
 362                 borderSize = self._parent._base._buttonSize * 0.2
 363                 x0, y0, w, h = self._parent.getSize()
 364                 x0 += borderSize
 365                 y0 += borderSize
 366                 widths = {}
 367                 heights = {}
 368                 for ctrl in self._parent._glGuiControlList:
 369                         x, y = ctrl._pos
 370                         w, h = ctrl.getMinSize()
 371                         if not x in widths:
 372                                 widths[x] = w
 373                         else:
 374                                 widths[x] = max(widths[x], w)
 375                         if not y in heights:
 376                                 heights[y] = h
 377                         else:
 378                                 heights[y] = max(heights[y], h)
 379                 self._size = sum(widths.values()) + borderSize * 2, sum(heights.values()) + borderSize * 2
 380                 if self._alignBottom:
 381                         y0 -= self._size[1] - self._parent.getSize()[3]
 382                         self._parent.setSize(x0 - borderSize, y0 - borderSize, self._size[0], self._size[1])
 383                 for ctrl in self._parent._glGuiControlList:
 384                         x, y = ctrl._pos
 385                         x1 = x0
 386                         y1 = y0
 387                         for n in xrange(0, x):
 388                                 if not n in widths:
 389                                         widths[n] = 3
 390                                 x1 += widths[n]
 391                         for n in xrange(0, y):
 392                                 if not n in heights:
 393                                         heights[n] = 3
 394                                 y1 += heights[n]
 395                         ctrl.setSize(x1, y1, widths[x], heights[y])
 396
 397         def getLayoutSize(self):
 398                 return self._size
 399
 400 class glButton(glGuiControl):
 401         def __init__(self, parent, imageID, tooltip, pos, callback, size = None):
 402                 self._buttonSize = size
 403                 self._hidden = False
 404                 super(glButton, self).__init__(parent, pos)
 405                 self._tooltip = tooltip
 406                 self._parent = parent
 407                 self._imageID = imageID
 408                 self._callback = callback
 409                 self._selected = False
 410                 self._focus = False
 411                 self._disabled = False
 412                 self._showExpandArrow = False
 413                 self._progressBar = None
 414                 self._altTooltip = ''
 415
 416         def setSelected(self, value):
 417                 self._selected = value
 418
 419         def setExpandArrow(self, value):
 420                 self._showExpandArrow = value
 421
 422         def setHidden(self, value):
 423                 self._hidden = value
 424
 425         def setDisabled(self, value):
 426                 self._disabled = value
 427
 428         def setProgressBar(self, value):
 429                 self._progressBar = value
 430
 431         def getProgressBar(self):
 432                 return self._progressBar
 433
 434         def setBottomText(self, value):
 435                 self._altTooltip = value
 436
 437         def getSelected(self):
 438                 return self._selected
 439
 440         def getMinSize(self):
 441                 if self._hidden:
 442                         return 0, 0
 443                 if self._buttonSize is not None:
 444                         if self._buttonSize > 0 and self._buttonSize < 1:
 445                                 return self._base._buttonSize * self._buttonSize, self._base._buttonSize * self._buttonSize
 446
 447                         return self._buttonSize, self._buttonSize
 448                 return self._base._buttonSize, self._base._buttonSize
 449
 450         def _getPixelPos(self):
 451                 x0, y0, w, h = self.getSize()
 452                 return x0 + w / 2, y0 + h / 2
 453
 454         def draw(self):
 455                 if self._hidden:
 456                         return
 457
 458                 cx = (self._imageID % 4) / 4
 459                 cy = int(self._imageID / 4) / 4
 460                 bs = self.getMinSize()[0]
 461                 pos = self._getPixelPos()
 462
 463                 glBindTexture(GL_TEXTURE_2D, self._base._glButtonsTexture)
 464                 #button sizing
 465                 scale = 1.0
 466                 if self._selected:
 467                         scale = 1.0
 468                 elif self._focus:
 469                         scale = 1.0
 470                 if self._disabled:
 471                         glColor4ub(128,128,128,128)
 472                 else:
 473                         glColor4ub(255,255,255,255)
 474                 openglHelpers.glDrawTexturedQuad(pos[0]-bs*scale/2, pos[1]-bs*scale/2, bs*scale, bs*scale, 0)
 475                 openglHelpers.glDrawTexturedQuad(pos[0]-bs*scale/2, pos[1]-bs*scale/2, bs*scale, bs*scale, self._imageID)
 476                 if self._showExpandArrow:
 477                         if self._selected:
 478                                 openglHelpers.glDrawTexturedQuad(pos[0]+bs*scale/2-bs*scale/4*1.2, pos[1]-bs*scale/2*1.2, bs*scale/4, bs*scale/4, 1)
 479                         else:
 480                                 openglHelpers.glDrawTexturedQuad(pos[0]+bs*scale/2-bs*scale/4*1.2, pos[1]-bs*scale/2*1.2, bs*scale/4, bs*scale/4, 1, 2)
 481                 glPushMatrix()
 482                 glTranslatef(pos[0], pos[1], 0)
 483                 glDisable(GL_TEXTURE_2D)
 484                 '''if self._focus:
 485                         glTranslatef(0, -0.55*bs*scale, 0)
 486
 487                         glPushMatrix()
 488                         glColor4ub(60,60,60,255)
 489                         glTranslatef(-1, -1, 0)
 490                         openglHelpers.glDrawStringCenter(self._tooltip)
 491                         glTranslatef(0, 2, 0)
 492                         openglHelpers.glDrawStringCenter(self._tooltip)
 493                         glTranslatef(2, 0, 0)
 494                         openglHelpers.glDrawStringCenter(self._tooltip)
 495                         glTranslatef(0, -2, 0)
 496                         openglHelpers.glDrawStringCenter(self._tooltip)
 497                         glPopMatrix()
 498
 499                         glColor4ub(255,255,255,255)
 500                         openglHelpers.glDrawStringCenter(self._tooltip)
 501                 '''
 502                 glPopMatrix()
 503                 progress = self._progressBar
 504                 if progress is not None:
 505                         glColor4ub(60,60,60,255)
 506                         openglHelpers.glDrawQuad(pos[0]-bs/2, pos[1]+bs/2, bs, bs / 4)
 507                         glColor4ub(255,255,255,255)
 508                         openglHelpers.glDrawQuad(pos[0]-bs/2+2, pos[1]+bs/2+2, (bs - 5) * progress + 1, bs / 4 - 4)
 509                 elif len(self._altTooltip) > 0:
 510                         glPushMatrix()
 511                         glTranslatef(pos[0], pos[1], 0)
 512                         glTranslatef(0, 0.6*bs, 0)
 513                         glTranslatef(0, 6, 0)
 514                         #glTranslatef(0.6*bs*scale, 0, 0)
 515
 516                         for line in self._altTooltip.split('\n'):
 517                                 glPushMatrix()
 518                                 glColor4ub(60,60,60,255)
 519                                 glTranslatef(-1, -1, 0)
 520                                 openglHelpers.glDrawStringCenter(line)
 521                                 glTranslatef(0, 2, 0)
 522                                 openglHelpers.glDrawStringCenter(line)
 523                                 glTranslatef(2, 0, 0)
 524                                 openglHelpers.glDrawStringCenter(line)
 525                                 glTranslatef(0, -2, 0)
 526                                 openglHelpers.glDrawStringCenter(line)
 527                                 glPopMatrix()
 528
 529                                 glColor4ub(255,255,255,255)
 530                                 openglHelpers.glDrawStringCenter(line)
 531                                 glTranslatef(0, 18, 0)
 532                         glPopMatrix()
 533
 534         def _checkHit(self, x, y):
 535                 if self._hidden or self._disabled:
 536                         return False
 537                 bs = self.getMinSize()[0]
 538                 pos = self._getPixelPos()
 539                 return -bs * 0.5 <= x - pos[0] <= bs * 0.5 and -bs * 0.5 <= y - pos[1] <= bs * 0.5
 540
 541         def OnMouseMotion(self, x, y):
 542                 if self._checkHit(x, y):
 543                         self._focus = True
 544                         return True
 545                 self._focus = False
 546                 return False
 547
 548         def OnMouseDown(self, x, y, button):
 549                 if self._checkHit(x, y):
 550                         self._callback(button)
 551                         return True
 552                 return False
 553
 554 class glRadioButton(glButton):
 555         def __init__(self, parent, imageID, tooltip, pos, group, callback):
 556                 super(glRadioButton, self).__init__(parent, imageID, tooltip, pos, self._onRadioSelect)
 557                 self._group = group
 558                 self._radioCallback = callback
 559                 self._group.append(self)
 560
 561         def setSelected(self, value):
 562                 self._selected = value
 563
 564         def _onRadioSelect(self, button):
 565                 self._base._focus = None
 566                 for ctrl in self._group:
 567                         if ctrl != self:
 568                                 ctrl.setSelected(False)
 569                 if self.getSelected():
 570                         self.setSelected(False)
 571                 else:
 572                         self.setSelected(True)
 573                 self._radioCallback(button)
 574
 575 class glComboButton(glButton):
 576         def __init__(self, parent, tooltip, defaultImageID, imageIDs, tooltips, pos, callback, stateCallback = None):
 577                 super(glComboButton, self).__init__(parent, defaultImageID, tooltip, pos, self._onComboOpenSelect)
 578                 self._imageIDs = imageIDs
 579                 self._tooltips = tooltips
 580                 self._comboCallback = callback
 581                 self._comboStateCallback = stateCallback
 582                 self._selection = 0
 583                 self._disabled = False
 584
 585         def _onComboOpenSelect(self, button):
 586                 if self.hasFocus():
 587                         self._base._focus = None
 588                 else:
 589                         self._base._focus = self
 590
 591         def draw(self):
 592                 if self._hidden:
 593                         return
 594
 595                 if self._comboStateCallback is not None and self.hasFocus() != self._selected:
 596                         self._comboStateCallback(self.hasFocus())
 597                 self._selected = self.hasFocus()
 598
 599                 super(glComboButton, self).draw()
 600
 601                 bs = self._base._buttonSize / 1.3
 602                 pos = self._getPixelPos()
 603
 604                 if not self._selected:
 605                         return
 606
 607                 glPushMatrix()
 608                 glTranslatef(pos[0]+bs*0.5, pos[1] + bs*0.5, 0)
 609                 glBindTexture(GL_TEXTURE_2D, self._base._glButtonsTexture)
 610                 for n in xrange(0, len(self._imageIDs)):
 611                         glTranslatef(0, bs, 0)
 612                         glColor4ub(255,255,255,255)
 613                         openglHelpers.glDrawTexturedQuad(-0.85*bs,-0.8*bs,bs,bs, 0)
 614                         openglHelpers.glDrawTexturedQuad(-0.85*bs,-0.8*bs,bs,bs, self._imageIDs[n])
 615                         glDisable(GL_TEXTURE_2D)
 616
 617                         glPushMatrix()
 618                         glTranslatef(-0.55*bs, 0.1*bs, 0)
 619
 620                         glPushMatrix()
 621                         glColor4ub(60,60,60,255)
 622                         glTranslatef(-1, -1, 0)
 623                         #openglHelpers.glDrawStringRight(self._tooltips[n])
 624                         glTranslatef(0, 2, 0)
 625                         #openglHelpers.glDrawStringRight(self._tooltips[n])
 626                         glTranslatef(2, 0, 0)
 627                         #openglHelpers.glDrawStringRight(self._tooltips[n])
 628                         glTranslatef(0, -2, 0)
 629                         #openglHelpers.glDrawStringRight(self._tooltips[n])
 630                         glPopMatrix()
 631
 632                         glColor4ub(255,255,255,255)
 633                         #openglHelpers.glDrawStringRight(self._tooltips[n])
 634                         glPopMatrix()
 635                 glPopMatrix()
 636
 637         def getValue(self):
 638                 return self._selection
 639
 640         def setDisabled(self, value):
 641                 self._disabled = value
 642
 643         def setValue(self, value):
 644                 self._selection = value
 645                 self._imageID = self._imageIDs[self._selection]
 646                 self._comboCallback()
 647
 648         def OnMouseDown(self, x, y, button):
 649                 if self._hidden or self._disabled:
 650                         return False
 651                 if self.hasFocus():
 652                         bs = self._base._buttonSize / 1.3
 653                         pos = self.getSize()
 654                         offsetX = pos[0] + (self._base._buttonSize - bs)
 655                         offsetY = pos[1] + self._base._buttonSize
 656
 657                         if 0 <= x - offsetX <= bs and 0 <= y - offsetY <= bs * len(self._imageIDs):
 658                                 self._selection = int((y - offsetY) / bs)
 659                                 self._imageID = self._imageIDs[self._selection]
 660                                 self._base._focus = None
 661                                 self._comboCallback()
 662                                 return True
 663                 return super(glComboButton, self).OnMouseDown(x, y, button)
 664
 665 class glFrame(glGuiContainer):
 666         def __init__(self, parent, pos):
 667                 super(glFrame, self).__init__(parent, pos)
 668                 self._selected = False
 669                 self._focus = False
 670                 self._hidden = False
 671
 672         def setSelected(self, value):
 673                 self._selected = value
 674
 675         def setHidden(self, value):
 676                 self._hidden = value
 677                 for child in self._glGuiControlList:
 678                         if self._base._focus == child:
 679                                 self._base._focus = None
 680
 681         def getSelected(self):
 682                 return self._selected
 683
 684         def getMinSize(self):
 685                 return self._base._buttonSize, self._base._buttonSize
 686
 687         def _getPixelPos(self):
 688                 x0, y0, w, h = self.getSize()
 689                 return x0, y0
 690
 691         def draw(self):
 692                 if self._hidden:
 693                         return
 694
 695                 bs = self._parent._buttonSize
 696                 pos = self._getPixelPos()
 697
 698                 size = self._layout.getLayoutSize()
 699                 glColor4ub(255,255,255,255)
 700                 openglHelpers.glDrawStretchedQuad(pos[0], pos[1], size[0], size[1], bs*0.75, 0)
 701                 #Draw the controls on the frame
 702                 super(glFrame, self).draw()
 703
 704         def _checkHit(self, x, y):
 705                 if self._hidden:
 706                         return False
 707                 pos = self._getPixelPos()
 708                 w, h = self._layout.getLayoutSize()
 709                 return 0 <= x - pos[0] <= w and 0 <= y - pos[1] <= h
 710
 711         def OnMouseMotion(self, x, y):
 712                 super(glFrame, self).OnMouseMotion(x, y)
 713                 if self._checkHit(x, y):
 714                         self._focus = True
 715                         return True
 716                 self._focus = False
 717                 return False
 718
 719         def OnMouseDown(self, x, y, button):
 720                 if self._checkHit(x, y):
 721                         super(glFrame, self).OnMouseDown(x, y, button)
 722                         return True
 723                 return False
 724
 725 class glNotification(glFrame):
 726         def __init__(self, parent, pos):
 727                 self._anim = None
 728                 self._previous_base_size = None
 729                 super(glNotification, self).__init__(parent, pos)
 730                 glGuiLayoutGrid(self)._alignBottom = False
 731                 self._label_text = "Notification"
 732                 self._label = glLabel(self, self._label_text, (0, 0))
 733                 self._buttonExtra = glButton(self, 31, "???", (1, 0), self.onExtraButton, 25)
 734                 self._button = glButton(self, 30, "", (2, 0), self.onClose, 25)
 735                 self._padding = glLabel(self, "", (0, 1))
 736                 self.updateLabelSize()
 737                 self.setHidden(True)
 738
 739         def setSize(self, x, y, w, h):
 740                 w, h = self._layout.getLayoutSize()
 741                 baseSize = self._base.GetSizeTuple()
 742                 if self._previous_base_size and self._previous_base_size != baseSize:
 743                         self.updateLabelSize()
 744                         w, h = self._layout.getLayoutSize()
 745
 746                 if self._anim is not None:
 747                         super(glNotification, self).setSize(baseSize[0] / 2 - w / 2, baseSize[1] - self._anim.getPosition() - self._base._buttonSize * 0.2, 1, 1)
 748                 else:
 749                         super(glNotification, self).setSize(baseSize[0] / 2 - w / 2, baseSize[1] - self._base._buttonSize * 0.2, 1, 1)
 750
 751         def updateLabelSize(self):
 752                 self._previous_base_size = self._base.GetSizeTuple()
 753                 self._label.setLabel(self._label_text)
 754                 self.updateLayout()
 755                 w, h = self._layout.getLayoutSize()
 756                 baseSize = self._base.GetSizeTuple()
 757                 crop = -1
 758                 while baseSize[0] < w:
 759                         text = self._label_text[:crop]
 760                         crop = crop - 1
 761                         self._label.setLabel(text + "...")
 762                         self.updateLayout()
 763                         w, h = self._layout.getLayoutSize()
 764                         if len(text) == 0:
 765                                 break
 766
 767         def draw(self):
 768                 self.setSize(0,0,0,0)
 769                 self.updateLayout()
 770                 super(glNotification, self).draw()
 771
 772         def message(self, text, extraButtonCallback = None, extraButtonIcon = None, extraButtonTooltip = None):
 773                 self._anim = animation(self._base, -20, 25, 1)
 774                 self.setHidden(False)
 775                 self._label_text = text
 776                 self._buttonExtra.setHidden(extraButtonCallback is None)
 777                 self._buttonExtra._imageID = extraButtonIcon
 778                 self._buttonExtra._tooltip = extraButtonTooltip
 779                 self._extraButtonCallback = extraButtonCallback
 780                 self._base._queueRefresh()
 781                 self.updateLabelSize()
 782                 self.setSize(0,0,0,0)
 783
 784         def onExtraButton(self, button):
 785                 self.onClose(button)
 786                 self._extraButtonCallback()
 787
 788         def onClose(self, button):
 789                 if self._anim is not None:
 790                         self._anim = animation(self._base, self._anim.getPosition(), -20, 1)
 791                 else:
 792                         self._anim = animation(self._base, 25, -20, 1)
 793
 794 class glLabel(glGuiControl):
 795         def __init__(self, parent, label, pos):
 796                 self._label = label
 797                 super(glLabel, self).__init__(parent, pos)
 798
 799         def setLabel(self, label):
 800                 self._label = label
 801
 802         def getMinSize(self):
 803                 w, h = openglHelpers.glGetStringSize(self._label)
 804                 return w + 10, h + 4
 805
 806         def _getPixelPos(self):
 807                 x0, y0, w, h = self.getSize()
 808                 return x0, y0
 809
 810         def draw(self):
 811                 x, y, w, h = self.getSize()
 812
 813                 glPushMatrix()
 814                 glTranslatef(x, y, 0)
 815
 816 #               glColor4ub(255,255,255,128)
 817 #               glBegin(GL_QUADS)
 818 #               glTexCoord2f(1, 0)
 819 #               glVertex2f( w, 0)
 820 #               glTexCoord2f(0, 0)
 821 #               glVertex2f( 0, 0)
 822 #               glTexCoord2f(0, 1)
 823 #               glVertex2f( 0, h)
 824 #               glTexCoord2f(1, 1)
 825 #               glVertex2f( w, h)
 826 #               glEnd()
 827
 828                 glTranslate(5, h - 5, 0)
 829                 glColor4ub(255,255,255,255)
 830                 openglHelpers.glDrawStringLeft(self._label)
 831                 glPopMatrix()
 832
 833         def _checkHit(self, x, y):
 834                 return False
 835
 836         def OnMouseMotion(self, x, y):
 837                 return False
 838
 839         def OnMouseDown(self, x, y, button):
 840                 return False
 841
 842 class glNumberCtrl(glGuiControl):
 843         def __init__(self, parent, value, pos, callback):
 844                 self._callback = callback
 845                 self._value = str(value)
 846                 self._selectPos = 0
 847                 self._maxLen = 6
 848                 self._inCallback = False
 849                 super(glNumberCtrl, self).__init__(parent, pos)
 850
 851         def setValue(self, value):
 852                 if self._inCallback:
 853                         return
 854                 self._value = str(value)
 855
 856         def getMinSize(self):
 857                 w, h = openglHelpers.glGetStringSize("VALUES")
 858                 return w + 10, h + 4
 859
 860         def _getPixelPos(self):
 861                 x0, y0, w, h = self.getSize()
 862                 return x0, y0
 863
 864         def draw(self):
 865                 x, y, w, h = self.getSize()
 866
 867                 glPushMatrix()
 868                 glTranslatef(x, y, 0)
 869
 870                 if self.hasFocus():
 871                         glColor4ub(255,255,255,255)
 872                 else:
 873                         glColor4ub(255,255,255,192)
 874                 glBegin(GL_QUADS)
 875                 glTexCoord2f(1, 0)
 876                 glVertex2f( w, 0)
 877                 glTexCoord2f(0, 0)
 878                 glVertex2f( 0, 0)
 879                 glTexCoord2f(0, 1)
 880                 glVertex2f( 0, h-1)
 881                 glTexCoord2f(1, 1)
 882                 glVertex2f( w, h-1)
 883                 glEnd()
 884
 885                 glTranslate(5, h - 5, 0)
 886                 glColor4ub(0,0,0,255)
 887                 openglHelpers.glDrawStringLeft(self._value)
 888                 if self.hasFocus():
 889                         glTranslate(openglHelpers.glGetStringSize(self._value[0:self._selectPos])[0] - 2, -1, 0)
 890                         openglHelpers.glDrawStringLeft('|')
 891                 glPopMatrix()
 892
 893         def _checkHit(self, x, y):
 894                 x1, y1, w, h = self.getSize()
 895                 return 0 <= x - x1 <= w and 0 <= y - y1 <= h
 896
 897         def OnMouseMotion(self, x, y):
 898                 return False
 899
 900         def OnMouseDown(self, x, y, button):
 901                 if self._checkHit(x, y):
 902                         self.setFocus()
 903                         return True
 904                 return False
 905
 906         def OnKeyChar(self, c):
 907                 self._inCallback = True
 908                 if c == wx.WXK_LEFT:
 909                         self._selectPos -= 1
 910                         self._selectPos = max(0, self._selectPos)
 911                 if c == wx.WXK_RIGHT:
 912                         self._selectPos += 1
 913                         self._selectPos = min(self._selectPos, len(self._value))
 914                 if c == wx.WXK_UP:
 915                         try:
 916                                 value = float(self._value)
 917                         except:
 918                                 pass
 919                         else:
 920                                 value += 0.1
 921                                 self._value = str(value)
 922                                 self._callback(self._value)
 923                 if c == wx.WXK_DOWN:
 924                         try:
 925                                 value = float(self._value)
 926                         except:
 927                                 pass
 928                         else:
 929                                 value -= 0.1
 930                                 if value > 0:
 931                                         self._value = str(value)
 932                                         self._callback(self._value)
 933                 if c == wx.WXK_BACK and self._selectPos > 0:
 934                         self._value = self._value[0:self._selectPos - 1] + self._value[self._selectPos:]
 935                         self._selectPos -= 1
 936                         self._callback(self._value)
 937                 if c == wx.WXK_DELETE:
 938                         self._value = self._value[0:self._selectPos] + self._value[self._selectPos + 1:]
 939                         self._callback(self._value)
 940                 if c == wx.WXK_TAB or c == wx.WXK_NUMPAD_ENTER or c == wx.WXK_RETURN:
 941                         if wx.GetKeyState(wx.WXK_SHIFT):
 942                                 self.focusPrevious()
 943                         else:
 944                                 self.focusNext()
 945                 if (ord('0') <= c <= ord('9') or c == ord('.')) and len(self._value) < self._maxLen:
 946                         self._value = self._value[0:self._selectPos] + chr(c) + self._value[self._selectPos:]
 947                         self._selectPos += 1
 948                         self._callback(self._value)
 949                 self._inCallback = False
 950
 951         def setFocus(self):
 952                 self._base._focus = self
 953                 self._selectPos = len(self._value)
 954                 return True
 955
 956 class glCheckbox(glGuiControl):
 957         def __init__(self, parent, value, pos, callback):
 958                 self._callback = callback
 959                 self._value = value
 960                 self._selectPos = 0
 961                 self._maxLen = 6
 962                 self._inCallback = False
 963                 super(glCheckbox, self).__init__(parent, pos)
 964
 965         def setValue(self, value):
 966                 if self._inCallback:
 967                         return
 968                 self._value = str(value)
 969
 970         def getValue(self):
 971                 return self._value
 972
 973         def getMinSize(self):
 974                 return 20, 20
 975
 976         def _getPixelPos(self):
 977                 x0, y0, w, h = self.getSize()
 978                 return x0, y0
 979
 980         def draw(self):
 981                 x, y, w, h = self.getSize()
 982
 983                 glPushMatrix()
 984                 glTranslatef(x, y, 0)
 985
 986                 glColor3ub(255,255,255)
 987                 if self._value:
 988                         openglHelpers.glDrawTexturedQuad(w/2-h/2,0, h, h, 28)
 989                 else:
 990                         openglHelpers.glDrawTexturedQuad(w/2-h/2,0, h, h, 29)
 991
 992                 glPopMatrix()
 993
 994         def _checkHit(self, x, y):
 995                 x1, y1, w, h = self.getSize()
 996                 return 0 <= x - x1 <= w and 0 <= y - y1 <= h
 997
 998         def OnMouseMotion(self, x, y):
 999                 return False
1000
1001         def OnMouseDown(self, x, y, button):
1002                 if self._checkHit(x, y):
1003                         self._value = not self._value
1004                         return True
1005                 return False
1006
1007 class glSlider(glGuiControl):
1008         def __init__(self, parent, value, minValue, maxValue, pos, callback):
1009                 super(glSlider, self).__init__(parent, pos)
1010                 self._callback = callback
1011                 self._focus = False
1012                 self._hidden = False
1013                 self._value = value
1014                 self._minValue = minValue
1015                 self._maxValue = maxValue
1016
1017         def setValue(self, value):
1018                 self._value = value
1019
1020         def getValue(self):
1021                 if self._value < self._minValue:
1022                         return self._minValue
1023                 if self._value > self._maxValue:
1024                         return self._maxValue
1025                 return self._value
1026
1027         def setRange(self, minValue, maxValue):
1028                 if maxValue < minValue:
1029                         maxValue = minValue
1030                 self._minValue = minValue
1031                 self._maxValue = maxValue
1032
1033         def getMinValue(self):
1034                 return self._minValue
1035
1036         def getMaxValue(self):
1037                 return self._maxValue
1038
1039         def setHidden(self, value):
1040                 self._hidden = value
1041
1042         def getMinSize(self):
1043                 return self._base._buttonSize * 0.2, self._base._buttonSize * 4
1044
1045         def _getPixelPos(self):
1046                 x0, y0, w, h = self.getSize()
1047                 minSize = self.getMinSize()
1048                 return x0 + w / 2 - minSize[0] / 2, y0 + h / 2 - minSize[1] / 2
1049
1050         def draw(self):
1051                 if self._hidden:
1052                         return
1053
1054                 w, h = self.getMinSize()
1055                 pos = self._getPixelPos()
1056
1057                 glPushMatrix()
1058                 glTranslatef(pos[0], pos[1]*1.05, 0)
1059                 glDisable(GL_TEXTURE_2D)
1060                 if self.hasFocus():
1061                         glColor4ub(177,205,54,255)
1062                 else:
1063                         glColor4ub(177,205,54,192)
1064                 glBegin(GL_QUADS)
1065                 glVertex2f( w/2,-h/2)
1066                 glVertex2f(-w/2,-h/2)
1067                 glVertex2f(-w/2, h/2)
1068                 glVertex2f( w/2, h/2)
1069                 glEnd()
1070                 scrollLength = h - w
1071                 if self._maxValue-self._minValue != 0:
1072                         valueNormalized = ((self.getValue()-self._minValue)/(self._maxValue-self._minValue))
1073                 else:
1074                         valueNormalized = 0
1075                 glTranslate(0.0,scrollLength/2,0)
1076                 if True:  # self._focus:
1077                         glColor4ub(90,104,27,255)
1078                         glPushMatrix()
1079                         glTranslate(-w/2,openglHelpers.glGetStringSize(str(self._minValue))[1]/2,0)
1080                         openglHelpers.glDrawStringRight(str(self._minValue))
1081                         glTranslate(0,-scrollLength,0)
1082                         openglHelpers.glDrawStringRight(str(self._maxValue))
1083                         glTranslate(w,scrollLength-scrollLength*valueNormalized,0)
1084                         openglHelpers.glDrawStringLeft(str(self.getValue()))
1085                         glPopMatrix()
1086                 glColor4ub(90,104,27,255)
1087                 glTranslate(0.0,-scrollLength*valueNormalized,0)
1088                 glBegin(GL_QUADS)
1089                 glVertex2f( w/2,-w/2)
1090                 glVertex2f(-w/2,-w/2)
1091                 glVertex2f(-w/2, w/2)
1092                 glVertex2f( w/2, w/2)
1093                 glEnd()
1094                 glPopMatrix()
1095
1096         def _checkHit(self, x, y):
1097                 if self._hidden:
1098                         return False
1099                 pos = self._getPixelPos()
1100                 w, h = self.getMinSize()
1101                 return -w/2 <= x - pos[0] <= w/2 and -h/2 <= y - pos[1] <= h/2
1102
1103         def setFocus(self):
1104                 self._base._focus = self
1105                 return True
1106
1107         def OnMouseMotion(self, x, y):
1108                 if self.hasFocus():
1109                         w, h = self.getMinSize()
1110                         scrollLength = h - w
1111                         pos = self._getPixelPos()
1112                         self.setValue(int(self._minValue + (self._maxValue - self._minValue) * -(y - pos[1]*1.05 - scrollLength/2) / scrollLength))
1113                         self._callback()
1114                         return True
1115                 if self._checkHit(x, y):
1116                         self._focus = True
1117                         return True
1118                 self._focus = False
1119                 return False
1120
1121         def OnMouseDown(self, x, y, button):
1122                 if self._checkHit(x, y):
1123                         self.setFocus()
1124                         self.OnMouseMotion(x, y)
1125                         return True
1126                 return False
1127
1128         def OnMouseUp(self, x, y):
1129                 if self.hasFocus():
1130                         self._base._focus = None
1131                         return True
1132                 return False