1 __copyright__ = "Copyright (C) 2013 David Braam - Released under terms of the AGPLv3 License"
9 from wx.lib import buttons
11 from Cura.util import profile
12 from Cura.util import resources
13 from Cura.gui.util import webcam
15 class printWindow(wx.Frame):
16 "Main user interface window"
18 def __init__(self, printerConnection):
19 super(printWindow, self).__init__(None, -1, title=_("Printing"))
20 self._printerConnection = printerConnection
21 self.lastUpdateTime = 0
23 self.SetSizer(wx.BoxSizer())
24 self.panel = wx.Panel(self)
25 self.GetSizer().Add(self.panel, 1, flag=wx.EXPAND)
26 self.sizer = wx.GridBagSizer(2, 2)
27 self.panel.SetSizer(self.sizer)
29 sb = wx.StaticBox(self.panel, label=_("Statistics"))
30 boxsizer = wx.StaticBoxSizer(sb, wx.VERTICAL)
32 self.powerWarningText = wx.StaticText(parent=self.panel,
34 label=_("Your computer is running on battery power.\nConnect your computer to AC power or your print might not finish."),
35 style=wx.ALIGN_CENTER)
36 self.powerWarningText.SetBackgroundColour('red')
37 self.powerWarningText.SetForegroundColour('white')
38 boxsizer.AddF(self.powerWarningText, flags=wx.SizerFlags().Expand().Border(wx.BOTTOM, 10))
39 self.powerManagement = power.PowerManagement()
40 self.powerWarningTimer = wx.Timer(self)
41 self.Bind(wx.EVT_TIMER, self.OnPowerWarningChange, self.powerWarningTimer)
42 self.OnPowerWarningChange(None)
43 self.powerWarningTimer.Start(10000)
45 self.statsText = wx.StaticText(self.panel, -1, _("Filament: ####.##m #.##g\nEstimated print time: #####:##\nMachine state:\nDetecting baudrateXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX"))
46 boxsizer.Add(self.statsText, flag=wx.LEFT, border=5)
48 self.sizer.Add(boxsizer, pos=(0, 0), span=(7, 1), flag=wx.EXPAND)
50 self.connectButton = wx.Button(self.panel, -1, _("Connect"))
51 #self.loadButton = wx.Button(self.panel, -1, 'Load')
52 self.printButton = wx.Button(self.panel, -1, _("Print"))
53 self.pauseButton = wx.Button(self.panel, -1, _("Pause"))
54 self.cancelButton = wx.Button(self.panel, -1, _("Cancel print"))
55 self.errorLogButton = wx.Button(self.panel, -1, _("Error log"))
56 self.progress = wx.Gauge(self.panel, -1, range=1000)
58 self.sizer.Add(self.connectButton, pos=(1, 1), flag=wx.EXPAND)
59 #self.sizer.Add(self.loadButton, pos=(1,1), flag=wx.EXPAND)
60 self.sizer.Add(self.printButton, pos=(2, 1), flag=wx.EXPAND)
61 self.sizer.Add(self.pauseButton, pos=(3, 1), flag=wx.EXPAND)
62 self.sizer.Add(self.cancelButton, pos=(4, 1), flag=wx.EXPAND)
63 self.sizer.Add(self.errorLogButton, pos=(5, 1), flag=wx.EXPAND)
64 self.sizer.Add(self.progress, pos=(7, 0), span=(1, 7), flag=wx.EXPAND)
66 nb = wx.Notebook(self.panel)
68 self.sizer.Add(nb, pos=(0, 2), span=(7, 4), flag=wx.EXPAND)
70 self.temperaturePanel = wx.Panel(nb)
71 sizer = wx.GridBagSizer(2, 2)
72 self.temperaturePanel.SetSizer(sizer)
74 self.temperatureSelect = wx.SpinCtrl(self.temperaturePanel, -1, '0', size=(21 * 3, 21), style=wx.SP_ARROW_KEYS)
75 self.temperatureSelect.SetRange(0, 400)
76 self.temperatureHeatUp = wx.Button(self.temperaturePanel, -1, str(int(profile.getProfileSettingFloat('print_temperature'))) + 'C')
77 self.bedTemperatureLabel = wx.StaticText(self.temperaturePanel, -1, _("BedTemp:"))
78 self.bedTemperatureSelect = wx.SpinCtrl(self.temperaturePanel, -1, '0', size=(21 * 3, 21), style=wx.SP_ARROW_KEYS)
79 self.bedTemperatureSelect.SetRange(0, 400)
80 self.bedTemperatureLabel.Show(False)
81 self.bedTemperatureSelect.Show(False)
83 self.temperatureGraph = TemperatureGraph(self.temperaturePanel)
85 sizer.Add(wx.StaticText(self.temperaturePanel, -1, _("Temp:")), pos=(0, 0))
86 sizer.Add(self.temperatureSelect, pos=(0, 1))
87 sizer.Add(self.temperatureHeatUp, pos=(0, 2))
88 sizer.Add(self.bedTemperatureLabel, pos=(1, 0))
89 sizer.Add(self.bedTemperatureSelect, pos=(1, 1))
90 sizer.Add(self.temperatureGraph, pos=(2, 0), span=(1, 3), flag=wx.EXPAND)
91 sizer.AddGrowableRow(2)
92 sizer.AddGrowableCol(2)
94 nb.AddPage(self.temperaturePanel, 'Temp')
96 self.directControlPanel = wx.Panel(nb)
98 sizer = wx.GridBagSizer(2, 2)
99 self.directControlPanel.SetSizer(sizer)
100 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Y100 F6000', 'G90'], 'print-move-y100.png'), pos=(0, 3))
101 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Y10 F6000', 'G90'], 'print-move-y10.png'), pos=(1, 3))
102 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Y1 F6000', 'G90'], 'print-move-y1.png'), pos=(2, 3))
104 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Y-1 F6000', 'G90'], 'print-move-y-1.png'), pos=(4, 3))
105 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Y-10 F6000', 'G90'], 'print-move-y-10.png'), pos=(5, 3))
106 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Y-100 F6000', 'G90'], 'print-move-y-100.png'), pos=(6, 3))
108 sizer.Add(PrintCommandButton(self, ['G91', 'G1 X-100 F6000', 'G90'], 'print-move-x-100.png'), pos=(3, 0))
109 sizer.Add(PrintCommandButton(self, ['G91', 'G1 X-10 F6000', 'G90'], 'print-move-x-10.png'), pos=(3, 1))
110 sizer.Add(PrintCommandButton(self, ['G91', 'G1 X-1 F6000', 'G90'], 'print-move-x-1.png'), pos=(3, 2))
112 sizer.Add(PrintCommandButton(self, ['G28 X0 Y0'], 'print-move-home.png'), pos=(3, 3))
114 sizer.Add(PrintCommandButton(self, ['G91', 'G1 X1 F6000', 'G90'], 'print-move-x1.png'), pos=(3, 4))
115 sizer.Add(PrintCommandButton(self, ['G91', 'G1 X10 F6000', 'G90'], 'print-move-x10.png'), pos=(3, 5))
116 sizer.Add(PrintCommandButton(self, ['G91', 'G1 X100 F6000', 'G90'], 'print-move-x100.png'), pos=(3, 6))
118 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Z10 F200', 'G90'], 'print-move-z10.png'), pos=(0, 8))
119 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Z1 F200', 'G90'], 'print-move-z1.png'), pos=(1, 8))
120 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Z0.1 F200', 'G90'], 'print-move-z0.1.png'), pos=(2, 8))
122 sizer.Add(PrintCommandButton(self, ['G28 Z0'], 'print-move-home.png'), pos=(3, 8))
124 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Z-0.1 F200', 'G90'], 'print-move-z-0.1.png'), pos=(4, 8))
125 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Z-1 F200', 'G90'], 'print-move-z-1.png'), pos=(5, 8))
126 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Z-10 F200', 'G90'], 'print-move-z-10.png'), pos=(6, 8))
128 sizer.Add(PrintCommandButton(self, ['G92 E0', 'G1 E2 F120'], 'extrude.png', size=(60, 20)), pos=(1, 10), span=(1, 3), flag=wx.EXPAND)
129 sizer.Add(PrintCommandButton(self, ['G92 E0', 'G1 E-2 F120'], 'retract.png', size=(60, 20)), pos=(2, 10), span=(1, 3), flag=wx.EXPAND)
131 nb.AddPage(self.directControlPanel, _("Jog"))
133 self.termPanel = wx.Panel(nb)
134 sizer = wx.GridBagSizer(2, 2)
135 self.termPanel.SetSizer(sizer)
137 f = wx.Font(8, wx.FONTFAMILY_MODERN, wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_NORMAL, False)
138 self.termLog = wx.TextCtrl(self.termPanel, style=wx.TE_MULTILINE | wx.TE_DONTWRAP)
139 self.termLog.SetFont(f)
140 self.termLog.SetEditable(0)
141 self.termInput = wx.TextCtrl(self.termPanel, style=wx.TE_PROCESS_ENTER)
142 self.termInput.SetFont(f)
143 self._termHistory = []
144 self._termHistoryIdx = 0
146 sizer.Add(self.termLog, pos=(0, 0), flag=wx.EXPAND)
147 sizer.Add(self.termInput, pos=(1, 0), flag=wx.EXPAND)
148 sizer.AddGrowableCol(0)
149 sizer.AddGrowableRow(0)
151 nb.AddPage(self.termPanel, _("Term"))
153 self.sizer.AddGrowableRow(6)
154 self.sizer.AddGrowableCol(3)
156 self.Bind(wx.EVT_CLOSE, self.OnClose)
157 self.connectButton.Bind(wx.EVT_BUTTON, self.OnConnect)
158 #self.loadButton.Bind(wx.EVT_BUTTON, self.OnLoad)
159 self.printButton.Bind(wx.EVT_BUTTON, self.OnPrint)
160 self.pauseButton.Bind(wx.EVT_BUTTON, self.OnPause)
161 self.cancelButton.Bind(wx.EVT_BUTTON, self.OnCancel)
162 self.errorLogButton.Bind(wx.EVT_BUTTON, self.OnErrorLog)
164 self.Bind(wx.EVT_BUTTON, lambda e: (self.temperatureSelect.SetValue(int(profile.getProfileSettingFloat('print_temperature'))), self.machineCom.sendCommand("M104 S%d" % (int(profile.getProfileSettingFloat('print_temperature'))))), self.temperatureHeatUp)
165 self.Bind(wx.EVT_SPINCTRL, self.OnTempChange, self.temperatureSelect)
166 self.Bind(wx.EVT_SPINCTRL, self.OnBedTempChange, self.bedTemperatureSelect)
168 self.Bind(wx.EVT_TEXT_ENTER, self.OnTermEnterLine, self.termInput)
169 self.termInput.Bind(wx.EVT_CHAR, self.OnTermKey)
175 self.statsText.SetMinSize(self.statsText.GetSize())
176 self.statsText.SetLabel(self._printerConnection.getStatusString())
178 self.UpdateButtonStates()
180 if webcam.hasWebcamSupport():
181 #Need to call the camera class on the GUI thread, or else it won't work. Shame as it hangs the GUI for about 2 seconds.
182 wx.CallAfter(self._webcamCheck)
184 self._printerConnection.addCallback(self._doPrinterConnectionUpdate)
186 def _webcamCheck(self):
187 self.cam = webcam.webcam()
188 if self.cam.hasCamera():
189 self.camPage = wx.Panel(self.tabs)
190 sizer = wx.GridBagSizer(2, 2)
191 self.camPage.SetSizer(sizer)
193 self.timelapsEnable = wx.CheckBox(self.camPage, -1, _("Enable timelapse movie recording"))
194 self.timelapsSavePath = wx.TextCtrl(self.camPage, -1, os.path.expanduser('~/timelaps_' + datetime.datetime.now().strftime('%Y-%m-%d_%H:%M') + '.mpg'))
195 sizer.Add(self.timelapsEnable, pos=(0, 0), span=(1, 2), flag=wx.EXPAND)
196 sizer.Add(self.timelapsSavePath, pos=(1, 0), span=(1, 2), flag=wx.EXPAND)
198 pages = self.cam.propertyPages()
199 self.cam.buttons = [self.timelapsEnable, self.timelapsSavePath]
201 button = wx.Button(self.camPage, -1, page)
202 button.index = pages.index(page)
203 sizer.Add(button, pos=(2, pages.index(page)))
204 button.Bind(wx.EVT_BUTTON, self.OnWebcamPropertyPageButton)
205 self.cam.buttons.append(button)
207 self.campreviewEnable = wx.CheckBox(self.camPage, -1, _("Show preview"))
208 sizer.Add(self.campreviewEnable, pos=(3, 0), span=(1, 2), flag=wx.EXPAND)
210 self.camPreview = wx.Panel(self.camPage)
211 sizer.Add(self.camPreview, pos=(4, 0), span=(1, 2), flag=wx.EXPAND)
213 self.tabs.AddPage(self.camPage, _("Camera"))
214 self.camPreview.timer = wx.Timer(self)
215 self.Bind(wx.EVT_TIMER, self.OnCameraTimer, self.camPreview.timer)
216 self.camPreview.timer.Start(500)
217 self.camPreview.Bind(wx.EVT_ERASE_BACKGROUND, self.OnCameraEraseBackground)
221 def OnPowerWarningChange(self, e):
222 type = self.powerManagement.get_providing_power_source_type()
223 if type == power.POWER_TYPE_AC and self.powerWarningText.IsShown():
224 self.powerWarningText.Hide()
227 elif type != power.POWER_TYPE_AC and not self.powerWarningText.IsShown():
228 self.powerWarningText.Show()
232 def OnClose(self, e):
233 self._printerConnection.closeActiveConnection()
234 self._printerConnection.removeCallback(self._doPrinterConnectionUpdate)
237 def OnConnect(self, e):
238 self._printerConnection.openActiveConnection()
243 def OnPrint(self, e):
244 self._printerConnection.startPrint()
246 def OnCancel(self, e):
247 self._printerConnection.cancelPrint()
249 def OnPause(self, e):
250 self._printerConnection.pause(not self._printerConnection.isPaused())
252 def OnErrorLog(self, e):
253 LogWindow(self._printerConnection.getErrorLog())
255 def OnTempChange(self, e):
256 if not self._printerConnection.isAbleToSendDirectCommand():
258 self._printerConnection.sendCommand("M104 S%d" % (self.temperatureSelect.GetValue()))
260 def OnBedTempChange(self, e):
261 if not self._printerConnection.isAbleToSendDirectCommand():
263 self._printerConnection.sendCommand("M140 S%d" % (self.bedTemperatureSelect.GetValue()))
265 def OnTermEnterLine(self, e):
266 if not self._printerConnection.isAbleToSendDirectCommand():
268 line = self.termInput.GetValue()
271 self._addTermLog('>%s\n' % (line))
272 self._printerConnection.sendCommand(line)
273 self._termHistory.append(line)
274 self._termHistoryIdx = len(self._termHistory)
275 self.termInput.SetValue('')
277 def OnTermKey(self, e):
278 if len(self._termHistory) > 0:
279 if e.GetKeyCode() == wx.WXK_UP:
280 self._termHistoryIdx -= 1
281 if self._termHistoryIdx < 0:
282 self._termHistoryIdx = len(self._termHistory) - 1
283 self.termInput.SetValue(self._termHistory[self._termHistoryIdx])
284 if e.GetKeyCode() == wx.WXK_DOWN:
285 self._termHistoryIdx -= 1
286 if self._termHistoryIdx >= len(self._termHistory):
287 self._termHistoryIdx = 0
288 self.termInput.SetValue(self._termHistory[self._termHistoryIdx])
291 def _addTermLog(self, line):
292 if len(self.termLog.GetValue()) > 10000:
293 self.termLog.SetValue(self.termLog.GetValue()[-10000:])
294 self.termLog.SetInsertionPointEnd()
295 self.termLog.AppendText(unicode(line, 'utf-8', 'replace'))
297 def _doPrinterConnectionUpdate(self, connection, extraInfo = None):
298 wx.CallAfter(self.__doPrinterConnectionUpdate, connection, extraInfo)
299 def __doPrinterConnectionUpdate(self, connection, extraInfo):
301 if self.lastUpdateTime + 0.5 > t:
303 self.lastUpdateTime = t
305 self.UpdateButtonStates()
306 if connection.isPrinting():
307 self.progress.SetValue(connection.getPrintProgress() * 1000)
309 self.progress.SetValue(0)
310 self.statsText.SetLabel(connection.getStatusString())
312 def UpdateButtonStates(self):
313 self.connectButton.Show(self._printerConnection.hasActiveConnection())
314 self.connectButton.Enable(not self._printerConnection.isActiveConnectionOpen())
315 self.pauseButton.Show(self._printerConnection.hasPause())
316 if not self._printerConnection.hasActiveConnection() or self._printerConnection.isActiveConnectionOpen():
317 self.printButton.Enable(not self._printerConnection.isPrinting())
318 self.pauseButton.Enable(self._printerConnection.isPrinting())
319 self.cancelButton.Enable(self._printerConnection.isPrinting())
321 self.printButton.Enable(False)
322 self.pauseButton.Enable(False)
323 self.cancelButton.Enable(False)
324 self.errorLogButton.Show(self._printerConnection.isInErrorState())
326 self.directControlPanel.Enable(self._printerConnection.isAbleToSendDirectCommand())
327 self.temperatureSelect.Enable(self._printerConnection.isAbleToSendDirectCommand())
328 self.temperatureHeatUp.Enable(self._printerConnection.isAbleToSendDirectCommand())
329 self.termInput.Enable(self._printerConnection.isAbleToSendDirectCommand())
331 def OnWebcamPropertyPageButton(self, e):
332 self.cam.openPropertyPage(e.GetEventObject().index)
334 def OnCameraTimer(self, e):
335 if not self.campreviewEnable.GetValue():
337 if self.machineCom is not None and self.machineCom.isPrinting():
339 self.cam.takeNewImage()
340 self.camPreview.Refresh()
342 def OnCameraEraseBackground(self, e):
345 dc = wx.ClientDC(self)
346 rect = self.GetUpdateRegion().GetBox()
347 dc.SetClippingRect(rect)
348 dc.SetBackground(wx.Brush(self.camPreview.GetBackgroundColour(), wx.SOLID))
349 if self.cam.getLastImage() is not None:
350 self.camPreview.SetMinSize((self.cam.getLastImage().GetWidth(), self.cam.getLastImage().GetHeight()))
352 dc.DrawBitmap(self.cam.getLastImage(), 0, 0)
356 class PrintCommandButton(buttons.GenBitmapButton):
357 def __init__(self, parent, commandList, bitmapFilename, size=(20, 20)):
358 self.bitmap = wx.Bitmap(resources.getPathForImage(bitmapFilename))
359 super(PrintCommandButton, self).__init__(parent.directControlPanel, -1, self.bitmap, size=size)
361 self.commandList = commandList
364 self.SetBezelWidth(1)
365 self.SetUseFocusIndicator(False)
367 self.Bind(wx.EVT_BUTTON, self.OnClick)
369 def OnClick(self, e):
370 if self.parent._printerConnection is None or self.parent._printerConnection.isAbleToSendDirectCommand():
372 for cmd in self.commandList:
373 self.parent._printerConnection.sendCommand(cmd)
376 class TemperatureGraph(wx.Panel):
377 def __init__(self, parent):
378 super(TemperatureGraph, self).__init__(parent)
380 self.Bind(wx.EVT_ERASE_BACKGROUND, self.OnEraseBackground)
381 self.Bind(wx.EVT_SIZE, self.OnSize)
382 self.Bind(wx.EVT_PAINT, self.OnDraw)
384 self._lastDraw = time.time() - 1.0
386 self._backBuffer = None
387 self.addPoint([0]*16, [0]*16, 0, 0)
388 self.SetMinSize((320, 200))
390 def OnEraseBackground(self, e):
394 if self._backBuffer is None or self.GetSize() != self._backBuffer.GetSize():
395 self._backBuffer = wx.EmptyBitmap(*self.GetSizeTuple())
396 self.UpdateDrawing(True)
399 dc = wx.BufferedPaintDC(self, self._backBuffer)
401 def UpdateDrawing(self, force=False):
403 if (not force and now - self._lastDraw < 1.0) or self._backBuffer is None:
407 dc.SelectObject(self._backBuffer)
409 dc.SetFont(wx.SystemSettings.GetFont(wx.SYS_SYSTEM_FONT))
410 w, h = self.GetSizeTuple()
411 bgLinePen = wx.Pen('#A0A0A0')
412 tempPen = wx.Pen('#FF4040')
413 tempSPPen = wx.Pen('#FFA0A0')
414 tempPenBG = wx.Pen('#FFD0D0')
415 bedTempPen = wx.Pen('#4040FF')
416 bedTempSPPen = wx.Pen('#A0A0FF')
417 bedTempPenBG = wx.Pen('#D0D0FF')
419 #Draw the background up to the current temperatures.
421 t0 = [0] * len(self._points[0][0])
425 for temp, tempSP, bedTemp, bedTempSP, t in self._points:
426 x1 = int(w - (now - t))
427 for x in xrange(x0, x1 + 1):
428 for n in xrange(0, len(temp)):
429 t = float(x - x0) / float(x1 - x0 + 1) * (temp[n] - t0[n]) + t0[n]
431 dc.DrawLine(x, h, x, h - (t * h / 300))
432 bt = float(x - x0) / float(x1 - x0 + 1) * (bedTemp - bt0) + bt0
433 dc.SetPen(bedTempPenBG)
434 dc.DrawLine(x, h, x, h - (bt * h / 300))
442 for x in xrange(w, 0, -30):
444 dc.DrawLine(x, 0, x, h)
446 for y in xrange(h - 1, 0, -h * 50 / 300):
448 dc.DrawLine(0, y, w, y)
449 dc.DrawText(str(tmpNr), 0, y - dc.GetFont().GetPixelSize().GetHeight())
451 dc.DrawLine(0, 0, w, 0)
452 dc.DrawLine(0, 0, 0, h)
456 t0 = [0] * len(self._points[0][0])
458 tSP0 = [0] * len(self._points[0][0])
460 for temp, tempSP, bedTemp, bedTempSP, t in self._points:
461 x1 = int(w - (now - t))
462 for x in xrange(x0, x1 + 1):
463 for n in xrange(0, len(temp)):
464 t = float(x - x0) / float(x1 - x0 + 1) * (temp[n] - t0[n]) + t0[n]
465 tSP = float(x - x0) / float(x1 - x0 + 1) * (tempSP[n] - tSP0[n]) + tSP0[n]
467 dc.DrawPoint(x, h - (tSP * h / 300))
469 dc.DrawPoint(x, h - (t * h / 300))
470 bt = float(x - x0) / float(x1 - x0 + 1) * (bedTemp - bt0) + bt0
471 btSP = float(x - x0) / float(x1 - x0 + 1) * (bedTempSP - btSP0) + btSP0
472 dc.SetPen(bedTempSPPen)
473 dc.DrawPoint(x, h - (btSP * h / 300))
474 dc.SetPen(bedTempPen)
475 dc.DrawPoint(x, h - (bt * h / 300))
483 self.Refresh(eraseBackground=False)
486 if len(self._points) > 0 and (time.time() - self._points[0][4]) > w + 20:
489 def addPoint(self, temp, tempSP, bedTemp, bedTempSP):
492 if bedTempSP is None:
494 self._points.append((temp[:], tempSP[:], bedTemp, bedTempSP, time.time()))
495 wx.CallAfter(self.UpdateDrawing)
498 class LogWindow(wx.Frame):
499 def __init__(self, logText):
500 super(LogWindow, self).__init__(None, title="Error log")
501 self.textBox = wx.TextCtrl(self, -1, logText, style=wx.TE_MULTILINE | wx.TE_DONTWRAP | wx.TE_READONLY)
502 self.SetSize((500, 400))