3
_' @ s d Z dd Zdd ZdS )a
****************************************
*** Program for the ray tracing ***
****************************************
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 raytracing import RT
ray_tracing=RT(ndeltaZZ, xx, yy, zz, deltaZZ, deltaX, deltaY, deltaZ, x_vec, y_vec, z_vec, d, cote_cd, kT_all, sigma_all, Mdot1, Mdot2, R1, R2, Teff_1, Teff_2, dwind_prim, dwind_sec, dir_obs, mp, zt1, pt1, yt1, zt2, pt2, yt2, treecdw1, treecdw2, tree, treevec)
c C s dd l }|j}|j|| }|dk r0|t| }|j|| d }d}t|dkrt||k r|| |j|d |d d } || |j|d |d d }
||jd|d |d |jtt||
t|| gtt||
t|| g }t||kr d| |d }t||krd| |j|| | d |j|d |d d } | dk r|d| | } d| |j|d |d d | }t|S )N g @g g ?g ?) mathpiatanabstansqrtlogmaxmin) zzptau0Rr r testZalptaux1x2Zterme r 3/local/network/htdocs/Lifeline/codes3/raytracing.pytauX s( ""\8
&r c) L C s dd l })dd l}*ddlm}+ g },|)j}-|*jt| d }.|*j||.| ||.| ||.| gj}/d}0x>t t| D ],}1|| }|| }|| }|}2|'}3|(}4||kr|}2|%}3|&}4|2|3ko|2|4k r|#j
|| || || ggd d }5|$j
|| || || gg\}6}7|6d }6|7d }7| |5 dkr d}8|}9|}:|!j
||7 | | |7 | |
|7 | gg\};}<|;d };d}=|;|6krd}=|j
|| |)j |d |d | gg\}>}?|?d }?|*j |j |? d d |j |? d d }@|d
}Ad |A d |A |:|*j |d |d |d d
}Bnd}8|}9|}:|"j
||7 | | |7 | |
|7 | gg\};}<|;d };d}=|;|6krd}=|j
| | | |)j |d |d | gg\}>}?|?d }?|*j |j |? d d |j |? d d }@|d
}Ad |A d |A |:|*j | | d |d |d d
}B|=dkr|*j|||g|*j
j|*j|||g }C|,j|
|5 |,jt||5 t|*j|C|d| d |0d }0|=dkr|8dk
r(|,j|B |9t| |0d | |)j | d | d d|: d|- |@ |: d | d }D|@dk rd}D|,j|D |"j
|/t|0d d d d f \}E}>|*j|Edd |Edd dk d }Ft|Fdkr d |A d |A |:|*j | |/t|0d d df d |/t|0d d df d |/t|0d d df d d
}Gx8t tt|Gd D ] }H|,j|G|Hd |j
| |/t|0d |Hd df | |)j |/t|0d |Hd df d |/t|0d |Hd df d | gg\}>}?|?d }?|*j |j |? d d |j |? d d }I|,j|9t| |0|Hd d | |)j | d | d d|: d|- |I |: d | d qW P |*j|Fdd |Fdd dkd }Jt|Jdkr.|Fd t|/t|0d df d kr.d |A d |A |:|*j | |/t|0d d df d |/t|0d d df d |/t|0d d df d d
}Gx8t tt|Gd D ] }H|,j|G|Hd |j
| |/t|0d |Hd df | |)j |/t|0d |Hd df d |/t|0d |Hd df d | gg\}>}?|?d }?|*j |j |? d d |j |? d d }I|,j|9t| |0|Hd d | |)j | d | d d|: d|- |I |: d | d qW P t|JdkrF|Fd }Kn |F|Jd }Kd |A d |A |:|*j | |/t|0d t|K|0 df d |/t|0d t|K|0 df d |/t|0d t|K|0 df d d
}Gx8t tt|Gd D ] }H|,j|G|Hd |j
| |/t|0d |Hd df | |)j |/t|0d |Hd df d |/t|0d |Hd df d | gg\}>}?|?d }?|*j |j |? d d |j |? d d }I|,j|9t| |0|Hd d | |)j | d | d d|: d|- |I |: d | d qW |K|0 d }0|8dkr|,j|B |9t||0d | |)j | d | d d|: d|- |@ |: d | d }D|@dk
rd}D|,j|D |!j
|/t|0d d d f \}E}>|*j|Edd |Edd dk d }Ft|Fdk rd |A d |A |:|*j |/t|0d d df d |/t|0d d df d |/t|0d d df d d
}Gx4t tt|Gd D ]}H|,j|G|Hd |j
|/t|0d |Hd df | |)j |/t|0d |Hd df d |/t|0d |Hd df d | gg\}>}?|?d }?|*j |j |? d d |j |? d d }I|,j|9t||0|Hd d | |)j | d | d d|: d|- |I |: d | d qW P |*j|Fdd |Fdd dkd }Jt|Jdkr|Fd t|/t|0d df d krd |A d |A |:|*j |/t|0d d df d |/t|0d d df d |/t|0d d df d d
}Gx4t tt|Gd D ]}H|,j|G|Hd |j
|/t|0d |Hd df | |)j |/t|0d |Hd df d |/t|0d |Hd df d | gg\}>}?|?d }?|*j |j |? d d |j |? d d }I|,j|9t||0|Hd d | |)j | d | d d|: d|- |I |: d | d
qW P t|Jdkr|Fd }Kn |F|Jd }Kd |A d |A |:|*j |/t|0d t|K|0 df d |/t|0d t|K|0 df d |/t|0d t|K|0 df d d
}Gx4t tt|Gd D ]}H|,j|G|Hd |j
|/t|0d |Hd df | |)j |/t|0d |Hd df d |/t|0d |Hd df d | gg\}>}?|?d }?|*j |j |? d d |j |? d d }I|,j|9t||0|Hd d | |)j | d | d d|: d|- |I |: d | d qnW |K|0 d }0|0| krvP qvW |,S ) Nr )KDTreeg ? inoutr uxuyghh!w>g?g333333?gffffff?g?g @@g @g Y@g g >@g @r r r r r r r )r numpy
scipy.spatialr r arangeintarrayTrangequeryr iloclinalgnormappendr dotr wherelen)LndeltaZZxxyyr deltaZZdeltaXdeltaYdeltaZx_vecy_vecz_vecdcote_cdkT_all sigma_allMdot1Mdot2R1R2Teff_1Teff_2tree_prim_wind
tree_sec_wind
dwind_prim dwind_secdir_obsmpzt1pt1yt1zt2pt2yt2treecdw1treecdw2treetreevecxmin1xmax2ymin1ymax2r npr ray_tracingr Z range_pointsZrayZincr_zz_Z test_distZmin1Zmax2 ind_closeZdistvecZ ind_closevecZ cote_chocZMdotradiusZdistcdwZ ind_closecdwinoutrienZ
ind_close2ZvwindZT0ZTwindZrdeltanZ
append_valZ
dist_resteZ index_negZ Twind_resteZirestZ
vwind_restZ
index_neg2Zindex_reprendrer r r RT- s *&".
.*8.
2*:
(.
L
&&v z*`&4v z*`
z*`
L
"&r v*`&4r v*`
v*` r] N)__doc__r r] r r r r s