1

import ctypes

     2

from LPCProto import *

     3

     4

LPC_STATUS={0xaa : "Started",

     5

            0x55 : "Stopped"}

     6

     7

class LPCAppProto(LPCProto):

     8

    def HandleTransaction(self, transaction):

     9

        self.TransactionLock.acquire()

    10

        try:

    11

            transaction.SetPseudoFile(self.serialPort)

    12

            # send command, wait ack (timeout)

    13

            transaction.SendCommand()

    14

            current_plc_status = transaction.GetCommandAck()

    15

            if current_plc_status is not None:

    16

                res = transaction.ExchangeData()

    17

            else:

    18

                raise LPCProtoError("controller did not answer as expected")

    19

        except Exception, e:

    20

            raise LPCProtoError("application mode transaction error : "+str(e))

    21

        finally:

    22

            self.TransactionLock.release()

    23

        return LPC_STATUS.get(current_plc_status,"Broken"), res

    24

    25

class LPCAppTransaction:

    26

    def __init__(self, command):

    27

        self.Command = command

    28

        self.pseudofile = None

    29

    30

    def SetPseudoFile(self, pseudofile):

    31

        self.pseudofile = pseudofile

    32

    33

    def SendCommand(self):

    34

        # send command thread

    35

        self.pseudofile.write(chr(self.Command))

    36

    37

    def GetCommandAck(self):

    38

        res = self.pseudofile.read(2)

    39

        if len(res) == 2:

    40

            comm_status, current_plc_status = map(ord, res)

    41

        else:

    42

            raise LPCProtoError("LPC transaction error - controller did not ack order")

    43

        # LPC returns command itself as an ack for command

    44

        if(comm_status == self.Command):

    45

            return current_plc_status

    46

        return None 

    47

    48

    def SendData(self, Data):

    49

        return self.pseudofile.write(Data)

    50

    51

    def GetData(self):

    52

        lengthstr = self.pseudofile.read(4)

    53

        # transform a byte string into length 

    54

        length = ctypes.cast(

    55

            ctypes.c_char_p(lengthstr), 

    56

            ctypes.POINTER(ctypes.c_uint32)).contents.value

    57

        return self.pseudofile.read(length)

    58

    59

    def ExchangeData(self): 

    60

        pass

    61

    62

class IDLETransaction(LPCAppTransaction):

    63

    def __init__(self):

    64

        LPCAppTransaction.__init__(self, 0x07)

    65

    ExchangeData = LPCAppTransaction.GetData

    66

    67

class STARTTransaction(LPCAppTransaction):

    68

    def __init__(self):

    69

        LPCAppTransaction.__init__(self, 0x01)

    70

    71

class STOPTransaction(LPCAppTransaction):

    72

    def __init__(self):

    73

        LPCAppTransaction.__init__(self, 0x02)

    74

    75

class RESETTransaction(LPCAppTransaction):

    76

    def __init__(self):

    77

        LPCAppTransaction.__init__(self, 0x03)

    78

    79

class SET_TRACE_VARIABLETransaction(LPCAppTransaction):

    80

    def __init__(self, data):

    81

        LPCAppTransaction.__init__(self, 0x04)

    82

        length = len(data)

    83

        # transform length into a byte string

    84

        # we presuppose endianess of LPC same as PC

    85

        lengthstr = ctypes.string_at(ctypes.pointer(ctypes.c_uint32(length)),4)

    86

        self.Data = lengthstr + data

    87

    88

    def ExchangeData(self):

    89

        self.SendData(self.Data)

    90

    91

class GET_TRACE_VARIABLETransaction(LPCAppTransaction):

    92

    def __init__(self):

    93

        LPCAppTransaction.__init__(self, 0x05)

    94

    ExchangeData = LPCAppTransaction.GetData

    95

    96

class GET_PLCIDTransaction(LPCAppTransaction):

    97

    def __init__(self):

    98

        LPCAppTransaction.__init__(self, 0x07)

    99

    ExchangeData = LPCAppTransaction.GetData

   100

   101

class GET_LOGCOUNTSTransaction(LPCAppTransaction):

   102

    def __init__(self):

   103

        LPCAppTransaction.__init__(self, 0x0B)

   104

    ExchangeData = LPCAppTransaction.GetData

   105

   106

class GET_LOGMSGTransaction(LPCAppTransaction):

   107

    def __init__(self,level,msgid):

   108

        LPCAppTransaction.__init__(self, 0x0C)

   109

        msgidstr = ctypes.string_at(ctypes.pointer(ctypes.c_int(msgid)),4)

   110

        self.Data = chr(level)+msgidstr

   111

   112

    def ExchangeData(self):

   113

        self.SendData(self.Data)

   114

        return self.GetData()

   115

   116

if __name__ == "__main__":

   117

    __builtins__.BMZ_DBG = True

   118

    TestConnection = LPCAppProto(6,115200,2)

   119

#    TestConnection.HandleTransaction(GET_PLCIDTransaction())

   120

    TestConnection.HandleTransaction(STARTTransaction())

   121

#    TestConnection.HandleTransaction(SET_TRACE_VARIABLETransaction(

   122

#           "\x03\x00\x00\x00"*200))

   123

#    TestConnection.HandleTransaction(STARTTransaction())

   124

    while True:

   125

        TestConnection.HandleTransaction(SET_TRACE_VARIABLETransaction(

   126

           "\x01\x00\x00\x00"+

   127

           "\x04"+

   128

           "\x01\x02\x02\x04"+

   129

           "\x01\x00\x00\x00"+

   130

           "\x08"+

   131

           "\x01\x02\x02\x04"+

   132

           "\x01\x02\x02\x04"+

   133

           "\x01\x00\x00\x00"+

   134

           "\x04"+

   135

           "\x01\x02\x02\x04"))

   136

    #status,res = TestConnection.HandleTransaction(GET_TRACE_VARIABLETransaction())

   137

    #print len(res)

   138

    #print "GOT : ", map(hex, map(ord, res))

   139

    #TestConnection.HandleTransaction(STOPTransaction())