# = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = =
# Script Name   : arch_file_updater.py
# Description   : This script designed to update architecture files 
#                 from a lower version to higher version
# Author        : Xifan Tang
# Email         : xifan@osfpga.org
# = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = =
import os
from os.path import dirname, abspath
import argparse
import logging
import shutil
from xml.dom import minidom
from datetime import timedelta
import time
import datetime
import re

#####################################################################
# Error codes
#####################################################################
error_codes = {
  "SUCCESS": 0,
  "ERROR": 1, 
  "OPTION_ERROR": 2, 
  "FILE_ERROR": 3 
}

#####################################################################
# Initialize logger 
#####################################################################
logging.basicConfig(format='%(levelname)s: %(message)s', level=logging.INFO);

#####################################################################
# Upgrade an architecture XML file from version 1.1 syntax to version 1.2
# Change rules:
# - Create a new child node of <sub_tile> under each node of <tile>
# - A <sub_tile> node herits the attribute 'name' of its parent <tile>
# - A <sub_tile> node herits the attribute 'capacity' of its parent <tile>
# - Any child nodes under a <tile> are moved to <sub_tile>
# - The attribute 'capacity' of parent <tile> is removed
#####################################################################
def convert_arch_xml_from_v1p1_to_v1p2(input_fname, output_fname):
  # Constants
  TILE_ROOT_TAG = "tiles"
  TILE_NODE_TAG = "tile"
  SUB_TILE_NODE_TAG = "sub_tile"
  NAME_TAG = "name" 
  CAPACITY_TAG = "capacity" 

  # Log runtime and status
  status = error_codes["SUCCESS"]
  start_time = time.time()

  log_str = "Converting \'" + input_fname + "\'" + " to " + "\'" + output_fname + "\'..."
  logging.info(log_str)
  # Parse the input file
  doc = minidom.parse(input_fname)

  # Iterate over <tile> nodes
  num_tile_roots = len(doc.getElementsByTagName(TILE_ROOT_TAG))
  if (num_tile_roots != 1):
    logging.info("Found " + str(num_tile_roots) + " <" + TILE_ROOT_TAG + ">")
    logging.error("Fail to find a require node (one and only one) <" + TILE_ROOT_TAG + "> under the root node!")
    return error_codes["ERROR"]
  tile_root = doc.getElementsByTagName(TILE_ROOT_TAG)[0]
  for tile_node in tile_root.getElementsByTagName(TILE_NODE_TAG):
    # Create a new child node <sub_tile>
    sub_tile_node = doc.createElement(SUB_TILE_NODE_TAG)
    # Add attributes to the new child node
    sub_tile_node.setAttribute(NAME_TAG, tile_node.getAttribute(NAME_TAG))
    if tile_node.hasAttribute(CAPACITY_TAG):
      sub_tile_node.setAttribute(CAPACITY_TAG, tile_node.getAttribute(CAPACITY_TAG))
      # Delete the out-of-date attributes
      tile_node.removeAttribute(CAPACITY_TAG)
    # Move other subelements to the new child node
    for child in tile_node.childNodes:
      # Add the node to the child node
      child_clone = child.cloneNode(deep=True)
      sub_tile_node.appendChild(child_clone)
    # Remove no longer required child nodes
    while (tile_node.hasChildNodes()):
      tile_node.removeChild(tile_node.firstChild)
    # Append the sub tile child to the tile node
    tile_node.appendChild(sub_tile_node)

  # Output the modified content
  with open(output_fname, "w") as output_xml_f:
    doc.writexml(output_xml_f, indent='', addindent=" ", newl='')
  doc.unlink()

  # Finish up
  end_time = time.time()
  time_diff = timedelta(seconds=(end_time - start_time))
  log_end_str1 = "[Done]"
  log_end_str2 = " took " + str(time_diff)
  logging.info(log_end_str1 + "." * (len(log_str) - len(log_end_str1) - len(log_end_str2)) + log_end_str2)
  return status

#####################################################################
# Main function
#####################################################################
if __name__ == '__main__':
  # Execute when the module is not initialized from an import statement

  # Parse the options and apply sanity checks
  parser = argparse.ArgumentParser(description='Convert an architecture file from a lower version to a higher version')
  parser.add_argument('--input_file',
                      required=True,
                      help='Path to input architecture file')
  parser.add_argument('--output_file',
                      default="converted_arch.xml",
                      help='Path to output converted architecture file')
  args = parser.parse_args()

  # Run conversion: from v1.1 syntax to v1.2 syntax
  exit(convert_arch_xml_from_v1p1_to_v1p2(args.input_file, args.output_file))