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

@author: Matthieu
"""

from CoolProp.Plots import PropsPlot
from CoolProp.State import State




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

fluide_reference='SulfurHexafluoride'
nom="SF$_6$"
Ph = PropsPlot(fluide_reference, 'Ph')
Ph.xlabel(r'$h$ $($kJ.kg$^{-1})$')
Ph.ylabel(r'$P$ $($kPa$)$')

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

S0=S1 = State(fluide_reference, dict(P=4000, H=0))
print(S0)


#### 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
#Ph.draw_isolines('H', [600,600],axis_limits=[[0.0, 2.1], [200.0, 700.0]])

# pour les titres
chaine="Diagramme $(P,h)$ de "+nom

if True:
    Ph.draw_isolines('S', [-0.0,-0.0],axis_limits=[[-160.0, -20.0], [250.0, 5000.0]])
    Ph.draw_isolines('S', [-0.1,-0.1],axis_limits=[[-160.0, -20.0], [250.0, 5000.0]])
    Ph.draw_isolines('S', [-0.2,-0.2],axis_limits=[[-160.0, -20.0], [250.0, 5000.0]])
    Ph.draw_isolines('S', [-0.3,-0.3],axis_limits=[[-160.0, -20.0], [250.0, 3740.0]])
    Ph.draw_isolines('S', [-0.3,-0.3],axis_limits=[[-160.0, -20.0], [3750.0,5000.0]])
    Ph.draw_isolines('S', [-0.4,-0.4],axis_limits=[[-160.0, -20.0], [250.0, 5000.0]])
    Ph.draw_isolines('S', [-0.5,-0.5],axis_limits=[[-160.0, -20.0], [250.0, 5000.0]])
    Ph.draw_isolines('S', [-0.6,-0.6],axis_limits=[[-160.0, -20.0], [250.0, 5000.0]])
    chaine+=" : isentropiques"

#### ISO-TITRE
if True:
    Ph.draw_isolines('Q', [0.1, 0.9],num=9)
    chaine+=" : isotitres"
    
#### ISO-T
if True:
    Ph.draw_isolines('T', [200, 350], num=9)
    chaine+=" : isothermes"
    
#### ISO-VOLUME MASSIQUE (en fait iso-densité)
if True:
    Ph.draw_isolines('D', [2, 600], num=9)
    chaine+=" : iso-volume massique"


#### le cadrage    
Ph.set_axis_limits([-160.0, 50.0, 0.0, 5000.0])

# pour le titre
if True:
    chaine="Diagramme $(P,h)$ de "+nom

Ph.title(chaine)

Ph.grid()
Ph.show()