Homec4science
Diffusion code_skeleton (master) Browse

R1692/merge_classes.py
5747c8377d30master

Last Change
Raw File

merge_classes.py
View Options

#!/usr/bin/env python
# -*- coding: utf-8 -*-
""" Module for merging class representations """
################################################################
import argparse
from class_dumper_classes import ClassDumperClasses
from class_reader import ClassReader
################################################################
def merge(files, output_filename):
"takes a list of files and produce a single one"
print files
classes = []
for class_file in files:
cls_reader = ClassReader()
classes.append(cls_reader.read(class_file))
print len(classes), " files to merge"
merged_classes = {}
for _class in classes:
for c in _class:
name = c.name
if c.name not in merged_classes:
merged_classes[c.name] = c
else:
pclass = merged_classes[c.name]
for encaps, methods in c.methods.iteritems():
for mname, method in methods.iteritems():
if mname in pclass.methods[encaps]:
if not pclass.methods[encaps][mname] == method:
print "Warming: ", ((c.name, encaps, mname))
else:
pclass.methods[encaps][mname] = method
for encaps, members in c.members.iteritems():
for mname, member in members.iteritems():
if mname in pclass.members[encaps]:
if not pclass.members[encaps][mname] == member:
print "Warming: ", ((c.name, encaps, mname))
else:
pclass.members[encaps][mname] = member
#for c in merged_classes:
# print merged_classes[c].name
dumper_class = ClassDumperClasses('output_filename')
fout = open(dumper_class.output_file, 'w')
for name, _class in merged_classes.iteritems():
dumper_class.dumpClass(_class, fout)
################################################################
def main():
parser = argparse.ArgumentParser(description='Classes descriptor merge')
parser.add_argument('class_files', nargs='+', help='The class files to merge')
parser.add_argument('--output_file', '-o',
help='The file where to put the classes description', required=True)
args = parser.parse_args()
args = vars(args)
class_files = args['class_files']
output_file = args['output_file']
merge(class_files, output_file)
if __name__ == '__main__':
main()
c4science · Help