/*
* JGrass - Free Open Source Java GIS http://www.jgrass.org
* (C) HydroloGIS - www.hydrologis.com
*
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* (http://www.eclipse.org/legal/epl-v10.html), and the HydroloGIS BSD
* License v1.0 (http://udig.refractions.net/files/hsd3-v10.html).
*/
package org.locationtech.udig.style.advanced.common.styleattributeclasses;
import static org.locationtech.udig.style.advanced.utils.Utilities.DEFAULT_MAXSCALE;
import static org.locationtech.udig.style.advanced.utils.Utilities.DEFAULT_MINSCALE;
import static org.locationtech.udig.style.advanced.utils.Utilities.sf;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import org.locationtech.udig.style.sld.SLD;
import org.geotools.styling.LineSymbolizer;
import org.geotools.styling.PointSymbolizer;
import org.geotools.styling.PolygonSymbolizer;
import org.geotools.styling.RasterSymbolizer;
import org.geotools.styling.Rule;
import org.geotools.styling.Symbolizer;
import org.geotools.styling.TextSymbolizer;
import org.geotools.xml.Configuration;
import org.geotools.xml.Encoder;
import org.geotools.xml.Parser;
import org.opengis.filter.Filter;
import org.locationtech.udig.style.advanced.utils.Utilities;
/**
* A wrapper for the {@link Rule} object to ease gui use.
*
* @author Andrea Antonello (www.hydrologis.com)
*/
public class RuleWrapper {
private Rule rule;
private String name;
private String maxScale;
private String minScale;
private List<SymbolizerWrapper> symbolizersWrapperList = new ArrayList<SymbolizerWrapper>();
private final FeatureTypeStyleWrapper parent;
public RuleWrapper( Rule rule, FeatureTypeStyleWrapper parent ) {
this.rule = rule;
this.parent = parent;
name = rule.getName();
try {
maxScale = String.valueOf(rule.getMaxScaleDenominator());
minScale = String.valueOf(rule.getMinScaleDenominator());
} catch (Exception e) {
maxScale = DEFAULT_MAXSCALE;
minScale = DEFAULT_MINSCALE;
}
List<Symbolizer> symbolizers = rule.symbolizers();
for( Symbolizer symbolizer : symbolizers ) {
SymbolizerWrapper wrapper = getWrapper(symbolizer);
symbolizersWrapperList.add(wrapper);
}
}
public FeatureTypeStyleWrapper getParent() {
return parent;
}
/**
* getter for the {@link Rule} that the {@link RuleWrapper} wraps.
*
* @return the backed rule.
*/
public Rule getRule() {
return rule;
}
private SymbolizerWrapper getWrapper( Symbolizer symbolizer ) {
SymbolizerWrapper symbolizerWrapper = null;
if (symbolizer instanceof PointSymbolizer) {
symbolizerWrapper = new PointSymbolizerWrapper(symbolizer, this);
} else if (symbolizer instanceof LineSymbolizer) {
symbolizerWrapper = new LineSymbolizerWrapper(symbolizer, this);
} else if (symbolizer instanceof PolygonSymbolizer) {
symbolizerWrapper = new PolygonSymbolizerWrapper(symbolizer, this);
} else if (symbolizer instanceof TextSymbolizer) {
symbolizerWrapper = new TextSymbolizerWrapper(symbolizer, this, getType());
} else if (symbolizer instanceof RasterSymbolizer) {
return null;
}
return symbolizerWrapper;
}
/**
* Returns the type of geometry/raster that the {@link SymbolizerWrapper} represents.
*
* @return the symbolizer type.
*/
public SLD getType() {
SymbolizerWrapper geometrySymbolizersWrapper = getGeometrySymbolizersWrapper();
if (geometrySymbolizersWrapper == null) {
return null;
}
Symbolizer symbolizer = geometrySymbolizersWrapper.getSymbolizer();
if (symbolizer instanceof PointSymbolizer) {
return SLD.POINT;
} else if (symbolizer instanceof LineSymbolizer) {
return SLD.LINE;
} else if (symbolizer instanceof PolygonSymbolizer) {
return SLD.POLYGON;
} else if (symbolizer instanceof RasterSymbolizer) {
return SLD.RASTER;
}
return null;
}
/**
* Getter for the used {@link SymbolizerWrapper}, for point, line or polygon.
*
* <p>Currently only one {@link Symbolizer} is supported in editing, so just the first is used.</p>
*
* @return the used {@link Symbolizer}.
*/
public SymbolizerWrapper getGeometrySymbolizersWrapper() {
for( SymbolizerWrapper symbolizerWrapper : symbolizersWrapperList ) {
if (!symbolizerWrapper.isTextSymbolizer()) {
return symbolizerWrapper;
}
}
DummySymbolizerWrapper geometrySymbolizersWrapper = new DummySymbolizerWrapper(
Utilities.createDefaultGeometrySymbolizer(SLD.POINT), null);
return geometrySymbolizersWrapper;
}
/**
* Getter for the used {@link TextSymbolizerWrapper}.
*
* <p>Currently only one {@link TextSymbolizer} is supported in editing, so just the first is used.</p>
*
* @return the used {@link TextSymbolizer}.
*/
public TextSymbolizerWrapper getTextSymbolizersWrapper() {
for( SymbolizerWrapper symbolizerWrapper : symbolizersWrapperList ) {
if (symbolizerWrapper.isTextSymbolizer()) {
return (TextSymbolizerWrapper) symbolizerWrapper;
}
}
return null;
}
/**
* Remove the {@link TextSymbolizerWrapper} from the ruleWrapper.
*/
public void removeTextSymbolizersWrapper() {
List<SymbolizerWrapper> removeSW = new ArrayList<SymbolizerWrapper>();
List<Symbolizer> removeS = new ArrayList<Symbolizer>();
List<Symbolizer> symbolizers = rule.symbolizers();
for( SymbolizerWrapper symbolizerWrapper : symbolizersWrapperList ) {
if (symbolizerWrapper.isTextSymbolizer()) {
Symbolizer symbolizer = symbolizerWrapper.getSymbolizer();
removeSW.add(symbolizerWrapper);
removeS.add(symbolizer);
}
}
symbolizersWrapperList.removeAll(removeSW);
symbolizers.removeAll(removeS);
}
/**
* Add a supplied or new {@link Symbolizer} to the {@link Rule}.
*
* @param newSymbolizer the new {@link Symbolizer} or null to create a new one.
* @param symbolizerClass the class in the case the symbolizer has to be created.
* @return the {@link SymbolizerWrapper} for the new {@link Symbolizer}.
*/
public <T> T addSymbolizer( Symbolizer newSymbolizer, Class<T> symbolizerClass ) {
SymbolizerWrapper wrapper = null;
if (newSymbolizer != null) {
if (newSymbolizer instanceof PointSymbolizer) {
wrapper = new PointSymbolizerWrapper(newSymbolizer, this);
} else if (newSymbolizer instanceof LineSymbolizer) {
wrapper = new LineSymbolizerWrapper(newSymbolizer, this);
} else if (newSymbolizer instanceof PolygonSymbolizer) {
wrapper = new PolygonSymbolizerWrapper(newSymbolizer, this);
} else if (newSymbolizer instanceof TextSymbolizer) {
wrapper = new TextSymbolizerWrapper(newSymbolizer, this, getType());
} else if (newSymbolizer instanceof RasterSymbolizer) {
// FIXME
return null;
}
} else {
if (symbolizerClass.isAssignableFrom(PointSymbolizerWrapper.class)) {
newSymbolizer = sf.createPointSymbolizer();
wrapper = new PointSymbolizerWrapper(newSymbolizer, this);
} else if (symbolizerClass.isAssignableFrom(LineSymbolizerWrapper.class)) {
newSymbolizer = sf.createLineSymbolizer();
wrapper = new LineSymbolizerWrapper(newSymbolizer, this);
} else if (symbolizerClass.isAssignableFrom(PolygonSymbolizerWrapper.class)) {
newSymbolizer = sf.createPolygonSymbolizer();
wrapper = new PolygonSymbolizerWrapper(newSymbolizer, this);
} else if (symbolizerClass.isAssignableFrom(TextSymbolizerWrapper.class)) {
newSymbolizer = sf.createTextSymbolizer();
wrapper = new TextSymbolizerWrapper(newSymbolizer, this, getType());
} else if (symbolizerClass.isAssignableFrom(RasterSymbolizer.class)) {
// FIXME
return null;
}
}
rule.symbolizers().add(newSymbolizer);
symbolizersWrapperList.add(wrapper);
return symbolizerClass.cast(wrapper);
}
/**
* Clear all the {@link Symbolizer}s and {@link SymbolizerWrapper}s.
*/
public void clear() {
rule.symbolizers().clear();
symbolizersWrapperList.clear();
}
public String getName() {
return name;
}
public void setName( String name ) {
this.name = name;
rule.setName(name);
}
public String getMaxScale() {
return maxScale;
}
public void setMaxScale( String maxScale ) {
this.maxScale = maxScale;
try {
rule.setMaxScaleDenominator(Double.parseDouble(maxScale));
} catch (Exception e) {
rule.setMaxScaleDenominator(Double.POSITIVE_INFINITY);
}
}
public String getMinScale() {
return minScale;
}
public void setMinScale( String minScale ) {
this.minScale = minScale;
try {
rule.setMinScaleDenominator(Double.parseDouble(minScale));
} catch (Exception e) {
rule.setMinScaleDenominator(Double.parseDouble(DEFAULT_MINSCALE));
}
}
// public String getFilter() throws IOException {
// Filter filter = rule.getFilter();
// if (filter == null) {
// return "";
// }
// // create the encoder with the filter 1.1 configuration
// Configuration configuration = new org.geotools.filter.v1_1.OGCConfiguration();
// Encoder encoder = new Encoder(configuration);
// // create an output stream
// ByteArrayOutputStream xml = new ByteArrayOutputStream();
// // encode
// encoder.setIndenting(true);
// encoder.encode(filter, org.geotools.filter.v1_1.OGC.Filter, xml);
// String filterXmlString = xml.toString();
// return filterXmlString;
// }
//
// public void setFilter( String filterXmlString ) throws Exception {
// // create the parser with the filter 1.0 configuration
// Configuration configuration = new org.geotools.filter.v1_1.OGCConfiguration();
// Parser parser = new Parser(configuration);
// InputStream xml = new ByteArrayInputStream(filterXmlString.getBytes());
// // parse
// Filter filter = (Filter) parser.parse(xml);
// rule.setFilter(filter);
// }
}