# -*- coding: utf-8 -*-
"""
Created on Sat Jul 12 14:53:20 2014

@author: Matthieu
"""

from CoolProp.Plots import PropsPlot



# Pour l'eau, la température doit rester comprise entre 273 et 647 K

ts_plot = PropsPlot('Water', 'Ts')
ts_plot.xlabel(r'$s$ $($kJ.kg$^{-1}$.K$^{-1})$')
ts_plot.ylabel(r'$T$ $($K$)$')

# Le if False / if True permet de gérer facilement ce qui est tracé ou non.


#### ISO-ENTHALPIQUES
# les limites sur s des axes ont été trouvées à la main
# car impossible de tracer les isoH brutalement
# les lignes commentées fonctionnent mais sont supprimées
# pour alléger le graphique
#ts_plot.draw_isolines('H', [600,600],axis_limits=[[0.0, 2.1], [200.0, 700.0]])

if True:
    ts_plot.draw_isolines('H', [800,800],axis_limits=[[0.0, 2.9], [200.0, 700.0]])
    #ts_plot.draw_isolines('H', [1000,1000],axis_limits=[[0.0, 3.5], [200.0, 700.0]])
    ts_plot.draw_isolines('H', [1200,1200],axis_limits=[[0.0, 4.25], [200.0, 700.0]])
    #ts_plot.draw_isolines('H', [1400,1400],axis_limits=[[0.0, 5.05], [200.0, 700.0]])
    ts_plot.draw_isolines('H', [1600,1600],axis_limits=[[0.0, 5.75], [200.0, 700.0]])
    #ts_plot.draw_isolines('H', [1800,1800],axis_limits=[[0.0, 6.45], [200.0, 700.0]])
    ts_plot.draw_isolines('H', [2000,2000],axis_limits=[[0.0, 7.25], [200.0, 700.0]])
    #ts_plot.draw_isolines('H', [2200,2200],axis_limits=[[0.0, 8], [200.0, 700.0]])
    ts_plot.draw_isolines('H', [2400,2400],axis_limits=[[1.0, 8.7], [200.0, 700.0]])
    #ts_plot.draw_isolines('H', [2600,2600],axis_limits=[[4.5, 9.4], [200.0, 700.0]])
    ts_plot.draw_isolines('H', [2800,2800],axis_limits=[[5.0, 10], [200.0, 700.0]])
    #ts_plot.draw_isolines('H', [2800,2800],axis_limits=[[5.5, 10], [200.0, 700.0]])
    ts_plot.draw_isolines('H', [3000,3000],axis_limits=[[6.0, 10], [200.0, 700.0]])
    ts_plot.title("Diagramme entropique de l'eau : isenthalpiques")

#### ISO-TITRE
if True:
    ts_plot.draw_isolines('Q', [0.1, 0.9],num=9)
    ts_plot.title("Diagramme entropique de l'eau : isotitres")
    
#### ISO-BARE
if True:
    ts_plot.draw_isolines('P', [10, 10000], num=9)
    ts_plot.title("Diagramme entropique de l'eau : isobares")
    
#### ISO-VOLUME MASSIQUE (en fait iso-densité)
if True:
    ts_plot.draw_isolines('D', [2, 600], num=7)
    ts_plot.title("Diagramme entropique de l'eau : iso-volume massique")

#### titre complet
if True:
    ts_plot.title("Diagramme entropique de l'eau")


#### le cadrage    
ts_plot.set_axis_limits([0.0, 10.0, 250.0, 700.0])

#### les légendes
ts_plot.axis.text(3.1e3, 300, "Courbe d'ébullition", fontsize=15,verticalalignment='center')
ts_plot.axis.arrow(3e3, 300, -1e3,130,shape='full',width=1,head_width=25,head_length=250,overhang=.5,lw=1,label='truc',length_includes_head=True,color='k')
ts_plot.axis.text(5.9e3, 350, "Courbe de rosée", fontsize=15,verticalalignment='center',horizontalalignment='right')
ts_plot.axis.arrow(6e3, 350, 1e3,50,shape='full',width=1,head_width=25,head_length=250,overhang=.5,lw=1,label='truc',length_includes_head=True,color='k')

ts_plot.grid()
ts_plot.show()