import CoolProp
import os

fluid_template = """.. _fluid_{fluid:s}:

{fluid_stars:s}
{fluid:s}
{fluid_stars:s}

References
==========

{references:s}
{aliases:s}

Fluid Information
=================

.. csv-table::
   :header-rows: 1
   :widths: 40, 60
   :file: {fluid:s}-info.csv
   
REFPROP Validation Data
=======================

.. note::

    You can download the script that generated the following figure here: :download:`(link to script)<REFPROPplots/{fluid:s}.py>`, right-click the link and then save as... or the equivalent in your browser.  You can also download this figure :download:`as a PDF<REFPROPplots/{fluid:s}.pdf>`. 

.. image:: REFPROPplots/{fluid:s}.png

Consistency Plots
=================

The following figure shows all the flash routines that are available for this fluid.  A red + is a failure of the flash routine, a black dot is a success.  Hopefully you will only see black dots

The red curve is the maximum temperature curve, and the blue curve is the melting line if one is available for the fluid.

.. note::

    You can download the script that generated the following figure here: :download:`(link to script)<Consistencyplots/{fluid:s}.py>`, right-click the link and then save as... or the equivalent in your browser.  You can also download this figure :download:`as a PDF<Consistencyplots/{fluid:s}.pdf>`. 

.. image:: Consistencyplots/{fluid:s}.png

"""

table_template = """ Parameter, Value
Molar mass [kg/mol],{mm:s}
CAS, {CAS:s}
ASHRAE, {ASHRAE:s}
Triple point temperature [K],{Tt:s}
Triple point pressure [Pa], {pt:s}
Critical point temperature [K], {Tc:s}
Critical point pressure [Pa], {pc:s}
Critical point density [kg/m3], {rhoc_mass:s}
Critical point density [mol/m3], {rhoc_molar:s}
"""
class FluidInfoTableGenerator(object):
    
    def __init__(self, name):
    
        self.name = name
        
    def write(self, path):
        def tos(n):
            ''' convert number to nicely formatted string '''
            n = str(n)
            if 'e' in n:
                n = n.replace('e',':math:`\times 10^{')
                n += '}`'
            else:
                return n
        molar_mass = CoolProp.CoolProp.PropsSI(self.name,'molemass')
        Tt = CoolProp.CoolProp.PropsSI(self.name,'Ttriple')
        Tc = CoolProp.CoolProp.PropsSI(self.name,'Tcrit')
        pc = CoolProp.CoolProp.PropsSI(self.name,'pcrit')
        pt = CoolProp.CoolProp.PropsSI(self.name,'ptriple')
        rhoc_mass = CoolProp.CoolProp.PropsSI(self.name,'rhomass_critical')
        rhoc_molar = CoolProp.CoolProp.PropsSI(self.name,'rhomolar_critical')
        CAS = CoolProp.CoolProp.get_fluid_param_string(self.name, "CAS")
        ASHRAE = CoolProp.CoolProp.get_fluid_param_string(self.name, "ASHRAE34")
            
        args = dict(mm = tos(molar_mass),
                    Tt = tos(Tt),
                    pt = tos(pt),
                    Tc = tos(Tc),
                    rhoc_mass = tos(rhoc_mass),
                    rhoc_molar = tos(rhoc_molar),
                    pc = tos(pc),
                    CAS = tos(CAS),
                    ASHRAE = tos(ASHRAE))
        out = table_template.format(**args)
        
        with open(os.path.join(path, self.name+'-info.csv'),'w') as fp:
            print 'writing', os.path.join(path, self.name+'-info.csv')
            fp.write(out)
            
class FluidGenerator(object):
    def __init__(self, fluid):
        self.fluid = fluid
        
    def write(self, path):
        
        # Write CSV table data for fluid information
        ITG = FluidInfoTableGenerator(self.fluid)
        ITG.write(path)
        
        aliases = ', '.join(['``' + a.strip() + '``' for a in CoolProp.CoolProp.get_fluid_param_string(self.fluid, 'aliases').strip().split(',') if a])
        if aliases:
            aliases = 'Aliases\n=======\n\n'+aliases + '\n'
        
        # Write RST file for fluid
        out = fluid_template.format(aliases = aliases,
                                    fluid = self.fluid,
                                    fluid_stars = '*'*len(self.fluid),
                                    references = ''
                                    )
        
        with open(os.path.join(path, self.fluid+'.rst'),'w') as fp:
            print 'writing', os.path.join(path, self.fluid+'.rst')
            fp.write(out)