ó |£*^c@sFdZddlZdZddgZed„Zd„Zd„ZdS( s± ********* 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 http://en.wikipedia.org/wiki/Shapefile for additional information. iÿÿÿÿNs!Ben Reilly (benwreilly@gmail.com)tread_shpt write_shpcCsxyddlm}Wntk r3tdƒ‚nXt|tƒsGdStjƒ}|j|ƒ}x|D]}g|jD]}|j ƒ^qy}xÜ|D]Ô}g|D]}|j |j |ƒƒ^q¥} |j ƒ} t t|| ƒƒ} |j ƒ| d<| jƒdkr(|j| jdƒ| ƒn| jƒdkr˜| jƒd} |r¨| jƒ| d <| jƒ| d <| jƒ| d <|j| jdƒ| j| ƒ| ƒqlxÁt| ƒD]°} | j| ƒ}| j| dƒ}|j|jƒ}|j|d|dƒ|j|d|dƒ|jƒ| d <|jƒ| d <|jƒ| d <~|j||| ƒqµWq˜q˜WqiW|S( sGenerates a networkx.DiGraph from shapefiles. 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. Returns ------- G : NetworkX graph Examples -------- >>> G=nx.read_shp('test.shp') # doctest: +SKIP References ---------- .. [1] http://en.wikipedia.org/wiki/Shapefile iÿÿÿÿ(togrs+read_shp requires OGR: http://www.gdal.org/NtShpNameiiitWkbtWkttJson(tosgeoRt ImportErrort isinstancetstrtnxtDiGraphtOpentschematGetNametGetFieldt GetFieldIndextgeometrytdicttziptGetGeometryTypetadd_nodet GetPoint_2Dt GetPointCountt ExportToWkbt ExportToWktt ExportToJsontadd_edgetrangetGeometryt wkbLineStringt AddPoint_2D(tpathtsimplifyRtnettshptlyrtxtfieldstftflddatatgt attributestlasttitpt1tpt2tsegment((sm/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/readwrite/nx_shp.pyRsH!   " + ("cs{yddlm‰Wntk r3tdƒ‚nXˆjƒ‡fd†}d‡‡fd†}ˆjdƒ}|j|ƒ}y|jdƒWnnX|jddˆj ƒ}x7|D]/}|j |}|||ƒ‰|ˆ|ƒqºWy|jdƒWnnX|jddˆj ƒ} i} i} iˆj t 6ˆjt6ˆjt6} x|jd tƒD]} |j| Œ}|| |ƒ‰xÍ| d jƒD]»\}}|d kr˜|d kr˜|d kr˜|dkr˜|| krFt|ƒ| kr | t|ƒ| | s    L