B _@sddlmZddlmZddlZddlmZddl m Z ddl Z ddl mZddlmZdZedZedZddZd d Zd d Zd dZdddZdS)) KineticGas)cubicNz6/home/ubuntu/Home/Documents/7_semester/irrev_prosjekt/zdata/zplots/alpha_T0/cCst|||t|S)N)npexp)nabr [/Users/vegardjervell/Documents/7_semester/irrev_prosjekt_renskriv/plotting/plot_alpha_t0.pyfit_funcsr c CstddtddD}tt|df}tt|}tjdd}tjdd|d d }d dtdD}| |d |d <x,tddD]}|j |||d d ||<qWx&tdD]}tj || ddqWt }| |dx`t|D]T\}} td| dt} t||ddg| d} t| ||<| j| j| jg||<qW|}dddg} xFtdD]:}t||tj|||ddtj| |ddqdWt|dtj||ddtjdddtjdddtjd d!d"dS)#NcSsg|]}|qSr r ).0ir r r sz$plot_convergence.. ) )figsizeg?)ncolsnrowsfigurehspacecSsg|]}dqS)Nr )r r r r r rsr)sharexF)visibleSRK Computingzorder approximationg?g?) mole_fracsNz$d_{-1}$z$d_0$z$d_1$black)color)fontsizezTime [s]zOrder of approximationZ convergence2iX)dpi)rarrayrangezeroslenpltrgsGridSpec add_subplotsetpget_xticklabelsrinit enumerateprinttime process_timerd_1d0d1 transposescaplotylabelxlabelsavefig) compsn_listd_list time_listfiggridaxsr eosrt0 analytical label_listr r r plot_convergences<    rIc Cs0tddtddD}tt|}t}||dxLt|D]@\}}td|dt }t ||dd g|d }t |||<qFWtd t j t||d d \}} t||d} tt|t|d} tj| t| |d|ddddtj||ddtj| t| | dddtd|tddS)NcSsg|]}|qSr r )r r r r r r<sz!plot_time_fit..rrzorder approximationg?g?)rrZFitting)rr)p02rrz--r ) linestyler!)r!red:)r!rOzexp :Zruntime)rr&r'r(r)rr0r1r2r3r4ropt curve_fitr polyfitlinspaceminmaxr*r:polyvalr=) r>r?rArEr rrFrGfit_polyZn_liner r r plot_time_fit;s$  " r\cCs td}dS)Nz 298_files/) plots_path) save_pathr r r files_298Tsr_*Fc Cs|d\}}t|d|}tddd}t}||dtt|} x:t|D].\} } t ||| d| g|d} | j || | <qXWt j || |d t d |||rt jd d t j|d ddS)a> Plot soret coefficient from kinetic gas theory for N'th order approximation with different m2/m1 and sigma2/sigma1 ratios :param N (int): Order of approximation :param compare (bool): Use same limits as Tipton, Tompson and Loyalka :param save (str): Filename to save figure as (defaults to False) ,rZg?g?rNrr)rr)labelz Plotted for :r)titleiX)r%N)splitr]rrUrr0r(r)r1rsoretr*r:r2legendr=) r>rTsavecomp1comp2r^x_listrE alpha_listr xalpha_T0r r r plot_testWs   ro)r`F)models.kineticgasrpycThermopack.pyctprnumpyrmatplotlib.pyplotpyplotr*matplotlib.gridspecgridspecr+r3scipy.optimizeoptimizerRZ matplotlib.cmcm root_path data_pathr]r rIr\r_ror r r r s      *