import csv  # import the python csv library
from Libraries.StandardFunctions import *
from Libraries.ChartFunctions import *
import WflowDeltashell.wflib as wfl
import os

# Needed if this .net thing is not loaded yet
import clr

clr.AddReference("System.Windows.Forms")
from System.Windows.Forms import OpenFileDialog, DialogResult


def getcsvname():
    themodel = wfl.GetModelByPartialName("wflow")

    dialog = OpenFileDialog()

    if themodel:
        dialog.InitialDirectory = os.path.join(
            themodel.DirectoryPath, themodel.DefaultOutputDirectoryName
        )
    else:
        dialog.InitialDirectory = "C:\\"

    dialog.Filter = "csv files (*.csv) | *.csv"
    dialog.FilterIndex = 1
    dialog.RestoreDirectory = False
    dialog.Title = "Select a WFlow result csv file: "

    if dialog.ShowDialog() == DialogResult.OK:
        thefile = dialog.FileName

    return thefile


def plotit(filename):
    colors = [
        Color.Red,
        Color.Black,
        Color.Green,
        Color.Black,
        Color.Orange,
        Color.YellowGreen,
        Color.Azure,
        Color.Cyan,
        Color.Brown,
        Color.Aquamarine,
        Color.CornflowerBlue,
        Color.DarkBlue,
        Color.BlanchedAlmond,
        Color.LightYellow,
        Color.SlateGray,
    ]

    runid = os.path.basename(os.path.dirname(filename))

    lineSeries = []

    olines = wfl.readwfcsv(filename)

    for yval in range(1, len(olines[0])):
        x = [[col[0], col[yval]] for col in olines]

        lineSeries.append(CreateLineSeries(x))

        lineSeries[-1].Color = colors[yval - 1]
        lineSeries[-1].Width = 1
        lineSeries[-1].PointerVisible = False
        lineSeries[-1].PointerSize = 0
        lineSeries[-1].PointerColor = Color.Red
        lineSeries[-1].PointerLineVisible = True
        lineSeries[-1].PointerLineColor = colors[yval - 1]
        lineSeries[-1].Transparency = 0
        lineSeries[-1].ShowInLegend = True
        lineSeries[-1].Title = "Id: " + str(yval)

    chart = CreateChart(lineSeries)

    # Configure the chart
    chart.TitleVisible = True
    chart.Title = "run: " + runid + ": " + os.path.basename(filename)
    chart.BackGroundColor = Color.White
    chart.Legend.Visible = True
    chart.Name = chart.Title

    # Configure the bottom axis
    chart.BottomAxis.Automatic = True
    chart.BottomAxis.Title = "Timestep"

    # Configure the left axis
    chart.LeftAxis.Title = os.path.basename(filename)

    # Show the chart
    view = OpenView(chart)


def complot(dirnames, file, ids):
    colors = [
        Color.Red,
        Color.Black,
        Color.Green,
        Color.Black,
        Color.Orange,
        Color.YellowGreen,
        Color.Azure,
        Color.Cyan,
        Color.Brown,
        Color.Aquamarine,
        Color.CornflowerBlue,
        Color.DarkBlue,
        Color.BlanchedAlmond,
        Color.LightYellow,
        Color.SlateGray,
    ]

    cnt = 0
    lineSeries = []
    for dirname in dirnames:
        runid = os.path.dirname(dirname)

        ftoread = os.path.join(dirname, file)

        if os.path.exists(ftoread):
            olines = wfl.readwfcsv(os.path.join(dirname, file))

            for yval in ids:
                x = [[col[0], col[yval]] for col in olines]

                lineSeries.append(CreateLineSeries(x))

                lineSeries[-1].Color = colors[cnt]
                lineSeries[-1].Width = 1
                lineSeries[-1].PointerVisible = False
                lineSeries[-1].PointerSize = 0
                lineSeries[-1].PointerColor = Color.Red
                lineSeries[-1].PointerLineVisible = True
                lineSeries[-1].PointerLineColor = colors[cnt]
                lineSeries[-1].Transparency = 0
                lineSeries[-1].ShowInLegend = True
                lineSeries[-1].Title = (
                    "Col: " + str(yval) + " Run: " + str(os.path.basename(runid))
                )
                cnt = cnt + 1

    casename = os.path.basename(os.path.dirname(os.path.dirname(dirname)))
    chart = CreateChart(lineSeries)

    # Configure the chart
    chart.TitleVisible = True
    chart.Title = casename.upper() + ":" + os.path.basename(file)
    chart.BackGroundColor = Color.White
    chart.Legend.Visible = True
    chart.Name = chart.Title

    # Configure the bottom axis
    chart.BottomAxis.Automatic = True
    chart.BottomAxis.Title = "Timestep"

    # Configure the left axis
    chart.LeftAxis.Title = os.path.basename(file)

    # Show the chart
    view = OpenView(chart)