U i @sdZddlZddlZddlmZdejd<ddlZddlZ ddl m Z ddl ZddlmZdz Generate heatmaps for all key OSM road attributes aggregated per grid cell. Enhancements: - Aggregates attributes per grid cell (mean for numeric, mode for categorical) - Uses grid resolution in output filenames - Handles missing data gracefully N)Path0Z USE_PYGEOS) colormapsFcCstd|dtj||ddd}|rjdd}|d||}|j||d id d |d }nNtj||d d ||<|d| }|j||did d |d}|j |d ddd}||fS)z'Aggregate road attribute per grid cell.u📊 Aggregating z per grid cell...innerZ intersects)howZ predicatecSs|}|js|jdStjS)Nr)modeemptyZilocnpnan)xZ mode_seriesr 6/data2/heatmap/scripts/visualize_all_attribute_grid.py mode_func%sz'aggregate_attributes..mode_funcZ index_right_modeT)columnsZinplaceZcoerce)errorsZ_meanleft)Z left_indexZright_onr) printgpdZsjoingroupbyZaggZ reset_indexrenamepdZ to_numericZmeanmerge)roadsgridattr categoricalZjoinedrZgroupedattr_colr r r aggregate_attributess  rc Cstjdd\}} |jd| d|j} |rZ|d|d|d|d} n|d|d|d} td|d |d |pd d |r|| } t | d krd} nd} |j || dd| ddd|t dd| Dddddd| j | dddn||j s6t||ddgnd\}}td }|jdd!|j ||d| d|||d"d#d$dddd% | j | ddd| j| d&d'd(| d)t||d*|d+|d,}tj|d-d.|d/t|td0|jdS)1N) )ZfigsizeZwhiteu — z (z, )u 🔎 Plotting z heatmap for z grid (layer=rz)...Ztab20ZSet2Tr)gRQ?z upper leftcSsg|] }t|qSr )str).0vr r r `sz*plot_attribute_heatmap..)Zbbox_to_anchorZloctitleZlabelszNo Data)colorlabel)columnaxlegendrcmap linewidth legend_kwds missing_kwdsZblackg?)r,r)r/_)rr#Zinferno)r)g333333?g{Gz?)r*Zshrinkpad) r+r.r/r,r-vminvmaxr0r1r )Zfontsizer4Zoff_Z _heatmap_z.pngi,Ztight)ZdpiZ bbox_inchesZ facecoloru💾 Saved heatmap: )pltZsubplotsZpatchZ set_facecolorZdissolveboundary capitalizerZdropnaZuniquelenZplotsortedrr Z nanpercentilerZget_cmapcopyZset_badZ set_titleZaxisZ tight_layoutZsavefigZ get_facecolorclosename)rrrstatesave_dir size_labelr layer_nameZfigr,Z state_outliner(Z unique_valsZ cmap_namer5r6Zcmap_objZout_pathr r r plot_attribute_heatmap;sl   &  0    rE__main__zUUsage: python -m scripts.visualize_all_attribute_grid [CELL_SIZE_METERS]r#i'iZkmZOSM_State_Filesz_roads_projected.gpkgZ_grid_z.gpkgu❌ Roads file not found: u❌ Grid file not found: u$📦 Loading road and grid data for z...Zroads_projected)layerZgrid_ZlaneswidthZheightzbuilding:levelsZsurfaceZmaterialZbridgeZtunnelrHZhighwayZbuildingcCsg|]}|tjkr|qSr rrr%ar r r r's r'cCsg|]}|tjkr|qSr rJrKr r r r's u📊 Road numeric attributes: u"🎨 Road categorical attributes: )rr)rrDTz_buildings_projected.gpkgZbuildings_projectedu📦 Loaded buildings data: u'⚠️ Failed to load buildings layer: cCsg|]}|tjkr|qSr  buildingsrrKr r r r's cCsg|]}|tjkr|qSr rMrKr r r r's u"📊 Building numeric attributes: u&🎨 Building categorical attributes: rNu> ✅ All grid-based attribute heatmaps generated successfully!)F)FN)5__doc__sysospathlibrenvironZ geopandasrZpandasrZmatplotlib.pyplotZpyplotr9Znumpyr Z matplotlibrrrE__name__r<argvrexitlowerrAintZ cell_sizerCZgdf_pathZ grid_pathparentrBexistsr;Z read_filerrZ numeric_attrsZcategorical_attrsZavailable_numericZavailable_categoricalrr>Zgrid_aggrrNZbuildings_pathr@ ExceptioneZb_available_numericZb_available_categoricalr r r r s|       V