U ;i@s|dZddlZddlmZddlmZddlZedddZ e dkrxe ej d krje d e d ed e ej d dS) z Extracts building features from a state's OpenStreetMap .pbf file. This script complements extract_state_roads.py by extracting building polygons with attributes like building type, height, and building:levels for heatmap analysis. N)Path)OSM)statec Cs|}t}|d|}||d}|s~t|d}|r^|d}td|jn td|tdt d||d }td |jd zt t |}|j d d idddd dddddddgd}|dks|j rtdtdWdStdt|dd|dtdt|jd |jkrX|d d}td |d|jkr~|d} td!| Wn<tk r} ztd"| t dW5d} ~ XYnXz|jd#d$}td%Wn2tk r} ztd&| W5d} ~ XYnXz"|j|d'd(d)td*|Wn<tk rh} ztd+| t dW5d} ~ XYnXtd,td-|td.|jtd/|jtd0t|dtd1td2td3td4dS)5NZOSM_State_Filesz.osm.pbfz*.pbfru📦 Using PBF file: u❌ No PBF file found in z= Run extract_state_roads.py first to download the OSM data.z_buildings.gpkgu 🏢 Extracting buildings from z...ZbuildingTZkeepFZheightzbuilding:levelszaddr:housenumberz addr:streetZamenityZshopZoffice)Z custom_filterZ filter_typeZ keep_nodesZkeep_relationsZextra_attributesu/⚠️ No building data found in this OSM file.z9 This state may not have detailed building data in OSM.u✅ Extracted ,z buildings from z data.u📋 Columns available: u🏘️ Top building types: u📏 Height stats: u ❌ Building extraction failed: i)Zepsgu&🌐 Reprojected to WGS84 (EPSG:4326).u⚠️ Reprojection failed: buildingsZGPKG)ZlayerZdriveru#💾 Saved extracted buildings to: u❌ Saving failed: u" ✅ Building extraction complete!u 📁 Folder: u 📥 Input: u 💾 Output: u🏢 Buildings extracted: z<------------------------------------------------------------z Next steps:z91. Run load_buildings.py to reproject buildings to meterszD2. Run visualize_all_attribute_grid.py to generate building heatmaps)lowerrcwdexistslistZglobprintnamesysexitrstrZget_data_by_custom_criteriaemptylen capitalizecolumnsZ value_countsheadZdescribe ExceptionZto_crsZto_file) rZbase_dirZ state_dirZpbf_fileZ pbf_filesZ output_pathZosmrZbuilding_typesZ height_statser1/data2/heatmap/scripts/extract_state_buildings.pyextract_state_buildingss          r__main__z=Usage: python -m scripts.extract_state_buildings z7Example: python -m scripts.extract_state_buildings ohior)__doc__rpathlibrZpyrosmrZ geopandasZgpdrr__name__rargvr rrrrrs  b