c0idZddlZddlZddlZddlmZddlZdZdZ dZ dZ e dk(r~e Z e jjZ e e j Zed zd Zed ez Zeedz Zeededz Zeededz ZeefD]0Zej7redeej(d 2edej9ded z ddedej:edej:ej<edZej<edeZ ejBs e jBredej(d e ee Z"e"jGededed ejIyy#e$r(Zed eej(d YdZ[sdZ[wwxYw)!u Calculates total road length per grid cell for a given state. ✅ Features: - Reads {state}_roads_projected.gpkg and {state}_grid_.gpkg - Computes total road length per tile (in meters) - Writes {state}_road_density_.gpkg N)PathGRID_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 scripts/calculate_density.pyresolve_cell_sizers}>")),=">IFGXFYYZ [  9~ !),IJ[I\\] ^  s ? A$AA$c\td|jjddi} tj|dg|ddgd}td t|d d |jj|d <td|jdd j}|dj|jd|d <|jdgj|j }td|S#t $r}t d||d }~wwxYw)z@Calculate total road length per grid cell using in-cell lengths.u5📊 Computing road segments within each grid cell...indexcell_id)columnsgeometry intersection)howz+Failed to compute road/grid intersections: Nu ✅ Created ,z road pieces clipped to cells.length_mu-🧮 Summing clipped lengths per grid cell...ru,✅ Density calculated with in-cell lengths.)print reset_indexrenamegpdoverlay Exception RuntimeErrorlenrlengthgroupbysummapfillnadrop set_indexr)roadsgrid grid_local intersectionsrgrouped grid_results rcalculate_densityr/#s2 AB!!#**GY3G*HJY :,   :. /  L]+A..L MN - 6 6 = =M* 9:##I.z:>>@G' 266w?FFqIJz//9+/6@@LK 89  YHNOUXXYs!D D+D&&D+ctjd}|jdd|jdddtd  |j S) Nz6Calculate road length per grid cell for a given state.) descriptionstatezState name (e.g., alabama).)help cell_size?z.Grid cell size in meters. Defaults to env var z if unset.)nargsr3)argparseArgumentParser add_argumentr parse_args)parsers rr:r:@sh  $ $LF &CD %%6$7zC    __main__u❌ ikmoutputz_roads_projected.gpkg_grid_z.gpkg_road_density_u❌ Missing input file: u📂 Working on z (z.1fz km grid)u 📥 Roads: u 📥 Grid: roads_projected)layergrid_u1⚠️ One of the input files is empty. Aborting. road_density_GPKG)rDdriveru💾 Saved density results to: )%__doc__r7rsyspathlibr geopandasrr rr/r:__name__argsr2lowerr4r errrexitgrid_size_labelbase_dir roads_path grid_path output_pathpathexists capitalizename read_filer)r*emptyresultto_fileresolver<rras + :  z #2Fi PQ L) *+ L( )* CMM*,= >E 3==E/1B*C DD {{djj AB ud +F NN; o5F&GPVNW +K,?,?,A+B CDI  SEl s F))G.GG