+ *jveRt^RIt^RIt^RIt^RIt^RIHtHt^RIH t ^RI H t H t ^RI t ^RIHt^RIHt^RIHt^RIHtRt^RIt^RIt^RItR$t.R%OtR t]P@!R 4t!RRlt"Rt#0R&mt$RRlt%RRlt&RRlt'R'RRllt(R(RRllt)R)RRllt*R*RRllt+Rt,R+R R!llt-R"R#lt.R# ]dR t]!R 4]!R 4Li;i),u: export_cog.py — Exportacao de campos do modelo Eta como Cloud Optimized GeoTIFF (COG) Cada campo e exportado como um arquivo .tif georeferenciado em WGS84 (EPSG:4326), com compressao DEFLATE, tiles 512x512 e overviews (piramides) para conformidade COG. Dependencia: rasterio >= 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 rasterioc\\RR4\\R^4\\R^4\\RR43#)zERetorna (compress, zlevel, predictor, tile_size) do config carregado. COG_COMPRESSDEFLATE COG_ZLEVEL COG_PREDICTOR COG_TILE_SIZEi)getattrconfigF/pesq/dados/sismom/SisMOM/sismom_fig/Figuras_Eta/scripts/export_cog.py _cog_paramsr=s> 3 ++-  ric$V^8dQhR\/#)return)int)formats"r __annotate__rIsFFSFrc\\RR4'g^#\\RR4'd^#\P\P,p\ VR, 4\P8d^#\P \P^, \P,,pVR8:d^#\P \ P!\P4\P,,p\\ P!VRRR74\P,#) zNumero de colunas a rolar para converter grade global 0-360 em -180..180. Retorna 0 quando cog.lon_180 esta desligado, a grade nao e global em lon, ou ja esta na convencao -180..180. k = indice do primeiro centro >= 180. COG_LON180F IRREGULAR_LONv@gf@left)sidegf@) rrNXDLONabsLON0nparanger searchsorted)span last_centerlonss r _lon_roll_kr,Is 6< / /v.. 99v{{ "D 4%<6;;&++Q&++ ==Kl" ;;699- ; ;D rtU8 9FII EErc\4pV'dQ\PV\P,,R, \PR, , pM,\P\PR, , p\ \RR4'd\ \P 4^8d\\P R,\P R,, 4p\\P R,4VR, ,p\\P R,\P ^,, \P^, , 4pMQ\P\PR, \P,,p\Pp\W\PV4#)aI Retorna o Affine transform rasterio para a grade atual. GrADS LON0/LAT0 sao centros de celula; from_origin espera o canto (corner) superior esquerdo do primeiro pixel. Correcao de meio pixel: ul_lon = LON0 - DLON/2 ul_lat = lat_max + dlat_top/2 Elimina o offset sistematico de ~DLON/2 visivel em grades esparsas (ex BESM). rg@ IRREGULAR_LATFg?) r,rr%r#rlenLATSfloatNYLAT0DLATr)kul_lon top_spacingul_latdlats r_get_transformr<]s  A++FKK/586;;;LLv{{S00v..3v{{3Ca3GFKKOfkk"o=> v{{2'+*;; B&++a.8VYY]KL C6;;>> vv{{D 99rcXV^8dQhR\PR\P/#rarrrr&ndarray)rs"rrrs"BJJ2::rc \Pp\P!V^,VR,\ V44p^RIHpV!W^RR\PR7pV!V4P\P4# \dd\P!T4p\TP^,4F(p\P!Y!TRT3,4TRT3&K* Tu#i;i)u Reamostrage de grade irregular (ex.: gaussiana) para grade regular. Interpola ao longo do eixo lat (eixo 0) usando os valores de config.LATS. Entrada / saída: (NY, NX) float32, S->N. )interp1dlinearF)axiskind bounds_error fill_value:NNNr/)rr2r&linspacer1scipy.interpolaterCnanastypefloat32 ImportError empty_likerangeshapeinterp)r? lats_irreglats_regrCfoutjs& r_regrid_to_regularrXs JZ]JrNC OLH .zQX$)bff>{!!"**-- mmC syy|$A (AqD BC1I% sABA+C54C5cXV^8dQhR\PR\P/#r>r@)rs"rrrs"RZZBJJrc f\P!V4pVP4'dVP4'dV#^RIHpV!VRRR7pV\ V4,# \Ed<^RIpTP4p\P!T4P4'd\P!T4p\P!T^\PR7p\P!TRR 1^R 13,TR^R 13,T^R 1RR 13,T^R 1R3,.4pTP4;_uu_4TPR\4\P !T^R 7p RRR4M +'giM;iX T,YV&EK"Tu#i;i) zPreenche NaN com o valor valido mais proximo (para visualizacao continua). Usa scipy.ndimage.distance_transform_edt quando disponivel; senao, dilata iterativamente com a media dos vizinhos validos (grades pequenas). )distance_transform_edtFT)return_distancesreturn_indicesN)constant_values:rNNignorerEr0r/)r&isnananyall scipy.ndimager[tuplerNwarningscopypadrKstackcatch_warnings simplefilterRuntimeWarningnanmean) r?maskr[idxrfrVmprinbs & r_fill_undef_nearestrssF 88C=D 88:: 8$TERVW5:  hhjhhsm!! AsArvv6AHHaQrT lAb!B$hK1R4"9q2r{STE((**%%h?ZZA.+***UCF  s2"A))A F03BF0/F  F0 F F0/F0cdV^8dQhR\PR\R\P/#)rdatavar_namer)r&rAstr)rs"rrrs)   s rzz rcvVP\P4pV\9d3\P!RRR7;_uu_4VR,pRRR4\ \ RR4'd \V4p\ \ RR4'd \V4p\P!V4p\4pV'd\P!W#)^R7p\P!\P!V4\P!\4V4pV# +'giL;i) u - Converte m->mm para precipitacao - Reamostrage para grade regular quando grade Y é irregular (gaussiana) - cog.fill_undef: preenche undef com vizinho mais proximo (sem mascara) - Flipa verticalmente (GrADS: S->N; rasterio: N->S) - Substitui NaN por NODATA - Retorna float32 r_)overinvalidg@@Nr.FCOG_FILL_UNDEFr`)rLr&rM _PRECIP_VARSerrstaterrrXrsflipudr,rollwhereraNODATA)rurvr?r7s&& r_prepare_arrayrs ++bjj !C< [[h 9 9,C:v.. %v'//!#& ))C.C  Aggc2A& ((288C="**V"4c :C J): 9s D(( D8 c ~V^8dQhR\PR\R\\,R\ R\/#)rr?fpathmetadata overviewsr)r&rArwrdictbool)rs"rrrsIVV V VtnV V  VrcZ\'g \R4h\P!\PP \PP V44RR7\P!V\PR7p\4wrErgRRRRR \PR \PR ^R \R \4R\ RVRVRVRRRVRVRR/pV'gX\"P$!VR3/VB;_uu_4p V P'V^4V'dV P(!R!/VBRRR4V#\*P,!RRR7;_uu_4p V P.p RRR4\"P$!X R3/VB;_uu_4p V P'V^4V'dV P(!R!/VBRRR4\"P$!V R4;_uu_4p V P1\2\4P64V P)RRR7RRR4\8P:!WR\<\>R\@\@RR 7 \PPCV 4'd\PD!V 4V# +'giT#;i +'giELE;i +'giEL;i +'giL;i \PPCX 4'd\PD!T 4ii;i)"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)dtypedriverGTiffrrMwidthheightcountcrs transformnodatacompresszlevel predictortiled blockxsize blockysizeBIGTIFFIF_SAFERwN.tifF)suffixdeletezr+ rio_overviewaverage)ns resampling)copy_src_overviewsrrrrrrr)# HAS_RASTERIOrNosmakedirspathdirnameabspathr&ascontiguousarrayrMrrr"r4 CRS_WGS84r<rrasterioopenwrite update_tagstempfileNamedTemporaryFilenamebuild_overviewsOVERVIEW_LEVELSr r rio_shutilrgCOMPRESS PREDICTOR TILE_SIZEexistsremove) r?rrr _compress_zlevel _predictor_tileprofiledsttmptmp_paths &&&& r write_cogrsW( <233KK 67$G  s"** 5C,7M)I gifiifiia in&figjdeejG$ ]]5# 1 1 1S IIc1 +(+2  $ $F5 A AS88 B ]]8S 4G 4 4 IIc1 +(+5 ]]8T * *c   1C1C D OO~)O D+  #    77>>( # # IIh  LI2 1 B A A5 4 4 + * 77>>( # # IIh  $sT-J( J1> K,-K 'K,28K*6K, J. 1 K  K  K, K) $K,,>L*cV^8dQhR\PR\R\R\R\R\R\ R\/#) rrurv timestampcog_dir title_extrar level_hpar)r&rArwrrr)rs"rrr4sf1E1E **1E1E1E 1E  1E  1E1E 1Ercb\W4p\PPVR4pV\9dRpRVR\P PW4RVRVP R4RR \P 2R \\4R R /p V'dWIR &Ve\V4V R&VeRV R2MRp V'd$RVPRR4P4 2MRp V V RVP R4 V R2p \PPW<4p \W}WR7#)a0 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. mmvariable descriptionunitsrz%Y-%m-%dT%H:%M:%SZmodelzEta03/BESM run rrz EPSG:4326rr_hPa %Y%m%d%Hr)rr)rr VAR_UNITSgetr|VAR_DESCstrftimeRUN_TAGrwrreplacelowerrrjoinr)rurvrrrrrr?rmetalev_str extra_tagfnamers&&&&&&& rexport_field_as_cogr4s-0  (C    2 .E< xv**8>uy))*>?(89s6{ { D)] N[&/&;!I;c"G?J!K''S1779:;PRIj 9#5#5j#A"B9+T RE GGLL (E S$ DDrc `V^8dQhR\R\R\R\R\R\/#)rdata_dirrvr sequentialverboser)rwrlist)rs"rrrhsD     rc \P!V4p\P!VRR7.pVFfp\P!WWR7p\ WWr4p VP V 4V'd&\RV RVPR4 RV 24KfKh V# \d=p T'd*\RT RTPR4 R T 24R p ? KR p ? KR p ? ii;i) zv Exporta todos os timesteps disponiveis de uma variavel como COG GeoTIFF. Returns ------- Lista de caminhos criados. Trrz [COG] rrz -> z [COG] ERRO : N) readerlist_available_timestampsrr read_fieldrappendprintr Exception) rrvrrr timestampssavedtrures &&&&& rexport_var_all_timestepsrhs11(;JKK$' E  P%%h8SD'CE LL  !AJJz,B+C4wOP  L  P hZqJ1G0H1#NOO PsA B C&*+C!!C&clV^8dQhR\R\R\R\R\R\R\/#)rr cog_base_dirrrr skip_existingr)rwrr)rs"rrrsN      rc :\P!VVVVVVR7#)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). )rrrrrr) accumulateexport_all_accumulations_as_cog)rrrrrrs&&&&&&r#export_all_24h_accumulations_as_cogrs)  5 5 $"!%  rc aaaaVwrporEp.pVPR4oV3Rlo\\R/4p\\R/4p \\R/4p .p VEFfoVPS^4p V ^8dV PS.4p V'dV 'd\;QJd!VVV3RlV 4F 'dK RM RM!VVV3RlV 44'dOV FFp\ P PS!S4S R V R S R 24pVPSW/R 34KH KV PS4K\ P PS!S4S R S R 24pV'd>\ P PV4'dVPSVVR 34EKUV PS4EKi V 'gV#/p/pV FoVPS^4p V ^8dxV PS.4pV PS.4p .p/pV F3pVV9gK VPV4pVPV4VVV&K5 V'dTMR VS&VVS&KR VS&K \P!WVVR 7pT EF7oTPS^4p T ^8dTPS/4pT PS.4p T PS.4pTPS/4pT FpTT9d?TPSTR R\T4,R,\T4,34KHTPT4pTeTT9dKf\TT,STS!S4YR7pTPSY/R 34K EK\TS,STS!S4TR7pTPSY/R 34EK: T# \d=pY{Uu.uFpTTR R\T4,3NK Muupiup,uR p?#R p?ii;i \d+pTPSTR \T434R p?EKR p?ii;i)a Worker por TIMESTEP: le o arquivo .bin UMA vez e gera TODOS os COGs das variaveis solicitadas (2D e 3D). Paralelismo correto = 1 task por arquivo. - Variaveis 2D: 1 COG por timestep ({VAR}_{YYYYMMDDHH}.tif) - Variaveis 3D: 1 COG por nivel em plot_levels ({VAR}_{N}hPa_{YYYYMMDDHH}.tif) args: (data_dir, timestamp, vars_list, cog_dir, sequential, overviews, skip_existing) returns: list of (var, timestamp, fpath_or_None, error_or_None) rcF<\S\4'd SV,#S#)N) isinstancer)var_out_dirs&r_var_dir&_worker_cog_timestep.._var_dirs *8T : :x}HHrVAR_NLEV VAR_LEVELSVAR_PLOT_LEVELSc 3<"TFQp\PP\PPS!S4S RV RS R244xKS R#5i)rhPa_rN)rrrr).0levrts_strrs& r '_worker_cog_timestep..sU+'cGGLL#3%qT&0NO&sAAFTrrrNrz leitura: znivel zhPa ausente em levels )rr)r)rrrrrcrrrrrindexrread_fields_selectiverrwr)args _data_dir_t_vars_seq_ovr_skipresults _var_nlev _var_levels_var_plot_levels vars_needednlev plot_lvlsr fpr var_level_map level_k_mapall_lvlskslkr7fieldsrvlev_datarrr rs& @@@@r_worker_cog_timestepr's9=5I5(DGkk*%FIvz"=Iv|"=Kv'8"=K}}S!$ !8(,,S"5Iss+'+sss+'+(( %ChsmuAcU$vhd5STBNNC#67%""3'GGLL#3%q0EFE..R56""3'+. MK}}S!$ !8#R0H(,,S"5IBB (? s+AIIaLBsG ! (*tM# !#K !%M# "T-- =T   4==a(Dax'OOC4,00b9 'OOC4"JJsB/$C(*R$s3x/2JJSQY]Z(\] s AyAX$5 ,  S"hsm"%BNNC#67%)c2x}X\]RT233: NA T{S{!1b$ c!f(<={SSST: 4 NNCT3q62 3 3 4sI5OC>P1P PPO=<P PP Q #QQ cV^8dQhR\R\R\R\R\R\R\R\R \/ #) rrrvars_to_exportrworkersrrrr)rwrrrr)rs"rrrsmxxxxx x  x  xxx xrc 2aaaaaa a!a"T;'g\Pp\P!V4pV'g\ RV R24/#S'Ed \ \R/4p \ \R/4p ^p VFTp V P V ^4p V ^8d0V P V .4pY'd \V4MT , p KKV ^, p KV \V4V ,p\ R\V4 RV RV RV 24VFRp V P V ^4p V ^8d)V P V .4p\ R V R V R V 24KD\ R V 24KT \P!VR R7/pVF@p\PPVV4p\P!VR R7VVV&KB VUu.uF pVVVVW6V3NK upo"VUu/uFpV.bK upo ^o^o^o^o\P!4o!VVVVV V!V"V3RlpV^8dm\VR7;_uu_4pS"Uu/uFpVP\V4VbK pp\!V4FpV!VP#44K RRR4MS"FpV!\V44K \P!4S!, pV^8d SV, M^pS'd\ RS RS RVR RVR R2 4S #uupiuupiuupi +'giLq;i)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 ''rrz [export_cog] z timesteps x z campos/ts = z COGs a gerar | workers=z [3D] z: nlev=z | plot_levels=z [2D] Trc&<S^, oS\S4, ^d,pVFgwr#rEV'd9S ^, o S'd&\RV RVPR4 RV 24KDKFS V,PV4S ^, o Ki S'dy\P!4S , pS'd!VS, \S4S, ,M^p\RSR R\S4 RVR R S R S RVR R VR R2RR7R#R#)z [ERRO] rrrz [3d/z z5.1fz%] ok=z err=z.0fzs ETA=sT)flushN)r1rrrtime)res_listpctrtsrerrelapsedetadonen_errn_okn_skiprt0tasksrs& r_process*export_all_fields_as_cog.._processhs   SZ#%#+ CU IcU!BKK ,C+DBseLMc !!%( $, iikB&G<@7T>c%j4&78aC d2YaE |2c$Z8vVE7WSMS 4   r) max_workersNz [export_cog] z COGs gerados, z erros | z.1fzs (z COG/s))r VAR_NAMESrrrrrr1rrrrr3rsubmitr'rresult)#rrr)rr*rrrr _var_nlev_g_var_plot_levels_gn_per_ts_v_nl_pl n_cogs_totalvar_dirsrdrr%r@pooltaskfutsfutr8speedr:r;r<r=rr>r?s#&&&&&f&& @@@@@@@rexport_all_fields_as_cogrTs6$77v'7'7N55h?J  ;H:QGH w$VZC $V->C B//"a(CQw(,,R4CH4A  !:1  C O,M(D7y B !B//"a(CQw(,,R4t73%seDEtn% !KK t,H GGLLs + A%  A 1nh }U E --nq"un-EDFEDiikB,{ W 5 5NSTedDKK 4d;TAeDT#D)&*6 5 D )$/ 0iikBG '! dWnE dV?5':#c%G 5 Lg  .>U6 5s*K7 K<3L8 L*LL L c0V^8dQhR\R\/#)rrr)rwr)rs"rrrs    rcH\'gR#\P!V4;_uu_4p\VP ^44^8pVP P RR4pT;'dTuuRRR4# +'giR#;i \dR#i;i)z Verificacao basica: abre o arquivo e confirma que tem overviews e tiles. Para validacao completa use: python -m cogdumper ou rio cogeo validate. FrN)rrrr1rrrr)rsrc has_overviewsis_tileds& r validate_cogrZs{ < ]]5 ! !S a 01A5MKKOOGU;H --X" ! ! ! s/BAA>3 B> B BB B! B!g)r >NEVEPRCVPRECPRGE)NF)rFN)FT)FTFF)NFr.TFF)/__doc__rr3rnumpyr&rrtypingrconcurrent.futuresrrrrasterio.shutilshutilrrasterio.transformr rasterio.crsrrasterio.enumsr rrNrrrrrrr from_epsgrr,r<r|rXrsrrrrrr'rTrZrrrrms6 (@ >(. )L   # MM$  F(:80 *6 NVz1EhL8jZx~ W>L 89 <=>sCCC