B }N_@sdddlZddlZddlZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS)NcCstj|}|S)zTest if file exist Args: filename (str): File path Returns: bool: Does file with filename exist? )ospathisfile)filenameexistsr#../pycThermopack/pyctp/plotutils.py file_existss r cCsRt|d}d}x0|D](}|}|dddkr:|d}qPqWtj||d}|S)zxLoad data from file Args: filename (str): File path Returns: ndarray: Data stored in file rr#)skiprows)opensplitnploadtxt)rfileZnlineslinewordsdatarrrloadFiles   rcCslt|d}d}d}xP|D]H}|}|dddkr^||kr`|t|d}|t|d}qPqW||fS)zGet last two words on line identefied by pointLabel. Args: filename (str): File path pointLabel (str): Searchable string Returns: str: Last two entries on line containg pointLabel r rr r )rrlen)r pointLabelrZcomp1Zcomp2rrrrr getBinary)s  rcCs\t|d}d}xH|D]@}|}|dddkrR|d|krTt|t|d}qPqW|S)zGet float from line identefied by pointLabel. Last entry on line covetered to float. Args: filename (str): File path pointLabel (str): Searchable string Returns: float: Last entry on line containg pointLabel r grr r )rrfloatr)rrrTrrrrrgetFloatAs   rcCsXt|d}d}xD|D]<}|}|dddkrN||krPt|t|d}qPqW|S)zGet integer from line identefied by pointLabel. Last word on line covetered to int. Args: filename (str): File path pointLabel (str): Searchable string Returns: int: Last entry on line containg pointLabel r rr r )rrintr)rrrnrrrrr getIntegerWs  r!cCstt|d}d}d}xX|D]P}|}|dddkrf||krht|t|d}t|t|d}qPqW||fS)aGet point (T,P) from line identefied by pointLabel. Last two words on line covetered to floats. Args: filename (str): File path pointLabel (str): Searchable string Returns: float: Last two entries on line containg pointLabel r grr rr )rrrr)rrrrPrrrrrgetPointms  r#cCsjt|d}t|d}|g|}x>t|D]2}x,t|D] }t|||fr>|||<Pq>Wq0W|S)zGet location of NaNs in data arrays Args: data (ndarray): Data containing NaNs Returns: array_like: Column indices giving first apperance of NaN rr )rshaperangemathisnan)rr mZnansijrrr getNaNindicess   r+cCsg}g}t|rt|}t|d\}}|dkr>|||gt|d\}}|dkrb|||gt|}t|d} | dkrd} n | dkrd} xttd D]h} | | } || } | d krqt| | f}x0t| D]$}|d | || f|d d |f<qW||qW||fS) zRead data from solid envelope data. Args: filename (str): File path Returns: lists: Lists of phase lines and critical points z#Critical point 1g?z#Critical point 2r rrN) r rr#appendr+rr$r%zeros)rlinesZcritsrZTc1ZPc1ZTc2ZPc2i_nansr(kr)istarti_nanrr*rrrget_solid_envelope_datas4  $r7cCs|g}g}g}g}d}t|rnt|}t|d}t|d}t|d}t|d} d} t|} xtt| D]h} | d} | | }|dkrqdt|df}x0tdD]$}|d||| f|dd|f<qW||qdWxt|D]t} | d| d} | | }|dkrqt|df}x2tdD]&}|d||| f|dd|f<qW||qWxt|D]|} | |d| d } | | }|dkrq\t|d f}x2td D]&}|d||| f|dd|f<qW||q\Wxt| D]} | ||dd| d } | | }|dkrqt|d f}x2td D]&}|d||| f|dd|f<q4W||qW|||||fS) zg[summary] Args: filename (str): File path Returns: [type]: [description] rz#Global phase diagram of type:z#Number of critical lines:z#Number of LLVE lines:z#Number of AZ lines:rNr.)r rr!r+r%rr1r0)rZVLEZVLLEZCRITZAZZKSTYPErZnCritZnLLVEZnAZZnVLEr3r)r5r6rr*rrrget_globa_binary_datasd      $ & & &r:) numpyrr&rr rrrr!r#r+r7r:rrrrs $