import os
import re

class EmcToolTable(object):
    ''' intended as bug-compatible Python replacement for the
    tooltable io used in iocontrol
    NB: old file formats not supported.
    '''

    ttype = { 'T' : int, 'P': int, 'Q':int,
              'X' : float, 'Y' : float, 'Z' : float,
              'A' : float, 'B' : float, 'C' : float,
              'U' : float, 'V' : float, 'W' : float,
              'I' : float, 'J' : float, 'D' : float }

    def __init__(self,filename,random_toolchanger):
         self.filename = filename
         self.random_toolchanger = random_toolchanger

    def load_table(self, tooltable,comments,fms):
        self.fakepocket = 0
        fp = open(self.filename)
        lno = 0
        for line in fp.readlines():
            lno += 1
            if not line.startswith(';') and line.strip():
                entry = self.parseline(lno,line.strip())
                if entry:
                    self.assign(tooltable,entry,comments,fms)
        fp.close()

    def save_table(self, tooltable, comments,fms):
        os.rename(self.filename,self.filename + '.bak')
        fp = open(self.filename, 'w')
        start = 0 if self.random_toolchanger else 1
        for p in range(start,len(tooltable)):
            t = tooltable[p]
            if t.toolno != -1:
                print >> fp, "T%d P%d" % (t.toolno, p if self.random_toolchanger else fms[p]),
                if t.diameter:  print >> fp, "D%f" % (t.diameter),
                if t.offset.x: print >> fp, "X%+f" % (t.offset.x),
                if t.offset.y: print >> fp, "Y%+f" % (t.offset.y),
                if t.offset.z: print >> fp, "Z%+f" % (t.offset.z),
                if t.offset.a: print >> fp, "A%+f" % (t.offset.a),
                if t.offset.b: print >> fp, "B%+f" % (t.offset.b),
                if t.offset.c: print >> fp, "C%+f" % (t.offset.c),
                if t.offset.u: print >> fp, "U%+f" % (t.offset.u),
                if t.offset.v: print >> fp, "V%+f" % (t.offset.v),
                if t.offset.w: print >> fp, "W%+f" % (t.offset.w),
                if t.offset.w: print >> fp, "W%+f" % (t.offset.w),
                if t.frontangle: print >> fp, "I%+f" % (t.frontangle),
                if t.backangle: print >> fp, "J%+f" % (t.backangle),
                if t.orientation: print >> fp, "Q%+d" % (t.orientation),
                if comments.has_key(p) and comments[p]:
                    print >> fp, ";%s" % (comments[p])
                else:
                    print >> fp
        fp.close()

    def assign(self,tooltable,entry,comments,fms):
        pocket = entry['P']
        if not self.random_toolchanger:
            self.fakepocket += 1
            if self.fakepocket >= len(tooltable):
                print "too many tools. skipping tool %d" % (toolno)
                return
            if not fms is None:
                fms[self.fakepocket] = pocket
            pocket = self.fakepocket
        if pocket < 0 or pocket > len(tooltable):
            print "max pocket number is %d. skipping tool %d" % (len(tooltable) - 1, toolno)
            return

        tooltable[pocket].zero()
        for (key,value) in entry.items():
            if key == 'T' : tooltable[pocket].toolno = value
            if key == 'Q' : tooltable[pocket].orientation = value
            if key == 'D' : tooltable[pocket].diameter = value
            if key == 'I' : tooltable[pocket].frontangle = value
            if key == 'J' : tooltable[pocket].backangle = value
            if key == 'X' : tooltable[pocket].offset.x = value
            if key == 'Y' : tooltable[pocket].offset.y = value
            if key == 'Z' : tooltable[pocket].offset.z = value
            if key == 'A' : tooltable[pocket].offset.a = value
            if key == 'B' : tooltable[pocket].offset.b = value
            if key == 'C' : tooltable[pocket].offset.c = value
            if key == 'U' : tooltable[pocket].offset.u = value
            if key == 'V' : tooltable[pocket].offset.v = value
            if key == 'W' : tooltable[pocket].offset.w = value
            if key == 'comment' : comments[pocket] = value # aaargh

    def parseline(self,lineno,line):
        """
        read a tooltable line
        if an entry was parsed successfully, return a  Tool() instance
        """
        line.rstrip("\n")
        if re.match('\A\s*T\d+',line):
            semi = line.find(";")
            if semi != -1:
                comment = line[semi+1:]
            else:
                comment = None
            entry = line.split(';')[0]
            result = dict()
            for field in entry.split():
                (name,value)  = re.search('([a-zA-Z])([+-]?\d*\.?\d*)',field).groups()
                if name:
                    key = name.upper()
                    result[key] = EmcToolTable.ttype[key](value)
                else:
                    print "%s:%d  bad line: '%s' " % (self.filename, lineno, entry)
            result['comment'] = comment
            return result
        print "%s:%d: unrecognized tool table entry   '%s'" % (self.filename,lineno,line)


    def restore_state(self,e):
        pass

    def save_state(self,e):
        pass