Cura/gui/firmwareInstall.py

   1 from __future__ import absolute_import
   2 import __init__
   3
   4 import os, glob, wx, threading, sys, time
   5
   6 from serial import Serial
   7
   8 from avr_isp import stk500v2
   9 from avr_isp import ispBase
  10 from avr_isp import intelHex
  11
  12 from util import machineCom
  13 from util import profile
  14
  15 def getDefaultFirmware():
  16         if profile.getPreference('machine_type') == 'ultimaker':
  17                 if sys.platform.startswith('linux'):
  18                         return os.path.join(os.path.dirname(os.path.abspath(__file__)), "../firmware/ultimaker_115200.hex")
  19                 else:
  20                         return os.path.join(os.path.dirname(os.path.abspath(__file__)), "../firmware/ultimaker_250000.hex")
  21         return None
  22
  23 class InstallFirmware(wx.Dialog):
  24         def __init__(self, filename = None, port = None):
  25                 super(InstallFirmware, self).__init__(parent=None, title="Firmware install", size=(250, 100))
  26                 if port == None:
  27                         port = profile.getPreference('serial_port')
  28                 if filename == None:
  29                         filename = getDefaultFirmware()
  30                 if filename == None:
  31                         wx.MessageBox('Cura does not ship with a default firmware for your machine.', 'Firmware update', wx.OK | wx.ICON_ERROR)
  32                         self.Destroy()
  33                         return
  34
  35                 sizer = wx.BoxSizer(wx.VERTICAL)
  36
  37                 self.progressLabel = wx.StaticText(self, -1, 'Reading firmware...')
  38                 sizer.Add(self.progressLabel, 0, flag=wx.ALIGN_CENTER)
  39                 self.progressGauge = wx.Gauge(self, -1)
  40                 sizer.Add(self.progressGauge, 0, flag=wx.EXPAND)
  41                 self.okButton = wx.Button(self, -1, 'Ok')
  42                 self.okButton.Disable()
  43                 self.okButton.Bind(wx.EVT_BUTTON, self.OnOk)
  44                 sizer.Add(self.okButton, 0, flag=wx.ALIGN_CENTER)
  45                 self.SetSizer(sizer)
  46
  47                 self.filename = filename
  48                 self.port = port
  49
  50                 threading.Thread(target=self.OnRun).start()
  51
  52                 self.ShowModal()
  53                 self.Destroy()
  54                 return
  55
  56         def OnRun(self):
  57                 hexFile = intelHex.readHex(self.filename)
  58                 wx.CallAfter(self.updateLabel, "Connecting to machine...")
  59                 programmer = stk500v2.Stk500v2()
  60                 programmer.progressCallback = self.OnProgress
  61                 if self.port == 'AUTO':
  62                         for self.port in machineCom.serialList():
  63                                 try:
  64                                         programmer.connect(self.port)
  65                                         break
  66                                 except ispBase.IspError:
  67                                         pass
  68                 else:
  69                         try:
  70                                 programmer.connect(self.port)
  71                         except ispBase.IspError:
  72                                 pass
  73
  74                 if programmer.isConnected():
  75                         wx.CallAfter(self.updateLabel, "Uploading firmware...")
  76                         try:
  77                                 programmer.programChip(hexFile)
  78                                 wx.CallAfter(self.updateLabel, "Done!")
  79                         except ispBase.IspError as e:
  80                                 wx.CallAfter(self.updateLabel, "Failed to write firmware.\n" + str(e))
  81
  82                         programmer.close()
  83                         wx.CallAfter(self.okButton.Enable)
  84                         return
  85                 wx.MessageBox('Failed to find machine for firmware upgrade\nIs your machine connected to the PC?', 'Firmware update', wx.OK | wx.ICON_ERROR)
  86                 wx.CallAfter(self.Close)
  87
  88         def updateLabel(self, text):
  89                 self.progressLabel.SetLabel(text)
  90                 self.Layout()
  91
  92         def OnProgress(self, value, max):
  93                 wx.CallAfter(self.progressGauge.SetRange, max)
  94                 wx.CallAfter(self.progressGauge.SetValue, value)
  95
  96         def OnOk(self, e):
  97                 self.Close()
  98
  99         def OnClose(self, e):
 100                 self.Destroy()
 101