Source code for temporal.temporal_raster3d_algebra

"""!@package grass.temporal

Temporal raster algebra

(C) 2013 by the GRASS Development Team
This program is free software under the GNU General Public
License (>=v2). Read the file COPYING that comes with GRASS
for details.

:authors: Thomas Leppelt and Soeren Gebbert

"""
from __future__ import print_function

try:
    import ply.lex as lex
    import ply.yacc as yacc
except:
    pass

from .temporal_raster_base_algebra import TemporalRasterBaseAlgebraParser,\
    TemporalRasterAlgebraLexer
import grass.pygrass.modules as pymod
from .space_time_datasets import Raster3DDataset


###############################################################################

[docs]class TemporalRaster3DAlgebraParser(TemporalRasterBaseAlgebraParser): """The temporal raster algebra class""" def __init__(self, pid=None, run=False, debug=True, spatial=False, register_null=False, dry_run=False, nprocs=1): TemporalRasterBaseAlgebraParser.__init__(self, pid=pid, run=run, debug=debug, spatial=spatial, register_null=register_null, dry_run=dry_run, nprocs=nprocs) self.m_mapcalc = pymod.Module('r3.mapcalc') self.m_mremove = pymod.Module('g.remove')
[docs] def parse(self, expression, basename = None, overwrite=False): # Check for space time dataset type definitions from temporal algebra l = TemporalRasterAlgebraLexer() l.build() l.lexer.input(expression) while True: tok = l.lexer.token() if not tok: break if tok.type == "STVDS" or tok.type == "STRDS" or tok.type == "STR3DS": raise SyntaxError("Syntax error near '%s'" %(tok.type)) self.lexer = TemporalRasterAlgebraLexer() self.lexer.build() self.parser = yacc.yacc(module=self, debug=self.debug, write_tables=False) self.overwrite = overwrite self.count = 0 self.stdstype = "str3ds" self.maptype = "raster_3d" self.mapclass = Raster3DDataset self.basename = basename self.expression = expression self.parser.parse(expression) return self.process_chain_dict
######################### Temporal functions ##############################
[docs] def p_statement_assign(self, t): # The expression should always return a list of maps. """ statement : stds EQUALS expr """ TemporalRasterBaseAlgebraParser.p_statement_assign(self, t)
[docs] def p_ts_neighbor_operation(self, t): # Examples: # A[1,0,-1] # B[-2] # C[1,-2,1,3] """ expr : stds L_SPAREN number COMMA number COMMA number R_SPAREN | stds L_SPAREN number R_SPAREN | stds L_SPAREN number COMMA number COMMA number COMMA number R_SPAREN """ # Check input stds. maplist = self.check_stds(t[1]) t_neighbor = 0 row_neighbor = None col_neighbor = None depth_neighbor = None if len(t) == 5: t_neighbor = t[3] elif len(t) == 9: row_neighbor = t[3] col_neighbor = t[5] depth_neighbor = t[7] elif len(t) == 11: t_neighbor = t[9] row_neighbor = t[3] col_neighbor = t[5] depth_neighbor = t[7] if self.run: resultlist = [] max_index = len(maplist) for map_i in maplist: # Get map index and temporal extent. map_index = maplist.index(map_i) new_index = map_index + t_neighbor if new_index < max_index and new_index >= 0: map_i_t_extent = map_i.get_temporal_extent() # Get neighboring map and set temporal extent. map_n = maplist[new_index] # Generate an intermediate map for the result map list. map_new = self.generate_new_map(map_n, bool_op = 'and', copy = True) map_new.set_temporal_extent(map_i_t_extent) # Create r.mapcalc expression string for the operation. if "cmd_list" in dir(map_new) and len(t) == 5: cmdstring = "%s" %(map_new.cmd_list) elif "cmd_list" not in dir(map_new) and len(t) == 5: cmdstring = "%s" %(map_n.get_id()) elif "cmd_list" in dir(map_new) and len(t) in (9,11): cmdstring = "%s[%s,%s,%s]" %(map_new.cmd_list, row_neighbor, col_neighbor, depth_neighbor) elif "cmd_list" not in dir(map_new) and len(t) in (9,11): cmdstring = "%s[%s,%s,%s]" %(map_n.get_id(), row_neighbor, col_neighbor, depth_neighbor) # Set new command list for map. map_new.cmd_list = cmdstring # Append map with updated command list to result list. resultlist.append(map_new) t[0] = resultlist
############################################################################### if __name__ == "__main__": import doctest doctest.testmod()