def test_setStartPoint_2(self):
args = {'geometry': 'test_geom', 'table': 'test_table'}
geomType = 'ST_MultiLineString'
utils.setStartPoint(geomType, args)
self.assertEqual(args['startpoint'],
'ST_StartPoint(ST_GeometryN(test_geom, 1))')
def findNearestNode(self, args, pt):
distance = self.iface.mapCanvas().getCoordinateTransform(
).mapUnitsPerPixel() * self.FIND_RADIUS
rect = QgsRectangle(pt.x() - distance,
pt.y() - distance,
pt.x() + distance,
pt.y() + distance)
canvasCrs = Utils.getDestinationCrs(
self.iface.mapCanvas().mapRenderer())
db = None
try:
dados = str(self.dock.comboConnections.currentText())
db = self.actionsDb[dados].connect()
con = db.con
srid, geomType = self.getSridAndGeomType(con, args)
if self.iface.mapCanvas().hasCrsTransformEnabled():
layerCrs = QgsCoordinateReferenceSystem()
Utils.createFromSrid(layerCrs, srid)
trans = QgsCoordinateTransform(canvasCrs, layerCrs)
pt = trans.transform(pt)
rect = trans.transform(rect)
args['canvas_srid'] = Utils.getCanvasSrid(canvasCrs)
args['srid'] = srid
args['x'] = pt.x()
args['y'] = pt.y()
args['minx'] = rect.xMinimum()
args['miny'] = rect.yMinimum()
args['maxx'] = rect.xMaximum()
args['maxy'] = rect.yMaximum()
Utils.setStartPoint(geomType, args)
Utils.setEndPoint(geomType, args)
Utils.setTransformQuotes(args)
# Getting nearest source
query1 = """
SELECT %(source)s,
ST_Distance(
%(startpoint)s,
ST_GeomFromText('POINT(%(x)f %(y)f)', %(srid)d)
) AS dist,
ST_AsText(%(transform_s)s%(startpoint)s%(transform_e)s)
FROM %(edge_table)s
WHERE ST_SetSRID('BOX3D(%(minx)f %(miny)f, %(maxx)f %(maxy)f)'::BOX3D, %(srid)d)
&& %(geometry)s ORDER BY dist ASC LIMIT 1""" % args
##QMessageBox.information(self.dock, self.dock.windowTitle(), query1)
cur1 = con.cursor()
cur1.execute(query1)
row1 = cur1.fetchone()
d1 = None
source = None
wkt1 = None
if row1:
d1 = row1[1]
source = row1[0]
wkt1 = row1[2]
# Getting nearest target
query2 = """
SELECT %(target)s,
ST_Distance(
%(endpoint)s,
ST_GeomFromText('POINT(%(x)f %(y)f)', %(srid)d)
) AS dist,
ST_AsText(%(transform_s)s%(endpoint)s%(transform_e)s)
FROM %(edge_table)s
WHERE ST_SetSRID('BOX3D(%(minx)f %(miny)f, %(maxx)f %(maxy)f)'::BOX3D, %(srid)d)
&& %(geometry)s ORDER BY dist ASC LIMIT 1""" % args
##QMessageBox.information(self.dock, self.dock.windowTitle(), query2)
cur2 = con.cursor()
cur2.execute(query2)
row2 = cur2.fetchone()
d2 = None
target = None
wkt2 = None
if row2:
d2 = row2[1]
target = row2[0]
wkt2 = row2[2]
# Checking what is nearer - source or target
d = None
node = None
wkt = None
if d1 and (not d2):
node = source
d = d1
wkt = wkt1
elif (not d1) and d2:
node = target
d = d2
wkt = wkt2
elif d1 and d2:
if d1 < d2:
node = source
d = d1
wkt = wkt1
else:
node = target
d = d2
wkt = wkt2
##QMessageBox.information(self.dock, self.dock.windowTitle(), str(d))
if (d == None) or (d > distance):
node = None
wkt = None
return False, None, None
return True, node, wkt
except psycopg2.DatabaseError, e:
QApplication.restoreOverrideCursor()
QMessageBox.critical(self.dock, self.dock.windowTitle(), '%s' % e)
return False, None, None
def test_setStartPoint_2(self):
args = { 'geometry': 'test_geom','table' : 'test_table'}
geomType = 'ST_MultiLineString'
utils.setStartPoint(geomType,args)
self.assertEqual(args['startpoint'], 'ST_StartPoint(ST_GeometryN(test_geom, 1))')
def findNearestNode(self, args, pt):
distance = self.iface.mapCanvas().getCoordinateTransform().mapUnitsPerPixel() * self.FIND_RADIUS
rect = QgsRectangle(pt.x() - distance, pt.y() - distance, pt.x() + distance, pt.y() + distance)
canvasCrs = Utils.getDestinationCrs(self.iface.mapCanvas().mapRenderer())
db = None
try:
dados = str(self.dock.comboConnections.currentText())
db = self.actionsDb[dados].connect()
con = db.con
srid, geomType = self.getSridAndGeomType(con, args)
if self.iface.mapCanvas().hasCrsTransformEnabled():
layerCrs = QgsCoordinateReferenceSystem()
Utils.createFromSrid(layerCrs, srid)
trans = QgsCoordinateTransform(canvasCrs, layerCrs)
pt = trans.transform(pt)
rect = trans.transform(rect)
args["canvas_srid"] = Utils.getCanvasSrid(canvasCrs)
args["srid"] = srid
args["x"] = pt.x()
args["y"] = pt.y()
args["minx"] = rect.xMinimum()
args["miny"] = rect.yMinimum()
args["maxx"] = rect.xMaximum()
args["maxy"] = rect.yMaximum()
Utils.setStartPoint(geomType, args)
Utils.setEndPoint(geomType, args)
Utils.setTransformQuotes(args)
# Getting nearest source
query1 = (
"""
SELECT %(source)s,
ST_Distance(
%(startpoint)s,
ST_GeomFromText('POINT(%(x)f %(y)f)', %(srid)d)
) AS dist,
ST_AsText(%(transform_s)s%(startpoint)s%(transform_e)s)
FROM %(edge_table)s
WHERE ST_SetSRID('BOX3D(%(minx)f %(miny)f, %(maxx)f %(maxy)f)'::BOX3D, %(srid)d)
&& %(geometry)s ORDER BY dist ASC LIMIT 1"""
% args
)
##QMessageBox.information(self.dock, self.dock.windowTitle(), query1)
cur1 = con.cursor()
cur1.execute(query1)
row1 = cur1.fetchone()
d1 = None
source = None
wkt1 = None
if row1:
d1 = row1[1]
source = row1[0]
wkt1 = row1[2]
# Getting nearest target
query2 = (
"""
SELECT %(target)s,
ST_Distance(
%(endpoint)s,
ST_GeomFromText('POINT(%(x)f %(y)f)', %(srid)d)
) AS dist,
ST_AsText(%(transform_s)s%(endpoint)s%(transform_e)s)
FROM %(edge_table)s
WHERE ST_SetSRID('BOX3D(%(minx)f %(miny)f, %(maxx)f %(maxy)f)'::BOX3D, %(srid)d)
&& %(geometry)s ORDER BY dist ASC LIMIT 1"""
% args
)
##QMessageBox.information(self.dock, self.dock.windowTitle(), query2)
cur2 = con.cursor()
cur2.execute(query2)
row2 = cur2.fetchone()
d2 = None
target = None
wkt2 = None
if row2:
d2 = row2[1]
target = row2[0]
wkt2 = row2[2]
# Checking what is nearer - source or target
d = None
node = None
wkt = None
if d1 and (not d2):
node = source
d = d1
wkt = wkt1
elif (not d1) and d2:
node = target
d = d2
wkt = wkt2
elif d1 and d2:
if d1 < d2:
node = source
d = d1
wkt = wkt1
else:
node = target
d = d2
wkt = wkt2
##QMessageBox.information(self.dock, self.dock.windowTitle(), str(d))
if (d == None) or (d > distance):
node = None
wkt = None
return False, None, None
return True, node, wkt
except psycopg2.DatabaseError, e:
QApplication.restoreOverrideCursor()
QMessageBox.critical(self.dock, self.dock.windowTitle(), "%s" % e)
return False, None, None