"""
Usage:
  models_regression.py [--num_cores=<num>]

  -h, --help             Show help text.
  -v, --version          Show version.
  --num_cores=<num>      Number of cores to be used by simulator
"""

from cmath import inf
from re import L, T
from docopt import docopt
import pandas as pd 
import numpy as np
import os 
from jinja2 import Template
import concurrent.futures
import shutil
import warnings
warnings.simplefilter(action='ignore', category=FutureWarning)

def call_simulator(file_name):
    """Call simulation commands to perform simulation.
    Args:
        file_name (str): Netlist file name.
    """
    os.system(f"Xyce -hspice-ext all {file_name} -l {file_name}.log")

def ext_measured(device, table):

    # Generate CSVs with truth tables
    df = pd.DataFrame(data=table)
    df.set_index(df.columns[0])
    new_header = df.iloc[0]
    df = df[1:]
    df.columns = new_header
    df.to_csv(f"{device}/{device}_measured.csv", index = False)

def ext_simulated(device, processes, volts, temps):

    # Get all corners simulated
    for process in processes:
        for volt in volts:
            for temp in temps:
                with open(f"device_netlists/gf180mcu_fd_sc_mcu7t5v0__{device}_1.spice") as f:
                    tmpl = Template(f.read())
                netlist_path  = f"{device}/{device}_netlists/{device}_{process}_{temp}c_{volt}v.spice"
                with open(netlist_path, "w") as netlist:
                    netlist.write(tmpl.render(process = process, volt = volt , temp = temp ))

                # Running ngspice for each netlist 
                with concurrent.futures.ProcessPoolExecutor(max_workers=workers_count) as executor:
                    executor.submit(call_simulator, netlist_path)

                df_simulated = pd.read_csv(f"{device}/simulated/{device}_{process}_{temp}c_{volt}v.csv",header=0)
                results = []
                for i, col in enumerate(df_simulated.columns):
                    if df_simulated.iloc[-1].iloc[i] > 2.5:
                        results.append(1)
                    else:
                        results.append(0)
                df_measured = pd.read_csv(f"{device}/{device}_measured.csv",header=0)
                df = df_measured
                df_measured.drop(df_measured.columns[len(df_measured.columns)-1], axis=1, inplace=True)
                df_measured['output'] = results[1:]
                df_measured.to_csv(f"{device}/simulated/{device}_{process}_{temp}c_{volt}v.csv",index= False)

def error_cal(device, processes, volts, temps):

    print (f"\n\nSimulation results of {device}")
    measured  = pd.read_csv(f"{device}/{device}_measured.csv")
    for process in processes:
        for volt in volts:
            for temp in temps:
                simulated = pd.read_csv(f"{device}/simulated/{device}_{process}_{temp}c_{volt}v.csv")

                res = (measured == simulated).all().all()
                print ("{:^5s} in PVT of {:^7s}, {:^3s}V, {:^3s}C functional simulation = {}".format(device, process, volt, temp, res))    
    print ("================================================================================================\n")

def main():

    devices = ["inv","nand2","or3"]

    # Generate truth tables data
    inv_table = [["input","output"],
                 [0,1],
                 [1,0]]

    nand2_table = [["input1","input2","output"],
                   [0,0,1],
                   [0,1,1],
                   [1,0,1],
                   [1,1,0]]

    or3_table = [["input1","input2","input3","output"],
                 [0,0,0,0],
                 [0,0,1,1],
                 [0,1,0,1],
                 [0,1,1,1],
                 [1,0,0,1],
                 [1,0,1,1],
                 [1,1,0,1],
                 [1,1,1,1]]

    tables = [inv_table,nand2_table,or3_table]

    processes   = ["typical","ff","ss"]
    volts       = ["5", "4.5", "5.5"]
    temps       = ["25", "-40", "125"]

    for i, device in enumerate(devices): 
        # Folder structure of measured values
        if os.path.exists(device) and os.path.isdir(device):
            shutil.rmtree(device)
        os.makedirs(device)

        # Folder structure of simulated values
        os.makedirs(f"{device}/{device}_netlists",exist_ok=True)
        os.makedirs(f"{device}/simulated",exist_ok=True)
        
        ext_measured (device, tables[i])
        # =========== Simulate ==============
        ext_simulated(device, processes, volts, temps)

        # ============ Results ==============
        error_cal    (device, processes, volts, temps)

# ================================================================
# -------------------------- MAIN --------------------------------
# ================================================================

if __name__ == "__main__":

    # Args
    arguments     = docopt(__doc__, version='comparator: 0.1')
    workers_count = os.cpu_count()*2 if arguments["--num_cores"] == None else int(arguments["--num_cores"])

    # Calling main function 
    main()