*j8@dZddlZddlZddlZddlmZmZmZeje jjZ e jZedz Zedz ZdZd#deded efd Zd adadad agadadad ad adadagaga ga!ga"da#da$da%da&da'd a(da)da*d a+d a,d a-da.da/da0da1da2d a3da4da5da6da7ga8ga9ia:ia;iaia?ia@daAdaBdaCgaDeEaFiaGdaHdaIded efdZJd$dZKdZLdeded efd ZMd%ded!ed efd"ZN eKy#eO$rYywxYw)&u config.py — Carrega configuracao de config.yaml e variables.yaml Os arquivos YAML ficam na raiz do projeto (um nivel acima de scripts/). Para usar arquivos alternativos, chame init_config() antes de importar qualquer outro modulo do projeto: import config config.init_config("/caminho/para/meu_config.yaml", "/caminho/para/minhas_variaveis.yaml") Ordem de busca dos arquivos padrao: 1. Variavel de ambiente CONFIG_FILE / VARIABLES_FILE 2. /config.yaml e /variables.yaml N)datetime timedeltadatez config.yamlzvariables.yamlc  ddl}tj|}|j st d|dt |d5}|j|cdddS#t$r tdwxYw#1swYyxYw)zCCarrega um arquivo YAML; lanca FileNotFoundError com mensagem util.rNzYPyYAML e necessario para ler os arquivos de configuracao. Instale com: pip install pyyamlz(Arquivo de configuracao nao encontrado: z7 Crie o arquivo ou passe --config / --vars-file no CLI.zutf-8)encoding)yaml ImportErrorpathlibPathexistsFileNotFoundErroropen safe_load)pathrfs B/pesq/dados/sismom/SisMOM/sismom_fig/Figuras_Eta/scripts/config.py _load_yamlr"s  << D ;;=6tf=E F   dW %~~a  & %   .    & %sA"A:"A7:B run_inputref_datereturncHt|j}t|dk(r|jr|S|jrKt|dkr=|j d}|xst j }|jd|zStd|d)u^ Resolve o identificador da rodada para o formato completo YYYYMMDDHH. Formatos aceitos ---------------- "2026060600" -> "2026060600" (YYYYMMDDHH, 10 digitos) "00" -> hoje + "00" (HH, 2 digitos — usa data do sistema ou ref_date) "0" -> hoje + "00" (H, 1 digito — completado com zero) "12" -> hoje + "12" Parameters ---------- run_input : tag completo ou apenas a hora (HH) ref_date : data de referencia (padrao: hoje, datetime.date.today()) Raises ------ ValueError se o formato nao for reconhecido. z%Y%m%dzFormato de rodada invalido: 'z8'. Use YYYYMMDDHH (ex: 2026060600) ou HH (ex: 00 ou 12).) strstriplenisdigitzfill_datetodaystrftime ValueError)rrhourr s rresolve_run_tagr$9s(I$$&I 9~ 1 1 3s9~2??1%,u{{}~~h'$..  ' {3@ A gQ?Fg@xDz>f4z.binz%Y%m%d%HsaidalogsxpngDEFLATErtemplatec|jddjddjddjddS) aConverte placeholders {yyyy}{mm}{dd}{hh} para formato strftime. Exemplos: {yyyy}{mm}{dd}{hh} -> %Y%m%d%H (10 chars, padrao 2D) {yyyy}{dd}{hh} -> %Y%d%H (8 chars, 3D sem mes) {yyyy}{mm}{dd} -> %Y%m%d (8 chars, sem hora) {yyyy}z%Y{mm}z%m{dd}z%d{hh}z%H)replace)r/s r_template_to_strftimer6us<  WXt $ WVt $ WVt $ WVt $ &r%c |xs$tjjdt}|xs$tjjdt}t j |at j |att}ttjdg}|d}|xs=tjjddxst|jdd}|s tdt|a tttd d ttd d ttd d ttd d at|dat|dat't"Dcgc]}t t)|t$zzc}a|d}t|dat|dat1|dat1|dat1|da|jdxsg} |jdxsg} | D cgc] } t1| c} a| D cgc] } t1| c} at=t8at=t:a t>ret.dkDr0t1tCt8dt8dz t.dz z d nda"tGjHt8tFjJa&nRt1|jdxsda"tGjNt4t4t.dz tDzzt.a&t@r*tGjHt:tFjJa(n4tGjNt2t2t,dz t6zzt,a(|d} t1| d a)t| d!a*t| d"jWd#ta,t| d$a-t| jd%d&} t]| a/t=| jd'd(a0t=| jd)d(a1t=| jd*d(a2|d+}t|jd,dxs tjjd-da3t|jd.da4thrtkthtnda6t|jd/d0a7t|jd1d2a8|jd3i}t|jd4d5a9t|jd6d7a:|jd8i}t|jd9d:a;t|jd;dad?a>t=|jd@d(a?t=|jdAd(a@|aAtD cgc]} | dB c} aBtD cic]} | dB| jdC| dBc} aCtD cic]} | dB| jdDdc} aDd}iaEtD]:}|t|dB<|tt|jdEdxsddz }<|aGtD cic]%} | dBt| jdEdxsd'c} aHtD cic],} | dBt| jdEdxsddkDrdFnd=.c} aItD cic]!} | dBt| jdGg#c} aKtD cic]!} | dBt| jdHg#c} aLtD cgc]} | jdId(s| dBc} aMttaOtD cic]M} | dB| jdJdK| jdLd | jdMd | jdId(dNOc} aPy cc}wcc} wcc} wcc} wcc} wcc} wcc} wcc} wcc} wcc} wcc} wcc} w)Oa* Inicializa (ou reinicializa) todas as constantes a partir dos arquivos YAML. Chamado automaticamente na importacao do modulo com os arquivos padrao. Pode ser chamado novamente pelo main.py quando o usuario passa --config ou --vars-file, ANTES de importar reader, export_cog etc. Parameters ---------- config_file : str | Path | None Caminho para config.yaml (None = usa padrao ou env CONFIG_FILE) vars_file : str | Path | None Caminho para variables.yaml (None = usa padrao ou env VARIABLES_FILE) CONFIG_FILEVARIABLES_FILE variablesrunRUN_TAGr&tagzTag da rodada nao definido. Use --run YYYYMMDDHH ou --run HH (ex: --run 00 usa hoje 00Z), ou defina a variavel de ambiente RUN_TAG.Nrntimesdt_hours)hoursgridnxnylon0lat0dlon lat_levels lon_levelsr'rr()dtypedlatmodelundefrM file_prefix {run_tag} file_suffixfile_timestampz{yyyy}{mm}{dd}{hh}fixedFyrev sequentialpaths data_baseSISMOM_DATA_BASEdata_dir output_dirr)log_dirr*figuredpir+extr,cogcompressr-zlevel predictorr tile_sizer. fill_undeflon_180name descriptionunitsnlevlevels plot_levelsprecipcmapviridisvminvmax)rprrrsro)Qosenvironget_DEFAULT_CONFIG_DEFAULT_VARIABLESr r _CONFIG_FILE _VARS_FILErrr"r$r<rintT0NTIMESDT_HOURSranger TIMESTAMPSNXNYfloatLON0LAT0DLON LAT_LEVELS LON_LEVELSbool IRREGULAR_LAT IRREGULAR_LONroundDLATnparrayfloat64LATSlinspaceLONSUNDEFDTYPEr5 FILE_PREFIX FILE_SUFFIXr6FILE_TIMESTAMP_FMT FIXED_RUNYREV SEQUENTIALrZDATA_DIR_TEMPLATE_resolve_path_templateDATA_DIR OUTPUT_DIRLOG_DIRDPIFIG_EXT COG_COMPRESS COG_ZLEVEL COG_PREDICTOR COG_TILE_SIZECOG_FILL_UNDEF COG_LON180 VARIABLES VAR_NAMESVAR_DESC VAR_UNITS VAR_INDEXmax NVARS_FIELDSVAR_NLEVVAR_NDIMlist VAR_LEVELSVAR_PLOT_LEVELS PRECIP_VARSset PRECIP_SET CMAP_CONFIG) config_file vars_filerun_tagcfgvars_rawr;raw_tagig _lat_lev_raw _lon_lev_rawvm_ts_tplpfigra_offset_vs r init_configrs.@O !OKU0@BT!UI<< ,L<< *J,'C*%))+r:H e*C  # ::>>)R ( # swwub! "  8  w'GGBQK(#gal*;S1=NGAbM*,B3x=!F3z?#H>CFmLm"yq8|44mLJ v;A qw YYYYMMDDHH (run_tag completo) {run_tag} -> YYYYMMDDHH (alias de {data}) {yyyy} -> YYYY (ano 4 digitos) {mm} -> MM (mes 2 digitos) {dd} -> DD (dia 2 digitos) {hh} -> HH (hora 2 digitos) Exemplos -------- template = "/dados/sismom/sismom_forecast/{data}/regional/eta/2D" run_tag = "2026060600" -> "/dados/sismom/sismom_forecast/2026060600/regional/eta/2D" template = "/dados/sismom/sismom_forecast/{yyyy}{mm}{dd}{hh}/regional/eta/2D" -> "/dados/sismom/sismom_forecast/2026060600/regional/eta/2D" rz{data}rRr1Nr>r2r?r3r@r4)rr5)r/rs rrr's, 7c'lR&7 g & g & gbqk * gal + gal + gam ,r%basec2|r3d|vr t||Stjj||dddStrtt|St r'tjjt |dddStjjd|S)uk Resolve o caminho dos dados para uma rodada. Prioridade ---------- 1. --data_base CLI (base != "") — retrocompat ou template 2. data_dir no config.yaml — suporta template ou caminho fixo 3. SISMOM_DATA_BASE env — base fixa (retrocompat) 4. data/ local — fallback de desenvolvimento {regionaleta2Ddata)rrtrjoinrrZ)rrs rbuild_data_dirrJs{ $;)$8 8ww||D':udCC%&7AAww||,gz5$OO 77<< ((r%)N)NNN)r&)P__doc__rtr numpyrrrrrr __file__parentresolve SCRIPTS_DIR PROJECT_ROOTrwrxrrr$r<r|r}r~rrrrrrrrrrrrrrrrrrrrZrrrrrrrrrrrrrrrrrrrrrrrrrrrrryrzr6rrrr Exceptionr%rrs  77w||H%,,446 !! !M1!$44!.!s!e!s!J 4b!X:Qs3DtDT "4  b u}  ee2[V{ZdEW"x"&7   U7 q*!-S]   B 22y2y "x2jTU\    su*Bk $Z &C &C &[|D S 3 3 F)C)s)C)6 M   s;DD  D