o F+$jC@sPdZddlZddlZddlZddlZddlmZmZddlm Z ddl m Z m Z z ddl Z ddlmZddlmZddlmZddlmZd ZWney[d Zed ed YnwddlZddlZddlZd ZdZgdZdZ dZ!dZ"e#dZ$ej%ej&dej'Z(ej)Z*ee*e(ej+ej'Z,hdZ-dej.de/dej.fddZ0  d8dej.de/de e1de2de/f dd Z3 ! d9dej.de/d"ed#e/d$e/de2de/fd%d&Z4 d:d'e/de/d#e/d(e2d)e2de5f d*d+Z6 d;d'e/d,e/d(e2d)e2de2d-e2de1fd.d/Z7d0d1Z8   d= 1.3 pip install rasterio Estrutura de saida: cog/ ├── TP2M/ │ ├── TP2M_2026060400.tif │ ├── TP2M_2026060401.tif │ └── ... ├── PREC/ │ ├── PREC_2026060406.tif │ └── acumulado_24h/ │ └── PREC_2026060500_acum24h.tif └── ... Valores: - Precipitacao (PREC, PRCV, PRGE, NEVE): convertidos de metros para mm - UNDEF do modelo substituido por NaN (nodata no GeoTIFF = -9999.0) - Arrays armazenados como float32 N)datetime timedelta)Optional)ProcessPoolExecutor as_completed) from_origin)CRS) ResamplingTFz,[export_cog] AVISO: rasterio nao encontrado.z0[export_cog] Instale com: pip install rasteriogi) ZDEFLATEr i>PRGEZNEVEPRCVPRECdatavar_namereturncCsD|tj}|tvr|d}t|}tt|tt|}|S)z - Converte m->mm para precipitacao - Flipa verticalmente (GrADS: S->N; rasterio: N->S) - Substitui NaN por NODATA - Retorna float32 g@@)astypenpfloat32 _PRECIP_VARSflipudwhereisnanNODATA)rrarrr A/dados/sismom/SisMOM/sismom_fig/Figuras_Eta/scripts/export_cog.py_prepare_arraySs  r"rfpathmetadata overviewscCststdtjtjtj|ddtj|tj d}ddt j t j dt tttttdttdd }|shtj|d fi|!}||d|rV|jdi|Wd |SWd |S1sawY|Stjd d d }|j}Wd n1s}wYzjtj|d fi|}||d|r|jdi|Wd n1swYt|d}|ttj|jdddWd n1swYtj ||dttdttdd Wtj!|rt"||Stj!|rt"|ww)a? Escreve arr (NY, NX) float32 como GeoTIFF tiled georeferenciado. Parameters ---------- arr : array (NY, NX) float32, preparado por _prepare_array fpath : caminho do arquivo .tif de saida metadata : tags GDAL a gravar nos metadados overviews : se True, embute overviews (piramides) no arquivo via rasterio.shutil.copy. ATENCAO: alguns visualizadores (ex: SisMOM) exibem cada nivel de overview como uma camada separada. Use False (padrao) para compatibilidade maxima. zrasterio e necessario.Texist_ok)dtypeZGTiffrrZIF_SAFER)driverr(widthZheightcountcrsZ transformnodatacompressZzlevel predictortiled blockxsize blockysizeZBIGTIFFwN.tifF)suffixdeletezr+Z rio_overviewaverage)nsZ resampling)Zcopy_src_overviewsr.r/r0r1r2r)r )# HAS_RASTERIO ImportErrorosmakedirspathdirnameabspathrascontiguousarrayrconfigNXNY CRS_WGS84 TRANSFORMrCOMPRESSZLEVEL PREDICTOR TILE_SIZErasterioopenwriteZ update_tagstempfileNamedTemporaryFilenameZbuild_overviewsOVERVIEW_LEVELSr r7 rio_shutilcopyexistsremove)rr#r$r%profiledsttmpZtmp_pathr r r! write_cogis|      rX timestampcog_dir title_extrac Cst||}tj|d}|tvrd}|tj||||ddtjtt dd}|r0||d<|r=d| d d nd} |d|d | d } t j || } t|| ||d S) a` Exporta um campo 2D como COG GeoTIFF. Parameters ---------- data : array (NY, NX) lido pelo reader var_name : nome da variavel timestamp : datetime do campo cog_dir : diretorio de saida title_extra: sufixo no nome do arquivo (ex: "acum24h") Returns ------- Caminho do arquivo .tif criado. rYmmz%Y-%m-%dT%H:%M:%SZzEta03/BESM run z EPSG:4326)variable descriptionunitsrZmodelr-r,r\_ %Y%m%d%Hr4)r$r%)r"rA VAR_UNITSgetrVAR_DESCstrftimeRUN_TAGstrrreplacelowerr;r=joinrX) rrrZr[r\r%rr`metaZ extra_tagfnamer#r r r!export_field_as_cogs$    rpdata_dir sequentialverbosec Cst|}tj|ddg}|D]O}z)tj||||d}t||||} || |r:td|d|dd| Wqt y_} z|rUtd|d|dd | WYd } ~ qd } ~ ww|S) z Exporta todos os timesteps disponiveis de uma variavel como COG GeoTIFF. Returns ------- Lista de caminhos criados. Tr&rrz [COG] rcrdz -> z [COG] ERRO : N) readerlist_available_timestampsr;r< read_fieldrpappendprintrh Exception) rqrr[rrrs timestampssavedtrr#er r r!export_var_all_timestepss"    r cog_base_dir skip_existingcCstj||||||dS)z Wrapper de compatibilidade -- delega para accumulate.export_all_accumulations_as_cog(). Nomenclatura atual: PREC_ACUM24h_2026060500.tif Janelas ACUM00Z e ACUM12Z calculadas a partir do horario do run (config.T0). )rqr[rrr%rrs) accumulateexport_all_accumulations_as_cog)rqrrrrsr%rr r r!#export_all_24h_accumulations_as_cogsrc sD|\}}}}}}g}d}g} |D]&} tj|| d|d} |r5tj| r5|| | dfq| | q| s?|Sz tj||d} Wntygz|fdd| DWYdSdww| D]5} zt | | | ||d} || | dfWqjtyz|| dt fWYdqjdww|S) a5 Worker por TIMESTEP: le o arquivo .bin UMA vez e gera TODOS os COGs das variaveis solicitadas. Paralelismo correto = 1 task por arquivo. args: (data_dir, timestamp, vars_list, cog_dir, sequential, overviews, skip_existing) returns: list of (var, timestamp, fpath_or_None, error_or_None) rdrbr4Nrtcsg|] }|ddfqS)Nz leitura: r .0v_trr r! Ssz(_worker_cog_timestep..)r%) rhr;r=rmrSryrvread_all_fieldsr{rprj) argsZ _data_dirZ_varsZ_out_dirZ_seqZ_ovrZ_skipresultsts_strZ vars_neededvarr#fieldsr rr!_worker_cog_timestep7s4  $"rvars_to_exportworkersc s ptj t}|stddiSr2tdt|dt dt|t d|tjdd fd d |D d d Dd d d d t  fdd} |dkrt |dfdd  D} t | D]} | | q~Wdn1swYn D]} | t | qt } | d kr| nd }rtddd| dd|dd S)a  Exporta todos os campos de todas as variaveis como COG GeoTIFF. Parameters ---------- data_dir : diretorio com os .bin cog_base_dir : diretorio raiz de saida dos COGs vars_to_export: lista de variaveis (None = todas) sequential : True se arquivos .bin usam marcadores Fortran workers : processos paralelos (1 = serial) verbose : exibir progresso skip_existing : pula arquivos ja existentes Returns ------- dict {var_name: [lista de caminhos]} z+[export_cog] Nenhum arquivo encontrado em ''z [export_cog] z timesteps x z variaveis = z COGs a gerar | workers=Tr&c sg|] }|fqSr r )rr~)rrqr%rrrrr r!rsz,export_all_fields_as_cog..cSsi|]}|gqSr r rr r r! sz,export_all_fields_as_cog..rcsd7td}|D]*\}}}}|r-d7r,td|d|dd|q||d7qrst}rM|tnd}tdd d td |d d dd |dd|dddddSdS)Nrdz [ERRO] rcrdrurz [3d/z z5.1fz%] ok=z err=z.0fzs ETA=sT)flush)lenrzrhrytime)res_listpctrtsr#errelapsedeta)donen_errn_okn_skipr}t0tasksrsr r!_processs4    z*export_all_fields_as_cog.._processr) max_workerscsi|] }t||qSr )submitr)rtask)poolr r!rsNz [export_cog] z COGs gerados, z erros | z.1fzs (z COG/s)) rA VAR_NAMESrvrwrzrr;r<rrrresultr)rqrrrrrrsr%rr|rfutsfutrrspeedr )rrqrrrrr%rr}rrrrrrrsr!export_all_fields_as_cog`sZ     rcCsxtsdSz-t|}t|ddk}|jdd}|o|WdWS1s*wYWdSty;YdSw)z Verificacao basica: abre o arquivo e confirma que tem overviews e tiles. Para validacao completa use: python -m cogdumper ou rio cogeo validate. Frrr0N)r9rJrKrr%rUrfr{)r#srcZ has_overviewsZis_tiledr r r! validate_cogs ( r)NF)rYF)FT)FTFF)NFrTFF)<__doc__r;rrMnumpyrrrtypingrconcurrent.futuresrrrJZrasterio.shutilshutilrQZrasterio.transformrZ rasterio.crsrZrasterio.enumsr r9r:rzrArvrrrIrPrFrGrHZ from_epsgrDLAT0rCDLATZ_UL_LATLON0Z_UL_LONDLONrErndarrayrjr"dictboolrXrplistrrrintrrr r r r!s         a 4 ) , f