a L9f>9@sddlZddlZddlmZddlmZddlm Z ddl Z ddl ZddlmZddlmZddlmZddlmZddlmZddlmZde jd <d e jd <d e jd <dejj_djejj_ e!dddZ"GdddZ#dS)N)deepcopy) HandlerPatch)log)frames)get_horizons_coord) body_dictg?zaxes.linewidthz font.sizei Nzagg.path.chunksizez{:.1f}WARNINGcCs2tjjtdgddddgd}d|j_|S)z prints a selection of body keys and the corresponding body names which may be provided to the HeliosphericConstellation class index)IDZBodyColor)orientcolumnsr r rKey)pd DataFrame from_dictrdropdrop_duplicatesr name)datarO/Users/putuwistika/Documents/TRILAK/hydralit-example/apps/extras/backmapping.pyprint_body_lists rc@s<eZdZdZgddfddZdddZdd d Zd dZdS)HeliosphericConstellationan Class which handles the selected bodies Parameters ---------- date: str body_list: list list of body keys to be used. Keys can be string of int. vsw_list: list, optional list of solar wind speeds at the position of the different bodies. Must have the same length as body_list. Default is an epmty list leading to vsw=400km/s used for every body. reference_long: float, optional Carrington longitute of reference position at the Sun reference_lat: float, optional Heliographic latitude of referene position at the Sun Nc stt|}tt||_||_||_td|jd}| t j dd|_ t |dkrhtt |d}gd}g}g} g} g} g} g} g}g}g}g}t|D]J\}}|vr܈|d}|d}|d }n.|}t|}||}t|g|||gzt||d}| t j dd}||||||jj|j jj}|d krp|d }|jj|j jj}||jj| |jj| |jj| || || |||j|||||d \}}|||jj|}|d kr|d }|||jdurl|||jj|j}|d krX|d }|||||jdur|jj|j}||Wqtytdtdt|d|jtdt|d|j||Yq0qfdd|D}||_t| |_ t!"t|j#|| | | | | |d|_$|jdur^||j$d<||j$d<|jdurt||j$d<|j$j%j&fiddidS)NiidSun)observerr) forestgreen mediumbluem saddlebrowntomatoolive steelblue darkmagentac darkslategrayyellowdarkolivegreenv@)vswhz!!! No ephemeris for target "z " for date zNo ephemeris for target "csi|]}||qSrr).0Zsel_keybodiesrr z6HeliosphericConstellation.__init__..)zSpacecraft/BodyuCarrington Longitude (°)u Latitude (°)zHeliocentric Distance (AU)z,Longitudinal separation to Earth's longitudez*Latitudinal separation to Earth's latitudeVswz)Magnetic footpoint longitude (Carrington)z7Longitudinal separation between body and reference_longzLLongitudinal separation between body's mangetic footpoint and reference_longz5Latitudinal separation between body and reference_latz text-alignleft)'listdictfromkeysrrdatereference_long reference_latr transform_torHeliographicCarringtonpos_Elennpzeros enumeratecopystrupdateappendlonvaluelatradius backmapping ValueErrorprintstwarningremovemaxmax_distrrkeys coord_tablestyleZset_properties) selfr< body_listvsw_listr=r>rA random_colsZ body_lon_listZ body_lat_listZbody_dist_listZlongsep_E_listZ latsep_E_listZ body_vsw_listZfootp_long_listZ longsep_listZ latsep_listZfootp_longsep_listibodybody_idbody_lab body_colorposZ longsep_EZlatsep_EsepalphaZbody_footp_longZlong_sepZlat_sepZbody_dict_shortrr3r__init__<s                    z"HeliosphericConstellation.__init__rc Cstjd}|}|jj}|jj}td} |||} t| | } |dur||| |} | dkrf| d} | dkrdt| } nt j } | | fS)a; Determine the longitudinal separation angle of a given spacecraft and a given reference longitude Parameters ---------- body_pos : astropy.coordinates.sky_coordinate.SkyCoord coordinate of the body in Carrington coordinates date: str e.g., '2020-03-22 12:30' reference_long: float Carrington longitude of reference point at Sun to which we determine the longitudinal separation vsw: float solar wind speed (km/s) used to determine the position of the magnetic footpoint of the body. Default is 400. out: sep: float longitudinal separation of body magnetic footpoint and reference longitude in degrees alpha: float backmapping angle M?j%?Ngf@r0gf) constaurJrKrMmathradiansdegreesabsrCnan) rYbody_posr<r=r/AUrbrJdistomegattrdrcrrrrNs    z%HeliosphericConstellation.backmappingTFr1c' Cs$ddl}tjd}tjtdddd\} } | |_td|j d d } t d } t |j D]\} }|j |d }|j |d }|j |d}|j |d}|}|j j}|jj}|jjj}|jj j}| jt||d||d|r| jt|t|gd|gd|d|rX|||}t| |}t|| |||| }| j|| |dqX|jdur|j}|dkrx|d}t|| ||||| | ||||}t|j ddg}tj|ddd|dddddddd }|r| j|| dd |d!d"| jd#d$d%}|jdurTd&d'} | j|gd(gd)tjt| d*id$d+}!| || |d,| td-|| |j d | d| j !j"j#dd.tj$d/|j d0| j%j&d1dd2d d3}"| '|"|j d kr| j(td|j d0d4d dd5d6n2|j d7kr6| j(td|j d0d8d dd5d6| j)|j*d9d:d;t+tj,d}$|#j.|$d |#j/|$d |#j0|$|#j1|$g}%|2| ||%}&| j3d?d7d@| j4dAdBdCdDdEdFdG| j5dH| j4dAd>dIdJdKdFdG| j5dH|r| j67d|dLkrt8|t9:| dS)Ma  Make a polar plot showing the Sun in the center (view from North) and the positions of the selected bodies Parameters ---------- plot_spirals: bool if True, the magnetic field lines connecting the bodies with the Sun are plotted plot_sun_body_line: bool if True, straight lines connecting the bodies with the Sun are plotted show_earth_centered_coord: bool if True, additional longitudinal tickmarks are shown with Earth at longitude 0 reference_vsw: int if defined, defines solar wind speed for reference. if not defined, 400 km/s is used outfile: string if provided, the plot is saved with outfile as filename rNrfpolar) projection) ) subplot_kwfigsizegy&1|?333333?gMbP?rgr+r,s)colorlabelg{Gz?:)r~r.g @g@g?gQ?blackg?) head_width head_length edgecolor facecolorlwzorderoverhangz--kz)field line connecting to ref. long. (vsw=z km/s))r)333333?gffffff? )locfontsizec[stjdd||ddd|dS)Nr?Tg?)length_includes_headr)mpatches FancyArrow)widthheight_rrr legend_arrowsz4HeliosphericConstellation.plot..legend_arrowzreference long.)rg333333?) patch_func)r handler_maprxi)nbins)rrg(\?kF) transformrrfillrrg6@)angle g? <)padg333333?)bottomgQ?x)axisrgGz?g{Gz?z Solar-MACHz DejaVu Serifrightr) fontfamilyrhavarzhttps://solar-mach.github.ioz DejaVu Sansr1);pylabrhripltsubplotsr:axrCarangerUrkrErrMrKrJrAplotdeg2radrlr=minarrowlegendrrr add_artistset_rlabel_positionset_theta_offsetset_rmaxset_rminyaxisget_major_locatorbase set_paramsCircle transData_b add_patch set_rgrids set_titler< tight_layoutsubplots_adjust get_positionx0y0rr _polar_twin tick_paramstext transFigurepatch set_alphasavefigrQpyplot)'rY plot_spiralsplot_sun_body_lineshow_earth_centered_coord reference_vsw transparentoutfileplrpfigrrrrr]r_r`raZbody_vswrorbZ dist_bodyZ body_longE_longZdist_ersrdZ alpha_bodyZ delta_refZ alpha_refZ arrow_distZref_arrZleg1rZleg2circlepos1offsetpos2ax2rrrrs    &         ( & ,     zHeliosphericConstellation.plotcCst|jj|ddd||d}||jd|jd|d|jddd d |j }|D]}| dq`|S) z| add an additional axes which is needed to plot additional longitudinal tickmarks with Earth at longitude 0 rttwinF)rurframeontheta_direction theta_offsetrzSr darkgreenr)rcolorsr) figureadd_axesget_theta_directionrrUr set_visibleset_theta_zero_locationrxaxis get_gridlines set_color)rYrrpositionr gridlinesZxaxrrrrJs     z%HeliosphericConstellation._polar_twin)r)TFTrFr1)__name__ __module__ __qualname____doc__rerNrrrrrrr+s h * |r)$ streamlitrQrjrFrmatplotlib.patchespatchesrmatplotlib.pyplotrrnumpyrCpandasrscipy.constants constantsrhZmatplotlib.legend_handlerrsunpyrsunpy.coordinatesrrZapps.extras.selected_bodiesrrcParamsoptionsdisplaymax_rowsformat float_formatsetLevelrrrrrrs(