chiark / gitweb /
Add webcam timelaps support for windows.
[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.STDOUT)\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, command, 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.command = command\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.printIdx != None:\r
72                         return;\r
73                 self.parent.sendCommand("G91")\r
74                 self.parent.sendCommand(self.command)\r
75                 self.parent.sendCommand("G90")\r
76                 e.Skip()\r
77 \r
78 class printWindow(wx.Frame):\r
79         "Main user interface window"\r
80         def __init__(self):\r
81                 super(printWindow, self).__init__(None, -1, title='Printing')\r
82                 self.machineCom = None\r
83                 self.machineConnected = False\r
84                 self.thread = None\r
85                 self.gcode = None\r
86                 self.gcodeList = None\r
87                 self.sendList = []\r
88                 self.printIdx = None\r
89                 self.temp = None\r
90                 self.bedTemp = None\r
91                 self.bufferLineCount = 4\r
92                 self.sendCnt = 0\r
93                 self.feedrateRatioOuterWall = 1.0\r
94                 self.feedrateRatioInnerWall = 1.0\r
95                 self.feedrateRatioFill = 1.0\r
96                 self.feedrateRatioSupport = 1.0\r
97                 self.pause = False\r
98                 self.termHistory = []\r
99                 self.termHistoryIdx = 0\r
100                 \r
101                 self.cam = None\r
102                 try:\r
103                         self.cam = webcam.webcam()\r
104                 except:\r
105                         pass\r
106 \r
107                 #self.SetIcon(icon.getMainIcon())\r
108                 \r
109                 self.SetSizer(wx.BoxSizer())\r
110                 self.panel = wx.Panel(self)\r
111                 self.GetSizer().Add(self.panel, 1, flag=wx.EXPAND)\r
112                 self.sizer = wx.GridBagSizer(2, 2)\r
113                 self.panel.SetSizer(self.sizer)\r
114                 \r
115                 sb = wx.StaticBox(self.panel, label="Statistics")\r
116                 boxsizer = wx.StaticBoxSizer(sb, wx.VERTICAL)\r
117                 self.statsText = wx.StaticText(self.panel, -1, "Filament: ####.##m #.##g\nPrint time: #####:##")\r
118                 boxsizer.Add(self.statsText, flag=wx.LEFT, border=5)\r
119                 \r
120                 self.sizer.Add(boxsizer, pos=(0,0), span=(5,1), flag=wx.EXPAND)\r
121                 \r
122                 self.connectButton = wx.Button(self.panel, -1, 'Connect')\r
123                 #self.loadButton = wx.Button(self.panel, -1, 'Load GCode')\r
124                 self.printButton = wx.Button(self.panel, -1, 'Print GCode')\r
125                 self.pauseButton = wx.Button(self.panel, -1, 'Pause')\r
126                 self.cancelButton = wx.Button(self.panel, -1, 'Cancel print')\r
127                 self.progress = wx.Gauge(self.panel, -1)\r
128                 \r
129                 self.sizer.Add(self.connectButton, pos=(0,1))\r
130                 #self.sizer.Add(self.loadButton, pos=(1,1))\r
131                 self.sizer.Add(self.printButton, pos=(2,1))\r
132                 self.sizer.Add(self.pauseButton, pos=(3,1))\r
133                 self.sizer.Add(self.cancelButton, pos=(4,1))\r
134                 self.sizer.Add(self.progress, pos=(5,0), span=(1,2), flag=wx.EXPAND)\r
135 \r
136                 nb = wx.Notebook(self.panel)\r
137                 self.sizer.Add(nb, pos=(0,3), span=(7,4), flag=wx.EXPAND)\r
138                 \r
139                 self.temperaturePanel = wx.Panel(nb)\r
140                 sizer = wx.GridBagSizer(2, 2)\r
141                 self.temperaturePanel.SetSizer(sizer)\r
142 \r
143                 self.temperatureSelect = wx.SpinCtrl(self.temperaturePanel, -1, '0', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
144                 self.temperatureSelect.SetRange(0, 400)\r
145                 self.bedTemperatureLabel = wx.StaticText(self.temperaturePanel, -1, "BedTemp:")\r
146                 self.bedTemperatureSelect = wx.SpinCtrl(self.temperaturePanel, -1, '0', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
147                 self.bedTemperatureSelect.SetRange(0, 400)\r
148                 self.bedTemperatureLabel.Show(False)\r
149                 self.bedTemperatureSelect.Show(False)\r
150                 \r
151                 self.temperatureGraph = temperatureGraph(self.temperaturePanel)\r
152                 \r
153                 sizer.Add(wx.StaticText(self.temperaturePanel, -1, "Temp:"), pos=(0,0))\r
154                 sizer.Add(self.temperatureSelect, pos=(0,1))\r
155                 sizer.Add(self.bedTemperatureLabel, pos=(1,0))\r
156                 sizer.Add(self.bedTemperatureSelect, pos=(1,1))\r
157                 sizer.Add(self.temperatureGraph, pos=(2,0), span=(1,2), flag=wx.EXPAND)\r
158                 sizer.AddGrowableRow(2)\r
159                 sizer.AddGrowableCol(0)\r
160 \r
161                 nb.AddPage(self.temperaturePanel, 'Temp')\r
162 \r
163                 self.directControlPanel = wx.Panel(nb)\r
164                 \r
165                 sizer = wx.GridBagSizer(2, 2)\r
166                 self.directControlPanel.SetSizer(sizer)\r
167                 sizer.Add(PrintCommandButton(self, 'G1 Y100 F6000', 'print-move-y100.png'), pos=(0,3))\r
168                 sizer.Add(PrintCommandButton(self, 'G1 Y10 F6000', 'print-move-y10.png'), pos=(1,3))\r
169                 sizer.Add(PrintCommandButton(self, 'G1 Y1 F6000', 'print-move-y1.png'), pos=(2,3))\r
170 \r
171                 sizer.Add(PrintCommandButton(self, 'G1 Y-1 F6000', 'print-move-y-1.png'), pos=(4,3))\r
172                 sizer.Add(PrintCommandButton(self, 'G1 Y-10 F6000', 'print-move-y-10.png'), pos=(5,3))\r
173                 sizer.Add(PrintCommandButton(self, 'G1 Y-100 F6000', 'print-move-y-100.png'), pos=(6,3))\r
174 \r
175                 sizer.Add(PrintCommandButton(self, 'G1 X-100 F6000', 'print-move-x-100.png'), pos=(3,0))\r
176                 sizer.Add(PrintCommandButton(self, 'G1 X-10 F6000', 'print-move-x-10.png'), pos=(3,1))\r
177                 sizer.Add(PrintCommandButton(self, 'G1 X-1 F6000', 'print-move-x-1.png'), pos=(3,2))\r
178 \r
179                 sizer.Add(PrintCommandButton(self, 'G28 X0 Y0', 'print-move-home.png'), pos=(3,3))\r
180 \r
181                 sizer.Add(PrintCommandButton(self, 'G1 X1 F6000', 'print-move-x1.png'), pos=(3,4))\r
182                 sizer.Add(PrintCommandButton(self, 'G1 X10 F6000', 'print-move-x10.png'), pos=(3,5))\r
183                 sizer.Add(PrintCommandButton(self, 'G1 X100 F6000', 'print-move-x100.png'), pos=(3,6))\r
184 \r
185                 sizer.Add(PrintCommandButton(self, 'G1 Z10 F200', 'print-move-z10.png'), pos=(0,7))\r
186                 sizer.Add(PrintCommandButton(self, 'G1 Z1 F200', 'print-move-z1.png'), pos=(1,7))\r
187                 sizer.Add(PrintCommandButton(self, 'G1 Z0.1 F200', 'print-move-z0.1.png'), pos=(2,7))\r
188 \r
189                 sizer.Add(PrintCommandButton(self, 'G28 Z0', 'print-move-home.png'), pos=(3,7))\r
190 \r
191                 sizer.Add(PrintCommandButton(self, 'G1 Z-0.1 F200', 'print-move-z-0.1.png'), pos=(4,7))\r
192                 sizer.Add(PrintCommandButton(self, 'G1 Z-1 F200', 'print-move-z-1.png'), pos=(5,7))\r
193                 sizer.Add(PrintCommandButton(self, 'G1 Z-10 F200', 'print-move-z-10.png'), pos=(6,7))\r
194 \r
195                 nb.AddPage(self.directControlPanel, 'Jog')\r
196 \r
197                 self.speedPanel = wx.Panel(nb)\r
198                 sizer = wx.GridBagSizer(2, 2)\r
199                 self.speedPanel.SetSizer(sizer)\r
200 \r
201                 self.outerWallSpeedSelect = wx.SpinCtrl(self.speedPanel, -1, '100', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
202                 self.outerWallSpeedSelect.SetRange(5, 1000)\r
203                 self.innerWallSpeedSelect = wx.SpinCtrl(self.speedPanel, -1, '100', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
204                 self.innerWallSpeedSelect.SetRange(5, 1000)\r
205                 self.fillSpeedSelect = wx.SpinCtrl(self.speedPanel, -1, '100', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
206                 self.fillSpeedSelect.SetRange(5, 1000)\r
207                 self.supportSpeedSelect = wx.SpinCtrl(self.speedPanel, -1, '100', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
208                 self.supportSpeedSelect.SetRange(5, 1000)\r
209                 \r
210                 sizer.Add(wx.StaticText(self.speedPanel, -1, "Outer wall:"), pos=(0,0))\r
211                 sizer.Add(self.outerWallSpeedSelect, pos=(0,1))\r
212                 sizer.Add(wx.StaticText(self.speedPanel, -1, "%"), pos=(0,2))\r
213                 sizer.Add(wx.StaticText(self.speedPanel, -1, "Inner wall:"), pos=(1,0))\r
214                 sizer.Add(self.innerWallSpeedSelect, pos=(1,1))\r
215                 sizer.Add(wx.StaticText(self.speedPanel, -1, "%"), pos=(1,2))\r
216                 sizer.Add(wx.StaticText(self.speedPanel, -1, "Fill:"), pos=(2,0))\r
217                 sizer.Add(self.fillSpeedSelect, pos=(2,1))\r
218                 sizer.Add(wx.StaticText(self.speedPanel, -1, "%"), pos=(2,2))\r
219                 sizer.Add(wx.StaticText(self.speedPanel, -1, "Support:"), pos=(3,0))\r
220                 sizer.Add(self.supportSpeedSelect, pos=(3,1))\r
221                 sizer.Add(wx.StaticText(self.speedPanel, -1, "%"), pos=(3,2))\r
222 \r
223                 nb.AddPage(self.speedPanel, 'Speed')\r
224                 \r
225                 self.termPanel = wx.Panel(nb)\r
226                 sizer = wx.GridBagSizer(2, 2)\r
227                 self.termPanel.SetSizer(sizer)\r
228                 \r
229                 f = wx.Font(8, wx.FONTFAMILY_MODERN, wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_NORMAL, False)\r
230                 self.termLog = wx.TextCtrl(self.termPanel, style=wx.TE_MULTILINE|wx.TE_DONTWRAP)\r
231                 self.termLog.SetFont(f)\r
232                 self.termLog.SetEditable(0)\r
233                 self.termInput = wx.TextCtrl(self.termPanel, style=wx.TE_PROCESS_ENTER)\r
234                 self.termInput.SetFont(f)\r
235 \r
236                 sizer.Add(self.termLog, pos=(0,0),flag=wx.EXPAND)\r
237                 sizer.Add(self.termInput, pos=(1,0),flag=wx.EXPAND)\r
238                 sizer.AddGrowableCol(0)\r
239                 sizer.AddGrowableRow(0)\r
240 \r
241                 nb.AddPage(self.termPanel, 'Term')\r
242                 \r
243                 if self.cam != None:\r
244                         self.camPage = wx.Panel(nb)\r
245                         sizer = wx.GridBagSizer(2, 2)\r
246                         self.camPage.SetSizer(sizer)\r
247                         \r
248                         nb.AddPage(self.camPage, 'Camera')\r
249                         self.camPage.timer = wx.Timer(self)\r
250                         self.Bind(wx.EVT_TIMER, self.OnCameraTimer, self.camPage.timer)\r
251                         self.camPage.timer.Start(500)\r
252                         self.camPage.Bind(wx.EVT_ERASE_BACKGROUND, self.OnCameraEraseBackground)\r
253 \r
254                 self.sizer.AddGrowableRow(3)\r
255                 self.sizer.AddGrowableCol(3)\r
256                 \r
257                 self.Bind(wx.EVT_CLOSE, self.OnClose)\r
258                 self.connectButton.Bind(wx.EVT_BUTTON, self.OnConnect)\r
259                 #self.loadButton.Bind(wx.EVT_BUTTON, self.OnLoad)\r
260                 self.printButton.Bind(wx.EVT_BUTTON, self.OnPrint)\r
261                 self.pauseButton.Bind(wx.EVT_BUTTON, self.OnPause)\r
262                 self.cancelButton.Bind(wx.EVT_BUTTON, self.OnCancel)\r
263                 \r
264                 self.Bind(wx.EVT_SPINCTRL, self.OnTempChange, self.temperatureSelect)\r
265                 self.Bind(wx.EVT_SPINCTRL, self.OnBedTempChange, self.bedTemperatureSelect)\r
266 \r
267                 self.Bind(wx.EVT_SPINCTRL, self.OnSpeedChange, self.outerWallSpeedSelect)\r
268                 self.Bind(wx.EVT_SPINCTRL, self.OnSpeedChange, self.innerWallSpeedSelect)\r
269                 self.Bind(wx.EVT_SPINCTRL, self.OnSpeedChange, self.fillSpeedSelect)\r
270                 self.Bind(wx.EVT_SPINCTRL, self.OnSpeedChange, self.supportSpeedSelect)\r
271                 self.Bind(wx.EVT_TEXT_ENTER, self.OnTermEnterLine, self.termInput)\r
272                 self.termInput.Bind(wx.EVT_CHAR, self.OnTermKey)\r
273                 \r
274                 self.Layout()\r
275                 self.Fit()\r
276                 self.Centre()\r
277 \r
278                 self.UpdateButtonStates()\r
279                 self.UpdateProgress()\r
280         \r
281         def OnCameraTimer(self, e):\r
282                 if self.printIdx != None:\r
283                         return\r
284                 self.cam.takeNewImage()\r
285                 self.camPage.Refresh()\r
286         \r
287         def OnCameraEraseBackground(self, e):\r
288                 dc = e.GetDC()\r
289                 if not dc:\r
290                         dc = wx.ClientDC(self)\r
291                         rect = self.GetUpdateRegion().GetBox()\r
292                         dc.SetClippingRect(rect)\r
293                 dc.SetBackground(wx.Brush(self.camPage.GetBackgroundColour(), wx.SOLID))\r
294                 dc.Clear()\r
295                 if self.cam.getLastImage() != None:\r
296                         dc.DrawBitmap(self.cam.getLastImage(), 0, 0)\r
297 \r
298         def UpdateButtonStates(self):\r
299                 self.connectButton.Enable(not self.machineConnected)\r
300                 #self.loadButton.Enable(self.printIdx == None)\r
301                 self.printButton.Enable(self.machineConnected and self.gcodeList != None and self.printIdx == None)\r
302                 self.pauseButton.Enable(self.printIdx != None)\r
303                 self.cancelButton.Enable(self.printIdx != None)\r
304                 self.temperatureSelect.Enable(self.machineConnected)\r
305                 self.bedTemperatureSelect.Enable(self.machineConnected)\r
306                 self.directControlPanel.Enable(self.machineConnected)\r
307         \r
308         def UpdateProgress(self):\r
309                 status = ""\r
310                 if self.gcode == None:\r
311                         status += "Loading gcode...\n"\r
312                 else:\r
313                         status += "Filament: %.2fm %.2fg\n" % (self.gcode.extrusionAmount / 1000, self.gcode.calculateWeight() * 1000)\r
314                         cost = self.gcode.calculateCost()\r
315                         if cost != False:\r
316                                 status += "Filament cost: %s\n" % (cost)\r
317                         status += "Print time: %02d:%02d\n" % (int(self.gcode.totalMoveTimeMinute / 60), int(self.gcode.totalMoveTimeMinute % 60))\r
318                 if self.printIdx == None:\r
319                         self.progress.SetValue(0)\r
320                         if self.gcodeList != None:\r
321                                 status += 'Line: -/%d\n' % (len(self.gcodeList))\r
322                 else:\r
323                         status += 'Line: %d/%d\n' % (self.printIdx, len(self.gcodeList))\r
324                         status += 'Height: %f\n' % (self.currentZ)\r
325                         self.progress.SetValue(self.printIdx)\r
326                 if self.temp != None:\r
327                         status += 'Temp: %d\n' % (self.temp)\r
328                 if self.bedTemp != None and self.bedTemp > 0:\r
329                         status += 'Bed Temp: %d\n' % (self.bedTemp)\r
330                         self.bedTemperatureLabel.Show(True)\r
331                         self.bedTemperatureSelect.Show(True)\r
332                         self.temperaturePanel.Layout()\r
333                 self.statsText.SetLabel(status.strip())\r
334                 #self.Layout()\r
335         \r
336         def OnConnect(self, e):\r
337                 if self.machineCom != None:\r
338                         self.machineCom.close()\r
339                         self.thread.join()\r
340                 self.machineCom = machineCom.MachineCom()\r
341                 self.thread = threading.Thread(target=self.PrinterMonitor)\r
342                 self.thread.start()\r
343                 self.UpdateButtonStates()\r
344         \r
345         def OnLoad(self, e):\r
346                 pass\r
347         \r
348         def OnPrint(self, e):\r
349                 if not self.machineConnected:\r
350                         return\r
351                 if self.gcodeList == None:\r
352                         return\r
353                 if self.printIdx != None:\r
354                         return\r
355                 self.currentZ = -1\r
356                 if self.cam != None:\r
357                         self.cam.startTimelaps(self.filename[: self.filename.rfind('.')] + ".mpg")\r
358                 self.printIdx = 1\r
359                 self.sendLine(0)\r
360                 self.sendCnt = self.bufferLineCount\r
361                 self.UpdateButtonStates()\r
362         \r
363         def OnCancel(self, e):\r
364                 if self.cam != None:\r
365                         self.cam.endTimelaps()\r
366                 self.printIdx = None\r
367                 self.pause = False\r
368                 self.pauseButton.SetLabel('Pause')\r
369                 self.sendCommand("M84")\r
370                 self.UpdateButtonStates()\r
371         \r
372         def OnPause(self, e):\r
373                 if self.pause:\r
374                         self.pause = False\r
375                         self.sendLine(self.printIdx)\r
376                         self.printIdx += 1\r
377                         self.pauseButton.SetLabel('Pause')\r
378                 else:\r
379                         self.pause = True\r
380                         self.pauseButton.SetLabel('Resume')\r
381         \r
382         def OnClose(self, e):\r
383                 global printWindowHandle\r
384                 printWindowHandle = None\r
385                 if self.machineCom != None:\r
386                         self.machineCom.close()\r
387                         self.thread.join()\r
388                 self.Destroy()\r
389 \r
390         def OnTempChange(self, e):\r
391                 self.sendCommand("M104 S%d" % (self.temperatureSelect.GetValue()))\r
392 \r
393         def OnBedTempChange(self, e):\r
394                 self.sendCommand("M140 S%d" % (self.bedTemperatureSelect.GetValue()))\r
395         \r
396         def OnSpeedChange(self, e):\r
397                 self.feedrateRatioOuterWall = self.outerWallSpeedSelect.GetValue() / 100.0\r
398                 self.feedrateRatioInnerWall = self.innerWallSpeedSelect.GetValue() / 100.0\r
399                 self.feedrateRatioFill = self.fillSpeedSelect.GetValue() / 100.0\r
400                 self.feedrateRatioSupport = self.supportSpeedSelect.GetValue() / 100.0\r
401         \r
402         def AddTermLog(self, line):\r
403                 self.termLog.AppendText(unicode(line, 'utf-8', 'replace'))\r
404         \r
405         def OnTermEnterLine(self, e):\r
406                 line = self.termInput.GetValue()\r
407                 if line == '':\r
408                         return\r
409                 self.termLog.AppendText('>%s\n' % (line))\r
410                 self.sendCommand(line)\r
411                 self.termHistory.append(line)\r
412                 self.termHistoryIdx = len(self.termHistory)\r
413                 self.termInput.SetValue('')\r
414 \r
415         def OnTermKey(self, e):\r
416                 if len(self.termHistory) > 0:\r
417                         if e.GetKeyCode() == wx.WXK_UP:\r
418                                 self.termHistoryIdx = self.termHistoryIdx - 1\r
419                                 if self.termHistoryIdx < 0:\r
420                                         self.termHistoryIdx = len(self.termHistory) - 1\r
421                                 self.termInput.SetValue(self.termHistory[self.termHistoryIdx])\r
422                         if e.GetKeyCode() == wx.WXK_DOWN:\r
423                                 self.termHistoryIdx = self.termHistoryIdx - 1\r
424                                 if self.termHistoryIdx >= len(self.termHistory):\r
425                                         self.termHistoryIdx = 0\r
426                                 self.termInput.SetValue(self.termHistory[self.termHistoryIdx])\r
427                 e.Skip()\r
428 \r
429         def LoadGCodeFile(self, filename):\r
430                 if self.printIdx != None:\r
431                         return\r
432                 #Send an initial M110 to reset the line counter to zero.\r
433                 lineType = 'CUSTOM'\r
434                 gcodeList = ["M110"]\r
435                 typeList = [lineType]\r
436                 for line in open(filename, 'r'):\r
437                         if line.startswith(';TYPE:'):\r
438                                 lineType = line[6:].strip()\r
439                         if ';' in line:\r
440                                 line = line[0:line.find(';')]\r
441                         line = line.strip()\r
442                         if len(line) > 0:\r
443                                 gcodeList.append(line)\r
444                                 typeList.append(lineType)\r
445                 gcode = gcodeInterpreter.gcode()\r
446                 gcode.loadList(gcodeList)\r
447                 print "Loaded: %s (%d)" % (filename, len(gcodeList))\r
448                 self.filename = filename\r
449                 self.gcode = gcode\r
450                 self.gcodeList = gcodeList\r
451                 self.typeList = typeList\r
452                 \r
453                 wx.CallAfter(self.progress.SetRange, len(gcodeList))\r
454                 wx.CallAfter(self.UpdateButtonStates)\r
455                 wx.CallAfter(self.UpdateProgress)\r
456                 \r
457         def sendCommand(self, cmd):\r
458                 if self.machineConnected:\r
459                         if self.printIdx == None or self.pause:\r
460                                 self.machineCom.sendCommand(cmd)\r
461                         else:\r
462                                 self.sendList.append(cmd)\r
463 \r
464         def sendLine(self, lineNr):\r
465                 if lineNr >= len(self.gcodeList):\r
466                         return False\r
467                 line = self.gcodeList[lineNr]\r
468                 try:\r
469                         if line == 'M0' or line == 'M1':\r
470                                 self.OnPause(None)\r
471                                 line = 'M105'\r
472                         if ('M104' in line or 'M109' in line) and 'S' in line:\r
473                                 n = int(re.search('S([0-9]*)', line).group(1))\r
474                                 wx.CallAfter(self.temperatureSelect.SetValue, n)\r
475                         if ('M140' in line or 'M190' in line) and 'S' in line:\r
476                                 n = int(re.search('S([0-9]*)', line).group(1))\r
477                                 wx.CallAfter(self.bedTemperatureSelect.SetValue, n)\r
478                         if self.typeList[lineNr] == 'WALL-OUTER':\r
479                                 line = re.sub('F([0-9]*)', lambda m: 'F' + str(int(int(m.group(1)) * self.feedrateRatioOuterWall)), line)\r
480                         if self.typeList[lineNr] == 'WALL-INNER':\r
481                                 line = re.sub('F([0-9]*)', lambda m: 'F' + str(int(int(m.group(1)) * self.feedrateRatioInnerWall)), line)\r
482                         if self.typeList[lineNr] == 'FILL':\r
483                                 line = re.sub('F([0-9]*)', lambda m: 'F' + str(int(int(m.group(1)) * self.feedrateRatioFill)), line)\r
484                         if self.typeList[lineNr] == 'SUPPORT':\r
485                                 line = re.sub('F([0-9]*)', lambda m: 'F' + str(int(int(m.group(1)) * self.feedrateRatioSupport)), line)\r
486                         if 'G1' in line and 'Z' in line:\r
487                                 z = float(re.search('Z([0-9\.]*)', line).group(1))\r
488                                 if self.cam != None and self.currentZ != z:\r
489                                         wx.CallAfter(self.cam.takeNewImage)\r
490                                         wx.CallAfter(self.camPage.Refresh)\r
491                                 self.currentZ = z\r
492                 except:\r
493                         print "Unexpected error:", sys.exc_info()\r
494                 checksum = reduce(lambda x,y:x^y, map(ord, "N%d%s" % (lineNr, line)))\r
495                 self.machineCom.sendCommand("N%d%s*%d" % (lineNr, line, checksum))\r
496                 return True\r
497 \r
498         def PrinterMonitor(self):\r
499                 while True:\r
500                         line = self.machineCom.readline()\r
501                         if line == None:\r
502                                 self.machineConnected = False\r
503                                 wx.CallAfter(self.UpdateButtonStates)\r
504                                 return\r
505                         if line == '':  #When we have a communication "timeout" and we're not sending gcode, then read the temperature.\r
506                                 if self.printIdx == None or self.pause:\r
507                                         self.machineCom.sendCommand("M105")\r
508                                 else:\r
509                                         wx.CallAfter(self.AddTermLog, '!!Comm timeout, forcing next line!!\n')\r
510                                         line = 'ok'\r
511                         if self.machineConnected:\r
512                                 while self.sendCnt > 0 and not self.pause:\r
513                                         self.sendLine(self.printIdx)\r
514                                         self.printIdx += 1\r
515                                         self.sendCnt -= 1\r
516                         if line.startswith("start"):\r
517                                 self.machineConnected = True\r
518                                 wx.CallAfter(self.UpdateButtonStates)\r
519                         elif 'T:' in line:\r
520                                 self.temp = float(re.search("[0-9\.]*", line.split('T:')[1]).group(0))\r
521                                 if 'B:' in line:\r
522                                         self.bedTemp = float(re.search("[0-9\.]*", line.split('B:')[1]).group(0))\r
523                                 self.temperatureGraph.addPoint(self.temp, self.temperatureSelect.GetValue(), self.bedTemp, self.bedTemperatureSelect.GetValue())\r
524                                 wx.CallAfter(self.UpdateProgress)\r
525                         elif line.strip() != 'ok':\r
526                                 wx.CallAfter(self.AddTermLog, line)\r
527                         if self.printIdx != None:\r
528                                 if line.startswith("ok"):\r
529                                         if len(self.sendList) > 0:\r
530                                                 self.machineCom.sendCommand(self.sendList.pop(0))\r
531                                         elif self.pause:\r
532                                                 self.sendCnt += 1\r
533                                         else:\r
534                                                 if self.sendLine(self.printIdx):\r
535                                                         self.printIdx += 1\r
536                                                 else:\r
537                                                         if self.cam != None:\r
538                                                                 self.cam.endTimelaps()\r
539                                                         self.printIdx = None\r
540                                                         wx.CallAfter(self.UpdateButtonStates)\r
541                                                 wx.CallAfter(self.UpdateProgress)\r
542                                 elif "resend" in line.lower() or "rs" in line:\r
543                                         try:\r
544                                                 lineNr=int(line.replace("N:"," ").replace("N"," ").replace(":"," ").split()[-1])\r
545                                         except:\r
546                                                 if "rs" in line:\r
547                                                         lineNr=int(line.split()[1])\r
548                                         self.printIdx = lineNr\r
549                                         #we should actually resend the line here, but we also get an "ok" for each error from Marlin. And thus we'll resend on the OK.\r
550 \r
551 class temperatureGraph(wx.Panel):\r
552         def __init__(self, parent):\r
553                 super(temperatureGraph, self).__init__(parent)\r
554                 \r
555                 self.Bind(wx.EVT_ERASE_BACKGROUND, self.OnEraseBackground)\r
556                 self.Bind(wx.EVT_SIZE, self.OnSize)\r
557                 self.Bind(wx.EVT_PAINT, self.OnDraw)\r
558                 \r
559                 self.lastDraw = time.time() - 1.0\r
560                 self.points = []\r
561                 self.backBuffer = None\r
562                 self.addPoint(0,0,0,0)\r
563         \r
564         def OnEraseBackground(self, e):\r
565                 pass\r
566         \r
567         def OnSize(self, e):\r
568                 if self.backBuffer == None or self.GetSize() != self.backBuffer.GetSize():\r
569                         self.backBuffer = wx.EmptyBitmap(*self.GetSizeTuple())\r
570                         self.UpdateDrawing(True)\r
571         \r
572         def OnDraw(self, e):\r
573                 dc = wx.BufferedPaintDC(self, self.backBuffer)\r
574         \r
575         def UpdateDrawing(self, force = False):\r
576                 now = time.time()\r
577                 if not force and now - self.lastDraw < 1.0:\r
578                         return\r
579                 self.lastDraw = now\r
580                 dc = wx.MemoryDC()\r
581                 dc.SelectObject(self.backBuffer)\r
582                 dc.Clear()\r
583                 w, h = self.GetSizeTuple()\r
584                 x0 = 0\r
585                 t0 = 0\r
586                 bt0 = 0\r
587                 tSP0 = 0\r
588                 btSP0 = 0\r
589                 tempPen = wx.Pen('#FF4040')\r
590                 tempSPPen = wx.Pen('#FFA0A0')\r
591                 tempPenBG = wx.Pen('#FFD0D0')\r
592                 bedTempPen = wx.Pen('#4040FF')\r
593                 bedTempSPPen = wx.Pen('#A0A0FF')\r
594                 bedTempPenBG = wx.Pen('#D0D0FF')\r
595                 for temp, tempSP, bedTemp, bedTempSP, t in self.points:\r
596                         x1 = int(w - (now - t))\r
597                         for x in xrange(x0, x1 + 1):\r
598                                 t = float(x - x0) / float(x1 - x0 + 1) * (temp - t0) + t0\r
599                                 bt = float(x - x0) / float(x1 - x0 + 1) * (bedTemp - bt0) + bt0\r
600                                 tSP = float(x - x0) / float(x1 - x0 + 1) * (tempSP - tSP0) + tSP0\r
601                                 btSP = float(x - x0) / float(x1 - x0 + 1) * (bedTempSP - btSP0) + btSP0\r
602                                 dc.SetPen(tempPenBG)\r
603                                 dc.DrawLine(x, h, x, h - (t * h / 300))\r
604                                 dc.SetPen(bedTempPenBG)\r
605                                 dc.DrawLine(x, h, x, h - (bt * h / 300))\r
606                                 dc.SetPen(tempSPPen)\r
607                                 dc.DrawPoint(x, h - (tSP * h / 300))\r
608                                 dc.SetPen(bedTempSPPen)\r
609                                 dc.DrawPoint(x, h - (btSP * h / 300))\r
610                                 dc.SetPen(tempPen)\r
611                                 dc.DrawPoint(x, h - (t * h / 300))\r
612                                 dc.SetPen(bedTempPen)\r
613                                 dc.DrawPoint(x, h - (bt * h / 300))\r
614                         t0 = temp\r
615                         bt0 = bedTemp\r
616                         tSP0 = tempSP\r
617                         btSP0 = bedTempSP\r
618                         x0 = x1 + 1\r
619                 \r
620                 del dc\r
621                 self.Refresh(eraseBackground=False)\r
622                 self.Update()\r
623                 \r
624                 if len(self.points) > 0 and (time.time() - self.points[0][4]) > w + 20:\r
625                         self.points.pop(0)\r
626 \r
627         def addPoint(self, temp, tempSP, bedTemp, bedTempSP):\r
628                 if bedTemp == None:\r
629                         bedTemp = 0\r
630                 if bedTempSP == None:\r
631                         bedTempSP = 0\r
632                 self.points.append((temp, tempSP, bedTemp, bedTempSP, time.time()))\r
633                 wx.CallAfter(self.UpdateDrawing)\r
634 \r