/*******************************************************************************
* Copyright (c) 2015 Voyager Search and MITRE
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Apache License, Version 2.0 which
* accompanies this distribution and is available at
* http://www.apache.org/licenses/LICENSE-2.0.txt
******************************************************************************/
package org.locationtech.spatial4j.io;
import org.locationtech.spatial4j.context.SpatialContext;
import org.locationtech.spatial4j.context.SpatialContextFactory;
import org.locationtech.spatial4j.shape.Shape;
import org.locationtech.spatial4j.shape.impl.PointImpl;
import org.junit.Test;
import java.text.ParseException;
import static org.junit.Assert.assertEquals;
public class WktCustomShapeParserTest extends WktShapeParserTest {
static class CustomShape extends PointImpl {
private final String name;
/**
* A simple constructor without normalization / validation.
*/
public CustomShape(String name, SpatialContext ctx) {
super(0, 0, ctx);
this.name = name;
}
}
public WktCustomShapeParserTest() {
super(makeCtx());
}
private static SpatialContext makeCtx() {
SpatialContextFactory factory = new SpatialContextFactory();
factory.readers.clear();
factory.readers.add( MyWKTShapeParser.class );
return factory.newSpatialContext();
}
@Test
public void testCustomShape() throws ParseException {
assertEquals("customShape", ((CustomShape) wkt("customShape()")).name);
assertEquals("custom3d", ((CustomShape) wkt("custom3d ()")).name);//number supported
}
@Test
public void testNextSubShapeString() throws ParseException {
WKTReader.State state = ((WKTReader)ctx.getFormats().getWktReader()).newState("OUTER(INNER(3, 5))");
state.offset = 0;
assertEquals("OUTER(INNER(3, 5))", state.nextSubShapeString());
assertEquals("OUTER(INNER(3, 5))".length(), state.offset);
state.offset = "OUTER(".length();
assertEquals("INNER(3, 5)", state.nextSubShapeString());
assertEquals("OUTER(INNER(3, 5)".length(), state.offset);
state.offset = "OUTER(INNER(".length();
assertEquals("3", state.nextSubShapeString());
assertEquals("OUTER(INNER(3".length(), state.offset);
}
public static class MyWKTShapeParser extends WKTReader {
public MyWKTShapeParser(SpatialContext ctx, SpatialContextFactory factory) {
super(ctx, factory);
}
@Override
protected State newState(String wkt) {
//First few lines compile, despite newState() being protected. Just proving extensibility.
WKTReader other = null;
if (false)
other.newState(wkt);
return new State(wkt);
}
@Override
public Shape parseShapeByType(State state, String shapeType) throws ParseException {
Shape result = super.parseShapeByType(state, shapeType);
if (result == null && shapeType.contains("custom")) {
state.nextExpect('(');
state.nextExpect(')');
return new CustomShape(shapeType, ctx);
}
return result;
}
}
}