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