def test_get_dataset_stack():
tiles = list_tiles_as_list(x=[120], y=[-20], satellites=[Satellite.LS5, Satellite.LS7],
acq_min=date(2014, 1, 1), acq_max=date(2014, 12, 31),
dataset_types=[DatasetType.ARG25, DatasetType.PQ25])
_log.info("\nFound %d tiles", len(tiles))
stack = get_dataset_data_stack(tiles, DatasetType.ARG25, Ls57Arg25Bands.BLUE.name, ndv=NDV,
mask_pqa_apply=True, mask_pqa_mask=[PqaMask.PQ_MASK_CLEAR])
_log.info("\nStack is %s", numpy.shape(stack))
def read_pixel_time_series(x, y, satellites, acq_min, acq_max, season, dataset_type,
mask_pqa_apply, mask_pqa_mask, band, x_offset, y_offset):
ndv = get_dataset_type_ndv(dataset_type)
tiles = get_tiles(x, y, satellites, acq_min, acq_max, season, dataset_type, mask_pqa_apply)
stack = get_dataset_data_stack(tiles, dataset_type, band.name, ndv=ndv,
x=x_offset, y=y_offset,
x_size=1, y_size=1,
mask_pqa_apply=mask_pqa_apply, mask_pqa_mask=mask_pqa_mask)
return [tile.end_datetime for tile in tiles], [s[0][0] for s in stack]
def run(self):
_log.info("Calculating statistics for chunk")
filtile = []
ndv = get_dataset_type_ndv(self.dataset_type)
data_type = get_dataset_type_data_type(self.dataset_type)
tiles = self.get_tiles()
filtile = tiles
if self.tidal_workflow:
filtile = []
lines = self.load_filterfile()
cnt = 0
_log.info("\tlength of original tiles is %d", len(tiles))
for tile in tiles:
#import pdb; pdb.set_trace()
cnt = cnt + 1
dataset = tile.datasets[self.dataset_type]
tdate = str(tile.end_datetime.strftime("%Y-%m-%d"))
if tdate in lines:
filtile.append(tile)
_log.info("\tlength of new filtered tiles is %d", len(filtile))
stack = get_dataset_data_stack(filtile,
self.dataset_type,
self.band.name,
ndv=ndv,
x=self.x_offset,
y=self.y_offset,
x_size=self.x_chunk_size,
y_size=self.y_chunk_size,
mask_pqa_apply=self.mask_pqa_apply,
mask_pqa_mask=self.mask_pqa_mask)
if len(stack) == 0:
return
# TODO get statistics to be generated from command line argument
if Statistic.COUNT in self.statistics:
#log_mem("Before COUNT")
# COUNT
stack_stat = calculate_stack_statistic_count(stack=stack, ndv=ndv)
numpy.save(self.get_statistic_filename(Statistic.COUNT),
stack_stat)
del stack_stat
if Statistic.MIN in self.statistics:
log_mem("Before MIN")
# MIN
stack_stat = calculate_stack_statistic_min(stack=stack,
ndv=ndv,
dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.MIN), stack_stat)
del stack_stat
if Statistic.MAX in self.statistics:
log_mem("Before MAX")
# MAX
stack_stat = calculate_stack_statistic_max(stack=stack,
ndv=ndv,
dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.MAX), stack_stat)
del stack_stat
if Statistic.MEAN in self.statistics:
log_mem("Before MEAN")
# MEAN
stack_stat = calculate_stack_statistic_mean(stack=stack,
ndv=ndv,
dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.MEAN), stack_stat)
del stack_stat
if Statistic.MEDIAN in self.statistics:
#log_mem("Before MEDIAN")
# MEAN
stack_stat = calculate_stack_statistic_median(stack=stack,
ndv=ndv,
dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.MEDIAN),
stack_stat)
del stack_stat
if Statistic.VARIANCE in self.statistics:
log_mem("Before VARIANCE")
# VARIANCE
stack_stat = calculate_stack_statistic_variance(stack=stack,
ndv=ndv,
dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.VARIANCE),
stack_stat)
del stack_stat
if Statistic.STANDARD_DEVIATION in self.statistics:
#log_mem("Before STANDARD_DEVIATION")
# STANDARD_DEVIATION
stack_stat = calculate_stack_statistic_standard_deviation(
stack=stack, ndv=ndv, dtype=data_type)
numpy.save(
self.get_statistic_filename(Statistic.STANDARD_DEVIATION),
stack_stat)
del stack_stat
for percentile in PERCENTILE:
if percentile in self.statistics:
log_mem("Before {p}".format(p=percentile.name))
stack_stat = calculate_stack_statistic_percentile(
stack=stack,
percentile=PERCENTILE[percentile],
ndv=ndv,
interpolation=self.interpolation)
numpy.save(self.get_statistic_filename(percentile), stack_stat)
del stack_stat
if Statistic.COUNT_OBSERVED in self.statistics:
#log_mem("Before OBSERVED COUNT")
# COUNT OBSERVED - note the copy=False is modifying the array so this is done last
stack_stat = calculate_stack_statistic_count_observed(stack=stack,
ndv=ndv)
numpy.save(self.get_statistic_filename(Statistic.COUNT_OBSERVED),
stack_stat)
del stack_stat
def run(self):
_log.info("Calculating statistics for chunk")
filtile = [ ]
ndv = get_dataset_type_ndv(self.dataset_type)
data_type = get_dataset_type_data_type(self.dataset_type)
tiles = self.get_tiles()
filtile = tiles
if self.tidal_workflow:
filtile = [ ]
lines = self.load_filterfile()
cnt=0
_log.info("\tlength of original tiles is %d", len(tiles))
for tile in tiles:
#import pdb; pdb.set_trace()
cnt=cnt+1
dataset = tile.datasets[self.dataset_type]
tdate= str(tile.end_datetime.strftime("%Y-%m-%d"))
if tdate in lines:
filtile.append(tile)
_log.info("\tlength of new filtered tiles is %d", len(filtile))
stack = get_dataset_data_stack(filtile, self.dataset_type, self.band.name, ndv=ndv,
x=self.x_offset, y=self.y_offset,
x_size=self.x_chunk_size, y_size=self.y_chunk_size,
mask_pqa_apply=self.mask_pqa_apply, mask_pqa_mask=self.mask_pqa_mask)
if len(stack) == 0:
return
# TODO get statistics to be generated from command line argument
if Statistic.COUNT in self.statistics:
#log_mem("Before COUNT")
# COUNT
stack_stat = calculate_stack_statistic_count(stack=stack, ndv=ndv)
numpy.save(self.get_statistic_filename(Statistic.COUNT), stack_stat)
del stack_stat
if Statistic.MIN in self.statistics:
log_mem("Before MIN")
# MIN
stack_stat = calculate_stack_statistic_min(stack=stack, ndv=ndv, dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.MIN), stack_stat)
del stack_stat
if Statistic.MAX in self.statistics:
log_mem("Before MAX")
# MAX
stack_stat = calculate_stack_statistic_max(stack=stack, ndv=ndv, dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.MAX), stack_stat)
del stack_stat
if Statistic.MEAN in self.statistics:
log_mem("Before MEAN")
# MEAN
stack_stat = calculate_stack_statistic_mean(stack=stack, ndv=ndv, dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.MEAN), stack_stat)
del stack_stat
if Statistic.MEDIAN in self.statistics:
#log_mem("Before MEDIAN")
# MEAN
stack_stat = calculate_stack_statistic_median(stack=stack, ndv=ndv, dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.MEDIAN), stack_stat)
del stack_stat
if Statistic.VARIANCE in self.statistics:
log_mem("Before VARIANCE")
# VARIANCE
stack_stat = calculate_stack_statistic_variance(stack=stack, ndv=ndv, dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.VARIANCE), stack_stat)
del stack_stat
if Statistic.STANDARD_DEVIATION in self.statistics:
#log_mem("Before STANDARD_DEVIATION")
# STANDARD_DEVIATION
stack_stat = calculate_stack_statistic_standard_deviation(stack=stack, ndv=ndv, dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.STANDARD_DEVIATION), stack_stat)
del stack_stat
for percentile in PERCENTILE:
if percentile in self.statistics:
log_mem("Before {p}".format(p=percentile.name))
stack_stat = calculate_stack_statistic_percentile(stack=stack, percentile=PERCENTILE[percentile],
ndv=ndv, interpolation=self.interpolation)
numpy.save(self.get_statistic_filename(percentile), stack_stat)
del stack_stat
if Statistic.COUNT_OBSERVED in self.statistics:
#log_mem("Before OBSERVED COUNT")
# COUNT OBSERVED - note the copy=False is modifying the array so this is done last
stack_stat = calculate_stack_statistic_count_observed(stack=stack, ndv=ndv)
numpy.save(self.get_statistic_filename(Statistic.COUNT_OBSERVED), stack_stat)
del stack_stat
def run(self):
_log.info("Calculating statistics for chunk")
ndv = get_dataset_type_ndv(self.dataset_type)
acq_min, acq_max, criteria = build_season_date_criteria(self.acq_min, self.acq_max, self.season,
seasons=SEASONS, extend=True)
_log.info("\tcriteria is %s", criteria)
dataset_types = [self.dataset_type]
if self.mask_pqa_apply:
dataset_types.append(DatasetType.PQ25)
tiles = list_tiles_as_generator(x=[self.x], y=[self.y], satellites=self.satellites,
acq_min=acq_min, acq_max=acq_max,
dataset_types=dataset_types, include=criteria)
stack = get_dataset_data_stack(tiles, self.dataset_type, self.band.name, ndv=ndv,
x=self.x_offset, y=self.y_offset,
x_size=self.x_chunk_size, y_size=self.y_chunk_size,
mask_pqa_apply=self.mask_pqa_apply, mask_pqa_mask=self.mask_pqa_mask)
if len(stack) == 0:
return
# TODO get statistics to be generated from command line argument
if Statistic.COUNT in self.statistics:
log_mem("Before COUNT")
# COUNT
print "COUNT"
stack_stat = calculate_stack_statistic_count(stack=stack, ndv=ndv)
numpy.save(self.get_statistic_filename(Statistic.COUNT), stack_stat)
del stack_stat
if Statistic.MIN in self.statistics:
log_mem("Before MIN")
# MIN
print "MIN"
stack_stat = calculate_stack_statistic_min(stack=stack, ndv=ndv)
numpy.save(self.get_statistic_filename(Statistic.MIN), stack_stat)
del stack_stat
if Statistic.MAX in self.statistics:
log_mem("Before MAX")
# MAX
print "MAX"
stack_stat = calculate_stack_statistic_max(stack=stack, ndv=ndv)
numpy.save(self.get_statistic_filename(Statistic.MAX), stack_stat)
del stack_stat
if Statistic.MEAN in self.statistics:
log_mem("Before MEAN")
# MEAN
print "MEAN"
stack_stat = calculate_stack_statistic_mean(stack=stack, ndv=ndv)
numpy.save(self.get_statistic_filename(Statistic.MEAN), stack_stat)
del stack_stat
for percentile in [Statistic.PERCENTILE_25, Statistic.PERCENTILE_50, Statistic.PERCENTILE_75, Statistic.PERCENTILE_90, Statistic.PERCENTILE_95]:
if percentile in self.statistics:
log_mem("Before {p}".format(p=percentile.name))
print "Before {p}".format(p=percentile.name)
stack_stat = calculate_stack_statistic_percentile(stack=stack, percentile=PERCENTILE[percentile], ndv=ndv)
numpy.save(self.get_statistic_filename(percentile), stack_stat)
del stack_stat
if Statistic.COUNT_OBSERVED in self.statistics:
log_mem("Before OBSERVED COUNT")
# COUNT OBSERVED - note the copy=False is modifying the array so this is done last
print "COUNT OBSERVED"
stack_stat = calculate_stack_statistic_count_observed(stack=stack, ndv=ndv)
numpy.save(self.get_statistic_filename(Statistic.COUNT_OBSERVED), stack_stat)
del stack_stat
log_mem("DONE")
def run(self):
_log.info("Calculating statistics for chunk")
ndv = get_dataset_type_ndv(self.dataset_type)
data_type = get_dataset_type_data_type(self.dataset_type)
tiles = self.get_tiles()
stack = get_dataset_data_stack(tiles, self.dataset_type, self.band.name, ndv=ndv,
x=self.x_offset, y=self.y_offset,
x_size=self.x_chunk_size, y_size=self.y_chunk_size,
mask_pqa_apply=self.mask_pqa_apply, mask_pqa_mask=self.mask_pqa_mask)
if len(stack) == 0:
return
# TODO get statistics to be generated from command line argument
if Statistic.COUNT in self.statistics:
log_mem("Before COUNT")
# COUNT
stack_stat = calculate_stack_statistic_count(stack=stack, ndv=ndv)
numpy.save(self.get_statistic_filename(Statistic.COUNT), stack_stat)
del stack_stat
if Statistic.MIN in self.statistics:
log_mem("Before MIN")
# MIN
stack_stat = calculate_stack_statistic_min(stack=stack, ndv=ndv, dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.MIN), stack_stat)
del stack_stat
if Statistic.MAX in self.statistics:
log_mem("Before MAX")
# MAX
stack_stat = calculate_stack_statistic_max(stack=stack, ndv=ndv, dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.MAX), stack_stat)
del stack_stat
if Statistic.MEAN in self.statistics:
log_mem("Before MEAN")
# MEAN
stack_stat = calculate_stack_statistic_mean(stack=stack, ndv=ndv, dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.MEAN), stack_stat)
del stack_stat
if Statistic.VARIANCE in self.statistics:
log_mem("Before VARIANCE")
# VARIANCE
stack_stat = calculate_stack_statistic_variance(stack=stack, ndv=ndv, dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.VARIANCE), stack_stat)
del stack_stat
if Statistic.STANDARD_DEVIATION in self.statistics:
log_mem("Before STANDARD_DEVIATION")
# STANDARD_DEVIATION
stack_stat = calculate_stack_statistic_standard_deviation(stack=stack, ndv=ndv, dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.STANDARD_DEVIATION), stack_stat)
del stack_stat
for percentile in PERCENTILE:
if percentile in self.statistics:
log_mem("Before {p}".format(p=percentile.name))
stack_stat = calculate_stack_statistic_percentile(stack=stack, percentile=PERCENTILE[percentile],
ndv=ndv, interpolation=self.interpolation)
numpy.save(self.get_statistic_filename(percentile), stack_stat)
del stack_stat
if Statistic.COUNT_OBSERVED in self.statistics:
log_mem("Before OBSERVED COUNT")
# COUNT OBSERVED - note the copy=False is modifying the array so this is done last
stack_stat = calculate_stack_statistic_count_observed(stack=stack, ndv=ndv)
numpy.save(self.get_statistic_filename(Statistic.COUNT_OBSERVED), stack_stat)
del stack_stat
log_mem("DONE")
def run(self):
_log.info("Calculating statistics for chunk")
ndv = get_dataset_type_ndv(self.dataset_type)
data_type = get_dataset_type_data_type(self.dataset_type)
tiles = self.get_tiles()
stack = get_dataset_data_stack(tiles, self.dataset_type, self.band.name, ndv=ndv,
x=self.x_offset, y=self.y_offset,
x_size=self.x_chunk_size, y_size=self.y_chunk_size,
mask_pqa_apply=self.mask_pqa_apply, mask_pqa_mask=self.mask_pqa_mask)
if len(stack) == 0:
return
# TODO get statistics to be generated from command line argument
if Statistic.COUNT in self.statistics:
log_mem("Before COUNT")
# COUNT
stack_stat = calculate_stack_statistic_count(stack=stack, ndv=ndv)
numpy.save(self.get_statistic_filename(Statistic.COUNT), stack_stat)
del stack_stat
if Statistic.MIN in self.statistics:
log_mem("Before MIN")
# MIN
stack_stat = calculate_stack_statistic_min(stack=stack, ndv=ndv, dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.MIN), stack_stat)
del stack_stat
if Statistic.MAX in self.statistics:
log_mem("Before MAX")
# MAX
stack_stat = calculate_stack_statistic_max(stack=stack, ndv=ndv, dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.MAX), stack_stat)
del stack_stat
if Statistic.MEAN in self.statistics:
log_mem("Before MEAN")
# MEAN
stack_stat = calculate_stack_statistic_mean(stack=stack, ndv=ndv, dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.MEAN), stack_stat)
del stack_stat
if Statistic.VARIANCE in self.statistics:
log_mem("Before VARIANCE")
# VARIANCE
stack_stat = calculate_stack_statistic_variance(stack=stack, ndv=ndv, dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.VARIANCE), stack_stat)
del stack_stat
if Statistic.STANDARD_DEVIATION in self.statistics:
log_mem("Before STANDARD_DEVIATION")
# STANDARD_DEVIATION
stack_stat = calculate_stack_statistic_standard_deviation(stack=stack, ndv=ndv, dtype=data_type)
numpy.save(self.get_statistic_filename(Statistic.STANDARD_DEVIATION), stack_stat)
del stack_stat
for percentile in PERCENTILE:
if percentile in self.statistics:
log_mem("Before {p}".format(p=percentile.name))
stack_stat = calculate_stack_statistic_percentile(stack=stack, percentile=PERCENTILE[percentile],
ndv=ndv, interpolation=self.interpolation)
numpy.save(self.get_statistic_filename(percentile), stack_stat)
del stack_stat
if Statistic.COUNT_OBSERVED in self.statistics:
log_mem("Before OBSERVED COUNT")
# COUNT OBSERVED - note the copy=False is modifying the array so this is done last
stack_stat = calculate_stack_statistic_count_observed(stack=stack, ndv=ndv)
numpy.save(self.get_statistic_filename(Statistic.COUNT_OBSERVED), stack_stat)
del stack_stat
log_mem("DONE")