o C$j@s8dZddlZddlZddlZedddlmZddl m Z ddl m ZddlmZzddlmZddlmZdZWneyJdZedYnwddlZdd ZeZd eiZd d Zd efddZd#ddZ d#ddZ!d#ddZ"      d$dej#d edededededefddZ$d ej#dej#fd!d"Z%dS)%uA plot_utils.py — Utilitários comuns de plot para o modelo Eta NZAgg)datetimeTFu@[plot_utils] cartopy não encontrado — usando matplotlib puro.cCsgd}tjd|S)u2Colormap branco→azul escuro para precipitação.))?rr)?g(\?r)gRQ?rr)gQ?gGz?)rg(\?r)r皙?333333?)333333?rg?precip)mcolorsZLinearSegmentedColormap from_list)colorsr A/dados/sismom/SisMOM/sismom_fig/Figuras_Eta/scripts/plot_utils.py _precip_cmaps rr cCst|tr t||S|S)zNConverte string de colormap (incluindo nomes proprios) para objeto matplotlib.) isinstancestr _CUSTOM_CMAPSget)Z cmap_namer r r _resolve_cmap5s  rvar_namecCs"tj|d\}}}t|||fS)z Retorna (cmap, vmin, vmax) para a variavel a partir de config.CMAP_CONFIG (carregado de variables.yaml). O cmap retornado pode ser string ou objeto matplotlib colormap. )viridisNN)config CMAP_CONFIGrr)rZraw_cmapvminvmaxr r rget_cmap_config<srocCst}|j||d}|jtjdddd|jtjdddd|jtjddd d|j d d d d dd}d|_ d|_ t d|_t d|_|S)u/Cria eixo com projeção PlateCarree (cartopy).)Z projectionZ50mrk) linewidthcolorg?z0.4rz0.6Tr0.7gffffff?--)Z draw_labelsrralpha linestyleF)ccrs PlateCarree add_subplotZ add_featurecfeatureZ COASTLINEZ with_scaleZBORDERSZSTATESZ gridlinesZ top_labelsZ right_labelsmtickerZ MaxNLocatorZxlocatorZylocator)figrectprojaxglr r r_setup_axes_cartopyKs   r/cCs*||}|d|jddddd|S)zCria eixo simples sem cartopy.equalTrr r!)rrr#)r'Z set_aspectgrid)r*r+r-r r r_setup_axes_plain[s  r2cCstrt||St||S)N) HAS_CARTOPYr/r2)r*r+r r r setup_axescs  r4data timestamp output_dir title_extraunits_overridereturnc  Cs^tj|dd|tj} |ptj|d} tj ||} | dur'| | } t |\} }}|dur4|} |dur:|}|dur@|}| t | }|durY|j rWt t|dnd}|durj|j rht t|dnd}||krr|d}tjd d }t|}tr|jtjtj| | ||td d }|jtjdtjd tjdtjd gtdn|jtjtj| | ||d d}|d|d|j||dddd}|j| dd|d}|rd|nd}|j|d| |d|ddd|r|dd nd}|rd|nd}|d|d |d!tj!}tj"#||}tj$|tj%d"d#t&||S)$u[ Plota um campo 2D e salva como imagem. Parameters ---------- data : array (NY, NX) var_name : nome da variável (config.VAR_NAMES) timestamp : datetime do campo output_dir : diretório de saída title_extra : texto adicional no título (ex: "Acumulado 24h") units_override: sobrescreve a unidade do config vmin/vmax_override: sobrescreve os limites do colormap cmap_override : sobrescreve o colormap convert_fn : função de conversão aplicada aos dados (ex: m→mm) Returns ------- Caminho do arquivo salvo. T)exist_okr5Nrb) )Zfigsizeauto)cmaprr transformshading)crs)rCrrrEuLongitude (°)u Latitude (°)Zverticalg{Gz?gQ?)r-Z orientationpadfraction )fontsizez %d/%m/%Y %HZu —  rA)rKrH _z%Y%m%d%H.Ztight)dpiZ bbox_inches)'osmakedirscopyastypenpfloat64r VAR_UNITSrVAR_DESCrisnansizefloat percentilepltfigurer4r3Z pcolormeshLONSLATSr%r&Z set_extentZ set_xlabelZ set_ylabelZcolorbarZ set_labelstrftimeZ set_titlereplacelowerFIG_EXTpathjoinZsavefigDPIclose)r6rr7r8r9r: vmin_override vmax_overrideZ cmap_override convert_fnarrunitsdescrCrrvalidr*r-ZimcbZtime_strextra extra_tagfnamefpathr r r plot_fieldmsh  "    $  rvrmcCs|dS)uDConverte metros → milímetros (para variáveis de precipitação).g@@r )rmr r rm_to_mmsrw)r)r5NNNNN)&__doc__rRnumpyrVZ matplotlibZuseZmatplotlib.pyplotZpyplotr^Zmatplotlib.colorsr r Zmatplotlib.tickerZtickerr)rZ cartopy.crsrGr%Zcartopy.featurefeaturer(r3 ImportErrorprintrrZ _PRECIP_CMAPrrrrr/r2r4ndarrayrvrwr r r rsb             f