com.github.ansell.shp.SHPUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of shp-debug Show documentation
Debugger for SHP files
There is a newer version: 0.0.8
/*
 * Copyright (c) 2016, Peter Ansell
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see .
 */
package com.github.ansell.shp;

import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.Rectangle;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.StandardCharsets;
import java.nio.file.Path;

import javax.imageio.ImageIO;

import org.geotools.data.shapefile.ShapefileDumper;
import org.geotools.data.simple.SimpleFeatureCollection;
import org.geotools.feature.simple.SimpleFeatureImpl;
import org.geotools.feature.simple.SimpleFeatureTypeImpl;
import org.geotools.geometry.jts.ReferencedEnvelope;
import org.geotools.map.MapContent;
import org.geotools.renderer.GTRenderer;
import org.geotools.renderer.lite.StreamingRenderer;
import org.opengis.feature.simple.SimpleFeature;
import org.opengis.feature.simple.SimpleFeatureType;
import org.opengis.feature.type.Name;

/**
 * Utilities for working with SHP files
 * 
 * @author Peter Ansell [email protected]
 */
public class SHPUtils {

	public static void renderImage(final MapContent map, final OutputStream output, final int imageWidth, String format)
			throws IOException {
		GTRenderer renderer = new StreamingRenderer();
		renderer.setMapContent(map);

		ReferencedEnvelope mapBounds = map.getMaxBounds();
		double heightToWidth = mapBounds.getSpan(1) / mapBounds.getSpan(0);
		Rectangle imageBounds = new Rectangle(0, 0, imageWidth, (int) Math.round(imageWidth * heightToWidth));

		BufferedImage image = new BufferedImage(imageBounds.width, imageBounds.height, BufferedImage.TYPE_INT_RGB);

		Graphics2D gr = image.createGraphics();
		gr.setPaint(Color.WHITE);
		gr.fill(imageBounds);

		renderer.paint(gr, imageBounds, mapBounds);
		ImageIO.write(image, format, output);
	}

	public static void writeShapefile(SimpleFeatureCollection fc, Path outputDir) throws IOException {
		ShapefileDumper dumper = new ShapefileDumper(outputDir.toFile());
		dumper.setCharset(StandardCharsets.UTF_8);
		// split when shp or dbf reaches 1000MB
		int maxSize = 1000 * 1024 * 1024;
		dumper.setMaxDbfSize(maxSize);
		dumper.dump(fc);
		// TODO: Create ZIP file from the contents for easy uploading
	}

	public static SimpleFeatureTypeImpl cloneSchema(SimpleFeatureType schema) {
		return changeSchemaName(schema, schema.getName());
	}

	public static SimpleFeatureTypeImpl changeSchemaName(SimpleFeatureType schema, Name outputSchemaName) {
		return new SimpleFeatureTypeImpl(outputSchemaName, schema.getAttributeDescriptors(),
				schema.getGeometryDescriptor(), schema.isAbstract(), schema.getRestrictions(), schema.getSuper(),
				schema.getDescription());
	}

	public static SimpleFeature changeSchemaName(SimpleFeature feature, SimpleFeatureType outputSchema) {
		return new SimpleFeatureImpl(feature.getAttributes(), outputSchema, feature.getIdentifier());
	}

}