ARXML_GEN.py

import re
from Tkinter import *
from textwrap import wrap
import tkFileDialog
import math
from DBC_Parser import DBCParse
from prettytable import PrettyTable
import xml.etree.ElementTree as ET

# Functions:
def AddBool(name,DataTypeElementsRoot):
	boolType = ET.SubElement(DataTypeElementsRoot,'BOOLEAN-TYPE')
	shortName = ET.SubElement(boolType,'SHORT-NAME')
	shortName.text = name
def AddStruct(message,DataTypeElementsRoot):
	recordType = ET.SubElement(DataTypeElementsRoot,'RECORD-TYPE')
	shortName = ET.SubElement(recordType,'SHORT-NAME')
	shortName.text = message.name + '_GEN_S'
	Elements = ET.SubElement(recordType,'ELEMENTS')
	for signal in message.signals:
		recordElement = ET.SubElement(Elements,'RECORD-ELEMENT')
		shortName = ET.SubElement(recordElement,'SHORT-NAME')
		shortName.text = signal
		typeRef = ET.SubElement(recordElement,'TYPE-TREF')
		typeRef.attrib = {'DEST':'BOOLEAN-TYPE'}
		typeRef.text = '/DataType/DT_'+signal

	

	
ET.register_namespace('', "http://autosar.org/3.2.2")
tree = ET.parse('COMH_Try_swc.arxml')
#----------------------------   Read DBC File --------------------------------------------------------
fileName = 'RIV_ADAS_Mule1_v8.dbc'
if fileName == None:
	sys.exit()

DBC_Name = fileName 

DBCParsObj = DBCParse(DBC_Name,'PARK')
DBCParsObj.printdbc()
#----------------------------   Read DBC File --------------------------------------------------------

root = tree.getroot()

topLevelPacks = root[0]

# for AR_Packs in topLevelPacks:
	# for AR_Pack in AR_Packs:
		# if 'SHORT-NAME' in AR_Pack.tag:
			# print AR_Pack.text

for AR_Packs in topLevelPacks:
	for AR_Pack in AR_Packs:
		if 'SHORT-NAME' in AR_Pack.tag:
			if AR_Pack.text == 'DataType':
				#print 'Datatype found' 	
				#print AR_Packs[1].tag
				DataType_Elements = AR_Packs[1]
				
# for Element in DataType_Elements:
	# print Element.tag

AddBool('DT_Raziur',DataType_Elements)

for message in DBCParsObj.Messages:
	if message == 'Misc_Report':
		AddStruct(DBCParsObj.Messages[message],DataType_Elements)
	
	

	
tree.write('new.xml')