o 9%j=@sJdZddlZddlZddlmZmZddlmZmZm 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 ddZddZ   d*deded e dededed ed!e d eeeeffd"d#Z  d+ded$ed e ded ed eeeeff 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 dddt t | t |Btj|| }Wdn1sdwYq |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). Nrrr)r,)dtypeignore)invalidover)rrrangerrreader read_fieldFileNotFoundErrornp zeros_likefloat32errstatewhereisnannan) r)r*r+r,dtaccstepr&tfieldrrrcompute_accumulations" &rAtifrextcCs |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 _ACUMZh_z%Y%m%d%H.)strftime)r*rr rCrrraccum_filenames rGcCsl|\}}}}}}}}ddl} tj||d|d} tj| ddt||d|d} tj| | } |rCtj| rC||d | ddfSt||||} | durW||d dd d fSzA| j| | || |t j ||d |d t |d t |dt |d|dddt jt | jd |d||d | dd fWSty}z||d dt |d fWYd}~Sd}~ww)z/Worker paralelo para gerar um COG de acumulado.rNrDrTexist_okrr rarquivo ausenteFmmr rrz%Y-%m-%dT%H:%M:%SZzEta03/BESM run ) variable descriptionunitsr%r rrrmodelZnodata)metadata overviews) export_cogospathjoinmakedirsrGexistsrAZ write_cogZ_prepare_arrayrVAR_DESCgetstrrFRUN_TAGZNODATA Exception)argsr)varwincog_dirr r,rR skip_existingZecogvar_dirfnamefpatharrerrr_accum_cog_workersB        "rhTr rarRrbverboseworkersc shddlm}m} tjddtdddtjD} fdd tjD} t| } |rmt d d td ttjd | d| D] } t d| ddd| ddd| ddd| d dqL|dkr| D]P}t |\}}}}}|r|dkr|rt d|d|d|qs|dkr|rt d|d|d|qs| | ||rt d|rd nd!dtj |qs| S||d"^fd#d| D}| |D]G}|\}}}}}|r|r|dkrd$nd%}t d|d|d|d|q| | ||r t d|rd nd!dtj |qWd&| S1s-wY| S)'aS 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. workers : processos paralelos (1 = serial) r)ProcessPoolExecutor as_completedTrHrJcSi|]}|gqSrr.0vrrr $z3export_all_accumulations_as_cog..c s*g|]}D] }||fqqSrr)ror_r`)r rar)rRr,rbr$rr &sz3export_all_accumulations_as_cog..[accum] Periodo: h | janelas x z variaveis = z acumulados | workers=z rZ10sz FH r3d-rz val. rz %Y%m%d %HZr rK [ERRO]  : [AVISO] ZSKIPzOK ) max_workerscsi|] }t||qSr)submitrh)ror?)poolrrrqCsz[AVISO]z[ERRO]N)concurrent.futuresrkrlrTrWr(r PRECIP_VARSlenprintrFrhr"rUbasenameresult)r)rar r,rRrbrirjrkrlsavedtasksn_totalrtaskr_ZwtypereerrZskippedfutsfuttagr)r rar)rRrr,rbr$rexport_all_accumulations_as_cog sl  (    $   &   r output_dirc Csddl}tj|ddt|d}ddtjD}dd d d d d |d dddd d d |dddddd d |dd}|rQtd|dt|dttjdtjD]} tj || d|d} tj| dd|D]} t | | d|tj d} tj | | } t || | |}|dur|rtd| d| ddqlz:t|| tr|| i|d n|| d }|j| || d|| d| |d || | |rtd!| Wqlty}z|rtd"| d| dd#|WYd}~qld}~wwqT|S)$z Gera figuras PNG dos acumulados de precipitacao. Parameters ---------- accum_hours : periodo do acumulo em horas (padrao: 24) rNTrHrJcSrmrrrnrrrrqdrrz*plot_all_accumulations..Pxi,i)rr0H2(<d)ZPRECZPRCVZPRGErtrurvz variaveisrDrr)r rCr|rzrz: arquivo ausente)r*datarr r%Z output_pathvmaxz OK ryr{)Z plot_utilsrTrWr(rrrZrrrUrVrGFIG_EXTrA isinstancedictZplot_accumulationr"r])r)rr r,riZpur$rZ vmax_tabler_rcr`rdrerfrrgrrrplot_all_accumulationsRs^   0 r)NNNr)F)rrB)rFFFTr )rFT)__doc__rTnumpyr5rrtypingrrrrr2intr(r[boolndarrayrArGrhrrrrrrs* o / + I