o ˷e/@s<dZddlZddlZddgZd ddZd ddZd dZdS) a ********* Shapefile ********* Generates a networkx.DiGraph from point and line shapefiles. "The Esri Shapefile or simply a shapefile is a popular geospatial vector data format for geographic information systems software. It is developed and regulated by Esri as a (mostly) open specification for data interoperability among Esri and other software products." See https://en.wikipedia.org/wiki/Shapefile for additional information. Nread_shp write_shpTc sd}tj|tddzddlm}Wnty#}ztd|d}~wwt|ts+dSt }| |}|dur?t d||D]} d d | j D} | D]r } | dur_|r^td qMfd d | D} tt| | } | | d <| |jkr|j| dfi| qM| |j|jfvrt| | ||D]}|\}}}|||||||qqM|rtd| dqMqA|S)a6Generates a networkx.DiGraph from shapefiles. .. deprecated:: 2.6 read_shp is deprecated and will be removed in NetworkX 3.0. See https://networkx.org/documentation/latest/auto_examples/index.html#geospatial. Point geometries are translated into nodes, lines into edges. Coordinate tuples are used as keys. Attributes are preserved, line geometries are simplified into start and end coordinates. Accepts a single shapefile or directory of many shapefiles. "The Esri Shapefile or simply a shapefile is a popular geospatial vector data format for geographic information systems software [1]_." Parameters ---------- path : file or string File, directory, or filename to read. simplify: bool If True, simplify line geometries to start and end coordinates. If False, and line feature geometry has multiple segments, the non-geometric attributes for that feature will be repeated for each edge comprising that feature. geom_attrs: bool If True, include the Wkb, Wkt and Json geometry attributes with each edge. NOTE: if these attributes are available, write_shp will use them to write the geometry. If nodes store the underlying coordinates for the edge geometry as well (as they do when they are read via this method) and they change, your geomety will be out of sync. strict: bool If True, raise NetworkXError when feature geometry is missing or GeometryType is not supported. If False, silently ignore missing or unsupported geometry in features. Returns ------- G : NetworkX graph Raises ------ ImportError If ogr module is not available. RuntimeError If file cannot be open or read. NetworkXError If strict=True and feature is missing geometry or GeometryType is not supported. Examples -------- >>> G = nx.read_shp("test.shp") # doctest: +SKIP References ---------- .. [1] https://en.wikipedia.org/wiki/Shapefile zread_shp is deprecated and will be removed in 3.0.See https://networkx.org/documentation/latest/auto_examples/index.html#geospatial. stacklevelrogrz+read_shp requires OGR: http://www.gdal.org/NzUnable to open cSsg|]}|qS)GetName.0xr r P/var/www/ideatree/venv/lib/python3.10/site-packages/networkx/readwrite/nx_shp.py izread_shp..z"Bad data: feature missing geometrycsg|] }|qSr )GetField GetFieldIndexr fr rrqsShpNamez GeometryType z not supported)warningswarnDeprecationWarningosgeor ImportError isinstancestrnxDiGraphOpen RuntimeErrorschemageometry NetworkXErrordictzipr GetGeometryTypewkbPointadd_node GetPoint_2D wkbLineStringwkbMultiLineStringedges_from_lineadd_edgeupdate)pathsimplify geom_attrsstrictmsgrerrnetshplyrfieldsgflddata attributesedgee1e2attrr rrrsRC       c csd}tj|tddzddlm}Wnty$}ztd|d}~ww||jkr|rZ|}| d}|rL| |d <| |d <| |d <| d| ||fVdStd| dD]G} | | } | | d} |}|r||j} | | d| d| | d| d| |d <| |d <| |d <~ | | |fVqcdS||jkrt|D]} || } t| |||EdHqdSdS) a Generate edges for each line in geom Written as a helper for read_shp Parameters ---------- geom: ogr line geometry To be converted into an edge or edges attrs: dict Attributes to be associated with all geoms simplify: bool If True, simplify the line as in read_shp geom_attrs: bool If True, add geom attributes to edge as in read_shp Returns ------- edges: generator of edges each edge is a tuple of form (node1_coord, node2_coord, attribute_dict) suitable for expanding into a networkx Graph add_edge call .. deprecated:: 2.6 zedges_from_line is deprecated and will be removed in 3.0.See https://networkx.org/documentation/latest/auto_examples/index.html#geospatial.rrrrz2edges_from_line requires OGR: http://www.gdal.org/NWkbWktJson)rrrrrrr&r*copy GetPointCount ExportToWkb ExportToWkt ExportToJsonr)rangeGeometry AddPoint_2Dr+GetGeometryCountGetGeometryRefr,)geomattrsr0r1r3rr4 edge_attrslastipt1pt2segmentgeom_ir r rr,sV          r,c sd}tj|tddzddlmWnty#}ztd|d}~wwfdd }dfd d }tjt j t j ifd d  d}||}z|dWnY|ddj}i} fdd} |D]} |j| } || | | | | |\} }||| qpz|dWnY|ddj}i}|jddD]}|j|} ||| | |d||\} }||| qd\}}dS)aWrites a networkx.DiGraph to two shapefiles, edges and nodes. .. deprecated:: 2.6 write_shp is deprecated and will be removed in 3.0. See https://networkx.org/documentation/latest/auto_examples/index.html#geospatial. Nodes and edges are expected to have a Well Known Binary (Wkb) or Well Known Text (Wkt) key in order to generate geometries. Also acceptable are nodes with a numeric tuple key (x,y). "The Esri Shapefile or simply a shapefile is a popular geospatial vector data format for geographic information systems software [1]_." Parameters ---------- G : NetworkX graph Directed graph outdir : directory path Output directory for the two shapefiles. Returns ------- None Examples -------- nx.write_shp(digraph, '/shapefiles') # doctest +SKIP References ---------- .. [1] https://en.wikipedia.org/wiki/Shapefile zwrite_shp is deprecated and will be removed in 3.0.See https://networkx.org/documentation/latest/auto_examples/index.html#geospatial.rrrrz,write_shp requires OGR: http://www.gdal.org/Ncs>d|vr |d}|Sd|vr|d}|St|djdkrrj}|d|d}}z|jdg|R|jdg|RW|Styqdd|D}dd|D}|jdg|R|jdg|RY|Swj}z |jdg|RW|Styd d|D}|jdg|RY|Sw) NrArBrtupler@cSg|]}t|qSr floatr r r rrrz2write_shp..netgeometry..cSrXr rYr r r rr rcSrXr rYr r r rrr) CreateGeometryFromWkbCreateGeometryFromWkttype__name__rJr*SetPoint TypeErrorr')keydatarN_from_to_ffrom_ftofkeyrr r netgeometrys:     zwrite_shp..netgeometrycsT|}||dur|D] \}}|||q|||dSN)Feature GetLayerDefn SetGeometryitemsSetField CreateFeatureDestroy)r"r7r;featurefieldrb)r9rr rcreate_features   z!write_shp..create_featurecsFt|vrt|||<nj||<|||}||dSri)r] OFTString FieldDefn CreateField)ravaluer8layernewfield)OGRTypesrr radd_fields_to_layer&s  z&write_shp..add_fields_to_layerzESRI Shapefilenodescs\i}|D]#\}}|dkr)|dkr)|dkr)|dkr)||vr%|||||||<q||fS)NrCrBrAr)rm)rbr8rxr;rarw)r{r rcreate_attributes=s z$write_shp..create_attributesedgesT)rb)NNri)rrrrrr UseExceptionsint OFTIntegerrrtrZOFTRealGetDriverByNameCreateDataSource DeleteLayer CreateLayerr'r|r*r~ get_edge_data)Goutdirr3r4rhrsdrvshpdirr| node_fieldsr}nrbr;r~ edge_fieldsr<r )rzr{r9rrrsP#          )TTT)TT)__doc__rnetworkxr__all__rr,rr r r rs   p H