def test_get_mapped_color(self):
vmin = 0
vmax = 10
bw_colormap = cmap.generate_hsv_bw_colormap(vmin=vmin, vmax=vmax)
mapped_hsvs = []
expected_hsvs = []
n = 10
for v in range(n):
mapped_hsvs.append(cmap.get_mapped_color(v, bw_colormap))
expected_hsvs.append({'h': 0, 's': 1, 'l': float(v)/n})
self.assertEquals(mapped_hsvs, expected_hsvs)
def test_generate_bw_colormap(self):
bw_colormap = cmap.generate_hsv_bw_colormap()
def get_data_map(project, query=None, data_entity=None, geom_id_entity=None,
geom_entity=None, wms_parameters={}, **kwargs):
""" Get a map image for a given project data map request. """
dao = manage.get_dao(project)
wms_parameters['layers'] = 'data'
if platform.system() == 'Java':
from georefine.util.mapping.gt_renderer import (
GeoToolsMapRenderer, mapSqlAlchemyConnectionParameters)
gt_renderer = GeoToolsMapRenderer()
sa_connection_parameters = dao.get_connection_parameters()
gt_connection_parameters = mapSqlAlchemyConnectionParameters(sa_connection_parameters)
sql = dao.get_sql(query)
# Render map image.
img = gt_renderer.renderMap(
connection_parameters=gt_connection_parameters,
sql=sql,
data_entity=data_entity,
geom_id_entity=geom_id_entity,
geom_entity=geom_entity,
map_parameters=wms_parameters,
**kwargs
)
else:
from georefine.util.mapping.ms_renderer import MapScriptRenderer
renderer = MapScriptRenderer()
# Generate data string from parameters.
# Generate styles from parameters.
driver = dao.connection.engine.url.drivername
if 'postgres' in driver:
connectiontype = 'POSTGIS'
ms_connection_str = "host=%s password=%s dbname=%s user=%s" % (
db.engine.url.host, db.engine.url.password,
db.engine.url.database, db.engine.url.username)
sql = dao.get_sql(query)
ms_data_str = ("geom FROM"
" (SELECT ST_SetSRID(subq.%s, 4326) as geom"
", subq.%s as geom_id, *"
" FROM (%s) as subq) as wrapped_subq"
" USING UNIQUE geom_id USING srid=4326"
% (
geom_entity['ID'],
geom_id_entity['ID'],
sql
)
)
# Spatialite needs some special handling in order to take advantage
# of spatial indices.
elif 'sqlite' in driver:
connectiontype = 'OGR'
ms_connection_str = dao.connection.engine.url.database
#frame_entity = {
#'EXPRESSION': 'func.BuildMbr(%s)' % wms_parameters['BBOX']
#}
#query_obj = dao.get_spatialite_spatial_query(
#query, geom_entity, frame_entity)
query_obj = dao.get_query(query)
sql = dao.query_to_raw_sql(query_obj)
ms_data_str = "SELECT %s AS 'geometry'" % geom_entity['ID']
if data_entity:
ms_data_str += ", %s as 'data'" % data_entity['ID']
ms_data_str += " FROM (%s) AS 'subq'" % sql
ms_data_str += " WHERE ST_Intersects(geometry, BuildMbr(%s))" % (
wms_parameters['BBOX'])
# Create classes for styling if there was a data entity.
if data_entity:
# Generate color bins.
color_bins = cmap.generate_colored_bins(
schema= 'rgb',
vmin=data_entity.get('vmin', 0),
vmax=data_entity.get('vmax', 1),
num_bins=data_entity.get('num_bins', 20),
colormap=data_entity.get(
'colormap', cmap.generate_hsv_bw_colormap()
),
include_bins=data_entity.get('include_bins', []),
include_values=data_entity.get('include_values', []),
)
# Add bottom/top bins.
color_bins.insert(0, ((None, color_bins[0][0][0]), color_bins[0][1]))
color_bins.append(((color_bins[-1][0][1], None), color_bins[-1][1]))
# Create classes from color bins.
classes = []
for color_bin in color_bins:
bin_ = color_bin[0]
cmin = bin_[0]
cmax = bin_[1]
color = color_bin[1]
if cmin is not None and cmax is not None:
cls = {
'expression': "(([data] >= %s) AND ([data] < %s))" % (
cmin, cmax),
'style': {
'color': color
}
}
elif cmin is None and cmax is not None:
cls = {
'expression': "([data] < %s)" % (cmax),
'style': {
'color': color
}
}
elif cmin is not None and cmax is None:
cls = {
'expression': "([data] >= %s)" % (cmin),
'style': {
'color': color
}
}
classes.append(cls)
layers = [{
'name': 'data',
'connection': ms_connection_str,
'connectiontype': connectiontype,
'data': ms_data_str,
'projection': 'init=epsg:4326',
'type': 'POLYGON',
'classes': classes,
}]
imgObj = renderer.render_map(
# use random bg color for transparency
# Otherwise it defaults to white.
imagecolor=(253, 27, 92),
wms_parameters=wms_parameters,
layers=layers,
)
img = renderer.imgObj_to_bytes(imgObj)
return img
