def transform(self, config='', **kwargs):
""" Transform the data using the config file """
if 'crs_target' in kwargs:
crs_target = kwargs.pop('crs_target', 'EPSG:4326')
self.epsg = crs_target
else:
crs_target = 'EPSG:4326'
self.epsg = 'EPSG:4326'
self.wkt = osr.SRS_WKT_WGS84
crs_target = get_coordinate_rerefence_system(extract_epsg(crs_target))
crs_source = get_coordinate_rerefence_system(extract_epsg(self.crs))
if config:
config = Models(config).config
self.properties = config_to_properties(config)
for x, feature in enumerate(self.features):
# Detect if correct geometry (Multipoints > Point)
feature = confirm_geometry(feature)
# Reproject data to target projection (crs_target=4326)
if kwargs.get('reproject', False):
reproject(feature, crs_source, crs_target)
# Transform Columns
if config:
feature = transform_columns(feature, config, **kwargs)
# Save feature to self
self[x] = feature
def read_shp(self, infile, **kwargs):
"""Reads a Shapefile and gives the results in GeoJSON format
Speed = 0.35ms/feature
"""
logging.info('Reading Shapefile: %s' % infile)
with fiona.drivers():
with fiona.open(infile) as source:
self.geometry = set([source.meta['schema']['geometry']])
self.properties = source.meta['schema']['properties']
# Read EPSG
crs = source.meta['crs']
if 'init' in crs:
self.epsg = crs['init'].upper()
elif 'crs_wkt' in source.meta:
self.wkt = source.meta['crs_wkt']
else:
logging.warning('Coordinate Reference System was not detected (default=EPSG:4326)')
self.epsg = 'EPSG:4326'
for feature in source:
if feature:
if feature['geometry']:
feature = confirm_geometry(feature)
self.features.append(feature)
else:
logging.warning('Could not find [geometry] in feature.')
def read_geojson(self, infile, **kwargs):
"""Reads a GeoJSON and gives the results in GeoJSON format
Speed = 0.091ms/feature"""
logging.info('Reading GeoJSON: %s' % infile)
if isinstance(infile, dict):
geojson = infile
else:
with open(infile) as f:
geojson = json.load(f)
if 'type' not in geojson:
raise ValueError('GeoJSON must contain a [type] "FeatureCollection" or "Feature"')
if 'crs' not in geojson:
logging.warning('Coordinate Reference System was not detected (default=EPSG:4326)')
self.epsg = 'EPSG:4326'
else:
self.epsg = 'EPSG:{}'.format(extract_epsg(geojson['crs']))
# Read GeoJSON Feature Collection
if geojson['type'] == 'FeatureCollection':
if not geojson['features']:
raise ValueError('FeatureCollection has [0] features.')
features = geojson['features']
# Read Single GeoJSON Feature
elif geojson['type'] == 'Feature':
features = [geojson]
properties = set()
self.geometry = set()
for feature in features:
# Add unique attribute keys to properties
if 'properties' in feature:
properties.update(feature['properties'].keys())
# Only add features with geometry
if feature.get('geometry'):
feature = confirm_geometry(feature)
self.features.append(feature)
self.geometry.update([feature['geometry']['type']])
else:
logging.warning('Could not find [geometry] in feature.')
# Creating basic properties for attribute table when building a shapefile.
self.properties = dict((key, 'str') for key in properties)
