a 7b3@slddlmZddlTddlZddlmZmZddlm Z ddl m Z ddl m Z e j Z Gd d d e j ZdS) )print_function)*N)platformexit)path)thermo)cubiccs:eZdZdZfddZddd Zd d Zd d ZZS)cpaz3 Interface to cubic plus association model csTtt|t|j|dd|_t|j|dd|_t|j|dd|_dS)z= Initialize cubic specific function pointers Z eoslibinitZinit_cpaZsaft_interfaceZ cpa_get_kijZ cpa_set_kijN) superr __init__getattrtpZget_export_nameeoslibinit_init_cpa s_get_kij s_set_kij)self __class__Y/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/pyctp/cpa.pyr sz cpa.__init__SRKvdWClassicDefaultc Cs|t|d}tt|}t|d}tt|} t|d} tt|} t|d} tt|} t|d}tt|}ttttttttttg |j_d|j_|| ||| || || | | tt| dt| d|_ dS)aInitialize cubic plus association model in thermopack Args: comps (str): Comma separated list of component names eos (str, optional): Cubic equation of state. Defaults to "SRK". mixing (str, optional): Mixture model. Defaults to "vdW". alpha (str, optional): Alpha model. Defaults to "Classic". parameter_reference (str, optional): Which parameters to use?. Defaults to "Default". asciiN ,) activatec_char_pencode c_len_typelenrargtypesrestypemaxsplitZnc)rcompsZeosZmixingalphaZparameter_referenceZeos_cZeos_lenZmixing_cZ mixing_lenZalpha_cZ alpha_lenZ comp_string_cZcomp_string_lenZ ref_string_cZref_string_lenrrrinit'sF        zcpa.initcCsf|t|}t|}tdd}ttttttg|j_d|j_|t|t||t |S)zGet attractive energy interaction parameter Args: c1 (int): Component one c2 (int): Component two Returns: kij (array_like): i-j interaction parameter (2 parameters) gN) rc_intc_doublePOINTERrr#r$byrefnparray)rc1c2c1_cc2_ckij_crrrget_kijWs   z cpa.get_kijcCs`|t|}t|}td|}ttttttg|j_d|j_|t|t||dS)zSet attractive energy interaction parameter Args: c1 (int): Component one c2 (int): Component two kij (array_like): i-j interaction parameter (2 parameters) r*N)rr+r,r-rr#r$r.)rr1r2Zkijr3r4r5rrrset_kijqs  z cpa.set_kij)rrrr) __name__ __module__ __qualname____doc__r r)r6r7 __classcell__rrrrr s  0r ) __future__rctypesZnumpyr/sysrrosrrr r!r rrrrs