Source code for script.raster3d
"""
Raster3d related functions to be used in Python scripts.
Usage:
::
from grass.script import raster3d as grass
grass.raster3d_info(map)
(C) 2008-2016 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.
.. sectionauthor:: Glynn Clements
.. sectionauthor:: Martin Landa <landa.martin gmail.com>
.. sectionauthor:: Soeren Gebbert <soeren.gebbert gmail.com>
"""
from __future__ import absolute_import
import os
import time
import string
from .core import read_command, write_command, fatal
from .utils import float_or_dms, parse_key_val
from grass.exceptions import CalledModuleError
[docs]def raster3d_info(map, env=None):
"""Return information about a raster3d map (interface to `r3.info`).
Example:
>>> mapcalc3d('volume = row() + col() + depth()')
>>> raster3d_info('volume') # doctest: +ELLIPSIS
{'vertical_units': '"units"', 'tbres': 1.0, ... 'south': 185000.0}
>>> run_command('g.remove', flags='f', type='raster_3d', name='volume')
0
:param str map: map name
:param env: environment
:return: parsed raster3d info
"""
def float_or_null(s):
if s == "NULL":
return None
else:
return float(s)
s = read_command("r3.info", flags="rg", map=map, env=env)
kv = parse_key_val(s)
for k in ["min", "max"]:
kv[k] = float_or_null(kv[k])
for k in ["north", "south", "east", "west", "top", "bottom"]:
kv[k] = float(kv[k])
for k in ["nsres", "ewres", "tbres"]:
kv[k] = float_or_dms(kv[k])
for k in ["rows", "cols", "depths"]:
kv[k] = int(kv[k])
for k in ["tilenumx", "tilenumy", "tilenumz"]:
kv[k] = int(kv[k])
for k in ["tiledimx", "tiledimy", "tiledimz"]:
kv[k] = int(kv[k])
return kv
[docs]def mapcalc3d(
exp, quiet=False, verbose=False, overwrite=False, seed=None, env=None, **kwargs
):
"""Interface to r3.mapcalc.
:param str exp: expression
:param bool quiet: True to run quietly (<tt>--q</tt>)
:param bool verbose: True to run verbosely (<tt>--v</tt>)
:param bool overwrite: True to enable overwriting the output (<tt>--o</tt>)
:param seed: an integer used to seed the random-number generator for the
rand() function, or 'auto' to generate a random seed
:param dict env: dictionary of environment variables for child process
:param kwargs:
"""
if seed == "auto":
seed = hash((os.getpid(), time.time())) % (2**32)
t = string.Template(exp)
e = t.substitute(**kwargs)
try:
write_command(
"r3.mapcalc",
file="-",
stdin=e,
env=env,
seed=seed,
quiet=quiet,
verbose=verbose,
overwrite=overwrite,
)
except CalledModuleError:
fatal(
_("An error occurred while running r3.mapcalc" " with expression: %s") % e
)