L!.i}dZddlZddlZddlZddlZddlmZddlZddl m Z ddl Z ddlZddlmZddlmZdZdZd+d Zd,d Zed k(rjej0d Zej5ddej5dddedej7Zej:j=Z eej>ZedzdZ$edez Z%e%ede$dz Z&e&jOse"de&ejFde"de&jPejRe&d e$!Z*e*jVre"d"ejFdejYd#e$d$Z-e%ede$d%z Z.e%ede$d&z Z/ee*dd e-e.' ee*ee/d(e"d)e/yy#e $r(Z!e"de!ejFdYdZ![!dZ![!wwxYw#e $rZ!e"d*e!YdZ![!ydZ![!wwxYw)-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_METERSc|xstjt}|tdtd t |S#t$r}td|dtd|d}~wwxYw)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$AA$length_mhotc jtjd\}}|j|||ddddddd  }|j|xsd |d d d|j dtj |r'tj |ddtd|ytjy)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 h||jvrtd|d|j td|j\}}}}t t j ||z |z }t t j ||z |z } |dks| dkr tdt||||} g} t|j||D]I\} } | %t| trt j| r.| j| t| fKt| | |f| tj d}t#j$|d d | |d d|jj'| tj 5}|j)|j+dd dddy#1swYyxYw) 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 T\\!8F8+?@AA xxHII!..D$d  4$;)34 5E D4K945 6F zVq[PQQD$ 9=I F4==$v,7 e =Zu5$**U:K  tU5\*+8 5/ VV F   HHOO vv   &-- *A.   s ="F((F1__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:rrsb '*+ 8)/V z $X $ $JF &CD %%6$7zC    D JJ   E%dnn5 #D),O%&.Hug^O3DEJJJ     $ZL12  +JOO+< => 3==]?:K+L MD zz 12 !"//1B& IEug^O3DDIIJug^O3DDIIJjuEZX7T9jL 2:,?@U  SEl 6 7 /u5667s03F5G%5G":GG"%G?* G::G?