o C$jRB@sdZddlZddlZddlmZmZddlmZmZm Z ddl Z ddl Z    d(dede de d e d eef d d Z d)dedededed e ejf ddZ  d*deded e ded ef ddZ  d+deded e dedededed eeeeffddZ  d,ded ed e deded eeeeff d!d"Z d-ded ededed eeeeff d#d$Z d)deded%eded ejf d&d'ZdS).a accumulate.py -- Acumulados de precipitacao para PREC, PRCV e PRGE Logica de acumulo: ------------------ O modelo tem saida HORARIA. A precipitacao NAO e definida na analise (hora 0), portanto o acumulo NUNCA inclui o timestep 0. Frequencia de acumulo configuravel via parametro accum_hours (padrao: 24). Para accum_hours = 24 (convencao meteorologica): Duas janelas sao geradas em paralelo: ACUM00Z: acumula de 00Z a 00Z (validade em horario 00Z) ACUM12Z: acumula de 12Z a 12Z (validade em horario 12Z) Para run iniciando em 00Z (ex: 2026060400): ACUM00Z: soma FH 1-24 -> val. 2026060500 (00Z dia seguinte) soma FH 25-48 -> val. 2026060600 ... ACUM12Z: soma FH 13-36 -> val. 2026060512 soma FH 37-60 -> val. 2026060612 ... Janela FH 109-132 seria o 5o ciclo, mas so ha FH ate 120 -> janela incompleta -> DESCARTADA Para run iniciando em 12Z (ex: 2026060412): ACUM12Z: soma FH 1-24 -> val. 2026060512 ACUM00Z: soma FH 13-36 -> val. 2026060600 ... Para outros periodos (ex: accum_hours = 6, 12, 48, 72): Janelas sequenciais comecando no FH dt_hours (primeiro FH disponivel): FH 1-6 -> val. YYYYMMDDHH (accum_hours=6) FH 7-12 -> val. YYYYMMDDHH ... Nomenclatura dos arquivos: PREC_ACUM24h_2026060500.tif -- acumulo de 24h, validade 2026-06-05 00Z PREC_ACUM12h_2026060512.tif -- acumulo de 12h, validade 2026-06-05 12Z PREC_ACUM6h_2026060506.tif -- acumulo de 6h, validade 2026-06-05 06Z N)datetime timedelta)ListDictOptionalt0ntimesdt_hours accum_hoursreturnc Cs|dur|ntj}|dur|ntj}|dur|ntj}||dkr/td|d|d|d|d|}||}|dkr|j}|dkrOd |fd d |fg}n|d kr^d |fd d |fg}n d |fd d |fg}g}|D]9\} } | } | |d||kr| |d|} |t| d } || | | | ||d | d7} | |d||ks|ql|jddd|Sd|d}g}|} | |d||kr| |d|} |t| d } ||| | | ||d | |7} | |d||ks|S)u Determina todas as janelas de acumulo validas para o run. Para accum_hours=24: gera janelas ACUM00Z e ACUM12Z (convencao sinótica). Para outros periodos: gera janelas sequenciais comecando no FH dt_hours. Janelas incompletas (menos de accum_hours/dt_hours arquivos disponiveis) sao automaticamente descartadas. Parameters ---------- t0 : inicio do run (padrao: config.T0) ntimes : numero de timesteps disponiveis (padrao: config.NTIMES) dt_hours : intervalo de saida do modelo em horas (padrao: config.DT_HOURS) accum_hours : periodo de acumulo em horas (padrao: 24) Deve ser multiplo de dt_hours. Exemplos: 6, 12, 24, 48, 72 Returns ------- Lista de dicts ordenada por (validade, tipo): { 'type' : str -- 'ACUM00Z' | 'ACUM12Z' | 'ACUM{N}h' 'start_fh' : int -- primeiro FH incluido (em horas) 'end_fh' : int -- ultimo FH incluido (em horas) 'validity' : datetime 'n_steps' : int -- numero de arquivos somados 'accum_hours': int -- periodo do acumulo em horas } Nrz accum_hours=z nao e multiplo de dt_hours=z. Escolha um valor multiplo de zh.rZACUM00ZZACUM12Z hours)typestart_fhend_fhvalidityn_stepsr cSs|d|dfS)Nrr)wrrA/dados/sismom/SisMOM/sismom_fig/Figuras_Eta/scripts/accumulate.pysz*get_accumulation_windows..)keyZACUMh) configT0NTIMESDT_HOURS ValueErrorhourrappendsort)rr r r Zmax_fhrZrun_hourZ cycle_startswindows accum_typerfhrrlabelrrrget_accumulation_windows8sx$         r(Fdata_dirvar_namewindow sequentialc Cstj}d}t|dD]]}|d||}tjt|d}z tj||||d} Wn ty3YdSw|dur@tj | tj d}tj ddt t | t |Btj|| }Wdn1scwYq |S) a Soma os campos de precipitacao de todos os FH dentro de uma janela. Parameters ---------- data_dir : diretorio com os arquivos .bin var_name : nome da variavel (PREC, PRCV ou PRGE) window : dict retornado por get_accumulation_windows() sequential : True se .bin usa marcadores Fortran Returns ------- np.ndarray (NY, NX) em metros (unidade original), ou None se qualquer arquivo estiver ausente (janela incompleta -> descartada). Nrrrr,)dtypeignoreinvalid)rrrangerrreader read_fieldFileNotFoundErrornp zeros_likefloat32errstatewhereisnannan) r)r*r+r,dtaccstepr&tfieldrrrcompute_accumulations" &rBtifrextcCs |d|d|dd|S)z Gera o nome do arquivo de acumulado. Exemplos: PREC_ACUM24h_2026060500.tif PREC_ACUM12h_2026060512.tif PREC_ACUM6h_2026060506.tif Z_ACUMZh_%Y%m%d%H.)strftime)r*rr rDrrraccum_filenames rHTcog_dir overviews skip_existingverbosecCsFddl}tj|ddt|d}ddtjD} |rZtd|d t|d ttjd t|ttjd |D] } td | ddd| ddd| ddd| ddq9tjD]} |D]} t | | d|d} tj || }|rtj |r|rtd| | |  |qat|| | |}|dur|rtd| d| dd| dd| dd qazI|j||| || tj| | d| dt| dt| dt| d| ddd tjt|jd! |d"| |  ||rtd#| Wqaty}z|rtd$| d| dd%|WYd}~qad}~wwq]| S)&a  Calcula e exporta todos os acumulados de PREC, PRCV e PRGE como COG GeoTIFF. Parameters ---------- accum_hours : periodo do acumulo em horas (padrao: 24) Para 24h: gera ACUM00Z e ACUM12Z. Para outros valores: janelas sequenciais. rNTexist_okr cSi|]}|gqSrr.0vrrr z3export_all_accumulations_as_cog..z[accum] Periodo: zh | z janelas x z variaveis = z acumuladosz rZ10sz FH r3d-r val. rz %Y%m%d %HZz SKIP z [AVISO]  z FHz: arquivo ausente -> descartadommr z%Y-%m-%dT%H:%M:%SZzEta03/BESM run ) variable descriptionunitsr%r rrrmodelZnodata)metadatarJz OK z [ERRO] : ) export_cogosmakedirsr(r PRECIP_VARSprintlenrGrHpathjoinexistsr"rBZ write_cogZ_prepare_arrayVAR_DESCgetstrRUN_TAGZNODATA Exception)r)rIr r,rJrKrLZecogr$savedrvarwinfnamefpatharrerrrexport_all_accumulations_as_cogsx   (           +rv output_dircCsddl}tj|ddt|d}ddtjD}dd d d d d dddd d d dddd dd d}tjD]} || i||d} |D]} t|| | |} | durQqCt| | d|tj d} tj || }d|d| dd| ddd| ddd| d d d! }zN|j | | | d|||jd"d| d# tj || d$| d d%d&tj }tj |r||krt||tj |r|| ||rtd'| WqCty}z|rtd(| d)| dd*|WYd}~qCd}~wwq3|S)+z Gera figuras PNG dos acumulados de precipitacao. Parameters ---------- accum_hours : periodo do acumulo em horas (padrao: 24) rNTrMrOcSrPrrrQrrrrTIrUz*plot_all_accumulations..Pxi,i)rr0H<()PRECPRCVPRGEr)r rDzAcum. zh (rz) [FH rZ03drWrrXz %d/%m %HZ]rZ) title_extraZ convert_fnZunits_overrideZ vmin_overrideZ vmax_override_rErFz [PNG accum] z [ERRO PNG accum] rYr`)Z plot_utilsrbrcr(rrdrkrBrHFIG_EXTrgrhrGZ plot_fieldZm_to_mmrirenamer"rern)r)rwr r,rLZpur$roZ vmax_tablerpvmaxrqrtrrrsrautorurrrplot_all_accumulations7sj        'rcCst||d||dS)zIAlias de compatibilidade -- chama plot_all_accumulations(accum_hours=24).r)r)rwr r,rL)r)r)rwr,rLrrrplot_all_24h_accumulationssrt_endc sfdddD}d}|D]<}tj||||d}|dur!t|}tjddtt|t|Btj||}Wdn1sDwYq |S)zHLegado: acumulado 24h terminando em t_end (4 x 6h). Mantido para compat.csg|] }t|dqS)r)r)rRrrrr sz,compute_24h_accumulation..)rr{rNr-r/r0)r3r4r6r7r9r:r;r<)r)r*rr,Z windows_4x6r>r@rArrrcompute_24h_accumulations &r)NNNr)F)rrC)rFFFT)rFT)FT)__doc__rbnumpyr6rrtypingrrrrr3intr(rlboolndarrayrBrHrvrrrrrrrs* o /  V L