U
    Aa	                     @   s  d dl mZmZmZ d dlZd dlmZ ddddddd	gZ	d
d Z
dd Zdd Zdd Zedkrdgd ZddddddgZdZedddd ZeeZd Ze Zeed eeD ]\ZZedeeee< qejed eee d  ed7 Ze Zeed eeD ]\ZZedeeee< qejed eee d  ed7 Ze Zed!d eeD ]\ZZedeeee< qlejed eee d  ed7 Ze Zeed eeD ]\ZZedeeee< qejed eee d  ed7 Ze Zeedd"dd#d$ eeD ]\ZZedeeee< q$ejed eee d  ed7 Zejd%d dd&d' ejdd(d)d&d' e   e!d* dS )+    )cubiccpaextended_cspNz	Two-phaseZLiquidZVapourzMinimum GibbsSingleZSolidZFakec                    s    fdd| D S )Nc                    s   g | ]}t | qS  )round).0xinr   ;/home/ubuntu/Home/Documents/9_semester/soret_model/tools.py
<listcomp>   s     zround_lst.<locals>.<listcomp>r   )xr   r   r
   r   	round_lst   s    r   c                 C   sH   t   }|dd tdddg}||d|\}}}}}|d |  S N	H2O,O2,N2PRg+?g-C6*?g-C6J?i͋ r   r   initnparraytwo_phase_tpflash)rhTeosr   ybetaVbetaLphr   r   r   humidity_2_mole_frac   s
    r   c           	      C   sH   t   }|dd tdddg}||d| \}}}}}| |d  S r   r   )	r   r   r   Zx_totZx_liqr   r   r   r   r   r   r   mole_frac_2_humidity   s
    r    c                 C   s*   | D ] }|D ]}t |dd qt   qd S )N )end)print)arrliner   r   r   r   	print_arr   s    r&   __main__   SRKr   PTSWZSPUNGCPAr   P   x   2   i     )labelH2OClassicZMBWR32N2d   black)colorZ   n   zVapour pressure)"pycThermopack.pyctpr   r   r   numpyr   matplotlib.pyplotpyplotpltZphasesr   r   r    r&   __name__eos_listnamescompslinspaceZ	temp_list
zeros_likeZ	pres_listr   eqr   	enumerateitempplotext_cspvlineshlineslegendsavefigr   r   r   r   <module>   s`   


