Cura/util/machineCom.py

   1 from __future__ import absolute_import
   2 import __init__
   3
   4 import os, glob, sys, time, math
   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 profile
  13
  14 try:
  15         import _winreg
  16 except:
  17         pass
  18
  19 def serialList():
  20     baselist=[]
  21     if os.name=="nt":
  22         try:
  23             key=_winreg.OpenKey(_winreg.HKEY_LOCAL_MACHINE,"HARDWARE\\DEVICEMAP\\SERIALCOMM")
  24             i=0
  25             while(1):
  26                 baselist+=[_winreg.EnumValue(key,i)[1]]
  27                 i+=1
  28         except:
  29             pass
  30     return baselist+glob.glob('/dev/ttyUSB*') + glob.glob('/dev/ttyACM*') +glob.glob("/dev/tty.usb*")+glob.glob("/dev/cu.*")+glob.glob("/dev/rfcomm*")
  31
  32 def baudrateList():
  33         return [250000, 230400, 115200, 57600, 38400, 19200, 9600]
  34
  35 class VirtualPrinter():
  36         def __init__(self):
  37                 self.readList = ['start\n', 'Marlin: Virtual Marlin!\n', '\x80\n']
  38                 self.temp = 0.0
  39                 self.targetTemp = 0.0
  40                 self.lastTempAt = time.time()
  41                 self.bedTemp = 1.0
  42                 self.bedTargetTemp = 1.0
  43
  44         def write(self, data):
  45                 if self.readList == None:
  46                         return
  47                 print "Send: %s" % (data.rstrip())
  48                 if 'M104' in data or 'M109' in data:
  49                         try:
  50                                 self.targetTemp = float(data[data.find('S')+1:])
  51                         except:
  52                                 pass
  53                 if 'M140' in data or 'M190' in data:
  54                         try:
  55                                 self.bedTargetTemp = float(data[data.find('S')+1:])
  56                         except:
  57                                 pass
  58                 if 'M105' in data:
  59                         self.readList.append("ok T:%.2f /%.2f B:%.2f /%.2f @:64\n" % (self.temp, self.targetTemp, self.bedTemp, self.bedTargetTemp))
  60                 else:
  61                         self.readList.append("ok\n")
  62
  63         def readline(self):
  64                 if self.readList == None:
  65                         return ''
  66                 n = 0
  67                 timeDiff = self.lastTempAt - time.time()
  68                 self.lastTempAt = time.time()
  69                 if abs(self.temp - self.targetTemp) > 1:
  70                         self.temp += math.copysign(timeDiff, self.targetTemp - self.temp)
  71                 if abs(self.bedTemp - self.bedTargetTemp) > 1:
  72                         self.bedTemp += math.copysign(timeDiff, self.bedTargetTemp - self.bedTemp)
  73                 while len(self.readList) < 1:
  74                         time.sleep(0.1)
  75                         n += 1
  76                         if n == 20:
  77                                 return ''
  78                         if self.readList == None:
  79                                 return ''
  80                 time.sleep(0.01)
  81                 print "Recv: %s" % (self.readList[0].rstrip())
  82                 return self.readList.pop(0)
  83
  84         def close(self):
  85                 self.readList = None
  86
  87 class MachineCom():
  88         def __init__(self, port = None, baudrate = None):
  89                 if port == None:
  90                         port = profile.getPreference('serial_port')
  91                 if baudrate == None:
  92                         if profile.getPreference('serial_baud') == 'AUTO':
  93                                 baudrate = 0
  94                         else:
  95                                 baudrate = int(profile.getPreference('serial_baud'))
  96                 self.serial = None
  97                 if port == 'AUTO':
  98                         programmer = stk500v2.Stk500v2()
  99                         for port in serialList():
 100                                 try:
 101                                         print "Connecting to: %s" % (port)
 102                                         programmer.connect(port)
 103                                         programmer.close()
 104                                         time.sleep(1)
 105                                         self.serial = self._openPortWithBaudrate(port, baudrate)
 106                                         break
 107                                 except ispBase.IspError as (e):
 108                                         print "Error while connecting to %s" % (port)
 109                                         print e
 110                                         pass
 111                                 except:
 112                                         print "Unexpected error while connecting to serial port:" + port, sys.exc_info()[0]
 113                         programmer.close()
 114                 elif port == 'VIRTUAL':
 115                         self.serial = VirtualPrinter()
 116                 else:
 117                         try:
 118                                 self.serial = self._openPortWithBaudrate(port, baudrate)
 119                         except:
 120                                 print "Unexpected error while connecting to serial port:" + port, sys.exc_info()[0]
 121                 print self.serial
 122
 123         def _openPortWithBaudrate(self, port, baudrate):
 124                 if baudrate != 0:
 125                         return Serial(port, baudrate, timeout=2)
 126                 for baudrate in baudrateList():
 127                         try:
 128                                 ser = Serial(port, baudrate, timeout=2)
 129                         except:
 130                                 print "Unexpected error while connecting to serial port:" + port, sys.exc_info()[0]
 131                                 continue
 132                         ser.setDTR(1)
 133                         time.sleep(0.1)
 134                         ser.setDTR(0)
 135                         time.sleep(0.2)
 136                         starttime = time.time()
 137                         for line in ser:
 138                                 if 'start' in line:
 139                                         ser.close()
 140                                         ser = Serial(port, baudrate, timeout=2)
 141                                         ser.setDTR(1)
 142                                         time.sleep(0.1)
 143                                         ser.setDTR(0)
 144                                         time.sleep(0.2)
 145                                         return ser
 146                                 if starttime - time.time() > 10:
 147                                         break
 148                         ser.close()
 149                 return None
 150
 151         def readline(self):
 152                 if self.serial == None:
 153                         return None
 154                 try:
 155                         ret = self.serial.readline()
 156                 except:
 157                         print "Unexpected error while reading serial port:", sys.exc_info()[0]
 158                         ret = ''
 159                 #if ret != '':
 160                 #       print "Recv: " + ret.rstrip()
 161                 return ret
 162
 163         def close(self):
 164                 if self.serial != None:
 165                         self.serial.close()
 166                 self.serial = None
 167
 168         def __del__(self):
 169                 self.close()
 170
 171         def isOpen(self):
 172                 return self.serial != None
 173
 174         def sendCommand(self, cmd):
 175                 if self.serial == None:
 176                         return
 177                 #print 'Send: ' + cmd
 178                 self.serial.write(cmd + '\n')