def showDialog(self):
if self.p1 == None or self.p2 == None or self.dummy == False:
self.p1 = None
self.p2 = None
self.m1 = None
QMessageBox.information(None, "Cancel", "No linesegment selected.")
else:
flags = Qt.WindowTitleHint | Qt.WindowSystemMenuHint | Qt.WindowMaximizeButtonHint # QgisGui.ModalDialogFlags
self.ctrl = ParallelLineGui(self.iface.mainWindow(), flags)
self.ctrl.initGui()
self.ctrl.show()
QObject.connect(self.ctrl, SIGNAL("okClicked(QString, double)"), self.createParallelLine)
QObject.connect( self.ctrl, SIGNAL("btnSelectVertex_clicked()"), self.selectVertex )
QObject.connect( self.ctrl, SIGNAL("unsetTool()"), self.unsetTool )
def showDialog(self):
if self.p1 == None or self.p2 == None or self.dummy == False:
self.p1 = None
self.p2 = None
self.m1 = None
QMessageBox.information(None, QCoreApplication.translate("ctools", "Cancel"), QCoreApplication.translate("ctools", "No linesegment selected."))
else:
flags = Qt.WindowTitleHint | Qt.WindowSystemMenuHint | Qt.WindowMaximizeButtonHint # QgisGui.ModalDialogFlags
self.ctrl = ParallelLineGui(self.iface.mainWindow(), flags)
self.ctrl.initGui()
self.ctrl.show()
self.ctrl.okClicked.connect(self.createParallelLine)
self.ctrl.btnSelectVertex_clicked.connect(self.selectVertex)
self.ctrl.unsetTool.connect(self.unsetTool)
class ParallelLineTool:
def __init__(self, iface, toolBar):
# Save reference to the QGIS interface
self.iface = iface
self.canvas = self.iface.mapCanvas()
# the 2 points of the segment
# p1 is always the left point
self.p1 = None
self.p2 = None
self.m1 = None
self.rb1 = QgsRubberBand(self.canvas, False)
self.pv = None
self.dummy = False
# Create actions
self.action_selectline = QAction(QIcon(":/plugins/cadtools/icons/select1line_v2.png"), "Select one linesegment", self.iface.mainWindow())
self.action_parallelline = QAction(QIcon(":/plugins/cadtools/icons/parallel.png"), "Create parallel line", self.iface.mainWindow())
self.action_selectline.setCheckable(True)
# Connect to signals for button behaviour
QObject.connect(self.action_selectline, SIGNAL("triggered()"), self.selectLineSegment)
QObject.connect(self.action_parallelline, SIGNAL("triggered()"), self.showDialog)
QObject.connect(self.canvas, SIGNAL("mapToolSet(QgsMapTool*)"), self.deactivate)
toolBar.addSeparator()
toolBar.addAction(self.action_selectline)
toolBar.addAction(self.action_parallelline)
def selectLineSegment(self):
mc = self.canvas
layer = mc.currentLayer()
# Set SingleSegmentFinder as current tool
self.tool = SingleSegmentFinderTool(self.canvas)
mc.setMapTool(self.tool)
self.action_selectline.setChecked(True)
# Connect to the SingleSegmentFinderTool
QObject.connect(self.tool, SIGNAL("segmentFound(PyQt_PyObject)"), self.storeSegmentPoints)
def storeSegmentPoints(self, result):
if result[0].x() < result[1].x():
self.p1 = result[0]
self.p2 = result[1]
elif result[0].x() == result[1].x():
self.p1 = result[0]
self.p2 = result[1]
else:
self.p1 = result[1]
self.p2 = result[0]
self.dummy = True
def showDialog(self):
if self.p1 == None or self.p2 == None or self.dummy == False:
self.p1 = None
self.p2 = None
self.m1 = None
QMessageBox.information(None, "Cancel", "No linesegment selected.")
else:
flags = Qt.WindowTitleHint | Qt.WindowSystemMenuHint | Qt.WindowMaximizeButtonHint # QgisGui.ModalDialogFlags
self.ctrl = ParallelLineGui(self.iface.mainWindow(), flags)
self.ctrl.initGui()
self.ctrl.show()
QObject.connect(self.ctrl, SIGNAL("okClicked(QString, double)"), self.createParallelLine)
QObject.connect( self.ctrl, SIGNAL("btnSelectVertex_clicked()"), self.selectVertex )
QObject.connect( self.ctrl, SIGNAL("unsetTool()"), self.unsetTool )
def selectVertex(self):
p1 = QgsPoint()
p2 = QgsPoint()
p1.setX(self.p1.x())
p1.setY(self.p1.y())
p2.setX(self.p2.x())
p2.setY( self.p2.y())
mc = self.canvas
mc.unsetMapTool(self.tool)
self.tool = SingleVertexFinderTool(mc)
mc.setMapTool(self.tool)
QObject.connect(self.tool, SIGNAL("singleVertexFound(PyQt_PyObject)"), self.storeVertexPoint)
self.p1 = p1
self.p2 = p2
self.rb1.reset()
color = QColor(255,0,0)
self.rb1.setColor(color)
self.rb1.setWidth(2)
self.rb1.addPoint(p1)
self.rb1.addPoint(p2)
self.rb1.show()
def storeVertexPoint(self, result):
self.pv = result[0]
def createParallelLine(self, method, distance):
# We need this because adding a layer to the mapcanvas deletes everything....
# Is this true? Why? What happens?
p1 = QgsPoint()
p2 = QgsPoint()
p1.setX(self.p1.x())
p1.setY(self.p1.y())
p2.setX(self.p2.x())
p2.setY( self.p2.y())
print str(method)
print str(distance)
if method == "fixed":
g = ParallelLine.calculateLine(self.p1, self.p2, distance)
cadutils.addGeometryToCadLayer(g)
self.canvas.refresh()
elif method == "vertex":
print "************************888"
points = [self.p1, self.p2]
g = QgsGeometry.fromPolyline(points)
g.translate( self.pv.x() - self.p1.x(), self.pv.y() - self.p1.y() )
# print str(g)
cadutils.addGeometryToCadLayer(g)
self.canvas.refresh()
# del self.m1
self.p1 = p1
self.p2 = p2
def unsetTool(self):
print "***************** unset tool"
mc = self.canvas
mc.unsetMapTool(self.tool)
self.action_selectline.setChecked(False)
def deactivate(self):
print "***************** deactivate parallellinetool"
self.dummy = False
self.rb1.reset()
self.action_selectline.setChecked(False)