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(255,255,255,255)
376 glTranslatef(0, -0.55*bs*scale, 0)
377 opengl.glDrawStringCenter(self._tooltip)
380 def _checkHit(self, x, y):
381 if self._hidden or self._disabled:
383 bs = self.getMinSize()[0]
384 pos = self._getPixelPos()
385 return -bs * 0.5 <= x - pos[0] <= bs * 0.5 and -bs * 0.5 <= y - pos[1] <= bs * 0.5
387 def OnMouseMotion(self, x, y):
388 if self._checkHit(x, y):
394 def OnMouseDown(self, x, y):
395 if self._checkHit(x, y):
400 class glRadioButton(glButton):
401 def __init__(self, parent, imageID, tooltip, pos, group, callback):
402 super(glRadioButton, self).__init__(parent, imageID, tooltip, pos, self._onRadioSelect)
404 self._radioCallback = callback
405 self._group.append(self)
407 def setSelected(self, value):
408 self._selected = value
410 def _onRadioSelect(self):
411 self._base._focus = None
412 for ctrl in self._group:
414 ctrl.setSelected(False)
415 if self.getSelected():
416 self.setSelected(False)
418 self.setSelected(True)
419 self._radioCallback()
421 class glComboButton(glButton):
422 def __init__(self, parent, tooltip, imageIDs, tooltips, pos, callback):
423 super(glComboButton, self).__init__(parent, imageIDs[0], tooltip, pos, self._onComboOpenSelect)
424 self._imageIDs = imageIDs
425 self._tooltips = tooltips
426 self._comboCallback = callback
429 def _onComboOpenSelect(self):
431 self._base._focus = None
433 self._base._focus = self
438 self._selected = self.hasFocus()
439 super(glComboButton, self).draw()
441 bs = self._base._buttonSize / 2
442 pos = self._getPixelPos()
444 if not self._selected:
448 glTranslatef(pos[0]+bs*0.5, pos[1] + bs*0.5, 0)
449 glBindTexture(GL_TEXTURE_2D, self._base._glButtonsTexture)
451 for n in xrange(0, len(self._imageIDs)):
452 glTranslatef(0, 1, 0)
453 glColor4ub(255,255,255,255)
454 opengl.glDrawTexturedQuad(-0.5,-0.5,1,1, self._imageIDs[n])
455 glDisable(GL_TEXTURE_2D)
458 glColor4ub(255,255,255,255)
459 glTranslatef(-0.55, 0.1, 0)
460 opengl.glDrawStringRight(self._tooltips[n])
465 return self._selection
467 def setValue(self, value):
468 self._selection = value
469 self._comboCallback()
471 def OnMouseDown(self, x, y):
472 if self._hidden or self._disabled:
475 bs = self._base._buttonSize / 2
476 pos = self._getPixelPos()
477 if 0 <= x - pos[0] <= bs and 0 <= y - pos[1] - bs <= bs * len(self._imageIDs):
478 self._selection = int((y - pos[1] - bs) / bs)
479 self._imageID = self._imageIDs[self._selection]
480 self._base._focus = None
481 self._comboCallback()
483 return super(glComboButton, self).OnMouseDown(x, y)
485 class glFrame(glGuiContainer):
486 def __init__(self, parent, pos):
487 super(glFrame, self).__init__(parent, pos)
488 self._selected = False
492 def setSelected(self, value):
493 self._selected = value
495 def setHidden(self, value):
498 def getSelected(self):
499 return self._selected
501 def getMinSize(self):
502 return self._base._buttonSize, self._base._buttonSize
504 def _getPixelPos(self):
505 x0, y0, w, h = self.getSize()
512 bs = self._parent._buttonSize
513 pos = self._getPixelPos()
516 glTranslatef(pos[0], pos[1], 0)
517 glBindTexture(GL_TEXTURE_2D, self._base._glButtonsTexture)
518 glEnable(GL_TEXTURE_2D)
520 size = self._layout.getLayoutSize()
521 glColor4ub(255,255,255,255)
527 # glVertex2f( size[0], 0)
531 # glVertex2f( 0, size[1])
533 # glVertex2f( size[0], size[1])
539 glTexCoord2f(0, tc/2)
541 glTexCoord2f(tc, tc/2)
544 glTexCoord2f(tc+tc, 0)
545 glVertex2f( size[0], 0)
547 glVertex2f( size[0] - bs, 0)
548 glTexCoord2f(tc, tc/2)
549 glVertex2f( size[0] - bs, bs)
550 glTexCoord2f(tc+tc, tc/2)
551 glVertex2f( size[0], bs)
553 glTexCoord2f(tc, tc/2)
554 glVertex2f( bs, size[1] - bs)
555 glTexCoord2f(0, tc/2)
556 glVertex2f( 0, size[1] - bs)
557 glTexCoord2f(0, tc/2+tc/2)
558 glVertex2f( 0, size[1])
559 glTexCoord2f(tc, tc/2+tc/2)
560 glVertex2f( bs, size[1])
562 glTexCoord2f(tc+tc, tc/2)
563 glVertex2f( size[0], size[1] - bs)
564 glTexCoord2f(tc, tc/2)
565 glVertex2f( size[0] - bs, size[1] - bs)
566 glTexCoord2f(tc, tc/2+tc/2)
567 glVertex2f( size[0] - bs, size[1])
568 glTexCoord2f(tc+tc, tc/2+tc/2)
569 glVertex2f( size[0], size[1])
572 glTexCoord2f(tc, tc/2)
573 glVertex2f( size[0]-bs, bs)
574 glTexCoord2f(tc, tc/2)
576 glTexCoord2f(tc, tc/2)
577 glVertex2f( bs, size[1]-bs)
578 glTexCoord2f(tc, tc/2)
579 glVertex2f( size[0]-bs, size[1]-bs)
582 glTexCoord2f(tc+tc, tc/2)
583 glVertex2f( size[0], bs)
584 glTexCoord2f(tc, tc/2)
585 glVertex2f( size[0]-bs, bs)
586 glTexCoord2f(tc, tc/2)
587 glVertex2f( size[0]-bs, size[1]-bs)
588 glTexCoord2f(tc+tc, tc/2)
589 glVertex2f( size[0], size[1]-bs)
592 glTexCoord2f(tc, tc/2)
594 glTexCoord2f(0, tc/2)
596 glTexCoord2f(0, tc/2)
597 glVertex2f( 0, size[1]-bs)
598 glTexCoord2f(tc, tc/2)
599 glVertex2f( bs, size[1]-bs)
603 glVertex2f( size[0]-bs, 0)
606 glTexCoord2f(tc, tc/2)
608 glTexCoord2f(tc, tc/2)
609 glVertex2f( size[0]-bs, bs)
612 glTexCoord2f(tc, tc/2)
613 glVertex2f( size[0]-bs, size[1]-bs)
614 glTexCoord2f(tc, tc/2)
615 glVertex2f( bs, size[1]-bs)
616 glTexCoord2f(tc, tc/2+tc/2)
617 glVertex2f( bs, size[1])
618 glTexCoord2f(tc, tc/2+tc/2)
619 glVertex2f( size[0]-bs, size[1])
622 glDisable(GL_TEXTURE_2D)
624 #Draw the controls on the frame
625 super(glFrame, self).draw()
627 def _checkHit(self, x, y):
630 pos = self._getPixelPos()
631 w, h = self._layout.getLayoutSize()
632 return 0 <= x - pos[0] <= w and 0 <= y - pos[1] <= h
634 def OnMouseMotion(self, x, y):
635 super(glFrame, self).OnMouseMotion(x, y)
636 if self._checkHit(x, y):
642 def OnMouseDown(self, x, y):
643 if self._checkHit(x, y):
644 super(glFrame, self).OnMouseDown(x, y)
648 class glLabel(glGuiControl):
649 def __init__(self, parent, label, pos):
651 super(glLabel, self).__init__(parent, pos)
653 def getMinSize(self):
654 w, h = opengl.glGetStringSize(self._label)
657 def _getPixelPos(self):
658 x0, y0, w, h = self.getSize()
662 x, y, w, h = self.getSize()
665 glTranslatef(x, y, 0)
667 # glColor4ub(255,255,255,128)
679 glTranslate(5, h - 5, 0)
680 glColor4ub(255,255,255,255)
681 opengl.glDrawStringLeft(self._label)
684 def _checkHit(self, x, y):
687 def OnMouseMotion(self, x, y):
690 def OnMouseDown(self, x, y):
693 class glNumberCtrl(glGuiControl):
694 def __init__(self, parent, value, pos, callback):
695 self._callback = callback
696 self._value = str(value)
699 self._inCallback = False
700 super(glNumberCtrl, self).__init__(parent, pos)
702 def setValue(self, value):
705 self._value = str(value)
707 def getMinSize(self):
708 w, h = opengl.glGetStringSize("VALUES")
711 def _getPixelPos(self):
712 x0, y0, w, h = self.getSize()
716 x, y, w, h = self.getSize()
719 glTranslatef(x, y, 0)
722 glColor4ub(255,255,255,255)
724 glColor4ub(255,255,255,192)
736 glTranslate(5, h - 5, 0)
737 glColor4ub(0,0,0,255)
738 opengl.glDrawStringLeft(self._value)
740 glTranslate(opengl.glGetStringSize(self._value[0:self._selectPos])[0] - 2, -1, 0)
741 opengl.glDrawStringLeft('|')
744 def _checkHit(self, x, y):
745 x1, y1, w, h = self.getSize()
746 return 0 <= x - x1 <= w and 0 <= y - y1 <= h
748 def OnMouseMotion(self, x, y):
751 def OnMouseDown(self, x, y):
752 if self._checkHit(x, y):
757 def OnKeyChar(self, c):
758 self._inCallback = True
761 self._selectPos = max(0, self._selectPos)
762 if c == wx.WXK_RIGHT:
764 self._selectPos = min(self._selectPos, len(self._value))
767 value = float(self._value)
772 self._value = str(value)
773 self._callback(self._value)
776 value = float(self._value)
782 self._value = str(value)
783 self._callback(self._value)
784 if c == wx.WXK_BACK and self._selectPos > 0:
785 self._value = self._value[0:self._selectPos - 1] + self._value[self._selectPos:]
787 self._callback(self._value)
788 if c == wx.WXK_DELETE:
789 self._value = self._value[0:self._selectPos] + self._value[self._selectPos + 1:]
790 self._callback(self._value)
792 if wx.GetKeyState(wx.WXK_SHIFT):
796 if (ord('0') <= c <= ord('9') or c == ord('.')) and len(self._value) < self._maxLen:
797 self._value = self._value[0:self._selectPos] + chr(c) + self._value[self._selectPos:]
799 self._callback(self._value)
800 self._inCallback = False
803 self._base._focus = self
804 self._selectPos = len(self._value)
807 class glCheckbox(glGuiControl):
808 def __init__(self, parent, value, pos, callback):
809 self._callback = callback
813 self._inCallback = False
814 super(glCheckbox, self).__init__(parent, pos)
816 def setValue(self, value):
819 self._value = str(value)
824 def getMinSize(self):
827 def _getPixelPos(self):
828 x0, y0, w, h = self.getSize()
832 x, y, w, h = self.getSize()
835 glTranslatef(x, y, 0)
837 glColor3ub(255,255,255)
839 opengl.glDrawTexturedQuad(w/2-h/2,0, h, h, 28)
841 opengl.glDrawTexturedQuad(w/2-h/2,0, h, h, 29)
845 def _checkHit(self, x, y):
846 x1, y1, w, h = self.getSize()
847 return 0 <= x - x1 <= w and 0 <= y - y1 <= h
849 def OnMouseMotion(self, x, y):
852 def OnMouseDown(self, x, y):
853 if self._checkHit(x, y):
854 self._value = not self._value
858 class glSlider(glGuiControl):
859 def __init__(self, parent, value, minValue, maxValue, pos, callback):
860 super(glSlider, self).__init__(parent, pos)
861 self._callback = callback
865 self._minValue = minValue
866 self._maxValue = maxValue
868 def setValue(self, value):
870 self._value = max(self._minValue, self._value)
871 self._value = min(self._maxValue, self._value)
876 def setRange(self, minValue, maxValue):
877 if maxValue < minValue:
879 self._minValue = minValue
880 self._maxValue = maxValue
881 self._value = max(minValue, self._value)
882 self._value = min(maxValue, self._value)
884 def getMinValue(self):
885 return self._minValue
887 def getMaxValue(self):
888 return self._maxValue
890 def setHidden(self, value):
893 def getMinSize(self):
894 return self._base._buttonSize * 0.2, self._base._buttonSize * 4
896 def _getPixelPos(self):
897 x0, y0, w, h = self.getSize()
898 minSize = self.getMinSize()
899 return x0 + w / 2 - minSize[0] / 2, y0 + h / 2 - minSize[1] / 2
905 w, h = self.getMinSize()
906 pos = self._getPixelPos()
909 glTranslatef(pos[0], pos[1], 0)
910 glDisable(GL_TEXTURE_2D)
912 glColor4ub(60,60,60,255)
914 glColor4ub(60,60,60,192)
916 glVertex2f( w/2,-h/2)
917 glVertex2f(-w/2,-h/2)
918 glVertex2f(-w/2, h/2)
919 glVertex2f( w/2, h/2)
922 glTranslate(0.0,scrollLength/2,0)
924 glColor4ub(0,0,0,255)
926 glTranslate(-w/2,opengl.glGetStringSize(str(self._minValue))[1]/2,0)
927 opengl.glDrawStringRight(str(self._minValue))
928 glTranslate(0,-scrollLength,0)
929 opengl.glDrawStringRight(str(self._maxValue))
930 if self._maxValue-self._minValue > 0:
931 glTranslate(w,scrollLength-scrollLength*((self._value-self._minValue)/(self._maxValue-self._minValue)),0)
932 opengl.glDrawStringLeft(str(self._value))
934 glColor4ub(255,255,255,240)
935 if self._maxValue - self._minValue != 0:
936 glTranslate(0.0,-scrollLength*((self._value-self._minValue)/(self._maxValue-self._minValue)),0)
938 glVertex2f( w/2,-w/2)
939 glVertex2f(-w/2,-w/2)
940 glVertex2f(-w/2, w/2)
941 glVertex2f( w/2, w/2)
945 def _checkHit(self, x, y):
948 pos = self._getPixelPos()
949 w, h = self.getMinSize()
950 return -w/2 <= x - pos[0] <= w/2 and -h/2 <= y - pos[1] <= h/2
953 self._base._focus = self
956 def OnMouseMotion(self, x, y):
958 w, h = self.getMinSize()
960 pos = self._getPixelPos()
961 self.setValue(int(self._minValue + (self._maxValue - self._minValue) * -(y - pos[1] - scrollLength/2) / scrollLength))
964 if self._checkHit(x, y):
970 def OnMouseDown(self, x, y):
971 if self._checkHit(x, y):
973 self.OnMouseMotion(x, y)
977 def OnMouseUp(self, x, y):
979 self._base._focus = None