12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826 |
- """
- Fast and exit-safe interface to GRASS C-library functions
- using ctypes and multiprocessing
- (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: Soeren Gebbert
- """
- from grass.exceptions import FatalError
- import sys
- from multiprocessing import Process, Lock, Pipe
- import logging
- from ctypes import byref, cast, c_int, c_void_p, CFUNCTYPE, POINTER
- from datetime import datetime
- import grass.lib.gis as libgis
- import grass.lib.raster as libraster
- import grass.lib.vector as libvector
- import grass.lib.date as libdate
- import grass.lib.raster3d as libraster3d
- import grass.lib.temporal as libtgis
- from grass.pygrass.rpc.base import RPCServerBase
- from grass.pygrass.raster import RasterRow
- from grass.pygrass.vector import VectorTopo
- from grass.script.utils import encode
- from grass.pygrass.utils import decode
- ###############################################################################
- class RPCDefs(object):
- # Function identifier and index
- STOP = 0
- HAS_TIMESTAMP = 1
- WRITE_TIMESTAMP = 2
- READ_TIMESTAMP = 3
- REMOVE_TIMESTAMP = 4
- READ_MAP_INFO = 5
- MAP_EXISTS = 6
- READ_MAP_INFO = 7
- AVAILABLE_MAPSETS = 8
- GET_DRIVER_NAME = 9
- GET_DATABASE_NAME = 10
- G_MAPSET = 11
- G_LOCATION = 12
- G_GISDBASE = 13
- READ_MAP_FULL_INFO = 14
- WRITE_SEMANTIC_LABEL = 15
- READ_SEMANTIC_LABEL = 16
- REMOVE_SEMANTIC_LABEL = 17
- G_FATAL_ERROR = 49
- TYPE_RASTER = 0
- TYPE_RASTER3D = 1
- TYPE_VECTOR = 2
- ###############################################################################
- def _read_map_full_info(lock, conn, data):
- """Read full map specific metadata from the spatial database using
- PyGRASS functions.
- :param lock: A multiprocessing.Lock instance
- :param conn: A multiprocessing.Pipe instance used to send True or False
- :param data: The list of data entries [function_id, maptype, name, mapset]
- """
- info = None
- try:
- maptype = data[1]
- name = data[2]
- mapset = data[3]
- if maptype == RPCDefs.TYPE_RASTER:
- info = _read_raster_full_info(name, mapset)
- elif maptype == RPCDefs.TYPE_VECTOR:
- info = _read_vector_full_info(name, mapset)
- except:
- raise
- finally:
- conn.send(info)
- ###############################################################################
- def _read_raster_full_info(name, mapset):
- """Read raster info, history and cats using PyGRASS RasterRow
- and return a dictionary. Colors should be supported in the
- future.
- """
- info = {}
- r = RasterRow(name=name, mapset=mapset)
- if r.exist() is True:
- r.open("r")
- for item in r.info:
- info[item[0]] = item[1]
- for item in r.hist:
- info[item[0]] = item[1]
- info["full_name"] = r.name_mapset()
- info["mtype"] = r.mtype
- if r.cats:
- info["cats_title"] = r.cats_title
- info["cats"] = list(r.cats)
- r.close()
- ts = libgis.TimeStamp()
- check = libgis.G_read_raster_timestamp(name, mapset, byref(ts))
- if check:
- dates = _convert_timestamp_from_grass(ts)
- info["start_time"] = dates[0]
- info["end_time"] = dates[1]
- if len(dates) > 2:
- info["time_unit"] = dates[2]
- return info
- ###############################################################################
- def _read_vector_full_info(name, mapset, layer=None):
- """Read vector info using PyGRASS VectorTopo
- and return a dictionary. C
- """
- info = {}
- v = VectorTopo(name=name, mapset=mapset)
- if v.exist() is True:
- v.open("r")
- # Bounding box
- for item in v.bbox().items():
- info[item[0]] = item[1]
- info["name"] = v.name
- info["mapset"] = v.mapset
- info["comment"] = v.comment
- info["full_name"] = v.full_name
- info["is_3d"] = v.is_3D()
- info["map_date"] = v.map_date
- info["maptype"] = v.maptype
- info["organization"] = v.organization
- info["person"] = v.person
- info["proj"] = v.proj
- info["proj_name"] = v.proj_name
- info["title"] = v.title
- info["thresh"] = v.thresh
- info["zone"] = v.zone
- vtypes = [
- "areas",
- "dblinks",
- "faces",
- "holes",
- "islands",
- "kernels",
- "lines",
- "nodes",
- "points",
- "updated_lines",
- "updated_nodes",
- "volumes",
- ]
- for vtype in vtypes:
- info[vtype] = v.number_of(vtype)
- info.update(v.num_primitives())
- if v.table is not None:
- info["columns"] = v.table.columns
- ts = libgis.TimeStamp()
- check = libgis.G_read_vector_timestamp(name, layer, mapset, byref(ts))
- if check:
- dates = _convert_timestamp_from_grass(ts)
- info["start_time"] = dates[0]
- info["end_time"] = dates[1]
- if len(dates) > 2:
- info["time_unit"] = dates[2]
- return info
- def _fatal_error(lock, conn, data):
- """Calls G_fatal_error()"""
- libgis.G_fatal_error("Fatal Error in C library server")
- ###############################################################################
- def _get_mapset(lock, conn, data):
- """Return the current mapset
- :param lock: A multiprocessing.Lock instance
- :param conn: A multiprocessing.Pipe instance used to send True or False
- :param data: The mapset as list entry 1 [function_id]
- :returns: Name of the current mapset
- """
- mapset = libgis.G_mapset()
- conn.send(decode(mapset))
- ###############################################################################
- def _get_location(lock, conn, data):
- """Return the current location
- :param lock: A multiprocessing.Lock instance
- :param conn: A multiprocessing.Pipe instance used to send True or False
- :param data: The mapset as list entry 1 [function_id]
- :returns: Name of the location
- """
- location = libgis.G_location()
- conn.send(decode(location))
- ###############################################################################
- def _get_gisdbase(lock, conn, data):
- """Return the current gisdatabase
- :param lock: A multiprocessing.Lock instance
- :param conn: A multiprocessing.Pipe instance used to send True or False
- :param data: The mapset as list entry 1 [function_id]
- :returns: Name of the gisdatabase
- """
- gisdbase = libgis.G_gisdbase()
- conn.send(decode(gisdbase))
- ###############################################################################
- def _get_driver_name(lock, conn, data):
- """Return the temporal database driver of a specific mapset
- :param lock: A multiprocessing.Lock instance
- :param conn: A multiprocessing.Pipe instance used to send True or False
- :param data: The mapset as list entry 1 [function_id, mapset]
- :returns: Name of the driver or None if no temporal database present
- """
- mapset = data[1]
- if not mapset:
- mapset = libgis.G_mapset()
- else:
- mapset = encode(mapset)
- drstring = libtgis.tgis_get_mapset_driver_name(mapset)
- conn.send(decode(drstring.data))
- ###############################################################################
- def _get_database_name(lock, conn, data):
- """Return the temporal database name of a specific mapset
- :param lock: A multiprocessing.Lock instance
- :param conn: A multiprocessing.Pipe instance used to send True or False
- :param data: The mapset as list entry 1 [function_id, mapset]
- :returns: Name of the database or None if no temporal database present
- """
- dbstring = None
- try:
- mapset = data[1]
- if not mapset:
- mapset = libgis.G_mapset()
- else:
- mapset = encode(mapset)
- dbstring = libtgis.tgis_get_mapset_database_name(mapset)
- dbstring = dbstring.data
- if dbstring:
- # We substitute GRASS variables if they are located in the database string
- # This behavior is in conjunction with db.connect
- dbstring = dbstring.replace(encode("$GISDBASE"), libgis.G_gisdbase())
- dbstring = dbstring.replace(encode("$LOCATION_NAME"), libgis.G_location())
- dbstring = dbstring.replace(encode("$MAPSET"), mapset)
- except:
- raise
- finally:
- conn.send(decode(dbstring))
- ###############################################################################
- def _available_mapsets(lock, conn, data):
- """Return all available mapsets the user can access as a list of strings
- :param lock: A multiprocessing.Lock instance
- :param conn: A multiprocessing.Pipe instance used to send True or False
- :param data: The list of data entries [function_id]
- :returns: Names of available mapsets as list of strings
- """
- count = 0
- mapset_list = []
- try:
- # Initialize the accessible mapset list, this is bad C design!!!
- libgis.G_get_mapset_name(0)
- mapsets = libgis.G_get_available_mapsets()
- while mapsets[count]:
- char_list = ""
- mapset = mapsets[count]
- permission = libgis.G_mapset_permissions(mapset)
- in_search_path = libgis.G_is_mapset_in_search_path(mapset)
- c = 0
- while mapset[c] != b"\x00":
- val = decode(mapset[c])
- char_list += val
- c += 1
- if permission >= 0 and in_search_path == 1:
- mapset_list.append(char_list)
- libgis.G_debug(
- 1,
- "c_library_server._available_mapsets: \n mapset: %s\n"
- " has permission %i\n in search path: %i"
- % (char_list, permission, in_search_path),
- )
- count += 1
- # We need to sort the mapset list, but the first one should be
- # the current mapset
- current_mapset = decode(libgis.G_mapset())
- if current_mapset in mapset_list:
- mapset_list.remove(current_mapset)
- mapset_list.sort()
- mapset_list.reverse()
- mapset_list.append(current_mapset)
- mapset_list.reverse()
- except:
- raise
- finally:
- conn.send(mapset_list)
- ###############################################################################
- def _has_timestamp(lock, conn, data):
- """Check if the file based GRASS timestamp is present and send
- True or False using the provided pipe.
- :param lock: A multiprocessing.Lock instance
- :param conn: A multiprocessing.Pipe instance used to send True or False
- :param data: The list of data entries [function_id, maptype, name,
- mapset, layer]
- """
- check = False
- try:
- maptype = data[1]
- name = data[2]
- mapset = data[3]
- layer = data[4]
- if maptype == RPCDefs.TYPE_RASTER:
- if libgis.G_has_raster_timestamp(name, mapset) == 1:
- check = True
- elif maptype == RPCDefs.TYPE_VECTOR:
- if libgis.G_has_vector_timestamp(name, layer, mapset) == 1:
- check = True
- elif maptype == RPCDefs.TYPE_RASTER3D:
- if libgis.G_has_raster3d_timestamp(name, mapset) == 1:
- check = True
- except:
- raise
- finally:
- conn.send(check)
- ###############################################################################
- def _read_timestamp(lock, conn, data):
- """Read the file based GRASS timestamp and send
- the result using the provided pipe.
- The tuple to be send via pipe: (return value of G_read_*_timestamp,
- timestamps).
- Please have a look at the documentation of G_read_raster_timestamp,
- G_read_vector_timestamp and G_read_raster3d_timestamp for the return
- values description.
- The timestamps to be send are tuples of values:
- - relative time (start, end, unit), start and end are of type
- integer, unit is of type string.
- - absolute time (start, end), start and end are of type datetime
- The end time may be None in case of a time instance.
- :param lock: A multiprocessing.Lock instance
- :param conn: A multiprocessing.Pipe instance used to send the result
- :param data: The list of data entries [function_id, maptype, name,
- mapset, layer]
- """
- check = False
- dates = None
- try:
- maptype = data[1]
- name = data[2]
- mapset = data[3]
- layer = data[4]
- ts = libgis.TimeStamp()
- if maptype == RPCDefs.TYPE_RASTER:
- check = libgis.G_read_raster_timestamp(name, mapset, byref(ts))
- elif maptype == RPCDefs.TYPE_VECTOR:
- check = libgis.G_read_vector_timestamp(name, layer, mapset, byref(ts))
- elif maptype == RPCDefs.TYPE_RASTER3D:
- check = libgis.G_read_raster3d_timestamp(name, mapset, byref(ts))
- dates = _convert_timestamp_from_grass(ts)
- except:
- raise
- finally:
- conn.send((check, dates))
- ###############################################################################
- def _write_timestamp(lock, conn, data):
- """Write the file based GRASS timestamp
- the return values of the called C-functions using the provided pipe.
- The value to be send via pipe is the return value of G_write_*_timestamp.
- Please have a look at the documentation of G_write_raster_timestamp,
- G_write_vector_timestamp and G_write_raster3d_timestamp for the return
- values description.
- :param lock: A multiprocessing.Lock instance
- :param conn: A multiprocessing.Pipe instance used to send True or False
- :param data: The list of data entries [function_id, maptype, name,
- mapset, layer, timestring]
- """
- check = -3
- try:
- maptype = data[1]
- name = data[2]
- # mapset = data[3]
- layer = data[4]
- timestring = data[5]
- ts = libgis.TimeStamp()
- check = libgis.G_scan_timestamp(byref(ts), timestring)
- if check != 1:
- logging.error("Unable to convert the timestamp: " + timestring)
- return -2
- if maptype == RPCDefs.TYPE_RASTER:
- check = libgis.G_write_raster_timestamp(name, byref(ts))
- elif maptype == RPCDefs.TYPE_VECTOR:
- check = libgis.G_write_vector_timestamp(name, layer, byref(ts))
- elif maptype == RPCDefs.TYPE_RASTER3D:
- check = libgis.G_write_raster3d_timestamp(name, byref(ts))
- except:
- raise
- finally:
- conn.send(check)
- ###############################################################################
- def _remove_timestamp(lock, conn, data):
- """Remove the file based GRASS timestamp
- the return values of the called C-functions using the provided pipe.
- The value to be send via pipe is the return value of G_remove_*_timestamp.
- Please have a look at the documentation of G_remove_raster_timestamp,
- G_remove_vector_timestamp and G_remove_raster3d_timestamp for the return
- values description.
- :param lock: A multiprocessing.Lock instance
- :param conn: A multiprocessing.Pipe instance used to send True or False
- :param data: The list of data entries [function_id, maptype, name,
- mapset, layer]
- """
- check = False
- try:
- maptype = data[1]
- name = data[2]
- mapset = data[3]
- layer = data[4]
- if maptype == RPCDefs.TYPE_RASTER:
- check = libgis.G_remove_raster_timestamp(name, mapset)
- elif maptype == RPCDefs.TYPE_VECTOR:
- check = libgis.G_remove_vector_timestamp(name, layer, mapset)
- elif maptype == RPCDefs.TYPE_RASTER3D:
- check = libgis.G_remove_raster3d_timestamp(name, mapset)
- except:
- raise
- finally:
- conn.send(check)
- ###############################################################################
- def _read_semantic_label(lock, conn, data):
- """Read the file based GRASS band identifier
- the result using the provided pipe.
- The result to be sent via pipe is the return value of
- Rast_read_semantic_label: either a semantic label string or None.
- :param lock: A multiprocessing.Lock instance
- :param conn: A multiprocessing.Pipe instance used to send True or False
- :param data: The list of data entries [function_id, maptype, name,
- mapset, layer, timestring]
- """
- semantic_label = None
- try:
- maptype = data[1]
- name = data[2]
- mapset = data[3]
- # layer = data[4]
- if maptype == RPCDefs.TYPE_RASTER:
- # Must use temporary variable to work around
- # ValueError: ctypes objects containing pointers cannot be pickled
- ret = libraster.Rast_read_semantic_label(name, mapset)
- if ret:
- semantic_label = decode(ret)
- else:
- logging.error(
- "Unable to read semantic label. " "Unsupported map type %s" % maptype
- )
- return -1
- except:
- raise
- finally:
- conn.send(semantic_label)
- ###############################################################################
- def _write_semantic_label(lock, conn, data):
- """Write the file based GRASS band identifier.
- Rises ValueError on invalid semantic label.
- Always sends back True.
- :param lock: A multiprocessing.Lock instance
- :param conn: A multiprocessing.Pipe instance used to send True or False
- :param data: The list of data entries [function_id, maptype, name,
- mapset, layer, timestring]
- """
- try:
- maptype = data[1]
- name = data[2]
- # mapset = data[3]
- # layer = data[4]
- semantic_label = data[5]
- if maptype == RPCDefs.TYPE_RASTER:
- if libraster.Rast_legal_semantic_label(semantic_label) is False:
- raise ValueError(_("Invalid semantic label"))
- libraster.Rast_write_semantic_label(name, semantic_label)
- else:
- logging.error(
- "Unable to write semantic label. " "Unsupported map type %s" % maptype
- )
- return -2
- except:
- raise
- finally:
- conn.send(True)
- ###############################################################################
- def _remove_semantic_label(lock, conn, data):
- """Remove the file based GRASS band identifier.
- The value to be send via pipe is the return value of G_remove_misc.
- :param lock: A multiprocessing.Lock instance
- :param conn: A multiprocessing.Pipe instance used to send True or False
- :param data: The list of data entries [function_id, maptype, name,
- mapset, layer, timestring]
- """
- check = False
- try:
- maptype = data[1]
- name = data[2]
- # mapset = data[3]
- # layer = data[4]
- if maptype == RPCDefs.TYPE_RASTER:
- check = libgis.G_remove_misc("cell_misc", "semantic_label", name)
- else:
- logging.error(
- "Unable to remove semantic label. " "Unsupported map type %s" % maptype
- )
- return -2
- except:
- raise
- finally:
- conn.send(check)
- ###############################################################################
- def _map_exists(lock, conn, data):
- """Check if a map exists in the spatial database
- The value to be send via pipe is True in case the map exists and False
- if not.
- :param lock: A multiprocessing.Lock instance
- :param conn: A multiprocessing.Pipe instance used to send True or False
- :param data: The list of data entries [function_id, maptype, name, mapset]
- """
- check = False
- try:
- maptype = data[1]
- name = data[2]
- mapset = data[3]
- if maptype == RPCDefs.TYPE_RASTER:
- mapset = libgis.G_find_raster(name, mapset)
- elif maptype == RPCDefs.TYPE_VECTOR:
- mapset = libgis.G_find_vector(name, mapset)
- elif maptype == RPCDefs.TYPE_RASTER3D:
- mapset = libgis.G_find_raster3d(name, mapset)
- if mapset:
- check = True
- except:
- raise
- finally:
- conn.send(check)
- ###############################################################################
- def _read_map_info(lock, conn, data):
- """Read map specific metadata from the spatial database using C-library
- functions
- :param lock: A multiprocessing.Lock instance
- :param conn: A multiprocessing.Pipe instance used to send True or False
- :param data: The list of data entries [function_id, maptype, name, mapset]
- """
- kvp = None
- try:
- maptype = data[1]
- name = data[2]
- mapset = data[3]
- if maptype == RPCDefs.TYPE_RASTER:
- kvp = _read_raster_info(name, mapset)
- elif maptype == RPCDefs.TYPE_VECTOR:
- kvp = _read_vector_info(name, mapset)
- elif maptype == RPCDefs.TYPE_RASTER3D:
- kvp = _read_raster3d_info(name, mapset)
- except:
- raise
- finally:
- conn.send(kvp)
- ###############################################################################
- def _read_raster_info(name, mapset):
- """Read the raster map info from the file system and store the content
- into a dictionary
- This method uses the ctypes interface to the gis and raster libraries
- to read the map metadata information
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: The key value pairs of the map specific metadata, or None in
- case of an error
- """
- kvp = {}
- if not libgis.G_find_raster(name, mapset):
- return None
- # Read the region information
- region = libraster.struct_Cell_head()
- libraster.Rast_get_cellhd(name, mapset, byref(region))
- kvp["north"] = region.north
- kvp["south"] = region.south
- kvp["east"] = region.east
- kvp["west"] = region.west
- kvp["nsres"] = region.ns_res
- kvp["ewres"] = region.ew_res
- kvp["rows"] = region.cols
- kvp["cols"] = region.rows
- maptype = libraster.Rast_map_type(name, mapset)
- if maptype == libraster.DCELL_TYPE:
- kvp["datatype"] = "DCELL"
- elif maptype == libraster.FCELL_TYPE:
- kvp["datatype"] = "FCELL"
- elif maptype == libraster.CELL_TYPE:
- kvp["datatype"] = "CELL"
- # Read range
- if libraster.Rast_map_is_fp(name, mapset):
- range = libraster.FPRange()
- libraster.Rast_init_fp_range(byref(range))
- ret = libraster.Rast_read_fp_range(name, mapset, byref(range))
- if ret < 0:
- logging.error(_("Unable to read range file"))
- return None
- if ret == 2:
- kvp["min"] = None
- kvp["max"] = None
- else:
- min = libgis.DCELL()
- max = libgis.DCELL()
- libraster.Rast_get_fp_range_min_max(byref(range), byref(min), byref(max))
- kvp["min"] = min.value
- kvp["max"] = max.value
- else:
- range = libraster.Range()
- libraster.Rast_init_range(byref(range))
- ret = libraster.Rast_read_range(name, mapset, byref(range))
- if ret < 0:
- logging.error(_("Unable to read range file"))
- return None
- if ret == 2:
- kvp["min"] = None
- kvp["max"] = None
- else:
- min = libgis.CELL()
- max = libgis.CELL()
- libraster.Rast_get_range_min_max(byref(range), byref(min), byref(max))
- kvp["min"] = min.value
- kvp["max"] = max.value
- return kvp
- ###############################################################################
- def _read_raster3d_info(name, mapset):
- """Read the 3D raster map info from the file system and store the content
- into a dictionary
- This method uses the ctypes interface to the gis and raster3d libraries
- to read the map metadata information
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: The key value pairs of the map specific metadata, or None in
- case of an error
- """
- kvp = {}
- if not libgis.G_find_raster3d(name, mapset):
- return None
- # Read the region information
- region = libraster3d.RASTER3D_Region()
- libraster3d.Rast3d_read_region_map(name, mapset, byref(region))
- kvp["north"] = region.north
- kvp["south"] = region.south
- kvp["east"] = region.east
- kvp["west"] = region.west
- kvp["nsres"] = region.ns_res
- kvp["ewres"] = region.ew_res
- kvp["tbres"] = region.tb_res
- kvp["rows"] = region.cols
- kvp["cols"] = region.rows
- kvp["depths"] = region.depths
- kvp["top"] = region.top
- kvp["bottom"] = region.bottom
- # We need to open the map, this function returns a void pointer
- # but we may need the correct type which is RASTER3D_Map, hence
- # the casting
- g3map = cast(
- libraster3d.Rast3d_open_cell_old(
- name,
- mapset,
- libraster3d.RASTER3D_DEFAULT_WINDOW,
- libraster3d.RASTER3D_TILE_SAME_AS_FILE,
- libraster3d.RASTER3D_NO_CACHE,
- ),
- POINTER(libraster3d.RASTER3D_Map),
- )
- if not g3map:
- logging.error(_("Unable to open 3D raster map <%s>" % (name)))
- return None
- maptype = libraster3d.Rast3d_file_type_map(g3map)
- if maptype == libraster.DCELL_TYPE:
- kvp["datatype"] = "DCELL"
- elif maptype == libraster.FCELL_TYPE:
- kvp["datatype"] = "FCELL"
- # Read range
- min = libgis.DCELL()
- max = libgis.DCELL()
- ret = libraster3d.Rast3d_range_load(g3map)
- if not ret:
- logging.error(_("Unable to load range of 3D raster map <%s>" % (name)))
- return None
- libraster3d.Rast3d_range_min_max(g3map, byref(min), byref(max))
- if min.value != min.value:
- kvp["min"] = None
- else:
- kvp["min"] = float(min.value)
- if max.value != max.value:
- kvp["max"] = None
- else:
- kvp["max"] = float(max.value)
- if not libraster3d.Rast3d_close(g3map):
- logging.error(_("Unable to close 3D raster map <%s>" % (name)))
- return None
- return kvp
- ###############################################################################
- def _read_vector_info(name, mapset):
- """Read the vector map info from the file system and store the content
- into a dictionary
- This method uses the ctypes interface to the vector libraries
- to read the map metadata information
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: The key value pairs of the map specific metadata, or None in
- case of an error
- """
- kvp = {}
- if not libgis.G_find_vector(name, mapset):
- return None
- # The vector map structure
- Map = libvector.Map_info()
- # We open the maps always in topology mode first
- libvector.Vect_set_open_level(2)
- with_topo = True
- # Code lend from v.info main.c
- if libvector.Vect_open_old_head2(byref(Map), name, mapset, "1") < 2:
- # force level 1, open fully
- # NOTE: number of points, lines, boundaries, centroids,
- # faces, kernels is still available
- libvector.Vect_set_open_level(1) # no topology
- with_topo = False
- if libvector.Vect_open_old2(byref(Map), name, mapset, "1") < 1:
- logging.error(
- _(
- "Unable to open vector map <%s>"
- % (libvector.Vect_get_full_name(byref(Map)))
- )
- )
- return None
- # Release the vector spatial index memory when closed
- libvector.Vect_set_release_support(byref(Map))
- # Read the extent information
- bbox = libvector.bound_box()
- libvector.Vect_get_map_box(byref(Map), byref(bbox))
- kvp["north"] = bbox.N
- kvp["south"] = bbox.S
- kvp["east"] = bbox.E
- kvp["west"] = bbox.W
- kvp["top"] = bbox.T
- kvp["bottom"] = bbox.B
- kvp["map3d"] = bool(libvector.Vect_is_3d(byref(Map)))
- # Read number of features
- if with_topo:
- kvp["points"] = libvector.Vect_get_num_primitives(
- byref(Map), libvector.GV_POINT
- )
- kvp["lines"] = libvector.Vect_get_num_primitives(byref(Map), libvector.GV_LINE)
- kvp["boundaries"] = libvector.Vect_get_num_primitives(
- byref(Map), libvector.GV_BOUNDARY
- )
- kvp["centroids"] = libvector.Vect_get_num_primitives(
- byref(Map), libvector.GV_CENTROID
- )
- kvp["faces"] = libvector.Vect_get_num_primitives(byref(Map), libvector.GV_FACE)
- kvp["kernels"] = libvector.Vect_get_num_primitives(
- byref(Map), libvector.GV_KERNEL
- )
- # Summarize the primitives
- kvp["primitives"] = (
- kvp["points"] + kvp["lines"] + kvp["boundaries"] + kvp["centroids"]
- )
- if kvp["map3d"]:
- kvp["primitives"] += kvp["faces"] + kvp["kernels"]
- # Read topology information
- kvp["nodes"] = libvector.Vect_get_num_nodes(byref(Map))
- kvp["areas"] = libvector.Vect_get_num_areas(byref(Map))
- kvp["islands"] = libvector.Vect_get_num_islands(byref(Map))
- kvp["holes"] = libvector.Vect_get_num_holes(byref(Map))
- kvp["volumes"] = libvector.Vect_get_num_primitives(
- byref(Map), libvector.GV_VOLUME
- )
- else:
- kvp["points"] = None
- kvp["lines"] = None
- kvp["boundaries"] = None
- kvp["centroids"] = None
- kvp["faces"] = None
- kvp["kernels"] = None
- kvp["primitives"] = None
- kvp["nodes"] = None
- kvp["areas"] = None
- kvp["islands"] = None
- kvp["holes"] = None
- kvp["volumes"] = None
- libvector.Vect_close(byref(Map))
- return kvp
- ###############################################################################
- def _convert_timestamp_from_grass(ts):
- """Convert a GRASS file based timestamp into the temporal framework
- format datetime or integer.
- A tuple of two datetime objects (start, end) is returned in case of
- absolute time.
- In case of relative time a tuple with start time, end time and the
- relative unit (start, end, unit) will be returned.
- Note:
- The end time will be set to None in case of a time instance.
- :param ts grass.lib.gis.TimeStamp object created by G_read_*_timestamp
- """
- dt1 = libgis.DateTime()
- dt2 = libgis.DateTime()
- count = c_int()
- libgis.G_get_timestamps(byref(ts), byref(dt1), byref(dt2), byref(count))
- if dt1.mode == libdate.DATETIME_ABSOLUTE:
- pdt1 = None
- pdt2 = None
- if count.value >= 1:
- pdt1 = datetime(
- int(dt1.year),
- int(dt1.month),
- int(dt1.day),
- int(dt1.hour),
- int(dt1.minute),
- int(dt1.second),
- )
- if count.value == 2:
- pdt2 = datetime(
- int(dt2.year),
- int(dt2.month),
- int(dt2.day),
- int(dt2.hour),
- int(dt2.minute),
- int(dt2.second),
- )
- # ATTENTION: We ignore the time zone
- # TODO: Write time zone support
- return (pdt1, pdt2)
- else:
- unit = None
- start = None
- end = None
- if count.value >= 1:
- if dt1.year > 0:
- unit = "years"
- start = dt1.year
- elif dt1.month > 0:
- unit = "months"
- start = dt1.month
- elif dt1.day > 0:
- unit = "days"
- start = dt1.day
- elif dt1.hour > 0:
- unit = "hours"
- start = dt1.hour
- elif dt1.minute > 0:
- unit = "minutes"
- start = dt1.minute
- elif dt1.second > 0:
- unit = "seconds"
- start = dt1.second
- if count.value == 2:
- if dt2.year > 0:
- end = dt2.year
- elif dt2.month > 0:
- end = dt2.month
- elif dt2.day > 0:
- end = dt2.day
- elif dt2.hour > 0:
- end = dt2.hour
- elif dt2.minute > 0:
- end = dt2.minute
- elif dt2.second > 0:
- end = dt2.second
- return (start, end, unit)
- ###############################################################################
- def _stop(lock, conn, data):
- libgis.G_debug(1, "Stop C-interface server")
- conn.close()
- lock.release()
- sys.exit()
- ###############################################################################
- def c_library_server(lock, conn):
- """The GRASS C-libraries server function designed to be a target for
- multiprocessing.Process
- :param lock: A multiprocessing.Lock
- :param conn: A multiprocessing.Pipe
- """
- def error_handler(data):
- """This function will be called in case of a fatal error in libgis"""
- # sys.stderr.write("Error handler was called\n")
- # We send an exception that will be handled in
- # the parent process, then close the pipe
- # and release any possible lock
- conn.send(FatalError())
- conn.close()
- lock.release()
- CALLBACK = CFUNCTYPE(c_void_p, c_void_p)
- CALLBACK.restype = c_void_p
- CALLBACK.argtypes = c_void_p
- cerror_handler = CALLBACK(error_handler)
- libgis.G_add_error_handler(cerror_handler, None)
- # Crerate the function array
- functions = [0] * 50
- functions[RPCDefs.STOP] = _stop
- functions[RPCDefs.HAS_TIMESTAMP] = _has_timestamp
- functions[RPCDefs.WRITE_TIMESTAMP] = _write_timestamp
- functions[RPCDefs.READ_TIMESTAMP] = _read_timestamp
- functions[RPCDefs.REMOVE_TIMESTAMP] = _remove_timestamp
- functions[RPCDefs.READ_MAP_INFO] = _read_map_info
- functions[RPCDefs.MAP_EXISTS] = _map_exists
- functions[RPCDefs.AVAILABLE_MAPSETS] = _available_mapsets
- functions[RPCDefs.GET_DRIVER_NAME] = _get_driver_name
- functions[RPCDefs.GET_DATABASE_NAME] = _get_database_name
- functions[RPCDefs.G_MAPSET] = _get_mapset
- functions[RPCDefs.G_LOCATION] = _get_location
- functions[RPCDefs.G_GISDBASE] = _get_gisdbase
- functions[RPCDefs.READ_MAP_FULL_INFO] = _read_map_full_info
- functions[RPCDefs.WRITE_SEMANTIC_LABEL] = _write_semantic_label
- functions[RPCDefs.READ_SEMANTIC_LABEL] = _read_semantic_label
- functions[RPCDefs.REMOVE_SEMANTIC_LABEL] = _remove_semantic_label
- functions[RPCDefs.G_FATAL_ERROR] = _fatal_error
- libgis.G_gisinit("c_library_server")
- libgis.G_debug(1, "Start C-interface server")
- while True:
- # Avoid busy waiting
- conn.poll(None)
- data = conn.recv()
- lock.acquire()
- functions[data[0]](lock, conn, data)
- lock.release()
- class CLibrariesInterface(RPCServerBase):
- """Fast and exit-safe interface to GRASS C-libraries functions
- This class implements a fast and exit-safe interface to the GRASS
- gis, raster, 3D raster and vector C-libraries functions.
- The C-libraries functions are called via ctypes in a subprocess
- using a pipe (multiprocessing.Pipe) to transfer the text messages.
- Hence, the process that uses the CLibrariesInterface will not be
- exited, if a G_fatal_error() was invoked in the subprocess.
- In this case the CLibrariesInterface object will simply start a
- new subprocess and restarts the pipeline.
- Usage:
- .. code-block:: python
- >>> import grass.script as gscript
- >>> import grass.temporal as tgis
- >>> gscript.use_temp_region()
- >>> gscript.run_command("g.region", n=80.0, s=0.0, e=120.0, w=0.0,
- ... t=1.0, b=0.0, res=10.0, res3=10.0)
- 0
- >>> tgis.init()
- >>> gscript.run_command("r.mapcalc", expression="test = 1", overwrite=True, quiet=True)
- 0
- >>> gscript.run_command("r3.mapcalc", expression="test = 1", overwrite=True, quiet=True)
- 0
- >>> gscript.run_command("v.random", output="test", n=10, overwrite=True, quiet=True)
- 0
- >>> gscript.run_command("r.timestamp", map="test", date='12 Mar 1995 10:34:40', overwrite=True, quiet=True)
- 0
- >>> gscript.run_command("r3.timestamp", map="test", date='12 Mar 1995 10:34:40', overwrite=True, quiet=True)
- 0
- >>> gscript.run_command("v.timestamp", map="test", date='12 Mar 1995 10:34:40', overwrite=True, quiet=True)
- 0
- # Check mapsets
- >>> ciface = tgis.CLibrariesInterface()
- >>> mapsets = ciface.available_mapsets()
- >>> mapsets[0] == tgis.get_current_mapset()
- True
- # Raster map
- >>> ciface = tgis.CLibrariesInterface()
- >>> check = ciface.raster_map_exists("test", tgis.get_current_mapset())
- >>> print check
- True
- >>> ciface.read_raster_info("test", tgis.get_current_mapset())
- {'rows': 12, 'north': 80.0, 'min': 1, 'datatype': 'CELL', 'max': 1, 'ewres': 10.0, 'cols': 8, 'west': 0.0, 'east': 120.0, 'nsres': 10.0, 'south': 0.0}
- >>> info = ciface.read_raster_full_info("test", tgis.get_current_mapset())
- >>> info # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
- {u'tbres': 1.0, ... 'keyword': 'generated by r.mapcalc',
- u'bottom': 0.0, 'end_time': None, 'title': 'test', u'south': 0.0}
- >>> info["start_time"]
- datetime.datetime(1995, 3, 12, 10, 34, 40)
- >>> info["end_time"]
- >>> check = ciface.has_raster_timestamp("test", tgis.get_current_mapset())
- >>> print check
- True
- >>> if check:
- ... res = ciface.read_raster_timestamp("test", tgis.get_current_mapset())
- ... if res[0]:
- ... print str(res[1][0]), str(res[1][0])
- ... ciface.remove_raster_timestamp("test", tgis.get_current_mapset())
- 1995-03-12 10:34:40 1995-03-12 10:34:40
- 1
- >>> ciface.has_raster_timestamp("test", tgis.get_current_mapset())
- False
- >>> ciface.write_raster_timestamp("test", tgis.get_current_mapset(), "13 Jan 1999 14:30:05")
- 1
- >>> ciface.has_raster_timestamp("test", tgis.get_current_mapset())
- True
- # 3D raster map
- >>> check = ciface.raster3d_map_exists("test", tgis.get_current_mapset())
- >>> print check
- True
- >>> ciface.read_raster3d_info("test", tgis.get_current_mapset())
- {'tbres': 1.0, 'rows': 12, 'north': 80.0, 'bottom': 0.0, 'datatype': 'DCELL', 'max': 1.0, 'top': 1.0, 'min': 1.0, 'cols': 8, 'depths': 1, 'west': 0.0, 'ewres': 10.0, 'east': 120.0, 'nsres': 10.0, 'south': 0.0}
- >>> check = ciface.has_raster3d_timestamp("test", tgis.get_current_mapset())
- >>> print check
- True
- >>> if check:
- ... res = ciface.read_raster3d_timestamp("test", tgis.get_current_mapset())
- ... if res[0]:
- ... print str(res[1][0]), str(res[1][0])
- ... ciface.remove_raster3d_timestamp("test", tgis.get_current_mapset())
- 1995-03-12 10:34:40 1995-03-12 10:34:40
- 1
- >>> ciface.has_raster3d_timestamp("test", tgis.get_current_mapset())
- False
- >>> ciface.write_raster3d_timestamp("test", tgis.get_current_mapset(), "13 Jan 1999 14:30:05")
- 1
- >>> ciface.has_raster3d_timestamp("test", tgis.get_current_mapset())
- True
- # Vector map
- >>> check = ciface.vector_map_exists("test", tgis.get_current_mapset())
- >>> print check
- True
- >>> kvp = ciface.read_vector_info("test", tgis.get_current_mapset())
- >>> kvp['points']
- 10
- >>> kvp = ciface.read_vector_full_info("test", tgis.get_current_mapset())
- >>> print kvp['points']
- 10
- >>> kvp['point']
- 10
- >>> kvp['area']
- 0
- >>> kvp['lines']
- 0
- >>> kvp['line']
- 0
- >>> 'columns' in kvp
- False
- >>> kvp["start_time"]
- datetime.datetime(1995, 3, 12, 10, 34, 40)
- >>> kvp["end_time"]
- >>> check = ciface.has_vector_timestamp("test", tgis.get_current_mapset(), None)
- >>> print check
- True
- >>> if check:
- ... res = ciface.read_vector_timestamp("test", tgis.get_current_mapset())
- ... if res[0]:
- ... print str(res[1][0]), str(res[1][0])
- ... ciface.remove_vector_timestamp("test", tgis.get_current_mapset())
- 1995-03-12 10:34:40 1995-03-12 10:34:40
- 1
- >>> ciface.has_vector_timestamp("test", tgis.get_current_mapset())
- False
- >>> ciface.write_vector_timestamp("test", tgis.get_current_mapset(), "13 Jan 1999 14:30:05")
- 1
- >>> ciface.has_vector_timestamp("test", tgis.get_current_mapset())
- True
- >>> ciface.get_driver_name()
- 'sqlite'
- >>> ciface.get_database_name().split("/")[-1]
- 'sqlite.db'
- >>> mapset = ciface.get_mapset()
- >>> location = ciface.get_location()
- >>> gisdbase = ciface.get_gisdbase()
- >>> ciface.fatal_error() # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE
- Traceback (most recent call last):
- raise FatalError("Exception raised: " + str(e) + " Message: " + message)
- FatalError: Exception raised: ...
- >>> ciface.fatal_error() # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE
- Traceback (most recent call last):
- raise FatalError("Exception raised: " + str(e) + " Message: " + message)
- FatalError: Exception raised: ...
- >>> ciface.fatal_error() # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE
- Traceback (most recent call last):
- raise FatalError("Exception raised: " + str(e) + " Message: " + message)
- FatalError: Exception raised: ...
- >>> ciface.fatal_error() # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE
- Traceback (most recent call last):
- raise FatalError("Exception raised: " + str(e) + " Message: " + message)
- FatalError: Exception raised: ...
- >>> ciface.stop()
- >>> gscript.del_temp_region()
- """
- def __init__(self):
- RPCServerBase.__init__(self)
- def start_server(self):
- self.client_conn, self.server_conn = Pipe(True)
- self.lock = Lock()
- self.server = Process(
- target=c_library_server, args=(self.lock, self.server_conn)
- )
- self.server.daemon = True
- self.server.start()
- def raster_map_exists(self, name, mapset):
- """Check if a raster map exists in the spatial database
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: True if exists, False if not
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.MAP_EXISTS, RPCDefs.TYPE_RASTER, name, mapset, None]
- )
- return self.safe_receive("raster_map_exists")
- def read_raster_info(self, name, mapset):
- """Read the raster map info from the file system and store the content
- into a dictionary
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: The key value pairs of the map specific metadata,
- or None in case of an error
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.READ_MAP_INFO, RPCDefs.TYPE_RASTER, name, mapset, None]
- )
- return self.safe_receive("read_raster_info")
- def read_raster_full_info(self, name, mapset):
- """Read raster info, history and cats using PyGRASS RasterRow
- and return a dictionary. Colors should be supported in the
- future.
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: The key value pairs of the map specific metadata,
- or None in case of an error
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.READ_MAP_FULL_INFO, RPCDefs.TYPE_RASTER, name, mapset, None]
- )
- return self.safe_receive("read_raster_full_info")
- def has_raster_timestamp(self, name, mapset):
- """Check if a file based raster timestamp exists
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: True if exists, False if not
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.HAS_TIMESTAMP, RPCDefs.TYPE_RASTER, name, mapset, None]
- )
- return self.safe_receive("has_raster_timestamp")
- def remove_raster_timestamp(self, name, mapset):
- """Remove a file based raster timestamp
- Please have a look at the documentation G_remove_raster_timestamp
- for the return values description.
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: The return value of G_remove_raster_timestamp
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.REMOVE_TIMESTAMP, RPCDefs.TYPE_RASTER, name, mapset, None]
- )
- return self.safe_receive("remove_raster_timestamp")
- def read_raster_timestamp(self, name, mapset):
- """Read a file based raster timestamp
- Please have a look at the documentation G_read_raster_timestamp
- for the return values description.
- The timestamps to be send are tuples of values:
- - relative time (start, end, unit), start and end are of type
- integer, unit is of type string.
- - absolute time (start, end), start and end are of type datetime
- The end time may be None in case of a time instance.
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: The return value of G_read_raster_timestamp
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.READ_TIMESTAMP, RPCDefs.TYPE_RASTER, name, mapset, None]
- )
- return self.safe_receive("read_raster_timestamp")
- def write_raster_timestamp(self, name, mapset, timestring):
- """Write a file based raster timestamp
- Please have a look at the documentation G_write_raster_timestamp
- for the return values description.
- Note:
- Only timestamps of maps from the current mapset can written.
- :param name: The name of the map
- :param mapset: The mapset of the map
- :param timestring: A GRASS datetime C-library compatible string
- :returns: The return value of G_write_raster_timestamp
- """
- self.check_server()
- self.client_conn.send(
- [
- RPCDefs.WRITE_TIMESTAMP,
- RPCDefs.TYPE_RASTER,
- name,
- mapset,
- None,
- timestring,
- ]
- )
- return self.safe_receive("write_raster_timestamp")
- def remove_raster_semantic_label(self, name, mapset):
- """Remove a file based raster semantic label
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: The return value of G_remove_misc
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.REMOVE_SEMANTIC_LABEL, RPCDefs.TYPE_RASTER, name, mapset, None]
- )
- return self.safe_receive("remove_raster_semantic_label")
- def read_raster_semantic_label(self, name, mapset):
- """Read a file based raster semantic label
- Returns semantic label or None
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: The return value of Rast_read_semantic_label
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.READ_SEMANTIC_LABEL, RPCDefs.TYPE_RASTER, name, mapset, None]
- )
- return self.safe_receive("read_raster_semantic_label")
- def write_raster_semantic_label(self, name, mapset, semantic_label):
- """Write a file based raster semantic label
- Note:
- Only semantic labels of maps from the current mapset can be written.
- :param name: The name of the map
- :param mapset: The mapset of the map
- :param semantic_label: semantic label
- :returns: always True
- """
- self.check_server()
- self.client_conn.send(
- [
- RPCDefs.WRITE_SEMANTIC_LABEL,
- RPCDefs.TYPE_RASTER,
- name,
- mapset,
- None,
- semantic_label,
- ]
- )
- return self.safe_receive("write_raster_semantic_label")
- def raster3d_map_exists(self, name, mapset):
- """Check if a 3D raster map exists in the spatial database
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: True if exists, False if not
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.MAP_EXISTS, RPCDefs.TYPE_RASTER3D, name, mapset, None]
- )
- return self.safe_receive("raster3d_map_exists")
- def read_raster3d_info(self, name, mapset):
- """Read the 3D raster map info from the file system and store the content
- into a dictionary
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: The key value pairs of the map specific metadata,
- or None in case of an error
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.READ_MAP_INFO, RPCDefs.TYPE_RASTER3D, name, mapset, None]
- )
- return self.safe_receive("read_raster3d_info")
- def has_raster3d_timestamp(self, name, mapset):
- """Check if a file based 3D raster timestamp exists
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: True if exists, False if not
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.HAS_TIMESTAMP, RPCDefs.TYPE_RASTER3D, name, mapset, None]
- )
- return self.safe_receive("has_raster3d_timestamp")
- def remove_raster3d_timestamp(self, name, mapset):
- """Remove a file based 3D raster timestamp
- Please have a look at the documentation G_remove_raster3d_timestamp
- for the return values description.
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: The return value of G_remove_raster3d_timestamp
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.REMOVE_TIMESTAMP, RPCDefs.TYPE_RASTER3D, name, mapset, None]
- )
- return self.safe_receive("remove_raster3d_timestamp")
- def read_raster3d_timestamp(self, name, mapset):
- """Read a file based 3D raster timestamp
- Please have a look at the documentation G_read_raster3d_timestamp
- for the return values description.
- The timestamps to be send are tuples of values:
- - relative time (start, end, unit), start and end are of type
- integer, unit is of type string.
- - absolute time (start, end), start and end are of type datetime
- The end time may be None in case of a time instance.
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: The return value of G_read_raster3d_timestamp
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.READ_TIMESTAMP, RPCDefs.TYPE_RASTER3D, name, mapset, None]
- )
- return self.safe_receive("read_raster3d_timestamp")
- def write_raster3d_timestamp(self, name, mapset, timestring):
- """Write a file based 3D raster timestamp
- Please have a look at the documentation G_write_raster3d_timestamp
- for the return values description.
- Note:
- Only timestamps of maps from the current mapset can written.
- :param name: The name of the map
- :param mapset: The mapset of the map
- :param timestring: A GRASS datetime C-library compatible string
- :returns: The return value of G_write_raster3d_timestamp
- """
- self.check_server()
- self.client_conn.send(
- [
- RPCDefs.WRITE_TIMESTAMP,
- RPCDefs.TYPE_RASTER3D,
- name,
- mapset,
- None,
- timestring,
- ]
- )
- return self.safe_receive("write_raster3d_timestamp")
- def vector_map_exists(self, name, mapset):
- """Check if a vector map exists in the spatial database
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: True if exists, False if not
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.MAP_EXISTS, RPCDefs.TYPE_VECTOR, name, mapset, None]
- )
- return self.safe_receive("vector_map_exists")
- def read_vector_info(self, name, mapset):
- """Read the vector map info from the file system and store the content
- into a dictionary
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: The key value pairs of the map specific metadata,
- or None in case of an error
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.READ_MAP_INFO, RPCDefs.TYPE_VECTOR, name, mapset, None]
- )
- return self.safe_receive("read_vector_info")
- def read_vector_full_info(self, name, mapset):
- """Read vector info using PyGRASS VectorTopo
- and return a dictionary.
- :param name: The name of the map
- :param mapset: The mapset of the map
- :returns: The key value pairs of the map specific metadata,
- or None in case of an error
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.READ_MAP_FULL_INFO, RPCDefs.TYPE_VECTOR, name, mapset, None]
- )
- return self.safe_receive("read_vector_full_info")
- def has_vector_timestamp(self, name, mapset, layer=None):
- """Check if a file based vector timestamp exists
- :param name: The name of the map
- :param mapset: The mapset of the map
- :param layer: The layer of the vector map
- :returns: True if exists, False if not
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.HAS_TIMESTAMP, RPCDefs.TYPE_VECTOR, name, mapset, layer]
- )
- return self.safe_receive("has_vector_timestamp")
- def remove_vector_timestamp(self, name, mapset, layer=None):
- """Remove a file based vector timestamp
- Please have a look at the documentation G_remove_vector_timestamp
- for the return values description.
- :param name: The name of the map
- :param mapset: The mapset of the map
- :param layer: The layer of the vector map
- :returns: The return value of G_remove_vector_timestamp
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.REMOVE_TIMESTAMP, RPCDefs.TYPE_VECTOR, name, mapset, layer]
- )
- return self.safe_receive("remove_vector_timestamp")
- def read_vector_timestamp(self, name, mapset, layer=None):
- """Read a file based vector timestamp
- Please have a look at the documentation G_read_vector_timestamp
- for the return values description.
- The timestamps to be send are tuples of values:
- - relative time (start, end, unit), start and end are of type
- integer, unit is of type string.
- - absolute time (start, end), start and end are of type datetime
- The end time may be None in case of a time instance.
- :param name: The name of the map
- :param mapset: The mapset of the map
- :param layer: The layer of the vector map
- :returns: The return value ofG_read_vector_timestamp and the timestamps
- """
- self.check_server()
- self.client_conn.send(
- [RPCDefs.READ_TIMESTAMP, RPCDefs.TYPE_VECTOR, name, mapset, layer]
- )
- return self.safe_receive("read_vector_timestamp")
- def write_vector_timestamp(self, name, mapset, timestring, layer=None):
- """Write a file based vector timestamp
- Please have a look at the documentation G_write_vector_timestamp
- for the return values description.
- Note:
- Only timestamps pf maps from the current mapset can written.
- :param name: The name of the map
- :param mapset: The mapset of the map
- :param timestring: A GRASS datetime C-library compatible string
- :param layer: The layer of the vector map
- :returns: The return value of G_write_vector_timestamp
- """
- self.check_server()
- self.client_conn.send(
- [
- RPCDefs.WRITE_TIMESTAMP,
- RPCDefs.TYPE_VECTOR,
- name,
- mapset,
- layer,
- timestring,
- ]
- )
- return self.safe_receive("write_vector_timestamp")
- def available_mapsets(self):
- """Return all available mapsets the user can access as a list of strings
- :returns: Names of available mapsets as list of strings
- """
- self.check_server()
- self.client_conn.send(
- [
- RPCDefs.AVAILABLE_MAPSETS,
- ]
- )
- return self.safe_receive("available_mapsets")
- def get_driver_name(self, mapset=None):
- """Return the temporal database driver of a specific mapset
- :param mapset: Name of the mapset
- :returns: Name of the driver or None if no temporal database present
- """
- self.check_server()
- self.client_conn.send([RPCDefs.GET_DRIVER_NAME, mapset])
- return self.safe_receive("get_driver_name")
- def get_database_name(self, mapset=None):
- """Return the temporal database name of a specific mapset
- :param mapset: Name of the mapset
- :returns: Name of the database or None if no temporal database present
- """
- self.check_server()
- self.client_conn.send([RPCDefs.GET_DATABASE_NAME, mapset])
- return self.safe_receive("get_database_name")
- def get_mapset(self):
- """Return the current mapset
- :returns: Name of the current mapset
- """
- self.check_server()
- self.client_conn.send(
- [
- RPCDefs.G_MAPSET,
- ]
- )
- return self.safe_receive("get_mapset")
- def get_location(self):
- """Return the location
- :returns: Name of the location
- """
- self.check_server()
- self.client_conn.send(
- [
- RPCDefs.G_LOCATION,
- ]
- )
- return self.safe_receive("get_location")
- def get_gisdbase(self):
- """Return the gisdatabase
- :returns: Name of the gisdatabase
- """
- self.check_server()
- self.client_conn.send(
- [
- RPCDefs.G_GISDBASE,
- ]
- )
- return self.safe_receive("get_gisdbase")
- def fatal_error(self, mapset=None):
- """Generate a fatal error in libgis.
- This function is only for testing purpose.
- """
- self.check_server()
- self.client_conn.send([RPCDefs.G_FATAL_ERROR])
- # The pipe should be closed in the checker thread
- return self.safe_receive("Fatal error")
- if __name__ == "__main__":
- import doctest
- doctest.testmod()
|