LibToJson.py

from liberty.parser import parse_liberty
from liberty.types import *
import json
import ast


def encode_escaped_string(s):
    if isinstance(s, EscapedString):
        return str(s)
    elif isinstance(s, WithUnit):
        return str(s)
    elif isinstance(s, ArithExpression):
        return str(s)
    elif isinstance(s, NameBitSelection):
        return str(s)
    else:
        type_name = s.__class__.__name__
        raise TypeError(f"Object of type '{type_name}' is not JSON serializable")


def escaped_string_convert(obj):
    if isinstance(obj, Group):
        new_obj = Group(escaped_string_convert(obj.group_name))
        for group in obj.groups:
            group = escaped_string_convert(group)
            new_obj.groups.append(group)
        for attr in obj.attributes:
            attr = escaped_string_convert(attr)
            new_obj.attributes.append(attr)
        for arg in obj.args:
            arg = escaped_string_convert(arg)
            new_obj.args.append(arg)
        for define in obj.defines:
            define = escaped_string_convert(define)
            new_obj.defines.append(define)
        obj = new_obj
    elif isinstance(obj, Attribute):
        obj.value = escaped_string_convert(obj.value)
        obj.name = escaped_string_convert(obj.name)
    elif isinstance(obj, Define):
        obj.attribute_name = escaped_string_convert(obj.attribute_name)
        obj.attribute_type = escaped_string_convert(obj.attribute_type)
        obj.group_name = escaped_string_convert(obj.group_name)
    elif isinstance(obj, EscapedString):
        obj = str(obj)
    elif isinstance(obj, list):
        if isinstance(obj[0], EscapedString):
            new_obj = list()
            for el in obj:
                try:
                    el1 = ast.literal_eval(str(el))
                    if isinstance(el1, tuple):
                        el = ast.literal_eval('[' + str(el) + ']')
                    else:
                        el = el1
                except:
                    el = str(el)
                new_obj.append(el)
            obj = new_obj
            if len(obj) == 1:
                obj = obj[0]
    return obj


def group_to_dict(group):
    out_dict = dict()
    attr_cnt = dict()
    for attr in group.attributes:
        if attr_cnt.get(attr.name) is None:
            attr_cnt[attr.name] = 1
        else:
            attr_cnt[attr.name] += 1
    for attr in group.attributes:
        if attr_cnt[attr.name] == 1:
            out_dict[attr.name] = attr.value
        else:
            if out_dict.get(attr.name) is None:
                out_dict[attr.name] = list()
            out_dict[attr.name].append(attr.value)
    if len(group.defines) != 0:
        out_dict['define'] = list()
        for define in group.defines:
            out_dict['define'].append(
                dict(attribute_name=define.attribute_name, attribute_type=define.attribute_type,
                     group_name=define.group_name))
    for gr in group.groups:
        if gr.args[0] is None:
            if out_dict.get(gr.group_name) is None:
                out_dict[gr.group_name] = list()
            out_dict[gr.group_name].append(group_to_dict(gr))
        else:
            if out_dict.get(gr.group_name) is None:
                out_dict[gr.group_name] = dict()
            if out_dict[gr.group_name].get(gr.args[0]) is None:
                out_dict[gr.group_name][gr.args[0]] = list()
            out_dict[gr.group_name][gr.args[0]].append(group_to_dict(gr))
    return out_dict


def lib_to_json(liberty_file, json_file):
    library = parse_liberty(open(liberty_file).read())
    library = escaped_string_convert(library)
    JSON_dict = dict()
    JSON_dict[library.group_name] = dict()
    JSON_dict[library.group_name][library.args[0]] = group_to_dict(library)
    json.dump(JSON_dict, open(json_file, 'w'), default=encode_escaped_string)