1 from __future__ import absolute_import
2 from __future__ import division
9 from wx import glcanvas
11 OpenGL.ERROR_CHECKING = False
12 from OpenGL.GL import *
14 from Cura.gui.util import opengl
16 class glGuiControl(object):
17 def __init__(self, parent, pos):
19 self._base = parent._base
21 self._size = (0,0, 1, 1)
22 self._parent.add(self)
24 def setSize(self, x, y, w, h):
25 self._size = (x, y, w, h)
33 def updateLayout(self):
37 for n in xrange(self._parent._glGuiControlList.index(self) + 1, len(self._parent._glGuiControlList)):
38 if self._parent._glGuiControlList[n].setFocus():
40 for n in xrange(0, self._parent._glGuiControlList.index(self)):
41 if self._parent._glGuiControlList[n].setFocus():
44 def focusPrevious(self):
45 for n in xrange(self._parent._glGuiControlList.index(self) -1, -1, -1):
46 if self._parent._glGuiControlList[n].setFocus():
48 for n in xrange(len(self._parent._glGuiControlList) - 1, self._parent._glGuiControlList.index(self), -1):
49 if self._parent._glGuiControlList[n].setFocus():
56 return self._base._focus == self
58 def OnMouseUp(self, x, y):
61 def OnKeyChar(self, key):
64 class glGuiContainer(glGuiControl):
65 def __init__(self, parent, pos):
66 self._glGuiControlList = []
67 glGuiLayoutButtons(self)
68 super(glGuiContainer, self).__init__(parent, pos)
71 self._glGuiControlList.append(ctrl)
74 def OnMouseDown(self, x, y):
75 for ctrl in self._glGuiControlList:
76 if ctrl.OnMouseDown(x, y):
80 def OnMouseUp(self, x, y):
81 for ctrl in self._glGuiControlList:
82 if ctrl.OnMouseUp(x, y):
86 def OnMouseMotion(self, x, y):
88 for ctrl in self._glGuiControlList:
89 if ctrl.OnMouseMotion(x, y):
94 for ctrl in self._glGuiControlList:
97 def updateLayout(self):
99 for ctrl in self._glGuiControlList:
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)
108 self._container = None
109 self._container = glGuiContainer(self, (0,0))
110 self._shownError = False
112 self._context = glcanvas.GLContext(self)
113 self._glButtonsTexture = None
114 self._glRobotTexture = None
115 self._buttonSize = 64
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)
126 def OnKeyChar(self, e):
127 if self._focus is not None:
128 self._focus.OnKeyChar(e.GetKeyCode())
131 def OnFocusLost(self, e):
135 def _OnGuiMouseLeftDown(self,e):
137 if self._container.OnMouseDown(e.GetX(), e.GetY()):
140 self.OnMouseLeftDown(e)
141 def _OnGuiMouseLeftUp(self, e):
142 if self._container.OnMouseUp(e.GetX(), e.GetY()):
145 self.OnMouseLeftUp(e)
147 def _OnGuiMouseMotion(self,e):
149 if not self._container.OnMouseMotion(e.GetX(), e.GetY()):
150 self.OnMouseMotion(e)
152 def _OnGuiPaint(self, e):
153 h = self.GetSize().GetHeight()
154 w = self.GetSize().GetWidth()
155 oldButtonSize = self._buttonSize
159 self._buttonSize = 32
161 self._buttonSize = 48
163 self._buttonSize = 64
165 self._buttonSize = 80
167 self._buttonSize = 96
168 if self._buttonSize != oldButtonSize:
169 self._container.updateLayout()
171 dc = wx.PaintDC(self)
173 self.SetCurrent(self._context)
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):
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
190 if self._glButtonsTexture is None:
191 self._glButtonsTexture = opengl.loadGLTexture('glButtons.png')
192 self._glRobotTexture = opengl.loadGLTexture('UltimakerRobot.png')
194 glDisable(GL_DEPTH_TEST)
196 glDisable(GL_LIGHTING)
197 glColor4ub(255,255,255,255)
199 glMatrixMode(GL_PROJECTION)
201 size = self.GetSize()
202 glOrtho(0, size.GetWidth()-1, size.GetHeight()-1, 0, -1000.0, 1000.0)
203 glMatrixMode(GL_MODELVIEW)
206 self._container.draw()
207 glBindTexture(GL_TEXTURE_2D, self._glRobotTexture)
208 glEnable(GL_TEXTURE_2D)
211 glTranslate(size.GetWidth() - 8,size.GetHeight() - 16,0)
212 s = self._buttonSize * 1.4
224 glDisable(GL_TEXTURE_2D)
227 def _OnEraseBackground(self,event):
228 #Workaround for windows background redraw flicker.
232 self._container.setSize(0, 0, self.GetSize().GetWidth(), self.GetSize().GetHeight())
233 self._container.updateLayout()
236 def OnMouseLeftDown(self,e):
238 def OnMouseLeftUp(self,e):
240 def OnMouseMotion(self, e):
242 def OnPaint(self, e):
246 if self._container is not None:
247 self._container.add(ctrl)
249 class glGuiLayoutButtons(object):
250 def __init__(self, parent):
251 self._parent = parent
252 self._parent._layout = self
255 bs = self._parent._base._buttonSize
256 x0, y0, w, h = self._parent.getSize()
258 for ctrl in self._parent._glGuiControlList:
261 x = w + pos[0] * gridSize - bs * 0.2
263 x = pos[0] * gridSize + bs * 0.2
265 y = h + pos[1] * gridSize * 1.2 - bs * 0.2
267 y = pos[1] * gridSize * 1.2 + bs * 0.2
268 ctrl.setSize(x, y, gridSize, gridSize)
270 def getLayoutSize(self):
271 _, _, w, h = self._parent.getSize()
274 class glGuiLayoutGrid(object):
275 def __init__(self, parent):
276 self._parent = parent
277 self._parent._layout = self
279 self._alignBottom = True
282 borderSize = self._parent._base._buttonSize * 0.2
283 x0, y0, w, h = self._parent.getSize()
288 for ctrl in self._parent._glGuiControlList:
290 w, h = ctrl.getMinSize()
294 widths[x] = max(widths[x], w)
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:
307 for n in xrange(0, x):
311 for n in xrange(0, y):
315 ctrl.setSize(x1, y1, widths[x], heights[y])
317 def getLayoutSize(self):
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
330 self._disabled = False
332 def setSelected(self, value):
333 self._selected = value
335 def setHidden(self, value):
338 def setDisabled(self, value):
339 self._disabled = value
341 def getSelected(self):
342 return self._selected
344 def getMinSize(self):
345 return self._base._buttonSize, self._base._buttonSize
347 def _getPixelPos(self):
348 x0, y0, w, h = self.getSize()
349 return x0 + w / 2, y0 + h / 2
355 cx = (self._imageID % 4) / 4
356 cy = int(self._imageID / 4) / 4
357 bs = self._base._buttonSize
358 pos = self._getPixelPos()
360 glBindTexture(GL_TEXTURE_2D, self._base._glButtonsTexture)
367 glColor4ub(128,128,128,128)
369 glColor4ub(255,255,255,255)
370 opengl.glDrawTexturedQuad(pos[0]-bs*scale/2, pos[1]-bs*scale/2, bs*scale, bs*scale, self._imageID)
372 glTranslatef(pos[0], pos[1], 0)
373 glDisable(GL_TEXTURE_2D)
375 glColor4ub(0,0,0,255)
376 glColor4ub(255,255,255,255)
377 glTranslatef(0, -0.55*bs*scale, 0)
378 opengl.glDrawStringCenter(self._tooltip)
381 def _checkHit(self, x, y):
382 if self._hidden or self._disabled:
384 bs = self.getMinSize()[0]
385 pos = self._getPixelPos()
386 return -bs * 0.5 <= x - pos[0] <= bs * 0.5 and -bs * 0.5 <= y - pos[1] <= bs * 0.5
388 def OnMouseMotion(self, x, y):
389 if self._checkHit(x, y):
395 def OnMouseDown(self, x, y):
396 if self._checkHit(x, y):
401 class glRadioButton(glButton):
402 def __init__(self, parent, imageID, tooltip, pos, group, callback):
403 super(glRadioButton, self).__init__(parent, imageID, tooltip, pos, self._onRadioSelect)
405 self._radioCallback = callback
406 self._group.append(self)
408 def setSelected(self, value):
409 self._selected = value
411 def _onRadioSelect(self):
412 self._base._focus = None
413 for ctrl in self._group:
415 ctrl.setSelected(False)
416 if self.getSelected():
417 self.setSelected(False)
419 self.setSelected(True)
420 self._radioCallback()
422 class glComboButton(glButton):
423 def __init__(self, parent, tooltip, imageIDs, tooltips, pos, callback):
424 super(glComboButton, self).__init__(parent, imageIDs[0], tooltip, pos, self._onComboOpenSelect)
425 self._imageIDs = imageIDs
426 self._tooltips = tooltips
427 self._comboCallback = callback
430 def _onComboOpenSelect(self):
432 self._base._focus = None
434 self._base._focus = self
439 self._selected = self.hasFocus()
440 super(glComboButton, self).draw()
442 bs = self._base._buttonSize / 2
443 pos = self._getPixelPos()
445 if not self._selected:
449 glTranslatef(pos[0]+bs*0.5, pos[1] + bs*0.5, 0)
450 glBindTexture(GL_TEXTURE_2D, self._base._glButtonsTexture)
452 for n in xrange(0, len(self._imageIDs)):
453 glTranslatef(0, 1, 0)
454 glColor3ub(255,255,255)
455 opengl.glDrawTexturedQuad(-0.5,-0.5,1,1, self._imageIDs[n])
456 glDisable(GL_TEXTURE_2D)
459 glColor4ub(0,0,0,255)
460 glTranslatef(-0.55, 0.1, 0)
461 opengl.glDrawStringRight(self._tooltips[n])
466 return self._selection
468 def setValue(self, value):
469 self._selection = value
470 self._comboCallback()
472 def OnMouseDown(self, x, y):
473 if self._hidden or self._disabled:
476 bs = self._base._buttonSize / 2
477 pos = self._getPixelPos()
478 if 0 <= x - pos[0] <= bs and 0 <= y - pos[1] - bs <= bs * len(self._imageIDs):
479 self._selection = int((y - pos[1] - bs) / bs)
480 self._imageID = self._imageIDs[self._selection]
481 self._base._focus = None
482 self._comboCallback()
484 return super(glComboButton, self).OnMouseDown(x, y)
486 class glFrame(glGuiContainer):
487 def __init__(self, parent, pos):
488 super(glFrame, self).__init__(parent, pos)
489 self._selected = False
493 def setSelected(self, value):
494 self._selected = value
496 def setHidden(self, value):
499 def getSelected(self):
500 return self._selected
502 def getMinSize(self):
503 return self._base._buttonSize, self._base._buttonSize
505 def _getPixelPos(self):
506 x0, y0, w, h = self.getSize()
513 bs = self._parent._buttonSize
514 pos = self._getPixelPos()
517 glTranslatef(pos[0], pos[1], 0)
518 glBindTexture(GL_TEXTURE_2D, self._base._glButtonsTexture)
519 glEnable(GL_TEXTURE_2D)
521 size = self._layout.getLayoutSize()
522 glColor4ub(255,255,255,255)
528 # glVertex2f( size[0], 0)
532 # glVertex2f( 0, size[1])
534 # glVertex2f( size[0], size[1])
540 glTexCoord2f(0, tc/2)
542 glTexCoord2f(tc, tc/2)
545 glTexCoord2f(tc+tc, 0)
546 glVertex2f( size[0], 0)
548 glVertex2f( size[0] - bs, 0)
549 glTexCoord2f(tc, tc/2)
550 glVertex2f( size[0] - bs, bs)
551 glTexCoord2f(tc+tc, tc/2)
552 glVertex2f( size[0], bs)
554 glTexCoord2f(tc, tc/2)
555 glVertex2f( bs, size[1] - bs)
556 glTexCoord2f(0, tc/2)
557 glVertex2f( 0, size[1] - bs)
558 glTexCoord2f(0, tc/2+tc/2)
559 glVertex2f( 0, size[1])
560 glTexCoord2f(tc, tc/2+tc/2)
561 glVertex2f( bs, size[1])
563 glTexCoord2f(tc+tc, tc/2)
564 glVertex2f( size[0], size[1] - bs)
565 glTexCoord2f(tc, tc/2)
566 glVertex2f( size[0] - bs, size[1] - bs)
567 glTexCoord2f(tc, tc/2+tc/2)
568 glVertex2f( size[0] - bs, size[1])
569 glTexCoord2f(tc+tc, tc/2+tc/2)
570 glVertex2f( size[0], size[1])
573 glTexCoord2f(tc, tc/2)
574 glVertex2f( size[0]-bs, bs)
575 glTexCoord2f(tc, tc/2)
577 glTexCoord2f(tc, tc/2)
578 glVertex2f( bs, size[1]-bs)
579 glTexCoord2f(tc, tc/2)
580 glVertex2f( size[0]-bs, size[1]-bs)
583 glTexCoord2f(tc+tc, tc/2)
584 glVertex2f( size[0], bs)
585 glTexCoord2f(tc, tc/2)
586 glVertex2f( size[0]-bs, bs)
587 glTexCoord2f(tc, tc/2)
588 glVertex2f( size[0]-bs, size[1]-bs)
589 glTexCoord2f(tc+tc, tc/2)
590 glVertex2f( size[0], size[1]-bs)
593 glTexCoord2f(tc, tc/2)
595 glTexCoord2f(0, tc/2)
597 glTexCoord2f(0, tc/2)
598 glVertex2f( 0, size[1]-bs)
599 glTexCoord2f(tc, tc/2)
600 glVertex2f( bs, size[1]-bs)
604 glVertex2f( size[0]-bs, 0)
607 glTexCoord2f(tc, tc/2)
609 glTexCoord2f(tc, tc/2)
610 glVertex2f( size[0]-bs, bs)
613 glTexCoord2f(tc, tc/2)
614 glVertex2f( size[0]-bs, size[1]-bs)
615 glTexCoord2f(tc, tc/2)
616 glVertex2f( bs, size[1]-bs)
617 glTexCoord2f(tc, tc/2+tc/2)
618 glVertex2f( bs, size[1])
619 glTexCoord2f(tc, tc/2+tc/2)
620 glVertex2f( size[0]-bs, size[1])
623 glDisable(GL_TEXTURE_2D)
625 #Draw the controls on the frame
626 super(glFrame, self).draw()
628 def _checkHit(self, x, y):
631 pos = self._getPixelPos()
632 w, h = self._layout.getLayoutSize()
633 return 0 <= x - pos[0] <= w and 0 <= y - pos[1] <= h
635 def OnMouseMotion(self, x, y):
636 super(glFrame, self).OnMouseMotion(x, y)
637 if self._checkHit(x, y):
643 def OnMouseDown(self, x, y):
644 if self._checkHit(x, y):
645 super(glFrame, self).OnMouseDown(x, y)
649 class glLabel(glGuiControl):
650 def __init__(self, parent, label, pos):
652 super(glLabel, self).__init__(parent, pos)
654 def getMinSize(self):
655 w, h = opengl.glGetStringSize(self._label)
658 def _getPixelPos(self):
659 x0, y0, w, h = self.getSize()
663 x, y, w, h = self.getSize()
666 glTranslatef(x, y, 0)
668 # glColor4ub(255,255,255,128)
680 glTranslate(5, h - 5, 0)
681 glColor4ub(255,255,255,255)
682 opengl.glDrawStringLeft(self._label)
685 def _checkHit(self, x, y):
688 def OnMouseMotion(self, x, y):
691 def OnMouseDown(self, x, y):
694 class glNumberCtrl(glGuiControl):
695 def __init__(self, parent, value, pos, callback):
696 self._callback = callback
697 self._value = str(value)
700 self._inCallback = False
701 super(glNumberCtrl, self).__init__(parent, pos)
703 def setValue(self, value):
706 self._value = str(value)
708 def getMinSize(self):
709 w, h = opengl.glGetStringSize("VALUES")
712 def _getPixelPos(self):
713 x0, y0, w, h = self.getSize()
717 x, y, w, h = self.getSize()
720 glTranslatef(x, y, 0)
723 glColor4ub(255,255,255,255)
725 glColor4ub(255,255,255,192)
737 glTranslate(5, h - 5, 0)
738 glColor4ub(0,0,0,255)
739 opengl.glDrawStringLeft(self._value)
741 glTranslate(opengl.glGetStringSize(self._value[0:self._selectPos])[0] - 2, -1, 0)
742 opengl.glDrawStringLeft('|')
745 def _checkHit(self, x, y):
746 x1, y1, w, h = self.getSize()
747 return 0 <= x - x1 <= w and 0 <= y - y1 <= h
749 def OnMouseMotion(self, x, y):
752 def OnMouseDown(self, x, y):
753 if self._checkHit(x, y):
758 def OnKeyChar(self, c):
759 self._inCallback = True
762 self._selectPos = max(0, self._selectPos)
763 if c == wx.WXK_RIGHT:
765 self._selectPos = min(self._selectPos, len(self._value))
768 value = float(self._value)
773 self._value = str(value)
774 self._callback(self._value)
777 value = float(self._value)
783 self._value = str(value)
784 self._callback(self._value)
785 if c == wx.WXK_BACK and self._selectPos > 0:
786 self._value = self._value[0:self._selectPos - 1] + self._value[self._selectPos:]
788 self._callback(self._value)
789 if c == wx.WXK_DELETE:
790 self._value = self._value[0:self._selectPos] + self._value[self._selectPos + 1:]
791 self._callback(self._value)
793 if wx.GetKeyState(wx.WXK_SHIFT):
797 if (ord('0') <= c <= ord('9') or c == ord('.')) and len(self._value) < self._maxLen:
798 self._value = self._value[0:self._selectPos] + chr(c) + self._value[self._selectPos:]
800 self._callback(self._value)
801 self._inCallback = False
804 self._base._focus = self
805 self._selectPos = len(self._value)
808 class glCheckbox(glGuiControl):
809 def __init__(self, parent, value, pos, callback):
810 self._callback = callback
814 self._inCallback = False
815 super(glCheckbox, self).__init__(parent, pos)
817 def setValue(self, value):
820 self._value = str(value)
825 def getMinSize(self):
828 def _getPixelPos(self):
829 x0, y0, w, h = self.getSize()
833 x, y, w, h = self.getSize()
836 glTranslatef(x, y, 0)
838 glColor3ub(255,255,255)
840 opengl.glDrawTexturedQuad(w/2-h/2,0, h, h, 28)
842 opengl.glDrawTexturedQuad(w/2-h/2,0, h, h, 29)
846 def _checkHit(self, x, y):
847 x1, y1, w, h = self.getSize()
848 return 0 <= x - x1 <= w and 0 <= y - y1 <= h
850 def OnMouseMotion(self, x, y):
853 def OnMouseDown(self, x, y):
854 if self._checkHit(x, y):
855 self._value = not self._value
859 class glSlider(glGuiControl):
860 def __init__(self, parent, value, minValue, maxValue, pos, callback):
861 super(glSlider, self).__init__(parent, pos)
862 self._callback = callback
866 self._minValue = minValue
867 self._maxValue = maxValue
869 def setValue(self, value):
871 self._value = max(self._minValue, self._value)
872 self._value = min(self._maxValue, self._value)
877 def setRange(self, minValue, maxValue):
878 if maxValue < minValue:
880 self._minValue = minValue
881 self._maxValue = maxValue
882 self._value = max(minValue, self._value)
883 self._value = min(maxValue, self._value)
885 def getMinValue(self):
886 return self._minValue
888 def getMaxValue(self):
889 return self._maxValue
891 def setHidden(self, value):
894 def getMinSize(self):
895 return self._base._buttonSize * 0.2, self._base._buttonSize * 4
897 def _getPixelPos(self):
898 x0, y0, w, h = self.getSize()
899 minSize = self.getMinSize()
900 return x0 + w / 2 - minSize[0] / 2, y0 + h / 2 - minSize[1] / 2
906 w, h = self.getMinSize()
907 pos = self._getPixelPos()
910 glTranslatef(pos[0], pos[1], 0)
911 glDisable(GL_TEXTURE_2D)
913 glColor4ub(60,60,60,255)
915 glColor4ub(60,60,60,192)
917 glVertex2f( w/2,-h/2)
918 glVertex2f(-w/2,-h/2)
919 glVertex2f(-w/2, h/2)
920 glVertex2f( w/2, h/2)
923 glTranslate(0.0,scrollLength/2,0)
925 glColor4ub(0,0,0,255)
927 glTranslate(-w/2,opengl.glGetStringSize(str(self._minValue))[1]/2,0)
928 opengl.glDrawStringRight(str(self._minValue))
929 glTranslate(0,-scrollLength,0)
930 opengl.glDrawStringRight(str(self._maxValue))
931 if self._maxValue-self._minValue > 0:
932 glTranslate(w,scrollLength-scrollLength*((self._value-self._minValue)/(self._maxValue-self._minValue)),0)
933 opengl.glDrawStringLeft(str(self._value))
935 glColor4ub(255,255,255,240)
936 if self._maxValue - self._minValue != 0:
937 glTranslate(0.0,-scrollLength*((self._value-self._minValue)/(self._maxValue-self._minValue)),0)
939 glVertex2f( w/2,-w/2)
940 glVertex2f(-w/2,-w/2)
941 glVertex2f(-w/2, w/2)
942 glVertex2f( w/2, w/2)
946 def _checkHit(self, x, y):
949 pos = self._getPixelPos()
950 w, h = self.getMinSize()
951 return -w/2 <= x - pos[0] <= w/2 and -h/2 <= y - pos[1] <= h/2
954 self._base._focus = self
957 def OnMouseMotion(self, x, y):
959 w, h = self.getMinSize()
961 pos = self._getPixelPos()
962 self.setValue(int(self._minValue + (self._maxValue - self._minValue) * -(y - pos[1] - scrollLength/2) / scrollLength))
965 if self._checkHit(x, y):
971 def OnMouseDown(self, x, y):
972 if self._checkHit(x, y):
974 self.OnMouseMotion(x, y)
978 def OnMouseUp(self, x, y):
980 self._base._focus = None