/*
* Geotoolkit - An Open Source Java GIS Toolkit
* http://www.geotoolkit.org
*
* (C) 2010, Geomatys
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library 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
* Lesser General Public License for more details.
*/
package org.geotoolkit.data.kml;
import com.vividsolutions.jts.geom.Coordinate;
import com.vividsolutions.jts.geom.CoordinateSequence;
import java.awt.Color;
import java.io.File;
import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.Iterator;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.stream.XMLStreamException;
import org.geotoolkit.data.kml.model.BasicLink;
import org.geotoolkit.data.kml.model.IconStyle;
import org.geotoolkit.data.kml.model.Kml;
import org.geotoolkit.data.kml.model.KmlException;
import org.geotoolkit.data.kml.model.KmlModelConstants;
import org.geotoolkit.data.kml.model.LabelStyle;
import org.geotoolkit.data.kml.model.Pair;
import org.geotoolkit.data.kml.model.Point;
import org.geotoolkit.data.kml.model.Style;
import org.geotoolkit.data.kml.model.StyleMap;
import org.geotoolkit.data.kml.model.StyleState;
import org.geotoolkit.data.kml.xml.KmlReader;
import org.junit.Test;
import org.opengis.feature.Feature;
import org.geotoolkit.data.kml.xml.KmlConstants;
import org.xml.sax.SAXException;
import static org.junit.Assert.*;
/**
*
* @author Samuel Andrés
* @module
*/
public class StyleMapTest extends org.geotoolkit.test.TestBase {
private static final double DELTA = 0.000000000001;
private static final String pathToTestFile = "src/test/resources/org/geotoolkit/data/kml/styleMap.kml";
@Test
public void styleMapReadTest() throws IOException, XMLStreamException, URISyntaxException, KmlException {
final Feature document;
{
final KmlReader reader = new KmlReader();
reader.setInput(new File(pathToTestFile));
final Kml kmlObjects = reader.read();
reader.dispose();
document = kmlObjects.getAbstractFeature();
}
assertEquals(KmlModelConstants.TYPE_DOCUMENT, document.getType());
assertEquals("StyleMap.kml", document.getPropertyValue(KmlConstants.TAG_NAME));
assertEquals(Boolean.TRUE, document.getPropertyValue(KmlConstants.TAG_OPEN));
Iterator<?> i = ((Iterable<?>) document.getPropertyValue(KmlConstants.TAG_STYLE_SELECTOR)).iterator();
assertTrue("Expected at least one element.", i.hasNext());
{
Style style = (Style) i.next();
assertEquals("normalState", style.getIdAttributes().getId());
IconStyle iconStyle0 = style.getIconStyle();
BasicLink icon0 = iconStyle0.getIcon();
assertEquals("http://maps.google.com/mapfiles/kml/pal3/icon55.png", icon0.getHref());
}
assertTrue("Expected at least 2 elements.", i.hasNext());
{
Style style = (Style) i.next();
assertEquals("highlightState", style.getIdAttributes().getId());
IconStyle iconStyle1 = style.getIconStyle();
assertEquals(1.1, iconStyle1.getScale(), DELTA);
BasicLink icon1 = iconStyle1.getIcon();
assertEquals("http://maps.google.com/mapfiles/kml/pal3/icon60.png", icon1.getHref());
LabelStyle labelStyle1 = style.getLabelStyle();
assertEquals(new Color(192, 0, 0, 255), labelStyle1.getColor());
assertEquals(1.1, labelStyle1.getScale(), DELTA);
}
assertTrue("Expected at least 3 elements.", i.hasNext());
{
StyleMap styleMap = (StyleMap) i.next();
assertEquals("styleMapExample", styleMap.getIdAttributes().getId());
assertEquals(2, styleMap.getPairs().size());
Pair pair0 = styleMap.getPairs().get(0);
assertEquals(new URI("#normalState"), pair0.getStyleUrl());
Pair pair1 = styleMap.getPairs().get(1);
assertEquals(StyleState.HIGHLIGHT, pair1.getKey());
assertEquals(new URI("#highlightState"), pair1.getStyleUrl());
}
assertFalse("Expected exactly 3 elements.", i.hasNext());
i = ((Iterable<?>) document.getPropertyValue(KmlConstants.TAG_FEATURES)).iterator();
assertTrue("Expected at least one element.", i.hasNext());
{
Feature placemark = (Feature) i.next();
assertEquals(KmlModelConstants.TYPE_PLACEMARK, placemark.getType());
assertEquals("StyleMap example", placemark.getPropertyValue(KmlConstants.TAG_NAME));
assertEquals(new URI("#styleMapExample"), placemark.getPropertyValue(KmlConstants.TAG_STYLE_URL));
Point point = (Point) placemark.getPropertyValue(KmlConstants.TAG_GEOMETRY);
CoordinateSequence coordinates = point.getCoordinateSequence();
assertEquals(1, coordinates.size());
Coordinate coordinate = coordinates.getCoordinate(0);
assertEquals(-122.368987, coordinate.x, DELTA);
assertEquals(37.817634, coordinate.y, DELTA);
assertEquals(0, coordinate.z, DELTA);
}
assertFalse("Expected exactly one element.", i.hasNext());
}
@Test
public void styleMapWriteTest() throws KmlException, IOException, XMLStreamException, ParserConfigurationException, SAXException, URISyntaxException {
// final KmlFactory kmlFactory = new DefaultKmlFactory();
//
// Coordinate coordinate = kmlFactory.createCoordinate(-122.368987, 37.817634, 0);
// Coordinates coordinates = kmlFactory.createCoordinates(Arrays.asList(coordinate));
// Point point = kmlFactory.createPoint(coordinates);
//
// Placemark placemark = kmlFactory.createPlacemark();
// placemark.setFeatureName("StyleMap example");
// placemark.setStyleUrl(new URI("#styleMapExample"));
// placemark.setAbstractGeometry(point);
//
// Style style0 = kmlFactory.createStyle();
// IconStyle iconStyle0 = kmlFactory.createIconStyle();
// BasicLink icon0 = kmlFactory.createBasicLink();
// icon0.setHref("http://maps.google.com/mapfiles/kml/pal3/icon55.png");
// iconStyle0.setIcon(icon0);
//
// LabelStyle labelStyle0 = kmlFactory.createLabelStyle();
// style0.setIconStyle(iconStyle0);
// style0.setLabelStyle(labelStyle0);
// IdAttributes idAttributes0 = kmlFactory.createIdAttributes("normalState", null);
// style0.setIdAttributes(idAttributes0);
//
// Style style1 = kmlFactory.createStyle();
// IconStyle iconStyle1 = kmlFactory.createIconStyle();
// iconStyle1.setScale(1.1);
// BasicLink icon1 = kmlFactory.createBasicLink();
// icon1.setHref("http://maps.google.com/mapfiles/kml/pal3/icon60.png");
// iconStyle1.setIcon(icon1);
//
// LabelStyle labelStyle1 = kmlFactory.createLabelStyle();
// labelStyle1.setColor(new Color(192, 0, 0, 255));
// labelStyle1.setScale(1.1);
// style1.setIconStyle(iconStyle1);
// style1.setLabelStyle(labelStyle1);
// IdAttributes idAttributes1 = kmlFactory.createIdAttributes("highlightState", null);
// style1.setIdAttributes(idAttributes1);
//
// StyleMap styleMap = kmlFactory.createStyleMap();
// Pair pair0 = kmlFactory.createPair();
// pair0.setStyleUrl(new URI("#normalState"));
//
// Pair pair1 = kmlFactory.createPair();
// pair1.setKey(StyleState.HIGHLIGHT);
// pair1.setStyleUrl(new URI("#highlightState"));
// styleMap.setPairs(Arrays.asList(pair0, pair1));
// IdAttributes idAttributes2 = kmlFactory.createIdAttributes("styleMapExample", null);
// styleMap.setIdAttributes(idAttributes2);
//
// Document document = kmlFactory.createDocument();
// document.setStyleSelectors(Arrays.asList(
// (AbstractStyleSelector) style0,
// (AbstractStyleSelector) style1,
// (AbstractStyleSelector) styleMap));
// document.setAbstractFeatures(Arrays.asList((AbstractFeature) placemark));
// document.setFeatureName("StyleMap.kml");
// document.setOpen(true);
//
// final Kml kml = kmlFactory.createKml(null, document, null, null);
//
// File temp = File.createTempFile("testStyleMap", ".kml");
// temp.deleteOnExit();
//
// KmlWriter writer = new KmlWriter();
// writer.setOutput(temp);
// writer.write(kml);
// writer.dispose();
//
// DomCompare.compare(new File(pathToTestFile), temp);
}
}