[GRASS-SVN] r47574 - grass-addons/grass7/vector/v.colors2
svn_grass at osgeo.org
svn_grass at osgeo.org
Thu Aug 11 17:26:30 EDT 2011
Author: martinl
Date: 2011-08-11 14:26:30 -0700 (Thu, 11 Aug 2011)
New Revision: 47574
Added:
grass-addons/grass7/vector/v.colors2/v.colors2.html
grass-addons/grass7/vector/v.colors2/v.colors2.py
Removed:
grass-addons/grass7/vector/v.colors2/v.colors.html
grass-addons/grass7/vector/v.colors2/v.colors.py
Modified:
grass-addons/grass7/vector/v.colors2/Makefile
Log:
v.colors renamed to v.colors2 (to avoid conflict with v.colors from trunk)
Modified: grass-addons/grass7/vector/v.colors2/Makefile
===================================================================
--- grass-addons/grass7/vector/v.colors2/Makefile 2011-08-11 21:24:52 UTC (rev 47573)
+++ grass-addons/grass7/vector/v.colors2/Makefile 2011-08-11 21:26:30 UTC (rev 47574)
@@ -1,6 +1,6 @@
MODULE_TOPDIR = ../..
-PGM = v.colors
+PGM = v.colors2
include $(MODULE_TOPDIR)/include/Make/Script.make
Deleted: grass-addons/grass7/vector/v.colors2/v.colors.html
===================================================================
--- grass-addons/grass7/vector/v.colors2/v.colors.html 2011-08-11 21:24:52 UTC (rev 47573)
+++ grass-addons/grass7/vector/v.colors2/v.colors.html 2011-08-11 21:26:30 UTC (rev 47574)
@@ -1,99 +0,0 @@
-<H2>DESCRIPTION</H2>
-
-<EM>v.colors</EM> is much like <em>r.colors</em>, but may be used for vector maps.
-You give it a vector map and numeric data column, together with color rules
-like you would do for a raster. It creates a new column in the database with
-R:G:B values suitable for use with '<tt>d.vect -a</tt>'.
-
-<P>
-How it works: it creates a dummy raster map with the same data range as
-the vector's column then runs <em>r.colors</em> for that temporary map.
-It then uses <em>r.what.colors</em> for each value found by <em>v.db.select</em>
-and uploads it a new column in the vector map's attribute database.
-
-<P>
-It is planned that this script will be replaced with a C display module which
-renders thematic vector maps directly instead requiring the overhead of
-saving the colors into the DB.
-
-<P>
-If the target column name given by the <b>rgb_column</b> option does
-not exist, it will be created. The default name is "<tt>GRASSRGB</tt>".
-
-<h2>EXAMPLES</h2>
-
-Create a random sample point map, query raster map values for those points,
-and colorize output.
-
-<div class="code"><pre>
-# Spearfish dataset
-g.region -d
-v.random out=rand5k_elev n=5000
-v.db.addtable map=rand5k_elev column='elevation double precision'
-v.what.rast vector=rand5k_elev raster=elevation.10m column=elevation
-v.colors map=rand5k_elev column=elevation color=bcyr
-
-# display colorized points map
-d.mon x0
-d.vect -a rand5k_elev
-</pre></div>
-
-<p>
-Colorizing the roads map in Spearfish:
-<div class="code"><pre>
-g.copy vect=roads,myroads
-echo "1 red
-2 yellow
-3 orange
-4 black
-5 grey" > colors.txt
-
-v.colors myroads column=cat rules=colors.txt
-
-# display colorized roads map
-d.mon x0
-d.vect -a myroads
-</pre></div>
-
-
-<P>
-Colorizing a TIN (polygons) generated by <em>v.delaunay</em>:
-
-<div class="code"><pre>
-# new columns for x,y,z of centroids
-v.db.addtable map=tin \
- columns="east double precision, north double precision, height double precision, GRASSRGB varchar(11)"
-
-# transfer geometry for colorizing (we need the centroid height)
-v.to.db tin option=coor columns="east,north,height"
-v.db.select tin
-
-v.colors tin column=height rgb_column=GRASSRGB color=rainbow
-
-# display colorized triangles
-d.mon x0
-d.vect -a tin
-</pre></div>
-
-
-<H2>SEE ALSO</H2>
-
-<EM>
-<A HREF="d.vect.html">d.vect</a> -z<br>
-<A HREF="r.colors.html">r.colors</A><BR>
-<A HREF="r.colors.stddev.html">r.colors.stddev</A><BR>
-<A HREF="r.what.color.html">r.what.color</A><BR>
-<A HREF="v.db.addcolumn">v.db.addcolumn</A><BR>
-<A HREF="v.db.select.html">v.db.select</A><BR>
-<A HREF="db.execute.html">db.execute</A>
-</EM>
-
-
-
-<H2>AUTHOR</H2>
-
-Hamish Bowman<BR>
-<i>Dunedin, New Zealand</i>
-
-<p>
-<i>Last changed: $Date$</i>
Deleted: grass-addons/grass7/vector/v.colors2/v.colors.py
===================================================================
--- grass-addons/grass7/vector/v.colors2/v.colors.py 2011-08-11 21:24:52 UTC (rev 47573)
+++ grass-addons/grass7/vector/v.colors2/v.colors.py 2011-08-11 21:26:30 UTC (rev 47574)
@@ -1,267 +0,0 @@
-#!/usr/bin/env python
-############################################################################
-#
-# MODULE: v.colors
-# AUTHOR: M. Hamish Bowman, Dept. Marine Science, Otago Univeristy,
-# New Zealand
-# Converted to Python by Glynn Clements
-# PURPOSE: Populate a GRASSRGB column with a color map and data column
-# Helper script for thematic mapping tasks
-#
-# COPYRIGHT: (c) 2008, 2010-2011 Hamish Bowman, and 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.
-#
-#############################################################################
-
-#%module
-#% description: Sets color rules for features in a vector map using a numeric attribute column.
-#% keywords: vector
-#% keywords: color table
-#%end
-#% option G_OPT_V_MAP
-#%end
-#%option G_OPT_V_FIELD
-#%end
-#%option G_OPT_DB_COLUMN
-#% description: Name of attribute column containing numeric data
-#% required : yes
-#%end
-#%option G_OPT_DB_COLUMN
-#% key: rgb_column
-#% description: Name of color column to populate with RGB values
-#% answer: GRASSRGB
-#% guisection: Colors
-#%end
-#%option
-#% key: range
-#% type: double
-#% required: no
-#% multiple: no
-#% key_desc: min,max
-#% description: Manually set range (min,max)
-#%end
-#%option
-#% key: color
-#% type: string
-#% key_desc: string
-#% options: aspect,aspectcolr,bcyr,bgyr,byg,byr,celsius,corine,curvature,differences,elevation,etopo2,evi,gdd,grey,grey1.0,grey255,grey.eq,grey.log,gyr,ndvi,population,precipitation,rainbow,ramp,random,ryb,ryg,sepia,slope,srtm,terrain,wave
-#% description: Type of color table
-#% required: no
-#% guisection: Colors
-#%end
-#%option G_OPT_R_INPUT
-#% key: raster
-#% required: no
-#% description: Name of raster map from which to copy color table
-#% guisection: Colors
-#%end
-#%option G_OPT_F_INPUT
-#% key: rules
-#% required: no
-#% description: Name of file containing rules
-#% guisection: Colors
-#%end
-#%flag
-#% key: s
-#% description: Save placeholder raster map for use with d.legend
-#%end
-#%flag
-#% key: n
-#% description: Invert colors
-#% guisection: Colors
-#%end
-
-## TODO: implement -e (equalized) and -g (logarithmic) methods in r.colors
-## 'v.db.select column= | wc -l' to set region size (1xLength)
-## then create r.in.ascii 1xLength matrix with data (WITHOUT uniq)
-## and run r.colors on that raster map.
-## or
-## v.to.rast, r.colors -g, then parse colr/ file. But that's resolution dependent
-
-import sys
-import os
-import atexit
-import string
-import grass.script as grass
-
-def cleanup():
- if tmp:
- grass.try_remove(tmp)
- if tmp_vcol:
- grass.try_remove(tmp_vcol)
- if tmp_colr:
- grass.run_command('g.remove', rast = tmp_colr, quiet = True)
-
-def main():
- color = options['color']
- column = options['column']
- layer = options['layer']
- map = options['map']
- range = options['range']
- raster = options['raster']
- rgb_column = options['rgb_column']
- rules = options['rules']
- flip = flags['n']
-
- global tmp, tmp_colr, tmp_vcol
- pid = os.getpid()
- tmp = tmp_colr = tmp_vcol = None
-
- mapset = grass.gisenv()['MAPSET']
- gisbase = os.getenv('GISBASE')
-
- # does map exist in CURRENT mapset?
- kv = grass.find_file(map, element = 'vector', mapset = mapset)
- if not kv['file']:
- grass.fatal(_("Vector map <%s> not found in current mapset") % map)
-
- vector = map.split('@', 1)
-
- # sanity check mutually exclusive color options
- if not options['color'] and not options['raster'] and not options['rules']:
- grass.fatal(_("Pick one of color, rules, or raster options"))
-
- if color:
- #### check the color rule is valid
- color_opts = os.listdir(os.path.join(gisbase, 'etc', 'colors'))
- color_opts += ['random', 'grey.eq', 'grey.log', 'rules']
- if color not in color_opts:
- grass.fatal(_("Invalid color rule <%s>\n") % color +
- _("Valid options are: %s") % ' '.join(color_opts))
- elif raster:
- if not grass.find_file(raster)['name']:
- grass.fatal(_("Raster raster map <%s> not found") % raster)
- elif rules:
- if not os.access(rules, os.R_OK):
- grass.fatal(_("Unable to read color rules file <%s>") % rules)
-
- # column checks
- # check input data column
- cols = grass.vector_columns(map, layer = layer)
- if column not in cols:
- grass.fatal(_("Column <%s> not found") % column)
- ncolumn_type = cols[column]['type']
- if ncolumn_type not in ["INTEGER", "DOUBLE PRECISION"]:
- grass.fatal(_("Column <%s> is not numeric but %s") % (column, ncolumn_type))
-
- # check if GRASSRGB column exists, make it if it doesn't
- table = grass.vector_db(map)[int(layer)]['table']
- if rgb_column not in cols:
- # RGB Column not found, create it
- grass.message(_("Creating column <%s>...") % rgb_column)
- if 0 != grass.run_command('v.db.addcolumn', map = map, layer = layer, column = "%s varchar(11)" % rgb_column):
- grass.fatal(_("Creating color column"))
- else:
- column_type = cols[rgb_column]['type']
- if column_type not in ["CHARACTER", "TEXT"]:
- grass.fatal(_("Column <%s> is not of compatible type (found %s)") % (rgb_column, column_type))
- else:
- num_chars = dict([(v[0], int(v[2])) for v in grass.db_describe(table)['cols']])[rgb_column]
- if num_chars < 11:
- grass.fatal(_("Color column <%s> is not wide enough (needs 11 characters)"), rgb_column)
-
- cvals = grass.vector_db_select(map, layer = int(layer), columns = column)['values'].values()
-
- # find data range
- if range:
- # order doesn't matter
- vals = range.split(',')
- else:
- grass.message(_("Scanning values..."))
- vals = [float(x[0]) for x in cvals]
-
- minval = min(vals)
- maxval = max(vals)
-
- grass.verbose(_("Range: [%s, %s]") % (minval, maxval))
- if not minval or not maxval:
- grass.fatal(_("Scanning data range"))
-
- # setup internal region
- grass.use_temp_region()
- grass.run_command('g.region', rows = 2, cols = 2)
-
- tmp_colr = "tmp_colr_%d" % pid
-
- # create dummy raster map
- if ncolumn_type == "INTEGER":
- grass.mapcalc("$tmp_colr = int(if(row() == 1, $minval, $maxval))",
- tmp_colr = tmp_colr, minval = minval, maxval = maxval)
- else:
- grass.mapcalc("$tmp_colr = double(if(row() == 1, $minval, $maxval))",
- tmp_colr = tmp_colr, minval = minval, maxval = maxval)
-
- if color:
- color_cmd = {'color': color}
- elif raster:
- color_cmd = {'raster': raster}
- elif rules:
- color_cmd = {'rules': rules}
-
- if flip:
- flip_flag = 'n'
- else:
- flip_flag = ''
-
- grass.run_command('r.colors', map = tmp_colr, flags = flip_flag, quiet = True, **color_cmd)
-
- tmp = grass.tempfile()
-
- # calculate colors and write SQL command file
- grass.message(_("Looking up colors..."))
-
- f = open(tmp, 'w')
- p = grass.feed_command('r.what.color', flags = 'i', input = tmp_colr, stdout = f)
- lastval = None
- for v in sorted(vals):
- if v == lastval:
- continue
- p.stdin.write('%f\n' % v)
- p.stdin.close()
- p.wait()
- f.close()
-
- tmp_vcol = "%s_vcol.sql" % tmp
- fi = open(tmp, 'r')
- fo = open(tmp_vcol, 'w')
- t = string.Template("UPDATE $table SET $rgb_column = '$colr' WHERE $column = $value;\n")
- found = 0
- for line in fi:
- [value, colr] = line.split(': ')
- colr = colr.strip()
- if len(colr.split(':')) != 3:
- continue
- fo.write(t.substitute(table = table, rgb_column = rgb_column, colr = colr, column = column, value = value))
- found += 1
- fi.close()
- fo.close()
-
- if not found:
- grass.fatal(_("No values found in color range"))
-
- # apply SQL commands to update the table with values
- grass.message(_("Writing %s colors...") % found)
-
- if 0 != grass.run_command('db.execute', input = tmp_vcol):
- grass.fatal(_("Processing SQL transaction"))
-
- if flags['s']:
- vcolors = "vcolors_%d" % pid
- grass.run_command('g.rename', rast = (tmp_colr, vcolors), quiet = True)
- grass.message(_("Raster map containing color rules saved to <%s>") % vcolors)
- # TODO save full v.colors command line history
- grass.run_command('r.support', map = vcolors,
- history = "",
- source1 = "vector map = %s" % map,
- source2 = "column = %s" % column,
- title = _("Dummy raster to use as thematic vector legend"),
- description = "generated by v.colors using r.mapcalc")
- grass.run_command('r.support', map = vcolors,
- history = _("RGB saved into <%s> using <%s%s%s>") % (rgb_column, color, raster, rules))
-
-if __name__ == "__main__":
- options, flags = grass.parser()
- atexit.register(cleanup)
- main()
Copied: grass-addons/grass7/vector/v.colors2/v.colors2.html (from rev 47573, grass-addons/grass7/vector/v.colors2/v.colors.html)
===================================================================
--- grass-addons/grass7/vector/v.colors2/v.colors2.html (rev 0)
+++ grass-addons/grass7/vector/v.colors2/v.colors2.html 2011-08-11 21:26:30 UTC (rev 47574)
@@ -0,0 +1,99 @@
+<H2>DESCRIPTION</H2>
+
+<EM>v.colors</EM> is much like <em>r.colors</em>, but may be used for vector maps.
+You give it a vector map and numeric data column, together with color rules
+like you would do for a raster. It creates a new column in the database with
+R:G:B values suitable for use with '<tt>d.vect -a</tt>'.
+
+<P>
+How it works: it creates a dummy raster map with the same data range as
+the vector's column then runs <em>r.colors</em> for that temporary map.
+It then uses <em>r.what.colors</em> for each value found by <em>v.db.select</em>
+and uploads it a new column in the vector map's attribute database.
+
+<P>
+It is planned that this script will be replaced with a C display module which
+renders thematic vector maps directly instead requiring the overhead of
+saving the colors into the DB.
+
+<P>
+If the target column name given by the <b>rgb_column</b> option does
+not exist, it will be created. The default name is "<tt>GRASSRGB</tt>".
+
+<h2>EXAMPLES</h2>
+
+Create a random sample point map, query raster map values for those points,
+and colorize output.
+
+<div class="code"><pre>
+# Spearfish dataset
+g.region -d
+v.random out=rand5k_elev n=5000
+v.db.addtable map=rand5k_elev column='elevation double precision'
+v.what.rast vector=rand5k_elev raster=elevation.10m column=elevation
+v.colors map=rand5k_elev column=elevation color=bcyr
+
+# display colorized points map
+d.mon x0
+d.vect -a rand5k_elev
+</pre></div>
+
+<p>
+Colorizing the roads map in Spearfish:
+<div class="code"><pre>
+g.copy vect=roads,myroads
+echo "1 red
+2 yellow
+3 orange
+4 black
+5 grey" > colors.txt
+
+v.colors myroads column=cat rules=colors.txt
+
+# display colorized roads map
+d.mon x0
+d.vect -a myroads
+</pre></div>
+
+
+<P>
+Colorizing a TIN (polygons) generated by <em>v.delaunay</em>:
+
+<div class="code"><pre>
+# new columns for x,y,z of centroids
+v.db.addtable map=tin \
+ columns="east double precision, north double precision, height double precision, GRASSRGB varchar(11)"
+
+# transfer geometry for colorizing (we need the centroid height)
+v.to.db tin option=coor columns="east,north,height"
+v.db.select tin
+
+v.colors tin column=height rgb_column=GRASSRGB color=rainbow
+
+# display colorized triangles
+d.mon x0
+d.vect -a tin
+</pre></div>
+
+
+<H2>SEE ALSO</H2>
+
+<EM>
+<A HREF="d.vect.html">d.vect</a> -z<br>
+<A HREF="r.colors.html">r.colors</A><BR>
+<A HREF="r.colors.stddev.html">r.colors.stddev</A><BR>
+<A HREF="r.what.color.html">r.what.color</A><BR>
+<A HREF="v.db.addcolumn">v.db.addcolumn</A><BR>
+<A HREF="v.db.select.html">v.db.select</A><BR>
+<A HREF="db.execute.html">db.execute</A>
+</EM>
+
+
+
+<H2>AUTHOR</H2>
+
+Hamish Bowman<BR>
+<i>Dunedin, New Zealand</i>
+
+<p>
+<i>Last changed: $Date$</i>
Copied: grass-addons/grass7/vector/v.colors2/v.colors2.py (from rev 47573, grass-addons/grass7/vector/v.colors2/v.colors.py)
===================================================================
--- grass-addons/grass7/vector/v.colors2/v.colors2.py (rev 0)
+++ grass-addons/grass7/vector/v.colors2/v.colors2.py 2011-08-11 21:26:30 UTC (rev 47574)
@@ -0,0 +1,267 @@
+#!/usr/bin/env python
+############################################################################
+#
+# MODULE: v.colors
+# AUTHOR: M. Hamish Bowman, Dept. Marine Science, Otago Univeristy,
+# New Zealand
+# Converted to Python by Glynn Clements
+# PURPOSE: Populate a GRASSRGB column with a color map and data column
+# Helper script for thematic mapping tasks
+#
+# COPYRIGHT: (c) 2008, 2010-2011 Hamish Bowman, and 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.
+#
+#############################################################################
+
+#%module
+#% description: Sets color rules for features in a vector map using a numeric attribute column.
+#% keywords: vector
+#% keywords: color table
+#%end
+#% option G_OPT_V_MAP
+#%end
+#%option G_OPT_V_FIELD
+#%end
+#%option G_OPT_DB_COLUMN
+#% description: Name of attribute column containing numeric data
+#% required : yes
+#%end
+#%option G_OPT_DB_COLUMN
+#% key: rgb_column
+#% description: Name of color column to populate with RGB values
+#% answer: GRASSRGB
+#% guisection: Colors
+#%end
+#%option
+#% key: range
+#% type: double
+#% required: no
+#% multiple: no
+#% key_desc: min,max
+#% description: Manually set range (min,max)
+#%end
+#%option
+#% key: color
+#% type: string
+#% key_desc: string
+#% options: aspect,aspectcolr,bcyr,bgyr,byg,byr,celsius,corine,curvature,differences,elevation,etopo2,evi,gdd,grey,grey1.0,grey255,grey.eq,grey.log,gyr,ndvi,population,precipitation,rainbow,ramp,random,ryb,ryg,sepia,slope,srtm,terrain,wave
+#% description: Type of color table
+#% required: no
+#% guisection: Colors
+#%end
+#%option G_OPT_R_INPUT
+#% key: raster
+#% required: no
+#% description: Name of raster map from which to copy color table
+#% guisection: Colors
+#%end
+#%option G_OPT_F_INPUT
+#% key: rules
+#% required: no
+#% description: Name of file containing rules
+#% guisection: Colors
+#%end
+#%flag
+#% key: s
+#% description: Save placeholder raster map for use with d.legend
+#%end
+#%flag
+#% key: n
+#% description: Invert colors
+#% guisection: Colors
+#%end
+
+## TODO: implement -e (equalized) and -g (logarithmic) methods in r.colors
+## 'v.db.select column= | wc -l' to set region size (1xLength)
+## then create r.in.ascii 1xLength matrix with data (WITHOUT uniq)
+## and run r.colors on that raster map.
+## or
+## v.to.rast, r.colors -g, then parse colr/ file. But that's resolution dependent
+
+import sys
+import os
+import atexit
+import string
+import grass.script as grass
+
+def cleanup():
+ if tmp:
+ grass.try_remove(tmp)
+ if tmp_vcol:
+ grass.try_remove(tmp_vcol)
+ if tmp_colr:
+ grass.run_command('g.remove', rast = tmp_colr, quiet = True)
+
+def main():
+ color = options['color']
+ column = options['column']
+ layer = options['layer']
+ map = options['map']
+ range = options['range']
+ raster = options['raster']
+ rgb_column = options['rgb_column']
+ rules = options['rules']
+ flip = flags['n']
+
+ global tmp, tmp_colr, tmp_vcol
+ pid = os.getpid()
+ tmp = tmp_colr = tmp_vcol = None
+
+ mapset = grass.gisenv()['MAPSET']
+ gisbase = os.getenv('GISBASE')
+
+ # does map exist in CURRENT mapset?
+ kv = grass.find_file(map, element = 'vector', mapset = mapset)
+ if not kv['file']:
+ grass.fatal(_("Vector map <%s> not found in current mapset") % map)
+
+ vector = map.split('@', 1)
+
+ # sanity check mutually exclusive color options
+ if not options['color'] and not options['raster'] and not options['rules']:
+ grass.fatal(_("Pick one of color, rules, or raster options"))
+
+ if color:
+ #### check the color rule is valid
+ color_opts = os.listdir(os.path.join(gisbase, 'etc', 'colors'))
+ color_opts += ['random', 'grey.eq', 'grey.log', 'rules']
+ if color not in color_opts:
+ grass.fatal(_("Invalid color rule <%s>\n") % color +
+ _("Valid options are: %s") % ' '.join(color_opts))
+ elif raster:
+ if not grass.find_file(raster)['name']:
+ grass.fatal(_("Raster raster map <%s> not found") % raster)
+ elif rules:
+ if not os.access(rules, os.R_OK):
+ grass.fatal(_("Unable to read color rules file <%s>") % rules)
+
+ # column checks
+ # check input data column
+ cols = grass.vector_columns(map, layer = layer)
+ if column not in cols:
+ grass.fatal(_("Column <%s> not found") % column)
+ ncolumn_type = cols[column]['type']
+ if ncolumn_type not in ["INTEGER", "DOUBLE PRECISION"]:
+ grass.fatal(_("Column <%s> is not numeric but %s") % (column, ncolumn_type))
+
+ # check if GRASSRGB column exists, make it if it doesn't
+ table = grass.vector_db(map)[int(layer)]['table']
+ if rgb_column not in cols:
+ # RGB Column not found, create it
+ grass.message(_("Creating column <%s>...") % rgb_column)
+ if 0 != grass.run_command('v.db.addcolumn', map = map, layer = layer, column = "%s varchar(11)" % rgb_column):
+ grass.fatal(_("Creating color column"))
+ else:
+ column_type = cols[rgb_column]['type']
+ if column_type not in ["CHARACTER", "TEXT"]:
+ grass.fatal(_("Column <%s> is not of compatible type (found %s)") % (rgb_column, column_type))
+ else:
+ num_chars = dict([(v[0], int(v[2])) for v in grass.db_describe(table)['cols']])[rgb_column]
+ if num_chars < 11:
+ grass.fatal(_("Color column <%s> is not wide enough (needs 11 characters)"), rgb_column)
+
+ cvals = grass.vector_db_select(map, layer = int(layer), columns = column)['values'].values()
+
+ # find data range
+ if range:
+ # order doesn't matter
+ vals = range.split(',')
+ else:
+ grass.message(_("Scanning values..."))
+ vals = [float(x[0]) for x in cvals]
+
+ minval = min(vals)
+ maxval = max(vals)
+
+ grass.verbose(_("Range: [%s, %s]") % (minval, maxval))
+ if not minval or not maxval:
+ grass.fatal(_("Scanning data range"))
+
+ # setup internal region
+ grass.use_temp_region()
+ grass.run_command('g.region', rows = 2, cols = 2)
+
+ tmp_colr = "tmp_colr_%d" % pid
+
+ # create dummy raster map
+ if ncolumn_type == "INTEGER":
+ grass.mapcalc("$tmp_colr = int(if(row() == 1, $minval, $maxval))",
+ tmp_colr = tmp_colr, minval = minval, maxval = maxval)
+ else:
+ grass.mapcalc("$tmp_colr = double(if(row() == 1, $minval, $maxval))",
+ tmp_colr = tmp_colr, minval = minval, maxval = maxval)
+
+ if color:
+ color_cmd = {'color': color}
+ elif raster:
+ color_cmd = {'raster': raster}
+ elif rules:
+ color_cmd = {'rules': rules}
+
+ if flip:
+ flip_flag = 'n'
+ else:
+ flip_flag = ''
+
+ grass.run_command('r.colors', map = tmp_colr, flags = flip_flag, quiet = True, **color_cmd)
+
+ tmp = grass.tempfile()
+
+ # calculate colors and write SQL command file
+ grass.message(_("Looking up colors..."))
+
+ f = open(tmp, 'w')
+ p = grass.feed_command('r.what.color', flags = 'i', input = tmp_colr, stdout = f)
+ lastval = None
+ for v in sorted(vals):
+ if v == lastval:
+ continue
+ p.stdin.write('%f\n' % v)
+ p.stdin.close()
+ p.wait()
+ f.close()
+
+ tmp_vcol = "%s_vcol.sql" % tmp
+ fi = open(tmp, 'r')
+ fo = open(tmp_vcol, 'w')
+ t = string.Template("UPDATE $table SET $rgb_column = '$colr' WHERE $column = $value;\n")
+ found = 0
+ for line in fi:
+ [value, colr] = line.split(': ')
+ colr = colr.strip()
+ if len(colr.split(':')) != 3:
+ continue
+ fo.write(t.substitute(table = table, rgb_column = rgb_column, colr = colr, column = column, value = value))
+ found += 1
+ fi.close()
+ fo.close()
+
+ if not found:
+ grass.fatal(_("No values found in color range"))
+
+ # apply SQL commands to update the table with values
+ grass.message(_("Writing %s colors...") % found)
+
+ if 0 != grass.run_command('db.execute', input = tmp_vcol):
+ grass.fatal(_("Processing SQL transaction"))
+
+ if flags['s']:
+ vcolors = "vcolors_%d" % pid
+ grass.run_command('g.rename', rast = (tmp_colr, vcolors), quiet = True)
+ grass.message(_("Raster map containing color rules saved to <%s>") % vcolors)
+ # TODO save full v.colors command line history
+ grass.run_command('r.support', map = vcolors,
+ history = "",
+ source1 = "vector map = %s" % map,
+ source2 = "column = %s" % column,
+ title = _("Dummy raster to use as thematic vector legend"),
+ description = "generated by v.colors using r.mapcalc")
+ grass.run_command('r.support', map = vcolors,
+ history = _("RGB saved into <%s> using <%s%s%s>") % (rgb_column, color, raster, rules))
+
+if __name__ == "__main__":
+ options, flags = grass.parser()
+ atexit.register(cleanup)
+ main()
More information about the grass-commit
mailing list