0i(dZddlZddlZddlZddlmZddlZddlm Z ddl Z ddl Z ddlmZddlmZmZmZddlmZdZdZdd Zd efd Zd Zd ZdefdZdZdZdZdZ e!dk(re yy)a Extract smooth hotspot contours from heatmap GeoTIFFs using continuous raster analysis. Instead of merging rectilinear grid cells (which creates blocky boundaries), this script treats the raster as continuous and uses skimage.measure.find_contours to extract smooth contour polygons at a threshold that captures the target area. Outputs: - PNG overlay showing contours on the heatmap - GeoJSON with smooth polygon boundaries (not blocky grid cells) N)Path)shapes)shape MultiPolygonPolygon)measureHOTSPOT_TARGET_AREA_KM2c||ntjt}|tdtd t |}|dkr td|S#t$r}td|d|d}~wwxYw)z7Return the requested hotspot area in square kilometers.Nz3Hotspot area must be provided via --target-area or .zInvalid hotspot area ''.rz'Hotspot area must be greater than zero.)osgetenvTARGET_AREA_ENV_VAR ValueErrorfloat)explicit_value raw_valueareaexcs #scripts/extract_hotspot_contours.pyresolve_target_arears"0"<"))L_B`IABUAVVW X  JY qyBCC K J1)B?@cIJs A A/A**A/c,dg}|r|j|g}|D]"}|j|j|$g}t}t|D]:}||vs|j s|j ||j |<|S)z6Find GeoTIFF heatmaps in the given location directory.z*road_density_*km.tif)extendglobsetsortedis_fileappendadd) location_dirextra_patternspatternsfilespattern unique_filesseen file_paths rfind_heatmap_rastersr(/s'(H' E \&&w/0L 5DE] D Y%6%6%8    * HHY #  raster_pathc$|jjd}|std|j|d}|j dstd|d t |dd}||d zfS#t$r}td |d |d}~wwxYw) zPExtract the grid size label (e.g., '2km') and corresponding cell size in meters._z"Cannot parse grid size label from kmz Grid label 'z' does not end with 'km'.NzInvalid grid label 'r g@@)stemsplitrnameendswithr)r*parts size_labelcell_kmrs rparse_size_labelr7@s    " "3 'E =k>N>N=OPQQrJ   t $< |3LMNNI 3B( w' '' I/ |2>?SHIsA11 B:B  Bcht|d}t|d}||z}|dz }||z }|tj|}t|dk(r t dtj |ddd} t t|t| dz } | | } td| d d |d d | S) zHCalculate density threshold that captures approximately the target area.r@BzNo valid data in rasterNr-u 📊 Threshold: .1fz (captures ~u km²)) absnpisfinitelenrsortminintprint) data transformtarget_area_km2 pixel_width pixel_height pixel_area_m2pixel_area_km2 target_pixels valid_data sorted_values threshold_idx thresholds r#calculate_threshold_for_target_arearQOsil#Ky|$L,.M"Y.N$n4Mbkk$'(J :!233GGJ'"-MM*C ,>,BCMm,I  #l?3:Ov VW r)ctjtj|||k\zddjtj}t j |d}|stjddgg|Sg}|D]}g}|D]4\} } |d| |dzz} |d | |d zz} |j| | f6t|d k\sM t|} | jr| js|j| |stjddgg|Sg}t|d D])\}}|j d z }|j|||d+tj|d|S#t$rYwxYw)zHExtract smooth contours from raster using skimage.measure.find_contours.r;rg?)level cluster_idarea_km2)columnsgeometrycrsr9)startr:)rTrUrWrW)rWrX)r>wherer?astypeuint8r find_contoursgpd GeoDataFramerr@ris_validis_empty Exception enumerater)rErFrPrX binary_maskcontourspolygonscontourcoordsrowcolxypolyclustersidxgeomrUs rextract_contours_from_rasterrtjs((2;;t, 0ABAqIPPQSQYQYZK$$[r]nanr?boundsrXrF)rusrcrEr{r}rXrFs r load_rasterrs t xx{!!),  88DFNBFFD9D88BKK-tRVV#***5 )4 Or)ctjd}|jdd|jdddtd  |jd d d d|j S)NzWExtract smooth hotspot contours from density GeoTIFFs using continuous raster analysis.) descriptionlocationz/Location name (e.g., alabama, egypt, thailand).)helpz --target-area target_areaz>Target hotspot area in square kilometers. Defaults to env var r )destrz--extra-patternrr!z1Additional glob pattern(s) for locating GeoTIFFs.)actionrr)argparseArgumentParser add_argumentr parse_args)parsers rrrs  $ $mF  )Z[   ##6"7q :    @     r)c t}|jj} t|j}td|z }|jst d|t||j}|st d|tdt|d|jdtd|D]}td |j t|\}}} } t#|| |} t%|| | | } | j&r td atdt| d| dj)} td| dd t+|\}}t-|||| | |}t/| |||}td|jtd|jtdy#t $r}t d||d}~wwxYw#t $r}td |Yd}~Dd}~wwxYw#t $r}td |Yd}~id}~wwxYw#t $r}td |Yd}~d}~wwxYw#t $rd}YwxYw)Nu❌ outputu"❌ Location directory not found: u$⚠️ No heatmap GeoTIFFs found in u🎯 Processing z heatmaps for z...u:🔬 Using continuous raster analysis with smooth contoursu 🗺️ u! ⚠️ Could not load raster: u ⚠️ u& ⚠️ Could not extract contours: u! ⚠️ No hotspots identified.u ✅ Found z hotspot region(s)rUu 📍 Total area: r<u km²unknownu 💾 Saved overlay: u 💾 Saved contours: u( ✅ Smooth contour extraction complete.)rrlowerrrr SystemExitrexistsr(r!rDr@ capitalizer2rrerQrtrsumr7rr)argsrrGerrr heatmap_rastersr*rEr}rrFrPhotspots total_arear5r,png_path geojson_paths rmainrs .>?>H,L    =l^LMM*<9L9LMO ? ~NOO S12.ATATAV@WWZ [\ FH&  ;++,-. 2=k2J /D&*i  ;D)_]I  3D)YPZ[H >> 5 6  c(m_,>?@j)--/  $Z$4I>? #,[9MJ!dFJRab#Hk:W  ' 78 ():):(;<=S'V 56s 04u&C/0(  5cU; <    Jse$ %    :3%@ A   #"J #sqG G*0 H>H4 I G'G""G'* H 3HH  H1H,,H14 I=II I'&I'__main__)N)"__doc__rmathr pathlibr geopandasramatplotlib.pyplotpyplotrnumpyr>rxrasterio.featuresrshapely.geometryrrrskimagerrrr(r7rQrtrrrrr__name__r)rrs  $99/"" ($ (6,D^ (d (B,>7B zFr)