from CoolProp.Plots.Plots import hs
import CoolProp
from CoolProp.CoolProp import Props
import matplotlib.pyplot as plt
import numpy as np

Fluid = 'Nitrogen'

fig = plt.figure()
ax = fig.add_subplot(111)

for Fluid in CoolProp.__fluids__:
    
    if Fluid == 'SES36':
        continue
        
    ax.cla()
    
    h_crit = Props('H','T',Props(Fluid,"Tcrit"),'D',Props(Fluid,"rhocrit"),Fluid)
    s_crit = Props('S','T',Props(Fluid,"Tcrit"),'D',Props(Fluid,"rhocrit"),Fluid)
   
    hL_Tmin = Props('H','T',Props(Fluid,"Tmin"),'Q',0,Fluid)
    hV_Tmin = Props('H','T',Props(Fluid,"Tmin"),'Q',1,Fluid)
    sL_Tmin = Props('S','T',Props(Fluid,"Tmin"),'Q',0,Fluid)
    sV_Tmin = Props('S','T',Props(Fluid,"Tmin"),'Q',1,Fluid)
    
    hs(Fluid, axis = ax)
    plt.plot(s_crit,h_crit,'rd')
    plt.plot([sL_Tmin,sV_Tmin],[hL_Tmin,hV_Tmin],'--')
    plt.gca().axhline(h_crit)
    plt.gca().axhline(hV_Tmin)
    
    # Two-Phase
    for T in np.linspace(Props(Fluid,"Tmin")+0.1,Props(Fluid,"Tcrit")-1e-3,30):
        for Q in np.linspace(0, 1, 30):
            try:
                h = Props("H",'Q',Q,'T',T,Fluid)
                s = Props("S",'Q',Q,'T',T,Fluid)
                T = Props("T",'S',s,'H',h,Fluid)
                #ax.plot(s,h,'o',mfc='none')
            except ValueError as VE:
                print T, Q , '|||', '"T","S",',s,',"H",',h,',"'+Fluid+'"', '|||', VE
                ax.plot(s,h,'o',mfc='none')
                
        
    for h in np.linspace(hL_Tmin, hV_Tmin + 1500,100):
        for s in np.linspace(sL_Tmin+0.01,sV_Tmin,100):
            try:
                h_pmax = Props('H','S',s,'P',6*Props(Fluid,'pcrit'),Fluid)
            except ValueError:
                h_pmax = 0
            htriple_s = (hV_Tmin-hL_Tmin)/(sV_Tmin-sL_Tmin)*(s-sL_Tmin)+hL_Tmin
            if h < htriple_s or h > h_pmax: continue
            try:
                T = Props("T",'S',s,'H',h,Fluid)
                #ax.plot(s,h,'o',mfc='none',ms=6)
            except ValueError:
                ax.plot(s,h,'s',mfc='none')
   

##     if Fluid =='Propane':
##         ps = Props("P",'T',300,'Q',0,Fluid);
##         hL = Props("H",'Q',0,'T',300,Fluid);
##         sL = Props("S",'Q',0,'T',300,Fluid);
##         h = Props("H",'P',ps,'T',299.5,Fluid);
##         s = Props("S",'P',ps,'T',299.5,Fluid);
##         print s,h,sL,hL
##         plt.plot(s,h,'o')
##         plt.plot(sL,hL,'d')
##         plt.gca().axvline(s)
##         plt.gca().axhline(268.75968916316691)
    
    fig.savefig('figs/'+Fluid+'.png',dpi=200)
    fig.savefig('figs/'+Fluid+'.pdf')