Cura/gui/util/openglGui.py

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