
    @h                        S r SSKrSSKJr  SSKrSSKJr  SS jr	\
S:X  GaL  \" \R                  5      S:  a  \" S	5        \R                  " S
5        \R                  S
   R                  5       r\" \R                  5      S:  a  \" \R                  S   5      OSr\S-   S3r\" S5      \-  r\\ S\ S3-  r\R-                  5       (       d  \" S\ 35        \R                  " S
5        \" S\R.                   35        S\ 3r\R2                  " \\S9r\R6                  (       a  \" S5        \R                  " S
5        \R9                  5        S\ S3r\\ S\ S3-  r\	" \SS\\S9  gg)u  
Visualize road‐density heatmap for a state using a blue/green colormap.

Features:
 - Loads {state}_road_density_{gridSize}.gpkg
 - Uses a blue⇢green color ramp
 - Supports specifying cell size (e.g., 2 km, 5 km)
 - Saves out a PNG image alongside the GeoPackage
    N)Pathlength_mBuGnc                 l   [         R                  " SS9u  pVU R                  UUUSSSSSSS	.S
9  UR                  U=(       d    SU S3SSS9  UR	                  S5        [         R
                  " 5         U(       a%  [         R                  " USSS9  [        SU 35        g[         R                  " 5         g)z>Plot density heatmap from grid data using blue/green colormap.)      )figsizeTr   zTotal Road Length (m)g333333?verticalg{Gz?)labelshrinkorientationpad)columncmapaxlegend	linewidthlegend_kwdszRoad Density ()   r   )fontsizer   offi,  tight)dpibbox_inchesu   💾 Saved heatmap to: N)	pltsubplotsplot	set_titleaxistight_layoutsavefigprintshow)gridr   r   title	save_pathfigr   s          2/data2/heatmap/scripts/visualize_heatmap_bluish.pyplot_heatmapr*      s    ll7+GCII,%	
   LL4N6(!4rrLJGGENI3G<'	{34
    __main__   zQUsage: python -m scripts.visualize_heatmap_bluish <STATE_NAME> [CELL_SIZE_METERS]   i  i  kmOSM_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_bluish.png)r   r   r&   r'   )r   r   NN)__doc__syspathlibr   	geopandasgpdmatplotlib.pyplotpyplotr   r*   __name__lenargvr#   exitlowerstateint	cell_size
grid_labelbase_dir
input_pathexistsname
layer_name	read_filer%   empty
capitalizer&   
output_png r+   r)   <module>rN      s|      6 z
388}qabHHQKE$'MA$5CHHQK 4ItO$B'J%&.Hug^J<uEEJ$ZL12	+JOO+<
=> -J==:6Dzz12!"/*VDEug^J<{KKJ jvUjY9 r+   