B L_X@sddlZddlmZddlmZddlmZddl Z ddl Z ddl Z ddl mZGdddZGdddZdd ZGd d d ZdS) N)Axes3D)CloughTocher2DInterpolatorc@sfeZdZddgdfddZdddZddd Zdd d Zd d ZddZddZ ddZ dddZ dS)Alpha_T0_empiricalg?wheightscCs|d\|_|_||_ttjt d}|j|j |d|jkdkrNnH|j|j |d|jkdkr|j|j|_|_d|_ nt d|d|j |d|jk|d|jk@|_ t|j d|_t|j d |_t|j d |_t|jt|jt|jt|jf|_t|j|_t|jt|j|_t|jt|j|_|jd krld |_|jd kr~d |_|d |_|d kr|j|_n0|dkr|j|_n|dkr|j|_nt ddS)a Callable object, composition and components are set upon initialization __call__(temp): returns alpha_T0 for component 1 at given composition and temperature, approximated from experimental data :param comps: 'comp1,comp2' :param mole_fracs: mole fractions of components :param method: 'wheights' (default), 'ct': CloughTocher, 'plane': linear interpolation ,z/alpha_t0.xlsxComp1Comp2Tz Components z not found in database.c_avgT_avg Alpha_T_0rrZplanectz1method must be either 'wheights', 'plane' or 'ct'N) splitcomp1comp2compspd read_excelospathdirname__file__locflip ValueErrordatanparraytolistc_listT_list alpha_listminmax cT_rangeslen num_pointsc_spanT_spancget_alpha_T0_wheights get_alpha_T0get_alpha_T0_planeget_alpha_T0_CloughTocher)selfr mole_fracsmethoddfr2^/Users/vegardjervell/Documents/7_semester/irrev_prosjekt_renskriv/models/alpha_t0_empirical.py__init__ s: $&          zAlpha_T0_empirical.__init__Fc s||_|jd}tt||jdk||jdk||jdk||jdkgrtdtt |ddtt |ddtt |jddd tt |jddd tt |jddd tt |jddd |j ||j d|j ||j d}td d tt|dddD}tdd |D}d|d<t|j |dg}t|j |dg}d} x|||||s~| |jkr~|j || } |j || } || |j | |j ||j ||j ||j ||j } x\| |jdkrF| sF| d7} |j || } |j || } || | ||||rd} PqW| rVd|| <|j ||}|j ||}| d7} qxW|j||} |||tfdd ttDfdd ttD}t| |}t||d||d| |dt| }|r.||||| |||||dksD| |jkrtdtt |ddtt |dd|jdtt |dd t|| gS)a :param comps (str) : comp1,comp2 following Thermopack convention :param c (float) : Volumetric concentration at which to get kinetic_gas :param T (float) : Temperature (K) at which to get kinetic_gas :param plot (bool or '2d' or '3d') : display fitted data :return float : kinetic_gas at the specified conditions, approximated from experimental data using the custom weighting scheme drr z c = z T = z is outside range c : (z, z), T : )cSsg|] \}}|qSr2r2).0i_r2r2r3 Wsz.cSs|dS)Nr r2)pairr2r2r3Wsz:Alpha_T0_empirical.get_alpha_T0_wheights..)keycSsg|]}dqS)Fr2)r9r:r2r2r3r<XsTcs4g|],}td|t|ddqS)Nr )rprod)r9r:)rr2r3r<zscs8g|]0}td|t|ddqS)Nr )rr@)r9r:)rAtotr2r3r<{siz Alpha_T0 at c = z for z# may be a bad approximation (R^2 = )plotr)anyrrr$warningswarnstrroundrr'r r(sorted enumerate check_insider& check_validr!sumranger%plot_fitr)r.TrCr)distdist_ind dist_ind_maskclose_cclose_Tr:new_cnew_T valid_point close_alphar fit_alphaR2r2)rArBr3r*@sR  >($"$  0Fz(Alpha_T0_empirical.get_alpha_T0_wheightsc Cs||_|jd}tt||jdk||jdk||jdk||jdkgrtdtt |ddtt |ddtt |jddd tt |jddd tt |jddd tt |jddd t t|j |j g |j||g}t|| gS) a :param comps (str) : comp1,comp2 following Thermopack convention :param c (float) : Volumetric concentration at which to get kinetic_gas :param T (float) : Temperature (K) at which to get kinetic_gas :param plot (bool or '2d' or '3d') : display fitted data :return float : kinetic_gas at the specified conditions, approximated from experimental data by CloughTocher-interpolation r5rr r6r7z c = z T = z is outside range c : (z, z), T : r8)rCr)rDrrr$rErFrGrHinterpolrr transposer!)r.rPrCr)rZr2r2r3r-s  >&z,Alpha_T0_empirical.get_alpha_T0_CloughTocherc s||_|jd}tt||jdk||jdk||jdk||jdkgrtdtt |ddtt |ddtt |jddd tt |jddd tt |jddd tt |jddd |j ||j d|j ||j d}td d tt|dddD}tdd |D}d}tg}xt|dkrF||jkrFtdd |D}d||<t|j ||g}t|j ||g} td|d} |j || } |j || } || |j | |j ||j | |j ||j ||j } x\| |jdkrn| sn| d7} |j || } |j || } || | || ||rd} PqW| r~d|| <|j ||}|j ||} x| |jdkr||jkr| d7} |j || } |j || } t|| gf}t| | gf}|||||rd|| <PqW|j ||}|j ||} |d7}qNW|j||}t|dkrt|| |g}t|d|d|d|d}t||d|d||d||d}nR|||tfdd ttDfdd ttD}t||}t||d| |d||dt|}|rn|||| ||||||dkrtdtt |ddtt |dd|jdtt |dd t|| gS)a :param comps (str) : comp1,comp2 following Thermopack convention :param c (float) : Volumetric concentration at which to get kinetic_gas :param T (float) : Temperature (K) at which to get kinetic_gas :param plot (bool or '2d' or '3d') : display fitted data :return float : kinetic_gas at the specified conditions, approximated from experimental data by fitting a plane to the closest three data points that enclose the desired point r5rr r6r7z c = z T = z is outside range c : (z, z), T : r8cSsg|] \}}|qSr2r2)r9r:r;r2r2r3r<sz9Alpha_T0_empirical.get_alpha_T0_plane..cSs|dS)Nr r2)r=r2r2r3r>sz7Alpha_T0_empirical.get_alpha_T0_plane..)r?cSsg|]}dqS)Fr2)r9r:r2r2r3r<scSsg|]}dqS)Fr2)r9r:r2r2r3r<sTcs4g|],}td|t|ddqS)Nr )rr@)r9r:)rAr2r3r<scs8g|]0}td|t|ddqS)Nr )rr@)r9r:)rArBr2r3r<sz Alpha_T0 at c = z for z may be a bad fit (R^2 = )rCr)rDrrr$rErFrGrHrr'r r(rIrJr%r& SkipCounterrL concatenaterKr!r]crossdotrMrNrOr)r.rPrCr)rQrRrSnrTrUr:rVrWrXZbox_cZbox_TrYZ close_pointsZ normal_vecrZrr[r2)rArBr3r,sv  >($ $$2  0 Fz%Alpha_T0_empirical.get_alpha_T0_planec Csdt|dkrdSt||||g}tj|tdd|Dfdd}ddtt|D}x|ddD]}t|d ||d d krx|d|||ddD]@}t|d ||d d krt||||d d krd SqWqpxf|d|||ddD]F}t|d ||d d krt||||d d krd SqWqpWdS) a :param x (ndarray) : list of x-points :param y (ndarray) : list of y-points :param c (float) : x-point :param T (float) : y-point :return bool : Is the point (c,T) inside the polygon spanned by (x1,y1),(x2,y2),... r7FcSsg|]}dqS)rr2)r9r:r2r2r3r<sz3Alpha_T0_empirical.check_inside..r )axiscSsg|]}|qSr2r2)r9r:r2r2r3r<sNrT)r%rrr]r_vstackrNr`) r.xyr)rPvecsindicesr:jr2r2r3rKs ""8 "<zAlpha_T0_empirical.check_insidec Csdt||||g}t||||g}x*t||D]\} } t| | dkr@dSq@WdS)a :param new_x (float) : x-point :param new_y (float) : y-point :param x_list (ndarray) : list of x-points :param y_list (ndarray) : list of y-points :param c (float) : x-value at point that is being approximated :param T (float) : y-value at point that is being approximated :return bool : If the point (new_x,new_y) will improve the approximated value at (c, T) rFT)rrr]zipra) r.Znew_xZnew_yx_listZy_listr)rPrhZnew_point_vecsvecZ new_point_vecr2r2r3rLs zAlpha_T0_empirical.check_validc std_tjjdks*jdkrtjjdjdfddt t |Ddtj||d d d d tj||d ddt dt dt dtt|ddtt|dtdt|dddt|ddtjdksjdkrt} | jddd} | jjdjdjddd d| j|||dd| j|||d d| d| d| dtd tt|dtjd!krt d"dS)#a Plots the fitted data, good for debugging and to view errors. Run __call__ with plot=True, plot='2d' or plot='3d' to activate NB: Does NOT display the plot. plt.show() must be called after function call to display Zplasma2dTr r cs g|]}j|qSr2)cmapr!)r9r:) max_alphar.r2r3r<;sz/Alpha_T0_empirical.plot_fit..)colorZgreenrfzInterpolation points)rqmarkerlabelblackzPoint to approximate)rqrszc [vol%]zT [K]zplots/selected/r6r;rNr73do) projectionr Zblue)rqrrredz$\alpha_T^\circ$ [-]z$R^2 = $)rnruTzArgument 'plot' can be: '2d': plot 2d selection of data points '3d': plot 3d fit and selection of data points True: plot both the above)cmget_cmapror#r!rCpltscatterrrNr%xlabelylabelprintrGrHsavefigclosefigure add_subplot set_xlabel set_ylabel set_zlabeltitleshow) r.rQrTrUrYr)rPrZr[figaxr2)rpr.r3rO0s0  4  (.&    zAlpha_T0_empirical.plot_fitcCs,|j|jd|jd|jddddddS) z~ scatter experimental data points in 3d :param ax: A_matr matplotlib.Axes3D instance on which to plot r r r rxrf()rqrrsN)r|r)r.rr2r2r3 plot_pointsZszAlpha_T0_empirical.plot_pointsc s<tdj\}}}}t||ddt||dt\t}|jddd} ||rt fddt t D}|j d |d d nbt fd dt t D} jjkrt| j| d } |jd | d ddtdj|d|d|ddS)z Plot the data-set selected upon initialization and the interpolation with the method selected upon initialization. :param dim_1d: Set to True if data is 1d ignoreg{Gz?rvru)rwcs$g|]}||dqS)r)r+)r9r:) T_list_1d c_list_1dr.r2r3r<qsz0Alpha_T0_empirical.plot_mesh..r5rt)rqcs,g|]$fddttDqS)cs,g|]$}|f|fdqS)r)r+)r9r:)r rrjr.r2r3r<usz;Alpha_T0_empirical.plot_mesh...)rNr%)r9)r rr.)rjr3r<usrg?)rqalphaz)Approximated $\alpha_T^\circ$ values for zc [%vol]zT [K]u$\alpha_T^\circ$ [–]N)rEfilterwarningsr$rlinspacemeshgridr{rrrrrNr%rCr+r-rshaper]Zplot_wireframerrrrr) r.Zdim_1dmin_cmax_cmin_Tmax_TrrZ alpha_vals_1dZ alpha_valsr2)r rrrr.r3 plot_meshas*  $    zAlpha_T0_empirical.plot_meshN)F)F)F)F) __name__ __module__ __qualname__r4r*r-r,rKrLrOrrr2r2r2r3r s5 K  `*rc@s0eZdZddZddZeZddZddZd S) r^cCs$||kr|d|_n||_||_dS)Nr )valskip)r.rrr2r2r3r4s zSkipCounter.__init__cCs6|j||jkr$|j|d7_n|j|7_|S)Nr )rr)r.otherr2r2r3__lADD__szSkipCounter.__lADD__cCs |j|kS)N)r)r.rr2r2r3__lt__szSkipCounter.__lt__cCs|jS)N)r)r.r2r2r3 __index__szSkipCounter.__index__N)rrrr4r__add__rrr2r2r2r3r^s r^csTg}g}g}g}g}xVtdddD]B}x:tdddD]&}tddddg}tddddg}||d||dtt } | t| } tfddttDfddttD} |d|} |tt|t |t| | |d||t| | |d||t| t| |d|q>Wq&Wt j |t t|dd t j |t t|d d t t j |t t|dd t j |t t|d d t t } | jd d d } | j ||t t|dd | j ||t t|d d | d| dt dS)z~ Testing procedure for different wheighted averages based on distance from point to different experimental points g{Gz?g?r6cs4g|],}td|t|ddqS)Nr )rr@)r9r:)rAr2r3r<sz!test_wheights..cs8g|]0}td|t|ddqS)Nr )rr@)r9r:)rArBr2r3r<sb)rqrArvru)rwr[Zr_minN)rrrexpsqrtrMrNr%appendr"r{r|absrrrrr)ZR2_listmin_rZ miss1_listZ miss2_listZ miss3_listZx_pZy_prfrgfw1w2rrr2)rArBr3 test_wheightssB  *  rc@s$eZdZddZddZddZdS) DB_BuildercCs ||_dS)N)filename)r.rr2r2r3r4szDB_Builder.__init__c Csft|j}|j|d|k|d|k@}t|d}t|d}t|d}t|d}||||fS)Nrrr r )rrrrr"r#) r.rrr1rrrrrr2r2r3 get_cT_ranges     zDB_Builder.get_cT_rangec CsZt|j}|d}|d}tddt||D}i}x>|D]6\}}||krf||d|q>d|gi||<q>Wx|D]}tt ||d||d<tt ||d||d<tt ||d||d<tt ||d||d <xrt ||dD]^\} }| ||\} } } } | ||d| <| ||d| <| ||d| <| ||d | <qWqWt |d }| d x|D]}xt ||dD]\} }| |d |d tt||d| d d tt||d| d d t||d| d t||d | dqW| dqWWdQRXdS)Nrrcss|]\}}||fVqdS)Nr2)r9c1c2r2r2r3 sz&DB_Builder.build_db..ZMixrrrrwz)Comp1, Comp2, min_c, max_c, min_T, max_T z, r )rrrsetrkkeysrrzerosr%rJropenwriterGrH)r.Zoutfiler1rrZ comp_tuplesZ comp_dictrrr:rrrrfiler2r2r3build_dbs6    zDB_Builder.build_dbN)rrrr4rrr2r2r2r3rs r)pandasrmatplotlib.pyplotpyplotr{ matplotlib.cmryZmpl_toolkits.mplot3drnumpyrrErscipy.interpolaterr\rr^rrr2r2r2r3s    |1