def __init__(self, back_ds, allocator, open_options, mode):
uid = uuid.uuid4()
with back_ds.acquire_driver_object(uid, allocator) as gdal_objs:
gdal_ds, lyr = gdal_objs
rect = None
if lyr is not None:
rect = lyr.GetExtent()
path = gdal_ds.GetDescription()
driver = gdal_ds.GetDriver().ShortName
sr = lyr.GetSpatialRef()
if sr is None:
wkt_stored = None
else:
wkt_stored = sr.ExportToWkt()
fields = BackGDALFileVector._fields_of_lyr(lyr)
type = conv.str_of_wkbgeom(lyr.GetGeomType())
layer = lyr.GetName()
super(BackGDALFileVector, self).__init__(back_ds=back_ds,
wkt_stored=wkt_stored,
mode=mode,
driver=driver,
open_options=open_options,
path=path,
uid=uid,
layer=layer,
fields=fields,
rect=rect,
type=type)
self._type_of_field_index = [
conv.type_of_oftstr(field['type']) for field in self.fields
]
def acreate_vector(self,
path,
geometry,
fields=(),
layer=None,
driver='ESRI Shapefile',
options=(),
sr=None):
"""Create a vector file anonymously in this DataSource. Only metadata are kept in memory.
See DataSource.create_vector
Example
-------
>>> lines = ds.acreate_vector('/path/to.shp', 'linestring')
>>> file_proj4 = lines.proj4_stored
"""
# Parameter checking ***************************************************
path = str(path)
geometry = conv.str_of_wkbgeom(conv.wkbgeom_of_str(geometry))
fields = _tools.normalize_fields_defn(fields)
if layer is None:
layer = '.'.join(ntpath.basename(path).split('.')[:-1])
else:
layer = str(layer)
driver = str(driver)
options = [str(arg) for arg in options]
if sr is not None:
sr = osr.GetUserInputAsWKT(sr)
# Construction dispatch ************************************************
if driver.lower() == 'memory':
# TODO: Check not concurrent
allocator = lambda: BackGDALFileVector.create_file(
'', geometry, fields, layer, 'Memory', options, sr)
prox = GDALMemoryVector(self, allocator, options)
elif True:
allocator = lambda: BackGDALFileVector.create_file(
path, geometry, fields, layer, driver, options, sr)
prox = GDALFileVector(self, allocator, options, 'w')
else:
pass
# DataSource Registering ***********************************************
self._register([], prox)
return prox
def __init__(self, back_ds, allocator, open_options):
# gdal_ds, lyr = self.create_file('', geometry, fields, layer, 'Memory', open_options, sr)
gdal_ds, lyr = allocator()
self._gdal_ds = gdal_ds
self._lyr = lyr
rect = None
if lyr is not None:
rect = lyr.GetExtent()
path = gdal_ds.GetDescription()
driver = gdal_ds.GetDriver().ShortName
sr = lyr.GetSpatialRef()
if sr is None:
wkt_stored = None
else:
wkt_stored = sr.ExportToWkt()
fields = BackGDALMemoryVector._fields_of_lyr(lyr)
type = conv.str_of_wkbgeom(lyr.GetGeomType())
layer = lyr.GetName()
super(BackGDALMemoryVector, self).__init__(
back_ds=back_ds,
wkt_stored=wkt_stored,
mode='w',
driver=driver,
open_options=open_options,
path=path,
layer=layer,
fields=fields,
rect=rect,
type=type
)
self._type_of_field_index = [
conv.type_of_oftstr(field['type'])
for field in self.fields
]
def __init__(self, ds, **kwargs):
# Opening informations
self.mode = kwargs.pop('mode')
self.open_options = kwargs.pop('open_options')
self.layer = kwargs.pop('layer')
# GDAL informations
if 'gdal_ds' in kwargs:
gdal_ds = kwargs.pop('gdal_ds')
lyr = kwargs.pop('lyr')
kwargs['path'] = gdal_ds.GetDescription()
kwargs['driver'] = gdal_ds.GetDriver().ShortName
kwargs['type'] = conv.str_of_wkbgeom(lyr.GetGeomType())
kwargs['fields'] = Vector._fields_of_lyr(lyr)
wkt = lyr.GetSpatialRef()
if wkt is not None:
wkt = wkt.ExportToWkt()
kwargs['wkt'] = wkt
self.path = kwargs.pop('path')
self.driver = kwargs.pop('driver')
self.type = kwargs.pop('type')
self.fields = kwargs.pop('fields')
super(Vector._Constants, self).__init__(ds, **kwargs)
def create_vector(self,
key,
path,
geometry,
fields=(),
layer=None,
driver='ESRI Shapefile',
options=(),
sr=None):
"""Create a vector file and register it under `key` in this DataSource. Only metadata are
kept in memory.
Parameters
----------
key: hashable (like a string)
File identifier within DataSource
path: string
geometry: string
name of a wkb geometry type
http://www.gdal.org/ogr__core_8h.html#a800236a0d460ef66e687b7b65610f12a
(see example below)
fields: sequence of dict
Attributes of fields, one dict per field. (see `Field attributes` below)
layer: None or string
driver: string
ogr driver to use when opening the file
http://www.gdal.org/ogr_formats.html
options: sequence of str
options for ogr
sr: string or None
Spatial reference of the new file
if None: don't set a spatial reference
if string:
if path: Use same projection as file at `path`
if textual spatial reference:
http://gdal.org/java/org/gdal/osr/SpatialReference.html#SetFromUserInput-java.lang.String-
Returns
-------
one of {GDALFileVector, GDALMemoryVector} depending on the `driver` parameter
Field attributes
----------------
Attributes:
'name': string
'type': string (see `Field type` below)
'precision': int
'width': int
'nullable': bool
'default': same as `type`
An attribute missing or None is kept to default value.
Field types
-----------
Binary key: 'binary', bytes, np.bytes_, aliases of np.bytes_
Date key: 'date'
DateTime key: 'datetime', datetime.datetime, np.datetime64, aliases of np.datetime64
Time key: 'time'
Integer key: 'integer' np.int32, aliases of np.int32
Integer64 key: 'integer64', int, np.int64, aliases of np.int64
Real key: 'real', float, np.float64, aliases of np.float64
String key: 'string', str, np.str_, aliases of np.str_
Integer64List key: 'integer64list', 'int list'
IntegerList key: 'integerlist'
RealList key: 'reallist', 'float list'
StringList key: 'stringlist', 'str list'
Example
-------
>>> ds.create_vector('lines', '/path/to.shp', 'linestring')
>>> geometry_type = ds.lines.type
>>> fields = [
{'name': 'name', 'type': str},
{'name': 'count', 'type': 'int32'},
{'name': 'area', 'type': np.float64, 'width': 5, precision: 18},
{'name': 'when', 'type': np.datetime64},
]
>>> ds.create_vector('zones', '/path/to.shp', 'polygon', fields)
>>> field0_type = ds.zones.fields[0]['type']
"""
# Parameter checking ***************************************************
self._validate_key(key)
path = str(path)
geometry = conv.str_of_wkbgeom(conv.wkbgeom_of_str(geometry))
fields = _tools.normalize_fields_defn(fields)
if layer is None:
layer = '.'.join(ntpath.basename(path).split('.')[:-1])
else:
layer = str(layer)
driver = str(driver)
options = [str(arg) for arg in options]
if sr is not None:
sr = osr.GetUserInputAsWKT(sr)
# Construction dispatch ************************************************
if driver.lower() == 'memory':
# TODO: Check not concurrent
allocator = lambda: BackGDALFileVector.create_file(
'', geometry, fields, layer, 'Memory', options, sr)
prox = GDALMemoryVector(self, allocator, options)
elif True:
allocator = lambda: BackGDALFileVector.create_file(
path, geometry, fields, layer, driver, options, sr)
prox = GDALFileVector(self, allocator, options, 'w')
else:
pass
# DataSource Registering ***********************************************
self._register([key], prox)
return prox
