org.geomajas.plugin.rasterizing.step.RasterTileStep Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of geomajas-plugin-rasterizing Show documentation
Geomajas server: Plugin rasterizing - Implementation
There is a newer version: 1.18.5
Show newest version
/*
 * This is part of Geomajas, a GIS framework, http://www.geomajas.org/.
 *
 * Copyright 2008-2015 Geosparc nv, http://www.geosparc.com/, Belgium.
 *
 * The program is available in open source according to the GNU Affero
 * General Public License. All contributions in this program are covered
 * by the Geomajas Contributors License Agreement. For full licensing
 * details, see LICENSE.txt in the project root.
 */

package org.geomajas.plugin.rasterizing.step;

import java.io.ByteArrayOutputStream;

import org.geomajas.configuration.NamedStyleInfo;
import org.geomajas.configuration.client.ClientMapInfo;
import org.geomajas.configuration.client.ClientVectorLayerInfo;
import org.geomajas.global.ExceptionCode;
import org.geomajas.global.GeomajasException;
import org.geomajas.layer.VectorLayer;
import org.geomajas.layer.pipeline.GetTileContainer;
import org.geomajas.layer.tile.TileMetadata;
import org.geomajas.plugin.caching.service.CacheCategory;
import org.geomajas.plugin.rasterizing.api.ImageService;
import org.geomajas.plugin.rasterizing.api.RasterizingContainer;
import org.geomajas.plugin.rasterizing.api.RasterizingPipelineCode;
import org.geomajas.plugin.rasterizing.command.dto.MapRasterizingInfo;
import org.geomajas.plugin.rasterizing.command.dto.VectorLayerRasterizingInfo;
import org.geomajas.service.DtoConverterService;
import org.geomajas.service.TestRecorder;
import org.geomajas.service.pipeline.PipelineCode;
import org.geomajas.service.pipeline.PipelineContext;
import org.geomajas.service.pipeline.PipelineStep;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;

/**
 * Pipeline step which rasterizes vector tiles.
 * 
 * @author Joachim Van der Auwera
 * @author Jan De Moerloose
 */
public class RasterTileStep implements PipelineStep {

	private final Logger log = LoggerFactory.getLogger(RasterTileStep.class);

	@Autowired
	private ImageService imageService;

	@Autowired
	private TestRecorder recorder;

	@Autowired
	private DtoConverterService converterService;

	private String id;

	public String getId() {
		return id;
	}

	public void setId(String id) {
		this.id = id;
	}

	public void execute(PipelineContext context, GetTileContainer tileContainer) throws GeomajasException {
		VectorLayer layer = context.get(PipelineCode.LAYER_KEY, VectorLayer.class);
		TileMetadata tileMetadata = context.get(PipelineCode.TILE_METADATA_KEY, TileMetadata.class);
		// put the image in a raster container
		RasterizingContainer rasterizingContainer = new RasterizingContainer();
		// determine the style
		NamedStyleInfo style = findStyle(layer, tileMetadata);
		// prepare the map configuration
		ClientMapInfo mapInfo = prepareMap(tileContainer, tileMetadata, style);
		byte[] image = renderMap(mapInfo);
		rasterizingContainer.setImage(image);
		context.put(RasterizingPipelineCode.CONTAINER_KEY, rasterizingContainer);
	}

	private byte[] renderMap(ClientMapInfo mapInfo) throws GeomajasException {
		ByteArrayOutputStream imageStream = new ByteArrayOutputStream(1024 * 10);
		try {
			imageService.writeMap(imageStream, mapInfo);
			recorder.record(CacheCategory.RASTER, "Rasterization success");
		} catch (Exception ex) { // NOSONAR
			recorder.record(CacheCategory.RASTER, "Rasterization failed");
			log.error("Problem while rasterizing tile, no image will be returned.", ex);
			throw new GeomajasException(ExceptionCode.LAYER_MODEL_IO_EXCEPTION, ex);
		}

		return imageStream.toByteArray();
	}

	private ClientMapInfo prepareMap(GetTileContainer tileContainer, TileMetadata tileMetadata, NamedStyleInfo style) {
		ClientMapInfo mapInfo = new ClientMapInfo();
		MapRasterizingInfo mapRasterizingInfo = new MapRasterizingInfo();
		mapRasterizingInfo.setBounds(converterService.toDto(tileContainer.getTile().getBounds()));
		mapInfo.setCrs(tileMetadata.getCrs());
		mapRasterizingInfo.setScale(tileMetadata.getScale());
		mapInfo.getWidgetInfo().put(MapRasterizingInfo.WIDGET_KEY, mapRasterizingInfo);
		ClientVectorLayerInfo clientVectorLayerInfo = new ClientVectorLayerInfo();
		clientVectorLayerInfo.setServerLayerId(tileMetadata.getLayerId());
		clientVectorLayerInfo.setNamedStyleInfo(style);
		VectorLayerRasterizingInfo vectorLayerRasterizingInfo = new VectorLayerRasterizingInfo();
		vectorLayerRasterizingInfo.setFilter(tileMetadata.getFilter());
		vectorLayerRasterizingInfo.setPaintGeometries(tileMetadata.isPaintGeometries());
		vectorLayerRasterizingInfo.setPaintLabels(tileMetadata.isPaintLabels());
		vectorLayerRasterizingInfo.setFilter(tileMetadata.getFilter());
		vectorLayerRasterizingInfo.setStyle(style);
		clientVectorLayerInfo.getWidgetInfo().put(VectorLayerRasterizingInfo.WIDGET_KEY, vectorLayerRasterizingInfo);
		mapInfo.getLayers().add(clientVectorLayerInfo);
		return mapInfo;
	}

	private NamedStyleInfo findStyle(VectorLayer layer, TileMetadata tileMetadata) {
		NamedStyleInfo style = tileMetadata.getStyleInfo();
		if (style == null) {
			// no style specified, take the first
			style = layer.getLayerInfo().getNamedStyleInfos().get(0);
		} else if (style.getFeatureStyles().isEmpty()) {
			// only name specified, find it
			style = layer.getLayerInfo().getNamedStyleInfo(style.getName());
		}
		return style;
	}
}