a }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)filenameexistsrV/Users/vegardjervell/Documents/9_semester/soret_model/pycThermopack/pyctp/plotutils.py file_existss r cCsPt|d}d}|D]*}|}|dddkr8|d}qq>qtj||d}|S)zxLoad data from file Args: filename (str): File path Returns: ndarray: Data stored in file rr#)skiprows)opensplitnploadtxt)rfileZnlineslinewordsdatarrrloadFiles  rcCsjt|d}d}d}|D]J}|}|dddkr\||vr`|t|d}|t|d}qqbq||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 rcCsZt|d}d}|D]B}|}|dddkrP|d|krTt|t|d}qqVq|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 rr r rrfloatr)rrrTrrrrrgetFloatAs  r cCsVt|d}d}|D]>}|}|dddkrL||vrPt|t|d}qqRq|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#cCsrt|d}d}d}|D]R}|}|dddkrd||vrht|t|d}t|t|d}qqjq||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 rrr rr r)rrrrPrrrrrgetPointms r%cCsdt|d}t|d}|g|}t|D]0}t|D]"}t|||fr:|||<q.q:q.|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"mnansijrrr getNaNindicess    r.cCsg}g}t|rt|}t|d\}}|dkr<|||gt|d\}}|dkr`|||gt|}t|d} | dkrd} n | dkrd} td D]d} | | } || } | d krqt| | f}t| D]$}|d | || f|d d |f<q||q||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   " r:cCsZg}g}g}g}d}t|rLt|}t|d}t|d}t|d}t|d} d} t|} t| D]d} | d} | | }|dkrqbt|df}tdD]$}|d||| f|dd|f<q||qbt|D]n} | d| d} | | }|dkrqt|df}tdD]&}|d||| f|dd|f<q ||qt|D]x} | |d| d } | | }|dkrvqHt|d f}td D]&}|d||| f|dd|f<q||qHt| D]} | ||dd| d } | | }|dkrqt|d f}td D]&}|d||| f|dd|f<q||q|||||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:rNr1)r rr#r.r'rr4r3)rZVLEZVLLEZCRITZAZZKSTYPErZnCritZnLLVEZnAZZnVLEr6r,r8r9rr-rrrget_globa_binary_datasd        "   $    $   $r=) numpyrr(rr rrr r#r%r.r:r=rrrrs $