o z!profiles_plot..cSg|]}|dqSrl set_fontnamerr[r&r&r'r upper leftlocpropZ Profiles__.png,tightdpi bbox_inchesN)pltsubplotsprintr9rrscatterplot tick_paramstwinxset_ylimnanminnanmax set_xlabel set_ylabelr>r?distance_2pointslinspacer;ceil set_xticksset_xticklabelsget_xticklabelsget_yticklabelslegendsavefig)rGrHm_names_nameoutdir figure_sizefigaxesax1rCx_axisrDax2Zdem_avefont1lon_slat_slon_elat_edistancexticklabelsfig_name fig_outputr&r&r' profiles_plotsR" $   rc  Csddg} tjdd| d\} } | } tdtdt|d}| j||dd|d| j||d d|d| jd d d d d d d d| }|j||ddd| t |t |d|jd d d d d dd ddddd}| d|| d|| d|t||||}tjdtt|dd d}| tjdt|dd d| dd|D| | }dd|D| jd|d d!|d"|d#}| |}| j|d$d%d&d'S)(z,plot master and slave data along one profilerIrJrrKrMrVrWrXr\r]r^r_Tr`rirjrkFrlrmrnrorsrtrurrxcSr{r&r|rr&r&r'r/rz profile_plot..cSrrrrr&r&r'r2rrrZProfile_rrrrrN)rrrrrr9rrrrrrrrrrr;rrrrrrr)rrrr m_profile s_profilerr dem_profilerrrrrrrrrrrrrr&r&r' profile_plots:   rcCsd}t|}t|}t|}t|}||} ||} t| ddt|t|t| dd} dtt| td| } || } | S)z@calculate distance[km] between two points based on their lat/long@rr)mathradianssincosatan2sqrt)rrrrRZ lon_s_radZ lat_s_radZ lon_e_radZ lat_e_radZdlonZdlatarSrr&r&r'r;s    8 rc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)ProfileaProfile class for profiles Example: import matplotlib.pyplot as plt from mimtpy.objects.profiles import Profile import mimtpy.objects.profiles as profiles pro_no, pro_lons, pro_lats = profiles.search_profiles(3,over_lat0, over_lon0, over_lat1, over_lon1, m_atr, s_atr) pro_obj = Profile(1, pi/6, pro_lon, pro_lat, m_overlay, s_overlay, over_lat0, over_lon0, m_atr, s_atr, outdir='$SCRATCHDIR/project/'+pro_No) pro_obj.profile_extract() c CsF||_||_||_||_||_||_||_||_| |_| |_ | |_ dS)N) r3 pro_anglepro_lonpro_lat m_overlay s_overlayrrrrr) selfr3rrrrrrrrrrr&r&r'__init__Xs zProfile.__init__cCs\t|jd}t|jd}t|j|j|d|_t|j|j|d|_|j|jfS)zz# @D0  N) ropenwrite writelinesr}min lon_startlon_endmax lat_startlat_endclose)rnameZgmt_filefr&r&r' profile_gmtxs   T        zProfile.profile_gmtcCsJ|j|j|j|j|j|jd}t|j|dd t |dS)zAsave each profile geometry info and data along value as json file)rrrrr2r4z.jsonrN) rrrrrtolistrrrrjsondumps)rr profile_datar&r&r' profile_jsons$zProfile.profile_jsoncCs6t|j\}}||j}|j}|j}|dtjdkr\|dtjdkr\t|}|dtjdkrHdt |}t ||||}n>|dtjdkr[t t ||}n*t|}|dkrxdt |}t ||||}nt dkrt t ||}|jtjd|_|jtjd|_|j|j|jf|_|j|j|jf|_tj|j|jdk<tj|j|jdk<|jd} |jd} |jd} |jd} || | \|_|_|| | \|_|_|jd} td | d |_|jd}td |d |_ |jd |j d t!|j"}|#||$|d S) z6extract m_overlay and s_overlay data along the profile-Zrhrr FILE_PATHz Sen([^/]+)/rrN)%rshaperrrrrpirrtanronesZanlgeastyper;row_nocolm_norrrnanrrrrrrr searchrrrr}r3rr)rrowscolmsrranglerZ tan_valuer row_startrow_endZ colm_startZcolm_endZ m_name_tmpZ s_name_tmpZpro_namer&r&r'profile_extractsN$          zProfile.profile_extractN) __name__ __module__ __qualname____doc__rrrrrr r&r&r&r'rMs     r)osr r>rrmatplotlib.pyplotpyplotrnumpyrr(rrrFrrrrr&r&r&r's  n36*