chiark / gitweb /
Add extrude/retract images for printing interface. Made exception handling a bit...
[cura.git] / Cura / gui / printWindow.py
1 from __future__ import absolute_import\r
2 import __init__\r
3 \r
4 import wx, threading, re, subprocess, sys, os, time\r
5 from wx.lib import buttons\r
6 \r
7 from gui import icon\r
8 from gui import toolbarUtil\r
9 from gui import webcam\r
10 from util import machineCom\r
11 from util import profile\r
12 from util import gcodeInterpreter\r
13 \r
14 printWindowMonitorHandle = None\r
15 \r
16 def printFile(filename):\r
17         global printWindowMonitorHandle\r
18         if printWindowMonitorHandle == None:\r
19                 printWindowMonitorHandle = printProcessMonitor()\r
20         printWindowMonitorHandle.loadFile(filename)\r
21 \r
22 def startPrintInterface(filename):\r
23         #startPrintInterface is called from the main script when we want the printer interface to run in a seperate process.\r
24         # It needs to run in a seperate process, as any running python code blocks the GCode sender pyton code (http://wiki.python.org/moin/GlobalInterpreterLock).\r
25         app = wx.App(False)\r
26         printWindowHandle = printWindow()\r
27         printWindowHandle.Show(True)\r
28         printWindowHandle.Raise()\r
29         printWindowHandle.OnConnect(None)\r
30         t = threading.Thread(target=printWindowHandle.LoadGCodeFile,args=(filename,))\r
31         t.daemon = True\r
32         t.start()\r
33         app.MainLoop()\r
34 \r
35 class printProcessMonitor():\r
36         def __init__(self):\r
37                 self.handle = None\r
38         \r
39         def loadFile(self, filename):\r
40                 if self.handle == None:\r
41                         self.handle = subprocess.Popen([sys.executable, sys.argv[0], '-r', filename], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)\r
42                         self.thread = threading.Thread(target=self.Monitor)\r
43                         self.thread.start()\r
44                 else:\r
45                         self.handle.stdin.write(filename + '\n')\r
46         \r
47         def Monitor(self):\r
48                 p = self.handle\r
49                 line = p.stdout.readline()\r
50                 while(len(line) > 0):\r
51                         #print line.rstrip()\r
52                         line = p.stdout.readline()\r
53                 p.wait()\r
54                 self.handle = None\r
55                 self.thread = None\r
56 \r
57 class PrintCommandButton(buttons.GenBitmapButton):\r
58         def __init__(self, parent, commandList, bitmapFilename, size=(20,20)):\r
59                 self.bitmap = toolbarUtil.getBitmapImage(bitmapFilename)\r
60                 super(PrintCommandButton, self).__init__(parent.directControlPanel, -1, self.bitmap, size=size)\r
61 \r
62                 self.commandList = commandList\r
63                 self.parent = parent\r
64 \r
65                 self.SetBezelWidth(1)\r
66                 self.SetUseFocusIndicator(False)\r
67 \r
68                 self.Bind(wx.EVT_BUTTON, self.OnClick)\r
69 \r
70         def OnClick(self, e):\r
71                 if self.parent.machineCom == None or self.parent.machineCom.isPrinting():\r
72                         return;\r
73                 for cmd in self.commandList:\r
74                         self.parent.machineCom.sendCommand(cmd)\r
75                 e.Skip()\r
76 \r
77 class printWindow(wx.Frame):\r
78         "Main user interface window"\r
79         def __init__(self):\r
80                 super(printWindow, self).__init__(None, -1, title='Printing')\r
81                 self.machineCom = None\r
82                 self.gcode = None\r
83                 self.gcodeList = None\r
84                 self.sendList = []\r
85                 self.temp = None\r
86                 self.bedTemp = None\r
87                 self.bufferLineCount = 4\r
88                 self.sendCnt = 0\r
89                 self.feedrateRatioOuterWall = 1.0\r
90                 self.feedrateRatioInnerWall = 1.0\r
91                 self.feedrateRatioFill = 1.0\r
92                 self.feedrateRatioSupport = 1.0\r
93                 self.pause = False\r
94                 self.termHistory = []\r
95                 self.termHistoryIdx = 0\r
96                 \r
97                 self.cam = None\r
98                 if webcam.hasWebcamSupport():\r
99                         self.cam = webcam.webcam()\r
100 \r
101                 #self.SetIcon(icon.getMainIcon())\r
102                 \r
103                 self.SetSizer(wx.BoxSizer())\r
104                 self.panel = wx.Panel(self)\r
105                 self.GetSizer().Add(self.panel, 1, flag=wx.EXPAND)\r
106                 self.sizer = wx.GridBagSizer(2, 2)\r
107                 self.panel.SetSizer(self.sizer)\r
108                 \r
109                 sb = wx.StaticBox(self.panel, label="Statistics")\r
110                 boxsizer = wx.StaticBoxSizer(sb, wx.VERTICAL)\r
111                 self.statsText = wx.StaticText(self.panel, -1, "Filament: ####.##m #.##g\nEstimated print time: #####:##\nMachine state:\nDetecting baudrateXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX")\r
112                 boxsizer.Add(self.statsText, flag=wx.LEFT, border=5)\r
113                 \r
114                 self.sizer.Add(boxsizer, pos=(0,0), span=(6,1), flag=wx.EXPAND)\r
115                 \r
116                 self.connectButton = wx.Button(self.panel, -1, 'Connect')\r
117                 #self.loadButton = wx.Button(self.panel, -1, 'Load GCode')\r
118                 self.printButton = wx.Button(self.panel, -1, 'Print GCode')\r
119                 self.pauseButton = wx.Button(self.panel, -1, 'Pause')\r
120                 self.cancelButton = wx.Button(self.panel, -1, 'Cancel print')\r
121                 self.machineLogButton = wx.Button(self.panel, -1, 'Error log')\r
122                 self.progress = wx.Gauge(self.panel, -1)\r
123                 \r
124                 self.sizer.Add(self.connectButton, pos=(0,1))\r
125                 #self.sizer.Add(self.loadButton, pos=(1,1))\r
126                 self.sizer.Add(self.printButton, pos=(2,1))\r
127                 self.sizer.Add(self.pauseButton, pos=(3,1))\r
128                 self.sizer.Add(self.cancelButton, pos=(4,1))\r
129                 self.sizer.Add(self.machineLogButton, pos=(5,1))\r
130                 self.sizer.Add(self.progress, pos=(6,0), span=(1,7), flag=wx.EXPAND)\r
131 \r
132                 nb = wx.Notebook(self.panel)\r
133                 self.sizer.Add(nb, pos=(0,3), span=(6,4), flag=wx.EXPAND)\r
134                 \r
135                 self.temperaturePanel = wx.Panel(nb)\r
136                 sizer = wx.GridBagSizer(2, 2)\r
137                 self.temperaturePanel.SetSizer(sizer)\r
138 \r
139                 self.temperatureSelect = wx.SpinCtrl(self.temperaturePanel, -1, '0', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
140                 self.temperatureSelect.SetRange(0, 400)\r
141                 self.bedTemperatureLabel = wx.StaticText(self.temperaturePanel, -1, "BedTemp:")\r
142                 self.bedTemperatureSelect = wx.SpinCtrl(self.temperaturePanel, -1, '0', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
143                 self.bedTemperatureSelect.SetRange(0, 400)\r
144                 self.bedTemperatureLabel.Show(False)\r
145                 self.bedTemperatureSelect.Show(False)\r
146                 \r
147                 self.temperatureGraph = temperatureGraph(self.temperaturePanel)\r
148                 \r
149                 sizer.Add(wx.StaticText(self.temperaturePanel, -1, "Temp:"), pos=(0,0))\r
150                 sizer.Add(self.temperatureSelect, pos=(0,1))\r
151                 sizer.Add(self.bedTemperatureLabel, pos=(1,0))\r
152                 sizer.Add(self.bedTemperatureSelect, pos=(1,1))\r
153                 sizer.Add(self.temperatureGraph, pos=(2,0), span=(1,2), flag=wx.EXPAND)\r
154                 sizer.AddGrowableRow(2)\r
155                 sizer.AddGrowableCol(1)\r
156 \r
157                 nb.AddPage(self.temperaturePanel, 'Temp')\r
158 \r
159                 self.directControlPanel = wx.Panel(nb)\r
160                 \r
161                 sizer = wx.GridBagSizer(2, 2)\r
162                 self.directControlPanel.SetSizer(sizer)\r
163                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Y100 F6000', 'G90'], 'print-move-y100.png'), pos=(0,3))\r
164                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Y10 F6000', 'G90'], 'print-move-y10.png'), pos=(1,3))\r
165                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Y1 F6000', 'G90'], 'print-move-y1.png'), pos=(2,3))\r
166 \r
167                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Y-1 F6000', 'G90'], 'print-move-y-1.png'), pos=(4,3))\r
168                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Y-10 F6000', 'G90'], 'print-move-y-10.png'), pos=(5,3))\r
169                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Y-100 F6000', 'G90'], 'print-move-y-100.png'), pos=(6,3))\r
170 \r
171                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 X-100 F6000', 'G90'], 'print-move-x-100.png'), pos=(3,0))\r
172                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 X-10 F6000', 'G90'], 'print-move-x-10.png'), pos=(3,1))\r
173                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 X-1 F6000', 'G90'], 'print-move-x-1.png'), pos=(3,2))\r
174 \r
175                 sizer.Add(PrintCommandButton(self, ['G28 X0 Y0'], 'print-move-home.png'), pos=(3,3))\r
176 \r
177                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 X1 F6000', 'G90'], 'print-move-x1.png'), pos=(3,4))\r
178                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 X10 F6000', 'G90'], 'print-move-x10.png'), pos=(3,5))\r
179                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 X100 F6000', 'G90'], 'print-move-x100.png'), pos=(3,6))\r
180 \r
181                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Z10 F200', 'G90'], 'print-move-z10.png'), pos=(0,8))\r
182                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Z1 F200', 'G90'], 'print-move-z1.png'), pos=(1,8))\r
183                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Z0.1 F200', 'G90'], 'print-move-z0.1.png'), pos=(2,8))\r
184 \r
185                 sizer.Add(PrintCommandButton(self, ['G28 Z0'], 'print-move-home.png'), pos=(3,8))\r
186 \r
187                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Z-0.1 F200', 'G90'], 'print-move-z-0.1.png'), pos=(4,8))\r
188                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Z-1 F200', 'G90'], 'print-move-z-1.png'), pos=(5,8))\r
189                 sizer.Add(PrintCommandButton(self, ['G91', 'G1 Z-10 F200', 'G90'], 'print-move-z-10.png'), pos=(6,8))\r
190 \r
191                 sizer.Add(PrintCommandButton(self, ['G92 E0', 'G1 E2 F120'], 'extrude.png', size=(60,20)), pos=(1,10), span=(1,3), flag=wx.EXPAND)\r
192                 sizer.Add(PrintCommandButton(self, ['G92 E0', 'G1 E-2 F120'], 'retract.png', size=(60,20)), pos=(2,10), span=(1,3), flag=wx.EXPAND)\r
193 \r
194                 nb.AddPage(self.directControlPanel, 'Jog')\r
195 \r
196                 self.speedPanel = wx.Panel(nb)\r
197                 sizer = wx.GridBagSizer(2, 2)\r
198                 self.speedPanel.SetSizer(sizer)\r
199 \r
200                 self.outerWallSpeedSelect = wx.SpinCtrl(self.speedPanel, -1, '100', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
201                 self.outerWallSpeedSelect.SetRange(5, 1000)\r
202                 self.innerWallSpeedSelect = wx.SpinCtrl(self.speedPanel, -1, '100', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
203                 self.innerWallSpeedSelect.SetRange(5, 1000)\r
204                 self.fillSpeedSelect = wx.SpinCtrl(self.speedPanel, -1, '100', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
205                 self.fillSpeedSelect.SetRange(5, 1000)\r
206                 self.supportSpeedSelect = wx.SpinCtrl(self.speedPanel, -1, '100', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
207                 self.supportSpeedSelect.SetRange(5, 1000)\r
208                 \r
209                 sizer.Add(wx.StaticText(self.speedPanel, -1, "Outer wall:"), pos=(0,0))\r
210                 sizer.Add(self.outerWallSpeedSelect, pos=(0,1))\r
211                 sizer.Add(wx.StaticText(self.speedPanel, -1, "%"), pos=(0,2))\r
212                 sizer.Add(wx.StaticText(self.speedPanel, -1, "Inner wall:"), pos=(1,0))\r
213                 sizer.Add(self.innerWallSpeedSelect, pos=(1,1))\r
214                 sizer.Add(wx.StaticText(self.speedPanel, -1, "%"), pos=(1,2))\r
215                 sizer.Add(wx.StaticText(self.speedPanel, -1, "Fill:"), pos=(2,0))\r
216                 sizer.Add(self.fillSpeedSelect, pos=(2,1))\r
217                 sizer.Add(wx.StaticText(self.speedPanel, -1, "%"), pos=(2,2))\r
218                 sizer.Add(wx.StaticText(self.speedPanel, -1, "Support:"), pos=(3,0))\r
219                 sizer.Add(self.supportSpeedSelect, pos=(3,1))\r
220                 sizer.Add(wx.StaticText(self.speedPanel, -1, "%"), pos=(3,2))\r
221 \r
222                 nb.AddPage(self.speedPanel, 'Speed')\r
223                 \r
224                 self.termPanel = wx.Panel(nb)\r
225                 sizer = wx.GridBagSizer(2, 2)\r
226                 self.termPanel.SetSizer(sizer)\r
227                 \r
228                 f = wx.Font(8, wx.FONTFAMILY_MODERN, wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_NORMAL, False)\r
229                 self.termLog = wx.TextCtrl(self.termPanel, style=wx.TE_MULTILINE|wx.TE_DONTWRAP)\r
230                 self.termLog.SetFont(f)\r
231                 self.termLog.SetEditable(0)\r
232                 self.termInput = wx.TextCtrl(self.termPanel, style=wx.TE_PROCESS_ENTER)\r
233                 self.termInput.SetFont(f)\r
234 \r
235                 sizer.Add(self.termLog, pos=(0,0),flag=wx.EXPAND)\r
236                 sizer.Add(self.termInput, pos=(1,0),flag=wx.EXPAND)\r
237                 sizer.AddGrowableCol(0)\r
238                 sizer.AddGrowableRow(0)\r
239 \r
240                 nb.AddPage(self.termPanel, 'Term')\r
241                 \r
242                 if self.cam != None and self.cam.hasCamera():\r
243                         self.camPage = wx.Panel(nb)\r
244                         sizer = wx.GridBagSizer(2, 2)\r
245                         self.camPage.SetSizer(sizer)\r
246                         \r
247                         self.timelapsEnable = wx.CheckBox(self.camPage, -1, 'Enable timelaps movie recording')\r
248                         sizer.Add(self.timelapsEnable, pos=(0,0), span=(1,2), flag=wx.EXPAND)\r
249                         \r
250                         pages = self.cam.propertyPages()\r
251                         self.cam.buttons = [self.timelapsEnable]\r
252                         for page in pages:\r
253                                 button = wx.Button(self.camPage, -1, page)\r
254                                 button.index = pages.index(page)\r
255                                 sizer.Add(button, pos=(1, pages.index(page)))\r
256                                 button.Bind(wx.EVT_BUTTON, self.OnPropertyPageButton)\r
257                                 self.cam.buttons.append(button)\r
258                         \r
259                         self.camPreview = wx.Panel(self.camPage)\r
260                         sizer.Add(self.camPreview, pos=(2,0), span=(1,2), flag=wx.EXPAND)\r
261                         \r
262                         nb.AddPage(self.camPage, 'Camera')\r
263                         self.camPreview.timer = wx.Timer(self)\r
264                         self.Bind(wx.EVT_TIMER, self.OnCameraTimer, self.camPreview.timer)\r
265                         self.camPreview.timer.Start(500)\r
266                         self.camPreview.Bind(wx.EVT_ERASE_BACKGROUND, self.OnCameraEraseBackground)\r
267 \r
268                 self.sizer.AddGrowableRow(5)\r
269                 self.sizer.AddGrowableCol(3)\r
270                 \r
271                 self.Bind(wx.EVT_CLOSE, self.OnClose)\r
272                 self.connectButton.Bind(wx.EVT_BUTTON, self.OnConnect)\r
273                 #self.loadButton.Bind(wx.EVT_BUTTON, self.OnLoad)\r
274                 self.printButton.Bind(wx.EVT_BUTTON, self.OnPrint)\r
275                 self.pauseButton.Bind(wx.EVT_BUTTON, self.OnPause)\r
276                 self.cancelButton.Bind(wx.EVT_BUTTON, self.OnCancel)\r
277                 self.machineLogButton.Bind(wx.EVT_BUTTON, self.OnMachineLog)\r
278                 \r
279                 self.Bind(wx.EVT_SPINCTRL, self.OnTempChange, self.temperatureSelect)\r
280                 self.Bind(wx.EVT_SPINCTRL, self.OnBedTempChange, self.bedTemperatureSelect)\r
281 \r
282                 self.Bind(wx.EVT_SPINCTRL, self.OnSpeedChange, self.outerWallSpeedSelect)\r
283                 self.Bind(wx.EVT_SPINCTRL, self.OnSpeedChange, self.innerWallSpeedSelect)\r
284                 self.Bind(wx.EVT_SPINCTRL, self.OnSpeedChange, self.fillSpeedSelect)\r
285                 self.Bind(wx.EVT_SPINCTRL, self.OnSpeedChange, self.supportSpeedSelect)\r
286                 self.Bind(wx.EVT_TEXT_ENTER, self.OnTermEnterLine, self.termInput)\r
287                 self.termInput.Bind(wx.EVT_CHAR, self.OnTermKey)\r
288                 \r
289                 self.Layout()\r
290                 self.Fit()\r
291                 self.Centre()\r
292                 \r
293                 self.statsText.SetMinSize(self.statsText.GetSize())\r
294 \r
295                 self.UpdateButtonStates()\r
296                 #self.UpdateProgress()\r
297         \r
298         def OnCameraTimer(self, e):\r
299                 if self.machineCom != None and self.machineCom.isPrinting():\r
300                         return\r
301                 self.cam.takeNewImage()\r
302                 self.camPreview.Refresh()\r
303         \r
304         def OnCameraEraseBackground(self, e):\r
305                 dc = e.GetDC()\r
306                 if not dc:\r
307                         dc = wx.ClientDC(self)\r
308                         rect = self.GetUpdateRegion().GetBox()\r
309                         dc.SetClippingRect(rect)\r
310                 dc.SetBackground(wx.Brush(self.camPreview.GetBackgroundColour(), wx.SOLID))\r
311                 if self.cam.getLastImage() != None:\r
312                         self.camPreview.SetMinSize((self.cam.getLastImage().GetWidth(), self.cam.getLastImage().GetHeight()))\r
313                         self.camPage.Fit()\r
314                         dc.DrawBitmap(self.cam.getLastImage(), 0, 0)\r
315                 else:\r
316                         dc.Clear()\r
317                 \r
318         def OnPropertyPageButton(self, e):\r
319                 self.cam.openPropertyPage(e.GetEventObject().index)\r
320 \r
321         def UpdateButtonStates(self):\r
322                 self.connectButton.Enable(self.machineCom == None or self.machineCom.isClosedOrError())\r
323                 #self.loadButton.Enable(self.machineCom == None or not (self.machineCom.isPrinting() or self.machineCom.isPaused()))\r
324                 self.printButton.Enable(self.machineCom != None and self.machineCom.isOperational() and not (self.machineCom.isPrinting() or self.machineCom.isPaused()))\r
325                 self.pauseButton.Enable(self.machineCom != None and (self.machineCom.isPrinting() or self.machineCom.isPaused()))\r
326                 if self.machineCom != None and self.machineCom.isPaused():\r
327                         self.pauseButton.SetLabel('Resume')\r
328                 else:\r
329                         self.pauseButton.SetLabel('Pause')\r
330                 self.cancelButton.Enable(self.machineCom != None and (self.machineCom.isPrinting() or self.machineCom.isPaused()))\r
331                 self.temperatureSelect.Enable(self.machineCom != None and self.machineCom.isOperational())\r
332                 self.bedTemperatureSelect.Enable(self.machineCom != None and self.machineCom.isOperational())\r
333                 self.directControlPanel.Enable(self.machineCom != None and self.machineCom.isOperational() and not self.machineCom.isPrinting())\r
334                 self.machineLogButton.Show(self.machineCom != None and self.machineCom.isError())\r
335                 if self.cam:\r
336                         for button in self.cam.buttons:\r
337                                 button.Enable(self.machineCom == None or not self.machineCom.isPrinting())\r
338         \r
339         def UpdateProgress(self):\r
340                 status = ""\r
341                 if self.gcode == None:\r
342                         status += "Loading gcode...\n"\r
343                 else:\r
344                         status += "Filament: %.2fm %.2fg\n" % (self.gcode.extrusionAmount / 1000, self.gcode.calculateWeight() * 1000)\r
345                         cost = self.gcode.calculateCost()\r
346                         if cost != False:\r
347                                 status += "Filament cost: %s\n" % (cost)\r
348                         status += "Estimated print time: %02d:%02d\n" % (int(self.gcode.totalMoveTimeMinute / 60), int(self.gcode.totalMoveTimeMinute % 60))\r
349                 if self.machineCom == None or not self.machineCom.isPrinting():\r
350                         self.progress.SetValue(0)\r
351                         if self.gcodeList != None:\r
352                                 status += 'Line: -/%d\n' % (len(self.gcodeList))\r
353                 else:\r
354                         printTime = self.machineCom.getPrintTime() / 60\r
355                         printTimeTotal = printTime * len(self.gcodeList) / self.machineCom.getPrintPos()\r
356                         printTimeLeft = printTimeTotal - printTime\r
357                         status += 'Line: %d/%d %d%%\n' % (self.machineCom.getPrintPos(), len(self.gcodeList), self.machineCom.getPrintPos() * 100 / len(self.gcodeList))\r
358                         if self.currentZ > 0:\r
359                                 status += 'Height: %0.1f\n' % (self.currentZ)\r
360                         status += 'Print time: %02d:%02d\n' % (int(printTime / 60), int(printTime % 60))\r
361                         if printTime < 1:\r
362                                 status += 'Print time left: Unknown\n'\r
363                         else:\r
364                                 status += 'Print time left: %02d:%02d\n' % (int(printTimeLeft / 60), int(printTimeLeft % 60))\r
365                         self.progress.SetValue(self.machineCom.getPrintPos())\r
366                 if self.machineCom != None:\r
367                         if self.machineCom.getTemp() > 0:\r
368                                 status += 'Temp: %d\n' % (self.machineCom.getTemp())\r
369                         if self.machineCom.getBedTemp() > 0:\r
370                                 status += 'Bed Temp: %d\n' % (self.machineCom.getBedTemp())\r
371                                 self.bedTemperatureLabel.Show(True)\r
372                                 self.bedTemperatureSelect.Show(True)\r
373                                 self.temperaturePanel.Layout()\r
374                         status += 'Machine state:%s\n' % (self.machineCom.getStateString())\r
375                 \r
376                 self.statsText.SetLabel(status.strip())\r
377         \r
378         def OnConnect(self, e):\r
379                 if self.machineCom != None:\r
380                         self.machineCom.close()\r
381                 self.machineCom = machineCom.MachineCom(callbackObject=self)\r
382                 self.UpdateButtonStates()\r
383         \r
384         def OnLoad(self, e):\r
385                 pass\r
386         \r
387         def OnPrint(self, e):\r
388                 if self.machineCom == None or not self.machineCom.isOperational():\r
389                         return\r
390                 if self.gcodeList == None:\r
391                         return\r
392                 if self.machineCom.isPrinting():\r
393                         return\r
394                 self.currentZ = -1\r
395                 if self.cam != None and self.timelapsEnable.GetValue():\r
396                         self.cam.startTimelaps(self.filename[: self.filename.rfind('.')] + ".mpg")\r
397                 self.machineCom.printGCode(self.gcodeList)\r
398                 self.UpdateButtonStates()\r
399         \r
400         def OnCancel(self, e):\r
401                 self.pauseButton.SetLabel('Pause')\r
402                 self.machineCom.cancelPrint()\r
403                 self.machineCom.sendCommand("M84")\r
404                 self.UpdateButtonStates()\r
405         \r
406         def OnPause(self, e):\r
407                 if self.machineCom.isPaused():\r
408                         self.machineCom.setPause(False)\r
409                 else:\r
410                         self.machineCom.setPause(True)\r
411         \r
412         def OnMachineLog(self, e):\r
413                 LogWindow('\n'.join(self.machineCom.getLog()))\r
414         \r
415         def OnClose(self, e):\r
416                 global printWindowHandle\r
417                 printWindowHandle = None\r
418                 if self.machineCom != None:\r
419                         self.machineCom.close()\r
420                 self.Destroy()\r
421 \r
422         def OnTempChange(self, e):\r
423                 self.machineCom.sendCommand("M104 S%d" % (self.temperatureSelect.GetValue()))\r
424 \r
425         def OnBedTempChange(self, e):\r
426                 self.machineCom.sendCommand("M140 S%d" % (self.bedTemperatureSelect.GetValue()))\r
427         \r
428         def OnSpeedChange(self, e):\r
429                 if self.machineCom == None:\r
430                         return\r
431                 self.machineCom.setFeedrateModifier('WALL-OUTER', self.outerWallSpeedSelect.GetValue() / 100.0)\r
432                 self.machineCom.setFeedrateModifier('WALL-INNER', self.innerWallSpeedSelect.GetValue() / 100.0)\r
433                 self.machineCom.setFeedrateModifier('FILL', self.fillSpeedSelect.GetValue() / 100.0)\r
434                 self.machineCom.setFeedrateModifier('SUPPORT', self.supportSpeedSelect.GetValue() / 100.0)\r
435         \r
436         def AddTermLog(self, line):\r
437                 self.termLog.AppendText(unicode(line, 'utf-8', 'replace'))\r
438                 l = len(self.termLog.GetValue())\r
439                 self.termLog.SetCaret(wx.Caret(self.termLog, (l, l)))\r
440         \r
441         def OnTermEnterLine(self, e):\r
442                 line = self.termInput.GetValue()\r
443                 if line == '':\r
444                         return\r
445                 self.termLog.AppendText('>%s\n' % (line))\r
446                 self.machineCom.sendCommand(line)\r
447                 self.termHistory.append(line)\r
448                 self.termHistoryIdx = len(self.termHistory)\r
449                 self.termInput.SetValue('')\r
450 \r
451         def OnTermKey(self, e):\r
452                 if len(self.termHistory) > 0:\r
453                         if e.GetKeyCode() == wx.WXK_UP:\r
454                                 self.termHistoryIdx = self.termHistoryIdx - 1\r
455                                 if self.termHistoryIdx < 0:\r
456                                         self.termHistoryIdx = len(self.termHistory) - 1\r
457                                 self.termInput.SetValue(self.termHistory[self.termHistoryIdx])\r
458                         if e.GetKeyCode() == wx.WXK_DOWN:\r
459                                 self.termHistoryIdx = self.termHistoryIdx - 1\r
460                                 if self.termHistoryIdx >= len(self.termHistory):\r
461                                         self.termHistoryIdx = 0\r
462                                 self.termInput.SetValue(self.termHistory[self.termHistoryIdx])\r
463                 e.Skip()\r
464 \r
465         def LoadGCodeFile(self, filename):\r
466                 if self.machineCom != None and self.machineCom.isPrinting():\r
467                         return\r
468                 #Send an initial M110 to reset the line counter to zero.\r
469                 prevLineType = lineType = 'CUSTOM'\r
470                 gcodeList = ["M110"]\r
471                 for line in open(filename, 'r'):\r
472                         if line.startswith(';TYPE:'):\r
473                                 lineType = line[6:].strip()\r
474                         if ';' in line:\r
475                                 line = line[0:line.find(';')]\r
476                         line = line.strip()\r
477                         if len(line) > 0:\r
478                                 if prevLineType != lineType:\r
479                                         gcodeList.append((line, lineType, ))\r
480                                 else:\r
481                                         gcodeList.append(line)\r
482                                 prevLineType = lineType\r
483                 gcode = gcodeInterpreter.gcode()\r
484                 gcode.loadList(gcodeList)\r
485                 #print "Loaded: %s (%d)" % (filename, len(gcodeList))\r
486                 self.filename = filename\r
487                 self.gcode = gcode\r
488                 self.gcodeList = gcodeList\r
489                 \r
490                 wx.CallAfter(self.progress.SetRange, len(gcodeList))\r
491                 wx.CallAfter(self.UpdateButtonStates)\r
492                 wx.CallAfter(self.UpdateProgress)\r
493                 wx.CallAfter(self.SetTitle, 'Printing: %s' % (filename))\r
494                 \r
495         def sendLine(self, lineNr):\r
496                 if lineNr >= len(self.gcodeList):\r
497                         return False\r
498                 line = self.gcodeList[lineNr]\r
499                 try:\r
500                         if ('M104' in line or 'M109' in line) and 'S' in line:\r
501                                 n = int(re.search('S([0-9]*)', line).group(1))\r
502                                 wx.CallAfter(self.temperatureSelect.SetValue, n)\r
503                         if ('M140' in line or 'M190' in line) and 'S' in line:\r
504                                 n = int(re.search('S([0-9]*)', line).group(1))\r
505                                 wx.CallAfter(self.bedTemperatureSelect.SetValue, n)\r
506                 except:\r
507                         print "Unexpected error:", sys.exc_info()\r
508                 checksum = reduce(lambda x,y:x^y, map(ord, "N%d%s" % (lineNr, line)))\r
509                 self.machineCom.sendCommand("N%d%s*%d" % (lineNr, line, checksum))\r
510                 return True\r
511 \r
512         def mcLog(self, message):\r
513                 #print message\r
514                 pass\r
515         \r
516         def mcTempUpdate(self, temp, bedTemp, targetTemp, bedTargetTemp):\r
517                 self.temperatureGraph.addPoint(temp, targetTemp, bedTemp, bedTargetTemp)\r
518                 if self.temperatureSelect.GetValue() != targetTemp:\r
519                         wx.CallAfter(self.temperatureSelect.SetValue, targetTemp)\r
520                 if self.bedTemperatureSelect.GetValue() != bedTargetTemp:\r
521                         wx.CallAfter(self.bedTemperatureSelect.SetValue, bedTargetTemp)\r
522         \r
523         def mcStateChange(self, state):\r
524                 if self.machineCom != None and state == self.machineCom.STATE_OPERATIONAL and self.cam != None:\r
525                         self.cam.endTimelaps()\r
526                 wx.CallAfter(self.UpdateButtonStates)\r
527                 wx.CallAfter(self.UpdateProgress)\r
528         \r
529         def mcMessage(self, message):\r
530                 wx.CallAfter(self.AddTermLog, message)\r
531         \r
532         def mcProgress(self, lineNr):\r
533                 wx.CallAfter(self.UpdateProgress)\r
534         \r
535         def mcZChange(self, newZ):\r
536                 self.currentZ = newZ\r
537                 if self.cam != None:\r
538                         wx.CallAfter(self.cam.takeNewImage)\r
539                         wx.CallAfter(self.camPreview.Refresh)\r
540 \r
541 class temperatureGraph(wx.Panel):\r
542         def __init__(self, parent):\r
543                 super(temperatureGraph, self).__init__(parent)\r
544                 \r
545                 self.Bind(wx.EVT_ERASE_BACKGROUND, self.OnEraseBackground)\r
546                 self.Bind(wx.EVT_SIZE, self.OnSize)\r
547                 self.Bind(wx.EVT_PAINT, self.OnDraw)\r
548                 \r
549                 self.lastDraw = time.time() - 1.0\r
550                 self.points = []\r
551                 self.backBuffer = None\r
552                 self.addPoint(0,0,0,0)\r
553                 self.SetMinSize((320,200))\r
554         \r
555         def OnEraseBackground(self, e):\r
556                 pass\r
557         \r
558         def OnSize(self, e):\r
559                 if self.backBuffer == None or self.GetSize() != self.backBuffer.GetSize():\r
560                         self.backBuffer = wx.EmptyBitmap(*self.GetSizeTuple())\r
561                         self.UpdateDrawing(True)\r
562         \r
563         def OnDraw(self, e):\r
564                 dc = wx.BufferedPaintDC(self, self.backBuffer)\r
565         \r
566         def UpdateDrawing(self, force = False):\r
567                 now = time.time()\r
568                 if not force and now - self.lastDraw < 1.0:\r
569                         return\r
570                 self.lastDraw = now\r
571                 dc = wx.MemoryDC()\r
572                 dc.SelectObject(self.backBuffer)\r
573                 dc.Clear()\r
574                 w, h = self.GetSizeTuple()\r
575                 bgLinePen = wx.Pen('#A0A0A0')\r
576                 tempPen = wx.Pen('#FF4040')\r
577                 tempSPPen = wx.Pen('#FFA0A0')\r
578                 tempPenBG = wx.Pen('#FFD0D0')\r
579                 bedTempPen = wx.Pen('#4040FF')\r
580                 bedTempSPPen = wx.Pen('#A0A0FF')\r
581                 bedTempPenBG = wx.Pen('#D0D0FF')\r
582 \r
583                 #Draw the background up to the current temperatures.\r
584                 x0 = 0\r
585                 t0 = 0\r
586                 bt0 = 0\r
587                 tSP0 = 0\r
588                 btSP0 = 0\r
589                 for temp, tempSP, bedTemp, bedTempSP, t in self.points:\r
590                         x1 = int(w - (now - t))\r
591                         for x in xrange(x0, x1 + 1):\r
592                                 t = float(x - x0) / float(x1 - x0 + 1) * (temp - t0) + t0\r
593                                 bt = float(x - x0) / float(x1 - x0 + 1) * (bedTemp - bt0) + bt0\r
594                                 dc.SetPen(tempPenBG)\r
595                                 dc.DrawLine(x, h, x, h - (t * h / 300))\r
596                                 dc.SetPen(bedTempPenBG)\r
597                                 dc.DrawLine(x, h, x, h - (bt * h / 300))\r
598                         t0 = temp\r
599                         bt0 = bedTemp\r
600                         tSP0 = tempSP\r
601                         btSP0 = bedTempSP\r
602                         x0 = x1 + 1\r
603 \r
604                 #Draw the grid\r
605                 for x in xrange(w, 0, -30):\r
606                         dc.SetPen(bgLinePen)\r
607                         dc.DrawLine(x, 0, x, h)\r
608                 for y in xrange(h-1, 0, -h * 50 / 300):\r
609                         dc.SetPen(bgLinePen)\r
610                         dc.DrawLine(0, y, w, y)\r
611                 dc.DrawLine(0, 0, w, 0)\r
612                 dc.DrawLine(0, 0, 0, h)\r
613                 \r
614                 #Draw the main lines\r
615                 x0 = 0\r
616                 t0 = 0\r
617                 bt0 = 0\r
618                 tSP0 = 0\r
619                 btSP0 = 0\r
620                 for temp, tempSP, bedTemp, bedTempSP, t in self.points:\r
621                         x1 = int(w - (now - t))\r
622                         for x in xrange(x0, x1 + 1):\r
623                                 t = float(x - x0) / float(x1 - x0 + 1) * (temp - t0) + t0\r
624                                 bt = float(x - x0) / float(x1 - x0 + 1) * (bedTemp - bt0) + bt0\r
625                                 tSP = float(x - x0) / float(x1 - x0 + 1) * (tempSP - tSP0) + tSP0\r
626                                 btSP = float(x - x0) / float(x1 - x0 + 1) * (bedTempSP - btSP0) + btSP0\r
627                                 dc.SetPen(tempSPPen)\r
628                                 dc.DrawPoint(x, h - (tSP * h / 300))\r
629                                 dc.SetPen(bedTempSPPen)\r
630                                 dc.DrawPoint(x, h - (btSP * h / 300))\r
631                                 dc.SetPen(tempPen)\r
632                                 dc.DrawPoint(x, h - (t * h / 300))\r
633                                 dc.SetPen(bedTempPen)\r
634                                 dc.DrawPoint(x, h - (bt * h / 300))\r
635                         t0 = temp\r
636                         bt0 = bedTemp\r
637                         tSP0 = tempSP\r
638                         btSP0 = bedTempSP\r
639                         x0 = x1 + 1\r
640                 \r
641                 del dc\r
642                 self.Refresh(eraseBackground=False)\r
643                 self.Update()\r
644                 \r
645                 if len(self.points) > 0 and (time.time() - self.points[0][4]) > w + 20:\r
646                         self.points.pop(0)\r
647 \r
648         def addPoint(self, temp, tempSP, bedTemp, bedTempSP):\r
649                 if bedTemp == None:\r
650                         bedTemp = 0\r
651                 if bedTempSP == None:\r
652                         bedTempSP = 0\r
653                 self.points.append((temp, tempSP, bedTemp, bedTempSP, time.time()))\r
654                 wx.CallAfter(self.UpdateDrawing)\r
655 \r
656 class LogWindow(wx.Frame):\r
657         def __init__(self, logText):\r
658                 super(LogWindow, self).__init__(None, title="Machine log")\r
659                 self.textBox = wx.TextCtrl(self, -1, logText, style=wx.TE_MULTILINE|wx.TE_DONTWRAP|wx.TE_READONLY)\r
660                 self.SetSize((500,400))\r
661                 self.Centre()\r
662                 self.Show(True)\r