#!/usr/bin/env python3
# Copyright 2020 Efabless Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import argparse
import re
import xml.etree.ElementTree as ET
import xml.dom.minidom as minidom

parser = argparse.ArgumentParser(
    description='Converts a TritonRoute DRC Report to a KLayout database')

parser.add_argument('--input', '-i', required=True,
                    help='')

parser.add_argument('--design-name', '-name', required=True,
                    help='')

parser.add_argument('--output', '-o', required=True,
                    help='')

args = parser.parse_args()

input_file_name = args.input
design_name = args.design_name
output_file_name = args.output

RE_VIOLATION = re.compile(r'violation type: (?P<type>\S+)\s+'
                          r'srcs: (?P<src1>\S+)( (?P<src2>\S+))?\s+'
                          r'bbox = \( (?P<llx>\S+), (?P<lly>\S+) \)'
                          r' - '
                          r'\( (?P<urx>\S+), (?P<ury>\S+) \) '
                          r'on Layer (?P<layer>\S+)', re.M)

def prettify(elem):
    """
    Return a pretty-printed XML string for the Element.
    """
    rough_string = ET.tostring(elem, 'utf-8')
    reparsed = minidom.parseString(rough_string)
    return reparsed.toprettyxml(indent="    ",
                                newl="\n")
                                # encoding='utf-8')

with open(input_file_name, 'r') as f:
    content = f.read()

cnt = 0
vio_dict  = {}
for match in RE_VIOLATION.finditer(content):
    cnt += 1
    type_ = match.group('type')
    src1 = match.group('src1')
    src2 = match.group('src2')
    llx = match.group('llx')
    lly = match.group('lly')
    urx = match.group('urx')
    ury = match.group('ury')
    layer = match.group('layer')

    item = ET.Element('item')
    ET.SubElement(item, 'category').text = type_
    ET.SubElement(item, 'cell').text = design_name
    ET.SubElement(item, 'visited').text = 'false'
    ET.SubElement(item, 'multiplicity').text = '1'
    values = ET.SubElement(item, 'values')
    box = ET.SubElement(values, 'value')
    box.text = f"box: ({llx},{lly};{urx},{ury})"
    layer_msg = ET.SubElement(values, 'value')
    layer_msg.text = f"text: 'On layer {layer}'"
    srcs = ET.SubElement(values, 'value')
    if src2:
        srcs.text = f"text: 'Between {src1} {src2}'"
    else:
        srcs.text = f"text: 'Between {src1}'"

    # create XML object
    if type_ not in vio_dict:
        vio_dict[type_] = []

    vio_dict[type_].append(item)

print("Found", cnt, "violations")

report_database = ET.Element('report-database')
categories = ET.SubElement(report_database, 'categories')
for type_ in vio_dict.keys():
    category = ET.SubElement(categories, 'category')
    ET.SubElement(category, 'name').text = type_

cells = ET.SubElement(report_database, 'cells')
cell = ET.SubElement(cells, 'cell')
ET.SubElement(cell, 'name').text = design_name

items = ET.Element('items')
for _, vios in vio_dict.items():
    for item in vios:
        items.append(item)

report_database.append(items)

report_database_str = prettify(report_database)

with open(output_file_name, 'w') as f:
    f.write(report_database_str)