Source code for pygrass.raster.abstract

# -*- coding: utf-8 -*-
"""
Created on Fri Aug 17 16:05:25 2012

@author: pietro
"""
from __future__ import (nested_scopes, generators, division, absolute_import,
                        with_statement, print_function, unicode_literals)
import ctypes

#
# import GRASS modules
#
from grass.script import fatal, gisenv
import grass.lib.gis as libgis
import grass.lib.raster as libraster

#
# import pygrass modules
#
from grass.pygrass import utils
from grass.pygrass.gis.region import Region
from grass.pygrass.errors import must_be_open
from grass.pygrass.shell.conversion import dict2html
from grass.pygrass.shell.show import raw_figure

#
# import raster classes
#
from grass.pygrass.raster.raster_type import TYPE as RTYPE, RTYPE_STR
from grass.pygrass.raster.category import Category
from grass.pygrass.raster.history import History

test_raster_name = "abstract_test_map"

# Define global variables to not exceed the 80 columns
INDXOUTRANGE = "The index (%d) is out of range, have you open the map?."
INFO = """{name}@{mapset}
rows: {rows}
cols: {cols}
north: {north} south: {south} nsres:{nsres}
east:  {east} west: {west} ewres:{ewres}
range: {min}, {max}
proj: {proj}
"""


[docs]class Info(object): def __init__(self, name, mapset=''): """Read the information for a raster map. :: >>> info = Info(test_raster_name) >>> info.read() >>> info # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE abstract_test_map@ rows: 4 cols: 4 north: 40.0 south: 0.0 nsres:10.0 east: 40.0 west: 0.0 ewres:10.0 range: 11, 44 ... <BLANKLINE> """ self.name = name self.mapset = mapset self.c_region = ctypes.pointer(libraster.struct_Cell_head()) self.c_range = None def _get_range(self): if self.mtype == 'CELL': self.c_range = ctypes.pointer(libraster.Range()) libraster.Rast_read_range(self.name, self.mapset, self.c_range) else: self.c_range = ctypes.pointer(libraster.FPRange()) libraster.Rast_read_fp_range(self.name, self.mapset, self.c_range) def _get_raster_region(self): libraster.Rast_get_cellhd(self.name, self.mapset, self.c_region)
[docs] def read(self): self._get_range() self._get_raster_region()
@property def north(self): return self.c_region.contents.north @property def south(self): return self.c_region.contents.south @property def east(self): return self.c_region.contents.east @property def west(self): return self.c_region.contents.west @property def top(self): return self.c_region.contents.top @property def bottom(self): return self.c_region.contents.bottom @property def rows(self): return self.c_region.contents.rows @property def cols(self): return self.c_region.contents.cols @property def nsres(self): return self.c_region.contents.ns_res @property def ewres(self): return self.c_region.contents.ew_res @property def tbres(self): return self.c_region.contents.tb_res @property def zone(self): return self.c_region.contents.zone @property def proj(self): return self.c_region.contents.proj @property def min(self): if self.c_range is None: return None return self.c_range.contents.min @property def max(self): if self.c_range is None: return None return self.c_range.contents.max @property def range(self): if self.c_range is None: return None, None return self.c_range.contents.min, self.c_range.contents.max @property def mtype(self): return RTYPE_STR[libraster.Rast_map_type(self.name, self.mapset)] def _get_units(self): return libraster.Rast_read_units(self.name, self.mapset) def _set_units(self, units): libraster.Rast_write_units(self.name, units) units = property(_get_units, _set_units) def _get_vdatum(self): return libraster.Rast_read_vdatum(self.name, self.mapset) def _set_vdatum(self, vdatum): libraster.Rast_write_vdatum(self.name, vdatum) vdatum = property(_get_vdatum, _set_vdatum) def __repr__(self): return INFO.format(name=self.name, mapset=self.mapset, rows=self.rows, cols=self.cols, north=self.north, south=self.south, east=self.east, west=self.west, top=self.top, bottom=self.bottom, nsres=self.nsres, ewres=self.ewres, tbres=self.tbres, zone=self.zone, proj=self.proj, min=self.min, max=self.max)
[docs] def keys(self): return ['name', 'mapset', 'rows', 'cols', 'north', 'south', 'east', 'west', 'top', 'bottom', 'nsres', 'ewres', 'tbres', 'zone', 'proj', 'min', 'max']
[docs] def items(self): return [(k, self.__getattribute__(k)) for k in self.keys()]
def __iter__(self): return ((k, self.__getattribute__(k)) for k in self.keys()) def _repr_html_(self): return dict2html(dict(self.items()), keys=self.keys(), border='1', kdec='b')
[docs]class RasterAbstractBase(object): """Raster_abstract_base: The base class from which all sub-classes inherit. It does not implement any row or map access methods: * Implements raster metadata information access (Type, ...) * Implements an open method that will be overwritten by the sub-classes * Implements the close method that might be overwritten by sub-classes (should work for simple row access) * Implements get and set region methods * Implements color, history and category handling * Renaming, deletion, ... """ def __init__(self, name, mapset="", *aopen, **kwopen): """The constructor need at least the name of the map *optional* field is the `mapset`. >>> ele = RasterAbstractBase(test_raster_name) >>> ele.name 'abstract_test_map' >>> ele.exist() True .. """ self.mapset = mapset self._name = name # Private attribute `_fd` that return the file descriptor of the map self._fd = None # Private attribute `_rows` that return the number of rows # in active window, When the class is instanced is empty and it is set # when you open the file, using Rast_window_rows() self._rows = None # Private attribute `_cols` that return the number of rows # in active window, When the class is instanced is empty and it is set # when you open the file, using Rast_window_cols() self._cols = None # self.region = Region() self.hist = History(self.name, self.mapset) self.cats = Category(self.name, self.mapset) self.info = Info(self.name, self.mapset) self._aopen = aopen self._kwopen = kwopen self._mtype = 'CELL' self._mode = 'r' self._overwrite = False def __enter__(self): self.open(*self._aopen, **self._kwopen) return self def __exit__(self, exc_type, exc_value, traceback): self.close() def _get_mtype(self): """Private method to get the Raster type""" return self._mtype def _set_mtype(self, mtype): """Private method to change the Raster type""" if mtype.upper() not in ('CELL', 'FCELL', 'DCELL'): str_err = "Raster type: {0} not supported ('CELL','FCELL','DCELL')" raise ValueError(_(str_err).format(mtype)) self._mtype = mtype self._gtype = RTYPE[self.mtype]['grass type'] mtype = property(fget=_get_mtype, fset=_set_mtype) def _get_mode(self): return self._mode def _set_mode(self, mode): if mode.upper() not in ('R', 'W'): str_err = _("Mode type: {0} not supported ('r', 'w')") raise ValueError(str_err.format(mode)) self._mode = mode mode = property(fget=_get_mode, fset=_set_mode) def _get_overwrite(self): return self._overwrite def _set_overwrite(self, overwrite): if overwrite not in (True, False): str_err = _("Overwrite type: {0} not supported (True/False)") raise ValueError(str_err.format(overwrite)) self._overwrite = overwrite overwrite = property(fget=_get_overwrite, fset=_set_overwrite) def _get_name(self): """Private method to return the Raster name""" return self._name def _set_name(self, newname): """Private method to change the Raster name""" if not utils.is_clean_name(newname): str_err = _("Map name {0} not valid") raise ValueError(str_err.format(newname)) if self.exist(): self.rename(newname) self._name = newname name = property(fget=_get_name, fset=_set_name) @must_be_open def _get_cats_title(self): return self.cats.title @must_be_open def _set_cats_title(self, newtitle): self.cats.title = newtitle cats_title = property(fget=_get_cats_title, fset=_set_cats_title) def __unicode__(self): return self.name_mapset() def __str__(self): """Return the string of the object""" return self.__unicode__() def __len__(self): return self._rows def __getitem__(self, key): """Return the row of Raster object, slice allowed.""" if isinstance(key, slice): # Get the start, stop, and step from the slice return (self.get_row(ii) for ii in range(*key.indices(len(self)))) elif isinstance(key, tuple): x, y = key return self.get(x, y) elif isinstance(key, int): if not self.is_open(): raise IndexError("Can not operate on a closed map. Call open() first.") if key < 0: # Handle negative indices key += self._rows if key >= self._rows: raise IndexError("The row index {0} is out of range [0, {1}).".format(key, self._rows)) return self.get_row(key) else: fatal("Invalid argument type.") def __iter__(self): """Return a constructor of the class""" return (self.__getitem__(irow) for irow in range(self._rows)) def _repr_png_(self): return raw_figure(utils.r_export(self))
[docs] def exist(self): """Return True if the map already exist, and set the mapset if were not set. call the C function `G_find_raster`. >>> ele = RasterAbstractBase(test_raster_name) >>> ele.exist() True """ if self.name: if self.mapset == '': mapset = utils.get_mapset_raster(self.name, self.mapset) self.mapset = mapset if mapset else '' return True if mapset else False return bool(utils.get_mapset_raster(self.name, self.mapset)) else: return False
[docs] def is_open(self): """Return True if the map is open False otherwise. >>> ele = RasterAbstractBase(test_raster_name) >>> ele.is_open() False """ return True if self._fd is not None and self._fd >= 0 else False
[docs] @must_be_open def close(self): """Close the map""" libraster.Rast_close(self._fd) # update rows and cols attributes self._rows = None self._cols = None self._fd = None
[docs] def remove(self): """Remove the map""" if self.is_open(): self.close() utils.remove(self.name, 'rast')
[docs] def fullname(self): """Return the full name of a raster map: name@mapset""" return "{name}@{mapset}".format(name=self.name, mapset=self.mapset)
[docs] def name_mapset(self, name=None, mapset=None): """Return the full name of the Raster. >>> ele = RasterAbstractBase(test_raster_name) >>> name = ele.name_mapset().split("@") >>> name ['abstract_test_map'] """ if name is None: name = self.name if mapset is None: self.exist() mapset = self.mapset gis_env = gisenv() if mapset and mapset != gis_env['MAPSET']: return "{name}@{mapset}".format(name=name, mapset=mapset) else: return name
[docs] def rename(self, newname): """Rename the map""" if self.exist(): utils.rename(self.name, newname, 'rast') self._name = newname
[docs] def set_region_from_rast(self, rastname='', mapset=''): """Set the computational region from a map, if rastername and mapset is not specify, use itself. This region will be used by all raster map layers that are opened in the same process. The GRASS region settings will not be modified. call C function `Rast_get_cellhd`, `Rast_set_window` """ if self.is_open(): fatal("You cannot change the region if map is open") raise region = Region() if rastname == '': rastname = self.name if mapset == '': mapset = self.mapset libraster.Rast_get_cellhd(rastname, mapset, region.byref()) self._set_raster_window(region)
[docs] def set_region(self, region): """Set the computational region that can be different from the current region settings. This region will be used by all raster map layers that are opened in the same process. The GRASS region settings will not be modified. """ if self.is_open(): fatal("You cannot change the region if map is open") raise self._set_raster_window(region)
def _set_raster_window(self, region): libraster.Rast_set_window(region.byref()) # update rows and cols attributes self._rows = libraster.Rast_window_rows() self._cols = libraster.Rast_window_cols()
[docs] @must_be_open def get_value(self, point, region=None): """This method returns the pixel value of a given pair of coordinates: :param point: pair of coordinates in tuple object or class object with coords() method """ # Check for tuple if type(point) != type([]) and type(point) != type(()): point = point.coords() if not region: region = Region() row, col = utils.coor2pixel(point, region) if col < 0 or col > region.cols or row < 0 or row > region.rows: return None line = self.get_row(int(row)) return line[int(col)]
[docs] @must_be_open def has_cats(self): """Return True if the raster map has categories""" if self.exist(): self.cats.read() if len(self.cats) != 0: return True return False
[docs] @must_be_open def num_cats(self): """Return the number of categories""" return len(self.cats)
[docs] @must_be_open def copy_cats(self, raster): """Copy categories from another raster map object""" self.cats.copy(raster.cats)
[docs] @must_be_open def sort_cats(self): """Sort categories order by range""" self.cats.sort()
[docs] @must_be_open def read_cats(self): """Read category from the raster map file""" self.cats.read(self)
[docs] @must_be_open def write_cats(self): """Write category to the raster map file""" self.cats.write(self)
[docs] @must_be_open def read_cats_rules(self, filename, sep=':'): """Read category from the raster map file""" self.cats.read_rules(filename, sep)
[docs] @must_be_open def write_cats_rules(self, filename, sep=':'): """Write category to the raster map file""" self.cats.write_rules(filename, sep)
[docs] @must_be_open def get_cats(self): """Return a category object""" cat = Category(name=self.name, mapset=self.mapset) cat.read() return cat
[docs] @must_be_open def set_cats(self, category): """The internal categories are copied from this object.""" self.cats.copy(category)
[docs] @must_be_open def get_cat(self, label): """Return a category given an index or a label""" return self.cats[label]
[docs] @must_be_open def set_cat(self, label, min_cat, max_cat=None, index=None): """Set or update a category""" self.cats.set_cat(index, (label, min_cat, max_cat))
if __name__ == "__main__": import doctest from grass.pygrass.modules import Module Module("g.region", n=40, s=0, e=40, w=0, res=10) Module("r.mapcalc", expression="%s = row() + (10 * col())" % (test_raster_name), overwrite=True) doctest.testmod() """Remove the generated vector map, if exist""" mset = utils.get_mapset_raster(test_raster_name, mapset='') if mset: Module("g.remove", flags='f', type='raster', name=test_raster_name)