def test_compute_paths(self):
self.test_prepare()
try:
self.r.reset()
except Exception as err:
self.fail("Path result resetting failed - {}".format(err.__str__()))
path_computation(origin, dest, self.g, self.r)
if list(self.r.path) != [53, 52, 13]:
self.fail("Path computation failed. Wrong sequence of links")
if list(self.r.path_nodes) != [5, 168, 166, 27]:
self.fail("Path computation failed. Wrong sequence of path nodes")
if list(self.r.milepost) != [0, 341, 1398, 2162]:
self.fail("Path computation failed. Wrong milepost results")
self.r.compute_path(origin, dest)
if list(self.r.path) != [53, 52, 13]:
self.fail("Path computation failed. Wrong sequence of links")
if list(self.r.path_nodes) != [5, 168, 166, 27]:
self.fail("Path computation failed. Wrong sequence of path nodes")
if list(self.r.milepost) != [0, 341, 1398, 2162]:
self.fail("Path computation failed. Wrong milepost results")
def test_compute_paths(self):
path_computation(origin, dest, self.g, self.r)
if list(self.r.path) != [53, 52, 13]:
self.fail("Path computation failed. Wrong sequence of links")
if list(self.r.path_nodes) != [5, 168, 166, 27]:
self.fail("Path computation failed. Wrong sequence of path nodes")
if list(self.r.milepost) != [0, 341, 1398, 2162]:
self.fail("Path computation failed. Wrong milepost results")
self.r.compute_path(origin, dest)
if list(self.r.path) != [53, 52, 13]:
self.fail("Path computation failed. Wrong sequence of links")
if list(self.r.path_nodes) != [5, 168, 166, 27]:
self.fail("Path computation failed. Wrong sequence of path nodes")
if list(self.r.milepost) != [0, 341, 1398, 2162]:
self.fail("Path computation failed. Wrong milepost results")
if list(self.r.path_link_directions) != [-1, -1, -1]:
self.fail("Path computation failed. Wrong link directions")
self.r.compute_path(dest, origin)
if list(self.r.path_link_directions) != [1, 1, 1]:
self.fail("Path computation failed. Wrong link directions")
def test_compute_paths(self):
path_computation(5, 2, self.g, self.r)
self.assertEqual(list(self.r.path), [12, 14], "Path computation failed. Wrong sequence of links")
self.assertEqual(list(self.r.path_link_directions), [1, 1], "Path computation failed. Wrong link directions")
self.assertEqual(list(self.r.path_nodes), [5, 6, 2], "Path computation failed. Wrong sequence of path nodes")
self.assertEqual(list(self.r.milepost), [0, 4, 9], "Path computation failed. Wrong milepost results")
def produces_path(self):
self.to_but.setEnabled(True)
if self.path_from.text().isdigit() and self.path_to.text().isdigit():
self.res.reset()
path_computation(int(self.path_from.text()),
int(self.path_to.text()), self.graph, self.res)
if self.res.path is not None:
## If you want to do selections instead of new layers, this is how to do it
# f = self.cb_link_id_field.currentText()
# t = ''
# for k in self.res.path[:-1]:
# t = t + f + "=" + str(k) + ' or '
# t = t + f + "=" + str(self.res.path[-1])
# expr = QgsExpression(t)
# it = self.line_layer.getFeatures(QgsFeatureRequest(expr))
#
# ids = [i.id() for i in it]
# self.line_layer.setSelectedFeatures(ids)
# If you want to create new layers
# This way is MUCH faster
crs = self.line_layer.dataProvider().crs().authid()
vl = QgsVectorLayer(
"LineString?crs={}".format(crs),
self.path_from.text() + " to " + self.path_to.text(),
"memory")
pr = vl.dataProvider()
# add fields
pr.addAttributes(self.line_layer.dataProvider().fields())
vl.updateFields(
) # tell the vector layer to fetch changes from the provider
# add a feature
all_links = []
for k in self.res.path:
fet = self.link_features[k]
all_links.append(fet)
# add all links to the temp layer
pr.addFeatures(all_links)
# add layer to the map
QgsProject.instance().addMapLayer(vl)
symbol = vl.renderer().symbol()
symbol.setWidth(1)
qgis.utils.iface.mapCanvas().refresh()
else:
qgis.utils.iface.messageBar().pushMessage(
"No path between " + self.path_from.text() + ' and ' +
self.path_to.text(),
'',
level=3)
def test_update_trace(self):
self.test_prepare()
try:
self.r.reset()
except:
self.fail('Path result resetting failed')
path_computation(origin, dest, self.g, self.r)
if list(self.r.path) != [53, 52, 13]:
self.fail('Path computation failed. Wrong sequence of links')
if list(self.r.path_nodes) != [5, 168, 166, 27]:
self.fail('Path computation failed. Wrong sequence of path nodes')
if list(self.r.milepost) != [0, 341, 1398, 2162]:
self.fail('Path computation failed. Wrong milepost results')
def test_update_trace(self):
self.test_prepare()
try:
self.r.reset()
except:
self.fail('Path result resetting failed')
path_computation(origin, dest, self.g, self.r)
if list(self.r.path) != [53, 52, 13]:
self.fail('Path computation failed. Wrong sequence of links')
if list(self.r.path_nodes) != [5, 168, 166, 27]:
self.fail('Path computation failed. Wrong sequence of path nodes')
if list(self.r.milepost) != [0, 341, 1398, 2162]:
self.fail('Path computation failed. Wrong milepost results')
def produces_path(self):
self.to_but.setEnabled(True)
if self.path_from.text().isdigit() and self.path_to.text().isdigit():
self.res.reset()
path_computation(int(self.path_from.text()),
int(self.path_to.text()), self.graph, self.res)
if self.res.path is not None:
# If you want to do selections instead of new layers
if self.rdo_selection.isChecked():
self.create_path_with_selection()
# If you want to create new layers
else:
self.create_path_with_scratch_layer()
else:
qgis.utils.iface.messageBar().pushMessage(
"No path between " + self.path_from.text() + " and " +
self.path_to.text(),
"",
level=3)
def finished(self, procedure):
ns = self.worker_thread.node_sequence
if len(ns) < 2:
return
all_links = []
for i in range(1, len(ns)):
self.res.reset()
path_computation(ns[i - 1], ns[i], self.graph, self.res)
all_links.extend(list(self.res.path))
if self.rdo_new_layer.isChecked():
self.create_path_with_scratch_layer(all_links)
else:
self.create_path_with_selection(all_links)
self.close()
if self.worker_thread.report is not None:
dlg2 = ReportDialog(self.iface, self.worker_thread.report)
dlg2.show()
dlg2.exec_()
def produces_path(self):
self.to_but.setEnabled(True)
if self.path_from.text().isdigit() and self.path_to.text().isdigit():
self.res.reset()
path_computation(int(self.path_from.text()),
int(self.path_to.text()), self.graph, self.res)
if self.res.path is not None:
## If you want to do selections instead of new layers, this is how to do it
# f = self.cb_link_id_field.currentText()
# t = ''
# for k in self.res.path[:-1]:
# t = t + f + "=" + str(k) + ' or '
# t = t + f + "=" + str(self.res.path[-1])
# expr = QgsExpression(t)
# it = self.line_layer.getFeatures(QgsFeatureRequest(expr))
#
# ids = [i.id() for i in it]
# self.line_layer.setSelectedFeatures(ids)
# If you want to create new layers
# This way is MUCH faster
crs = self.line_layer.dataProvider().crs().authid()
vl = QgsVectorLayer(
"LineString?crs={}".format(crs),
self.path_from.text() + " to " + self.path_to.text(),
"memory")
pr = vl.dataProvider()
# add fields
pr.addAttributes(self.line_layer.dataProvider().fields())
vl.updateFields(
) # tell the vector layer to fetch changes from the provider
# add a feature
all_links = []
for k in self.res.path:
fet = self.link_features[k]
all_links.append(fet)
# add all links to the temp layer
pr.addFeatures(all_links)
# add layer to the map
QgsMapLayerRegistry.instance().addMapLayer(vl)
# format the layer with a thick line
registry = QgsSymbolLayerV2Registry.instance()
lineMeta = registry.symbolLayerMetadata("SimpleLine")
symbol = QgsLineSymbolV2()
lineLayer = lineMeta.createSymbolLayer({
'width':
'1',
'color':
self.random_rgb(),
'offset':
'0',
'penstyle':
'solid',
'use_custom_dash':
'0',
'joinstyle':
'bevel',
'capstyle':
'square'
})
symbol.deleteSymbolLayer(0)
symbol.appendSymbolLayer(lineLayer)
renderer = QgsSingleSymbolRendererV2(symbol)
vl.setRendererV2(renderer)
qgis.utils.iface.mapCanvas().refresh()
else:
qgis.utils.iface.messageBar().pushMessage(
"No path between " + self.path_from.text() + ' and ' +
self.path_to.text(),
'',
level=3)