a
    7b                     @   s`   d dl mZ d dlT d dlZd dlmZmZ d dlm	Z	 ddl
mZ ejZG dd	 d	ejZdS )
    )print_function)*N)platformexit)path   )thermoc                       s*   e Zd ZdZ fddZdddZ  ZS )petsz
    Interface to PETS
    c                    s(   t t|   t| j| dd| _dS )z<
        Initialize pets specific function pointers
        Z
eoslibinitZ	init_petsN)superr	   __init__getattrtpZget_export_nameeoslibinit_init_pets)self	__class__ Z/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/pyctp/pets.pyr      s    zpets.__init__Defaultc                 C   sL   |    t|d}tt|}ttg| j_d| j_| || d| _dS )a  Initialize he PeTS equation of state for the LJ fluid
        truncated and shifted at 2.5 sigma. Reference:
        Heier et al. 2018 (10.1080/00268976.2018.1447153)

        Args:
            parameter_reference (str, optional): Wath parameters to use. Defaults to "Default".
        asciiNr   )	activatec_char_pencode
c_len_typelenr   argtypesrestypeZnc)r   Zparameter_referenceZref_string_cZref_string_lenr   r   r   init#   s    z	pets.init)r   )__name__
__module____qualname____doc__r   r   __classcell__r   r   r   r   r	      s   r	   )
__future__r   ctypesZnumpynpsysr   r   osr    r   r   Z
thermopackr	   r   r   r   r   <module>   s   