def testAppendStreetGrow(self):
map = createMap(self.testdatadir)
map.open(mode="a")
streets, group = map.getLayerAndGroupByName("00_Streets")
streets.open("a")
# Add a new street
newstreet = CellElementPolyline(((16.175, 58.5901), (16.176, 58.5905)), objtype=2)
newbbox = newstreet.bboxrec.union(streets.bboxrec)
newbbox = newbbox.buffer(streets.scale)
streets.bboxrec = newbbox
actual = Set([ce for ce in streets.getCellElements()])
expected = actual.copy()
newstreet.discretizeGeometry(streets.getCell(1))
# streets.addCellElement(newstreet)
# expected.add(newstreet)
actual = Set([ce for ce in streets.getCellElements()])
# Test if the new streets exists in the layer object
self.assertEqual(actual, expected)
# Test if the new street was written to file
streets.close()
del streets
del map
map = createMap(self.testdatadir)
map.open(mode="a")
streets, group = map.getLayerAndGroupByName("00_Streets")
streets.open("r")
# actual = Set([ce.serialize(cell) for ce in streets.getCellElements()])
# expected = Set([ce.serialize(cell) for ce in expected])
actual = Set([ce for ce in streets.getCellElements()])
self.assertEqual(len(actual), len(expected))
if actual != expected:
print "difference:",actual.symmetric_difference(expected)
print "actual: ",actual
print "expected: ",expected
self.assertEqual(actual, expected, "Cell elements don't match after read back")
def testAppendStreet(self):
map = createMap(self.testdatadir)
map.open(mode="a")
streets, group = map.getLayerAndGroupByName("00_Streets")
streets.open("a")
actual = Set([ce for ce in streets.getCellElements()])
expected = actual.copy()
# Add a new street
newstreet = CellElementPolyline(((16.185, 58.5912), (16.186, 58.5915)), objtype=2)
newstreet.discretizeGeometry(streets.getCell(1))
streets.addCellElement(newstreet)
expected.add(newstreet)
actual = Set([ce for ce in streets.getCellElements()])
# Test if the new streets exists in the layer object
self.assertEqual(actual, expected)
# Test if the new street was written to file
streets.close()
del streets
del map
map = createMap(self.testdatadir)
map.open()
streets, group = map.getLayerAndGroupByName("00_Streets")
streets.open("r")
# actual = Set([ce.serialize(cell) for ce in streets.getCellElements()])
# expected = Set([ce.serialize(cell) for ce in expected])
actual = Set([ce for ce in streets.getCellElements()])
expected = Set([ce for ce in expected])
self.assertEqual(len(actual), len(expected))
if actual != expected:
for m in actual:
if m.objtype == 2:
print streets.scale[0]
print m
print newstreet
print m==newstreet
self.assertEqual(actual, expected, "Cell elements don't match after read back")
def testAddGroup(self):
map = createMap(self.testdatadir)
map.open(mode='a')
map.bigendian = True
trailgroup = GroupNormal(map, name="00_Trails")
map.addGroup(trailgroup)
trailgroup.open("w")
# Add trail layer and feature to new group
trails = Layer(map,
name="00_Trails",
filename="00trails",
layertype=LayerTypePolyline)
trails.open(mode='w')
map.addLayer(trails)
trails.setXScale(1e-5)
trails.setYScale(1e-5)
trailgroup.addLayer(trails)
newtrailwkt = "LINESTRING (16.185 58.5912, 16.186 58.5915)"
newtrail = CellElementPolyline(trails, wkt=newtrailwkt)
newtrail.discretizeGeometry(trails.getCell(1))
cellelementrefs = trails.addCellElement(newtrail)
feature = FeatureNormal(name="Apgatan",
layerindex=map.getLayerIndex(trails),
objtype=29,
cellelementreflist=cellelementrefs)
trailgroup.addFeature(feature)
map.writeImage('trails.img')
map.close()
os.system("cat " + os.path.join(self.testdatadir, "00map.ini"))
map = createMap(self.testdatadir)
map.open('r')
trailgroup = map.getGroupByName("00_Trails")
trailgroup.open()
print "Trails", trails
def testAddGroup(self):
map = createMap(self.testdatadir)
map.open(mode='a')
map.bigendian = True
trailgroup = GroupNormal(map, name="00_Trails")
map.addGroup(trailgroup)
trailgroup.open("w")
# Add trail layer and feature to new group
trails = Layer(map, name="00_Trails", filename="00trails", layertype=LayerTypePolyline)
trails.open(mode='w')
map.addLayer(trails)
trails.setXScale(1e-5)
trails.setYScale(1e-5)
trailgroup.addLayer(trails)
newtrailwkt = "LINESTRING (16.185 58.5912, 16.186 58.5915)"
newtrail = CellElementPolyline(trails, wkt=newtrailwkt)
newtrail.discretizeGeometry(trails.getCell(1))
cellelementrefs = trails.addCellElement(newtrail)
feature = FeatureNormal(name="Apgatan", layerindex=map.getLayerIndex(trails),
objtype=29,
cellelementreflist=cellelementrefs)
trailgroup.addFeature(feature)
map.writeImage('trails.img')
map.close()
os.system("cat " +os.path.join(self.testdatadir, "00map.ini"))
map = createMap(self.testdatadir)
map.open('r')
trailgroup = map.getGroupByName("00_Trails")
trailgroup.open()
print "Trails",trails
def testAddPolylineLayer(self):
trails = Layer(self.map, name="Trails", filename="trails", layertype=LayerTypePolyline)
trails.open(mode='w')
self.map.addLayer(trails)
# Add a new trail
newtrail = CellElementPolyline(((16.185, 58.5912), (16.186, 58.5915)), objtype=0)
newtrail.discretizeGeometry(trails.getCell(1))
trails.addCellElement(newtrail)
self.map.close()
map = createMap(self.testdatadir)
map.open()
trails, group = map.getLayerAndGroupByName("Trails")
trails.open('r')
actual = Set(trails.getCellElements())
expected = Set([newtrail])
self.assertSetsEqual(actual, expected)
def testAddFeature(self):
map = createMap(self.testdatadir)
map.open(mode='a')
roads = map.getGroupByName("00_Roads")
roads.open('a')
streets, streetsgroup = map.getLayerAndGroupByName("00_Streets")
nfeatures=roads.getFeatureCount()
self.assertEqual(roads.getFeatureCount(), 29)
newstreetwkt = "LINESTRING (16.185 58.5912, 16.186 58.5915)"
newstreet = CellElementPolyline(streets, wkt=newstreetwkt)
newstreet.discretizeGeometry(streets.getCell(1))
cellelementrefs = streets.addCellElement(newstreet)
feature = FeatureStreet(name="Apgatan",
layerindex=map.getLayerIndex(streets),
objtype=29,
cellelementreflist=cellelementrefs)
roads.addFeature(feature)
# Check that the # of features increased
self.assertEqual(roads.getFeatureCount(), nfeatures+1)
found=False
for i in range(0, roads.getFeatureCount()):
tmpfeature = roads.getFeatureByIndex(i)
if tmpfeature == feature:
print feature
found = True
self.assertTrue(found, "New feature not found")
names = [roads.getFeatureByIndex(i).name for i in range(0, roads.getFeatureCount())]
self.assertTrue(areSorted(names), "Features are not sorted")
print roads.layers
# Re-open map and verify that the new feature is present
map.close()
del map
del streets
map = createMap(self.testdatadir)
map.open()
roads = map.getGroupByName("00_Roads")
print roads.layers
roads.open('r')
found = False
for i in range(0, roads.getFeatureCount()):
tmpfeature = roads.getFeatureByIndex(i)
print tmpfeature
if tmpfeature.name == feature.name:
streets, streetsgroup = map.getLayerAndGroupByName("00_Streets")
if tmpfeature == feature:
found = True
# Check that the # of features increased
self.assertEqual(roads.getFeatureCount(), nfeatures+1)
self.assertTrue(found, "New feature not found after re-opening the map")
names = [roads.getFeatureByIndex(i).name for i in range(0, roads.getFeatureCount())]
self.assertTrue(areSorted(names), "Features are not sorted")
def testAddFeature(self):
map = createMap(self.testdatadir)
map.open(mode='a')
roads = map.getGroupByName("00_Roads")
roads.open('a')
streets, streetsgroup = map.getLayerAndGroupByName("00_Streets")
nfeatures = roads.getFeatureCount()
self.assertEqual(roads.getFeatureCount(), 29)
newstreetwkt = "LINESTRING (16.185 58.5912, 16.186 58.5915)"
newstreet = CellElementPolyline(streets, wkt=newstreetwkt)
newstreet.discretizeGeometry(streets.getCell(1))
cellelementrefs = streets.addCellElement(newstreet)
feature = FeatureStreet(name="Apgatan",
layerindex=map.getLayerIndex(streets),
objtype=29,
cellelementreflist=cellelementrefs)
roads.addFeature(feature)
# Check that the # of features increased
self.assertEqual(roads.getFeatureCount(), nfeatures + 1)
found = False
for i in range(0, roads.getFeatureCount()):
tmpfeature = roads.getFeatureByIndex(i)
if tmpfeature == feature:
print feature
found = True
self.assertTrue(found, "New feature not found")
names = [
roads.getFeatureByIndex(i).name
for i in range(0, roads.getFeatureCount())
]
self.assertTrue(areSorted(names), "Features are not sorted")
print roads.layers
# Re-open map and verify that the new feature is present
map.close()
del map
del streets
map = createMap(self.testdatadir)
map.open()
roads = map.getGroupByName("00_Roads")
print roads.layers
roads.open('r')
found = False
for i in range(0, roads.getFeatureCount()):
tmpfeature = roads.getFeatureByIndex(i)
print tmpfeature
if tmpfeature.name == feature.name:
streets, streetsgroup = map.getLayerAndGroupByName(
"00_Streets")
if tmpfeature == feature:
found = True
# Check that the # of features increased
self.assertEqual(roads.getFeatureCount(), nfeatures + 1)
self.assertTrue(found,
"New feature not found after re-opening the map")
names = [
roads.getFeatureByIndex(i).name
for i in range(0, roads.getFeatureCount())
]
self.assertTrue(areSorted(names), "Features are not sorted")
