imcpython/IMCPython.py at master · claudiacor/imcpython · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
#!/usr/bin/env/python
from jinja2 import Environment, FileSystemLoader
import os
from xml.etree import ElementTree


class IMCPython:
    def __init__(self):
        with open('IMC.xml', 'rt') as f:
            self.tree = ElementTree.parse(f)
        self.abbrev_lista = []
        self.mabbrev_lista = []
        self.mid_lista = []
        self.type_lista = []
        self.abb_type_form = []
        self.THIS_DIR = os.path.dirname(os.path.abspath(__file__))
        self.j2_env = Environment(loader=FileSystemLoader(self.THIS_DIR), trim_blocks = True)
        self.out = ""
        self.newfact = ""
        self.serial = ""
        self.form_list = []


    def parse_message(self):
        for node in self.tree.iter('message'):
            m_id = node.attrib.get('id')
            m_name = node.attrib.get('name')
            m_abbrev = node.attrib.get('abbrev')
            m_source = node.attrib.get('source')
            self.parse_field(node)
            self.get_serial()
            if m_id is not None:
                self.mid_lista.append(m_id)

            if m_abbrev is not None:
                self.mabbrev_lista.append(m_abbrev)

            self.mabbrev_mid = zip(self.mabbrev_lista,self.mid_lista)
            aux = self.j2_env.get_template('Template.py').render(abb_type_form_list = self.abb_type_form, mabbrev_mid_list = self.mabbrev_mid)
            fact = self.j2_env.get_template('Fact.py').render(mabbrev_mid_list = self.mabbrev_mid)
            #print self.mabbrev_mid
            self.out += aux
            self.out += "\n"
            self.newfact += fact
            self.newfact += "\n"
            #print "tuplo: " + str(self.mabbrev_mid)
            self.mabbrev_lista = []
            self.serial = ""
            self.mid_lista = []
            self.form_list = []
            self.type_lista = []
            self.abbrev_lista = []

    def parse_field(self, message_node):
       # print "#Parse:"
        for field_node in message_node.iter('field'):
            name = field_node.attrib.get('name')
            self.abbrev = field_node.attrib.get('abbrev')
            Type = field_node.attrib.get('type')
            unit = field_node.attrib.get('unit')

            if self.abbrev is not None and (self.abbrev != "description"):
                self.abbrev_lista.append(self.abbrev)

            if Type is not None:
            	#print "\ttype: " + type
               # print "\tabbrev: " + self.abbrev
            	form = self.format(Type)
                #print "form: " + form
                #print "form_list: " + str(self.form_list)
                self.form_list.append(form)
                self.type_lista.append(Type)

       	self.abb_type_form = zip(self.abbrev_lista,self.type_lista,self.form_list)
        #print self.abb_type_form

    def get_serial(self):
    	self.serial = self.serial.join(self.form_list)
        #print self.serial
        #print "\n"

    def format(self, Type):
        if(Type == 'int8_t'):
            return '\'b\''
        elif(Type == 'uint8_t'):
            return '\'B\''
        elif(Type == 'int16_t'):
            return '\'h\''
        elif(Type == 'uint16_t'):
            return '\'H\''
        elif(Type == 'int32_t'):
            return '\'i\''
        elif(Type == 'uint32_t'):
            return '\'I\''
        elif(Type == 'int64_t'):
            return '\'q\''
        elif(Type == 'fp32_t'):
            return '\'f\''
        elif(Type == 'fp64_t'):
            return '\'d\''
        elif(Type == 'plaintext'):
            #o len tem que ser do q esta dentro da variavel
            return '\'s\''
	elif(Type == 'rawdata'):
	    return '\'s\''
        else:
            return "\'?\'"

nw = IMCPython()
nw.parse_message()

    #generate Definitions.py
with open('src_generated/Definitions.py','wb') as f:
    f.write('import struct\n')
with open("src_generated/Definitions.py", "ab") as f:
    f.write("from copy import deepcopy\n")
with open("src_generated/Definitions.py", "ab") as f:
    f.write("from Message import *\n")
with open("src_generated/Definitions.py", "ab") as f:
    f.write("import Factory\n")
with open("src_generated/Definitions.py", "ab") as f:
    f.write(nw.out.encode('latin-1'))

    #generate Factory.py
with open("src_generated/Factory.py", 'wb') as n:
    n.write("import Definitions as IMC\n")
with open("src_generated/Factory.py", 'ab') as n:
    n.write('def produce(id):\n')
with open("src_generated/Factory.py", 'ab') as n:
    n.write(nw.newfact.encode('latin-1'))

    #copy message.py to src_generated
with open('Message.py','rb') as ms:
    data = ms.read()
    with open('src_generated/Message.py','wb') as m:
        m.write(data)