12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331 |
- # -*- coding: utf-8 -*-
- """!@package grass.temporal
- @brief GRASS Python scripting module (temporal GIS functions)
- Temporal GIS related functions to be used in temporal GIS Python library package.
- Usage:
- @code
- import grass.temporal as tgis
- ...
- @endcode
- (C) 2008-2011 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.
- @author Soeren Gebbert
- """
- from abstract_dataset import *
- from temporal_granularity import *
- from temporal_relationships import *
- ###############################################################################
- class abstract_space_time_dataset(abstract_dataset):
- """!Abstract space time dataset class
- This class represents a space time dataset. Convenient functions
- to select, update, insert or delete objects of this type in the SQL
- temporal database exists as well as functions to register or unregister
- raster maps.
- Parts of the temporal logic are implemented in the SQL temporal database,
- like the computation of the temporal and spatial extent as well as the
- collecting of metadata.
- """
- def __init__(self, ident):
- self.reset(ident)
- self.map_counter = 0
- def get_new_map_instance(self, ident=None):
- """!Return a new instance of a map dataset which is associated with the type of this class
- @param ident: The unique identifier of the new object
- """
- raise IOError("This method must be implemented in the subclasses")
- def get_map_register(self):
- """!Return the name of the map register table"""
- raise IOError("This method must be implemented in the subclasses")
- def set_map_register(self, name):
- """!Set the name of the map register table
- This table stores all map names which are registered in this space time dataset.
- @param name: The name of the register table
- """
- raise IOError("This method must be implemented in the subclasses")
-
- def print_self(self):
- """!Print the content of the internal structure to stdout"""
- self.base.print_self()
- if self.is_time_absolute():
- self.absolute_time.print_self()
- if self.is_time_relative():
- self.relative_time.print_self()
- self.spatial_extent.print_self()
- self.metadata.print_self()
- def print_info(self):
- """!Print information about this class in human readable style"""
-
- if self.get_type() == "strds":
- # 1 2 3 4 5 6 7
- # 0123456789012345678901234567890123456789012345678901234567890123456789012345678
- print ""
- print " +-------------------- Space Time Raster Dataset -----------------------------+"
- if self.get_type() == "str3ds":
- # 1 2 3 4 5 6 7
- # 0123456789012345678901234567890123456789012345678901234567890123456789012345678
- print ""
- print " +-------------------- Space Time Raster3d Dataset ---------------------------+"
- if self.get_type() == "stvds":
- # 1 2 3 4 5 6 7
- # 0123456789012345678901234567890123456789012345678901234567890123456789012345678
- print ""
- print " +-------------------- Space Time Vector Dataset -----------------------------+"
- print " | |"
- self.base.print_info()
- if self.is_time_absolute():
- self.absolute_time.print_info()
- if self.is_time_relative():
- self.relative_time.print_info()
- self.spatial_extent.print_info()
- self.metadata.print_info()
- print " +----------------------------------------------------------------------------+"
- def print_shell_info(self):
- """!Print information about this class in shell style"""
- self.base.print_shell_info()
- if self.is_time_absolute():
- self.absolute_time.print_shell_info()
- if self.is_time_relative():
- self.relative_time.print_shell_info()
- self.spatial_extent.print_shell_info()
- self.metadata.print_shell_info()
- def set_initial_values(self, temporal_type, semantic_type, \
- title=None, description=None):
- """!Set the initial values of the space time dataset
- @param temporal_type: The temporal type of this space time dataset (absolute or relative)
- @param semantic_type: The semantic type of this dataset
- @param title: The title
- @param description: The description of this dataset
- """
- if temporal_type == "absolute":
- self.set_time_to_absolute()
- elif temporal_type == "relative":
- self.set_time_to_relative()
- else:
- core.fatal(_("Unknown temporal type \"%s\"") % (temporal_type))
- self.base.set_semantic_type(semantic_type)
- self.metadata.set_title(title)
- self.metadata.set_description(description)
- def get_semantic_type(self):
- """!Return the semantic type of this dataset"""
- return self.base.get_semantic_type()
- def get_initial_values(self):
- """!Return the initial values: temporal_type, semantic_type, title, description"""
-
- temporal_type = self.get_temporal_type()
- semantic_type = self.base.get_semantic_type()
- title = self.metadata.get_title()
- description = self.metadata.get_description()
- return temporal_type, semantic_type, title, description
- def get_granularity(self):
- """!Return the granularity"""
-
- temporal_type = self.get_temporal_type()
- if temporal_type == "absolute":
- granularity = self.absolute_time.get_granularity()
- elif temporal_type == "relative":
- granularity = self.relative_time.get_granularity()
- return granularity
- def set_granularity(self, granularity):
- temporal_type = self.get_temporal_type()
-
- if temporal_type == "absolute":
- self.set_time_to_absolute()
- self.absolute_time.set_granularity(granularity)
- elif temporal_type == "relative":
- self.set_time_to_relative()
- self.relative_time.set_granularity(granularity)
- else:
- core.fatal(_("Unknown temporal type \"%s\"") % (temporal_type))
- def set_relative_time_unit(self, unit):
- """!Set the relative time unit which may be of type: years, months, days, hours, minutes or seconds
-
- All maps registered in a (relative time) space time dataset must have the same unit
- """
- temporal_type = self.get_temporal_type()
-
- if temporal_type == "relative":
- if not self.check_relative_time_unit(unit):
- core.fatal(_("Unsupported temporal unit: %s") % (unit))
- self.relative_time.set_unit(unit)
- def get_map_time(self):
- """!Return the type of the map time, interval, point, mixed or invalid"""
-
- temporal_type = self.get_temporal_type()
- if temporal_type == "absolute":
- map_time = self.absolute_time.get_map_time()
- elif temporal_type == "relative":
- map_time = self.relative_time.get_map_time()
- return map_time
- def count_temporal_types(self, maps=None, dbif=None):
- """!Return the temporal type of the registered maps as dictionary
- The map list must be ordered by start time
- The temporal type can be:
- * point -> only the start time is present
- * interval -> start and end time
- * invalid -> No valid time point or interval found
- @param maps: A sorted (start_time) list of abstract_dataset objects
- @param dbif: The database interface to be used
- """
- if maps == None:
- maps = get_registered_maps_as_objects(where=None, order="start_time", dbif=dbif)
-
- time_invalid = 0
- time_point = 0
- time_interval = 0
- tcount = {}
- for i in range(len(maps)):
- # Check for point and interval data
- if maps[i].is_time_absolute():
- start, end, tz = maps[i].get_absolute_time()
- if maps[i].is_time_relative():
- start, end, unit = maps[i].get_relative_time()
- if start != None and end != None:
- time_interval += 1
- elif start != None and end == None:
- time_point += 1
- else:
- time_invalid += 1
- tcount["point"] = time_point
- tcount["interval"] = time_interval
- tcount["invalid"] = time_invalid
- return tcount
- def count_gaps(self, maps=None, dbif=None):
- """!Count the number of gaps between temporal neighbors
-
- @param maps: A sorted (start_time) list of abstract_dataset objects
- @param dbif: The database interface to be used
- @return The numbers of gaps between temporal neighbors
- """
- if maps == None:
- maps = get_registered_maps_as_objects(where=None, order="start_time", dbif=dbif)
-
- gaps = 0
- # Check for gaps
- for i in range(len(maps)):
- if i < len(maps) - 1:
- relation = maps[i + 1].temporal_relation(maps[i])
- if relation == "after":
- gaps += 1
- return gaps
-
- def print_temporal_relation_matrix(self, maps=None, dbif=None):
- """!Print the temporal relation matrix of all registered maps to stdout
- The temporal relation matrix includes the temporal relations between
- all registered maps. The relations are strings stored in a list of lists.
-
- @param maps: a ordered by start_time list of map objects
- @param dbif: The database interface to be used
- """
-
- if maps == None:
- maps = get_registered_maps_as_objects(where=None, order="start_time", dbif=dbif)
-
- print_temporal_relations(maps, maps)
- def get_temporal_relation_matrix(self, maps=None, dbif=None):
- """!Return the temporal relation matrix of all registered maps as list of lists
- The map list must be ordered by start time
- The temporal relation matrix includes the temporal relations between
- all registered maps. The relations are strings stored in a list of lists.
-
- @param maps: a ordered by start_time list of map objects
- @param dbif: The database interface to be used
- """
- if maps == None:
- maps = get_registered_maps_as_objects(where=None, order="start_time", dbif=dbif)
- return get_temporal_relation_matrix(maps, maps)
- def count_temporal_relations(self, maps=None, dbif=None):
- """!Count the temporal relations between the registered maps.
- The map list must be ordered by start time. Temporal relations are counted
- by analysing the sparse upper right side temporal relationships matrix.
- @param maps: A sorted (start_time) list of abstract_dataset objects
- @param dbif: The database interface to be used
- @return A dictionary with counted temporal relationships
- """
-
- if maps == None:
- maps = get_registered_maps_as_objects(where=None, order="start_time", dbif=dbif)
- return count_temporal_relations(maps, maps)
- def check_temporal_topology(self, maps=None, dbif=None):
- """!Check the temporal topology
- Correct topology means, that time intervals are not overlap or
- that intervals does not contain other intervals. Equal time intervals or
- points of time are not allowed.
- The map list must be ordered by start time
- Allowed and not allowed temporal relationships for correct topology
- after -> allowed
- before -> allowed
- follows -> allowed
- precedes -> allowed
- equivalent -> not allowed
- during -> not allowed
- contains -> not allowed
- overlaps -> not allowed
- overlapped -> not allowed
- starts -> not allowed
- finishes -> not allowed
- started -> not allowed
- finished -> not allowed
- @param maps: A sorted (start_time) list of abstract_dataset objects
- @return True if topology is correct
- """
- if maps == None:
- maps = get_registered_maps_as_objects(where=None, order="start_time", dbif=dbif)
- relations = self.count_temporal_relations(maps)
- map_time = self.get_map_time()
- if map_time == "interval" or map_time == "mixed":
- if relations.has_key("equivalent"):
- return False
- if relations.has_key("during"):
- return False
- if relations.has_key("contains"):
- return False
- if relations.has_key("overlaps"):
- return False
- if relations.has_key("overlapped"):
- return False
- if relations.has_key("starts"):
- return False
- if relations.has_key("finishes"):
- return False
- if relations.has_key("started"):
- return False
- if relations.has_key("finished"):
- return False
- elif map_time == "point":
- if relations.has_key("equivalent"):
- return False
- else:
- return False
- return True
- def sample_by_dataset(self, stds, method=None, spatial=False, dbif=None):
- """!Sample this space time dataset with the temporal topology of a second space time dataset
- The sample dataset must have "interval" as temporal map type, so all sample maps have valid interval time.
- In case spatial is True, the spatial overlap between temporal related maps is performed. Only
- temporal related and spatial overlapping maps are returned.
-
- Return all registered maps as ordered (by start_time) object list with
- "gap" map objects (id==None). Each list entry is a list of map objects
- which are potentially located in temporal relation to the actual granule of the second space time dataset.
- Each entry in the object list is a dict. The actual sampler map and its temporal extent (the actual granule) and
- the list of samples are stored:
- list = self.sample_by_dataset(stds=sampler, method=["during","overlap","contain","equal"])
- for entry in list:
- granule = entry["granule"]
- maplist = entry["samples"]
- for map in maplist:
- map.select()
- map.print_info()
- A valid temporal topology (no overlapping or inclusion allowed) is needed to get correct results in case of gaps
- in the sample dataset.
-
- Gaps between maps are identified as unregistered maps with id==None.
- The map objects are initialized with the id and the temporal extent of the granule (temporal type, start time, end time).
- In case more map information are needed, use the select() method for each listed object.
- @param stds: The space time dataset to be used for temporal sampling
- @param method: This option specifies what sample method should be used. In case the registered maps are of temporal point type,
- only the start time is used for sampling. In case of mixed of interval data the user can chose between:
- * start: Select maps of which the start time is located in the selection granule
- map : s
- granule: s-----------------e
- map : s--------------------e
- granule: s-----------------e
- map : s--------e
- granule: s-----------------e
- * during: Select maps which are temporal during the selection granule
- map : s-----------e
- granule: s-----------------e
- * overlap: Select maps which temporal overlap the selection granule
- map : s-----------e
- granule: s-----------------e
- map : s-----------e
- granule: s----------e
- * contain: Select maps which temporally contain the selection granule
- map : s-----------------e
- granule: s-----------e
- * equal: Select maps which temporally equal to the selection granule
- map : s-----------e
- granule: s-----------e
- All these methods can be combined. Method must be of type tuple including the identification strings.
- @param spatial: If set True additional the spatial overlapping is used for selection -> spatio-temporal relation.
- The returned map objects will have temporal and spatial extents
- @param dbif: The database interface to be used
- In case nothing found None is returned
- """
-
- use_start = False
- use_during = False
- use_overlap = False
- use_contain = False
- use_equal = False
- # Initialize the methods
- if method:
- for name in method:
- if name == "start":
- use_start = True
- if name == "during":
- use_during = True
- if name == "overlap":
- use_overlap = True
- if name == "contain":
- use_contain = True
- if name == "equal":
- use_equal = True
- else:
- use_during = True
- use_overlap = True
- use_contain = True
- use_equal = True
- if self.get_temporal_type() != stds.get_temporal_type():
- core.error(_("The space time datasets must be of the same temporal type"))
- return None
- if stds.get_map_time() != "interval":
- core.error(_("The temporal map type of the sample dataset must be interval"))
- return None
- # In case points of time are available, disable the interval specific methods
- if self.get_map_time() == "point":
- use_start = True
- use_during = False
- use_overlap = False
- use_contain = False
- use_equal = False
- dbif, connect = init_dbif(dbif)
- obj_list = []
- sample_maps = stds.get_registered_maps_as_objects_with_gaps(where=None, dbif=dbif)
-
- for granule in sample_maps:
- # Read the spatial extent
- if spatial == True:
- granule.spatial_extent.select(dbif)
- start, end = granule.get_valid_time()
- where = create_temporal_relation_sql_where_statement(start, end, use_start, \
- use_during, use_overlap, use_contain, use_equal)
-
- maps = self.get_registered_maps_as_objects(where, "start_time", dbif)
- result = {}
- result["granule"] = granule
- num_samples = 0
- maplist = []
- if maps:
- for map in maps:
- # Read the spatial extent
- if spatial == True:
- map.spatial_extent.select(dbif)
- # Ignore spatial disjoint maps
- if not granule.spatial_overlapping(map):
- continue
- num_samples += 1
- maplist.append(copy.copy(map))
- # Fill with empty map in case no spatio-temporal relations found
- if not maps or num_samples == 0:
- map = self.get_new_map_instance(None)
- if self.is_time_absolute():
- map.set_absolute_time(start, end)
- elif self.is_time_relative():
- map.set_relative_time(start, end, self.get_relative_time_unit())
- maplist.append(copy.copy(map))
- result["samples"] = maplist
- obj_list.append(copy.copy(result))
- if connect == True:
- dbif.close()
- return obj_list
- def get_registered_maps_as_objects_by_granularity(self, gran=None, dbif=None):
- """!Return all registered maps as ordered (by start_time) object list with
- "gap" map objects (id==None) for temporal topological operations using the
- granularity of the space time dataset as increment. Each list entry is a list of map objects
- which are potentially located in the actual granule.
- A valid temporal topology (no overlapping or inclusion allowed) is needed to get correct results.
-
- The dataset must have "interval" as temporal map type, so all maps have valid interval time.
- Gaps between maps are identified as unregistered maps with id==None.
- The objects are initialized with the id and the temporal extent (temporal type, start time, end time).
- In case more map information are needed, use the select() method for each listed object.
- @param gran: The granularity to be used
- @param dbif: The database interface to be used
- In case nothing found None is returned
- """
- dbif, connect = init_dbif(dbif)
- obj_list = []
- if gran == None:
- gran = self.get_granularity()
- start, end = self.get_valid_time()
- while start < end:
- if self.is_time_absolute():
- next = increment_datetime_by_string(start, gran)
- else:
- next = start + gran
- where = "(start_time <= '%s' and end_time >= '%s')" % (start, next)
- rows = self.get_registered_maps("id", where, "start_time", dbif)
- if rows:
- if len(rows) > 1:
- core.warning(_("More than one map found in a granule. Temporal granularity seems to be invalid or the chosen granularity is not a greatest common divider of all intervals and gaps in the dataset."))
-
- maplist = []
- for row in rows:
- # Take the first map
- map = self.get_new_map_instance(rows[0]["id"])
- if self.is_time_absolute():
- map.set_absolute_time(start, next)
- elif self.is_time_relative():
- map.set_relative_time(start, next, self.get_relative_time_unit())
- maplist.append(copy.copy(map))
- obj_list.append(copy.copy(maplist))
- start = next
- if connect == True:
- dbif.close()
- return obj_list
- def get_registered_maps_as_objects_with_gaps(self, where=None, dbif=None):
- """!Return all registered maps as ordered (by start_time) object list with
- "gap" map objects (id==None) for temporal topological operations
- Gaps between maps are identified as maps with id==None
- The objects are initialized with the id and the temporal extent (temporal type, start time, end time).
- In case more map information are needed, use the select() method for each listed object.
- @param where: The SQL where statement to select a subset of the registered maps without "WHERE"
- @param dbif: The database interface to be used
- In case nothing found None is returned
- """
- dbif, connect = init_dbif(dbif)
- obj_list = []
-
- maps = self.get_registered_maps_as_objects(where, "start_time", dbif)
- if maps and len(maps) > 0:
- for i in range(len(maps)):
- obj_list.append(maps[i])
- # Detect and insert gaps
- if i < len(maps) - 1:
- relation = maps[i + 1].temporal_relation(maps[i])
- if relation == "after":
- start1, end1 = maps[i].get_valid_time()
- start2, end2 = maps[i + 1].get_valid_time()
- end = start2
- if end1:
- start = end1
- else:
- start = start1
- map = self.get_new_map_instance(None)
- if self.is_time_absolute():
- map.set_absolute_time(start, end)
- elif self.is_time_relative():
- map.set_relative_time(start, end, self.get_relative_time_unit())
- obj_list.append(copy.copy(map))
- if connect == True:
- dbif.close()
- return obj_list
- def get_registered_maps_as_objects(self, where=None, order="start_time", dbif=None):
- """!Return all registered maps as ordered object list for temporal topological operations
- The objects are initialized with the id and the temporal extent (temporal type, start time, end time).
- In case more map information are needed, use the select() method for each listed object.
- @param where: The SQL where statement to select a subset of the registered maps without "WHERE"
- @param order: The SQL order statement to be used to order the objects in the list without "ORDER BY"
- @param dbif: The database interface to be used
- In case nothing found None is returned
- """
- dbif, connect = init_dbif(dbif)
- obj_list = []
-
- rows = self.get_registered_maps("id,start_time,end_time", where, order, dbif)
- count = 0
- if rows:
- for row in rows:
- core.percent(count, len(rows), 1)
- map = self.get_new_map_instance(row["id"])
- if self.is_time_absolute():
- map.set_absolute_time(row["start_time"], row["end_time"])
- elif self.is_time_relative():
- map.set_relative_time(row["start_time"], row["end_time"], self.get_relative_time_unit())
- obj_list.append(copy.copy(map))
- count += 1
- core.percent(1, 1, 1)
- if connect == True:
- dbif.close()
- return obj_list
- def get_registered_maps(self, columns=None, where = None, order = None, dbif=None):
- """!Return sqlite rows of all registered maps.
-
- In case columns are not specified, each row includes all columns specified in the datatype specific view
- @param columns: Columns to be selected as SQL compliant string
- @param where: The SQL where statement to select a subset of the registered maps without "WHERE"
- @param order: The SQL order statement to be used to order the objects in the list without "ORDER BY"
- @param dbif: The database interface to be used
- In case nothing found None is returned
- """
- dbif, connect = init_dbif(dbif)
- rows = None
- if self.get_map_register():
- # Use the correct temporal table
- if self.get_temporal_type() == "absolute":
- map_view = self.get_new_map_instance(None).get_type() + "_view_abs_time"
- else:
- map_view = self.get_new_map_instance(None).get_type() + "_view_rel_time"
- if columns:
- sql = "SELECT %s FROM %s WHERE %s.id IN (SELECT id FROM %s)" % (columns, map_view, map_view, self.get_map_register())
- else:
- sql = "SELECT * FROM %s WHERE %s.id IN (SELECT id FROM %s)" % (map_view, map_view, self.get_map_register())
- if where:
- sql += " AND %s" % (where)
- if order:
- sql += " ORDER BY %s" % (order)
-
- try:
- dbif.cursor.execute(sql)
- rows = dbif.cursor.fetchall()
- except:
- if connect == True:
- dbif.close()
- core.error(_("Unable to get map ids from register table <%s>") % (self.get_map_register()))
- raise
- if connect == True:
- dbif.close()
- return rows
- def delete(self, dbif=None, execute=True):
- """!Delete a space time dataset from the temporal database
- This method removes the space time dataset from the temporal database and drops its map register table
- @param dbif: The database interface to be used
- @param execute: If True the SQL DELETE and DROP table statements will be executed.
- If False the prepared SQL statements are returned and must be executed by the caller.
- @return The SQL statements if execute == False, else an empty string
- """
- # First we need to check if maps are registered in this dataset and
- # unregister them
- core.verbose(_("Delete space time %s dataset <%s> from temporal database") % (self.get_new_map_instance(ident=None).get_type(), self.get_id()))
- statement = ""
- dbif, connect = init_dbif(dbif)
- # SELECT all needed information from the database
- self.metadata.select(dbif)
- if self.get_map_register():
- core.verbose(_("Drop map register table: %s") % (self.get_map_register()))
- rows = self.get_registered_maps("id", None, None, dbif)
- # Unregister each registered map in the table
- if rows:
- num_maps = len(rows)
- count = 0
- for row in rows:
- core.percent(count, num_maps, 1)
- # Unregister map
- map = self.get_new_map_instance(row["id"])
- statement += self.unregister_map(map=map, dbif=dbif, execute=False)
- count += 1
- core.percent(1, 1, 1)
- # Safe the DROP table statement
- statement += "DROP TABLE " + self.get_map_register() + ";\n"
- # Remove the primary key, the foreign keys will be removed by trigger
- statement += self.base.get_delete_statement()
- if execute == True:
- dbif.execute_transaction(statement)
- self.reset(None)
- if connect == True:
- dbif.close()
- if execute:
- return ""
- return statement
- def register_map(self, map, dbif=None):
- """!Register a map in the space time dataset.
- This method takes care of the registration of a map
- in a space time dataset.
- In case the map is already registered this function will break with a warning
- and return False
- @param dbif: The database interface to be used
- """
- dbif, connect = init_dbif(dbif)
- if map.is_in_db(dbif) == False:
- dbif.close()
- core.fatal(_("Only maps with absolute or relative valid time can be registered"))
- if map.get_layer():
- core.verbose(_("Register %s map <%s> with layer %s in space time %s dataset <%s>") % (map.get_type(), map.get_map_id(), map.get_layer(), map.get_type(), self.get_id()))
- else:
- core.verbose(_("Register %s map <%s> in space time %s dataset <%s>") % (map.get_type(), map.get_map_id(), map.get_type(), self.get_id()))
- # First select all data from the database
- map.select(dbif)
- if not map.check_valid_time():
- if map.get_layer():
- core.fatal(_("Map <%s> with layer %s has invalid time") % map.get_map_id(), map.get_layer())
- else:
- core.fatal(_("Map <%s> has invalid time") % map.get_map_id())
- map_id = map.base.get_id()
- map_name = map.base.get_name()
- map_mapset = map.base.get_mapset()
- map_register_table = map.get_stds_register()
- map_rel_time_unit = map.get_relative_time_unit()
- map_ttype = map.get_temporal_type()
- #print "Map register table", map_register_table
- # Get basic info
- stds_name = self.base.get_name()
- stds_mapset = self.base.get_mapset()
- stds_register_table = self.get_map_register()
- stds_ttype = self.get_temporal_type()
-
- # The gathered SQL statemets are stroed here
- statement = ""
- # Check temporal types
- if stds_ttype != map_ttype:
- if map.get_layer():
- core.fatal(_("Temporal type of space time dataset <%s> and map <%s> with layer %s are different") % (self.get_id(), map.get_map_id(), map.get_layer()))
- else:
- core.fatal(_("Temporal type of space time dataset <%s> and map <%s> are different") % (self.get_id(), map.get_map_id()))
- # In case no map has been registered yet, set the relative time unit from the first map
- if (self.metadata.get_number_of_maps() == None or self.metadata.get_number_of_maps() == 0) and \
- self.map_counter == 0 and self.is_time_relative():
- self.set_relative_time_unit(map_rel_time_unit)
- statement += self.relative_time.get_update_all_statement_mogrified(dbif)
- core.verbose(_("Set temporal unit for space time %s dataset <%s> to %s") % (map.get_type(), self.get_id(), map_rel_time_unit))
- stds_rel_time_unit = self.get_relative_time_unit()
- # Check the relative time unit
- if self.is_time_relative() and (stds_rel_time_unit != map_rel_time_unit):
- if map.get_layer():
- core.fatal(_("Relative time units of space time dataset <%s> and map <%s> with layer %s are different") % (self.get_id(), map.get_map_id(), map.get_layer()))
- else:
- core.fatal(_("Relative time units of space time dataset <%s> and map <%s> are different") % (self.get_id(), map.get_map_id()))
- #print "STDS register table", stds_register_table
- if stds_mapset != map_mapset:
- dbif.close()
- core.fatal(_("Only maps from the same mapset can be registered"))
- # Check if map is already registered
- if stds_register_table:
- if dbmi.paramstyle == "qmark":
- sql = "SELECT id FROM " + stds_register_table + " WHERE id = (?)"
- else:
- sql = "SELECT id FROM " + stds_register_table + " WHERE id = (%s)"
- try:
- dbif.cursor.execute(sql, (map_id,))
- row = dbif.cursor.fetchone()
- except:
- row = None
- core.warning(_("Error in strds_register_table request"))
- raise
- if row and row[0] == map_id:
- if connect == True:
- dbif.close()
- if map.get_layer():
- core.warning(_("Map <%s> with layer %s is already registered.") % (map.get_map_id(), map.get_layer()))
- else:
- core.warning(_("Map <%s> is already registered.") % (map.get_map_id()))
- return ""
- # Create tables
- sql_path = get_sql_template_path()
- # We need to create the map raster register table before we can register the map
- if map_register_table == None:
- # Create a unique id
- uuid_rand = "map_" + str(uuid.uuid4()).replace("-", "")
- map_register_table = uuid_rand + "_" + self.get_type() + "_register"
-
- # Read the SQL template
- sql = open(os.path.join(sql_path, "map_stds_register_table_template.sql"), 'r').read()
- # Create the raster, raster3d and vector tables
- sql = sql.replace("GRASS_MAP", map.get_type())
- sql = sql.replace("MAP_NAME", map_name + "_" + map_mapset )
- sql = sql.replace("TABLE_NAME", uuid_rand )
- sql = sql.replace("MAP_ID", map_id)
- sql = sql.replace("STDS", self.get_type())
-
- statement += sql
- # Set the stds register table name and put it into the DB
- map.set_stds_register(map_register_table)
- statement += map.metadata.get_update_statement_mogrified(dbif)
-
- if map.get_layer():
- core.verbose(_("Created register table <%s> for %s map <%s> with layer %s") % \
- (map_register_table, map.get_type(), map.get_map_id(), map.get_layer()))
- else:
- core.verbose(_("Created register table <%s> for %s map <%s>") % \
- (map_register_table, map.get_type(), map.get_map_id()))
- # We need to create the table and register it
- if stds_register_table == None:
- # Create table name
- stds_register_table = stds_name + "_" + stds_mapset + "_" + map.get_type() + "_register"
- # Read the SQL template
- sql = open(os.path.join(sql_path, "stds_map_register_table_template.sql"), 'r').read()
- # Create the raster, raster3d and vector tables
- sql = sql.replace("GRASS_MAP", map.get_type())
- sql = sql.replace("SPACETIME_NAME", stds_name + "_" + stds_mapset )
- sql = sql.replace("SPACETIME_ID", self.base.get_id())
- sql = sql.replace("STDS", self.get_type())
- statement += sql
- # Set the map register table name and put it into the DB
- self.set_map_register(stds_register_table)
- statement += self.metadata.get_update_statement_mogrified(dbif)
- core.verbose(_("Created register table <%s> for space time %s dataset <%s>") % \
- (stds_register_table, map.get_type(), self.get_id()))
- # We need to execute the statement at this time
- if statement != "":
- dbif.execute_transaction(statement)
- statement = ""
- # Register the stds in the map stds register table
- # Check if the entry is already there
- if dbmi.paramstyle == "qmark":
- sql = "SELECT id FROM " + map_register_table + " WHERE id = ?"
- else:
- sql = "SELECT id FROM " + map_register_table + " WHERE id = %s"
- try:
- dbif.cursor.execute(sql, (self.base.get_id(),))
- row = dbif.cursor.fetchone()
- except:
- row = None
- # In case of no entry make a new one
- if row == None:
- if dbmi.paramstyle == "qmark":
- sql = "INSERT INTO " + map_register_table + " (id) " + "VALUES (?);\n"
- else:
- sql = "INSERT INTO " + map_register_table + " (id) " + "VALUES (%s);\n"
- statement += dbif.mogrify_sql_statement((sql, (self.base.get_id(),)))
- # Now put the raster name in the stds map register table
- if dbmi.paramstyle == "qmark":
- sql = "INSERT INTO " + stds_register_table + " (id) " + "VALUES (?);\n"
- else:
- sql = "INSERT INTO " + stds_register_table + " (id) " + "VALUES (%s);\n"
- statement += dbif.mogrify_sql_statement((sql, (map_id,)))
- # Now execute the insert transaction
- dbif.execute_transaction(statement)
- if connect == True:
- dbif.close()
- # increase the counter
- self.map_counter += 1
- def unregister_map(self, map, dbif = None, execute=True):
- """!Unregister a map from the space time dataset.
- This method takes care of the un-registration of a map
- from a space time dataset.
- @param map: The map object to unregister
- @param dbif: The database interface to be used
- @param execute: If True the SQL DELETE and DROP table statements will be executed.
- If False the prepared SQL statements are returned and must be executed by the caller.
- @return The SQL statements if execute == False, else an empty string, None in case of a failure
- """
- statement = ""
- dbif, connect = init_dbif(dbif)
- # First select needed data from the database
- map.metadata.select(dbif)
- map_id = map.get_id()
- map_register_table = map.get_stds_register()
- stds_register_table = self.get_map_register()
- if map.get_layer():
- core.verbose(_("Unregister %s map <%s> with layer %s") % (map.get_type(), map.get_map_id(), map.get_layer()))
- else:
- core.verbose(_("Unregister %s map <%s>") % (map.get_type(), map.get_map_id()))
- # Check if the map is registered in the space time raster dataset
- if dbmi.paramstyle == "qmark":
- sql = "SELECT id FROM " + map_register_table + " WHERE id = ?"
- else:
- sql = "SELECT id FROM " + map_register_table + " WHERE id = %s"
- try:
- dbif.cursor.execute(sql, (self.base.get_id(),))
- row = dbif.cursor.fetchone()
- except:
- row = None
- # Break if the map is not registered
- if row == None:
- if map.get_layer():
- core.warning(_("Map <%s> with layer %s is not registered in space time dataset <%s>") %(map.get_map_id(), map.get_layer(), self.base.get_id()))
- else:
- core.warning(_("Map <%s> is not registered in space time dataset <%s>") %(map.get_map_id(), self.base.get_id()))
- if connect == True:
- dbif.close()
- return ""
- # Remove the space time raster dataset from the raster dataset register
- if map_register_table != None:
- if dbmi.paramstyle == "qmark":
- sql = "DELETE FROM " + map_register_table + " WHERE id = ?;\n"
- else:
- sql = "DELETE FROM " + map_register_table + " WHERE id = %s;\n"
- statement += dbif.mogrify_sql_statement((sql, (self.base.get_id(),)))
- # Remove the raster map from the space time raster dataset register
- if stds_register_table != None:
- if dbmi.paramstyle == "qmark":
- sql = "DELETE FROM " + stds_register_table + " WHERE id = ?;\n"
- else:
- sql = "DELETE FROM " + stds_register_table + " WHERE id = %s;\n"
- statement += dbif.mogrify_sql_statement((sql, (map_id,)))
- if execute == True:
- dbif.execute_transaction(statement)
-
- if connect == True:
- dbif.close()
- # decrease the counter
- self.map_counter -= 1
- if execute:
- return ""
- return statement
-
- def update_from_registered_maps(self, dbif = None):
- """!This methods updates the spatial and temporal extent as well as
- type specific metadata. It should always been called after maps are registered
- or unregistered/deleted from the space time dataset.
- The update of the temporal extent checks if the end time is set correctly.
- In case the registered maps have no valid end time (None) the maximum start time
- will be used. If the end time is earlier than the maximum start time, it will
- be replaced by the maximum start time.
- An other solution to automate this is to use the deactivated trigger
- in the SQL files. But this will result in a huge performance issue
- in case many maps are registered (>1000).
-
- @param dbif: The database interface to be used
- """
- core.verbose(_("Update metadata, spatial and temporal extent from all registered maps of <%s>") % (self.get_id()))
- # Nothing to do if the register is not present
- if not self.get_map_register():
- return
- dbif, connect = init_dbif(dbif)
- map_time = None
- use_start_time = False
- # Get basic info
- stds_name = self.base.get_name()
- stds_mapset = self.base.get_mapset()
- sql_path = get_sql_template_path()
- #We create a transaction
- sql_script = ""
-
- # Update the spatial and temporal extent from registered maps
- # Read the SQL template
- sql = open(os.path.join(sql_path, "update_stds_spatial_temporal_extent_template.sql"), 'r').read()
- sql = sql.replace("GRASS_MAP", self.get_new_map_instance(None).get_type())
- sql = sql.replace("SPACETIME_NAME", stds_name + "_" + stds_mapset )
- sql = sql.replace("SPACETIME_ID", self.base.get_id())
- sql = sql.replace("STDS", self.get_type())
- sql_script += sql
- sql_script += "\n"
- # Update type specific metadata
- sql = open(os.path.join(sql_path, "update_" + self.get_type() + "_metadata_template.sql"), 'r').read()
- sql = sql.replace("GRASS_MAP", self.get_new_map_instance(None).get_type())
- sql = sql.replace("SPACETIME_NAME", stds_name + "_" + stds_mapset )
- sql = sql.replace("SPACETIME_ID", self.base.get_id())
- sql = sql.replace("STDS", self.get_type())
- sql_script += sql
- sql_script += "\n"
- dbif.execute_transaction(sql_script)
-
- # Read and validate the selected end time
- self.select()
- if self.is_time_absolute():
- start_time, end_time, tz = self.get_absolute_time()
- else:
- start_time, end_time, unit = self.get_relative_time()
- # In case no end time is set, use the maximum start time of all registered maps as end time
- if end_time == None:
- use_start_time = True
- else:
- # Check if the end time is smaller than the maximum start time
- if self.is_time_absolute():
- sql = """SELECT max(start_time) FROM GRASS_MAP_absolute_time WHERE GRASS_MAP_absolute_time.id IN
- (SELECT id FROM SPACETIME_NAME_GRASS_MAP_register);"""
- sql = sql.replace("GRASS_MAP", self.get_new_map_instance(None).get_type())
- sql = sql.replace("SPACETIME_NAME", stds_name + "_" + stds_mapset )
- else:
- sql = """SELECT max(start_time) FROM GRASS_MAP_relative_time WHERE GRASS_MAP_relative_time.id IN
- (SELECT id FROM SPACETIME_NAME_GRASS_MAP_register);"""
- sql = sql.replace("GRASS_MAP", self.get_new_map_instance(None).get_type())
- sql = sql.replace("SPACETIME_NAME", stds_name + "_" + stds_mapset )
- dbif.cursor.execute(sql)
- row = dbif.cursor.fetchone()
- if row != None:
- # This seems to be a bug in sqlite3 Python driver
- if dbmi.__name__ == "sqlite3":
- tstring = row[0]
- # Convert the unicode string into the datetime format
- if self.is_time_absolute():
- if tstring.find(":") > 0:
- time_format = "%Y-%m-%d %H:%M:%S"
- else:
- time_format = "%Y-%m-%d"
- max_start_time = datetime.strptime(tstring, time_format)
- else:
- max_start_time = row[0]
- else:
- max_start_time = row[0]
- if end_time < max_start_time:
- use_start_time = True
-
- # Set the maximum start time as end time
- if use_start_time:
- if self.is_time_absolute():
- sql = """UPDATE STDS_absolute_time SET end_time =
- (SELECT max(start_time) FROM GRASS_MAP_absolute_time WHERE GRASS_MAP_absolute_time.id IN
- (SELECT id FROM SPACETIME_NAME_GRASS_MAP_register)
- ) WHERE id = 'SPACETIME_ID';"""
- sql = sql.replace("GRASS_MAP", self.get_new_map_instance(None).get_type())
- sql = sql.replace("SPACETIME_NAME", stds_name + "_" + stds_mapset )
- sql = sql.replace("SPACETIME_ID", self.base.get_id())
- sql = sql.replace("STDS", self.get_type())
- elif self.is_time_relative():
- sql = """UPDATE STDS_relative_time SET end_time =
- (SELECT max(start_time) FROM GRASS_MAP_relative_time WHERE GRASS_MAP_relative_time.id IN
- (SELECT id FROM SPACETIME_NAME_GRASS_MAP_register)
- ) WHERE id = 'SPACETIME_ID';"""
- sql = sql.replace("GRASS_MAP", self.get_new_map_instance(None).get_type())
- sql = sql.replace("SPACETIME_NAME", stds_name + "_" + stds_mapset )
- sql = sql.replace("SPACETIME_ID", self.base.get_id())
- sql = sql.replace("STDS", self.get_type())
- dbif.execute_transaction(sql)
- # Count the temporal map types
- maps = self.get_registered_maps_as_objects(dbif=dbif)
- tlist = self.count_temporal_types(maps)
- if tlist["interval"] > 0 and tlist["point"] == 0 and tlist["invalid"] == 0:
- map_time = "interval"
- elif tlist["interval"] == 0 and tlist["point"] > 0 and tlist["invalid"] == 0:
- map_time = "point"
- elif tlist["interval"] > 0 and tlist["point"] > 0 and tlist["invalid"] == 0:
- map_time = "mixed"
- else:
- map_time = "invalid"
- # Compute the granularity
- if map_time != "invalid":
- # Smallest supported temporal resolution
- if self.is_time_absolute():
- gran = compute_absolute_time_granularity(maps)
- elif self.is_time_relative():
- gran = compute_relative_time_granularity(maps)
- else:
- gran = None
- # Set the map time type and update the time objects
- if self.is_time_absolute():
- self.absolute_time.select(dbif)
- self.metadata.select(dbif)
- if self.metadata.get_number_of_maps() > 0:
- self.absolute_time.set_map_time(map_time)
- self.absolute_time.set_granularity(gran)
- else:
- self.absolute_time.set_map_time(None)
- self.absolute_time.set_granularity(None)
- self.absolute_time.update_all(dbif)
- else:
- self.relative_time.select(dbif)
- self.metadata.select(dbif)
- if self.metadata.get_number_of_maps() > 0:
- self.relative_time.set_map_time(map_time)
- self.relative_time.set_granularity(gran)
- else:
- self.relative_time.set_map_time(None)
- self.relative_time.set_granularity(None)
- self.relative_time.update_all(dbif)
- if connect == True:
- dbif.close()
- ###############################################################################
- def create_temporal_relation_sql_where_statement(start, end, use_start=True, use_during=False,
- use_overlap=False, use_contain=False, use_equal=False):
- """!Create a SQL WHERE statement for temporal relation selection of maps in space time datasets
- @param start: The start time
- @param end: The end time
- @param use_start: Select maps of which the start time is located in the selection granule
- map : s
- granule: s-----------------e
- map : s--------------------e
- granule: s-----------------e
- map : s--------e
- granule: s-----------------e
- @param use_during: during: Select maps which are temporal during the selection granule
- map : s-----------e
- granule: s-----------------e
- @param use_overlap: Select maps which temporal overlap the selection granule
- map : s-----------e
- granule: s-----------------e
- map : s-----------e
- granule: s----------e
- @param use_contain: Select maps which temporally contain the selection granule
- map : s-----------------e
- granule: s-----------e
- @param use_equal: Select maps which temporally equal to the selection granule
- map : s-----------e
- granule: s-----------e
- """
- where = "("
- if use_start:
- where += "(start_time >= '%s' and start_time < '%s') " % (start, end)
- if use_during:
- if use_start:
- where += " OR "
- where += "((start_time > '%s' and end_time < '%s') OR " % (start, end)
- where += "(start_time >= '%s' and end_time < '%s') OR " % (start, end)
- where += "(start_time > '%s' and end_time <= '%s'))" % (start, end)
- if use_overlap:
- if use_start or use_during:
- where += " OR "
- where += "((start_time < '%s' and end_time > '%s' and end_time < '%s') OR " % (start, start, end)
- where += "(start_time < '%s' and start_time > '%s' and end_time > '%s'))" % (end, start, end)
- if use_contain:
- if use_start or use_during or use_overlap:
- where += " OR "
- where += "((start_time < '%s' and end_time > '%s') OR " % (start, end)
- where += "(start_time <= '%s' and end_time > '%s') OR " % (start, end)
- where += "(start_time < '%s' and end_time >= '%s'))" % (start, end)
- if use_equal:
- if use_start or use_during or use_overlap or use_contain:
- where += " OR "
- where += "(start_time = '%s' and end_time = '%s')" % (start, end)
- where += ")"
-
- # Catch empty where statement
- if where == "()":
- where = None
- return where
|