Cura/util/machineCom.py

   1 from __future__ import absolute_import
   2 import __init__
   3
   4 import os, glob, 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 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, 115200, 57600, 38400, 19200, 9600]
  34
  35 class VirtualPrinter():
  36         def __init__(self):
  37                 self.readList = ['start\n']
  38                 self.temp = 0.0
  39                 self.targetTemp = 0.0
  40                 self.bedTemp = 1.0
  41                 self.bedTargetTemp = 1.0
  42
  43         def write(self, data):
  44                 if self.readList == None:
  45                         return
  46                 print "Send: %s" % (data.rstrip())
  47                 if 'M104' in data or 'M109' in data:
  48                         try:
  49                                 self.targetTemp = float(data[data.find('S')+1:])
  50                         except:
  51                                 pass
  52                 if 'M140' in data or 'M190' in data:
  53                         try:
  54                                 self.bedTargetTemp = float(data[data.find('S')+1:])
  55                         except:
  56                                 pass
  57                 if 'M105' in data:
  58                         self.readList.append("ok T:%f /%f B:%f /%f @:64\n" % (self.temp, self.targetTemp, self.bedTemp, self.bedTargetTemp))
  59                 else:
  60                         self.readList.append("ok\n")
  61
  62         def readline(self):
  63                 if self.readList == None:
  64                         return ''
  65                 n = 0
  66                 self.temp = (self.temp + self.targetTemp) / 2
  67                 self.bedTemp = (self.bedTemp + self.bedTargetTemp) / 2
  68                 while len(self.readList) < 1:
  69                         time.sleep(0.1)
  70                         n += 1
  71                         if n == 20:
  72                                 return ''
  73                         if self.readList == None:
  74                                 return ''
  75                 time.sleep(0.01)
  76                 print "Recv: %s" % (self.readList[0].rstrip())
  77                 return self.readList.pop(0)
  78
  79         def close(self):
  80                 self.readList = None
  81
  82 class MachineCom():
  83         def __init__(self, port = None, baudrate = None):
  84                 if port == None:
  85                         port = profile.getPreference('serial_port')
  86                 if baudrate == None:
  87                         if profile.getPreference('serial_baud') == 'AUTO':
  88                                 baudrate = 0
  89                         else:
  90                                 baudrate = int(profile.getPreference('serial_baud'))
  91                 self.serial = None
  92                 if port == 'AUTO':
  93                         programmer = stk500v2.Stk500v2()
  94                         for port in serialList():
  95                                 try:
  96                                         print "Connecting to: %s" % (port)
  97                                         programmer.connect(port)
  98                                         programmer.close()
  99                                         time.sleep(1)
 100                                         self.serial = self._openPortWithBaudrate(port, baudrate)
 101                                         break
 102                                 except ispBase.IspError as (e):
 103                                         print "Error while connecting to %s" % (port)
 104                                         print e
 105                                         pass
 106                                 except:
 107                                         print "Unexpected error while connecting to serial port:" + port, sys.exc_info()[0]
 108                         programmer.close()
 109                 elif port == 'VIRTUAL':
 110                         self.serial = VirtualPrinter()
 111                 else:
 112                         try:
 113                                 self.serial = self._openPortWithBaudrate(port, baudrate)
 114                         except:
 115                                 print "Unexpected error while connecting to serial port:" + port, sys.exc_info()[0]
 116                 print self.serial
 117
 118         def _openPortWithBaudrate(self, port, baudrate):
 119                 if baudrate != 0:
 120                         return Serial(port, baudrate, timeout=2)
 121                 for baudrate in baudrateList():
 122                         try:
 123                                 ser = Serial(port, baudrate, timeout=2)
 124                         except:
 125                                 print "Unexpected error while connecting to serial port:" + port, sys.exc_info()[0]
 126                                 continue
 127                         starttime = time.time()
 128                         for line in ser:
 129                                 if line.startswith('start'):
 130                                         ser.close()
 131                                         return Serial(port, baudrate, timeout=2)
 132                                 if starttime - time.time() > 10:
 133                                         break
 134                         ser.close()
 135                 return None
 136
 137         def readline(self):
 138                 if self.serial == None:
 139                         return None
 140                 ret = self.serial.readline()
 141                 #if ret != '':
 142                 #       print "Recv: " + ret.rstrip()
 143                 return ret
 144
 145         def close(self):
 146                 if self.serial != None:
 147                         self.serial.close()
 148                 self.serial = None
 149
 150         def __del__(self):
 151                 self.close()
 152
 153         def isOpen(self):
 154                 return self.serial != None
 155
 156         def sendCommand(self, cmd):
 157                 if self.serial == None:
 158                         return
 159                 #print 'Send: ' + cmd
 160                 self.serial.write(cmd + '\n')