1 from __future__ import absolute_import
\r
4 import wx, threading, re, subprocess, sys, os
\r
5 from wx.lib import buttons
\r
8 from gui import toolbarUtil
\r
9 from util import machineCom
\r
10 from util import profile
\r
11 from util import gcodeInterpreter
\r
13 printWindowMonitorHandle = None
\r
15 def printFile(filename):
\r
16 global printWindowMonitorHandle
\r
17 if printWindowMonitorHandle == None:
\r
18 printWindowMonitorHandle = printProcessMonitor()
\r
19 printWindowMonitorHandle.loadFile(filename)
\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
26 printWindowHandle = printWindow()
\r
27 printWindowHandle.Show(True)
\r
28 printWindowHandle.Raise()
\r
29 printWindowHandle.OnConnect(None)
\r
30 printWindowHandle.LoadGCodeFile(filename)
\r
33 class printProcessMonitor():
\r
37 def loadFile(self, filename):
\r
38 if self.handle == None:
\r
39 self.handle = subprocess.Popen([sys.executable, sys.argv[0], '-r', filename], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
\r
40 self.thread = threading.Thread(target=self.Monitor)
\r
43 self.handle.stdin.write(filename + '\n')
\r
47 line = p.stdout.readline()
\r
48 while(len(line) > 0):
\r
50 line = p.stdout.readline()
\r
55 class PrintCommandButton(buttons.GenBitmapButton):
\r
56 def __init__(self, parent, command, bitmapFilename, size=(20,20)):
\r
57 self.bitmap = toolbarUtil.getBitmapImage(bitmapFilename)
\r
58 super(PrintCommandButton, self).__init__(parent.directControlPanel, -1, self.bitmap, size=size)
\r
60 self.command = command
\r
61 self.parent = parent
\r
63 self.SetBezelWidth(1)
\r
64 self.SetUseFocusIndicator(False)
\r
66 self.Bind(wx.EVT_BUTTON, self.OnClick)
\r
68 def OnClick(self, e):
\r
69 self.parent.sendCommand("G91")
\r
70 self.parent.sendCommand(self.command)
\r
73 class printWindow(wx.Frame):
\r
74 "Main user interface window"
\r
76 super(printWindow, self).__init__(None, -1, title='Printing')
\r
77 self.machineCom = None
\r
78 self.machineConnected = False
\r
81 self.gcodeList = None
\r
83 self.printIdx = None
\r
86 self.bufferLineCount = 4
\r
89 #self.SetIcon(icon.getMainIcon())
\r
91 self.SetSizer(wx.BoxSizer())
\r
92 self.panel = wx.Panel(self)
\r
93 self.GetSizer().Add(self.panel, 1, flag=wx.EXPAND)
\r
94 self.sizer = wx.GridBagSizer(2, 2)
\r
95 self.panel.SetSizer(self.sizer)
\r
97 sb = wx.StaticBox(self.panel, label="Statistics")
\r
98 boxsizer = wx.StaticBoxSizer(sb, wx.VERTICAL)
\r
99 self.statsText = wx.StaticText(self.panel, -1, "Filament: ####.##m #.##g\nPrint time: #####:##")
\r
100 boxsizer.Add(self.statsText, flag=wx.LEFT, border=5)
\r
102 self.sizer.Add(boxsizer, pos=(0,0), span=(4,1), flag=wx.EXPAND)
\r
104 self.connectButton = wx.Button(self.panel, -1, 'Connect')
\r
105 #self.loadButton = wx.Button(self.panel, -1, 'Load GCode')
\r
106 self.printButton = wx.Button(self.panel, -1, 'Print GCode')
\r
107 self.cancelButton = wx.Button(self.panel, -1, 'Cancel print')
\r
108 self.progress = wx.Gauge(self.panel, -1)
\r
110 h = self.connectButton.GetSize().GetHeight()
\r
111 self.temperatureSelect = wx.SpinCtrl(self.panel, -1, '0', size=(21*3,21), style=wx.SP_ARROW_KEYS)
\r
112 self.temperatureSelect.SetRange(0, 400)
\r
113 self.bedTemperatureLabel = wx.StaticText(self.panel, -1, "BedTemp:")
\r
114 self.bedTemperatureSelect = wx.SpinCtrl(self.panel, -1, '0', size=(21*3,21), style=wx.SP_ARROW_KEYS)
\r
115 self.bedTemperatureSelect.SetRange(0, 400)
\r
116 self.bedTemperatureLabel.Show(False)
\r
117 self.bedTemperatureSelect.Show(False)
\r
119 self.sizer.Add(self.connectButton, pos=(0,1))
\r
120 #self.sizer.Add(self.loadButton, pos=(1,1))
\r
121 self.sizer.Add(self.printButton, pos=(2,1))
\r
122 self.sizer.Add(self.cancelButton, pos=(3,1))
\r
123 self.sizer.Add(self.progress, pos=(4,0), span=(1,2), flag=wx.EXPAND)
\r
125 self.sizer.Add(wx.StaticText(self.panel, -1, "Temp:"), pos=(0,3))
\r
126 self.sizer.Add(self.temperatureSelect, pos=(0,4))
\r
127 self.sizer.Add(self.bedTemperatureLabel, pos=(0,5))
\r
128 self.sizer.Add(self.bedTemperatureSelect, pos=(0,6))
\r
130 self.directControlPanel = wx.Panel(self.panel)
\r
131 self.sizer.Add(self.directControlPanel, pos=(1,3), span=(5,4))
\r
133 sizer = wx.GridBagSizer(2, 2)
\r
134 self.directControlPanel.SetSizer(sizer)
\r
135 sizer.Add(PrintCommandButton(self, 'G1 Y100 F6000', 'print-move-y100.png'), pos=(0,3))
\r
136 sizer.Add(PrintCommandButton(self, 'G1 Y10 F6000', 'print-move-y10.png'), pos=(1,3))
\r
137 sizer.Add(PrintCommandButton(self, 'G1 Y1 F6000', 'print-move-y1.png'), pos=(2,3))
\r
139 sizer.Add(PrintCommandButton(self, 'G1 Y-1 F6000', 'print-move-y-1.png'), pos=(4,3))
\r
140 sizer.Add(PrintCommandButton(self, 'G1 Y-10 F6000', 'print-move-y-10.png'), pos=(5,3))
\r
141 sizer.Add(PrintCommandButton(self, 'G1 Y-100 F6000', 'print-move-y-100.png'), pos=(6,3))
\r
143 sizer.Add(PrintCommandButton(self, 'G1 X-100 F6000', 'print-move-x-100.png'), pos=(3,0))
\r
144 sizer.Add(PrintCommandButton(self, 'G1 X-10 F6000', 'print-move-x-10.png'), pos=(3,1))
\r
145 sizer.Add(PrintCommandButton(self, 'G1 X-1 F6000', 'print-move-x-1.png'), pos=(3,2))
\r
147 sizer.Add(PrintCommandButton(self, 'G28 X0 Y0', 'print-move-home.png'), pos=(3,3))
\r
149 sizer.Add(PrintCommandButton(self, 'G1 X1 F6000', 'print-move-x1.png'), pos=(3,4))
\r
150 sizer.Add(PrintCommandButton(self, 'G1 X10 F6000', 'print-move-x10.png'), pos=(3,5))
\r
151 sizer.Add(PrintCommandButton(self, 'G1 X100 F6000', 'print-move-x100.png'), pos=(3,6))
\r
153 sizer.Add(PrintCommandButton(self, 'G1 Z10 F200', 'print-move-z10.png'), pos=(0,7))
\r
154 sizer.Add(PrintCommandButton(self, 'G1 Z1 F200', 'print-move-z1.png'), pos=(1,7))
\r
155 sizer.Add(PrintCommandButton(self, 'G1 Z0.1 F200', 'print-move-z0.1.png'), pos=(2,7))
\r
157 sizer.Add(PrintCommandButton(self, 'G28 Z0', 'print-move-home.png'), pos=(3,7))
\r
159 sizer.Add(PrintCommandButton(self, 'G1 Z-0.1 F200', 'print-move-z-0.1.png'), pos=(4,7))
\r
160 sizer.Add(PrintCommandButton(self, 'G1 Z-1 F200', 'print-move-z-1.png'), pos=(5,7))
\r
161 sizer.Add(PrintCommandButton(self, 'G1 Z-10 F200', 'print-move-z-10.png'), pos=(6,7))
\r
163 self.sizer.AddGrowableRow(3)
\r
164 self.sizer.AddGrowableCol(0)
\r
166 self.Bind(wx.EVT_CLOSE, self.OnClose)
\r
167 self.connectButton.Bind(wx.EVT_BUTTON, self.OnConnect)
\r
168 #self.loadButton.Bind(wx.EVT_BUTTON, self.OnLoad)
\r
169 self.printButton.Bind(wx.EVT_BUTTON, self.OnPrint)
\r
170 self.cancelButton.Bind(wx.EVT_BUTTON, self.OnCancel)
\r
172 self.Bind(wx.EVT_SPINCTRL, self.OnTempChange, self.temperatureSelect)
\r
173 self.Bind(wx.EVT_SPINCTRL, self.OnBedTempChange, self.bedTemperatureSelect)
\r
179 self.UpdateButtonStates()
\r
180 self.UpdateProgress()
\r
182 def UpdateButtonStates(self):
\r
183 self.connectButton.Enable(not self.machineConnected)
\r
184 #self.loadButton.Enable(self.printIdx == None)
\r
185 self.printButton.Enable(self.machineConnected and self.gcodeList != None and self.printIdx == None)
\r
186 self.cancelButton.Enable(self.printIdx != None)
\r
187 self.temperatureSelect.Enable(self.machineConnected)
\r
188 self.directControlPanel.Enable(self.machineConnected)
\r
190 def UpdateProgress(self):
\r
192 if self.gcode != None:
\r
193 status += "Filament: %.2fm %.2fg\n" % (self.gcode.extrusionAmount / 1000, self.gcode.calculateWeight() * 1000)
\r
194 cost = self.gcode.calculateCost()
\r
196 status += "Filament cost: %s\n" % (cost)
\r
197 status += "Print time: %02d:%02d\n" % (int(self.gcode.totalMoveTimeMinute / 60), int(self.gcode.totalMoveTimeMinute % 60))
\r
198 if self.printIdx == None:
\r
199 self.progress.SetValue(0)
\r
200 if self.gcodeList != None:
\r
201 status += 'Line: -/%d\n' % (len(self.gcodeList))
\r
203 status += 'Line: %d/%d\n' % (self.printIdx, len(self.gcodeList))
\r
204 self.progress.SetValue(self.printIdx)
\r
205 if self.temp != None:
\r
206 status += 'Temp: %d\n' % (self.temp)
\r
207 if self.bedTemp != None and self.bedTemp > 0:
\r
208 status += 'Bed Temp: %d\n' % (self.bedTemp)
\r
209 self.bedTemperatureLabel.Show(True)
\r
210 self.bedTemperatureSelect.Show(True)
\r
211 self.statsText.SetLabel(status.strip())
\r
214 def OnConnect(self, e):
\r
215 if self.machineCom != None:
\r
216 self.machineCom.close()
\r
218 self.machineCom = machineCom.MachineCom()
\r
219 self.thread = threading.Thread(target=self.PrinterMonitor)
\r
220 self.thread.start()
\r
221 self.UpdateButtonStates()
\r
223 def OnLoad(self, e):
\r
226 def OnPrint(self, e):
\r
227 if not self.machineConnected:
\r
229 if self.gcodeList == None:
\r
231 if self.printIdx != None:
\r
235 self.sendCnt = self.bufferLineCount
\r
236 self.UpdateButtonStates()
\r
238 def OnCancel(self, e):
\r
239 self.printIdx = None
\r
240 self.sendCommand("M84")
\r
241 self.UpdateButtonStates()
\r
243 def OnClose(self, e):
\r
244 global printWindowHandle
\r
245 printWindowHandle = None
\r
246 if self.machineCom != None:
\r
247 self.machineCom.close()
\r
251 def OnTempChange(self, e):
\r
252 self.sendCommand("M104 S%d" % (self.temperatureSelect.GetValue()))
\r
254 def OnBedTempChange(self, e):
\r
255 self.sendCommand("M140 S%d" % (self.bedTemperatureSelect.GetValue()))
\r
257 def LoadGCodeFile(self, filename):
\r
258 if self.printIdx != None:
\r
260 #Send an initial M110 to reset the line counter to zero.
\r
261 gcodeList = ["M110"]
\r
262 for line in open(filename, 'r'):
\r
264 line = line[0:line.find(';')]
\r
265 line = line.strip()
\r
267 gcodeList.append(line)
\r
268 gcode = gcodeInterpreter.gcode()
\r
269 gcode.loadList(gcodeList)
\r
270 print "Loaded: %s (%d)" % (filename, len(gcodeList))
\r
271 self.progress.SetRange(len(gcodeList))
\r
273 self.gcodeList = gcodeList
\r
274 self.UpdateButtonStates()
\r
275 self.UpdateProgress()
\r
277 def sendCommand(self, cmd):
\r
278 if self.machineConnected:
\r
279 if self.printIdx == None:
\r
280 self.machineCom.sendCommand(cmd)
\r
282 self.sendList.append(cmd)
\r
284 def sendLine(self, lineNr):
\r
285 if lineNr >= len(self.gcodeList):
\r
287 checksum = reduce(lambda x,y:x^y, map(ord, "N%d%s" % (lineNr, self.gcodeList[lineNr])))
\r
288 self.machineCom.sendCommand("N%d%s*%d" % (lineNr, self.gcodeList[lineNr], checksum))
\r
291 def PrinterMonitor(self):
\r
293 line = self.machineCom.readline()
\r
295 self.machineConnected = False
\r
296 wx.CallAfter(self.UpdateButtonStates)
\r
298 if self.machineConnected:
\r
299 while self.sendCnt > 0:
\r
300 self.sendLine(self.printIdx)
\r
303 elif line.startswith("start"):
\r
304 self.machineConnected = True
\r
305 wx.CallAfter(self.UpdateButtonStates)
\r
307 self.temp = float(re.search("[0-9\.]*", line.split('T:')[1]).group(0))
\r
309 self.bedTemp = float(re.search("[0-9\.]*", line.split('B:')[1]).group(0))
\r
310 wx.CallAfter(self.UpdateProgress)
\r
311 if self.printIdx == None:
\r
312 if line == '': #When we have a communication "timeout" and we're not sending gcode, then read the temperature.
\r
313 self.machineCom.sendCommand("M105")
\r
315 if line.startswith("ok"):
\r
316 if len(self.sendList) > 0:
\r
317 self.machineCom.sendCommand(self.sendList.pop(0))
\r
319 if self.sendLine(self.printIdx):
\r
322 self.printIdx = None
\r
323 wx.CallAfter(self.UpdateButtonStates)
\r
324 wx.CallAfter(self.UpdateProgress)
\r
325 elif "resend" in line.lower() or "rs" in line:
\r
327 lineNr=int(line.replace("N:"," ").replace("N"," ").replace(":"," ").split()[-1])
\r
330 lineNr=int(line.split()[1])
\r
331 self.printIdx = lineNr
\r
332 #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