Cura/gui/printWindow.py

   1 from __future__ import absolute_import\r
   2 import __init__\r
   3 \r
   4 import wx, threading, re\r
   5 \r
   6 from gui import machineCom\r
   7 from gui import icon\r
   8 from util import profile\r
   9 from util import gcodeInterpreter\r
  10 \r
  11 printWindowHandle = None\r
  12 \r
  13 def printFile(filename):\r
  14         global printWindowHandle\r
  15         if printWindowHandle == None:\r
  16                 printWindowHandle = printWindow()\r
  17                 printWindowHandle.OnConnect(None)\r
  18         printWindowHandle.Show(True)\r
  19         printWindowHandle.Raise()\r
  20         printWindowHandle.LoadGCodeFile(filename)\r
  21 \r
  22 class printWindow(wx.Frame):\r
  23         "Main user interface window"\r
  24         def __init__(self):\r
  25                 super(printWindow, self).__init__(None, -1, title='Printing')\r
  26                 self.machineCom = None\r
  27                 self.machineConnected = False\r
  28                 self.thread = None\r
  29                 self.gcode = None\r
  30                 self.gcodeList = None\r
  31                 self.sendList = []\r
  32                 self.printIdx = None\r
  33                 self.temp = None\r
  34                 self.bufferLineCount = 4\r
  35                 self.sendCnt = 0\r
  36 \r
  37                 #self.SetIcon(icon.getMainIcon())\r
  38                 \r
  39                 self.SetSizer(wx.BoxSizer())\r
  40                 self.panel = wx.Panel(self)\r
  41                 self.GetSizer().Add(self.panel, 1, flag=wx.EXPAND)\r
  42                 self.sizer = wx.GridBagSizer(2, 2)\r
  43                 self.panel.SetSizer(self.sizer)\r
  44                 \r
  45                 sb = wx.StaticBox(self.panel, label="Statistics")\r
  46                 boxsizer = wx.StaticBoxSizer(sb, wx.VERTICAL)\r
  47                 self.statsText = wx.StaticText(self.panel, -1, "Filament: ####.##m #.##g\nPrint time: #####:##")\r
  48                 boxsizer.Add(self.statsText, flag=wx.LEFT, border=5)\r
  49                 \r
  50                 self.sizer.Add(boxsizer, pos=(0,0), span=(4,1), flag=wx.EXPAND)\r
  51                 \r
  52                 self.connectButton = wx.Button(self.panel, -1, 'Connect')\r
  53                 #self.loadButton = wx.Button(self.panel, -1, 'Load GCode')\r
  54                 self.printButton = wx.Button(self.panel, -1, 'Print GCode')\r
  55                 self.cancelButton = wx.Button(self.panel, -1, 'Cancel print')\r
  56                 self.progress = wx.Gauge(self.panel, -1)\r
  57                 \r
  58                 h = self.connectButton.GetSize().GetHeight()\r
  59                 self.temperatureSelect = wx.SpinCtrl(self.panel, -1, '0', size=(21*3,21), style=wx.SP_ARROW_KEYS)\r
  60                 self.temperatureSelect.SetRange(0, 400)\r
  61                 \r
  62                 self.sizer.Add(self.connectButton, pos=(0,1))\r
  63                 #self.sizer.Add(self.loadButton, pos=(1,1))\r
  64                 self.sizer.Add(self.printButton, pos=(2,1))\r
  65                 self.sizer.Add(self.cancelButton, pos=(3,1))\r
  66                 self.sizer.Add(self.progress, pos=(4,0), span=(1,2), flag=wx.EXPAND)\r
  67 \r
  68                 self.sizer.Add(wx.StaticText(self.panel, -1, "Temp:"), pos=(0,3))\r
  69                 self.sizer.Add(self.temperatureSelect, pos=(0,4))\r
  70 \r
  71                 self.sizer.AddGrowableRow(3)\r
  72                 self.sizer.AddGrowableCol(0)\r
  73                 \r
  74                 self.Bind(wx.EVT_CLOSE, self.OnClose)\r
  75                 self.connectButton.Bind(wx.EVT_BUTTON, self.OnConnect)\r
  76                 #self.loadButton.Bind(wx.EVT_BUTTON, self.OnLoad)\r
  77                 self.printButton.Bind(wx.EVT_BUTTON, self.OnPrint)\r
  78                 self.cancelButton.Bind(wx.EVT_BUTTON, self.OnCancel)\r
  79                 \r
  80                 self.Bind(wx.EVT_SPINCTRL, self.OnTempChange, self.temperatureSelect)\r
  81                 \r
  82                 self.Layout()\r
  83                 self.Fit()\r
  84                 self.Centre()\r
  85                 \r
  86                 self.UpdateButtonStates()\r
  87                 self.UpdateProgress()\r
  88         \r
  89         def UpdateButtonStates(self):\r
  90                 self.connectButton.Enable(not self.machineConnected)\r
  91                 #self.loadButton.Enable(self.printIdx == None)\r
  92                 self.printButton.Enable(self.machineConnected and self.gcodeList != None and self.printIdx == None)\r
  93                 self.cancelButton.Enable(self.printIdx != None)\r
  94         \r
  95         def UpdateProgress(self):\r
  96                 status = ""\r
  97                 if self.gcode != None:\r
  98                         status += "Filament: %.2fm %.2fg\n" % (self.gcode.extrusionAmount / 1000, self.gcode.calculateWeight() * 1000)\r
  99                         cost_kg = float(profile.getPreference('filament_cost_kg'))\r
 100                         cost_meter = float(profile.getPreference('filament_cost_meter'))\r
 101                         if cost_kg > 0.0 and cost_meter > 0.0:\r
 102                                 status += "Filament cost: %.2f / %.2f\n" % (self.gcode.calculateWeight() * cost_kg, self.gcode.extrusionAmount / 1000 * cost_meter)\r
 103                         elif cost_kg > 0.0:\r
 104                                 status += "Filament cost: %.2f\n" % (self.gcode.calculateWeight() * cost_kg)\r
 105                         elif cost_meter > 0.0:\r
 106                                 status += "Filament cost: %.2f\n" % (self.gcode.extrusionAmount / 1000 * cost_meter)\r
 107                         status += "Print time: %02d:%02d\n" % (int(self.gcode.totalMoveTimeMinute / 60), int(self.gcode.totalMoveTimeMinute % 60))\r
 108                 if self.printIdx == None:\r
 109                         self.progress.SetValue(0)\r
 110                         if self.gcodeList != None:\r
 111                                 status += 'Line: -/%d\n' % (len(self.gcodeList))\r
 112                 else:\r
 113                         self.progress.SetValue(self.printIdx)\r
 114                         status += 'Line: %d/%d\n' % (self.printIdx, len(self.gcodeList))\r
 115                 if self.temp != None:\r
 116                         status += 'Temp: %d\n' % (self.temp)\r
 117                 self.statsText.SetLabel(status.strip())\r
 118                 self.Layout()\r
 119         \r
 120         def OnConnect(self, e):\r
 121                 if self.machineCom != None:\r
 122                         self.machineCom.close()\r
 123                         self.thread.join()\r
 124                 self.machineCom = machineCom.MachineCom()\r
 125                 self.thread = threading.Thread(target=self.PrinterMonitor)\r
 126                 self.thread.start()\r
 127                 self.UpdateButtonStates()\r
 128         \r
 129         def OnLoad(self, e):\r
 130                 pass\r
 131         \r
 132         def OnPrint(self, e):\r
 133                 if not self.machineConnected:\r
 134                         return\r
 135                 if self.gcodeList == None:\r
 136                         return\r
 137                 if self.printIdx != None:\r
 138                         return\r
 139                 self.printIdx = 1\r
 140                 self.sendLine(0)\r
 141                 self.sendCnt = self.bufferLineCount\r
 142                 self.UpdateButtonStates()\r
 143         \r
 144         def OnCancel(self, e):\r
 145                 self.printIdx = None\r
 146                 self.UpdateButtonStates()\r
 147         \r
 148         def OnClose(self, e):\r
 149                 global printWindowHandle\r
 150                 printWindowHandle = None\r
 151                 if self.machineCom != None:\r
 152                         self.machineCom.close()\r
 153                         self.thread.join()\r
 154                 self.Destroy()\r
 155 \r
 156         def OnTempChange(self, e):\r
 157                 self.sendCommand("M104 S%d" % (self.temperatureSelect.GetValue()))\r
 158 \r
 159         def LoadGCodeFile(self, filename):\r
 160                 if self.printIdx != None:\r
 161                         return\r
 162                 #Send an initial M110 to reset the line counter to zero.\r
 163                 gcodeList = ["M110"]\r
 164                 for line in open(filename, 'r'):\r
 165                         if ';' in line:\r
 166                                 line = line[0:line.find(';')]\r
 167                         line = line.strip()\r
 168                         if len(line) > 0:\r
 169                                 gcodeList.append(line)\r
 170                 gcode = gcodeInterpreter.gcode()\r
 171                 gcode.loadList(gcodeList)\r
 172                 print "Loaded: %s (%d)" % (filename, len(gcodeList))\r
 173                 self.progress.SetRange(len(gcodeList))\r
 174                 self.gcode = gcode\r
 175                 self.gcodeList = gcodeList\r
 176                 self.UpdateButtonStates()\r
 177                 self.UpdateProgress()\r
 178                 \r
 179         def sendCommand(self, cmd):\r
 180                 if self.machineConnected:\r
 181                         if self.printIdx == None:\r
 182                                 self.machineCom.sendCommand(cmd)\r
 183                         else:\r
 184                                 self.sendList.append(cmd)\r
 185 \r
 186         def sendLine(self, lineNr):\r
 187                 if lineNr >= len(self.gcodeList):\r
 188                         return False\r
 189                 checksum = reduce(lambda x,y:x^y, map(ord, "N%d%s" % (lineNr, self.gcodeList[lineNr])))\r
 190                 self.machineCom.sendCommand("N%d%s*%d" % (lineNr, self.gcodeList[lineNr], checksum))\r
 191                 return True\r
 192 \r
 193         def PrinterMonitor(self):\r
 194                 while True:\r
 195                         line = self.machineCom.readline()\r
 196                         if line == None:\r
 197                                 self.machineConnected = False\r
 198                                 wx.CallAfter(self.UpdateButtonStates)\r
 199                                 return\r
 200                         if self.machineConnected:\r
 201                                 while self.sendCnt > 0:\r
 202                                         self.sendLine(self.printIdx)\r
 203                                         self.printIdx += 1\r
 204                                         self.sendCnt -= 1\r
 205                         elif line.startswith("start"):\r
 206                                 self.machineConnected = True\r
 207                                 wx.CallAfter(self.UpdateButtonStates)\r
 208                         if 'T:' in line:\r
 209                                 self.temp = float(re.search("[0-9\.]*", line.split('T:')[1]).group(0))\r
 210                                 wx.CallAfter(self.UpdateProgress)\r
 211                         if self.printIdx == None:\r
 212                                 if line == '':  #When we have a communication "timeout" and we're not sending gcode, then read the temperature.\r
 213                                         self.machineCom.sendCommand("M105")\r
 214                         else:\r
 215                                 if line.startswith("ok"):\r
 216                                         if len(self.sendList) > 0:\r
 217                                                 self.machineCom.sendCommand(self.sendList.pop(0))\r
 218                                         else:\r
 219                                                 if self.sendLine(self.printIdx):\r
 220                                                         self.printIdx += 1\r
 221                                                 else:\r
 222                                                         self.printIdx = None\r
 223                                                         wx.CallAfter(self.UpdateButtonStates)\r
 224                                                 wx.CallAfter(self.UpdateProgress)\r
 225                                 elif "resend" in line.lower() or "rs" in line:\r
 226                                         try:\r
 227                                                 lineNr=int(line.replace("N:"," ").replace("N"," ").replace(":"," ").split()[-1])\r
 228                                         except:\r
 229                                                 if "rs" in line:\r
 230                                                         lineNr=int(line.split()[1])\r
 231                                         self.printIdx = lineNr\r
 232                                         #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
 233 \r