3
__i @ sZ d Z dd Zdd Zdd Zdd Zd d
Zd d Zd
d ZdddZdd Z dd Z
dS )a
***************************************************
*** Program for the computation of the shock ***
*** in adiabatic colliding wind binaries ***
***************************************************
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 ashock import ashock
rien=ashock(Mdot1,Mdot2,vinf1,vinf2, mass_ratio,ex,omega, a, Per, R1, R2, Teff1, Teff2, nbr_points_shock_2D, nbr_points_shock_3D, nbr_points_width, nbr_bin_profile, direct, cut_lim, wind_prim, wind_sec, phase, incl,ipar, mu)
Use of the Canto et al. (1996) formalism
# Parameters
# ==========
Mdot - Mass loss rate [Msun/yr]
vinf - terminal velocity of the winds [cm/s]
ex - eccentricity
omega - argument of the periapsis [radian]
a - semi-major axis [Rsun]
R - stellar radius [Rsun]
Teff - Effective temperature of the star [K]
nbr_points_shock_2D - number of points to create the 2D shape of the shock
nbr_points_shock_3D - number of points to create the 3D shape of the shock
nstep_width - number of points to discretize the inside of the shock
nbr_bin_profile - number of points in the line profile
direct - directory where saving files
cut_lim - Limit of the tangential velocity to see the emission
wind_prim - h5 file containing the wind distribution of the primary
wind_sec - h5 file containing the wind distribution of the secondary
phase - phase of the system (not corrected from omega)
incl - inclination of the orbit
ipar - index of the system parameter
mu - Mean molecular weight
# Versions
# ========
v1 - 16/03/18
c
C sP dd l }
d}d}| d |d |d k rd}|jd|| || gg\}}|d }|
j|
j| d d |
j| d d }|
j|dkrd}|| kr| }|| d |d |d k r$d}|j|| || gg\}}|d }|
j| j| d d | j| d d }| j| d dkr| j| d dkr|d }|d }d
}| j| d }| j| d }|d }x~|dkr|dkr|d
kr| j| d }| j| d }|d }|}d }n,d
}| j| d }| j| d }|d }|}qnW |
j| j| d d | j| d d }|| kr$| }| d |d |d kr@|| d |d |d kr@|j| | || gg\}}|d }|
j| j| d d | j| d d }| j| d dkr| j| d dkr|d }|d }d
}| j| d }| j| d }|d }x~|dkr|dkr|d
krX| j| d }| j| d }|d }|}d }n,d
}| j| d }| j| d }|d }|}q W |
j| j| d d | j| d d }|jd| | || gg\}}|d }|
j|
j| d d |
j| d d }|
j| d dkr|
j| d dkr|d }|d }d
}|
j| d }|
j| d }|d }x~|dkr|dkr|d
kr|
j| d }|
j| d }|d }|}d }n,d
}|
j| d }|
j| d }|d }|}qhW |
j|
j| d d |
j| d d }|
j|dkr$d}|| kr2| }|| kr@| }|| || S ) N g cA g ?uxuyFg updown)numpyquerysqrtilocisfinite)xyR1R2Mdot1Mdot2dtree_prim_wind
tree_sec_wind
dwind_prim dwind_secvinf1vinf2npvv1vv2rien ind_closeZ
ind_closepZ
ind_closemZstadeZdw1Zdw2 r //local/network/htdocs/Lifeline/codes3/ashock.pycomp_eta5 s * *(
*
8 *(
* *(
*
r! c C sN dd l } t| d||||||||| |
|
}
|
dkr4dS | |d| j|
S d S )Nr g g ?)mathr! r
)r
r r r r r r r r r r r r" etar r r find_x0 s
r$ c C s& dd l }|j|d || d | S )Nr r )r r
)ZcenterxZcenteryradiusr
r r r r plot_star s r& c C s: dd l }|}| }|j|d |d }|j|| || S )Nr r )r r
arctan2)r zr ZoppZadjZhypor r r angleyz s
r) c C s dd l }| |j| S )Nr )r" cos)vecangler" r r r projx s r- c C s dd l }| |j| |j| S )Nr )r" sinr* )r+ angle1angle2r" r r r projy s r1 c C s dd l }| |j| |j| S )Nr )r" r. )r+ r/ r0 r" r r r projz s r2 皙?c C sx dd l }| j}|| |j|| ddd }|| |j|| ddd }|j|j|d |d dd}| j||k d S )Nr r )axisr )r trianglesrollmaxr
set_mask) triangr
r alphar r5 ZxtriZytriZmaxir r r
apply_mask s r; c
C s dd l }| \}}}|\}}} g }
g } xTtt|D ]D} | j|j|| ||f||f |
j|j|| ||f|| f q2W | |
fS )Nr )r rangelenappendinterp)
p1p2r
r x1y1z1x2y2z2ZreszZresyir r r find_xy s
$rI c + C st" ddl m} ddlm} ddlm} dd l}dd l}ddlm } dd l
j }dd l }dd l
} dd l}
ddlm} ddlm} ddlm}
dd lj}| \}}}}}}}}}}}}}} } } } } }!}"}#}$}%}&}'}(|j})|
j|"jd d
}*|
j|#jd d
}+||j|j|*j d d d |*j d d d fdlt |*j d d d fj dd},||j|j|+j d d d |+j d d d fdmt |+j d d d fj dd}-|| || }.|d}/|d}0|d}1|dd }2||0 dn }||0 do }||/ }||/ }||/ }t ||g}3|j | |j!dd }4|4|j"t!| d d |3 |j"t!| d }5|)d }6|.dkr
|)d|. }7|)d }6d|)d |7 |j#|)d }8|8|7|6 }9|6|9 }6xLt$|9dkrt$|8dkrd|6|7 |j#|6 }8|8|7|6 }9|6|9 }6qW |6|)d k r
|)d |6 }6|$}:|:};|:|; ||j%|; d||j&|; }|j(|>|=}?|?d|) krd}?|d|d d||j&|? }@|j)||@ }A|t*||@ | d |@|||||,|-|*|+||f d }B|Bd }Bt+| d t,|& d }C|Cj-t,|Bd |Cj-t,|B| d! |Cj-t,|B|@ d" |Cj. t |B|g}B|$|( }$|$dk rd|) |$ }$d#d$ t/d%D \}D}E}F}G}H}I}J}K}L}M}N}O}P}Q}Rd&d$ t/d'D \
}S}T}U}V}W}X}Y}Z}[}\}]}^}_d(d$ t/d)D \}`}a}b}c}d}e|j | d* |) d+| }fd*|) d+| }g|j0| |@ ||@ d,}ht1dd||@ |h}i|j0| |@ ||@ d,}jt1dd||@ |j}k|jd }j|j | |6 | d-|) d }l|j | d. d |)|6 | d. d
}m|m|j#|m }ndd|. |l|j#|l d }og }pxH|oD ]@}q|j2t$|n|q |j3t$|n|q kd }r|pj4|m|rd qTW |j5|p}p|j%|p|j%|p|l }s|s|j%|l |@ }t|j2|+d d/kd }u|+j |u }v|B}wd}xg }yg }zx|tD ]}{|j'|wd |xd }||j'|@|w d |xd }}t6|w|x|||||@|,|-|*|+||
}~|~d
kr|{|w|@||d |j'|~ ||d |j'|~ |}d }d| }||{|x |w }w|{}x|yj4|w |zj4|~ qW |j5|z}z|j5|y|@ }y|t|@ }tdddg}d}xXt/| D ]J}t!|d
d0 dkrLd1t,|d
d2 t,| d3 }t7| |y|@ }|t|@ |j&|f| }|t|@ |j%|f| }
xt/t |yD
]}|| dk rl|j)|j'|| d || d |@|| |Ak rl|j#|A|@|| }|j8|| |j'|| d || d }t$||j&| t!|j9|| ||< t$||j%| t!|j9|| ||< |j:|j;|| || || g||j|| || }|)|j:|j;|| |@ || || g||j|Kd~ |Ld }tA|j | d | |}tB|j | d | ||}tC|j | d | ||}|j|W||j%|% |j&|$ ||j&|% ||j%|% |j%|$ f}WtA|j | d | |}tB|j | d | ||}tC|j | d | ||}|j|W||j%|% |j&|$ ||j&|% ||j%|% |j%|$ f}W|j|P|f}P|j|^d
d|j | f}^|j|Q|j | d | f}Q|j|R|j | d | f}R|j|P|f}P|j|^d d|j | f}^|j|Q|j | d | f}Q|j|R|j | d | f}RqW qW |j|X|Sf}y|j|Y|Tf}t|j|Z|Uf}|j|J|Df}D|j|K|Ef}E|j|L|Ff}F|j|M|Gf}G|j|N|Hf}H|j|O|If}I|j5|[d d }|j5|\d d }|j5|Pd d }|j5|Qd d }tDt |Ƀ| }|j|d|ͅ |tDtE| | d/ tDtE| | d/ | f}|j|d|ͅ |tDtE| | d/ tDtE| | d/ | f}|j|d|ͅ |tDtE| | d/ tDtE| | d/ | f}|j|d|ͅ |tDtE| | d/ tDtE| | d/ | f}|jFd||@ d|jGt |Ѓ d? }|jH||@ ||@ }tI|||@ ||@ d@dA |jJd
|jK }|jL }|jM|Ӄ |jN|||dB |jM|҃ |jO|h|idCdD |jO|j|kdCdD |jO|h|i dCdD |jO|j|k dCdD |jPdEdF}|jQdG |jRdt |P |jSdH |jTdI |jUdd |jVdd
|jW| dJ t,|& dK |j. d}|jFd||@ d }ِqW |jX|j dM|jGt |у| |j"t@|у| dN |j"t@|у| }|jH||@ ||@ }tI|||@ ||@ d@dA |jJd
|jK }|jL }|jM|Ӄ |jN||| |dB |jM|҃ |jO|h|idCdD |jO|j|kdCdD |jO|h|i dCdD |jO|j|k dCdD |jPdEdF}|jQdOt,|ك dP |jRt@|Q| t |Q| |jSdH |jTdI |jUdd |jVdd
|jW| dQ t,|& dK |j. |[|\|]|P|QdR}|
j-|| dS t,|& dT t |[}dU| }t@dV|jG|j't |yt@|y d t |tt@|t d t |ȃt@|ȃ d | g}t@|D}dWt |G }t@|H}dWt |H }||j&|$ |j%|% }||j&|% }||j%|$ |j%|% }|j5|||g|jg r )argsz/histograms/xstag_parwz cm
z R1
z d
c s s | ]
}g V qd S )Nr ).0_r r r * s zashock.. c s s | ]
}g V qd S )Nr )rY rZ r r r r[ + s
c s s | ]
}g V qd S )Nr )rY rZ r r r r[ , s g v@g f@( g-C 6?g @@g ?g Y@z* /z *)r4 r r g cAg @ghh!w>g @g 0@white)colorg $@g "@g333333?)r: )levelsblack)cz%.2f)formatzTemperature [keV]zy/Dzx/Dz/plots/temperature_xy_parz.pdfgvIh%<=g 4@g 3@z Density [z g/cm^3]z/plots/density_par)r
r r( kTrhoz/plots/char_shock_parz.datg?g j@g?z/histograms/ray_tracing_parZ_binz.dataz# Temperature / Factor
z*** Ray tracing ***gQ@-(;g Q?gq=
ףp?gqىEa
z/histograms/bin_parz # Tangential velocity (km/s)
z /histograms/emiss_part_shock_parz:# Part of the emissivity which depends only on the shock.
z/histograms/temp_emiss_parz9# Temperature of the cell where the emission is created.
z*# Each line of a bin of the line profile.
rm g H~Ag H~Ag9UY>g9UY>rm rm rn rm rn rm rn rm rn rm rn rm rm )r r )r r rm rn )r r )r r rm rn rm rm rm rm rm rm rm )[Z
scipy.spatialrJ Zscipy.optimizerK scipyrL r r"
constantesrM matplotlib.pyplotpyplot
matplotlibosZpandasZmpl_toolkits.mplot3drN Z
raytracingrO Z
astropy.iorP matplotlib.tritripiZread_hdfZ reset_indexreshape concatenater r= Tr7 arangefloatfloortanabsr. r* r
atan2atanr$ openstrwritecloser< linspacer& wherenanminr> arrayr! printacossignarccosdotlinalgnormr) r minr- r1 r2 introundCircleceil
Triangulationr; figuregcfgca
add_artist tricontourfplotcolorbar set_labelset_climylabelxlabelxlimylimsavefigsortjoinmap( liste_paramrJ rK rL r r" rM pltrs rt pdrN rO rP rv r r r r
mass_ratioexomegarj perr r ZTeff_1ZTeff_2nbr_points_shock_2Dnbr_points_shock_3DZ nstep_widthnbr_bin_profiledirectcut_lim wind_primwind_secphaseincliparmuM_conjrw r r r r betarR rS rT rU vmaxr+ ZvvvZthetainfasympZdeltafZ
deltathetaMEdEcoisii
phase_truer Z
ouverture_minxstagZfxstagZx_cdw1Zy_cdw1Zz_cdw1Zx_cdw2Zy_cdw2Zz_cdw2Zx_cdw1gZy_cdw1gZz_cdw1gZx_cdw2gZy_cdw2gZz_cdw2gZkT_allZrho_allZ sigma_allZx_cdZy_cdZz_cdZvol_cdZvrad_cdZx_skewZy_skewZz_skewZxx_allZyy_allZzz_allZcote_cdZvt_vecZvt1_vecZvp1_vecZvt2_vecZvp2_vecZ
slope_vec1Z
slope_vec2Zphi_vecdphiZxstar1Zystar1Zxstar2Zystar2Z theta_vecZ
theta1_vecZ
ThetaCotThetapointZthe1_vecZ pointhereindZr_testZy_vecZind_halfZdwind_sec_halfZx_oldZy_oldZx_vecZeta_vecZ y_vecherer1r2r# ZpentexaxislmaxZicutsentenceZxcutZycutZzcutZcutZRprimer, ZthetahlatZ the1_vechZthetahm1Z the1_vechm1Z
slope_vechdthetaZdthe1Zr1_vechZr2_vechr r r r Zsig0r: Zterm1Zterm2Zterm3Zterm4Z sigma_vecZrho1Zrho2rh Z
thickness1Z
thickness2Zl0_1Zl0_2Zrl1Zl011Zrl2Zl022Zl1_widthZl2_widthZ increasevolZx_cdw1g1Zy_cdw1g1Zz_cdw1g1Zx_cdw2g1Zy_cdw2g1Zz_cdw2g1Z x_cdw1g1aZ y_cdw1g1aZ z_cdw1g1aZ x_cdw2g1aZ y_cdw2g1aZ z_cdw2g1aZvt1Zvt2Z x_cdw1g11Z y_cdw1g11Z z_cdw1g11Z x_cdw2g11Z y_cdw2g11Z z_cdw2g11Zvp1Zvp2ZkT1ZkT2ZvxZvyZvzZz_vecZx_capZy_capZkT_capZrho_capZ nbr_layerZx_layerZy_layerZkT_layerZ rho_layerZcircle1Zcircle2rc Ztriang2figaxcbarfactortableZ
nbr_pt_totZdeltaZZZndeltaZZxmin1Zxmax2ymin1Zymax2ZdeltaXZdeltaYZdeltaZZdir_obsZensemble_pointstreeZtreecdw1Ztreecdw2ZtreevecZ
temp_emissZ emiss_part1rH ZfRTxxyyzzrg ZvolZvradyt1Zzt1Zpt1yt2Zzt2Zpt2Zvis1Zvis2Zind_vradZ ray_tracingZfbinZfhistoZftempr r r ashock s$
8RR 0
(. . "0
0, &
(
8
D,((Vb
nz *.:* >* (8<
66
RR
VV&&
HH (BBBB
J
T (. 666
&
>@ BD&&6X*
"
r N)r3 ) __doc__r! r$ r& r) r- r1 r2 r; rI r r r r r 1 s g