@Test
public void testMappingAndUnmappingOfWays() throws Exception {
// Import
String filename = OSMImportOp.class.getResource("ways.xml").getFile();
File file = new File(filename);
geogit.command(OSMImportOp.class).setDataSource(file.getAbsolutePath()).call();
WorkingTree workTree = geogit.getRepository().getWorkingTree();
long unstaged = workTree.countUnstaged("node").getCount();
assertTrue(unstaged > 0);
unstaged = workTree.countUnstaged("way").getCount();
assertTrue(unstaged > 0);
geogit.command(AddOp.class).call();
geogit.command(CommitOp.class).setMessage("msg").call();
// map
Map<String, AttributeDefinition> fields = Maps.newHashMap();
Map<String, List<String>> mappings = Maps.newHashMap();
mappings.put("highway", Lists.newArrayList("residential"));
fields.put("geom", new AttributeDefinition("geom", FieldType.LINESTRING));
fields.put("name", new AttributeDefinition("name", FieldType.STRING));
MappingRule mappingRule = new MappingRule("residential", mappings, fields);
List<MappingRule> mappingRules = Lists.newArrayList();
mappingRules.add(mappingRule);
Mapping mapping = new Mapping(mappingRules);
geogit.command(OSMMapOp.class).setMapping(mapping).call();
// check that mapping was correctly performed
Optional<RevFeature> revFeature =
geogit
.command(RevObjectParse.class)
.setRefSpec("HEAD:residential/31347480")
.call(RevFeature.class);
assertTrue(revFeature.isPresent());
Optional<RevFeatureType> featureType =
geogit.command(ResolveFeatureType.class).setRefSpec("HEAD:residential/31347480").call();
assertTrue(featureType.isPresent());
ImmutableList<Optional<Object>> values = revFeature.get().getValues();
assertEquals(4, values.size());
// modify a mapped feature. We add a new coordinate to the geometry (0,1) and change the
// value of 'name' tag to "newvalue"
ArrayList<Coordinate> coords =
Lists.newArrayList(((Geometry) values.get(2).get()).getCoordinates());
coords.add(new Coordinate(0, 1));
assertEquals(31347480l, values.get(0).get());
GeometryFactory gf = new GeometryFactory();
SimpleFeatureBuilder fb =
new SimpleFeatureBuilder((SimpleFeatureType) featureType.get().type());
fb.set("geom", gf.createLineString(coords.toArray(new Coordinate[0])));
fb.set("name", "newname");
fb.set("id", 31347480l);
fb.set("nodes", values.get(3).get());
SimpleFeature newFeature = fb.buildFeature("31347480");
geogit.getRepository().getWorkingTree().insert("residential", newFeature);
Optional<RevFeature> mapped =
geogit
.command(RevObjectParse.class)
.setRefSpec("WORK_HEAD:residential/31347480")
.call(RevFeature.class);
assertTrue(mapped.isPresent());
values = mapped.get().getValues();
assertEquals(
"LINESTRING (7.1960069 50.7399033, 7.195868 50.7399081, 7.1950788 50.739912, 7.1949262 50.7399053, "
+ "7.1942463 50.7398686, 7.1935778 50.7398262, 7.1931011 50.7398018, 7.1929987 50.7398009, 7.1925978 50.7397889, "
+ "7.1924199 50.7397781, 0 1)",
values.get(2).get().toString());
assertEquals(31347480l, ((Long) values.get(0).get()).longValue());
assertEquals("newname", values.get(1).get().toString());
// unmap
geogit.command(OSMUnmapOp.class).setPath("residential").call();
// Check that raw OSM data was updated
// First, we check that the corresponding way has been modified
Optional<RevFeature> unmapped =
geogit
.command(RevObjectParse.class)
.setRefSpec("WORK_HEAD:way/31347480")
.call(RevFeature.class);
assertTrue(unmapped.isPresent());
values = unmapped.get().getValues();
assertEquals(
"LINESTRING (7.1960069 50.7399033, 7.195868 50.7399081, 7.1950788 50.739912, 7.1949262 50.7399053, "
+ "7.1942463 50.7398686, 7.1935778 50.7398262, 7.1931011 50.7398018, 7.1929987 50.7398009, 7.1925978 50.7397889, "
+ "7.1924199 50.7397781, 0 1)",
values.get(7).get().toString());
assertEquals("lit:yes|highway:residential|name:newname", values.get(3).get().toString());
// now we get the 'nodes' field in the unmapped feature and check take the id of its last
// node, which refers to the node that we have added to the geometry
int WAY_NODES_FIELD = 6;
String nodes = values.get(WAY_NODES_FIELD).get().toString();
String[] nodeIds = nodes.split(";");
String newNodeId = nodeIds[nodeIds.length - 1];
// and we check that the node has been added to the 'node' tree and has the right
// coordinates.
Optional<RevFeature> newNode =
geogit
.command(RevObjectParse.class)
.setRefSpec("WORK_HEAD:node/" + newNodeId)
.call(RevFeature.class);
assertTrue(newNode.isPresent());
values = newNode.get().getValues();
int NODE_GEOM_FIELD = 6;
assertEquals("POINT (0 1)", values.get(NODE_GEOM_FIELD).get().toString());
}
@Test
public void testMappingAndUnmappingOfNodesWithAlias() throws Exception {
// Import
String filename = OSMImportOp.class.getResource("nodes.xml").getFile();
File file = new File(filename);
geogit.command(OSMImportOp.class).setDataSource(file.getAbsolutePath()).call();
geogit.command(AddOp.class).call();
geogit.command(CommitOp.class).setMessage("msg").call();
// Map
Map<String, AttributeDefinition> fields = Maps.newHashMap();
Map<String, List<String>> mappings = Maps.newHashMap();
mappings.put("highway", Lists.newArrayList("bus_stop"));
fields.put("geom", new AttributeDefinition("geom", FieldType.POINT));
fields.put("name", new AttributeDefinition("name_alias", FieldType.STRING));
MappingRule mappingRule = new MappingRule("busstops", mappings, fields);
List<MappingRule> mappingRules = Lists.newArrayList();
mappingRules.add(mappingRule);
Mapping mapping = new Mapping(mappingRules);
geogit.command(OSMMapOp.class).setMapping(mapping).call();
Optional<RevFeature> revFeature =
geogit
.command(RevObjectParse.class)
.setRefSpec("HEAD:busstops/507464799")
.call(RevFeature.class);
assertTrue(revFeature.isPresent());
Optional<RevFeatureType> featureType =
geogit.command(ResolveFeatureType.class).setRefSpec("HEAD:busstops/507464799").call();
assertTrue(featureType.isPresent());
ImmutableList<Optional<Object>> values = revFeature.get().getValues();
assertEquals(3, values.size());
String wkt = "POINT (7.1959361 50.739397)";
assertEquals(wkt, values.get(2).get().toString());
assertEquals(507464799l, values.get(0).get());
// unmap without having made any changes and check that the canonical folders are not
// modified
geogit.command(OSMUnmapOp.class).setPath("busstops").call();
WorkingTree workTree = geogit.getRepository().getWorkingTree();
long unstaged = workTree.countUnstaged("way").getCount();
assertEquals(0, unstaged);
unstaged = workTree.countUnstaged("node").getCount();
assertEquals(0, unstaged);
// Modify a node
GeometryFactory gf = new GeometryFactory();
SimpleFeatureBuilder fb =
new SimpleFeatureBuilder((SimpleFeatureType) featureType.get().type());
fb.set("geom", gf.createPoint(new Coordinate(0, 1)));
fb.set("name_alias", "newname");
fb.set("id", 507464799l);
SimpleFeature newFeature = fb.buildFeature("507464799");
geogit.getRepository().getWorkingTree().insert("busstops", newFeature);
// check that it was correctly inserted in the working tree
Optional<RevFeature> mapped =
geogit
.command(RevObjectParse.class)
.setRefSpec("WORK_HEAD:busstops/507464799")
.call(RevFeature.class);
assertTrue(mapped.isPresent());
values = mapped.get().getValues();
assertEquals("POINT (0 1)", values.get(2).get().toString());
assertEquals(507464799l, ((Long) values.get(0).get()).longValue());
assertEquals("newname", values.get(1).get().toString());
// unmap
geogit.command(OSMUnmapOp.class).setPath("busstops").call();
unstaged = workTree.countUnstaged("node").getCount();
assertEquals(1, unstaged);
// check that the unmapped node has the changes we introduced
Optional<RevFeature> unmapped =
geogit
.command(RevObjectParse.class)
.setRefSpec("WORK_HEAD:node/507464799")
.call(RevFeature.class);
assertTrue(unmapped.isPresent());
values = unmapped.get().getValues();
assertEquals("POINT (0 1)", values.get(6).get().toString());
assertEquals(
"bus:yes|public_transport:platform|highway:bus_stop|VRS:ortsteil:Hoholz|name:newname|VRS:ref:68566|VRS:gemeinde:BONN",
values.get(3).get().toString());
// check that unchanged nodes keep their attributes
Optional<RevFeature> unchanged =
geogit
.command(RevObjectParse.class)
.setRefSpec("WORK_HEAD:node/1633594723")
.call(RevFeature.class);
values = unchanged.get().getValues();
assertEquals("14220478", values.get(4).get().toString());
assertEquals("1355097351000", values.get(2).get().toString());
assertEquals("2", values.get(1).get().toString());
}
