a J5d?@@sdZddlZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZdd gZd d Zd d ZddZddZddZddZddZd&ddZddZddZddZdd Zd!d"Zd#d$Zd'd%d ZdS)(a> This modules provides a conversion to / from a ragged (or "jagged") array representation of the geometries. A ragged array is an irregular array of arrays of which each element can have a different length. As a result, such an array cannot be represented as a standard, rectangular nD array. The coordinates of geometries can be represented as arrays of arrays of coordinate pairs (possibly multiple levels of nesting, depending on the geometry type). Geometries, as a ragged array of coordinates, can be efficiently represented as contiguous arrays of coordinates provided that there is another data structure that keeps track of which range of coordinate values corresponds to a given geometry. This can be done using offsets, counts, or indices. This module currently implements offsets into the coordinates array. This is the ragged array representation defined by the the Apache Arrow project as "variable size list array" (https://arrow.apache.org/docs/format/Columnar.html#variable-size-list-layout). See for example https://cfconventions.org/Data/cf-conventions/cf-conventions-1.9/cf-conventions.html#representations-features for different options. The exact usage of the Arrow list array with varying degrees of nesting for the different geometry types is defined by the GeoArrow project: https://github.com/geoarrow/geoarrow N)creation) GeometryTypeget_coordinate_dimension get_parts get_rings get_type_id)get_coordinates)is_emptyto_ragged_arrayfrom_ragged_arraycCsXt||d}t|}|rPt|d}|tt|}tj||tjdd}|dfS)N include_zrZaxis) r r anynpZnonzeroarangeleninsertnan)arrrcoordsemptiesindicesrrQ/var/www/html/django/DPS/env/lib/python3.9/site-packages/shapely/_ragged_array.py_get_arrays_point0s rcCsRtt|dd}t||dkrNtj|d|dt|fd|dd}|S)NrrZconstant)Zconstant_values)rrbincountcumsumrpad)rnoffsetsrrr_indices_to_offsets>sr#cCs4t|dd\}}t|t|}t||d}||ffS)NT return_indexr )rr#rr )rr_ part_indicesr"rrrr_get_arrays_multipointKs r(cCs*t|d|d\}}t|t|}||ffS)NTr%r)r r#r)rrrrr"rrr_get_arrays_linestringWsr*cCsTt|dd\}}t|t|}t|d|d\}}tt|dd}|||ffSNTr$r)r)rr#rr rrrr)rrarr_flatr'offsets2rroffsets1rrr_get_arrays_multilinestring_s r/cCsTt|dd\}}t|t|}t|d|d\}}tt|dd}|||ffSr+)rr#rr rrrr)rrr, ring_indicesr-rrr.rrr_get_arrays_polygonls r1c Cs~t|dd\}}t|t|}t|dd\}}tt|dd}t|d|d\}} tt| dd} || ||ffS)NTr$rr)) rr#rrrrrrr ) rrr,r'offsets3Z arr_flat2r0r-rrr.rrr_get_arrays_multipolygonysr3cCst|}|dur.tt|t|dk}tt|}||dk}t|dkrt|d}|tj kr|t ||\}}n|tj krt ||\}}nz|tj krt||\}}n`|tjkrt||\}}nF|tjkrt||\}}n,|tjkrt||\}}ntd|jdnt|dkrt|tj tjhkrLtj}t||\}}nrt|tj tjhkrxtj}t||\}}nFt|tj tjhkrtj}t||\}}ntdd d |Dd ntdd d |Dd |||fS) a1 Converts geometries to a ragged array representation using a contiguous array of coordinates and offset arrays. This function converts an array of geometries to a ragged array (i.e. irregular array of arrays) of coordinates, represented in memory using a single contiguous array of the coordinates, and up to 3 offset arrays that keep track where each sub-array starts and ends. This follows the in-memory layout of the variable size list arrays defined by Apache Arrow, as specified for geometries by the GeoArrow project: https://github.com/geoarrow/geoarrow. Parameters ---------- geometries : array_like Array of geometries (1-dimensional). include_z : bool, default None If False, return 2D geometries. If True, include the third dimension in the output (if a geometry has no third dimension, the z-coordinates will be NaN). By default, will infer the dimensionality from the input geometries. Note that this inference can be unreliable with empty geometries (for a guaranteed result, it is recommended to specify the keyword). Returns ------- tuple of (geometry_type, coords, offsets) geometry_type : GeometryType The type of the input geometries (required information for roundtrip). coords : np.ndarray Contiguous array of shape (n, 2) or (n, 3) of all coordinates of all input geometries. offsets: tuple of np.ndarray Offset arrays that make it possible to reconstruct the geometries from the flat coordinates array. The number of offset arrays depends on the geometry type. See https://github.com/geoarrow/geoarrow/blob/main/format.md for details. Notes ----- Mixed singular and multi geometry types of the same basic type are allowed (e.g., Point and MultiPoint) and all singular types will be treated as multi types. GeometryCollections and other mixed geometry types are not supported. See also -------- from_ragged_array Examples -------- Consider a Polygon with one hole (interior ring): >>> import shapely >>> polygon = shapely.Polygon( ... [(0, 0), (10, 0), (10, 10), (0, 10)], ... holes=[[(2, 2), (3, 2), (2, 3)]] ... ) >>> polygon This polygon can be thought of as a list of rings (first ring is the exterior ring, subsequent rings are the interior rings), and each ring as a list of coordinate pairs. This is very similar to how GeoJSON represents the coordinates: >>> import json >>> json.loads(shapely.to_geojson(polygon))["coordinates"] [[[0.0, 0.0], [10.0, 0.0], [10.0, 10.0], [0.0, 10.0], [0.0, 0.0]], [[2.0, 2.0], [3.0, 2.0], [2.0, 3.0], [2.0, 2.0]]] This function will return a similar list of lists of lists, but using a single contiguous array of coordinates, and multiple arrays of offsets: >>> geometry_type, coords, offsets = shapely.to_ragged_array([polygon]) >>> geometry_type >>> coords array([[ 0., 0.], [10., 0.], [10., 10.], [ 0., 10.], [ 0., 0.], [ 2., 2.], [ 3., 2.], [ 2., 3.], [ 2., 2.]]) >>> offsets (array([0, 5, 9]), array([0, 2])) As an example how to interpret the offsets: the i-th ring in the coordinates is represented by ``offsets[0][i]`` to ``offsets[0][i+1]``: >>> exterior_ring_start, exterior_ring_end = offsets[0][0], offsets[0][1] >>> coords[exterior_ring_start:exterior_ring_end] array([[ 0., 0.], [10., 0.], [10., 10.], [ 0., 10.], [ 0., 0.]]) NrrGeometry type  is not supportedz,Geometry type combination is not supported (cSsg|]}t|jqSrrname.0trrr z#to_ragged_array..)cSsg|]}t|jqSrr8r:rrrr=%r>)rZasarrayrrr uniquerrrPOINTr LINESTRINGr*POLYGONr1 MULTIPOINTr(MULTILINESTRINGr/ MULTIPOLYGONr3 ValueErrorr9set)Z geometriesrZ geom_typestyprr"rrrr sVm           cCs@t|}t|jdd}|rs,        $