mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-02-27 03:43:53 +08:00
71f547219c
f034dee pass dirname explicitly to generator helperfunctions and whitespace d2fb5f0 small generator cleanup ab04e55 Update generator.py d86d7f1 Update generator.py 031acc5 Update generator.py 19a4249 Update generator.py 9c49b88 Update generator.py git-subtree-dir: opendbc git-subtree-split: f034deec7dda00c4c33a614ad104ce51aeb604b9
50 lines
1.4 KiB
Python
Executable File
50 lines
1.4 KiB
Python
Executable File
#!/usr/bin/env python2
|
|
import os
|
|
import re
|
|
|
|
cur_path = os.path.dirname(os.path.realpath(__file__))
|
|
generator_path = os.path.join(cur_path, '../')
|
|
include_pattern = re.compile(r'CM_ "IMPORT (.*?)"')
|
|
|
|
|
|
def read_dbc(dir_name, filename):
|
|
with open(os.path.join(dir_name, filename)) as file_in:
|
|
return file_in.read()
|
|
|
|
|
|
def create_dbc(dir_name, filename):
|
|
dbc_file_in = read_dbc(dir_name, filename)
|
|
|
|
includes = include_pattern.findall(dbc_file_in)
|
|
|
|
output_filename = filename.replace('.dbc', '_generated.dbc')
|
|
output_file_location = os.path.join(generator_path, output_filename)
|
|
|
|
with open(output_file_location, 'w') as dbc_file_out:
|
|
dbc_file_out.write('CM_ "AUTOGENERATED FILE, DO NOT EDIT"\n')
|
|
|
|
for include_filename in reversed(includes):
|
|
include_file_header = '\n\nCM_ "Imported file %s starts here"\n' % include_filename
|
|
dbc_file_out.write(include_file_header)
|
|
|
|
include_file = read_dbc(dir_name, include_filename)
|
|
dbc_file_out.write(include_file)
|
|
|
|
dbc_file_out.write('\nCM_ "%s starts here"\n' % filename)
|
|
|
|
core_dbc = include_pattern.sub('', dbc_file_in)
|
|
dbc_file_out.write(core_dbc)
|
|
|
|
|
|
for dir_name, _, filenames in os.walk(cur_path):
|
|
if dir_name == cur_path:
|
|
continue
|
|
|
|
print dir_name
|
|
for filename in filenames:
|
|
if filename.startswith('_'):
|
|
continue
|
|
|
|
print filename
|
|
create_dbc(dir_name, filename)
|