def get_or_create(self, uri, reprocess=False, *args, **kwargs):
# ingest file to db
ds, created = super(DatasetManager,
self).get_or_create(uri, *args, **kwargs)
# set Dataset entry_title
ds.entry_title = 'SAR NRCS'
ds.save()
# Unless reprocess==True, we may not need to do the following... (see
# managers.py in sar doppler processor)
#visExists = ... # check if visualization(s) already created
#if visExists and not reprocess:
# warnings.warn('NO VISUALISATIONS CREATED - update managers.py')
# return ds, created
n = Nansat(nansat_filename(uri))
n.reproject_GCPs()
n.resize(pixelsize=500)
lon, lat = n.get_corners()
lat_max = min(lat.max(), 85)
d = Domain(
NSR(3857), '-lle %f %f %f %f -ts %d %d' %
(lon.min(), lat.min(), lon.max(), lat_max, n.shape()[1],
n.shape()[0]))
# Get all NRCS bands
s0bands = []
pp = []
for key, value in n.bands().iteritems():
try:
if value['standard_name'] == standard_name:
s0bands.append(key)
pp.append(value['polarization'])
except KeyError:
continue
''' Create data products
'''
mm = self.__module__.split('.')
module = '%s.%s' % (mm[0], mm[1])
mp = media_path(module, n.fileName)
# ppath = product_path(module, n.fileName)
# Create png's for each band
num_products = 0
for band in s0bands:
print 'Visualize', band
s0_tmp = n[band]
n_tmp = Nansat(domain=n, array=s0_tmp)
n_tmp.reproject_GCPs()
n_tmp.reproject(d)
s0 = n_tmp[1]
n_tmp = None
mask = np.ones(s0.shape, np.uint8)
mask[np.isnan(s0) + (s0 <= 0)] = 0
s0 = np.log10(s0) * 10.
meta = n.bands()[band]
product_filename = '%s_%s.png' % (meta['short_name'],
meta['polarization'])
nansatFigure(s0, mask, polarization_clims[meta['polarization']][0],
polarization_clims[meta['polarization']][1], mp,
product_filename)
# Get DatasetParameter
param = Parameter.objects.get(short_name=meta['short_name'])
dsp, created = DatasetParameter.objects.get_or_create(
dataset=ds, parameter=param)
# Create Visualization
geom, created = GeographicLocation.objects.get_or_create(
geometry=WKTReader().read(n.get_border_wkt()))
vv, created = Visualization.objects.get_or_create(
uri='file://localhost%s/%s' % (mp, product_filename),
title='%s %s polarization' %
(param.standard_name, meta['polarization']),
geographic_location=geom)
# Create VisualizationParameter
vp, created = VisualizationParameter.objects.get_or_create(
visualization=vv, ds_parameter=dsp)
return ds, True
def get_or_create(self, uri, reprocess=False, *args, **kwargs):
# ingest file to db
ds, created = super(DatasetManager,
self).get_or_create(uri, *args, **kwargs)
fn = nansat_filename(uri)
n = Nansat(fn)
# Reproject to leaflet projection
xlon, xlat = n.get_corners()
d = Domain(
NSR(3857), '-lle %f %f %f %f -tr 1000 1000' %
(xlon.min(), xlat.min(), xlon.max(), xlat.max()))
n.reproject(d)
# Get band numbers of required bands according to standard names
speedBandNum = n._get_band_number({'standard_name': 'wind_speed'})
dirBandNum = n._get_band_number(
{'standard_name': 'wind_from_direction'})
# Get numpy arrays of the bands
speed = n[speedBandNum]
dir = n[dirBandNum]
## It probably wont work with nansatmap...
#nmap = Nansatmap(n, resolution='l')
#nmap.pcolormesh(speed, vmax=18)
#nmap.quiver(-speed*np.sin(dir), speed*np.cos(dir), step=10, scale=300,
# width=0.002)
# Set paths - this code should be inherited but I think there is an
# issue in generalising the first line that defines the current module
mm = self.__module__.split('.')
module = '%s.%s' % (mm[0], mm[1])
mp = media_path(module, n.fileName)
ppath = product_path(module, n.fileName)
filename = os.path.basename(n.fileName).split('.')[0] + '.' + \
os.path.basename(n.fileName).split('.')[1] + '.png'
# check uniqueness of parameter
param1 = Parameter.objects.get(standard_name=n.get_metadata(
bandID=speedBandNum, key='standard_name'))
param2 = Parameter.objects.get(standard_name=n.get_metadata(
bandID=dirBandNum, key='standard_name'))
n.write_figure(os.path.join(mp, filename),
bands=speedBandNum,
mask_array=n['swathmask'],
mask_lut={0: [128, 128, 128]},
transparency=[128, 128, 128])
# Get DatasetParameter
dsp1, created = DatasetParameter.objects.get_or_create(
dataset=ds, parameter=param1)
# Create Visualization
geom, created = GeographicLocation.objects.get_or_create(
geometry=WKTReader().read(n.get_border_wkt()))
vv, created = Visualization.objects.get_or_create(
uri='file://localhost%s/%s' % (mp, filename),
title='%s' % (param1.standard_name),
geographic_location=geom)
# Create VisualizationParameter
vp, created = VisualizationParameter.objects.get_or_create(
visualization=vv, ds_parameter=dsp1)
return ds, True
