12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190 |
- """
- @package vdigit.mapwindow
- @brief Map display canvas for wxGUI vector digitizer
- Classes:
- - mapwindow::VDigitWindow
- (C) 2011-2013 by the GRASS Development Team
- This program is free software under the GNU General Public License
- (>=v2). Read the file COPYING that comes with GRASS for details.
- @author Martin Landa <landa.martin gmail.com>
- """
- import wx
- import tempfile
- from grass.pydispatch.signal import Signal
- from dbmgr.dialogs import DisplayAttributesDialog
- from core.gcmd import RunCommand, GMessage, GError
- from core.debug import Debug
- from mapwin.buffered import BufferedMapWindow
- from core.settings import UserSettings
- from core.utils import ListOfCatsToRange, _
- from core.globalvar import QUERYLAYER
- from vdigit.dialogs import VDigitCategoryDialog, VDigitZBulkDialog, VDigitDuplicatesDialog
- from gui_core import gselect
- from gui_core.wrap import PseudoDC
- class VDigitWindow(BufferedMapWindow):
- """A Buffered window extended for vector digitizer.
- """
- def __init__(self, parent, giface, Map, properties, tree=None,
- id=wx.ID_ANY, lmgr=None,
- style=wx.NO_FULL_REPAINT_ON_RESIZE, **kwargs):
- BufferedMapWindow.__init__(self, parent=parent, giface=giface, Map=Map,
- properties=properties,
- style=style, **kwargs)
- self.lmgr = lmgr
- self.tree = tree
- self.pdcVector = PseudoDC()
- self.toolbar = self.parent.GetToolbar('vdigit')
- self.digit = None # wxvdigit.IVDigit
- self._digitizingInfo = False # digitizing with info
- # Emitted when info about digitizing updated
- # Parameter text is a string with information
- # currently used only for coordinates of mouse cursor + segmnt and
- # total feature length
- self.digitizingInfo = Signal('VDigitWindow.digitizingInfo')
- # Emitted when some info about digitizing is or will be availbale
- self.digitizingInfoAvailable = Signal('VDigitWindow.digitizingInfo')
- # Emitted when some info about digitizing is or will be availbale
- # digitizingInfo signal is emmited only between digitizingInfoAvailable
- # and digitizingInfoUnavailable signals
- self.digitizingInfoUnavailable = Signal('VDigitWindow.digitizingInfo')
- self.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown)
- self.mouseMoving.connect(self._mouseMovingToDigitizingInfo)
- def GetDisplay(self):
- if self.digit:
- return self.digit.GetDisplay()
- return None
- def GetDigit(self):
- """Get digit class"""
- return self.digit
- def SetToolbar(self, toolbar):
- """Set up related toolbar
- """
- self.toolbar = toolbar
- def _mouseMovingToDigitizingInfo(self, x, y):
- e, n = x, y
- precision = int(UserSettings.Get(group='projection', key='format',
- subkey='precision'))
- if self.toolbar.GetAction() != 'addLine' or \
- self.toolbar.GetAction('type') not in ('line', 'boundary') or \
- len(self.polycoords) == 0:
- # we cannot provide info, so find out if it is something new
- if self._digitizingInfo:
- self._digitizingInfo = False
- self.digitizingInfoUnavailable.emit()
- return
- # else, we can provide info, so find out if it is first time
- if not self._digitizingInfo:
- self._digitizingInfo = True
- self.digitizingInfoAvailable.emit()
- # for linear feature show segment and total length
- distance_seg = self.Distance(self.polycoords[-1],
- (e, n), screen=False)[0]
- distance_tot = distance_seg
- for idx in range(1, len(self.polycoords)):
- distance_tot += self.Distance(self.polycoords[idx - 1],
- self.polycoords[idx],
- screen=False)[0]
- text = "seg: %.*f; tot: %.*f" % (precision, distance_seg,
- precision, distance_tot)
- self.digitizingInfo.emit(text=text)
- def OnKeyDown(self, event):
- """Key pressed"""
- shift = event.ShiftDown()
- kc = event.GetKeyCode()
- event = None
- if not shift:
- if kc == ord('P'):
- event = wx.CommandEvent(winid=self.toolbar.addPoint)
- tool = self.toolbar.OnAddPoint
- elif kc == ord('L'):
- event = wx.CommandEvent(winid=self.toolbar.addLine)
- tool = self.toolbar.OnAddLine
- if event:
- self.toolbar.OnTool(event)
- tool(event)
- def _updateMap(self):
- if not self.toolbar or \
- not self.toolbar.GetLayer():
- return
- # set region
- self.digit.GetDisplay().UpdateRegion()
- # re-calculate threshold for digitization tool
- # self.parent.digit.GetDisplay().GetThreshold()
- # draw map
- # self.pdcVector.Clear()
- self.pdcVector.RemoveAll()
- item = None
- if self.tree:
- try:
- item = self.tree.FindItemByData(
- 'maplayer', self.toolbar.GetLayer())
- except TypeError:
- pass
- if not self.tree or \
- (self.tree and item and
- self.tree.IsItemChecked(item)):
- self.redrawAll = True
- self.digit.GetDisplay().DrawMap()
- # translate tmp objects (pointer position)
- if self.toolbar.GetAction() == 'moveLine' and \
- hasattr(self, "moveInfo"):
- if 'beginDiff' in self.moveInfo:
- # move line
- for id in self.moveInfo['id']:
- self.pdcTmp.TranslateId(id,
- self.moveInfo['beginDiff'][0],
- self.moveInfo['beginDiff'][1])
- del self.moveInfo['beginDiff']
- def OnLeftDownAddLine(self, event):
- """Left mouse button pressed - add new feature
- """
- try:
- mapLayer = self.toolbar.GetLayer().GetName()
- except:
- return
- if self.toolbar.GetAction('type') in ['point', 'centroid']:
- # add new point / centroiud
- east, north = self.Pixel2Cell(self.mouse['begin'])
- nfeat, fids = self.digit.AddFeature(
- self.toolbar.GetAction('type'),
- [(east, north)])
- if nfeat < 1:
- return
- self.UpdateMap(render=False) # redraw map
- # add new record into atribute table
- if UserSettings.Get(
- group='vdigit', key="addRecord", subkey='enabled'):
- # select attributes based on layer and category
- cats = {
- fids[0]: {
- UserSettings.Get(
- group='vdigit',
- key="layer",
- subkey='value'): (
- UserSettings.Get(
- group='vdigit',
- key="category",
- subkey='value'),
- )}}
- posWindow = self.ClientToScreen(
- (self.mouse['end'][0] + self.dialogOffset,
- self.mouse['end'][1] + self.dialogOffset))
- addRecordDlg = DisplayAttributesDialog(
- parent=self, map=mapLayer, cats=cats, pos=posWindow,
- action="add", ignoreError=True)
- if self.toolbar.GetAction('type') == 'centroid':
- for fid in fids:
- self._geomAttrb(fid, addRecordDlg, 'area')
- self._geomAttrb(fid, addRecordDlg, 'perimeter')
- if addRecordDlg.IsFound():
- addRecordDlg.ShowModal()
- addRecordDlg.Destroy()
- elif self.toolbar.GetAction('type') in ["line", "boundary", "area"]:
- # add new point to the line
- self.polycoords.append(
- self.Pixel2Cell(
- event.GetPosition()))
- self.DrawLines(pdc=self.pdcTmp)
- def _geomAttrb(self, fid, dialog, attrb):
- """Define geometry attributes
- """
- mapLayer = self.toolbar.GetLayer()
- if self.tree:
- item = self.tree.FindItemByData('maplayer', mapLayer)
- vdigit = self.tree.GetLayerInfo(item, key='vdigit')
- else:
- item = vdigit = None
- if not vdigit or \
- 'geomAttr' not in vdigit or \
- attrb not in vdigit['geomAttr']:
- return
- val = -1
- if attrb == 'length':
- val = self.digit.GetLineLength(fid)
- type = attrb
- elif attrb == 'area':
- val = self.digit.GetAreaSize(fid)
- type = attrb
- elif attrb == 'perimeter':
- val = self.digit.GetAreaPerimeter(fid)
- type = 'length'
- if val > 0:
- layer = int(
- UserSettings.Get(
- group='vdigit',
- key="layer",
- subkey='value'))
- column = vdigit['geomAttr'][attrb]['column']
- val = UnitsConvertValue(
- val, type, vdigit['geomAttr'][attrb]['units'])
- dialog.SetColumnValue(layer, column, val)
- dialog.OnReset()
- def _geomAttrbUpdate(self, fids):
- """Update geometry atrributes of currently selected features
- :param fid: list feature id
- """
- mapLayer = self.parent.toolbars['vdigit'].GetLayer()
- vectorName = mapLayer.GetName()
- if self.tree:
- item = self.tree.FindItemByData('maplayer', mapLayer)
- vdigit = self.tree.GetLayerInfo(item, key='vdigit')
- else:
- item = vdigit = None
- if not vdigit or 'geomAttr' not in vdigit:
- return
- dbInfo = gselect.VectorDBInfo(vectorName)
- sqlfile = tempfile.NamedTemporaryFile(mode="w")
- for fid in fids:
- for layer, cats in self.digit.GetLineCats(fid).iteritems():
- table = dbInfo.GetTable(layer)
- for attrb, item in vdigit['geomAttr'].iteritems():
- val = -1
- if attrb == 'length':
- val = self.digit.GetLineLength(fid)
- type = attrb
- elif attrb == 'area':
- val = self.digit.GetAreaSize(fid)
- type = attrb
- elif attrb == 'perimeter':
- val = self.digit.GetAreaPerimeter(fid)
- type = 'length'
- if val < 0:
- continue
- val = UnitsConvertValue(val, type, item['units'])
- for cat in cats:
- sqlfile.write(
- 'UPDATE %s SET %s = %f WHERE %s = %d;\n' %
- (table, item['column'], val, dbInfo.GetKeyColumn(layer), cat))
- sqlfile.file.flush()
- RunCommand('db.execute',
- parent=True,
- quiet=True,
- input=sqlfile.name)
- def _updateATM(self):
- """Update open Attribute Table Manager
- .. todo::
- use AddDataRow() instead
- """
- if not self.lmgr:
- return
- # update ATM
- digitVector = self.toolbar.GetLayer().GetName()
- for atm in self.lmgr.dialogs['atm']:
- atmVector = atm.GetVectorName()
- if atmVector == digitVector:
- layer = UserSettings.Get(
- group='vdigit', key="layer", subkey='value')
- # TODO: use AddDataRow instead
- atm.LoadData(layer)
- def OnLeftDownEditLine(self, event):
- """Left mouse button pressed - edit linear feature - add new
- vertex.
- """
- self.polycoords.append(self.Pixel2Cell(self.mouse['begin']))
- self.moveInfo['id'].append(wx.NewId())
- self.DrawLines(pdc=self.pdcTmp)
- def OnLeftDownMoveLine(self, event):
- """Left mouse button pressed - vector digitizer move
- feature/vertex, edit linear feature
- """
- self.moveInfo = dict()
- # geographic coordinates of initial position (left-down)
- self.moveInfo['begin'] = None
- # list of ids to modify
- self.moveInfo['id'] = list()
- # set pen
- if self.toolbar.GetAction() in ["moveVertex", "editLine"]:
- pcolor = UserSettings.Get(group='vdigit', key="symbol",
- subkey=["highlight", "color"])
- self.pen = self.polypen = wx.Pen(colour=pcolor,
- width=2, style=wx.SHORT_DASH)
- self.pdcTmp.SetPen(self.polypen)
- def OnLeftDownDisplayCA(self, event):
- """Left mouse button pressed - vector digitizer display categories
- or attributes action
- """
- try:
- mapLayer = self.toolbar.GetLayer().GetName()
- except:
- return
- coords = self.Pixel2Cell(self.mouse['begin'])
- # unselect
- self.digit.GetDisplay().SetSelected([])
- # select feature by point
- cats = {}
- self.digit.GetDisplay().SelectLineByPoint(coords)
- if not self.digit.GetDisplay().GetSelected():
- for key in ('attributes', 'category'):
- if self.parent.dialogs[key] and \
- self.parent.dialogs[key].IsShown():
- self.parent.dialogs[key].Hide()
- self.UpdateMap(render=False, renderVector=True)
- return
- if UserSettings.Get(group='vdigit', key='checkForDupl',
- subkey='enabled'):
- lines = self.digit.GetDisplay().GetSelected()
- else:
- lines = (
- self.digit.GetDisplay().GetSelected()[0],
- ) # only first found
- for line in lines:
- cats[line] = self.digit.GetLineCats(line)
- posWindow = self.ClientToScreen(
- (self.mouse['end'][0] + self.dialogOffset,
- self.mouse['end'][1] + self.dialogOffset))
- if self.toolbar.GetAction() == "displayAttrs":
- # select attributes based on coordinates (all layers)
- if self.parent.dialogs['attributes'] is None:
- self.parent.dialogs['attributes'] = \
- DisplayAttributesDialog(parent=self, map=mapLayer,
- cats=cats,
- action="update")
- else:
- # upgrade dialog
- self.parent.dialogs['attributes'].UpdateDialog(cats=cats)
- if self.parent.dialogs['attributes'] and \
- self.parent.dialogs['attributes'].mapDBInfo:
- if len(cats.keys()) > 0:
- # highlight feature & re-draw map
- if not self.parent.dialogs['attributes'].IsShown():
- self.parent.dialogs['attributes'].Show()
- else:
- if self.parent.dialogs['attributes'] and \
- self.parent.dialogs['attributes'].IsShown():
- self.parent.dialogs['attributes'].Hide()
- else: # displayCats
- if self.parent.dialogs['category'] is None:
- # open new dialog
- dlg = VDigitCategoryDialog(parent=self,
- vectorName=mapLayer,
- cats=cats,
- pos=posWindow,
- title=_("Update categories"))
- self.parent.dialogs['category'] = dlg
- else:
- # update currently open dialog
- self.parent.dialogs['category'].UpdateDialog(cats=cats)
- if self.parent.dialogs['category']:
- if len(cats.keys()) > 0:
- # highlight feature & re-draw map
- if not self.parent.dialogs['category'].IsShown():
- self.parent.dialogs['category'].Show()
- else:
- if self.parent.dialogs['category'].IsShown():
- self.parent.dialogs['category'].Hide()
- self.UpdateMap(render=False, renderVector=True)
- def OnLeftDownCopyCA(self, event):
- """Left mouse button pressed - vector digitizer copy
- categories or attributes action
- """
- if not hasattr(self, "copyCatsList"):
- self.copyCatsList = []
- else:
- self.copyCatsIds = []
- self.mouse['box'] = 'box'
- def OnLeftDownCopyLine(self, event):
- """Left mouse button pressed - vector digitizer copy lines
- action
- """
- if not hasattr(self, "copyIds"):
- self.copyIds = []
- self.layerTmp = None
- def OnLeftDownBulkLine(self, event):
- """Left mouse button pressed - vector digitizer label 3D
- vector lines
- """
- if len(self.polycoords) > 1: # start new line
- self.polycoords = []
- self.ClearLines(pdc=self.pdcTmp)
- self.polycoords.append(self.Pixel2Cell(event.GetPosition()))
- if len(self.polycoords) == 1:
- begin = self.Pixel2Cell(self.polycoords[-1])
- end = self.Pixel2Cell(self.mouse['end'])
- else:
- end = self.Pixel2Cell(self.polycoords[-1])
- begin = self.Pixel2Cell(self.mouse['begin'])
- self.DrawLines(self.pdcTmp, polycoords=(begin, end))
- def OnLeftDownUndo(self, event):
- """Left mouse button pressed with control key - vector
- digitizer undo functionality
- """
- if self.mouse["use"] != "pointer" or not self.toolbar:
- return
- action = self.toolbar.GetAction()
- if (action == "addLine" and self.toolbar.GetAction('type')
- in ["line", "boundary", "area"]) or action == "editLine":
- # add line or boundary -> remove last point from the line
- try:
- removed = self.polycoords.pop()
- Debug.msg(
- 4, "VDigitWindow.OnMiddleDown(): polycoords_poped=%s" %
- [removed, ])
- # self.mouse['begin'] = self.Cell2Pixel(self.polycoords[-1])
- except:
- pass
- if action == "editLine":
- # remove last vertex & line
- if len(self.moveInfo['id']) > 1:
- self.moveInfo['id'].pop()
- self.UpdateMap(render=False, renderVector=False)
- elif action in ["deleteLine", "deleteArea", "moveLine", "splitLine",
- "addVertex", "removeVertex", "moveVertex",
- "copyCats", "flipLine", "mergeLine",
- "snapLine", "connectLine", "copyLine",
- "queryLine", "breakLine", "typeConv"]:
- # various tools -> unselected selected features
- self.digit.GetDisplay().SetSelected([])
- if action in ["moveLine", "moveVertex", "editLine"] and \
- hasattr(self, "moveInfo"):
- del self.moveInfo
- elif action == "copyCats":
- try:
- del self.copyCatsList
- del self.copyCatsIds
- except AttributeError:
- pass
- elif action == "copyLine":
- del self.copyIds
- if self.layerTmp:
- self.Map.DeleteLayer(self.layerTmp)
- self.UpdateMap(render=True, renderVector=False)
- del self.layerTmp
- self.polycoords = []
- self.UpdateMap(render=False) # render vector
- elif action == "zbulkLine":
- # reset polyline
- self.polycoords = []
- self.digit.GetDisplay().SetSelected([])
- self.UpdateMap(render=False)
- self.redrawAll = True
- self.UpdateMap(render=False, renderVector=False)
- def _onLeftDown(self, event):
- """Left mouse button donw - vector digitizer various actions
- """
- try:
- mapLayer = self.toolbar.GetLayer().GetName()
- except:
- GMessage(parent=self,
- message=_("No vector map selected for editing."))
- event.Skip()
- return
- action = self.toolbar.GetAction()
- if not action:
- GMessage(
- parent=self, message=_(
- "Nothing to do. "
- "Choose appropriate tool from digitizer toolbar."))
- event.Skip()
- return
- if action not in ("moveVertex",
- "addVertex",
- "removeVertex",
- "editLine"):
- # set pen
- self.pen = wx.Pen(
- colour=UserSettings.Get(
- group='vdigit',
- key='symbol',
- subkey=[
- 'newSegment',
- 'color']),
- width=2,
- style=wx.SHORT_DASH)
- self.polypen = wx.Pen(
- colour=UserSettings.Get(
- group='vdigit',
- key='symbol',
- subkey=[
- 'newLine',
- 'color']),
- width=2,
- style=wx.SOLID)
- if action in ("addVertex",
- "removeVertex",
- "splitLines"):
- # unselect
- self.digit.GetDisplay().SetSelected([])
- if action == "addLine":
- self.OnLeftDownAddLine(event)
- elif action == "editLine" and \
- hasattr(self, "moveInfo"):
- self.OnLeftDownEditLine(event)
- elif action in ("moveLine", "moveVertex", "editLine") and \
- not hasattr(self, "moveInfo"):
- self.OnLeftDownMoveLine(event)
- elif action in ("displayAttrs"
- "displayCats"):
- self.OnLeftDownDisplayCA(event)
- elif action in ("copyCats",
- "copyAttrs"):
- self.OnLeftDownCopyCA(event)
- elif action == "copyLine":
- self.OnLeftDownCopyLine(event)
- elif action == "zbulkLine":
- self.OnLeftDownBulkLine(event)
- def OnLeftUpVarious(self, event):
- """Left mouse button released - vector digitizer various
- actions
- """
- pos1 = self.Pixel2Cell(self.mouse['begin'])
- pos2 = self.Pixel2Cell(self.mouse['end'])
- nselected = 0
- action = self.toolbar.GetAction()
- # -> delete line || move line || move vertex
- if action in ("moveVertex",
- "editLine"):
- if len(self.digit.GetDisplay().GetSelected()) == 0:
- nselected = int(
- self.digit.GetDisplay().SelectLineByPoint(pos1)['line'] != -1)
- if action == "editLine":
- try:
- selVertex = self.digit.GetDisplay(
- ).GetSelectedVertex(pos1)[0]
- except IndexError:
- selVertex = None
- if selVertex:
- # self.UpdateMap(render=False)
- ids = self.digit.GetDisplay().GetSelected(grassId=False)
- # move this line to tmp layer
- self.polycoords = []
- for id in ids:
- if id % 2: # register only vertices
- e, n = self.Pixel2Cell(
- self.pdcVector.GetIdBounds(id)[0:2])
- self.polycoords.append((e, n))
- self.digit.GetDisplay().DrawSelected(False)
- if selVertex < ids[-1] / 2:
- # choose first or last node of line
- self.moveInfo['id'].reverse()
- self.polycoords.reverse()
- else:
- # unselect
- self.digit.GetDisplay().SetSelected([])
- del self.moveInfo
- self.UpdateMap(render=False)
- elif action in ("copyCats",
- "copyAttrs"):
- if not hasattr(self, "copyCatsIds"):
- # 'from' -> select by point
- nselected = int(
- self.digit.GetDisplay().SelectLineByPoint(pos1)['line'] != -1)
- if nselected:
- self.copyCatsList = self.digit.GetDisplay().GetSelected()
- else:
- # -> 'to' -> select by bbox
- self.digit.GetDisplay().SetSelected([])
- # return number of selected features (by box/point)
- nselected = self.digit.GetDisplay().SelectLinesByBox((pos1, pos2))
- if nselected == 0:
- nselected = int(
- self.digit.GetDisplay().SelectLineByPoint(pos1)['line'] != -1)
- if nselected > 0:
- self.copyCatsIds = self.digit.GetDisplay().GetSelected()
- elif action == "queryLine":
- selected = self.digit.SelectLinesByQuery(bbox=(pos1, pos2))
- nselected = len(selected)
- if nselected > 0:
- self.digit.GetDisplay().SetSelected(selected)
- else:
- # -> moveLine || deleteLine, etc. (select by point/box)
- if action == 'moveLine' and \
- len(self.digit.GetDisplay().GetSelected()) > 0:
- nselected = 0
- else:
- if action == 'deleteArea':
- nselected = int(
- self.digit.GetDisplay().SelectAreaByPoint(pos1)['area'] != -1)
- else:
- if action == 'moveLine':
- drawSeg = True
- else:
- drawSeg = False
- nselected = self.digit.GetDisplay().SelectLinesByBox(
- bbox=(pos1, pos2), drawSeg=drawSeg)
- if nselected == 0:
- nselected = int(
- self.digit.GetDisplay().SelectLineByPoint(pos1)['line'] != -1)
- if nselected > 0:
- if action in ("moveLine", "moveVertex") and \
- hasattr(self, "moveInfo"):
- # get pseudoDC id of objects which should be redrawn
- if action == "moveLine":
- # -> move line
- self.moveInfo['id'] = self.digit.GetDisplay(
- ).GetSelected(grassId=False)
- else: # moveVertex
- self.moveInfo['id'] = self.digit.GetDisplay(
- ).GetSelectedVertex(pos1)
- if len(self.moveInfo['id']) == 0: # no vertex found
- self.digit.GetDisplay().SetSelected([])
- #
- # check for duplicates
- #
- if UserSettings.Get(
- group='vdigit', key='checkForDupl', subkey='enabled'):
- dupl = self.digit.GetDisplay().GetDuplicates()
- self.UpdateMap(render=False)
- if dupl:
- posWindow = self.ClientToScreen(
- (self.mouse['end'][0] + self.dialogOffset,
- self.mouse['end'][1] + self.dialogOffset))
- dlg = VDigitDuplicatesDialog(
- parent=self, data=dupl, pos=posWindow)
- if dlg.ShowModal() == wx.ID_OK:
- self.digit.GetDisplay().UnSelect(dlg.GetUnSelected())
- # update selected
- self.UpdateMap(render=False)
- if action != "editLine":
- # -> move line || move vertex
- self.UpdateMap(render=False)
- else: # no vector object found
- if not (action in ("moveLine",
- "moveVertex") and
- hasattr(self, "moveInfo") and
- len(self.moveInfo['id']) > 0):
- # avoid left-click when features are already selected
- self.UpdateMap(render=False, renderVector=False)
- def OnLeftUpModifyLine(self, event):
- """Left mouse button released - vector digitizer split line,
- add/remove vertex action
- """
- pos1 = self.Pixel2Cell(self.mouse['begin'])
- pointOnLine = self.digit.GetDisplay().SelectLineByPoint(pos1)['point']
- if not pointOnLine:
- return
- if self.toolbar.GetAction() in ["splitLine", "addVertex"]:
- self.UpdateMap(render=False) # highlight object
- self.DrawCross(
- pdc=self.pdcTmp, coords=self.Cell2Pixel(
- (pointOnLine[0], pointOnLine[1])), size=5)
- else: # removeVertex
- # get only id of vertex
- try:
- id = self.digit.GetDisplay().GetSelectedVertex(pos1)[0]
- except IndexError:
- id = None
- if id:
- x, y = self.pdcVector.GetIdBounds(id)[0:2]
- self.pdcVector.RemoveId(id)
- self.UpdateMap(render=False) # highlight object
- self.DrawCross(pdc=self.pdcTmp, coords=(x, y),
- size=5)
- else:
- # unselect
- self.digit.GetDisplay().SetSelected([])
- self.UpdateMap(render=False)
- def OnLeftUpCopyLine(self, event):
- """Left mouse button released - vector digitizer copy feature
- action
- """
- pos1 = self.Pixel2Cell(self.mouse['begin'])
- pos2 = self.Pixel2Cell(self.mouse['end'])
- if UserSettings.Get(group='vdigit', key='bgmap',
- subkey='value', settings_type='internal') == '':
- # no background map -> copy from current vector map layer
- nselected = self.digit.GetDisplay().SelectLinesByBox((pos1, pos2))
- if nselected > 0:
- # highlight selected features
- self.UpdateMap(render=False)
- else:
- self.UpdateMap(render=False, renderVector=False)
- else:
- # copy features from background map
- self.copyIds = self.digit.SelectLinesFromBackgroundMap(
- bbox=(pos1,
- pos2))
- if len(self.copyIds) > 0:
- color = UserSettings.Get(group='vdigit', key='symbol',
- subkey=['highlight', 'color'])
- colorStr = str(color[0]) + ":" + str(color[1]
- ) + ":" + str(color[2])
- dVectTmp = [
- 'd.vect',
- 'map=%s' %
- UserSettings.Get(
- group='vdigit',
- key='bgmap',
- subkey='value',
- settings_type='internal'),
- 'cats=%s' %
- ListOfCatsToRange(
- self.copyIds),
- '-i',
- 'color=%s' %
- colorStr,
- 'fill_color=%s' %
- colorStr,
- 'type=point,line,boundary,centroid',
- 'width=2']
- if not self.layerTmp:
- self.layerTmp = self.Map.AddLayer(ltype='vector',
- name=QUERYLAYER,
- command=dVectTmp)
- else:
- self.layerTmp.SetCmd(dVectTmp)
- else:
- if self.layerTmp:
- self.Map.DeleteLayer(self.layerTmp)
- self.layerTmp = None
- self.UpdateMap(render=True, renderVector=True)
- def OnLeftUpBulkLine(self, event):
- """Left mouse button released - vector digitizer z-bulk line
- action
- """
- # select lines to be labeled
- pos1 = self.polycoords[0]
- pos2 = self.polycoords[1]
- nselected = self.digit.GetDisplay().SelectLinesByBox((pos1, pos2))
- if nselected > 0:
- # highlight selected features
- self.UpdateMap(render=False)
- self.DrawLines(pdc=self.pdcTmp) # redraw temp line
- else:
- self.UpdateMap(render=False, renderVector=False)
- def OnLeftUpConnectLine(self, event):
- """Left mouse button released - vector digitizer connect line
- action
- """
- if len(self.digit.GetDisplay().GetSelected()) > 0:
- self.UpdateMap(render=False)
- def _onLeftUp(self, event):
- """Left mouse button released"""
- if event.ControlDown():
- return
- if hasattr(self, "moveInfo"):
- if len(self.digit.GetDisplay().GetSelected()) == 0:
- self.moveInfo['begin'] = self.Pixel2Cell(
- self.mouse['begin']) # left down
- # eliminate initial mouse moving efect
- self.mouse['begin'] = self.mouse['end']
- action = self.toolbar.GetAction()
- if action in ("deleteLine",
- "deleteArea",
- "moveLine",
- "moveVertex",
- "copyCats",
- "copyAttrs",
- "editLine",
- "flipLine",
- "mergeLine",
- "snapLine",
- "queryLine",
- "breakLine",
- "typeConv",
- "connectLine"):
- self.OnLeftUpVarious(event)
- elif action in ("splitLine",
- "addVertex",
- "removeVertex"):
- self.OnLeftUpModifyLine(event)
- elif action == "copyLine":
- self.OnLeftUpCopyLine(event)
- elif action == "zbulkLine" and \
- len(self.polycoords) == 2:
- self.OnLeftUpBulkLine(event)
- elif action == "connectLine":
- self.OnLeftUpConnectLine(event)
- if len(self.digit.GetDisplay().GetSelected()) > 0:
- self.redrawAll = None
- def _onRightDown(self, event):
- # digitization tool (confirm action)
- action = self.toolbar.GetAction()
- if action in ("moveLine", "moveVertex") and \
- hasattr(self, "moveInfo"):
- pFrom = self.moveInfo['begin']
- pTo = self.Pixel2Cell(event.GetPosition())
- move = (pTo[0] - pFrom[0],
- pTo[1] - pFrom[1])
- if action == "moveLine":
- # move line
- if self.digit.MoveSelectedLines(move) < 0:
- return
- elif action == "moveVertex":
- # move vertex
- fid = self.digit.MoveSelectedVertex(pFrom, move)
- if fid < 0:
- return
- self._geomAttrbUpdate([fid, ])
- del self.moveInfo
- def _onRightUp(self, event):
- """Right mouse button released (confirm action)
- """
- action = self.toolbar.GetAction()
- if action == "addLine" and \
- self.toolbar.GetAction('type') in ["line", "boundary", "area"]:
- # -> add new line / boundary
- try:
- mapName = self.toolbar.GetLayer().GetName()
- except:
- mapName = None
- GError(parent=self,
- message=_("No vector map selected for editing."))
- if mapName:
- if self.toolbar.GetAction('type') == 'line':
- line = True
- else:
- line = False
- if len(self.polycoords) < 2: # ignore 'one-point' lines
- return
- nfeat, fids = self.digit.AddFeature(
- self.toolbar.GetAction('type'), self.polycoords)
- if nfeat < 0:
- return
- position = self.Cell2Pixel(self.polycoords[-1])
- self.polycoords = []
- self.UpdateMap(render=False)
- self.redrawAll = True
- self.Refresh()
- # add new record into atribute table
- if self._addRecord() and (line is True or (not line and nfeat > 0)):
- posWindow = self.ClientToScreen(
- (position[0] + self.dialogOffset, position[1] + self.dialogOffset))
- # select attributes based on layer and category
- cats = {
- fids[0]: {
- UserSettings.Get(
- group='vdigit',
- key="layer",
- subkey='value'): (
- UserSettings.Get(
- group='vdigit',
- key="category",
- subkey='value'),
- )}}
- addRecordDlg = DisplayAttributesDialog(
- parent=self, map=mapName, cats=cats, pos=posWindow,
- action="add", ignoreError=True)
- for fid in fids:
- self._geomAttrb(fid, addRecordDlg, 'length')
- # auto-placing centroid
- self._geomAttrb(fid, addRecordDlg, 'area')
- self._geomAttrb(fid, addRecordDlg, 'perimeter')
- if addRecordDlg.IsFound():
- addRecordDlg.ShowModal()
- addRecordDlg.Destroy()
- elif action == "deleteLine":
- # -> delete selected vector features
- if self.digit.DeleteSelectedLines() < 0:
- return
- self._updateATM()
- elif action == "deleteArea":
- # -> delete selected vector areas
- if self.digit.DeleteSelectedAreas() < 0:
- return
- self._updateATM()
- elif action == "splitLine":
- # split line
- if self.digit.SplitLine(self.Pixel2Cell(self.mouse['begin'])) < 0:
- return
- elif action == "addVertex":
- # add vertex
- fid = self.digit.AddVertex(self.Pixel2Cell(self.mouse['begin']))
- if fid < 0:
- return
- elif action == "removeVertex":
- # remove vertex
- fid = self.digit.RemoveVertex(self.Pixel2Cell(self.mouse['begin']))
- if fid < 0:
- return
- self._geomAttrbUpdate([fid, ])
- elif action in ("copyCats", "copyAttrs"):
- if action == 'copyCats':
- if self.digit.CopyCats(self.copyCatsList,
- self.copyCatsIds, copyAttrb=False) < 0:
- return
- else:
- if self.digit.CopyCats(self.copyCatsList,
- self.copyCatsIds, copyAttrb=True) < 0:
- return
- del self.copyCatsList
- del self.copyCatsIds
- self._updateATM()
- elif action == "editLine" and \
- hasattr(self, "moveInfo"):
- line = self.digit.GetDisplay().GetSelected()[0]
- if self.digit.EditLine(line, self.polycoords) < 0:
- return
- del self.moveInfo
- elif action == "flipLine":
- if self.digit.FlipLine() < 0:
- return
- elif action == "mergeLine":
- if self.digit.MergeLine() < 0:
- return
- elif action == "breakLine":
- if self.digit.BreakLine() < 0:
- return
- elif action == "snapLine":
- if self.digit.SnapLine() < 0:
- return
- elif action == "connectLine":
- if len(self.digit.GetDisplay().GetSelected()) > 1:
- if self.digit.ConnectLine() < 0:
- return
- elif action == "copyLine":
- if self.digit.CopyLine(self.copyIds) < 0:
- return
- del self.copyIds
- if self.layerTmp:
- self.Map.DeleteLayer(self.layerTmp)
- self.UpdateMap(render=True, renderVector=False)
- del self.layerTmp
- elif action == "zbulkLine" and len(self.polycoords) == 2:
- pos1 = self.polycoords[0]
- pos2 = self.polycoords[1]
- selected = self.digit.GetDisplay().GetSelected()
- dlg = VDigitZBulkDialog(
- parent=self,
- title=_("Z bulk-labeling dialog"),
- nselected=len(selected))
- if dlg.ShowModal() == wx.ID_OK:
- if self.digit.ZBulkLines(pos1, pos2, dlg.value.GetValue(),
- dlg.step.GetValue()) < 0:
- return
- self.UpdateMap(render=False)
- elif action == "typeConv":
- # -> feature type conversion
- # - point <-> centroid
- # - line <-> boundary
- if self.digit.TypeConvForSelectedLines() < 0:
- return
- if action != "addLine":
- # unselect and re-render
- self.digit.GetDisplay().SetSelected([])
- self.polycoords = []
- self.UpdateMap(render=False)
- def _onMouseMoving(self, event):
- self.mouse['end'] = event.GetPosition()
- Debug.msg(5, "VDigitWindow.OnMouseMoving(): coords=%f,%f" %
- (self.mouse['end'][0], self.mouse['end'][1]))
- action = self.toolbar.GetAction()
- if action == "addLine" and \
- self.toolbar.GetAction('type') in ["line", "boundary", "area"]:
- if len(self.polycoords) > 0:
- self.MouseDraw(pdc=self.pdcTmp,
- begin=self.Cell2Pixel(self.polycoords[-1]))
- elif action in ["moveLine", "moveVertex", "editLine"] \
- and hasattr(self, "moveInfo"):
- dx = self.mouse['end'][0] - self.mouse['begin'][0]
- dy = self.mouse['end'][1] - self.mouse['begin'][1]
- # draw lines on new position
- if action == "moveLine" and \
- len(self.moveInfo['id']) > 0:
- # move line
- for id in self.moveInfo['id']:
- self.pdcTmp.TranslateId(id, dx, dy)
- elif action in ["moveVertex", "editLine"]:
- # move vertex ->
- # (vertex, left vertex, left line,
- # right vertex, right line)
- # do not draw static lines
- if action == "moveVertex" and \
- len(self.moveInfo['id']) > 0:
- self.polycoords = []
- self.pdcTmp.RemoveId(self.moveInfo['id'][0])
- if self.moveInfo['id'][1] > 0: # previous vertex
- x, y = self.Pixel2Cell(
- self.pdcTmp.GetIdBounds(
- self.moveInfo['id'][1])[
- 0:2])
- self.pdcTmp.RemoveId(self.moveInfo['id'][1] + 1)
- self.polycoords.append((x, y))
- self.polycoords.append(self.Pixel2Cell(self.mouse['end']))
- if self.moveInfo['id'][2] > 0: # next vertex
- x, y = self.Pixel2Cell(
- self.pdcTmp.GetIdBounds(
- self.moveInfo['id'][2])[
- 0:2])
- self.pdcTmp.RemoveId(self.moveInfo['id'][2] - 1)
- self.polycoords.append((x, y))
- self.ClearLines(pdc=self.pdcTmp)
- self.DrawLines(pdc=self.pdcTmp)
- if action == "editLine":
- self.MouseDraw(pdc=self.pdcTmp,
- begin=self.Cell2Pixel(self.polycoords[-1]))
- self.Refresh() # TODO: use RefreshRect()
- self.mouse['begin'] = self.mouse['end']
- elif action == "zbulkLine":
- if len(self.polycoords) == 1:
- # draw mouse moving
- self.MouseDraw(self.pdcTmp)
- def _zoom(self, event):
- tmp1 = self.mouse['end']
- tmp2 = self.Cell2Pixel(self.moveInfo['begin'])
- dx = tmp1[0] - tmp2[0]
- dy = tmp1[1] - tmp2[1]
- self.moveInfo['beginDiff'] = (dx, dy)
- for id in self.moveInfo['id']:
- self.pdcTmp.RemoveId(id)
- def _addRecord(self):
- return UserSettings.Get(
- group='vdigit', key="addRecord", subkey='enabled')
|