Zh ddlmZddlmZddlmZddlmZddlmZddlmZddlmZddl m Z dd l m Z dd lmZdd lmZdd lmZdd lmZdd lZGddeZy ))Frame) Instruction) Integrator)schemes)Field)Group)IntVar)colorizeN)std) hdf5writer)Boundary)SimpleNamespacec~eZdZdZdZfdZfdZedZdZdZ dZ d Z d Z d Z d Zd ZdZddZxZS) SimulationaThe main simulation class for running dust coagulation simulations. `dustpy.Simulation`` is a child of ``simframe.Frame``. For setting simple initial conditions use ``Simulation.ini``, For making the simulation grids use ``Simulation.makegrids()``, For initialization use ``Simulation.initialize()``, For running simulations use ``Simulation.run()``. Please have a look at the documentation of ``simframe`` for further details.DustPyc *t|d?i|td?itd?iddddddddd d td?id d tjzd tj zddtj zdtd?idddddtj zdtj zdtd?idtjzdtjzddd|_t|d|_ d|j_ t|d|j_ d|jj_ d|jj_dd g|jj_t|d!|j_d|jj _d|jj _t|d"|j_d|jj&_d|jj&_d|jj&_ d|jj&_d|jj&_d|jj&_d|jj&_d|j_t|d#|j_d|jj6_d|jj6_d|jj6_gd$|jj6_d|j_t|d%|j_d|jj>_ d|jj>_!d|jj>_"gd&|jj>_d|j_#d|j_$d|j_%d|j_&d|j_'t|d'|j_(d|jjP_)d|jjP_d(d)g|jjP_t|d*|j_*d|jjT_+d|jjT_,d|jjT_-d|jjT_"d+d,g|jjT_d|j_.d|j_/d|j_0t|d-|j_1d|jjb_)t|d.|jjb_2d|jjbjd_3d|jjbjd_4d|jjbjd_d|jjbjd_"d|jjbjd_d|jjbjd_gd/|jjbjd_d|jjb_5d|jjb_gd0|jjb_gd1|j_t|d2|_6d|jl_7t|d!|jl_d|jlj _d|jlj _d|jl_8d|jl_9d|jl_d|jl_:d|jl_;d|jl_<d|jl_=d|jl_>d|jl_?d|jl_&t|d3|jl_*d|jljT_,d|jljT_-d|jljT_"d+d,g|jljT_d|jl_.d|jl_/d|jl_@t|d4|jl_Ad|jlj_Bd|jlj_1d5d6g|jlj_t|d-|jl_1d|jljb_d|jljb_Cd7d8g|jljb_gd9|jl_t|d:|_Dd|j_Ed|j_Fd|j_Gd|j_ d|j_Hd|j_Id|j_Jd;g|j_t|d<|_Kd|j_Ld|j_Md|j_Nd|j_@gd=|j_gd>|_d|_Oy)@zMain simulation class.g-C6?F$@g{Gz?g ?gUUUUUUgQ?gY@) aIniMaxallowDriftingParticleserosionMassRatiod2gRatiodistExp excavatedMassfragmentDistribution rhoMonomervFraggMbP?g?gffffff@ggN@)alphaMdiskmuSigmaExpSigmaRcg-q=j@d@@)NmbpdmminmmaxNrrminrmaxg@g@)MRT)dustgasgridstarzDust quantities descriptionNzBackreaction coefficientsABzBoundary conditionszCoagulation quantitieszMixing parameters)radturbvertFluxes)advdifftot ProbabilitiesfragstickSourcesextr> VelocitieszRelative velocities)azibrownr8r9r:r>r@driftmaxreldeltarhosfillaStHrho backreactionvDepskernelpSzGas quantitiesz Source termszTorque parametersLambdarSviscr8) r r/rcsHpnurQnmfpPetatorquerXzGrid quantitiesOmegaKzStellar quantities)r-r.r/L)r3r2r1r0)Psuper__init__rcM_sunm_pauR_sun_inirr0rNrRr6r7updaterboundaryinnerouter coagulationrA stick_indrUlf_indrm_indphirTrKr8r9r:Fir<r=r>rMrPrVrQrLrWr@rXcoagrChydSigma SigmaFloorrOrSrIrErFrHr1rr[rar\r_r r^r]r`r/rbrYrZr2r*rriNmmrcr3rdr-r.t)selfkwargs __class__s `/mounts/lovelace/software/anaconda3/envs/py312/lib/python3.12/site-packages/dustpy/simulation.pyrgzSimulation.__init__#s "6"$ & 0AV\aVYNSMQSUZ_PTKO BD 0A/>/@%JNqww,GJ155yMPLOPQPTPTH AC/@0?0A1JPJNHKJLQTT'JOPQPTPT* BD0A0?0Ar!''zGI!''zGLBD0A5') & J$,=>  !& 9"; #'  #'  *-s &"45JK #'  #'  % 6!8 &* #*. '"& $( !'+ $'+ $$( ! 2EF " # # "9  Tx8     5      Do>    %w/ Di8     $en  #  Dl;  2GH "  $ " " # # #: #  9 Y +;<!$4IJ"&"&          4^<    #U^ " 2EF!% #+S/4\:   $e_ J$,=>        %J $,@A     0 7 c`|jdkDrd}|dj|jz }|dz }|djdz }|djdz }|d jd z }|dz }|td d z }t ||j t |y )z#This functions runs the simulation.rz DustPy v{} z Documentation: {}z"https://stammler.github.io/dustpy/z PyPI: {}z https://pypi.org/project/dustpy/z GitHub: {}z#https://github.com/stammler/dustpy/z) Please cite Stammler & Birnstiel (2022).blueN) verbosityformat __version__r printcheckmassconservationrfrun)rmsgrs rrzSimulation.runs >>A C >(()9)9: :C 4KC (//46 6C (//24 4C (//57 7C 4KC 8H&Q QC #J ""$  rc|jS)z1Parameter set for setting the initial conditions.)rmrs rinizSimulation.inisyyrcD|j|jy)aFunction creates radial and mass grid. Notes ----- The grids are set up with the parameters given in ``Simulation.ini``. If you want to have a custom radial grid you have to set the array of grid cell interfaces ``Simulation.grid.ri``, before calling ``Simulation.makegrids()``. You cannot have a customized mass grid, because the coagulation alorithm strictly needs a logarithmic grid. Always set your mass grid parameters with ``Simulation.ini``.N) _makemassgrid_makeradialgridrs r makegridszSimulation.makegridss  rctj|jjj}tj|jjj }tj ||z }t||jjjzdz}tjtj|jjjtj|jjj |d}t||dd|j_ t||dd|j_ y) zFunction sets the radial mass grid. If ``Simulation.grid.rint`` is not given it uses the parameters set in ``Simulation.ini``.rnumbasez# of mass binsTr5constantz Mass grid [g]N) nplog10rr2r(r)ceilintr'logspacerr~r)rlogmminlogmmaxdecadesr~rs rrzSimulation._makemassgrids((488==--.((488==--.'''G+, 488==... /! 3 KK!3!34bhh HHMM  7 %'c 3 ""2TC  !4A rcJ|jjtjtj|j jj tj|j jj|j jjdzd}|j jj}n(|jj}|jddz }d|dd|ddzz}tj|dddz|dddzz z}t||d d |j_t||d d |j_ t||d d |j_t||dd |j_ y)zKFunction sets the mass grid using the parameters set in ``Simulation.ini``.Nrrrrg?z# of radial grid cellsTrzRadial grid cell centers [cm]z Radial grid cell interfaces [cm]uRadial grid annulus area [cm²])r2r}rrrrr+r,r*shapepirr|r6)rr}r*r|r6s rrzSimulation._makeradialgridsK 99<< RXXdhhmm&8&89288 ""<$)-)9)9!);#GB!!BB!qB CR2ab6! " EE2ab619r#2wz) * "":TK  !!@4Q  ""DtU  !!BTS rc |jjj td|jjj td|jjj td|jjj td|jjj td|jjj td|jjj td|jj td |jd kDr d }d }|td dz }t|tdd}t|tj j#|jjj|jjj|jj\}}}||f}d}||krd}dj%|}dj%t||}|dz }|dj%|d |jj|d z }|dj%|d|jj|dz }t|}t|tdd}t||jjj } |jjj } |jjj} |jjj} |jj} |jjj}tj j'| | | |\}}}||f}d}||krd}dj%|}dj%t||}|dz }|dj%|d |jj|d z }|dj%|d|jj|dz }t|}t|tdd}t|tj j)| | | | | |\}}}||f}d}||krd}dj%|}dj%t||}|dz }|dj%|d |jj|d z }|dj%|d|jj|dz }t|}t|yy)zQFunction checks for mass conservation and prints the maximum relative mass error.Nz/'Simulation.dust.coagulation.stick' is not set.z3'Simulation.dust.coagulation.stick_ind' is not set.z+'Simulation.dust.coagulation.A' is not set.z-'Simulation.dust.coagulation.eps' is not set.z0'Simulation.dust.coagulation.lf_ind' is not set.z0'Simulation.dust.coagulation.rm_ind' is not set.z-'Simulation.dust.coagulation.phi' is not set.z'sim.grid.m' is not set.rgvIh%<=rz"Checking for mass conservation... yellow)colorz - Sticking:redgreenz{:9.2e}z max. rel. error: {:} z for particle collision z( m[{:d}] = {:9.2e} g with z m[{:d}] = {:9.2e} grz - Full fragmentation:z - Erosion:z m[{:d}] = {:9.2e} g )r0rrrA RuntimeErrorrsr6rUrtrurvr2rrr rr dust_f check_mass_conservation_stickingr*check_mass_conservation_full_fragmentationcheck_mass_conservation_erosion)rerraccrerrmaxijtuprerrorr6rUklfkrmrrvs rrz Simulation.checkmassconservations 99 & & .AC C 99 * * 2EG G 99 " " *=? ? 99 $ $ ,?A A 99 ' ' /BD D 99 ' ' /BD D 99 $ $ ,?A A 99;; 9: : >>A FC 8A"*, ,C #J,H=C #J::FF %%++TYY-B-B-L-Ldiikk[LFAqa&CE$$V,E299e,.C 5 5C >EEA CF+- -C 4;;A CF+- -C3-C #J6hGC #J %%''A))''++C))''..C))''..C A))''++C::PP33 LFAqa&CE$$V,E299e,.C 5 5C >EEA CF+- -C 4;;A CF+- -C3-C #J+8EEA CF+- -C 6==A CF+- -C3-C #JS rct|jjtr$t|jjts|j |j t|dd|_d|j _tjj|j _ tj|j tjddd t j"zg|j _|j'|j)|j+|j-|j.@t1tj2j4|j2j6d |j2j8id t1tj:j4|j:j6|j:j<|j:j>j@d d g}tC|j d|_||j._"tjjF|j._$tjjJ|j._&|jNtQ|_'|j.jS|jUy)zFunction initializes the simulation frame. Function sets all fields that are None with a standard value. If the grids are not set, it will call ``Simulation.makegrids()`` first.NzTime [s]r4皙?r&r$)rrhs&Dust: implicit 1st-order direct solver controllerr5)roSextz%Gas: implicit 1st-order direct solverzDefault integrator)+ isinstancer2r~rr*rrr cflr simdtrnrhstack geomspacerhyear snapshots_initializestar_initializegrid_initializegas_initializedust integratorrr0 impl_1_directrz_rhsr1rorXrCr instructionsprepare_implicit_dust preparatorfinalize_implicit_dust finalizerwriterr _finalizeupdate)rrs r initializezSimulation.initialize{s $)),,.ju6U NN  66>D"*=DFDFFJ WWZZDFFN!yydDb9!&&@A DFF      ?? "CHH22 IIOO(-tyy~~(*(P  CGG11 HHNN,0HH,=,=(, ()P L )$8:DO+7DOO (),)F)FDOO &(+(F(FDOO % ;; $,DK !!# rc /t|jj}t|jjt|jjf}t|jj|jjz}t|jjdzt|jjf}t|jjt|jjt|jjf}|jj ct |tj|d|j_tjj |jj _ tjj|\}}}} } } } |jjj(t ||dd|jj_ |jjj(t || dd|jj_|jjj(t || d d|jj_|jjj (t || d d|jj_|jjj"(t || d d|jj_|jjj$(t ||d d|jj_|jjj&(t ||d d|jj_|jj(ct |tj|d|j_tjj(|jj(_ |jj*j,]|j.j0j2tj4|z} t || d|jj*_|jj*j6]|j.j0j2tj4|z} t || d|jj*_|jj*j8]|j.j0j2tj4|z} t || d|jj*_|jjct |tj|d|j_tjj|jj_ |jj:j<wt |tj|d|jj:_tjj>|jj:j<_ |jj:j@wt |tj|d|jj:_ tjjB|jj:j@_ |jj:jDwt |tj|d|jj:_"tjjF|jj:jD_ |jjH0t |tj4|d|j_$|jjJct |tj|d|j_%tjjJ|jjJ_ |jjLct |tj|d|j_&tjjL|jjL_ |jjNct |tj|d|j_'tjjP|jjN_ |jjRS|j.jjTtj4|z}t ||d|j_)|jjVjXwt |tj|d|jjV_,tjjZ|jjVjX_ |jjVj$wt |tj|d|jjV_tjj\|jjVj$_ |jj^j`wt |tj|d|jj^_0tjjb|jj^j`_ |jj^jd:t |tj|d|jj^_2|jj^jfwt |tj|d|jj^_3tjjh|jj^jf_ |jj^jDwt |tj|d |jj^_"tjjj|jj^jD_ |jjlct |tj|d!|j_6tjjn|jjl_ |jjpjX]|j.jjrtj4|z}t ||d"|jjp_,|jjpjtjvt |tj|d#|jjpjt_;tjjx|jjpjtjv_ |jjpjtjzt |tj|d$|jjpjt_=tjj||jjpjtjz_ |jjpjtj,t |tj|d%|jjpjt_tjj~|jjpjtj,_ |jjpjtj6t |tj|d&|jjpjt_tjj|jjpjtj6_ |jjpjtj8t |tj|d'|jjpjt_tjj|jjpjtj8_ |jjpjtjDt |tj|d(|jjpjt_"tjj|jjpjtjD_ |jjpjwt |tj|d)|jjp_Ctjj|jjpj_ |jjpj,wt |tj|d*|jjp_tjj|jjpj,_ |jj|jj?d+tjj|z}t ||d,|j_G|jjtjj|}tj||jjkd+|jjz|}t ||d-|j_Htjj|jj_Ltjj|jj_N|jjt ||jjd.d/|j_Ot |tj|d0|j_P|jjjdt|jj|jj|jjd12|jj_R|jjjt|jjddd3|jjddd3|jjddd3d4d+|jjd3z5|jj_V|jjjj|jjjjtjj||jj|jjd6<y#YxYw)7z&Function to initialize dust quantitiesrNzParticle size [cm]r4zFragment normalization factorsTrzRemnant mass distributionzIndex of largest fragmentzSmaller index of remnantzFragment distributionzSticking matrixz$Non-zero elements of sticking matrixuDiffusivity [cm²/s]zRadial mixing parameterzTurbulent mixing parameterzVertical mixing parameterzDust-to-gas ratiozAdvective flux [g/cm/s]zDiffusive flux [g/cm/s]zTotal flux [g/cm/s]zFilling factorzScale heights [cm]uCollision kernel [cm²/s]u+Midplane mass density per mass bin [g/cm³]uSolid state density [g/cm³]zFragmentation probabilityzSticking probabilityuCoagulation sources [g/cm²/s]External sources [g/cm²/s]Hydrodynamic sources [g/cm²/s]uTot sources [g/cm²/s]z Stokes numberzFragmentation velocity [cm/s]z"Relative azimuthal velocity [cm/s]z(Relative Brownian motion velocity [cm/s]zRelative radial velocity [cm/s]z"Relative turbulent velocity [cm/s]z*Relative vertical settling velocity [cm/s]zTotal relative velocity [cm/s]zMaximum drift velocity [cm/s]Radial velocity [cm/s]r'Floor value of surface density [g/cm²]u%Surface density per mass bin [g/cm²]*Previous value of surface density [g/cm²]r5copy+Right-hand side of matrix equation [g/cm²] const_grad) conditionrval)rvalue.)Yrr2r*r~r0rNrrzerosr rncoagulation_parametersrrr6rUrtrurvrArsrTrKr8rr1ronesr9r:rwr<F_advr=F_diffr>F_totrMrPrVrQ rho_midplanerLrrWr@p_fragp_stickrXrxS_coagrCryS_hydS_totrOSt_Epstein_StokesIrSrrIrEvrel_azimuthal_driftrFvrel_brownian_motionvrel_radial_driftvrel_turbulent_motionvrel_vertical_settlingvrel_totrH vdriftmaxvradrr{rzMRN_distributionwhere Sigma_derivdifferentiatorjacobian jacobinator _SigmaOldrrorpr r|r}rq setboundaryenforce_floor_value)rshape1shape2 shape2ravelshape2p1shape3rArsr6rUrtrurvrKrLrr{rzs rrzSimulation._initializedustsdiill#diill#S%67499<< 45  %a'TYY\\):;diill#S IILL&tyy||,. 99;; bhhv&6,@BDIIK"%((**DIIKK 8;8W8W 95y!S&&# 99 " " *&+a%EPT'VDII ! ! # 99 $ $ ,(-c'BT)SDII ! ! % 99 ' ' /+0f*EPT,VDII ! ! ( 99 ' ' /+0f*Dt,UDII ! ! ( 99 $ $ ,(-c'>)ODII ! ! % 99 & & .*/e):T+KDII ! ! ' 99 * * 2.3i-S^b/dDII ! ! + 99;; bhhv&6,BDDIIK"%((**DIIKK  99??   &HHLL&&8E"'e)B#DDIIOO  99??   'HHLL&&8E#(e)E$GDIIOO 99??   'HHLL&&8E#(e)D$FDIIOO  99== !$)%8:DIIM$'HHLLDIIMM ! 99<<   #$T288,'@ BDIILL '*xx~~DIILL   $ 99<<   $ %dBHH-'@!BDIILL (+DIILL   % 99<<   #$T288,'< >DIILL '*xx~~DIILL   $ 99>> !"4*%57DIIN 99;; bhhv&6,@BDIIK"%((**DIIKK  99   #$T288,%@ BDII '*xxDII   $ 99== !$)%RTDIIM$'HH$9$9DIIMM ! 99>> !88==++bggfo=D"d(FHDIIN 99;;   #$T288,%@ BDIIKK '*xxDIIKK   $ 99;;   $ %dBHH-%;!=DIIKK (+(8(8DIIKK   % 99;;   #$T288,%E GDIIKK '*xxDIIKK   $ 99;;?? "#D"((+%BDDIIKKO 99;;?? "#D"((+%FHDIIKKO&)hhnnDIIKKOO # 99;;?? "#D"((+%=?DIIKKO&)hhnnDIIKKOO # 99<<  rxx(%46DIIL#&88#>#>DIILL 99;;   #HHMM''"''&/9E$e)H JDIIKK  99;;??   &"'bhh/%I#KDIIKKOO *-((*G*GDIIKKOO   ' 99;;?? ($)$1%O%QDIIKKOO !,/HH,I,IDIIKKOO ! ! ) 99;;??   &"'bhh/%F#HDIIKKOO *-((*D*DDIIKKOO   ' 99;;??   '#(rxx0%I$KDIIKKOO +.88+I+IDIIKKOO ( 99;;??   '#(rxx0%Q$SDIIKKOO +.88+J+JDIIKKOO ( 99;;??   &"'bhh/%E#GDIIKKOO *-((*;*;DIIKKOO   ' 99;;   '#(rxx0%D$FDIIKK +.88+=+=DIIKK ( 99;;?? "#D"((+%=?DIIKKO&)hhmmDIIKKOO #  II    99   'SXX0066J#(j.W$YDII  99?? "HH--d3EHHUdii&:&:: !5!55"$E$e)PRDIIO),)=)= &&)hh&7&7 # $ $))///[bfh tRXX &&SU  99   # # +'/   & (DII   $ 99   # # +'/ DbD! TrT" "%$))..r22 (DII   $   ,,.   ,,. $$T*#'99?? C a  sOA^^A^ ct|jj}t|jjdz}|jjS|j jjt j|z}t||d|j_|jjct|t j|d|j_ tjj|jj_ |jjct|t j|d|j_tjj|jj_ |jj ct|t j|d|j_tjj |jj _ |jj"ct|t j|d|j_tjj"|jj"_ |jj$ct|t j|d |j_tjj&|jj$_ |jj(S|j jj(t j|z}t||d |j_|jj*ct|t j|d |j_tjj,|jj*_ |jj.ct|t j|d |j_tjj.|jj._ |jj0ct|t j|d |j_tjj2|jj0_ |jj4ct|t j|d|j_tjj6|jj4_ |jj8j:wt|t j|d|jj8_tjj<|jj8j:_ |jj8j>:t|t j|d|jj8_|jj8j@wt|t j|d|jj8_ tjjB|jj8j@_ |jjD3t|dt j|zd|j_"|jjFt jHtjjK|jjL|j jjN|j jjP|j jjR}t jT||jjD}t||d|j_#tjjV|jjF_,|jjZct|t j|d|j_-tjj\|jjZ_ |jj^j`:t|t j|d|jj^_0|jj^jbwt|t j|d|jj^_1tjjd|jj^jb_ |jjbjfwt|t j|d|jjb_3tjjh|jjbjf_ |jjbjjwt|t j|d|jjb_5tjjl|jjbjj_ t|t jn|jjFdd|j_8t|t j|d|j_9|jtjvjx:t|t j|d|jtjv_<|jtjvjz:t|t j|d|jtjv_=|jj}|jj~jt|jjL|jj|jjF|jj~_@|jj~jjd |jj~jj|jj~jt|jjLddd!|jjddd!|jjFddd!|jj~_E|jj~jjd"|jjDd!|jj~jj|jjF|jjpd#<y)$z%Function to initialize gas quantitiesrNzTurbulent alpha parameterr4zIsothermal sound speed [cm/s]zPressure gradient parameterzGas flux interfaces [g/cm/s]zPressure scale height [cm]zMidplane mean free path [cm]zMean molecular weight [g]uMiplane number density [1/cm³]uKinematic viscosity [cm²/s]uMidplane pressure [g/cm/s²]uMiplane mass density [g/cm³]rruTotal sources [g/cm²/s]g0.++ruSurface density [g/cm²]zTemperature [K]u=Specific angular momentum injection rate by torque [cm²/s²]z+Effective velocity imposed by torque [cm/s]z!Viscous accretion velocity [cm/s]rrTrrz Pull factorz Push factorrrr.)Frr2r*r1rrrrrr[rr cs_isothermalrnra eta_midplanerwr\r_ mfp_midplaner r^ n_midplaner]r` P_midplanerQrrXryrrCr>rr{rzarraylyndenbellpringle1974r|r"r!rmaximumrrr/ T_passiverbrYrSvtorquerZvviscr8r zeros_likerrr0rRr6r7rrorpr r} setconditionrrq)rr shape1p1rr SigmaGass rrzSimulation._initializegassdiill# %a' 88>> !HHLL&&8E"e)DFDHHN 88;; bhhv&6,KMDHHK"%''"7"7DHHKK  88<<  rxx(%BDDHHL#&77#7#7DHHLL 88;; bhhx&8,JLDHHK"%''**DHHKK  88;; bhhv&6,HJDHHK"%''**DHHKK  88<<  rxx'7-KMDHHL#&77#7#7DHHLL 88;; 2776?2Bb,GIDHHK 88:: tRXXf%5+LNDHHJ!$!3!3DHHJJ  88;; bhhv&6,JLDHHK"%''**DHHKK  88:: tRXXf%5+IKDHHJ!$!3!3DHHJJ  88<<  rxx'7-LNDHHL#&77#7#7DHHLL 88::>> !"4*%FHDHHJJN%(WW]]DHHJJNN " 88::>> !"4*%BDDHHJJN 88::>> !"4*%?ADHHJJN%(WW]]DHHJJNN " 88   &"'gbggfo-;d#fDHH  88>> !xx = = TXX\\11488<<3H3H$((,,J\J\!^_Hzz(DHH,?,?@H"4/IKDHHN%(WW%5%5" 88:: tRXXf%5+<>DHHJ!$!2!2DHHJJ  88?? ! ! )%*4&1A7v&xDHHOO " 88??   $ %dBHHV,<2_!aDHHOO (+DHHOO   % 88::?? "#D"((6*:0SUDHHJJO&)ggmmDHHJJOO # 88::>> !"4&)9/GIDHHJJN%(WW\\DHHJJNN "# "--/=iptvdBHH %!NP 99 ! ! # # +',bggfo=(BDII " " $ 99 ! ! # # +',bhhv&M(CDII " " $  88   " " *&. TYY\\488>>';DHH   # HH   # # 0 0 > ++- 88   " " *&. DbD!499<<"#5txx~~dd7K'MDHH   # HH   # # 0 0txx**2. 0 ++-"&((..3rcht|jj}|jjct |t j |d|j_tjj|jj_|jjy)z&Function to initialize grid quantitiesNzKeplerian frequency [1/s]r4) rr2r*rcrrrr rnr)rr s rrzSimulation._initializegrid spdiill# 99   #$T288F+;1L NDII '*xxDII   $ rc|jjPt|dd|j_tjj|jj_|jj ;t||jjj d|j_|jj;t||jjjd|j_|jj;t||jjjd|j_ |jjy)z-Function to initialize the stellar quantitiesNrzLuminosity [erg/s]r4zMass [g]z Radius [cm]zEffective temperature [K]) r3rdrr luminosityrnr-rr.r/rrs rrzSimulation._initializestar+s 99;; b6JKDIIK"%(("5"5DIIKK  99;; dhhmmoo,68DIIK 99;; dhhmmoo,9;DIIK 99;; dhhmmoo,GIDIIK rcB t|jjtr$t|jjts t dt |jjD])\}}|j|jjus)n|dk(r/t|jj|jjz}t||jjdd|j_ t|tj|d|j_t"jj$|jj_t"j(j*|j,_gd|jj0_gd |j_d d g|jj2_t"j(j4|j_t"j(j8|j_|d k(rlt=t"jj>|jjd |jj id}||jj<n |dk(r tAd|dk(r tAdt d|dk(rtC|jdr |j` tC|jdr |j`d|jj_gd|jj0_gd|j_gd|jj2_t"j(jD|j_t"j(jF|j_|dk(rt"j(jH|j,_t=tJjL|jj|jj2jNdddd }||jj<|j,jQd!tRjTznt d"t d#|jjW|jY|jZd$kDr2d%j]t_|d&t_|d&}ta|yy)'aFunction sets the dust integrator. Parameters ---------- scheme : string, optional, default : "explicit" Possible values {"explicit", "implicit"} method : string, optional, default : "cash-karp" Possible values for explicit integration {"cash-karp"} Possible values for implicit integration {"direct", "gmres", "bicgstab}zNThe simulation frame has to be initialized before calling setdustintegrator().implicitrTrrr4rGrJrCr>directrrrgmresz%GMRES method is not implemented, yet.bicgstabz!BiCGSTAB is not implemented, yet.z(Invalid method for implicit integration.explicitrrN)r@rHr8rI)rKrLrMrNrOrPrQrRrSrTrUrwrVrWrX)rxryrCr> cash-karprg?)dYdxrUrXz2Dust: explicit 5th-order adaptive Cash-Karp methodrz(Invalid method for explicit integration.zUnknown integration scheme.rz5Setting dust integrator scheme: {} method: {}r)1rr2r~rr*r enumeraterrYr0rzrrrrrr rrrrrrnrSrXrrrrrrNotImplementedErrorhasattrprepare_explicit_dustfinalize_explicit_dust dt_adaptiverexpl_5_cash_karp_adptvr>suggestrhrrrrrr r)rschememethodrinstr rs rsetdustintegratorzSimulation.setdustintegrator@s$)),,.ju6U`b b!!=!=> GAtvv(  Z tyy||DIILL89K #(diioo3_fj#lDII #4**WYDIIN+.((*;*;DIIOO '!WWZZDFFN#>DIIKK !]DII #(%.DIIKK *-)F)FDOO &(+(F(FDOO %!"388#9#9#'99??/4diinn/1/W % 37,,Q/7");==:%)*MNN"#MNN z !tyy+.II'tyy&)IIN+/DIIOO '#EDIIKK !cDII "?DIIKK *-)F)FDOO &(+(F(FDOO %$"%!4!4#7#A#A#'99??7;yy{{6947/10d %37,,Q/r!&&y)##MNN<= = !!# >>A KRR((66*BDC #J r)r'r()__name__ __module__ __qualname____doc__rgrpropertyrrrrrrrrrrr6 __classcell__)rs@rrrsl THfP*  AS(eN;zW3rM1^ *wrr)simframerrrrsimframe.framerrr simframe.utils.colorr dustpy.constants constantsrhdustpyr simframe.io.writersr dustpy.utils.boundaryr dustpy.utils.simplenamespacernumpyrrrerrrGs=  !)**8bbr