3
_{A @ s@ d Z dd Zdd Zdd Zdd Zd d
Zd d Zd
d ZdS )a
*********************************************
*** Program for the computation of the ***
*** radiative inhibition ***
*********************************************
This code is part of the LIFELINE program.
Copyright (C) 2020-2021 University of Liege (Belgium)
Enmanuelle Mossoux (STAR Institute)
This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public
License as published by the Free Software Foundation, either version 3 of the License, or any later version.
This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied
warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
details.
You should have received a copy of the GNU General Public License along with this program.
If not, see .
Run: from radiative_inhibition.py import radiative_inhibition
radiative_inhibition_comp(Mdot1, Mdot2, Per, mass_ratio, R1, R2, Teff_1, Teff_2, d, direct, mu, H_mass_frac, beta1, beta2, ipar)
# Parameters
# ==========
Mdoti - The mass-loss rate [Msun/yr]
Per - Orbital period [days]
mass_ratio - M1/M2
Ri - Stellar radius [Rsun]
Teff_i - Effective temperature [K]
d - Separation [cm]
direct - Directory where to save the results
mu - Mean molecular weight. Default value: 0.62 (totally ionized gas)
H_mass_frac - Fractional mass abundance of H. Default value: 0.7381 (Sun)
betai - Parameter of the beta law of the wind velocity for each star. Default value: 1.
ipar - Index of the binary system
# Versions
# ========
v1 - 1/3/2019 - Application of radiative inhibition_models.py for a particular binary.
c C s& dd l }|j|d || d | S )N )numpysqrt)ZcenterxZcenteryradiusxnp r =/local/network/htdocs/Lifeline/codes3/radiative_inhibition.py plot_star+ s r
c C s dd l }| | | d }d| | d| |d }|| | }dd|d d| d| |d }d| d|d | d| |d d d|d d| |d| |d }|||| d }dd| d|
d|
| }d| ||| | d d| |
d|
| d d| d|
d|
| }d | | | d }t||| | | | | }||| || || | | }d| d| ||| | | d| }d| |
d| |
| d| | | d }||j| S )
Nr r g @g ? g @g ?g )mathabsr )M1Gamma1R1M2Gamma2R2grav sound_velduchcudalpha1alpha2r
AcCzfactorK1ZdK1duK2ZdK2duZdAduZBcZdBduZdhduZomegacr r r vz1 s (T T $,0r" c C s d| | d| | |d S )Ng @g ?r g r )zMGammar r r r r uzD s r& c C s dd l } |t| d | j |d }
d|d | d | | | | d|d | | |t| | |
| | } | d S )Nr g @r g ?g @)r
r pi) r urrr dMdotdsigma1r deltarr radgravityr
rho1eqr r r dv_ligne_centreI s Tr0 c C s dd l }dd l}| \}}|j|d |d }||d |j |d }| d |d }|j||j| |j||j| }d| | || |
d| d | ||| || || || |
|d | || | |
|| t||| ||| || |
| }|j||j| |j||j| }d| | || || || || || || | ||| |
|| || t||| ||| || |
| }t||| ||| dkrd| | || |
d| d | ||| || || || |
|d | || | }d| | || || || || || || | ||| |
|| }|j|d |d S )Nr r g @g ?g @g )r r
r r' cossinr ) puthetarr( uthetaturtr) r* thetatheta2r r+
deltathetar rad1rad2grav2r- irr r
r yvherer. soundanglereq1eq2r r r dv_hors_ligne_centreO s (("hXrD c " C s dd l }dd l}| \}}|j|d |d }||d |j |d }| d |d }t||| ||| | }|j||j| |j||j| }| |
|| t||| ||| || |
| d |d | |
}t||| ||| dkrd}d}|d | d d| | |d d| d| |
|d d| | d | || ||| |d | d |
|d ||| |d| | } | d |d | d| | || |
|d d| | d | || ||| |d | d |
|d d| | || || |d| | } d | | || |
d| d | ||| || || || |
|d | || | |
|| t||| ||| || |
| } |j||j| |j||j| }| | | t||| ||| || |
| d |d | |
}t||| ||| dkrd}|d | d d| | |d d| d| || |d d| | d | || ||| |d | d || |d || ||
|d| | } | d |d | d| | || || |d d| | d | || ||| |d | d || |d || || |
|d| | } d | | || || || || || || | ||| |
|| || t||| ||| || |
| }!|j| s|j| s|j| s|j| r|j||j| |j||j| }|d | d d| | |d d| d| |
|d d| | d | || ||| |d | d |
|d ||| } | d |d | d| | || |
|d d| | d | || ||| |d | d |
|d d| | || || } |d | d d| | |d d| d| || |d d| | d | || ||| |d | d || |d || ||
} | d |d | d| | || || |d d| | d | || ||| |d | d || |d || || |
} d | | || |
d| d | ||| || || || |
|d | || | } d | | || || || || || || | ||| |
|| }!|jd| | d|! | d| | d|! | gS )
Nr r g @r g g @g @ g ?) r r
r r' r r1 r2 isnanarray)"r3 r4 r( r5 r6 r) r* r7 r8 r r+ r9 r r: r; r< r- r= r r
r r> r? r. r@ Zconst2rA Zconst1Zeq1xZeq1yrB Zeq2yZeq2xrC r r r jacb s> (P"(L"0(hXrH c t s, dd l }dd l}dd l}ddlm} dd lj}dd lj}dd l }ddl
m }m } dd l
}dd l}dd l} | j } |d} |d} |dd } |dd }!|d
d }"|d }#|d
|d d }$|d
d }%|d}&d|
|" d|$ }'|j}(|jddddddd d d dddddddddddddddg})|jdddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0g}*|jd1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHg}+dIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\dBd]d\d^g},|j|)}-d_|(d` |
da | |db d` }.|.dc| }/|.|/ }0|| }|| d }|| }|| d }d`| |k }1
xt|1D
]}2|2dkr|| }}|| }}|/|0 }0}/| | }} || }}|3|4 }4}3de|jd| |0 | }5de|jd| |/ | }6|5}4|6}3|5dcdf| |
d | |6dcdf| |
d | g}7t|7d( dg |
d dh t||gdi db dj| dk r|5dcdf| |
d | }5|6dcdf| |
d | }6d_|# |( |dg |d d` }8d_|# |( | dg |d d` }9|j|&| |$| }:|"|8 dk |0dg |( | |! |% };|"|9 dk |/dg |( | |! |% }<||$|5 d_ |( |
d d` }=|jt|-| }>|jt|*t|>dl t|>dl dl |= }?|,t|>dl |? }@|+t|>dl |? }A||$|6 d_ |( |
d d` }=|jt|-| }>|jt|*t|>dl t|>dl dl |= }?|,t|>dl |? }B|j|,}Bdcd||
dm| d` dc|B dc|B ||
dm| d` }C||
dm| dkrdcd|dm| d` dc|B dc|B |dm| d` }Cdcddc|@ dc|@ d }D|0dc|; |/dc|< }E|8|D |9|C }Ft|
|0|;| |:}Gtdm| |0|;| |:}Hdcd_|/ dc|< |G|H|G d` |H|0 dc|; }I|dc|@|G|H|G d` |Fdc|@ |E }Jd| |0 dc|; |:d` }K|K |H }Lt |0|;||/|<|| |:|
|H|I|G|@|B}M|jd|& | |$ }N|M |L}O|
d }Pd|
| |L |P }Q g}Rdng}S|Og}T|Og}Udng}V g}Wdng}Xxtt|Qd D ]}Y|O|P }O|
|O }Zdcd||O d` dc|@ dc|@ ||O d` }[|8d_|j |Od` }\|9d_|( |Zd` }]dn}^t|Z|krdcdc||Z d` dc|B dc|B ||Z d` }^dk|'dc|@ |A |\|[ |]|^ |!|N|@ }_| |/ dc|< |Zd` | |0 dc|; |Od` }`|t |5fdo |O|J|:|P|@|_|`fdp}a|aj |Rj |Wj |Sj dn |Xj dn |Tj |O |Uj |O |Vj dn qBW tt |T}Q|j|T}Tdn}b|(dq }c|(|c }d|Tdd }Odcd||O d` dc|@ dc|@ ||O d` }[|8d_|( |Od` }\dk|'dc|@ |A |\ |[ |!|N|@ }_| |0 dc|; |Od` }`x\tt|dD ]J}e|Rd dn|Sd }f|Rd }g|b|c }bt!drt"t#|bds |( dt |Wj |j$|b |j%|b |Xj |j%|b |j$|b |j|
d` |Od` d|O |
|j$|b }Z|j&t|Z|k d }?dcd||Z d` dc|B dc|B ||Z d` }^|9d_|( |Zd` }]t |?dk
rdn|^|?< dk|'dc|@ |A |] |^ |!|N|@ }h| |/ dc|< |Zd` }i|j'|
|O|j$|b |Z }jdu}ky|xDtt|Qd D ].}Y|b|(d k rl|j|Y |j(||
k rl|Zdn }hdv fdwdxdy}l|fdk r|gdz |gdz fdf}mn |gdz |gdz f|fdz |fdz ff}m|t)|gdm |fdm g|m |f|g|O|Y |J|b|j|Y |:|P|c|@|_|Y |h|Y |i|Y |`|Y |Yft*|l|kd{d|id}}n|g |f|nj+d~k rX|nj,d }k|nd d |nd d |R|Yd < |S|Yd < |Yt|Qd` k r|R|Yd` }g|S|Yd` }f|Wj |j$|b |j%|b |Xj |j%|b |j$|b |Wd d` |Xd d` |4d` k r8|Wd dk
r,|(|j(|Xd |Wd }on|j(|Xd |Wd }o|4|j$|o |Wd< |4|j%|o |Xd< q8W |j-|U|T|j$|b f}U|j-|V|T|j%|b f}VW n@ t.k
r |Wd tt |U }W|Xd tt |U }XP Y nX qW |j|W}W|j|X}X|j|U}U|j|V}Vdn|W|j|U|U |V|V |
| k|V|U ||
k @ |Udnk@ < dn|X|j|U|U |V|V |
| k|V|U ||
k @ |Udnk@ < |j/| |j0j1dt"|2d d t"| d r|j2dt"|2d d t"| d nd |j3t4|j|U|
|j|V|
|W|Xd}p|pj5| d t"|2d d t"| d ddd |1dkr|j0j1dt"| d rt|j2dt"| d nd |j3t4|j|U|
|j|V|
|W|Xd}p|pj5| d t"| d ddd ~pddl6m7}q |q|||4|
|1|2|| d }rqRW | j }st!dt"|s| d d t"|s| d d d S )Nr ) constante)minimizeminimize_scalarRsunMsunGg @@cg Y@sigma_tg @sigma_kmpmekbr g @g p@g @@g L@g @g L@g @g j@g YAg CAg ecBg 9Ag @TAg VxBg jHAg Ag P`Bg hAg 6Bg ØBg 8|Ag Bg WHBg חAg vH7Bg ļBg *uAg kDBg 聰vBg 9zAg '\ RBg Bg;On?gV-?g rh?g)\(?gzG?gS㥻?g;On?g^I +?gGz?gx&1?gx&1?g=
ףp=?gQ ?gI +?gX9v?g+η?g+?gʡE?g&1 ?gʡE?guV?gbX9ȶ?gbX9?gh|?5?gMb?g(\?g"~j?gy&1?gMbX?g(\?gV-?gK7?gHzG?g)\(?g'1 Z?g%C?gS?gx&1?g#~j?g;On?grh|?gq=
ףp?gS?g1 Zd?g{Gz?g9v?g @r r g @g ?g v@g@gGz?rE g mBgp=
v@gh㈵> cAg @g? Zbounded)boundsmethodargsg `f@zTheta: g f@z on 180 degreeg?Zineqc s( | d d | d d d d S )Nr r r r )r )r( r4 r r 0 s z+radiative_inhibition_comp..)typefung?maxiterd )rW rY rH constraintstoloptionsTg I@r Z wind_starZ_paramz.h5)r r> uxuyz/winds/wind_starZwindt)formatZwind_star2_paramz/winds/wind_star2_param)plotsz/windszI have worked during g N@z min or g @z hours.g ~~Ag ~~Agyy?gAOb?gŃ̷?gyy?gAOb?)rV rU rg rg rg rg rg rg rg rg )8sysr
r
constantesrI matplotlib.pyplotpyplotmatplotlib.mlabmlab
matplotlibZscipy.optimizerJ rK osZpandastimer' rG uniqueranger minmaxargminr intmeanr& r" r0 r appendlenprintstrroundr1 r2 wherearccosatanrD rH successr\ concatenateKeyboardInterruptchdirpathexistsremoveZ DataFramedictZto_hdfZ plots_windsrf )ttype1type2Mdot1Mdot2Per
mass_ratior r ZTeff_1ZTeff_2r directmu H_mass_fracbeta1beta2iparrh r
r rI pltrm rn rJ rK ro pdrp time1rL rM r light_velrP rQ ZmhrR rT Zsigma_er' Z
abbott82_teffZ
abbott82_densZk_vecZ alpha_vecZTuniqmass_sumr r Znbr_starZistarvinf2vinf1v1v2vlocalZL1ZL2r r r Zrhomeanind0indr k1r ZK2cZK1cc1c2r r r r* r ZzcZvcZVth1r) r+ ZnurZur_vecZ
utheta_vecZrvecZx_vecZy_vecrb rc r= r2r ZfluxZflux2r! r: r- Zsol1r7 r9 Znutheta_Zuthetaurr; r< r8 ZtoleZconsZbndsZsolangleZdf_xyzrf nothingtime3r )r( r4 r radiative_inhibition_comp s
:::4
$
8> $$ . .
@8 4,
"0002$
0( "",0( &
b
"""
::
T*0
4* r N)__doc__r
r" r& r0 rD rH r r r r r ' s '