a 7b@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_/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/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#)Zskiprows)opensplitnpZloadtxt)rfilenlineslinewordsdatarrrloadFiles  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 )r rlen)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 r rfloatr)rrrTrrrrrgetFloatAs  rcCsVt|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 )r rintr)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!mZnansijrrr 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   " r8cCsZg}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:rNr/)r rr"r,r&rr2r1)rZVLEZVLLEZCRITZAZZKSTYPErZnCritZnLLVEZnAZZnVLEr4r*r6r7rr+rrrget_globa_binary_datasd        "   $    $   $r;) Znumpyrr'rr rrrr"r$r,r8r;rrrrs $