123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885 |
- """!@package grass.script.tgis_abstract_datasets
- @brief GRASS Python scripting module (temporal GIS functions)
- Temporal GIS related functions to be used in temporal GIS Python library package.
- Usage:
- @code
- from grass.script import tgis_abstract_datasets as grass
- ...
- @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
- """
- import uuid
- import copy
- from temporal_extent import *
- from spatial_extent import *
- from metadata import *
- class abstract_dataset(object):
- """This is the base class for all datasets (raster, vector, raster3d, strds, stvds, str3ds)"""
- def reset(self, ident):
- """Reset the internal structure and set the identifier"""
- raise IOError("This method must be implemented in the subclasses")
- def get_type(self):
- """Return the type of this class"""
- raise IOError("This method must be implemented in the subclasses")
-
- def get_new_instance(self, ident):
- """Return a new instance with the type of this class"""
- raise IOError("This method must be implemented in the subclasses")
- def get_id(self):
- return self.base.get_id()
- def get_absolute_time(self):
- """Returns a tuple of the start, the end valid time and the timezone of the map
- @return A tuple of (start_time, end_time, timezone)
- """
-
- start = self.absolute_time.get_start_time()
- end = self.absolute_time.get_end_time()
- tz = self.absolute_time.get_timezone()
-
- return (start, end, tz)
-
- def get_relative_time(self):
- """Returns the relative time interval (start_time, end_time) or None if not present"""
- start = self.relative_time.get_start_time()
- end = self.relative_time.get_end_time()
- return (start, end)
- def get_temporal_type(self):
- """Return the temporal type of this dataset"""
- return self.base.get_ttype()
-
- def get_spatial_extent(self):
- """Return a tuple of spatial extent (north, south, east, west, top, bottom) """
-
- north = self.spatial_extent.get_north()
- south = self.spatial_extent.get_south()
- east = self.spatial_extent.get_east()
- west = self.spatial_extent.get_west()
- top = self.spatial_extent.get_top()
- bottom = self.spatial_extent.get_bottom()
-
- return (north, south, east, west, top, bottom)
-
- def select(self, dbif=None):
- """Select temporal dataset entry from database and fill up the internal structure"""
- self.base.select(dbif)
- if self.is_time_absolute():
- self.absolute_time.select(dbif)
- if self.is_time_relative():
- self.relative_time.select(dbif)
- self.spatial_extent.select(dbif)
- self.metadata.select(dbif)
-
- def is_in_db(self, dbif=None):
- """Check if the temporal dataset entry is in the database"""
- return self.base.is_in_db(dbif)
- def delete(self):
- """Delete temporal dataset entry from database if it exists"""
- raise IOError("This method must be implemented in the subclasses")
- def insert(self, dbif=None):
- """Insert temporal dataset entry into database from the internal structure"""
- self.base.insert(dbif)
- if self.is_time_absolute():
- self.absolute_time.insert(dbif)
- if self.is_time_relative():
- self.relative_time.insert(dbif)
- self.spatial_extent.insert(dbif)
- self.metadata.insert(dbif)
- def update(self, dbif=None):
- """Update temporal dataset entry of database from the internal structure"""
- self.base.update(dbif)
- if self.is_time_absolute():
- self.absolute_time.update(dbif)
- if self.is_time_relative():
- self.relative_time.update(dbif)
- self.spatial_extent.update(dbif)
- self.metadata.update(dbif)
- 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() == "raster":
- # 1 2 3 4 5 6 7
- # 0123456789012345678901234567890123456789012345678901234567890123456789012345678
- print ""
- print " +-------------------- Raster Dataset ----------------------------------------+"
- if self.get_type() == "raster3d":
- # 1 2 3 4 5 6 7
- # 0123456789012345678901234567890123456789012345678901234567890123456789012345678
- print ""
- print " +-------------------- Raster3d Dataset --------------------------------------+"
- if self.get_type() == "vector":
- # 1 2 3 4 5 6 7
- # 0123456789012345678901234567890123456789012345678901234567890123456789012345678
- print ""
- print " +-------------------- Vector Dataset ----------------------------------------+"
- 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_time_to_absolute(self):
- self.base.set_ttype("absolute")
- def set_time_to_relative(self):
- self.base.set_ttype("relative")
- def is_time_absolute(self):
- if self.base.D.has_key("temporal_type"):
- return self.base.get_ttype() == "absolute"
- else:
- return None
- def is_time_relative(self):
- if self.base.D.has_key("temporal_type"):
- return self.base.get_ttype() == "relative"
- else:
- return None
- def temporal_relation(self, map):
- """Return the temporal relation of this and the provided temporal map"""
- if self.is_time_absolute() and map.is_time_absolute():
- return self.absolute_time.temporal_relation(map.absolute_time)
- if self.is_time_relative() and map.is_time_relative():
- return self.relative_time.temporal_relation(map.relative_time)
- return None
- ###############################################################################
- class abstract_map_dataset(abstract_dataset):
- """This is the base class for all maps (raster, vector, raster3d)
- providing additional function to set the valid time and the spatial extent.
- """
-
- def get_new_stds_instance(self, ident):
- """Return a new space time dataset instance in which maps are stored with the type of this class"""
- raise IOError("This method must be implemented in the subclasses")
-
- def get_stds_register(self):
- """Return the space time dataset register table name in which stds are listed in which this map is registered"""
- raise IOError("This method must be implemented in the subclasses")
-
- def set_stds_register(self, name):
- """Set the space time dataset register table name in which stds are listed in which this map is registered"""
- raise IOError("This method must be implemented in the subclasses")
-
- def set_absolute_time(self, start_time, end_time=None, timezone=None):
- """Set the absolute time interval with start time and end time
-
- @start_time a datetime object specifying the start time of the map
- @end_time a datetime object specifying the end time of the map
- @timezone Thee timezone of the map
-
- """
- if start_time != None and not isinstance(start_time, datetime) :
- core.fatal(_("Start time must be of type datetime"))
- if end_time != None and not isinstance(end_time, datetime) :
- core.fatal(_("End time must be of type datetime"))
- if start_time != None and end_time != None:
- if start_time >= end_time:
- core.error(_("End time must be later than start time"))
- return False
- self.base.set_ttype("absolute")
-
- self.absolute_time.set_start_time(start_time)
- self.absolute_time.set_end_time(end_time)
- self.absolute_time.set_timezone(timezone)
- return True
- def update_absolute_time(self, start_time, end_time=None, timezone=None, dbif = None):
- """Update the absolute time
- @start_time a datetime object specifying the start time of the map
- @end_time a datetime object specifying the end time of the map
- @timezone Thee timezone of the map
- """
- connect = False
- if dbif == None:
- dbif = sql_database_interface()
- dbif.connect()
- connect = True
- self.set_absolute_time(start_time, end_time, timezone)
- self.absolute_time.update_all(dbif)
- self.base.update(dbif)
- if connect == True:
- dbif.close()
- def set_relative_time(self, start_time, end_time=None):
- """Set the relative time interval
-
- @start_time A double value in days
- @end_time A double value in days
- """
- if start_time != None and end_time != None:
- if abs(float(start_time)) >= abs(float(end_time)):
- core.error(_("End time must be greater than start time"))
- return False
- self.base.set_ttype("relative")
-
- self.relative_time.set_start_time(float(start_time))
- if end_time != None:
- self.relative_time.set_end_time(float(end_time))
- else:
- self.relative_time.set_end_time(None)
- return True
- def update_relative_time(self, start_time, end_time=None, dbif = None):
- """Set the relative time interval
- @interval A double value in days
- """
- connect = False
- if dbif == None:
- dbif = sql_database_interface()
- dbif.connect()
- connect = True
- self.set_relative_time(start_time, end_time)
- self.relative_time.update_all(dbif)
- self.base.update(dbif)
- if connect == True:
- dbif.close()
- def set_spatial_extent(self, north, south, east, west, top=0, bottom=0):
- """Set the spatial extent of the map"""
- self.spatial_extent.set_spatial_extent(north, south, east, west, top, bottom)
-
- def delete(self, dbif=None):
- """Delete a map entry from database if it exists
-
- Remove dependent entries:
- * Remove the map entry in each space time dataset in which this map is registered
- * Remove the space time dataset register table
- """
- connect = False
- if dbif == None:
- dbif = sql_database_interface()
- dbif.connect()
- connect = True
- if self.is_in_db(dbif):
-
- # First we unregister from all dependent space time datasets
- self.unregister(dbif)
- # Remove the strds register table
- if self.get_stds_register():
- sql = "DROP TABLE " + self.get_stds_register()
- #print sql
- try:
- dbif.cursor.execute(sql)
- except:
- core.error("Unable to remove space time dataset register table " + self.get_stds_register())
- core.verbose("Delete " + self.get_type() + " dataset <" + self.get_id() + "> from temporal database")
- # Delete yourself from the database, trigger functions will take care of dependencies
- self.base.delete(dbif)
- self.reset(None)
- if connect == True:
- dbif.close()
- def unregister(self, dbif=None):
- """ Remove the map entry in each space time dataset in which this map is registered
- """
- core.verbose("Unregister " + self.get_type() + " dataset <" + self.get_id() + "> from space time datasets")
-
- connect = False
- if dbif == None:
- dbif = sql_database_interface()
- dbif.connect()
- connect = True
- # Get all datasets in which this map is registered
- rows = self.get_registered_datasets(dbif)
- # For each stds in which the map is registered
- if rows:
- for row in rows:
- # Create a space time dataset object to remove the map
- # from its register
- stds = self.get_new_stds_instance(row["id"])
- stds.select(dbif)
- stds.unregister_map(self, dbif)
- # Take care to update the space time dataset after
- # the map has been unregistred
- stds.update_from_registered_maps(dbif)
- if connect == True:
- dbif.close()
-
- def get_registered_datasets(self, dbif=None):
- """Return all space time dataset ids in which this map is registered as
- sqlite3 rows with column "id" or None if this map is not registered in any
- space time dataset.
- """
- connect = False
- if dbif == None:
- dbif = sql_database_interface()
- dbif.connect()
- connect = True
- rows = None
- try:
- if self.get_stds_register() != None:
- # Select all stds tables in which this map is registered
- sql = "SELECT id FROM " + self.get_stds_register()
- dbif.cursor.execute(sql)
- rows = dbif.cursor.fetchall()
- except:
- core.error("Unable to select space time dataset register table " + self.get_stds_register())
- if connect == True:
- dbif.close()
-
- return rows
- ###############################################################################
- 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 int 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)
- def get_new_instance(self, ident):
- """Return a new instance with the type of this class"""
- raise IOError("This method must be implemented in the subclasses")
- def get_new_map_instance(self, ident):
- """Return a new instance of a map dataset which is associated with the type of this class"""
- 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"""
- raise IOError("This method must be implemented in the subclasses")
- def set_initial_values(self, granularity, temporal_type, semantic_type, \
- title=None, description=None):
- 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 \"" + temporal_type + "\"")
- self.base.set_semantic_type(semantic_type)
- self.metadata.set_title(title)
- self.metadata.set_description(description)
- def get_temporal_relation_matrix(self, dbif=None):
- """Return the temporal relation matrix between all registered maps
- """
- connect = False
- if dbif == None:
- dbif = sql_database_interface()
- dbif.connect()
- connect = True
- matrix = []
- maps = self.get_registered_maps_as_objects(dbif=dbif, where=None, order="start_time")
- # Create the temporal relation matrix
- # Add the map names first
- row = []
- for map in maps:
- row.append(map.get_id())
- matrix.append(row)
- for mapA in maps:
- row = []
- for mapB in maps:
- row.append(mapA.temporal_relation(mapB))
- matrix.append(row)
- if connect == True:
- dbif.close()
- return matrix
- def get_registered_maps_as_objects(self, dbif=None, where = None, order = None):
- """Return all registered maps as ordered object list
- Each row includes all columns specified in the datatype specific view
- In case nothing found None is returned
- """
- connect = False
- if dbif == None:
- dbif = sql_database_interface()
- dbif.connect()
- connect = True
- obj_list = []
-
- rows = self.get_registered_maps(dbif, where, order)
- if rows:
- for row in rows:
- map = self.get_new_map_instance(row["id"])
- map.select(dbif)
- obj_list.append(copy.copy(map))
- if connect == True:
- dbif.close()
- return obj_list
- def get_registered_maps(self, dbif=None, where = None, order = None):
- """Return sqlite rows of all registered maps.
-
- Each row includes all columns specified in the datatype specific view
-
- In case nothing found None is returned
- """
- connect = False
- if dbif == None:
- dbif = sql_database_interface()
- dbif.connect()
- connect = True
- 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"
- 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)
- print sql
- try:
- dbif.cursor.execute(sql)
- rows = dbif.cursor.fetchall()
- except:
- 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):
- """Delete a space time dataset from the database"""
- # 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()))
- connect = False
- if dbif == None:
- dbif = sql_database_interface()
- dbif.connect()
- connect = True
- if self.get_map_register():
- rows = self.get_registered_maps(dbif)
- # Unregister each registered map in the table
- if rows:
- for row in rows:
- # Unregister map
- map = self.get_new_map_instance(row["id"])
- self.unregister_map(map, dbif)
- try:
- # Drop remove the map register table
- sql = "DROP TABLE " + self.get_map_register()
- dbif.cursor.execute(sql)
- except:
- core.error(_("Unable to drop table <%s>") % (self.get_map_register()))
- raise
- # Remove the primary key, the foreign keys will be removed by trigger
- self.base.delete(dbif)
- self.reset(None)
- if connect == True:
- dbif.close()
-
- 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
- """
- connect = False
- if dbif == None:
- dbif = sql_database_interface()
- dbif.connect()
- connect = True
- if map.is_in_db(dbif) == False:
- core.fatal("Only maps with absolute or relative valid time can be registered")
- core.verbose("Register " + map.get_type() + " map: " + map.get_id() + " in space time " + map.get_type() + " dataset <" + self.get_id() + ">")
- # First select all data from the database
- map.select(dbif)
- 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()
- #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()
- #print "STDS register table", stds_register_table
- if stds_mapset != map_mapset:
- core.fatal("Only maps from the same mapset can be registered")
- # Check if map is already registred
- if stds_register_table:
- sql = "SELECT id FROM " + stds_register_table + " WHERE id = (?)"
- dbif.cursor.execute(sql, (map_id,))
- row = dbif.cursor.fetchone()
- # In case of no entry make a new one
- if row and row[0] == map_id:
- core.warning("Map " + map_id + "is already registered.")
- return False
- # Create tables
- sql_path = get_sql_template_path()
- # We need to create the stmap raster register table bevor 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())
- try:
- dbif.cursor.executescript(sql)
- except:
- try:
- # Drop stds register table
- sql = "DROP TABLE " + map_register_table
- dbif.cursor.execute(sql)
- except:
- core.error(_("Unable to drop table <%s>" % (map_register_table)))
- raise
- try:
- dbif.cursor.executescript(sql_script)
- except:
- core.error("Unable to create the space time " + map.get_type() +\
- " dataset register table for " + map.get_type() + " map <" + map.get_id())
- raise
- # Set the stds register table name and put it into the DB
- map.set_stds_register(map_register_table)
- map.metadata.update(dbif)
-
- core.verbose("Created register table <" + map_register_table + "> for " + map.get_type() + " map <" + map.get_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())
- sql_script = ""
- #sql_script += "BEGIN TRANSACTION;\n"
- sql_script += sql
- #sql_script += "\n"
- #sql_script += "END TRANSACTION;"
- try:
- dbif.cursor.executescript(sql_script)
- except:
- try:
- # Drop map register table
- sql = "DROP TABLE " + stds_register_table
- dbif.cursor.execute(sql)
- except:
- core.error(_("Unable to drop table <%s>" % (stds_register_table)))
- raise
- try:
- dbif.cursor.executescript(sql_script)
- except:
- core.error("Unable to create the " + map.get_type() +\
- " map register table for space time " + map.get_type() + " dataset <" + map.get_id())
- raise
- # Trigger have been disabled due to peformance issues while registration
- ## We need raster specific trigger
- #sql = open(os.path.join(sql_path, "stds_" + map.get_type() + "_register_trigger_template.sql"), 'r').read()
- #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())
- #sql_script = ""
- #sql_script += "BEGIN TRANSACTION;\n"
- #sql_script += sql
- #sql_script += "\n"
- #sql_script += "END TRANSACTION;"
- #dbif.cursor.executescript(sql_script)
- # Set the map register table name and put it into the DB
- self.set_map_register(stds_register_table)
- self.metadata.update(dbif)
- core.verbose("Created register table <" + stds_register_table + "> for space time " + map.get_type() + " dataset <" + self.get_id() + ">")
- # Register the stds in the map stds register table
- # Check if the entry is already there
- sql = "SELECT id FROM " + map_register_table + " WHERE id = ?"
- dbif.cursor.execute(sql, (self.base.get_id(),))
- row = dbif.cursor.fetchone()
- # In case of no entry make a new one
- if row == None:
- sql = "INSERT INTO " + map_register_table + " (id) " + "VALUES (?)"
- #print sql
- dbif.cursor.execute(sql, (self.base.get_id(),))
- # Now put the raster name in the stds map register table
- sql = "INSERT INTO " + stds_register_table + " (id) " + "VALUES (?)"
- #print sql
- dbif.cursor.execute(sql, (map_id,))
- if connect == True:
- dbif.close()
-
- return True
- def unregister_map(self, map, dbif = None):
- """Unregister a map from the space time dataset.
- This method takes care of the unregistration of a map
- from a space time dataset.
- """
- connect = False
- if dbif == None:
- dbif = sql_database_interface()
- dbif.connect()
- connect = True
- if map.is_in_db(dbif) == False:
- core.fatal("Unable to find map <" + map.get_id() + "> in temporal database")
- core.info("Unregister " + map.get_type() + " map: " + map.get_id())
- # First select all data from the database
- map.select(dbif)
- map_id = map.base.get_id()
- map_register_table = map.get_stds_register()
- # Get basic info
- stds_register_table = self.get_map_register()
- # Check if the map is registered in the space time raster dataset
- sql = "SELECT id FROM " + map_register_table + " WHERE id = ?"
- dbif.cursor.execute(sql, (self.base.get_id(),))
- row = dbif.cursor.fetchone()
- # Break if the map is not registered
- if row == None:
- core.warning("Map " + map_id + " is not registered in space time dataset " + self.base.get_id())
- return False
- # Remove the space time raster dataset from the raster dataset register
- if map_register_table != None:
- sql = "DELETE FROM " + map_register_table + " WHERE id = ?"
- dbif.cursor.execute(sql, (self.base.get_id(),))
- # Remove the raster map from the space time raster dataset register
- if stds_register_table != None:
- sql = "DELETE FROM " + stds_register_table + " WHERE id = ?"
- dbif.cursor.execute(sql, (map_id,))
- if connect == True:
- dbif.close()
-
- 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.
- An other solution to automate this is to use the diactivated trigger
- in the SQL files. But this will result in a huge performance issue
- in case many maps are registred (>1000).
- """
- core.info("Update metadata, spatial and temporal extent from all registered maps of <" + self.get_id() + ">")
- connect = False
- if dbif == None:
- dbif = sql_database_interface()
- dbif.connect()
- connect = True
- # 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 = ""
- sql_script += "BEGIN TRANSACTION;\n"
-
- # 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"
- sql_script += "END TRANSACTION;"
- dbif.cursor.executescript(sql_script)
- if connect == True:
- dbif.close()
|