7fVddlZddlZddlmZddlZddlmZddlZddlm Z ddl m Z ddlmZddlZddlmZddlZddlddlddlmZmZmZmZmaddlZ dZ"d Z#d Z$dd Z%dd Z&dd Z'ddZ(ddZ)dZ*dZ+ddZ,y)N)datetime) relativedelta)Line2D)interp2d)*)JSON_DOWNLOAD_URL JSON_VOLCANO START_DATEEND_DATEELNINOSc |j}|jdztz}g}|jr t}nd}|jr t |||j r t||jr\t|j|j|j}|rt|||jnt|||j|j rpg}|gk(r+t|j|j|j}t#||j$kr d}t'||j(rW|j*rKt-|j(|j*\|_|_d|j(d|j*}n|j.rt1||j.d\}}} t-|d|d\|_|_|j dk(r|j(d|j*dg} t3|j(dz t5|j(dzg|_t3|j*dz t5|j*dzg|_|j.dd |j(d|j*}n d }t'||j.r|j.d} n3|j(r'|j*r|j(d |j*} t7|jt8s%t9|jj;d d n|jj;d d } t7|jt8s%t9|jj;d d n|jj;d d } |j<d d | d | d |j d }|j dk(rd}nd}|jr8|j?r|j?jAs t}tC|j(|j*|||}|j dk(s|j dk(s|dk(rt9|j$dz}nK|j dk(r,|jDrdt#|d}n dt#|d}nd|j d}||dd}|jdvs |jDr!tG||j|jD}|j dk(r&|jHrtK||jH}tM||j*|j(||jN|jP||jR|j.rGtUjV d| dddd |j.dd!"tUjXd#dd$%tUjZ}tUj\}|j^rtUj`||jbrtUjd||fStg||j$}ti|jjtm|jj}|jnr1t7|jnt st |jnn|}|gk7rXtq||d&}|Dcic]}||}}|d&js||d'<|jtjwtx|d'<|jjdkDrJt{|jjDcgc]+}t}j~||t9|dzz(-}}ng}|d)jd*+|jjzt#|z jtj|jjdz ,|d-<|d-jsfd.|d-<t|}|j d/k(rt|||\}}nAd}|r$|jd)0}|jd1}t|||j||}|jr|s t|||}|jns|gk7rt||||}tUjX|d2d#3tUjtUjZ}tUj\}|j^r!td4|tUj`||jbrtUjd||fSy5cc}wcc}w)6N/z;Error: The number of dates is less than the rolling window.z Latitude: z , Longitude: rmapz - Latitude: zfError: Please provide valid coordinates or volcano name. Try using --list to get a list of volcanoes._-z.pngstrengthTFdailybarz day precipitation (mm)zCumulative precipitation over z days (mm)zDaily precipitation over z days (mm/day) z precipitation (mm)zCumulative precipitation)titleylabely2label)WMYred^2)colormarkerslabelzordersmallg333333?)fontsizeframeon framealphaDate Eruptionsr#r&rollfirst)method)upperr#c|SN)xcolorss Q/home/exouser/operations/rsmas_insar/tools/Precip/src/precip/plotter_functions.pyz!prompt_subplots..s fQiannualbydropz upper left)handleslocr)zSaving:)NN)Odirr use_sshconnect_jetstreamcheckcheck_nc4_fileslistvolcanoes_listdownloadgenerate_date_list start_dateend_dateaveragedownload_jetstream_parallelparalleldload_site_list_parallelstylelenr/ ValueErrorlatitude longitudeadapt_coordinates volcano_nameextract_volcanoes_infominmax isinstancestrreplaceoutdir get_transport is_activesql_extract_precipitationcumulate#weekly_monthly_yearly_precipitation interpolateinterpolate_mapmap_precipitationcolorbarisolinesvlimpltscatterlegendgcfgca save_flagsavefig show_flagshowvolcano_rain_frame color_schemebins quantile_name add_event adapt_eventsrr-applydate_to_decimal_yearrangempatchesPatchrankastypeintclipfrom_nested_to_floatannual_plotter sort_values reset_index bar_plotterlogelnino plot_elninosplot_eruptions tight_layoutprint)inpsgpm_dirvolcano_json_dir date_listssheruption_datesmsgrlaloidvolcano_positionsaveNamestrStartstrEnd save_pathr precipitationrlabelsfigaxesquantiledate eruption_dictilegend_handlesaxsr7s @r8prompt_subplotsrs3hhGxx#~ 4I ||! zz% yy'( }}&t t||T  ' 3 F %Wi G zz ?*4??DMM4<<XI y>TYY &OCS/ ! ==T^^,=dmmT^^,\ )DM4> }T^^NPTPaPabcPd'e $ND",=d1gtAw,O )DM4>zzU"$(MM!$4dnnQ6G#H !$T]]!3a!7T]]9Ka9O P "%dnn"5"93t~~;NQR;R!S((+,M$-- VZVdVdUefE|CS/ !   ,,Q/ ]]t~~--$..)9:H@J4??\_@`3t'//R8fjfufuf}f}BDFgG>9CDNNTX9YT$..1_mN R *.-:OPN5CCDT4ZCMC*7v)>)B)B=)QM+ &*7)@)@)F)FG[)\M+ & 99q=bghlhqhqbrs]^hnn6!9HsSTUVSVxDWXsNs N$1#8#=#=W#=#MPTPYPY#Y]`an]o"o!w!wx{!|"B"BIMIRIRSTIT"B"U g"/w!7!;!;>^r1+M>8UXYN >|gN ggiwwy >> I & ;;y ! >> HHJDy GDts - e0ec tj|}|j|j}|S#tjj $rq}t d|t dtjj|st|tt|}tj|}Yd}~|Sd}~wwxYw)NzError: zLoading from local file)requestsgetraise_for_statusjson exceptionsRequestExceptionrospathexistsdownload_volcano_jsonropenload)jsonfileurlresponsedataerrfs r8get_volcano_jsonrs<<$!!#}} K    / / y '(ww~~h' !(,= > Nyy| Ks59CA&CCct|t}g}|dD]$}|dd|vs|j|dd&|D] }t||S)z Retrieves a list of volcano names from a JSON file. Args: jsonfile (str): The path to the JSON file containing volcano data. Returns: None features properties VolcanoName)rrappendr)rr volcanoNamejvolcanos r8rHrHsr H&7 8DK * ? \?= ) <   q}= >? g r:c gd}t|t}g}g}tjtdj }tjt dj }|dD]} | dd|k(s| dd} | dd} tj| dddj } | dd } | d d d } tj| dd dj }t| d| d| d|| |k\r| |kr|j| |s| dd}|j| | ||g|rtj||}|S|gk7r?t|}tdtdtd|D]}td|tdtd|  fS#d }YxYw)am Extracts information about a specific volcano from a JSON file. Args: jsonfile (str): The path to the JSON file containing volcano data. volcanoName (str): The name of the volcano to extract information for. Returns: tuple: A tuple containing the start dates of eruptions, a date list, and the coordinates of the volcano. )VolcanoStartEndzMax Explosivityz%Y%m%drrr VolcanoNumber StartDategeometry coordinatesNEndDateNonez, id: z eruption started z and ended ExplosivityIndexMax)columnsz!---------------------------------zSorting eruptions by date...zExtracted eruption in date: r) rrrstrptimer rr rrpd DataFramesorted)rrr column_namesr start_dates frame_data first_daylast_dayrrnamestartrendstrendfds r8rXrXsBL H&7 8DKJ!!*h7<<>I  8499;H* = \?= )[ 8<1BlOM2D%%q{'ChOTTVEJ- 6K%dd+K ''<)ChOTTV TF&$6ug[N O !ex&7""5),(=>!!4U";<-=0 \\*l ; b[)  12 ,- 12 5A 0! 4 5 -. "I  R ''M s *F??Gc |rv|dDcgc]}|dz }}|dDcgc] }|dzdz }}dt|djz}|j||dd|d }|j||S|r!||dj j } n||dj d} | D]}t j|dd d d |td gd gdd dddgz }|Scc}wcc}w)Nr-r?xWr,blackvzVolcanic Events)r#r$r%r&dashed) )r6r# linestyledashes linewidthr)r"rzVolcanic event)r#rrr&r) rRuniquerkrnotnaindexrjaxvliner) rrrrrr6y scatter_sizeeruption eruptionss r8rris. )+6 7qQU 7 7$1+$> ?qa1f] ? ?]6%:%A%A%C!DD ;;q!73,Vg;hh'!- "<"B"B"DEKK "- "<"B"B"DEkR W aw(5TUVWvqcA3gSX`p}~AAN ' 8 ?s C0C5c |jddd}tj|jd}tj|jd}t |||}tj |j |j|jd|z}tj |j |j|jd|z}|||}|S#|d}YxYw)aO Interpolates a precipitation map using scipy.interpolate.interp2d. Parameters: dataframe (pandas.DataFrame): The input dataframe containing the precipitation data. resolution (int): The resolution factor for the interpolated map. Default is 5. Returns: numpy.ndarray: The interpolated precipitation map. Precipitationrr)rnparangeshaperlinspacerYrZ) dataframe resolutionvaluesr6r interpolatornew_xnew_y new_valuess r8reres/215 &,,q/"A &,,q/"AAq&)L KK&,,q/**D EE KK&,,q/**D EEeU+J 1s C;;Dc,tjjd|}t|tst |d}|j }d||dk<||ddt j||d|d}|dk7rt j||dt S) N01m)rregionrwhiter)levelsr7extent linewidths)inliner)) pygmtdatasetsload_earth_reliefr[rrrjcontourclabel)rrrgridgrid_npconts r8 add_isolinesrs >> + +uV + LD fc "VAYkkGGGaKDG ;;tF76VY ZD z 43 Jr:cgd}t|tjk(r|jddd} n,t|tk(r||dj d} n|} t j| jd} |sd} | j} n |d} |d} |d|d|d|dg} |s t| } nt| |d} | j| | | | | | j|d|d| j|d|d| j}|j|d | j |d y ) aj Plot a map of precipitation. Args: precipitation_series (pd.DataFrame or dict or ndarray): The precipitation data. If a DataFrame, it should have a column named 'Precipitation' containing the precipitation values. If a dict, it should have keys representing dates in the format 'YYYY-MM-DD' and values as precipitation arrays. If an ndarray, it should be a 2D array of precipitation values. lo (list): The longitude range [lo_min, lo_max]. la (list): The latitude range [la_min, la_max]. date (list): The list of dates for which precipitation is plotted. colorbar (str): The colormap for the plot. levels (list): The contour levels for the isolines. vlim (tuple, optional): The minimum and maximum values for the colorbar. Defaults to None. Returns: None )imrrz%Y-%m-%d)axisrT)r)vminvmaxrcmaprrN)typerrrdictstrftimerflip transposerZrimshowylimxlimrg set_labelr)precipitation_serieslolarrgrrrim_ypreciprrrrjcbars r8rfrfsZ( C !R\\1%))/:1=a@ " #t +%d1g&6&6z&BC& WWV%%'a 0F zz|AwAweRUBqE2a5 )F 6"66$7CJJvDtFJ CHHRUBqE CHHRUBqE 3<<>D NN6(#$ CIIfWor:c |tjd|r;|jd}|jd}d}t t |}nd}|d }|d}tj |d |dk(r,tjd tjgd |stjd tj|dt|djdzdz}tj|||d|d|dk(rt|d jdz} t|d jdzdz} tjt | | z dzdzD cgc] } | d| zz c} t | | z dzdzD cgc]} dt!| d| zzddzc} tj"} | j|d |dd|d| j%ddd|t'j(d|d gz }|Scc} wcc} w)!N) )figsizer/r<Tr>g?g{Gz?Decimalrr)g?rr)diYearr?rr#)r#widthalphaFr')ticksrcumsumgrayg?zCumulative precipitation (mm)ir))rotationlabelpadrr.)rjsubplotsrrr{rRryscaleyticksxlabelrrrZrrYxticksr\twinx set_ylabelr|r}) rrrrrr0r6rr4rrrplot2s r8rrsZLL!%11V1< %11t1<  #m$ % ) $fAJJvh  d{ 5 *+  6IIfWo v&**,s2q8 9EGGAq g.e1E5M),002a78- *..0A59: u8Ja7O1PQA%1Q3-QCHKNQVKV[\J\`aIaCb[c}~[^adejnopqnqerastvtwax[x[c d  - *M(,CF\akn o 83RTU8>>fY>OPQQ R[cs H4*H9 c |dj}|dj}t|dz}t|dzdz}tjdddddgid\}}|d} |d} |ddz}|ddz}| j|||d d d t|d j#z g}t ||dzD]0}|j%|d|ddz|k(j'2| j)t ||dz|dd| j+t |dz|z dzDcgc] }|d|zz c}t |dz|z dzDcgc]}t-|d|zzc}| j/t dDcgc] }dd|zz c}gd| j1d| j3d| j5|d| j5d| j+t |dz|z dzDcgc] }|d|zz c}t |dz|z dzDcgc]}t-|d|zzc}||dfScc}wcc}wcc}wcc}wcc}w)Nr,rr width_ratios) ) gridspec_kwr+rr%rr,)r#r$r%r#rrpurple)heightr# gUUUUUU?gUUUUUU?) 010203040506070809101112Monthr.rz Total (mm))rYrZrrjr9r{ color_count by_season volc_raincopyrrarrayrRrkr7rainfallrrsumbarh set_yticksr\ set_xticks set_xlabelr? set_title)rrr first_date last_daterrrrax0ax1rr rain_by_yearrain_jdates_jbin_sizer6rtotalsyearks r8rr+sy)--/Ji(,,.I  a E i1nq !C QA/GQYZIC q'C q'C  i 1$Ai A%AKK1M'236SQ^_eQfQmQmQoMpCpKrFeSU#c mO4]95MQR5RVZ5Z[__abcHHU5#a% &8HD NN537U?q2H+JKaEQqSMKjoruxyry}BsBGHrHkJNKefcRW[\]^[^R_N`NKLNNuRy9!TT1H_9I*%I/ I4I9I> c tjj}|ddg|ddgd}|dj}t |djdzd z}d t |d j z}dD]E}t|D]5\} } | |kDr t| |} |rt| d \} } t| d \} } | | k(r'|j| | g| d z | d z g||d d|nw|j| dg| d z | d z g||d d||jd| g| d z | d z g||d d|n*tj| | g|dz |dz g||d dd||d |D cgc]} | jc} vs|jtj||d ||d 8H|Scc} w)NuStrong El Niñor"uStrong La Niña)z strong ninoz strong ninar,r5r/riUr,g(\?rg?)r#r1rg ?g/nbg?rr.)rjcmbwrrZrrRrr rYdivmodplot get_labelrr|r})rrrrr7rr4rrx1x2y1y2rs r8rras 66::D"3i):;TRSWVgLh iF  " & & (C x(,,.4: ;E]6299;<#JaAKKM#JJ%%hnn6!9Qrs|  0%#& %'XX Up(0G(T0B0<~/d3#l r: