chiark / gitweb /
Updated webcam support a bit, still needs more work to access properties, and enable...
[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.machineCom == None or self.parent.machineCom.isPrinting():\r
72                         return;\r
73                 if self.command.startswith('G1'):\r
74                         self.parent.machineCom.sendCommand("G91")\r
75                 self.parent.machineCom.sendCommand(self.command)\r
76                 if self.command.startswith('G1'):\r
77                         self.parent.machineCom.sendCommand("G90")\r
78                 e.Skip()\r
79 \r
80 class printWindow(wx.Frame):\r
81         "Main user interface window"\r
82         def __init__(self):\r
83                 super(printWindow, self).__init__(None, -1, title='Printing')\r
84                 self.machineCom = None\r
85                 self.gcode = None\r
86                 self.gcodeList = None\r
87                 self.sendList = []\r
88                 self.temp = None\r
89                 self.bedTemp = None\r
90                 self.bufferLineCount = 4\r
91                 self.sendCnt = 0\r
92                 self.feedrateRatioOuterWall = 1.0\r
93                 self.feedrateRatioInnerWall = 1.0\r
94                 self.feedrateRatioFill = 1.0\r
95                 self.feedrateRatioSupport = 1.0\r
96                 self.pause = False\r
97                 self.termHistory = []\r
98                 self.termHistoryIdx = 0\r
99                 \r
100                 self.cam = None\r
101                 if webcam.hasWebcamSupport():\r
102                         self.cam = webcam.webcam()\r
103 \r
104                 #self.SetIcon(icon.getMainIcon())\r
105                 \r
106                 self.SetSizer(wx.BoxSizer())\r
107                 self.panel = wx.Panel(self)\r
108                 self.GetSizer().Add(self.panel, 1, flag=wx.EXPAND)\r
109                 self.sizer = wx.GridBagSizer(2, 2)\r
110                 self.panel.SetSizer(self.sizer)\r
111                 \r
112                 sb = wx.StaticBox(self.panel, label="Statistics")\r
113                 boxsizer = wx.StaticBoxSizer(sb, wx.VERTICAL)\r
114                 self.statsText = wx.StaticText(self.panel, -1, "Filament: ####.##m #.##g\nPrint time: #####:##\nMachine state: Detecting baudrate")\r
115                 boxsizer.Add(self.statsText, flag=wx.LEFT, border=5)\r
116                 \r
117                 self.sizer.Add(boxsizer, pos=(0,0), span=(5,1), flag=wx.EXPAND)\r
118                 \r
119                 self.connectButton = wx.Button(self.panel, -1, 'Connect')\r
120                 #self.loadButton = wx.Button(self.panel, -1, 'Load GCode')\r
121                 self.printButton = wx.Button(self.panel, -1, 'Print GCode')\r
122                 self.pauseButton = wx.Button(self.panel, -1, 'Pause')\r
123                 self.cancelButton = wx.Button(self.panel, -1, 'Cancel print')\r
124                 self.progress = wx.Gauge(self.panel, -1)\r
125                 \r
126                 self.sizer.Add(self.connectButton, pos=(0,1))\r
127                 #self.sizer.Add(self.loadButton, pos=(1,1))\r
128                 self.sizer.Add(self.printButton, pos=(2,1))\r
129                 self.sizer.Add(self.pauseButton, pos=(3,1))\r
130                 self.sizer.Add(self.cancelButton, pos=(4,1))\r
131                 self.sizer.Add(self.progress, pos=(5,0), span=(1,2), flag=wx.EXPAND)\r
132 \r
133                 nb = wx.Notebook(self.panel)\r
134                 self.sizer.Add(nb, pos=(0,3), span=(7,4), flag=wx.EXPAND)\r
135                 \r
136                 self.temperaturePanel = wx.Panel(nb)\r
137                 sizer = wx.GridBagSizer(2, 2)\r
138                 self.temperaturePanel.SetSizer(sizer)\r
139 \r
140                 self.temperatureSelect = wx.SpinCtrl(self.temperaturePanel, -1, '0', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
141                 self.temperatureSelect.SetRange(0, 400)\r
142                 self.bedTemperatureLabel = wx.StaticText(self.temperaturePanel, -1, "BedTemp:")\r
143                 self.bedTemperatureSelect = wx.SpinCtrl(self.temperaturePanel, -1, '0', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
144                 self.bedTemperatureSelect.SetRange(0, 400)\r
145                 self.bedTemperatureLabel.Show(False)\r
146                 self.bedTemperatureSelect.Show(False)\r
147                 \r
148                 self.temperatureGraph = temperatureGraph(self.temperaturePanel)\r
149                 \r
150                 sizer.Add(wx.StaticText(self.temperaturePanel, -1, "Temp:"), pos=(0,0))\r
151                 sizer.Add(self.temperatureSelect, pos=(0,1))\r
152                 sizer.Add(self.bedTemperatureLabel, pos=(1,0))\r
153                 sizer.Add(self.bedTemperatureSelect, pos=(1,1))\r
154                 sizer.Add(self.temperatureGraph, pos=(2,0), span=(1,2), flag=wx.EXPAND)\r
155                 sizer.AddGrowableRow(2)\r
156                 sizer.AddGrowableCol(0)\r
157 \r
158                 nb.AddPage(self.temperaturePanel, 'Temp')\r
159 \r
160                 self.directControlPanel = wx.Panel(nb)\r
161                 \r
162                 sizer = wx.GridBagSizer(2, 2)\r
163                 self.directControlPanel.SetSizer(sizer)\r
164                 sizer.Add(PrintCommandButton(self, 'G1 Y100 F6000', 'print-move-y100.png'), pos=(0,3))\r
165                 sizer.Add(PrintCommandButton(self, 'G1 Y10 F6000', 'print-move-y10.png'), pos=(1,3))\r
166                 sizer.Add(PrintCommandButton(self, 'G1 Y1 F6000', 'print-move-y1.png'), pos=(2,3))\r
167 \r
168                 sizer.Add(PrintCommandButton(self, 'G1 Y-1 F6000', 'print-move-y-1.png'), pos=(4,3))\r
169                 sizer.Add(PrintCommandButton(self, 'G1 Y-10 F6000', 'print-move-y-10.png'), pos=(5,3))\r
170                 sizer.Add(PrintCommandButton(self, 'G1 Y-100 F6000', 'print-move-y-100.png'), pos=(6,3))\r
171 \r
172                 sizer.Add(PrintCommandButton(self, 'G1 X-100 F6000', 'print-move-x-100.png'), pos=(3,0))\r
173                 sizer.Add(PrintCommandButton(self, 'G1 X-10 F6000', 'print-move-x-10.png'), pos=(3,1))\r
174                 sizer.Add(PrintCommandButton(self, 'G1 X-1 F6000', 'print-move-x-1.png'), pos=(3,2))\r
175 \r
176                 sizer.Add(PrintCommandButton(self, 'G28 X0 Y0', 'print-move-home.png'), pos=(3,3))\r
177 \r
178                 sizer.Add(PrintCommandButton(self, 'G1 X1 F6000', 'print-move-x1.png'), pos=(3,4))\r
179                 sizer.Add(PrintCommandButton(self, 'G1 X10 F6000', 'print-move-x10.png'), pos=(3,5))\r
180                 sizer.Add(PrintCommandButton(self, 'G1 X100 F6000', 'print-move-x100.png'), pos=(3,6))\r
181 \r
182                 sizer.Add(PrintCommandButton(self, 'G1 Z10 F200', 'print-move-z10.png'), pos=(0,7))\r
183                 sizer.Add(PrintCommandButton(self, 'G1 Z1 F200', 'print-move-z1.png'), pos=(1,7))\r
184                 sizer.Add(PrintCommandButton(self, 'G1 Z0.1 F200', 'print-move-z0.1.png'), pos=(2,7))\r
185 \r
186                 sizer.Add(PrintCommandButton(self, 'G28 Z0', 'print-move-home.png'), pos=(3,7))\r
187 \r
188                 sizer.Add(PrintCommandButton(self, 'G1 Z-0.1 F200', 'print-move-z-0.1.png'), pos=(4,7))\r
189                 sizer.Add(PrintCommandButton(self, 'G1 Z-1 F200', 'print-move-z-1.png'), pos=(5,7))\r
190                 sizer.Add(PrintCommandButton(self, 'G1 Z-10 F200', 'print-move-z-10.png'), pos=(6,7))\r
191 \r
192                 nb.AddPage(self.directControlPanel, 'Jog')\r
193 \r
194                 self.speedPanel = wx.Panel(nb)\r
195                 sizer = wx.GridBagSizer(2, 2)\r
196                 self.speedPanel.SetSizer(sizer)\r
197 \r
198                 self.outerWallSpeedSelect = wx.SpinCtrl(self.speedPanel, -1, '100', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
199                 self.outerWallSpeedSelect.SetRange(5, 1000)\r
200                 self.innerWallSpeedSelect = wx.SpinCtrl(self.speedPanel, -1, '100', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
201                 self.innerWallSpeedSelect.SetRange(5, 1000)\r
202                 self.fillSpeedSelect = wx.SpinCtrl(self.speedPanel, -1, '100', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
203                 self.fillSpeedSelect.SetRange(5, 1000)\r
204                 self.supportSpeedSelect = wx.SpinCtrl(self.speedPanel, -1, '100', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
205                 self.supportSpeedSelect.SetRange(5, 1000)\r
206                 \r
207                 sizer.Add(wx.StaticText(self.speedPanel, -1, "Outer wall:"), pos=(0,0))\r
208                 sizer.Add(self.outerWallSpeedSelect, pos=(0,1))\r
209                 sizer.Add(wx.StaticText(self.speedPanel, -1, "%"), pos=(0,2))\r
210                 sizer.Add(wx.StaticText(self.speedPanel, -1, "Inner wall:"), pos=(1,0))\r
211                 sizer.Add(self.innerWallSpeedSelect, pos=(1,1))\r
212                 sizer.Add(wx.StaticText(self.speedPanel, -1, "%"), pos=(1,2))\r
213                 sizer.Add(wx.StaticText(self.speedPanel, -1, "Fill:"), pos=(2,0))\r
214                 sizer.Add(self.fillSpeedSelect, pos=(2,1))\r
215                 sizer.Add(wx.StaticText(self.speedPanel, -1, "%"), pos=(2,2))\r
216                 sizer.Add(wx.StaticText(self.speedPanel, -1, "Support:"), pos=(3,0))\r
217                 sizer.Add(self.supportSpeedSelect, pos=(3,1))\r
218                 sizer.Add(wx.StaticText(self.speedPanel, -1, "%"), pos=(3,2))\r
219 \r
220                 nb.AddPage(self.speedPanel, 'Speed')\r
221                 \r
222                 self.termPanel = wx.Panel(nb)\r
223                 sizer = wx.GridBagSizer(2, 2)\r
224                 self.termPanel.SetSizer(sizer)\r
225                 \r
226                 f = wx.Font(8, wx.FONTFAMILY_MODERN, wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_NORMAL, False)\r
227                 self.termLog = wx.TextCtrl(self.termPanel, style=wx.TE_MULTILINE|wx.TE_DONTWRAP)\r
228                 self.termLog.SetFont(f)\r
229                 self.termLog.SetEditable(0)\r
230                 self.termInput = wx.TextCtrl(self.termPanel, style=wx.TE_PROCESS_ENTER)\r
231                 self.termInput.SetFont(f)\r
232 \r
233                 sizer.Add(self.termLog, pos=(0,0),flag=wx.EXPAND)\r
234                 sizer.Add(self.termInput, pos=(1,0),flag=wx.EXPAND)\r
235                 sizer.AddGrowableCol(0)\r
236                 sizer.AddGrowableRow(0)\r
237 \r
238                 nb.AddPage(self.termPanel, 'Term')\r
239                 \r
240                 if self.cam != None and self.cam.hasCamera():\r
241                         self.camPage = wx.Panel(nb)\r
242                         sizer = wx.GridBagSizer(2, 2)\r
243                         self.camPage.SetSizer(sizer)\r
244                         \r
245                         self.camPreview = wx.Panel(self.camPage)\r
246                         sizer.Add(self.camPreview, pos=(0,0), flag=wx.EXPAND)\r
247                         \r
248                         nb.AddPage(self.camPage, 'Camera')\r
249                         self.camPreview.timer = wx.Timer(self)\r
250                         self.Bind(wx.EVT_TIMER, self.OnCameraTimer, self.camPreview.timer)\r
251                         self.camPreview.timer.Start(500)\r
252                         self.camPreview.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.machineCom != None and not self.machineCom.isPrinting():\r
283                         return\r
284                 self.cam.takeNewImage()\r
285                 self.camPreview.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.camPreview.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(self.machineCom == None or self.machineCom.isClosedOrError())\r
300                 #self.loadButton.Enable(self.machineCom == None or not (self.machineCom.isPrinting() or self.machineCom.isPaused()))\r
301                 self.printButton.Enable(self.machineCom != None and self.machineCom.isOperational() and not (self.machineCom.isPrinting() or self.machineCom.isPaused()))\r
302                 self.pauseButton.Enable(self.machineCom != None and (self.machineCom.isPrinting() or self.machineCom.isPaused()))\r
303                 if self.machineCom != None and self.machineCom.isPaused():\r
304                         self.pauseButton.SetLabel('Resume')\r
305                 else:\r
306                         self.pauseButton.SetLabel('Pause')\r
307                 self.cancelButton.Enable(self.machineCom != None and (self.machineCom.isPrinting() or self.machineCom.isPaused()))\r
308                 self.temperatureSelect.Enable(self.machineCom != None and self.machineCom.isOperational())\r
309                 self.bedTemperatureSelect.Enable(self.machineCom != None and self.machineCom.isOperational())\r
310                 self.directControlPanel.Enable(self.machineCom != None and self.machineCom.isOperational())\r
311         \r
312         def UpdateProgress(self):\r
313                 status = ""\r
314                 if self.gcode == None:\r
315                         status += "Loading gcode...\n"\r
316                 else:\r
317                         status += "Filament: %.2fm %.2fg\n" % (self.gcode.extrusionAmount / 1000, self.gcode.calculateWeight() * 1000)\r
318                         cost = self.gcode.calculateCost()\r
319                         if cost != False:\r
320                                 status += "Filament cost: %s\n" % (cost)\r
321                         status += "Print time: %02d:%02d\n" % (int(self.gcode.totalMoveTimeMinute / 60), int(self.gcode.totalMoveTimeMinute % 60))\r
322                 if self.machineCom == None or not self.machineCom.isPrinting():\r
323                         self.progress.SetValue(0)\r
324                         if self.gcodeList != None:\r
325                                 status += 'Line: -/%d\n' % (len(self.gcodeList))\r
326                 else:\r
327                         status += 'Line: %d/%d\n' % (self.machineCom.getPrintPos(), len(self.gcodeList))\r
328                         status += 'Height: %f\n' % (self.currentZ)\r
329                         self.progress.SetValue(self.machineCom.getPrintPos())\r
330                 if self.machineCom != None:\r
331                         if self.machineCom.getTemp() > 0:\r
332                                 status += 'Temp: %d\n' % (self.machineCom.getTemp())\r
333                         if self.machineCom.getBedTemp() > 0:\r
334                                 status += 'Bed Temp: %d\n' % (self.machineCom.getBedTemp())\r
335                                 self.bedTemperatureLabel.Show(True)\r
336                                 self.bedTemperatureSelect.Show(True)\r
337                                 self.temperaturePanel.Layout()\r
338                         status += 'Machine state: %s\n' % (self.machineCom.getStateString())\r
339                 \r
340                 self.statsText.SetLabel(status.strip())\r
341         \r
342         def OnConnect(self, e):\r
343                 if self.machineCom != None:\r
344                         self.machineCom.close()\r
345                 self.machineCom = machineCom.MachineCom(callbackObject=self)\r
346                 self.UpdateButtonStates()\r
347         \r
348         def OnLoad(self, e):\r
349                 pass\r
350         \r
351         def OnPrint(self, e):\r
352                 if self.machineCom == None or not self.machineCom.isOperational():\r
353                         return\r
354                 if self.gcodeList == None:\r
355                         return\r
356                 if self.machineCom.isPrinting():\r
357                         return\r
358                 self.currentZ = -1\r
359                 if self.cam != None:\r
360                         self.cam.startTimelaps(self.filename[: self.filename.rfind('.')] + ".mpg")\r
361                 self.machineCom.printGCode(self.gcodeList)\r
362                 self.UpdateButtonStates()\r
363         \r
364         def OnCancel(self, e):\r
365                 if self.cam != None:\r
366                         self.cam.endTimelaps()\r
367                 self.pauseButton.SetLabel('Pause')\r
368                 self.machineCom.cancelPrint()\r
369                 self.machineCom.sendCommand("M84")\r
370                 self.UpdateButtonStates()\r
371         \r
372         def OnPause(self, e):\r
373                 if self.machineCom.isPaused():\r
374                         self.machineCom.setPause(False)\r
375                 else:\r
376                         self.machineCom.setPause(True)\r
377         \r
378         def OnClose(self, e):\r
379                 global printWindowHandle\r
380                 printWindowHandle = None\r
381                 if self.machineCom != None:\r
382                         self.machineCom.close()\r
383                 self.Destroy()\r
384 \r
385         def OnTempChange(self, e):\r
386                 self.machineCom.sendCommand("M104 S%d" % (self.temperatureSelect.GetValue()))\r
387 \r
388         def OnBedTempChange(self, e):\r
389                 self.machineCom.sendCommand("M140 S%d" % (self.bedTemperatureSelect.GetValue()))\r
390         \r
391         def OnSpeedChange(self, e):\r
392                 if self.machineCom == None:\r
393                         return\r
394                 self.machineCom.setFeedrateModifier('WALL-OUTER', self.outerWallSpeedSelect.GetValue() / 100.0)\r
395                 self.machineCom.setFeedrateModifier('WALL-INNER', self.innerWallSpeedSelect.GetValue() / 100.0)\r
396                 self.machineCom.setFeedrateModifier('FILL', self.fillSpeedSelect.GetValue() / 100.0)\r
397                 self.machineCom.setFeedrateModifier('SUPPORT', self.supportSpeedSelect.GetValue() / 100.0)\r
398         \r
399         def AddTermLog(self, line):\r
400                 self.termLog.AppendText(unicode(line, 'utf-8', 'replace'))\r
401         \r
402         def OnTermEnterLine(self, e):\r
403                 line = self.termInput.GetValue()\r
404                 if line == '':\r
405                         return\r
406                 self.termLog.AppendText('>%s\n' % (line))\r
407                 self.machineCom.sendCommand(line)\r
408                 self.termHistory.append(line)\r
409                 self.termHistoryIdx = len(self.termHistory)\r
410                 self.termInput.SetValue('')\r
411 \r
412         def OnTermKey(self, e):\r
413                 if len(self.termHistory) > 0:\r
414                         if e.GetKeyCode() == wx.WXK_UP:\r
415                                 self.termHistoryIdx = self.termHistoryIdx - 1\r
416                                 if self.termHistoryIdx < 0:\r
417                                         self.termHistoryIdx = len(self.termHistory) - 1\r
418                                 self.termInput.SetValue(self.termHistory[self.termHistoryIdx])\r
419                         if e.GetKeyCode() == wx.WXK_DOWN:\r
420                                 self.termHistoryIdx = self.termHistoryIdx - 1\r
421                                 if self.termHistoryIdx >= len(self.termHistory):\r
422                                         self.termHistoryIdx = 0\r
423                                 self.termInput.SetValue(self.termHistory[self.termHistoryIdx])\r
424                 e.Skip()\r
425 \r
426         def LoadGCodeFile(self, filename):\r
427                 if self.machineCom != None and self.machineCom.isPrinting():\r
428                         return\r
429                 #Send an initial M110 to reset the line counter to zero.\r
430                 prevLineType = lineType = 'CUSTOM'\r
431                 gcodeList = ["M110"]\r
432                 for line in open(filename, 'r'):\r
433                         if line.startswith(';TYPE:'):\r
434                                 lineType = line[6:].strip()\r
435                         if ';' in line:\r
436                                 line = line[0:line.find(';')]\r
437                         line = line.strip()\r
438                         if len(line) > 0:\r
439                                 if prevLineType != lineType:\r
440                                         gcodeList.append((line, lineType, ))\r
441                                 else:\r
442                                         gcodeList.append(line)\r
443                                 prevLineType = lineType\r
444                 gcode = gcodeInterpreter.gcode()\r
445                 gcode.loadList(gcodeList)\r
446                 print "Loaded: %s (%d)" % (filename, len(gcodeList))\r
447                 self.filename = filename\r
448                 self.gcode = gcode\r
449                 self.gcodeList = gcodeList\r
450                 \r
451                 wx.CallAfter(self.progress.SetRange, len(gcodeList))\r
452                 wx.CallAfter(self.UpdateButtonStates)\r
453                 wx.CallAfter(self.UpdateProgress)\r
454                 \r
455         def sendLine(self, lineNr):\r
456                 if lineNr >= len(self.gcodeList):\r
457                         return False\r
458                 line = self.gcodeList[lineNr]\r
459                 try:\r
460                         if ('M104' in line or 'M109' in line) and 'S' in line:\r
461                                 n = int(re.search('S([0-9]*)', line).group(1))\r
462                                 wx.CallAfter(self.temperatureSelect.SetValue, n)\r
463                         if ('M140' in line or 'M190' in line) and 'S' in line:\r
464                                 n = int(re.search('S([0-9]*)', line).group(1))\r
465                                 wx.CallAfter(self.bedTemperatureSelect.SetValue, n)\r
466                 except:\r
467                         print "Unexpected error:", sys.exc_info()\r
468                 checksum = reduce(lambda x,y:x^y, map(ord, "N%d%s" % (lineNr, line)))\r
469                 self.machineCom.sendCommand("N%d%s*%d" % (lineNr, line, checksum))\r
470                 return True\r
471 \r
472         def mcLog(self, message):\r
473                 print message\r
474         \r
475         def mcTempUpdate(self, temp, bedTemp):\r
476                 self.temperatureGraph.addPoint(temp, self.temperatureSelect.GetValue(), bedTemp, self.bedTemperatureSelect.GetValue())\r
477         \r
478         def mcStateChange(self, state):\r
479                 wx.CallAfter(self.UpdateButtonStates)\r
480                 wx.CallAfter(self.UpdateProgress)\r
481         \r
482         def mcMessage(self, message):\r
483                 wx.CallAfter(self.AddTermLog, message)\r
484         \r
485         def mcProgress(self, lineNr):\r
486                 wx.CallAfter(self.UpdateProgress)\r
487         \r
488         def mcZChange(self, newZ):\r
489                 if self.cam != None:\r
490                         wx.CallAfter(self.cam.takeNewImage)\r
491                         wx.CallAfter(self.camPreview.Refresh)\r
492 \r
493 class temperatureGraph(wx.Panel):\r
494         def __init__(self, parent):\r
495                 super(temperatureGraph, self).__init__(parent)\r
496                 \r
497                 self.Bind(wx.EVT_ERASE_BACKGROUND, self.OnEraseBackground)\r
498                 self.Bind(wx.EVT_SIZE, self.OnSize)\r
499                 self.Bind(wx.EVT_PAINT, self.OnDraw)\r
500                 \r
501                 self.lastDraw = time.time() - 1.0\r
502                 self.points = []\r
503                 self.backBuffer = None\r
504                 self.addPoint(0,0,0,0)\r
505         \r
506         def OnEraseBackground(self, e):\r
507                 pass\r
508         \r
509         def OnSize(self, e):\r
510                 if self.backBuffer == None or self.GetSize() != self.backBuffer.GetSize():\r
511                         self.backBuffer = wx.EmptyBitmap(*self.GetSizeTuple())\r
512                         self.UpdateDrawing(True)\r
513         \r
514         def OnDraw(self, e):\r
515                 dc = wx.BufferedPaintDC(self, self.backBuffer)\r
516         \r
517         def UpdateDrawing(self, force = False):\r
518                 now = time.time()\r
519                 if not force and now - self.lastDraw < 1.0:\r
520                         return\r
521                 self.lastDraw = now\r
522                 dc = wx.MemoryDC()\r
523                 dc.SelectObject(self.backBuffer)\r
524                 dc.Clear()\r
525                 w, h = self.GetSizeTuple()\r
526                 x0 = 0\r
527                 t0 = 0\r
528                 bt0 = 0\r
529                 tSP0 = 0\r
530                 btSP0 = 0\r
531                 tempPen = wx.Pen('#FF4040')\r
532                 tempSPPen = wx.Pen('#FFA0A0')\r
533                 tempPenBG = wx.Pen('#FFD0D0')\r
534                 bedTempPen = wx.Pen('#4040FF')\r
535                 bedTempSPPen = wx.Pen('#A0A0FF')\r
536                 bedTempPenBG = wx.Pen('#D0D0FF')\r
537                 for temp, tempSP, bedTemp, bedTempSP, t in self.points:\r
538                         x1 = int(w - (now - t))\r
539                         for x in xrange(x0, x1 + 1):\r
540                                 t = float(x - x0) / float(x1 - x0 + 1) * (temp - t0) + t0\r
541                                 bt = float(x - x0) / float(x1 - x0 + 1) * (bedTemp - bt0) + bt0\r
542                                 tSP = float(x - x0) / float(x1 - x0 + 1) * (tempSP - tSP0) + tSP0\r
543                                 btSP = float(x - x0) / float(x1 - x0 + 1) * (bedTempSP - btSP0) + btSP0\r
544                                 dc.SetPen(tempPenBG)\r
545                                 dc.DrawLine(x, h, x, h - (t * h / 300))\r
546                                 dc.SetPen(bedTempPenBG)\r
547                                 dc.DrawLine(x, h, x, h - (bt * h / 300))\r
548                                 dc.SetPen(tempSPPen)\r
549                                 dc.DrawPoint(x, h - (tSP * h / 300))\r
550                                 dc.SetPen(bedTempSPPen)\r
551                                 dc.DrawPoint(x, h - (btSP * h / 300))\r
552                                 dc.SetPen(tempPen)\r
553                                 dc.DrawPoint(x, h - (t * h / 300))\r
554                                 dc.SetPen(bedTempPen)\r
555                                 dc.DrawPoint(x, h - (bt * h / 300))\r
556                         t0 = temp\r
557                         bt0 = bedTemp\r
558                         tSP0 = tempSP\r
559                         btSP0 = bedTempSP\r
560                         x0 = x1 + 1\r
561                 \r
562                 del dc\r
563                 self.Refresh(eraseBackground=False)\r
564                 self.Update()\r
565                 \r
566                 if len(self.points) > 0 and (time.time() - self.points[0][4]) > w + 20:\r
567                         self.points.pop(0)\r
568 \r
569         def addPoint(self, temp, tempSP, bedTemp, bedTempSP):\r
570                 if bedTemp == None:\r
571                         bedTemp = 0\r
572                 if bedTempSP == None:\r
573                         bedTempSP = 0\r
574                 self.points.append((temp, tempSP, bedTemp, bedTempSP, time.time()))\r
575                 wx.CallAfter(self.UpdateDrawing)\r
576 \r