&hSrSSKrSSKJr SSKJr SSKJrJrJ r SSK J r SSK J r SSKJr SS KJrJrJrJr \ "5 SS KJrJrJrJr SS KJrJrJrJrJr SSS5 S S 1r S 1r!SSjr"SSjr#SSjr$Sr%Sr&Sr'SSjr(SSjr)SSjr*g!,(df  N=f)z7Low level functions to read and write OGR data sources.N)BytesIO)Path) HAS_ARROW_APIHAS_ARROW_WRITE_API HAS_PYARROW)GDALEnv)detect_write_driver)DataSourceError) _mask_to_wkb_preprocess_options_key_valueget_vsi_path_or_buffervsi_path)ogr_open_arrowogr_read ogr_writeogr_write_arrow)_get_driver_metadata_itemget_gdal_versionget_gdal_version_stringogr_driver_supports_vsiogr_driver_supports_write FlatGeobufGPKGc U(a [U5O0n[[U540SU_SU_SU_SU_SU_SU_SU=(d S_S U_S U _S [U 5_S U _S U _SU _SU_SU_SU_6$)a Read OGR data source into numpy arrays. IMPORTANT: non-linear geometry types (e.g., MultiSurface) are converted to their linear approximations. Parameters ---------- path_or_buffer : pathlib.Path or str, or bytes buffer A dataset path or URI, raw buffer, or file-like object with a read method. layer : int or str, optional (default: first layer) If an integer is provided, it corresponds to the index of the layer with the data source. If a string is provided, it must match the name of the layer in the data source. Defaults to first layer in data source. encoding : str, optional (default: None) If present, will be used as the encoding for reading string values from the data source. By default will automatically try to detect the native encoding and decode to ``UTF-8``. columns : list-like, optional (default: all columns) List of column names to import from the data source. Column names must exactly match the names in the data source, and will be returned in the order they occur in the data source. To avoid reading any columns, pass an empty list-like. If combined with ``where`` parameter, must include columns referenced in the ``where`` expression or the data may not be correctly read; the data source may return empty results or raise an exception (behavior varies by driver). read_geometry : bool, optional (default: True) If True, will read geometry into WKB. If False, geometry will be None. force_2d : bool, optional (default: False) If the geometry has Z values, setting this to True will cause those to be ignored and 2D geometries to be returned skip_features : int, optional (default: 0) Number of features to skip from the beginning of the file before returning features. If greater than available number of features, an empty DataFrame will be returned. Using this parameter may incur significant overhead if the driver does not support the capability to randomly seek to a specific feature, because it will need to iterate over all prior features. max_features : int, optional (default: None) Number of features to read from the file. where : str, optional (default: None) Where clause to filter features in layer by attribute values. If the data source natively supports SQL, its specific SQL dialect should be used (eg. SQLite and GeoPackage: `SQLITE`_, PostgreSQL). If it doesn't, the `OGRSQL WHERE`_ syntax should be used. Note that it is not possible to overrule the SQL dialect, this is only possible when you use the SQL parameter. Examples: ``"ISO_A3 = 'CAN'"``, ``"POP_EST > 10000000 AND POP_EST < 100000000"`` bbox : tuple of (xmin, ymin, xmax, ymax), optional (default: None) If present, will be used to filter records whose geometry intersects this box. This must be in the same CRS as the dataset. If GEOS is present and used by GDAL, only geometries that intersect this bbox will be returned; if GEOS is not available or not used by GDAL, all geometries with bounding boxes that intersect this bbox will be returned. Cannot be combined with ``mask`` keyword. mask : Shapely geometry, optional (default: None) If present, will be used to filter records whose geometry intersects this geometry. This must be in the same CRS as the dataset. If GEOS is present and used by GDAL, only geometries that intersect this geometry will be returned; if GEOS is not available or not used by GDAL, all geometries with bounding boxes that intersect the bounding box of this geometry will be returned. Requires Shapely >= 2.0. Cannot be combined with ``bbox`` keyword. fids : array-like, optional (default: None) Array of integer feature id (FID) values to select. Cannot be combined with other keywords to select a subset (``skip_features``, ``max_features``, ``where``, ``bbox``, or ``mask``). Note that the starting index is driver and file specific (e.g. typically 0 for Shapefile and 1 for GeoPackage, but can still depend on the specific file). The performance of reading a large number of features usings FIDs is also driver specific. sql : str, optional (default: None) The SQL statement to execute. Look at the sql_dialect parameter for more information on the syntax to use for the query. When combined with other keywords like ``columns``, ``skip_features``, ``max_features``, ``where``, ``bbox``, or ``mask``, those are applied after the SQL query. Be aware that this can have an impact on performance, (e.g. filtering with the ``bbox`` or ``mask`` keywords may not use spatial indexes). Cannot be combined with the ``layer`` or ``fids`` keywords. sql_dialect : str, optional (default: None) The SQL dialect the ``sql`` statement is written in. Possible values: - **None**: if the data source natively supports SQL, its specific SQL dialect will be used by default (eg. SQLite and Geopackage: `SQLITE`_, PostgreSQL). If the data source doesn't natively support SQL, the `OGRSQL`_ dialect is the default. - '`OGRSQL`_': can be used on any data source. Performance can suffer when used on data sources with native support for SQL. - '`SQLITE`_': can be used on any data source. All spatialite_ functions can be used. Performance can suffer on data sources with native support for SQL, except for Geopackage and SQLite as this is their native SQL dialect. return_fids : bool, optional (default: False) If True, will return the FIDs of the feature that were read. datetime_as_string : bool, optional (default: False) If True, will return datetime dtypes as detected by GDAL as a string array (which can be used to extract timezone info), instead of a datetime64 array. **kwargs Additional driver-specific dataset open options passed to OGR. Invalid options will trigger a warning. Returns ------- (dict, fids, geometry, data fields) Returns a tuple of meta information about the data source in a dict, an ndarray of FIDs corresponding to the features that were read or None (if return_fids is False), an ndarray of geometry objects or None (if data source does not include geometry or read_geometry is False), a tuple of ndarrays for each field in the data layer. Meta is: { "crs": "", "fields": , "dtypes": "encoding": "", "geometry_type": "" } .. _OGRSQL: https://gdal.org/user/ogr_sql_dialect.html#ogr-sql-dialect .. _OGRSQL WHERE: https://gdal.org/user/ogr_sql_dialect.html#where .. _SQLITE: https://gdal.org/user/sql_sqlite_dialect.html#sql-sqlite-dialect .. _spatialite: https://www.gaia-gis.it/gaia-sins/spatialite-sql-latest.html layerencodingcolumns read_geometryforce_2d skip_features max_featuresrwherebboxmaskfidssql sql_dialect return_fidsdataset_kwargsdatetime_as_string)r rr r )path_or_bufferrrrrrr r!r"r#r$r%r&r'r(r*kwargsr)s ^C:\Users\julio\OneDrive\Documentos\Trabajo\Ideas Frescas\venv\Lib\site-packages\pyogrio/raw.pyreadr.'sz?E26:"N ~.      $     $ "&Q   $          &! ".# c [(d [S5eSSKJn [ 5nUS:a [ S5eUbUS:a [ S5eSU;aUR S5nOSnUbUU:aUnSnUS :aUnSn[U4UUUUUUU U U U U UUUS S .UD6nUunnUby/nSnUR5nURU5 U[U5- nUXg-:aOM<URUURS 9RXg5R5nODUS:a.UR!5RU5R5nOUR!5nSSS5 UU4$![a Mf=f!,(df  WW4$=f) aRead OGR data source into a pyarrow Table. See docstring of `read` for parameters. Returns ------- (dict, pyarrow.Table) Returns a tuple of meta information about the data source in a dict, and a pyarrow Table with data. Meta is: { "crs": "", "fields": , "encoding": "", "geometry_type": "", "geometry_name": "", } z{pyarrow required to read using 'read_arrow'. You can use 'open_arrow' to read data with an alternative Arrow implementationr)Tablez'skip_features' must be >= 0Nz'max_features' must be >= 0 batch_size)rT)rrrrrr"r#r$r%r&r'r(r r2 use_pyarrow)schema)r RuntimeErrorpyarrowr1r ValueErrorpop open_arrowread_next_batchappendlen StopIteration from_batchesr7slicecombine_chunksread_all)r+rrrrrr r!r"r#r$r%r&r'r(r,r1 gdal_versionr2gdal_skip_featuressourcemetareaderbatchescountbatchtables r- read_arrowrNsN ; D  #%Lq788L1$4677vZZ -  L:$=! y *  #    (! " # $  f  #GE "224ENN5)SZ'E!=>? ""76=="A}3!  Q OO%++M:IIKEOO%Ea d ;#%C  d ;s1E<(7E+BE<+ E95E<8E99E<< F c [(d [S5eU(a [U5O0n[[ U540SU_SU_SU_SU_SU_SU_SU=(d S _S U_S U _S [ U 5_S U _SU _SU _SU_SU_SU_SU_6$)a; Open OGR data source as a stream of Arrow record batches. See docstring of `read` for parameters. The returned object is reading from a stream provided by OGR and must not be accessed after the OGR dataset has been closed, i.e. after the context manager has been closed. By default this functions returns a generic stream object implementing the `Arrow PyCapsule Protocol`_ (i.e. having an ``__arrow_c_stream__`` method). This object can then be consumed by your Arrow implementation of choice that supports this protocol. Optionally, you can specify ``use_pyarrow=True`` to directly get the stream as a `pyarrow.RecordBatchReader`. .. _Arrow PyCapsule Protocol: https://arrow.apache.org/docs/format/CDataInterface/PyCapsuleInterface.html Other Parameters ---------------- batch_size : int (default: 65_536) Maximum number of features to retrieve in a batch. use_pyarrow : bool (default: False) If True, return a pyarrow RecordBatchReader instead of a generic ArrowStream object. In the default case, this stream object needs to be passed to another library supporting the Arrow PyCapsule Protocol to consume the stream of data. Examples -------- >>> from pyogrio.raw import open_arrow >>> import pyarrow as pa >>> import shapely >>> >>> with open_arrow(path) as source: >>> meta, stream = source >>> # wrap the arrow stream object in a pyarrow RecordBatchReader >>> reader = pa.RecordBatchReader.from_stream(stream) >>> geom_col = meta["geometry_name"] or "wkb_geometry" >>> for batch in reader: >>> geometries = shapely.from_wkb(batch[geom_col]) The returned `stream` object needs to be consumed by a library implementing the Arrow PyCapsule Protocol. In the above example, pyarrow is used through its RecordBatchReader. For this case, you can also specify ``use_pyarrow=True`` to directly get this result as a short-cut: >>> with open_arrow(path, use_pyarrow=True) as source: >>> meta, reader = source >>> geom_col = meta["geometry_name"] or "wkb_geometry" >>> for batch in reader: >>> geometries = shapely.from_wkb(batch[geom_col]) Returns ------- (dict, pyarrow.RecordBatchReader or ArrowStream) Returns a tuple of meta information about the data source in a dict, and a data stream object (a generic ArrowStream object, or a pyarrow RecordBatchReader if `use_pyarrow` is set to True). Meta is: { "crs": "", "fields": , "encoding": "", "geometry_type": "", "geometry_name": "", } z'GDAL>= 3.6 required to read using arrowrrrrrr r!rr"r#r$r%r&r'r(r)r2r6)rr8r rr r )r+rrrrrr r!r"r#r$r%r&r'r(r2r6r,r)s r-r<r<Vst =DEE>D26:"N ~.      $     $ "&Q   $          &! "# $ % r/cSSKJs Jn /nU(aiURU5nUR S5HCnUR R SS5S:wdM%URUR S5 ME U$)z&Convert metadata xml to list of names.rNOptionscopevectorrastername)xml.etree.ElementTreeetree ElementTree fromstringiterattribgetr>)xmlEToptionsrootoptions r-_parse_options_namesrbsh'&G }}S!ii)F}}  (3x?v}}V45* Nr/cUbUb [S5eUnX4HinUcMUR5HMupE[U[5(d[SUS35e[U[5(aM@[SUS35e Mk X4$)zValidate the metadata.z,Cannot pass both metadata and layer_metadataz metadata key z must be a stringzmetadata value )r:items isinstancestr)dataset_metadatalayer_metadatametadatarHkvs r-_validate_metadatarls  %KL L!"2   !!S))$}QC7H%IJJ!!S))$qc9J%KLL %3  ++r/cN[U=(d 05n[U=(d 05nU(at[U5n[[US55n[[US55nUR5H+upX;aXU'MX;aXU'M[ SUSUS35e XE4$)zBPreprocess kwargs and split in dataset and layer creation options.DMD_CREATIONOPTIONLISTDS_LAYER_CREATIONOPTIONLISTzunrecognized option 'z' for driver '')r rbrrdr:) driverdataset_options layer_optionsr,r) layer_kwargsdataset_option_nameslayer_option_namesrjrks r-_preprocess_options_kwargsrws2?3HbIN01D"EL .v63 %f.F G 2 %f.K L LLNDA($%q!("#Q #8>&QR!STT #  ''r/c[U[5(a]Uc [S5eUS;a[SU35e[U5(d[ US[ 535eU(a [ S5eOJ[US5(a [U[5(d [ S5e[U5nUc [U5n[U5(d[ US[ 535eU(aUS :Xa[5S ::a [S 5eX4$) aVValidate and return path and driver. Parameters ---------- path : str or io.BytesIO path to output file on writeable file system or an io.BytesIO object to allow writing to memory. Will raise NotImplementedError if an open file handle is passed. driver : str, optional (default: None) The OGR format driver used to write the vector file. By default attempts to infer driver from path. Must be provided to write to a file-like object. append : bool, optional (default: False) True if path and driver is being tested for append support Returns ------- (path, driver) z2driver must be provided to write to in-memory file> OpenFileGDBESRI Shapefilez/writing to in-memory file is not supported for z5 does not support ability to write in-memory in GDAL z+append is not supported for in-memory fileswritezwriting to an open file handle is not yet supported; instead, write to a BytesIO instance and then read bytes from that to write to the file handlez. does not support write functionality in GDAL r)r4rzGappend to FlatGeobuf is not supported for GDAL <= 3.5.0 due to segfault)rerr:rr rNotImplementedErrorhasattrrrr rrr8)pathrqr>s r-_get_write_path_driverrs/*$  >QR R 6 6NvhWX X'v..!(O*,-/  %&ST T  w   4(>(>! Y  ~ >(.F %V , ,hD&() +  &L(-=-?9-L U   <r/c |URSS5 [XU S9up[XU5upUb$U c!URS5=(a U[;n UbUc[ R "SSS9 [UUUU5unn[U40SU_S U_S U_S U_S U_S U_SU_SU_SU _SU _SU _SU _SU _SU_SU_SU_SU_6 g)aWrite geometry and field data to an OGR file format. Parameters ---------- path : str or io.BytesIO path to output file on writeable file system or an io.BytesIO object to allow writing to memory. Will raise NotImplementedError if an open file handle is passed; use BytesIO instead. NOTE: support for writing to memory is limited to specific drivers. geometry : ndarray of WKB encoded geometries or None If None, geometries will not be written to output file field_data : list-like of shape (num_fields, num_records) contains one record per field to be written in same order as fields fields : list-like contains field names field_mask : list-like of ndarrays or None, optional (default: None) contains mask arrays indicating null values of the field at the same position in the outer list, or None to indicate field does not have a mask array layer : str, optional (default: None) layer name to create. If writing to memory and layer name is not provided, it layer name will be set to a UUID4 value. driver : string, optional (default: None) The OGR format driver used to write the vector file. By default attempts to infer driver from path. Must be provided to write to memory. geometry_type : str, optional (default: None) Possible values are: "Unknown", "Point", "LineString", "Polygon", "MultiPoint", "MultiLineString", "MultiPolygon" or "GeometryCollection". This parameter does not modify the geometry, but it will try to force the layer type of the output file to this value. Use this parameter with caution because using a wrong layer geometry type may result in errors when writing the file, may be ignored by the driver, or may result in invalid files. crs : str, optional (default: None) WKT-encoded CRS of the geometries to be written. encoding : str, optional (default: None) If present, will be used as the encoding for writing string values to the file. Use with caution, only certain drivers support encodings other than UTF-8. promote_to_multi : bool, optional (default: None) If True, will convert singular geometry types in the data to their corresponding multi geometry type for writing. By default, will convert mixed singular and multi geometry types to multi geometry types for drivers that do not support mixed singular and multi geometry types. If False, geometry types will not be promoted, which may result in errors or invalid files when attempting to write mixed singular and multi geometry types to drivers that do not support such combinations. nan_as_null : bool, default True For floating point columns (float32 / float64), whether NaN values are written as "null" (missing value). Defaults to True because in pandas NaNs are typically used as missing value. Note that when set to False, behaviour is format specific: some formats don't support NaNs by default (e.g. GeoJSON will skip this property) or might treat them as null anyway (e.g. GeoPackage). append : bool, optional (default: False) If True, the data source specified by path already exists, and the driver supports appending to an existing data source, will cause the data to be appended to the existing records in the data source. Not supported for writing to in-memory files. NOTE: append support is limited to specific drivers and GDAL versions. dataset_metadata : dict, optional (default: None) Metadata to be stored at the dataset level in the output file; limited to drivers that support writing metadata, such as GPKG, and silently ignored otherwise. Keys and values must be strings. layer_metadata : dict, optional (default: None) Metadata to be stored at the layer level in the output file; limited to drivers that support writing metadata, such as GPKG, and silently ignored otherwise. Keys and values must be strings. metadata : dict, optional (default: None) alias of layer_metadata dataset_options : dict, optional Dataset creation options (format specific) passed to OGR. Specify as a key-value dictionary. layer_options : dict, optional Layer creation options (format specific) passed to OGR. Specify as a key-value dictionary. gdal_tz_offsets : dict, optional (default: None) Used to handle GDAL timezone offsets for each field contained in dict. **kwargs Additional driver-specific dataset creation options passed to OGR. Invalid options will trigger a warning. dtypesNr>Multi'crs' was not provided. The output dataset will not have projection information defined and may not be usable in other systems. stacklevelrrqgeometry geometry_type field_data field_maskfieldscrsrpromote_to_multi nan_as_nullr>rgrhr)rtgdal_tz_offsets) r;rrl startswithDRIVERS_NO_MIXED_SINGLE_MULTIwarningswarnrwr)rrrrrrrqrrrrrr>rgrhrirrrsrr,r)rts r-r{r{KsTZ JJx)$vFLD'9(($ 0 8  $ $W - 877       $>$ NL   $    * *& &!""#$(%r/c X[(d [S5e[US5(d [S5e[ XUS9upSU;a [S5eUb&Uc [S5eUc[ R "S S S 9 [XU 5up[X=3.8 required to write using arrow__arrow_c_stream__zThe provided data is not recognized as Arrow data. The object should implement the Arrow PyCapsule Protocol (i.e. have a '__arrow_c_stream__' method).rrzGThe 'promote_to_multi' option is not supported when writing using ArrowNz#'geometry_type' keyword is requiredrrr) rrq arrow_objr geometry_namerrr>rgrhr)rt) rr8r~r:rrrrlrwr)rrrrqrrrrr>rgrhrirrrsr,r)rts r- write_arrowrst  DEE 92 3 3 ,  *$vFLDV# U    BC C ; MM  (:(($ $>$ NL ## )%%!r/)NNNTFrNNNNNNNFF)NNNTFrNNNNNNNF)NNNTFrNNNNNNNFr3F)F)NNNNNNNTFNNNNNN) NNNNNNFNNNNN)+__doc__riorpathlibrpyogrio._compatrrr pyogrio._envr pyogrio.corer pyogrio.errorsr pyogrio.utilr r r r pyogrio._iorrrr pyogrio._ogrrrrrrrDRIVERS_NO_MIXED_DIMENSIONSr.rNr<rbrlrwrr{rr/r-rs=KK ,* YPP !        #qn       !x|       %rj",((.AR     )[B     NoYs C C*