U _@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|SN)npexp)nabr X/home/ubuntu/Home/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 <tddD]}|j |||d d ||<qtdD]}tj || ddqt }| |dt|D]T\}} td| dt} t||ddg| d} t| ||<| j| j| jg||<q|}dddg} tdD]:}t||tj|||ddtj| |ddqVt|dtj||ddtjdddtjdddtjd d!d"dS)#NcSsg|]}|qSr r .0ir r r sz$plot_convergence.. ) )figsize皙?)ncolsnrowsfigurehspacecSsg|]}dqSrr r r r r rsr)sharexF)visibleSRK Computingorder approximation皙?皙? mole_fracsNz$d_{-1}$z$d_0$z$d_1$blackcolor)fontsizezTime [s]zOrder of approximationZ convergence2Xdpi)rarrayrangezeroslenpltrgsGridSpec add_subplotsetpget_xticklabelsrinit enumerateprinttime process_timerd_1d0d1 transposescaplotylabelxlabelsavefig) compsn_listd_list time_listfiggridaxsreosrt0 analytical label_listr r r plot_convergences<      rSc Cs,tddtddD}tt|}t}||dt|D]@\}}td|dt }t ||dd g|d }t |||<qDtd 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<sz!plot_time_fit..rr r!r"r#r$ZFitting)rr)p02rrz--r') linestyler)r(red:)r)rYzexp :Zruntime)rr0r1r2r3rr:r;r<r=r>ropt curve_fitr polyfitlinspaceminmaxr4rDpolyvalrG) rHrIrKrOrrrPrQfit_polyZn_liner r r plot_time_fit;s$  " rfcCs td}dS)Nz 298_files/) plots_path) save_pathr r r files_298Tsri*Fc Cs|d\}}t|d|}tddd}t}||dtt|} t|D].\} } t ||| d| g|d} | j || | <qVt 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) ,rdrg?rXrrr$)labelz Plotted for :r&)titler-r.N)splitrgrr_rr:r2r3r;rsoretr4rDr<legendrG) rHr&Tsavecomp1comp2rhx_listrO alpha_listrxalpha_T0r r r plot_testWs   ry)rjF)models.kineticgasrpycThermopack.pyctprnumpyrmatplotlib.pyplotpyplotr4matplotlib.gridspecgridspecr5r=scipy.optimizeoptimizer\Z matplotlib.cmcm root_path data_pathrgr rSrfriryr r r r s      *