import tkPlotXY
import numpy
import Converter.PyTree as C
import Converter.Internal as Internal

# Change some default parameters of tkPlotXY by using the default_values dictionnary
# AXIS
tkPlotXY.default_values['Axis']['axis_x_min'] = 0.
tkPlotXY.default_values['Axis']['axis_x_max'] = 1.
tkPlotXY.default_values['Axis']['axis_x_autoscale'] = False
# GRID
tkPlotXY.default_values['Grid']['Mx_grid_color'] = 'black'
tkPlotXY.default_values['Grid']['My_grid_color'] = 'black'
# LEGEND
tkPlotXY.default_values['Legend']['legend_border_color'] = 'black'
# CURVE
tkPlotXY.default_values['Curve']['line_width'] = 1.5

# Load cgns files
walls = C.convertFile2PyTree('wall1.cgns')

# Create a data zone for each cgns file
# Keys can be later used as legend label
data = {}
data['wall'] = {
    'x/c':Internal.getNodeFromName(walls, 'CoordinateX')[1],
    'Cp' :Internal.getNodeFromName(walls, 'Cp')[1],
    'Cf' :Internal.getNodeFromName(walls, 'Cf')[1]
}

# Plot
desktop, win = tkPlotXY.createTkDesktop()
desktop.setData(data)

# Create a graph
graph0 = desktop.createGraph('Graph_Cf', '1:1', figsize=(6,4))
graph0.updateSubPlotParams({'isActive':True, 'right':0.97, 'top':0.97, 'left':0.2, 'bottom':0.13, 'wspace':0.3})

# Each new curve must be created first and then added to the graph with addCurve()
curve = tkPlotXY.Curve(zone=["wall"], varx='x/c', vary='Cf', legend_label='wall')
graph0.addCurve('1:1', curve)

# Open window for curve editing when starting tkPlotXY
desktop.cmd_editCurves()
win.mainloop()