L!.i}SrSSKrSSKrSSKrSSKrSSKJr SSKrSSK J r SSK r SSKrSSKJr SSKJr SrSrS+S jrS,S jr\S :XGab\R0"S S9r\R5SSS9 \R5SSS\S3S9 \R75r\R:R=5r\"\R>5r\S-S3r$\"S5\- r%\%\S\$S3- r&\&RO5(d\""S\&35 \RF"S5 \""S\&RP35 \RR"\&S \$3S!9r*\*RV(a\""S"5 \RF"S5 \RY5S#\$S$3r-\%\S\$S%3- r.\%\S\$S&3- r/\"\*SS \-\.S'9 \"\*\\/SS(9 \""S)\/35 gg!\ a(r!\""S\!35 \RF"S5 Sr!C!GNSr!C!ff=f!\ ar!\""S*\!35 Sr!C!gSr!C!ff=f)-u Visualize road density heatmap for a given state. ✅ Features: - Loads {state}_road_density_.gpkg - Automatically chooses column & color map - Supports multiple grid resolutions (2km, 5km, etc.) - Saves optional PNG output for reporting N)Path) rasterize) from_originGRID_CELL_SIZE_METERScU=(d [R"[5nUc[S[S35e[ U5$![an[SUS[S35UeSnAff=f)z8Resolve cell size from CLI args or environment variable.Nz8Cell size must be passed as an argument or provided via .zInvalid cell size 'z'. Provide an integer or set )osgetenvCELL_SIZE_ENV_VAR ValueErrorint)explicit_value raw_valueexcs +/data2/heatmap/scripts/visualize_heatmap.pyresolve_cell_sizers}>")),=">IFGXFYYZ [  9~ !),IJ[I\\] ^  s A A,A''A,length_mhotc l[R"SS9upVURUUUSSSSSSS .S 9nURU=(d S US 3S SS9 UR S5 [R "5 U(a%[R "USSS9 [SU35 g[R"5 g)z$Plot density heatmap from grid data.) )figsizeTrzTotal Road Length (meters)g333333?g{Gz?vertical)labelshrinkpad orientation)columncmapaxlegend linewidth legend_kwdszRoad Density ()r)fontsizeroffi,tight)dpi bbox_inchesu💾 Saved heatmap to: N) pltsubplotsplot set_titleaxis tight_layoutsavefigprintshow)gridrrtitle save_pathfigr grid_plots r plot_heatmapr9+sll7+GC   1%   ILL4N6(!4rrLJGGEN I3G< ' {34  c X0R;a[SUS35eURc [S5eURupEpg[ [ R "Xd- U- 55n[ [ R "Xu- U- 55n US::dU S::a [S5e[XGX5n /n [URX5HVupU b0[U [5(a[ R"U 5(aM:U RU [U 545 MX [U X4U [R SS9n["R$"US S U US SURR'5U [R S 9 nUR)UR+S5S 5 SSS5 g!,(df  g=f) z/Rasterize the density grid and save as GeoTIFF.zColumn 'z' not found in grid.Nz-Grid CRS is undefined; cannot export GeoTIFF.rz5Invalid grid bounds; cannot derive raster dimensions.float32)shapes out_shape transformfilldtypewGTiff)driverheightwidthcountrAcrsr?nodata)columnsr rI total_boundsr mathceilrzipgeometry isinstancefloatisnanappendrnpnanrasterioopento_wktwriteastype)r4 cell_sizeout_pathrminxminymaxxmaxyrGrFr?r=geomvaluerasterdsts rsave_density_geotiffrfGs} \\!8F8+?@AA xxHII!..D  4;)34 5E DK945 6F zVq[PQQD =I F4==$,7  =Zu55$**U:K:K  tU5\*+8 / VV F   HHOO vv   &-- *A.   s "F// F=__main__z4Visualize road density using grid GeoPackage output.) descriptionstatezState name (e.g., alabama).)helpr\?z.Grid cell size in meters. Defaults to env var z if unset.)nargsrju❌ rDikmOSM_State_Files_road_density_z.gpkgu❌ File not found: u📦 Loading density data from road_density_)layeru!⚠️ No features found to plot.z Road Density (z grid)z.pngz.tif)rrr5r6)ru"🗺️ Saved GeoTIFF heatmap to: u⚠️ Could not save GeoTIFF: )rrNN)r)0__doc__argparserMr syspathlibr geopandasgpdmatplotlib.pyplotpyplotr+numpyrUrWrasterio.featuresrrasterio.transformrr rr9rf__name__ArgumentParserparser add_argument parse_argsargsrilowerr\r errr2exitgrid_size_labelbase_dir input_pathexistsname read_filer4empty capitalizer5 output_png output_tifr:rrs[ '*+ 8)/V z  $ $JF &CD %%6$7zC    D JJ   E%dnn5 #D),O%&.Hug^O3DEJJJ      $ZL12   +JOO+< => ==]?:K+L MD zz 12  !"//1B& IEug^O3DDIIJug^O3DDIIJjuEZX7T9jL 2:,?@U  SEl  6 7 /u5667s0-F0G!0G6GG!G<' G77G<