wxdigit.py 69 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041
  1. """
  2. @package vdigit.wxdigit
  3. @brief wxGUI vector digitizer (base class)
  4. Code based on wxVdigit C++ component from GRASS 6.4.0
  5. (gui/wxpython/vdigit). Converted to Python in 2010/12-2011/01.
  6. List of classes:
  7. - wxdigit::VDigitError
  8. - wxdigit::IVDigit
  9. .. todo::
  10. Read large amounts of data from Vlib into arrays, which could
  11. then be processed using NumPy and rendered using glDrawArrays or
  12. glDrawElements, so no per-line/per-vertex processing in Python. Bulk
  13. data processing with NumPy is much faster than iterating in Python
  14. (and NumPy would be an excellent candidate for acceleration via
  15. e.g. OpenCL or CUDA; I'm surprised it hasn't happened already).
  16. (C) 2007-2014 by the GRASS Development Team
  17. This program is free software under the GNU General Public License
  18. (>=v2). Read the file COPYING that comes with GRASS for details.
  19. @author Martin Landa <landa.martin gmail.com>
  20. """
  21. import grass.script.core as grass
  22. from grass.pydispatch.signal import Signal
  23. from core.gcmd import GError
  24. from core.debug import Debug
  25. from core.settings import UserSettings
  26. from core.utils import _
  27. from vdigit.wxdisplay import DisplayDriver, GetLastError
  28. try:
  29. from grass.lib.gis import *
  30. from grass.lib.vector import *
  31. from grass.lib.vedit import *
  32. from grass.lib.dbmi import *
  33. except ImportError:
  34. pass
  35. class VDigitError:
  36. def __init__(self, parent):
  37. """Class for managing error messages of vector digitizer
  38. :param parent: parent window for dialogs
  39. """
  40. self.parent = parent
  41. self.caption = _('Digitization Error')
  42. def NoMap(self, name = None):
  43. """No map for editing"""
  44. if name:
  45. message = _('Unable to open vector map <%s>.') % name
  46. else:
  47. message = _('No vector map open for editing.')
  48. GError(message + ' ' + _('Operation canceled.'),
  49. parent = self.parent,
  50. caption = self.caption)
  51. def WriteLine(self):
  52. """Writing line failed
  53. """
  54. GError(message = _('Writing new feature failed. '
  55. 'Operation canceled.\n\n'
  56. 'Reason: %s') % GetLastError(),
  57. parent = self.parent,
  58. caption = self.caption)
  59. def ReadLine(self, line):
  60. """Reading line failed
  61. """
  62. GError(message = _('Reading feature id %d failed. '
  63. 'Operation canceled.') % line,
  64. parent = self.parent,
  65. caption = self.caption)
  66. def DbLink(self, dblink):
  67. """No dblink available
  68. """
  69. GError(message = _('Database link %d not available. '
  70. 'Operation canceled.') % dblink,
  71. parent = self.parent,
  72. caption = self.caption)
  73. def Driver(self, driver):
  74. """Staring driver failed
  75. """
  76. GError(message = _('Unable to start database driver <%s>. '
  77. 'Operation canceled.') % driver,
  78. parent = self.parent,
  79. caption = self.caption)
  80. def Database(self, driver, database):
  81. """Opening database failed
  82. """
  83. GError(message = _('Unable to open database <%(db)s> by driver <%(driver)s>. '
  84. 'Operation canceled.') % { 'db' : database, 'driver' : driver},
  85. parent = self.parent,
  86. caption = self.caption)
  87. def DbExecute(self, sql):
  88. """Sql query failed
  89. """
  90. GError(message = _("Unable to execute SQL query '%s'. "
  91. "Operation canceled.") % sql,
  92. parent = self.parent,
  93. caption = self.caption)
  94. def DeadLine(self, line):
  95. """Dead line
  96. """
  97. GError(message = _("Feature id %d is marked as dead. "
  98. "Operation canceled.") % line,
  99. parent = self.parent,
  100. caption = self.caption)
  101. def FeatureType(self, ftype):
  102. """Unknown feature type
  103. """
  104. GError(message = _("Unsupported feature type %d. "
  105. "Operation canceled.") % ftype,
  106. parent = self.parent,
  107. caption = self.caption)
  108. class IVDigit:
  109. def __init__(self, mapwindow, driver = DisplayDriver):
  110. """Base class for vector digitizer (ctypes interface)
  111. :param mapwindow: reference to a map window
  112. """
  113. self.poMapInfo = None # pointer to Map_info
  114. self.mapWindow = mapwindow
  115. # background map
  116. self.bgMapInfo = Map_info()
  117. self.poBgMapInfo = self.popoBgMapInfo = None
  118. # TODO: replace this by using giface
  119. if not mapwindow.parent.IsStandalone():
  120. goutput = mapwindow.parent.GetLayerManager().GetLogWindow()
  121. log = goutput.GetLog(err = True)
  122. progress = mapwindow.parent._giface.GetProgress()
  123. else:
  124. log = sys.stderr
  125. progress = None
  126. self.toolbar = mapwindow.parent.toolbars['vdigit']
  127. self._error = VDigitError(parent = self.mapWindow)
  128. self._display = driver(device = mapwindow.pdcVector,
  129. deviceTmp = mapwindow.pdcTmp,
  130. mapObj = mapwindow.Map,
  131. window = mapwindow,
  132. glog = log,
  133. gprogress = progress)
  134. # GRASS lib
  135. self.poPoints = Vect_new_line_struct()
  136. self.poCats = Vect_new_cats_struct()
  137. # self.SetCategory()
  138. # layer / max category
  139. self.cats = dict()
  140. self._settings = dict()
  141. self.UpdateSettings() # -> self._settings
  142. # undo/redo
  143. self.changesets = list()
  144. self.changesetCurrent = -1 # first changeset to apply
  145. if self.poMapInfo:
  146. self.InitCats()
  147. self.emit_signals = False
  148. # signals which describes features changes during digitization,
  149. # activate them using EmitSignals method
  150. # currently implemented for functionality used by wx.iclass (for scatter plot)
  151. # signals parameter description:
  152. # old_bboxs - list of bboxes of boundary features, which covers changed areas
  153. # it is bbox of old state (before edit)
  154. # old_areas_cats - list of area categories of boundary features of old state (before edit)
  155. # same position in both lists corresponds to same feature
  156. # new_bboxs = list of bboxes of created features / after edit
  157. # new_areas_cats list of areas cats of created features / after edit
  158. # same position in both lists corresponds to same features
  159. # for description of items in bbox and area_cats lists see return value of _getaAreaBboxCats
  160. # TODO currently it is not possible to identify corresponded features
  161. # in old and new lists (requires changed to vector updated format)
  162. # TODO return feature type
  163. #TODO handle errors?
  164. self.featureAdded = Signal('IVDigit.featureAdded')
  165. self.areasDeleted = Signal('IVDigit.areasDeleted')
  166. self.vertexMoved = Signal('IVDigit.vertexMoved')
  167. self.vertexAdded = Signal('IVDigit.vertexAdded')
  168. self.vertexRemoved = Signal('IVDigit.vertexRemoved')
  169. self.featuresDeleted = Signal('IVDigit.featuresDeleted')
  170. self.featuresMoved = Signal('IVDigit.featuresMoved')
  171. self.lineEdited = Signal('IVDigit.lineEdited')
  172. def __del__(self):
  173. Debug.msg(1, "IVDigit.__del__()")
  174. Vect_destroy_line_struct(self.poPoints)
  175. self.poPoints = None
  176. Vect_destroy_cats_struct(self.poCats)
  177. self.poCats = None
  178. if self.poBgMapInfo:
  179. Vect_close(self.poBgMapInfo)
  180. self.poBgMapInfo = self.popoBgMapInfo = None
  181. del self.bgMapInfo
  182. def EmitSignals(self, emit):
  183. """Activate/deactivate signals which describes features changes during digitization.
  184. """
  185. self.emit_signals = emit
  186. def CloseBackgroundMap(self):
  187. """Close background vector map"""
  188. if not self.poBgMapInfo:
  189. return
  190. Vect_close(self.poBgMapInfo)
  191. self.poBgMapInfo = self.popoBgMapInfo = None
  192. def OpenBackgroundMap(self, bgmap):
  193. """Open background vector map
  194. .. todo::
  195. support more background maps then only one
  196. :param bgmap: name of vector map to be opened
  197. :type bgmap: str
  198. :return: pointer to map_info
  199. :return: None on error
  200. """
  201. name = create_string_buffer(GNAME_MAX)
  202. mapset = create_string_buffer(GMAPSET_MAX)
  203. if not G_name_is_fully_qualified(bgmap, name, mapset):
  204. name = str(bgmap)
  205. mapset = str(G_find_vector2(bgmap, ''))
  206. else:
  207. name = str(name.value)
  208. mapset = str(mapset.value)
  209. if (name == Vect_get_name(self.poMapInfo) and \
  210. mapset == Vect_get_mapset(self.poMapInfo)):
  211. self.poBgMapInfo = self.popoBgMapInfo = None
  212. self._error.NoMap(bgmap)
  213. return
  214. self.poBgMapInfo = pointer(self.bgMapInfo)
  215. self.popoBgMapInfo = pointer(self.poBgMapInfo)
  216. if Vect_open_old(self.poBgMapInfo, name, mapset) == -1:
  217. self.poBgMapInfo = self.popoBgMapInfo = None
  218. self._error.NoMap(bgmap)
  219. return
  220. def _getSnapMode(self):
  221. """Get snapping mode
  222. - snap to vertex
  223. - snap to nodes
  224. - no snapping
  225. :return: snap mode
  226. """
  227. threshold = self._display.GetThreshold()
  228. if threshold > 0.0:
  229. if UserSettings.Get(group = 'vdigit', key = 'snapToVertex', subkey = 'enabled'):
  230. return SNAPVERTEX
  231. else:
  232. return SNAP
  233. else:
  234. return NO_SNAP
  235. def _getNewFeaturesLayer(self):
  236. """Returns layer of new feature (from settings)"""
  237. if UserSettings.Get(group = 'vdigit', key = "categoryMode", subkey = 'selection') == 2:
  238. layer = -1 # -> no category
  239. else:
  240. layer = UserSettings.Get(group = 'vdigit', key = "layer", subkey = 'value')
  241. return layer
  242. def _getNewFeaturesCat(self):
  243. """Returns category of new feature (from settings)"""
  244. if UserSettings.Get(group = 'vdigit', key = "categoryMode", subkey = 'selection') == 2:
  245. cat = -1
  246. else:
  247. cat = self.SetCategory()
  248. return cat
  249. def _breakLineAtIntersection(self, line, pointsLine):
  250. """Break given line at intersection
  251. :param line: line id
  252. :param pointsLine: line geometry
  253. :return: number of modified lines
  254. """
  255. if not self._checkMap():
  256. return -1
  257. if not Vect_line_alive(self.poMapInfo, line):
  258. return 0
  259. if not pointsLine:
  260. if Vect_read_line(self.poMapInfo, self.poPoints, None, line) < 0:
  261. self._error.ReadLine(line)
  262. return -1
  263. points = self.poPoints
  264. else:
  265. points = pointsLine
  266. listLine = Vect_new_boxlist(0)
  267. listRef = Vect_new_list()
  268. listBreak = Vect_new_list()
  269. pointsCheck = Vect_new_line_struct()
  270. lineBox = bound_box()
  271. # find all relevant lines
  272. Vect_get_line_box(self.poMapInfo, line, byref(lineBox))
  273. Vect_select_lines_by_box(self.poMapInfo, byref(lineBox),
  274. GV_LINES, listLine)
  275. # check for intersection
  276. Vect_list_append(listBreak, line)
  277. Vect_list_append(listRef, line)
  278. for i in range(listLine.contents.n_values):
  279. lineBreak = listLine.contents.id[i]
  280. if lineBreak == line:
  281. continue
  282. ltype = Vect_read_line(self.poMapInfo, pointsCheck, None, lineBreak)
  283. if not (ltype & GV_LINES):
  284. continue
  285. if Vect_line_check_intersection(self.poPoints, pointsCheck,
  286. WITHOUT_Z):
  287. Vect_list_append(listBreak, lineBreak)
  288. ret = Vect_break_lines_list(self.poMapInfo, listBreak, listRef,
  289. GV_LINES, None)
  290. Vect_destroy_line_struct(pointsCheck)
  291. Vect_destroy_boxlist(listLine)
  292. Vect_destroy_list(listBreak)
  293. Vect_destroy_list(listRef)
  294. return ret
  295. def _addChangeset(self):
  296. # disable redo
  297. changesetLast = len(self.changesets) - 1
  298. if self.changesetCurrent < changesetLast and len(self.changesets) > 0:
  299. del self.changesets[self.changesetCurrent + 1:changesetLast + 1]
  300. self.toolbar.EnableRedo(False)
  301. data = list()
  302. for i in range(Vect_get_num_updated_lines(self.poMapInfo) - 1, -1, -1):
  303. line = Vect_get_updated_line(self.poMapInfo, i)
  304. offset = Vect_get_updated_line_offset(self.poMapInfo, i)
  305. data.append({ 'line' : line,
  306. 'offset' : offset })
  307. self.changesetCurrent += 1
  308. self.changesets.insert(self.changesetCurrent, data)
  309. Vect_reset_updated(self.poMapInfo)
  310. def _applyChangeset(self, changeset, undo):
  311. """Apply changeset (undo/redo changeset)
  312. :param changeset: changeset id
  313. :param undo: True for undo otherwise redo
  314. :type undo: bool
  315. :return: 1 changeset applied
  316. :return: 0 changeset not applied
  317. :return: -1 on error
  318. """
  319. if changeset < 0 or changeset >= len(self.changesets):
  320. return -1
  321. ret = 0
  322. actions = self.changesets[changeset]
  323. if undo:
  324. firstaction = 0
  325. lastaction = len(actions)
  326. step = 1
  327. else:
  328. firstaction = len(actions) - 1
  329. lastaction = -1
  330. step = -1
  331. for i in range(firstaction, lastaction, step):
  332. action = actions[i]
  333. line = action['line']
  334. if action['offset'] > 0:
  335. if Vect_line_alive(self.poMapInfo, line):
  336. Debug.msg(3, "IVDigit._applyChangeset(): changeset=%d, action=add, line=%d -> deleted",
  337. changeset, line)
  338. Vect_delete_line(self.poMapInfo, line)
  339. ret = 1
  340. else:
  341. Debug.msg(3, "Digit.ApplyChangeset(): changeset=%d, action=add, line=%d dead",
  342. changeset, line)
  343. else: # delete
  344. offset = abs(action['offset'])
  345. if not Vect_line_alive(self.poMapInfo, line):
  346. Debug.msg(3, "Digit.ApplyChangeset(): changeset=%d, action=delete, line=%d -> added",
  347. changeset, line)
  348. if Vect_restore_line(self.poMapInfo, line, offset) < 0:
  349. return -1
  350. ret = 1
  351. else:
  352. Debug.msg(3, "Digit.ApplyChangeset(): changeset=%d, action=delete, line=%d alive",
  353. changeset, line)
  354. action['offset'] *= -1
  355. Vect_reset_updated(self.poMapInfo)
  356. return ret
  357. def AddFeature(self, ftype, points):
  358. """Add new feature
  359. :param ftype: feature type (point, line, centroid, boundary)
  360. :param points: tuple of points ((x, y), (x, y), ...)
  361. :return: tuple (number of added features, feature ids)
  362. """
  363. layer = self._getNewFeaturesLayer()
  364. cat = self._getNewFeaturesCat()
  365. if ftype == 'point':
  366. vtype = GV_POINT
  367. elif ftype == 'line':
  368. vtype = GV_LINE
  369. elif ftype == 'centroid':
  370. vtype = GV_CENTROID
  371. elif ftype == 'boundary':
  372. vtype = GV_BOUNDARY
  373. elif ftype == 'area':
  374. vtype = GV_AREA
  375. else:
  376. GError(parent = self.mapWindow,
  377. message = _("Unknown feature type '%s'") % ftype)
  378. return (-1, None)
  379. if vtype & GV_LINES and len(points) < 2:
  380. GError(parent = self.mapWindow,
  381. message = _("Not enough points for line"))
  382. return (-1, None)
  383. self.toolbar.EnableUndo()
  384. ret = self._addFeature(vtype, points, layer, cat,
  385. self._getSnapMode(), self._display.GetThreshold())
  386. if ret[0] > -1 and self.emit_signals:
  387. self.featureAdded.emit(new_bboxs=[self._createBbox(points)],
  388. new_areas_cats=[[{layer : [cat]}, None]])
  389. return ret
  390. def DeleteSelectedLines(self):
  391. """Delete selected features
  392. :return: number of deleted features
  393. """
  394. if not self._checkMap():
  395. return -1
  396. # collect categories for deleting if requested
  397. deleteRec = UserSettings.Get(group = 'vdigit', key = 'delRecord', subkey = 'enabled')
  398. catDict = dict()
  399. old_bboxs = []
  400. old_areas_cats = []
  401. if deleteRec:
  402. for i in self._display.selected['ids']:
  403. if Vect_read_line(self.poMapInfo, None, self.poCats, i) < 0:
  404. self._error.ReadLine(i)
  405. if self.emit_signals:
  406. ret = self._getLineAreaBboxCats(i)
  407. if ret:
  408. old_bboxs += ret[0]
  409. old_areas_cats += ret[1]
  410. # catDict was not used -> put into comment
  411. #cats = self.poCats.contents
  412. #for j in range(cats.n_cats):
  413. # if cats.field[j] not in catDict.keys():
  414. # catDict[cats.field[j]] = list()
  415. # catDict[cats.field[j]].append(cats.cat[j])
  416. poList = self._display.GetSelectedIList()
  417. nlines = Vedit_delete_lines(self.poMapInfo, poList)
  418. Vect_destroy_list(poList)
  419. self._display.selected['ids'] = list()
  420. if nlines > 0:
  421. if deleteRec:
  422. self._deleteRecords(catDict)
  423. self._addChangeset()
  424. self.toolbar.EnableUndo()
  425. if self.emit_signals:
  426. self.featuresDeleted.emit(old_bboxs=old_bboxs,
  427. old_areas_cats=old_areas_cats)
  428. return nlines
  429. def _deleteRecords(self, cats):
  430. """Delete records from attribute table
  431. :param cats: directory field/list of cats
  432. """
  433. handle = dbHandle()
  434. poHandle = pointer(handle)
  435. stmt = dbString()
  436. poStmt = pointer(stmt)
  437. for dblink in range(Vect_get_num_dblinks(self.poMapInfo)):
  438. poFi = Vect_get_dblink(self.poMapInfo, dblink)
  439. if poFi is None:
  440. self._error.DbLink(dblink)
  441. return -1
  442. Fi = poFi.contents
  443. if Fi.number not in cats.keys():
  444. continue
  445. poDriver = db_start_driver(Fi.driver)
  446. if poDriver is None:
  447. self._error.Driver(Fi.driver)
  448. return -1
  449. db_init_handle(poHandle)
  450. db_set_handle(poHandle, Fi.database, None)
  451. if db_open_database(poDriver, poHandle) != DB_OK:
  452. self._error.Database(Fi.driver, Fi.database)
  453. return -1
  454. db_init_string(poStmt)
  455. db_set_string(poStmt, "DELETE FROM %s WHERE" % Fi.table)
  456. n_cats = 0
  457. for cat in cats[Fi.number]:
  458. if n_cats > 0:
  459. db_append_string(poStmt, " or")
  460. db_append_string(poStmt, " %s = %d" % (Fi.key, cat))
  461. n_cats += 1
  462. if n_cats > 0 and \
  463. db_execute_immediate(poDriver, poStmt) != DB_OK:
  464. self._error.DbExecute(db_get_string(poStmt))
  465. return -1
  466. db_close_database_shutdown_driver(poDriver)
  467. def DeleteSelectedAreas(self):
  468. """Delete selected areas (centroid+boundaries)
  469. :return: number of deleted
  470. """
  471. if len(self._display.selected['ids']) < 1:
  472. return 0
  473. poList = self._display.GetSelectedIList()
  474. cList = poList.contents
  475. nareas = 0
  476. old_bboxs = []
  477. old_areas_cats = []
  478. for i in range(cList.n_values):
  479. if Vect_get_line_type(self.poMapInfo, cList.value[i]) != GV_CENTROID:
  480. continue
  481. if self.emit_signals:
  482. area = Vect_get_centroid_area(self.poMapInfo, cList.value[i]);
  483. if area > 0:
  484. bbox, cats = self._getaAreaBboxCats(area)
  485. old_bboxs += bbox
  486. old_areas_cats += cats
  487. nareas += Vedit_delete_area_centroid(self.poMapInfo, cList.value[i])
  488. if nareas > 0:
  489. self._addChangeset()
  490. self.toolbar.EnableUndo()
  491. if self.emit_signals:
  492. self.areasDeleted.emit(old_bboxs=old_bboxs,
  493. old_areas_cats=old_areas_cats)
  494. return nareas
  495. def _getLineAreaBboxCats(self, ln_id):
  496. """Helper function
  497. :param ln_id: id of feature
  498. :return: None if the feature does not exists
  499. :return: list of :func:`_getaAreaBboxCats`
  500. """
  501. ltype = Vect_read_line(self.poMapInfo, None, None, ln_id)
  502. if ltype == GV_CENTROID:
  503. #TODO centroid opttimization, can be edited also its area -> it will appear two times in new_ lists
  504. return self._getCentroidAreaBboxCats(ln_id)
  505. else:
  506. return [self._getBbox(ln_id)], [self._getLineAreasCategories(ln_id)]
  507. def _getCentroidAreaBboxCats(self, centroid):
  508. """Helper function
  509. :param centroid: id of an centroid
  510. :return: None if area does not exists
  511. :return: see return of :func:`_getaAreaBboxCats`
  512. """
  513. if not Vect_line_alive(self.poMapInfo, centroid):
  514. return None
  515. area = Vect_get_centroid_area(self.poMapInfo, centroid)
  516. if area > 0:
  517. return self._getaAreaBboxCats(area)
  518. else:
  519. return None
  520. def _getaAreaBboxCats(self, area):
  521. """Helper function
  522. :param area: area id
  523. :return: list of categories :func:`_getLineAreasCategories` and
  524. list of bboxes :func:`_getBbox` of area boundary
  525. features
  526. """
  527. po_b_list = Vect_new_list()
  528. Vect_get_area_boundaries(self.poMapInfo, area, po_b_list);
  529. b_list = po_b_list.contents
  530. geoms = []
  531. areas_cats = []
  532. if b_list.n_values > 0:
  533. for i_line in range(b_list.n_values):
  534. line = b_list.value[i_line];
  535. geoms.append(self._getBbox(abs(line)))
  536. areas_cats.append(self._getLineAreasCategories(abs(line)))
  537. Vect_destroy_list(po_b_list);
  538. return geoms, areas_cats
  539. def _getLineAreasCategories(self, ln_id):
  540. """Helper function
  541. :param line_id: id of boundary feature
  542. :return: categories of areas on the left, right side of the feature
  543. :return: format: [[{layer : [cat]}, None]] means:
  544. area to the left (list of layers which has cats list as values),
  545. area to the right (no area there in this case (None))
  546. :return: [] the feature is not boundary or does not exists
  547. """
  548. if not Vect_line_alive (self.poMapInfo, ln_id):
  549. return []
  550. ltype = Vect_read_line(self.poMapInfo, None, None, ln_id)
  551. if ltype != GV_BOUNDARY:
  552. return []
  553. cats = [None, None]
  554. left = c_int()
  555. right = c_int()
  556. if Vect_get_line_areas(self.poMapInfo, ln_id, pointer(left), pointer(right)) == 1:
  557. areas = [left.value, right.value]
  558. for i, a in enumerate(areas):
  559. if a > 0:
  560. centroid = Vect_get_area_centroid(self.poMapInfo, a)
  561. if centroid <= 0:
  562. continue
  563. c = self._getCategories(centroid)
  564. if c:
  565. cats[i] = c
  566. return cats
  567. def _getCategories(self, ln_id):
  568. """Helper function
  569. :param line_id: id of feature
  570. :return: list of the feature categories [{layer : cats}, next layer...]
  571. :return: None feature does not exist
  572. """
  573. if not Vect_line_alive (self.poMapInfo, ln_id):
  574. return none
  575. poCats = Vect_new_cats_struct()
  576. if Vect_read_line(self.poMapInfo, None, poCats, ln_id) < 0:
  577. Vect_destroy_cats_struct(poCats)
  578. return None
  579. cCats = poCats.contents
  580. cats = {}
  581. for j in range(cCats.n_cats):
  582. if cats.has_key(cCats.field[j]):
  583. cats[cCats.field[j]].append(cCats.cat[j])
  584. else:
  585. cats[cCats.field[j]] = [cCats.cat[j]]
  586. Vect_destroy_cats_struct(poCats)
  587. return cats
  588. def _getBbox(self, ln_id):
  589. """Helper function
  590. :param line_id: id of line feature
  591. :return: bbox bounding box of the feature
  592. :return: None feature does not exist
  593. """
  594. if not Vect_line_alive (self.poMapInfo, ln_id):
  595. return None
  596. poPoints = Vect_new_line_struct()
  597. if Vect_read_line(self.poMapInfo, poPoints, None, ln_id) < 0:
  598. Vect_destroy_line_struct(poPoints)
  599. return []
  600. geom = self._convertGeom(poPoints)
  601. bbox = self._createBbox(geom)
  602. Vect_destroy_line_struct(poPoints)
  603. return bbox
  604. def _createBbox(self, points):
  605. """Helper function
  606. :param points: list of points [(x, y), ...] to be bbox created for
  607. :return: bbox bounding box of points {'maxx':, 'maxy':, 'minx':, 'miny'}
  608. """
  609. bbox = {}
  610. for pt in points:
  611. if not bbox.has_key('maxy'):
  612. bbox['maxy'] = pt[1]
  613. bbox['miny'] = pt[1]
  614. bbox['maxx'] = pt[0]
  615. bbox['minx'] = pt[0]
  616. continue
  617. if bbox['maxy'] < pt[1]:
  618. bbox['maxy'] = pt[1]
  619. elif bbox['miny'] > pt[1]:
  620. bbox['miny'] = pt[1]
  621. if bbox['maxx'] < pt[0]:
  622. bbox['maxx'] = pt[0]
  623. elif bbox['minx'] > pt[0]:
  624. bbox['minx'] = pt[0]
  625. return bbox
  626. def _convertGeom(self, poPoints):
  627. """Helper function convert geom from ctypes line_pts to python
  628. list
  629. :return: coords in python list [(x, y),...]
  630. """
  631. Points = poPoints.contents
  632. pts_geom = []
  633. for j in range(Points.n_points):
  634. pts_geom.append((Points.x[j], Points.y[j]))
  635. return pts_geom
  636. def MoveSelectedLines(self, move):
  637. """Move selected features
  638. :param move: direction (x, y)
  639. """
  640. if not self._checkMap():
  641. return -1
  642. nsel = len(self._display.selected['ids'])
  643. if nsel < 1:
  644. return -1
  645. thresh = self._display.GetThreshold()
  646. snap = self._getSnapMode()
  647. poList = self._display.GetSelectedIList()
  648. if self.emit_signals:
  649. old_bboxs = []
  650. old_areas_cats = []
  651. for sel_id in self._display.selected['ids']:
  652. ret = self._getLineAreaBboxCats(sel_id)
  653. if ret:
  654. old_bboxs += ret[0]
  655. old_areas_cats += ret[1]
  656. Vect_set_updated(self.poMapInfo, 1)
  657. n_up_lines_old = Vect_get_num_updated_lines(self.poMapInfo)
  658. nlines = Vedit_move_lines(self.poMapInfo, self.popoBgMapInfo, int(self.poBgMapInfo is not None),
  659. poList,
  660. move[0], move[1], 0,
  661. snap, thresh)
  662. Vect_destroy_list(poList)
  663. if nlines > 0 and self.emit_signals:
  664. new_bboxs = []
  665. new_areas_cats = []
  666. n_up_lines = Vect_get_num_updated_lines(self.poMapInfo)
  667. for i in range(n_up_lines_old, n_up_lines):
  668. new_id = Vect_get_updated_line(self.poMapInfo, i)
  669. ret = self._getLineAreaBboxCats(new_id)
  670. if ret:
  671. new_bboxs += ret[0]
  672. new_areas_cats += ret[1]
  673. if nlines > 0 and self._settings['breakLines']:
  674. for i in range(1, nlines):
  675. self._breakLineAtIntersection(nlines + i, None, changeset)
  676. if nlines > 0:
  677. self._addChangeset()
  678. self.toolbar.EnableUndo()
  679. if self.emit_signals:
  680. self.featuresMoved.emit(new_bboxs=new_bboxs,
  681. old_bboxs=old_bboxs,
  682. old_areas_cats=old_areas_cats,
  683. new_areas_cats=new_areas_cats)
  684. return nlines
  685. def MoveSelectedVertex(self, point, move):
  686. """Move selected vertex of the line
  687. :param point: location point
  688. :param move: x,y direction
  689. :return: id of new feature
  690. :return: 0 vertex not moved (not found, line is not selected)
  691. :return: -1 on error
  692. """
  693. if not self._checkMap():
  694. return -1
  695. if len(self._display.selected['ids']) != 1:
  696. return -1
  697. # move only first found vertex in bbox
  698. poList = self._display.GetSelectedIList()
  699. if self.emit_signals:
  700. cList = poList.contents
  701. old_bboxs = [self._getBbox(cList.value[0])]
  702. old_areas_cats = [self._getLineAreasCategories(cList.value[0])]
  703. Vect_set_updated(self.poMapInfo, 1)
  704. n_up_lines_old = Vect_get_num_updated_lines(self.poMapInfo)
  705. Vect_reset_line(self.poPoints)
  706. Vect_append_point(self.poPoints, point[0], point[1], 0.0)
  707. moved = Vedit_move_vertex(self.poMapInfo, self.popoBgMapInfo, int(self.poBgMapInfo is not None),
  708. poList, self.poPoints,
  709. self._display.GetThreshold(type = 'selectThresh'),
  710. self._display.GetThreshold(),
  711. move[0], move[1], 0.0,
  712. 1, self._getSnapMode())
  713. Vect_destroy_list(poList)
  714. if moved > 0 and self.emit_signals:
  715. n_up_lines = Vect_get_num_updated_lines(self.poMapInfo)
  716. new_bboxs = []
  717. new_areas_cats = []
  718. for i in range(n_up_lines_old, n_up_lines):
  719. new_id = Vect_get_updated_line(self.poMapInfo, i)
  720. new_bboxs.append(self._getBbox(new_id))
  721. new_areas_cats.append(self._getLineAreasCategories(new_id))
  722. if moved > 0 and self._settings['breakLines']:
  723. self._breakLineAtIntersection(Vect_get_num_lines(self.poMapInfo),
  724. None)
  725. if moved > 0:
  726. self._addChangeset()
  727. self.toolbar.EnableUndo()
  728. if self.emit_signals:
  729. self.vertexMoved.emit(new_bboxs=new_bboxs,
  730. new_areas_cats=new_areas_cats,
  731. old_areas_cats=old_areas_cats,
  732. old_bboxs=old_bboxs)
  733. return moved
  734. def AddVertex(self, coords):
  735. """Add new vertex to the selected line/boundary on position 'coords'
  736. :param coords: coordinates to add vertex
  737. :return: id of new feature
  738. :return: 0 nothing changed
  739. :return: -1 on failure
  740. """
  741. added = self._ModifyLineVertex(coords, add = True)
  742. if added > 0:
  743. self.toolbar.EnableUndo()
  744. return added
  745. def RemoveVertex(self, coords):
  746. """Remove vertex from the selected line/boundary on position 'coords'
  747. :param coords: coordinates to remove vertex
  748. :return: id of new feature
  749. :return: 0 nothing changed
  750. :return: -1 on failure
  751. """
  752. deleted = self._ModifyLineVertex(coords, add = False)
  753. if deleted > 0:
  754. self.toolbar.EnableUndo()
  755. return deleted
  756. def SplitLine(self, point):
  757. """Split/break selected line/boundary on given position
  758. :param point: point where to split line
  759. :return: 1 line modified
  760. :return: 0 nothing changed
  761. :return: -1 error
  762. """
  763. thresh = self._display.GetThreshold('selectThresh')
  764. if not self._checkMap():
  765. return -1
  766. poList = self._display.GetSelectedIList()
  767. Vect_reset_line(self.poPoints)
  768. Vect_append_point(self.poPoints, point[0], point[1], 0.0)
  769. ret = Vedit_split_lines(self.poMapInfo, poList,
  770. self.poPoints, thresh, None)
  771. Vect_destroy_list(poList)
  772. if ret > 0:
  773. self._addChangeset()
  774. self.toolbar.EnableUndo()
  775. return ret
  776. def EditLine(self, line, coords):
  777. """Edit existing line/boundary
  778. :param line: feature id to be modified
  779. :param coords: list of coordinates of modified line
  780. :return: feature id of new line
  781. :return: -1 on error
  782. """
  783. if not self._checkMap():
  784. return -1
  785. if len(coords) < 2:
  786. self.DeleteSelectedLines()
  787. return 0
  788. if not Vect_line_alive(self.poMapInfo, line):
  789. self._error.DeadLine(line)
  790. return -1
  791. # read original feature
  792. ltype = Vect_read_line(self.poMapInfo, None, self.poCats, line)
  793. if ltype < 0:
  794. self._error.ReadLine(line)
  795. return -1
  796. if self.emit_signals:
  797. old_bboxs = [self._getBbox(line)]
  798. old_areas_cats = [self._getLineAreasCategories(line)]
  799. # build feature geometry
  800. Vect_reset_line(self.poPoints)
  801. for p in coords:
  802. Vect_append_point(self.poPoints, p[0], p[1], 0.0)
  803. # apply snapping (node or vertex)
  804. snap = self._getSnapMode()
  805. if snap != NO_SNAP:
  806. modeSnap = not (snap == SNAP)
  807. Vedit_snap_line(self.poMapInfo, self.popoBgMapInfo,
  808. int(self.poBgMapInfo is not None),
  809. -1, self.poPoints, self._display.GetThreshold(), modeSnap)
  810. newline = Vect_rewrite_line(self.poMapInfo, line, ltype,
  811. self.poPoints, self.poCats)
  812. if newline > 0 and self.emit_signals:
  813. new_geom = [self._getBbox(newline)]
  814. new_areas_cats = [self._getLineAreasCategories(newline)]
  815. if newline > 0 and self._settings['breakLines']:
  816. self._breakLineAtIntersection(newline, None)
  817. if newline > 0:
  818. self._addChangeset()
  819. self.toolbar.EnableUndo()
  820. if self.emit_signals:
  821. self.lineEdited.emit(old_bboxs=old_bboxs,
  822. old_areas_cats=old_areas_cats,
  823. new_bboxs=new_bboxs,
  824. new_areas_cats=new_areas_cats)
  825. return newline
  826. def FlipLine(self):
  827. """Flip selected lines/boundaries
  828. :return: number of modified lines
  829. :return: -1 on error
  830. """
  831. if not self._checkMap():
  832. return -1
  833. nlines = Vect_get_num_lines(self.poMapInfo)
  834. poList = self._display.GetSelectedIList()
  835. ret = Vedit_flip_lines(self.poMapInfo, poList)
  836. Vect_destroy_list(poList)
  837. if ret > 0:
  838. self._addChangeset()
  839. self.toolbar.EnableUndo()
  840. return ret
  841. def MergeLine(self):
  842. """Merge selected lines/boundaries
  843. :return: number of modified lines
  844. :return: -1 on error
  845. """
  846. if not self._checkMap():
  847. return -1
  848. poList = self._display.GetSelectedIList()
  849. ret = Vedit_merge_lines(self.poMapInfo, poList)
  850. Vect_destroy_list(poList)
  851. if ret > 0:
  852. self._addChangeset()
  853. self.toolbar.EnableUndo()
  854. return ret
  855. def BreakLine(self):
  856. """Break selected lines/boundaries
  857. :return: number of modified lines
  858. :return: -1 on error
  859. """
  860. if not self._checkMap():
  861. return -1
  862. poList = self._display.GetSelectedIList()
  863. ret = Vect_break_lines_list(self.poMapInfo, poList, None,
  864. GV_LINES, None)
  865. Vect_destroy_list(poList)
  866. if ret > 0:
  867. self._addChangeset()
  868. self.toolbar.EnableUndo()
  869. return ret
  870. def SnapLine(self):
  871. """Snap selected lines/boundaries
  872. :return: on success
  873. :return: -1 on error
  874. """
  875. if not self._checkMap():
  876. return -1
  877. nlines = Vect_get_num_lines(self.poMapInfo)
  878. poList = self._display.GetSelectedIList()
  879. Vect_snap_lines_list(self.poMapInfo, poList,
  880. self._display.GetThreshold(), None)
  881. Vect_destroy_list(poList)
  882. if nlines < Vect_get_num_lines(self.poMapInfo):
  883. self._addChangeset()
  884. self.toolbar.EnableUndo()
  885. def ConnectLine(self):
  886. """Connect selected lines/boundaries
  887. :return: 1 lines connected
  888. :return: 0 lines not connected
  889. :return: -1 on error
  890. """
  891. if not self._checkMap():
  892. return -1
  893. poList = self._display.GetSelectedIList()
  894. ret = Vedit_connect_lines(self.poMapInfo, poList,
  895. self._display.GetThreshold())
  896. Vect_destroy_list(poList)
  897. if ret > 0:
  898. self._addChangeset()
  899. self.toolbar.EnableUndo()
  900. return ret
  901. def CopyLine(self, ids = []):
  902. """Copy features from (background) vector map
  903. :param ids: list of line ids to be copied
  904. :return: number of copied features
  905. :return: -1 on error
  906. """
  907. if not self._checkMap():
  908. return -1
  909. nlines = Vect_get_num_lines(self.poMapInfo)
  910. poList = self._display.GetSelectedIList(ids)
  911. ret = Vedit_copy_lines(self.poMapInfo, self.poBgMapInfo,
  912. poList)
  913. Vect_destroy_list(poList)
  914. if ret > 0 and self.poBgMapInfo and self._settings['breakLines']:
  915. for i in range(1, ret):
  916. self._breakLineAtIntersection(nlines + i, None)
  917. if ret > 0:
  918. self._addChangeset()
  919. self.toolbar.EnableUndo()
  920. return ret
  921. def CopyCats(self, fromId, toId, copyAttrb = False):
  922. """Copy given categories to objects with id listed in ids
  923. :param cats: ids of 'from' feature
  924. :param ids: ids of 'to' feature(s)
  925. :return: number of modified features
  926. :return: -1 on error
  927. """
  928. if len(fromId) < 1 or len(toId) < 1:
  929. return 0
  930. poCatsFrom = self.poCats
  931. poCatsTo = Vect_new_cats_struct();
  932. nlines = 0
  933. for fline in fromId:
  934. if not Vect_line_alive(self.poMapInfo, fline):
  935. continue
  936. if Vect_read_line(self.poMapInfo, None, poCatsFrom, fline) < 0:
  937. self._error.ReadLine(fline)
  938. return -1
  939. for tline in toId:
  940. if not Vect_line_alive(self.poMapInfo, tline):
  941. continue
  942. ltype = Vect_read_line(self.poMapInfo, self.poPoints, poCatsTo, tline)
  943. if ltype < 0:
  944. self._error.ReadLine(fline)
  945. return -1
  946. catsFrom = poCatsFrom.contents
  947. for i in range(catsFrom.n_cats):
  948. if not copyAttrb:
  949. # duplicate category
  950. cat = catsFrom.cat[i]
  951. else:
  952. # duplicate attributes
  953. cat = self.cats[catsFrom.field[i]] + 1
  954. self.cats[catsFrom.field[i]] = cat
  955. poFi = Vect_get_field(self.poMapInfo, catsFrom.field[i])
  956. if not poFi:
  957. self._error.DbLink(i)
  958. return -1
  959. fi = poFi.contents
  960. driver = db_start_driver(fi.driver)
  961. if not driver:
  962. self._error.Driver(fi.driver)
  963. return -1
  964. handle = dbHandle()
  965. db_init_handle(byref(handle))
  966. db_set_handle(byref(handle), fi.database, None)
  967. if db_open_database(driver, byref(handle)) != DB_OK:
  968. db_shutdown_driver(driver)
  969. self._error.Database(fi.driver, fi.database)
  970. return -1
  971. stmt = dbString()
  972. db_init_string(byref(stmt))
  973. db_set_string(byref(stmt),
  974. "SELECT * FROM %s WHERE %s=%d" % (fi.table, fi.key,
  975. catsFrom.cat[i]))
  976. cursor = dbCursor()
  977. if db_open_select_cursor(driver, byref(stmt), byref(cursor),
  978. DB_SEQUENTIAL) != DB_OK:
  979. db_close_database_shutdown_driver(driver)
  980. return -1
  981. table = db_get_cursor_table(byref(cursor))
  982. ncols = db_get_table_number_of_columns(table)
  983. sql = "INSERT INTO %s VALUES (" % fi.table
  984. # fetch the data
  985. more = c_int()
  986. while True:
  987. if db_fetch(byref(cursor), DB_NEXT, byref(more)) != DB_OK:
  988. db_close_database_shutdown_driver(driver)
  989. return -1
  990. if not more.value:
  991. break
  992. value_string = dbString()
  993. for col in range(ncols):
  994. if col > 0:
  995. sql += ","
  996. column = db_get_table_column(table, col)
  997. if db_get_column_name(column) == fi.key:
  998. sql += "%d" % cat
  999. continue
  1000. value = db_get_column_value(column)
  1001. db_convert_column_value_to_string(column, byref(value_string))
  1002. if db_test_value_isnull(value):
  1003. sql += "NULL"
  1004. else:
  1005. ctype = db_sqltype_to_Ctype(db_get_column_sqltype(column))
  1006. if ctype != DB_C_TYPE_STRING:
  1007. sql += db_get_string(byref(value_string))
  1008. else:
  1009. sql += "'%s'" % db_get_string(byref(value_string))
  1010. sql += ")"
  1011. db_set_string(byref(stmt), sql)
  1012. if db_execute_immediate(driver, byref(stmt)) != DB_OK:
  1013. db_close_database_shutdown_driver(driver)
  1014. return -1
  1015. db_close_database_shutdown_driver(driver)
  1016. G_free(poFi)
  1017. if Vect_cat_set(poCatsTo, catsFrom.field[i], cat) < 1:
  1018. continue
  1019. if Vect_rewrite_line(self.poMapInfo, tline, ltype, self.poPoints, poCatsTo) < 0:
  1020. self._error.WriteLine()
  1021. return -1
  1022. nlines +=1
  1023. Vect_destroy_cats_struct(poCatsTo)
  1024. if nlines > 0:
  1025. self.toolbar.EnableUndo()
  1026. return nlines
  1027. def _selectLinesByQueryThresh(self):
  1028. """Generic method used for SelectLinesByQuery() -- to get
  1029. threshold value"""
  1030. thresh = 0.0
  1031. if UserSettings.Get(group = 'vdigit', key = 'query', subkey = 'selection') == 0:
  1032. thresh = UserSettings.Get(group = 'vdigit', key = 'queryLength', subkey = 'thresh')
  1033. if UserSettings.Get(group = 'vdigit', key = "queryLength", subkey = 'than-selection') == 0:
  1034. thresh = -1 * thresh
  1035. else:
  1036. thresh = UserSettings.Get(group = 'vdigit', key = 'queryDangle', subkey = 'thresh')
  1037. if UserSettings.Get(group = 'vdigit', key = "queryDangle", subkey = 'than-selection') == 0:
  1038. thresh = -1 * thresh
  1039. return thresh
  1040. def SelectLinesByQuery(self, bbox):
  1041. """Select features by query
  1042. .. todo::
  1043. layer / 3D
  1044. :param bbox: bounding box definition
  1045. """
  1046. if not self._checkMap():
  1047. return -1
  1048. thresh = self._selectLinesByQueryThresh()
  1049. query = QUERY_UNKNOWN
  1050. if UserSettings.Get(group = 'vdigit', key = 'query', subkey = 'selection') == 0:
  1051. query = QUERY_LENGTH
  1052. else:
  1053. query = QUERY_DANGLE
  1054. ftype = GV_POINTS | GV_LINES # TODO: 3D
  1055. layer = 1 # TODO
  1056. ids = list()
  1057. poList = Vect_new_list()
  1058. coList = poList.contents
  1059. if UserSettings.Get(group = 'vdigit', key = 'query', subkey = 'box'):
  1060. Vect_reset_line(self.poPoints)
  1061. x1, y1 = bbox[0]
  1062. x2, y2 = bbox[1]
  1063. z1 = z2 = 0.0
  1064. Vect_append_point(self.poPoints, x1, y1, z1)
  1065. Vect_append_point(self.poPoints, x2, y1, z2)
  1066. Vect_append_point(self.poPoints, x2, y2, z1)
  1067. Vect_append_point(self.poPoints, x1, y2, z2)
  1068. Vect_append_point(self.poPoints, x1, y1, z1)
  1069. Vect_select_lines_by_polygon(self.poMapInfo, self.poPoints, 0, None,
  1070. ftype, poList)
  1071. if coList.n_values == 0:
  1072. return ids
  1073. Vedit_select_by_query(self.poMapInfo,
  1074. ftype, layer, thresh, query,
  1075. poList)
  1076. for i in range(coList.n_values):
  1077. ids.append(int(coList.value[i]))
  1078. Debug.msg(3, "IVDigit.SelectLinesByQuery(): lines=%d", coList.n_values)
  1079. Vect_destroy_list(poList)
  1080. return ids
  1081. def IsVector3D(self):
  1082. """Check if open vector map is 3D
  1083. """
  1084. if not self._checkMap():
  1085. return False
  1086. return Vect_is_3d(self.poMapInfo)
  1087. def GetLineLength(self, line):
  1088. """Get line length
  1089. :param line: feature id
  1090. :return: line length
  1091. :return: -1 on error
  1092. """
  1093. if not self._checkMap():
  1094. return -1
  1095. if not Vect_line_alive(self.poMapInfo, line):
  1096. return -1
  1097. ltype = Vect_read_line(self.poMapInfo, self.poPoints, None, line)
  1098. if ltype < 0:
  1099. self._error.ReadLine(line)
  1100. return ret
  1101. length = -1
  1102. if ltype & GV_LINES: # lines & boundaries
  1103. length = Vect_line_length(self.poPoints)
  1104. return length
  1105. def GetAreaSize(self, centroid):
  1106. """Get area size
  1107. :param centroid: centroid id
  1108. :return: area size
  1109. :return: -1 on error
  1110. """
  1111. if not self._checkMap():
  1112. return -1
  1113. ltype = Vect_read_line(self.poMapInfo, None, None, centroid)
  1114. if ltype < 0:
  1115. self._error.ReadLine(line)
  1116. return ret
  1117. if ltype != GV_CENTROID:
  1118. return -1
  1119. area = Vect_get_centroid_area(self.poMapInfo, centroid)
  1120. size = -1
  1121. if area > 0:
  1122. if not Vect_area_alive(self.poMapInfo, area):
  1123. return size
  1124. size = Vect_get_area_area(self.poMapInfo, area)
  1125. return size
  1126. def GetAreaPerimeter(self, centroid):
  1127. """Get area perimeter
  1128. :param centroid: centroid id
  1129. :return: area size
  1130. :return: -1 on error
  1131. """
  1132. if not self._checkMap():
  1133. return -1
  1134. ltype = Vect_read_line(self.poMapInfo, None, None, centroid)
  1135. if ltype < 0:
  1136. self._error.ReadLine(line)
  1137. return ret
  1138. if ltype != GV_CENTROID:
  1139. return -1
  1140. area = Vect_get_centroid_area(self.poMapInfo, centroid)
  1141. perimeter = -1
  1142. if area > 0:
  1143. if not Vect_area_alive(self.poMapInfo, area):
  1144. return -1
  1145. Vect_get_area_points(self.poMapInfo, area, self.poPoints)
  1146. perimeter = Vect_area_perimeter(self.poPoints)
  1147. return perimeter
  1148. def SetLineCats(self, line, layer, cats, add = True):
  1149. """Set categories for given line and layer
  1150. :param line: feature id
  1151. :param layer: layer number (-1 for first selected line)
  1152. :param cats: list of categories
  1153. :param add: if True to add, otherwise do delete categories
  1154. :return: new feature id (feature need to be rewritten)
  1155. :return: -1 on error
  1156. """
  1157. if not self._checkMap():
  1158. return -1
  1159. if line < 1 and len(self._display.selected['ids']) < 1:
  1160. return -1
  1161. update = False
  1162. if line == -1:
  1163. update = True
  1164. line = self._display.selected['ids'][0]
  1165. if not Vect_line_alive(self.poMapInfo, line):
  1166. return -1
  1167. ltype = Vect_read_line(self.poMapInfo, self.poPoints, self.poCats, line)
  1168. if ltype < 0:
  1169. self._error.ReadLine(line)
  1170. return -1
  1171. for c in cats:
  1172. if add:
  1173. Vect_cat_set(self.poCats, layer, c)
  1174. else:
  1175. Vect_field_cat_del(self.poCats, layer, c)
  1176. newline = Vect_rewrite_line(self.poMapInfo, line, ltype,
  1177. self.poPoints, self.poCats)
  1178. if newline > 0:
  1179. self._addChangeset()
  1180. self.toolbar.EnableUndo()
  1181. if update:
  1182. # update line id since the line was rewritten
  1183. self._display.selected['ids'][0] = newline
  1184. return newline
  1185. def TypeConvForSelectedLines(self):
  1186. """Feature type conversion for selected objects.
  1187. Supported conversions:
  1188. - point <-> centroid
  1189. - line <-> boundary
  1190. :return: number of modified features
  1191. :return: -1 on error
  1192. """
  1193. if not self._checkMap():
  1194. return -1
  1195. poList = self._display.GetSelectedIList()
  1196. ret = Vedit_chtype_lines(self.poMapInfo, poList)
  1197. Vect_destroy_list(poList)
  1198. if ret > 0:
  1199. self._addChangeset()
  1200. self.toolbar.EnableUndo()
  1201. return ret
  1202. def Undo(self, level = -1):
  1203. """Undo action
  1204. :param level: levels to undo (0 to revert all)
  1205. :return: id of current changeset
  1206. """
  1207. changesetLast = len(self.changesets) - 1
  1208. if changesetLast < 0:
  1209. return changesetLast
  1210. if level < 0 and self.changesetCurrent > changesetLast:
  1211. self.changesetCurrent = changesetLast
  1212. elif level == 0:
  1213. # 0 -> undo all
  1214. level = -1 * self.changesetCurrent - 1
  1215. Debug.msg(2, "Digit.Undo(): changeset_last=%d, changeset_current=%d, level=%d",
  1216. changesetLast, self.changesetCurrent, level)
  1217. if level < 0: # undo
  1218. if self.changesetCurrent + level < -1:
  1219. return self.changesetCurrent
  1220. for changeset in range(self.changesetCurrent, self.changesetCurrent + level, -1):
  1221. self._applyChangeset(changeset, undo = True)
  1222. elif level > 0: # redo
  1223. if self.changesetCurrent + 1 > changesetLast:
  1224. return self.changesetCurrent
  1225. for changeset in range(self.changesetCurrent + 1, self.changesetCurrent + 1 + level):
  1226. self._applyChangeset(changeset, undo = False)
  1227. self.changesetCurrent += level
  1228. Debug.msg(2, "Digit.Undo(): changeset_current=%d, changeset_last=%d",
  1229. self.changesetCurrent, changesetLast)
  1230. self.mapWindow.UpdateMap(render = False)
  1231. if self.changesetCurrent < 0: # disable undo tool
  1232. self.toolbar.EnableUndo(False)
  1233. else:
  1234. self.toolbar.EnableUndo(True)
  1235. if self.changesetCurrent < changesetLast:
  1236. self.toolbar.EnableRedo(True)
  1237. else:
  1238. self.toolbar.EnableRedo(False)
  1239. def ZBulkLines(self, pos1, pos2, start, step):
  1240. """Z-bulk labeling
  1241. :param pos1: reference line (start point)
  1242. :param pos1: reference line (end point)
  1243. :param start: starting value
  1244. :param step: step value
  1245. :return: number of modified lines
  1246. :return: -1 on error
  1247. """
  1248. if not self._checkMap():
  1249. return -1
  1250. poList = self._display.GetSelectedIList()
  1251. ret = Vedit_bulk_labeling(self.poMapInfo, poList,
  1252. pos1[0], pos1[1], pos2[0], pos2[1],
  1253. start, step)
  1254. Vect_destroy_list(poList)
  1255. if ret > 0:
  1256. self._addChangeset()
  1257. self.toolbar.EnableUndo()
  1258. return ret
  1259. def GetDisplay(self):
  1260. """Get display driver instance"""
  1261. return self._display
  1262. def OpenMap(self, name, update=True, tmp = False):
  1263. """Open vector map for editing
  1264. :param map: name of vector map to be set up
  1265. :type map: str
  1266. :param tmp: True to open temporary vector map
  1267. """
  1268. Debug.msg (3, "AbstractDigit.SetMapName map=%s" % name)
  1269. if '@' in name:
  1270. name, mapset = name.split('@')
  1271. else:
  1272. mapset = grass.gisenv()['MAPSET']
  1273. self.poMapInfo = self._display.OpenMap(str(name), str(mapset), update, tmp)
  1274. if self.poMapInfo:
  1275. self.InitCats()
  1276. return self.poMapInfo
  1277. def CloseMap(self):
  1278. """Close currently open vector map
  1279. """
  1280. if not self._checkMap():
  1281. return
  1282. # print extra line before building message
  1283. sys.stdout.write(os.linesep)
  1284. # build topology, close map
  1285. self._display.CloseMap()
  1286. def InitCats(self):
  1287. """Initialize categories information
  1288. :return: 0 on success
  1289. :return: -1 on error
  1290. """
  1291. self.cats.clear()
  1292. if not self._checkMap():
  1293. return -1
  1294. ndblinks = Vect_get_num_dblinks(self.poMapInfo)
  1295. for i in range(ndblinks):
  1296. fi = Vect_get_dblink(self.poMapInfo, i).contents
  1297. if fi:
  1298. self.cats[fi.number] = None
  1299. # find max category
  1300. nfields = Vect_cidx_get_num_fields(self.poMapInfo)
  1301. Debug.msg(2, "wxDigit.InitCats(): nfields=%d", nfields)
  1302. for i in range(nfields):
  1303. field = Vect_cidx_get_field_number(self.poMapInfo, i)
  1304. ncats = Vect_cidx_get_num_cats_by_index(self.poMapInfo, i)
  1305. if field <= 0:
  1306. continue
  1307. for j in range(ncats):
  1308. cat = c_int()
  1309. type = c_int()
  1310. id = c_int()
  1311. Vect_cidx_get_cat_by_index(self.poMapInfo, i, j,
  1312. byref(cat), byref(type), byref(id))
  1313. if field in self.cats:
  1314. if cat > self.cats[field]:
  1315. self.cats[field] = cat.value
  1316. else:
  1317. self.cats[field] = cat.value
  1318. Debug.msg(3, "wxDigit.InitCats(): layer=%d, cat=%d", field, self.cats[field])
  1319. # set default values
  1320. for field, cat in self.cats.iteritems():
  1321. if cat == None:
  1322. self.cats[field] = 0 # first category 1
  1323. Debug.msg(3, "wxDigit.InitCats(): layer=%d, cat=%d", field, self.cats[field])
  1324. def _checkMap(self):
  1325. """Check if map is open
  1326. """
  1327. if not self.poMapInfo:
  1328. self._error.NoMap()
  1329. return False
  1330. return True
  1331. def _addFeature(self, ftype, coords, layer, cat, snap, threshold):
  1332. """Add new feature(s) to the vector map
  1333. :param ftype: feature type (GV_POINT, GV_LINE, GV_BOUNDARY, ...)
  1334. :param coords: tuple of coordinates ((x, y), (x, y), ...)
  1335. :param layer: layer number (-1 for no cat)
  1336. :param cat: category number
  1337. :param snap: snap to node/vertex
  1338. :param threshold: threshold for snapping
  1339. :return: tuple (number of added features, list of fids)
  1340. :return: number of features -1 on error
  1341. """
  1342. fids = list()
  1343. if not self._checkMap():
  1344. return (-1, None)
  1345. is3D = bool(Vect_is_3d(self.poMapInfo))
  1346. Debug.msg(2, "IVDigit._addFeature(): npoints=%d, layer=%d, cat=%d, snap=%d",
  1347. len(coords), layer, cat, snap)
  1348. if not (ftype & (GV_POINTS | GV_LINES | GV_AREA)): # TODO: 3D
  1349. self._error.FeatureType(ftype)
  1350. return (-1, None)
  1351. # set category
  1352. Vect_reset_cats(self.poCats)
  1353. if layer > 0 and ftype != GV_AREA:
  1354. Vect_cat_set(self.poCats, layer, cat)
  1355. self.cats[layer] = max(cat, self.cats.get(layer, 1))
  1356. # append points
  1357. Vect_reset_line(self.poPoints)
  1358. for c in coords:
  1359. Vect_append_point(self.poPoints, c[0], c[1], 0.0)
  1360. if ftype & (GV_BOUNDARY | GV_AREA):
  1361. # close boundary
  1362. points = self.poPoints.contents
  1363. last = points.n_points - 1
  1364. if self._settings['closeBoundary']:
  1365. Vect_append_point(self.poPoints, points.x[0], points.y[0], points.z[0])
  1366. elif Vect_points_distance(points.x[0], points.y[0], points.z[0],
  1367. points.x[last], points.y[last], points.z[last],
  1368. is3D) <= threshold:
  1369. points.x[last] = points.x[0]
  1370. points.y[last] = points.y[0]
  1371. points.z[last] = points.z[0]
  1372. if snap != NO_SNAP:
  1373. # apply snapping (node or vertex)
  1374. modeSnap = not (snap == SNAP)
  1375. Vedit_snap_line(self.poMapInfo, self.popoBgMapInfo, int(self.poBgMapInfo is not None),
  1376. -1, self.poPoints, threshold, modeSnap)
  1377. if ftype == GV_AREA:
  1378. ltype = GV_BOUNDARY
  1379. else:
  1380. ltype = ftype
  1381. newline = Vect_write_line(self.poMapInfo, ltype, self.poPoints, self.poCats)
  1382. if newline < 0:
  1383. self._error.WriteLine()
  1384. return (-1, None)
  1385. fids.append(newline)
  1386. # add centroids for left/right area
  1387. if ftype & GV_AREA:
  1388. left = right = -1
  1389. bpoints = Vect_new_line_struct()
  1390. cleft = c_int()
  1391. cright = c_int()
  1392. Vect_get_line_areas(self.poMapInfo, newline,
  1393. byref(cleft), byref(cright))
  1394. left = cleft.value
  1395. right = cright.value
  1396. Debug.msg(3, "IVDigit._addFeature(): area - left=%d right=%d",
  1397. left, right)
  1398. # check if area exists and has no centroid inside
  1399. if layer > 0 and (left > 0 or right > 0):
  1400. Vect_cat_set(self.poCats, layer, cat)
  1401. self.cats[layer] = max(cat, self.cats.get(layer, 0))
  1402. x = c_double()
  1403. y = c_double()
  1404. if left > 0 and \
  1405. Vect_get_area_centroid(self.poMapInfo, left) == 0:
  1406. # if Vect_get_area_points(self.poMapInfo, left, bpoints) > 0 and
  1407. # Vect_find_poly_centroid(bpoints, byref(x), byref(y)) == 0:
  1408. if Vect_get_point_in_area(self.poMapInfo, left, byref(x), byref(y)) == 0:
  1409. Vect_reset_line(bpoints)
  1410. Vect_append_point(bpoints, x.value, y.value, 0.0)
  1411. newc = Vect_write_line(self.poMapInfo, GV_CENTROID,
  1412. bpoints, self.poCats)
  1413. if newc < 0:
  1414. self._error.WriteLine()
  1415. return (len(fids), fids)
  1416. else:
  1417. fids.append(newc)
  1418. if right > 0 and \
  1419. Vect_get_area_centroid(self.poMapInfo, right) == 0:
  1420. # if Vect_get_area_points(byref(self.poMapInfo), right, bpoints) > 0 and
  1421. # Vect_find_poly_centroid(bpoints, byref(x), byref(y)) == 0:
  1422. if Vect_get_point_in_area(self.poMapInfo, right, byref(x), byref(y)) == 0:
  1423. Vect_reset_line(bpoints)
  1424. Vect_append_point(bpoints, x.value, y.value, 0.0)
  1425. newc = Vect_write_line(self.poMapInfo, GV_CENTROID,
  1426. bpoints, self.poCats)
  1427. if newc < 0:
  1428. self._error.WriteLine()
  1429. return (len(fids, fids))
  1430. else:
  1431. fids.append(newc)
  1432. Vect_destroy_line_struct(bpoints)
  1433. # break line or boundary at intersection
  1434. if self._settings['breakLines']:
  1435. self._breakLineAtIntersection(newline, self.poPoints)
  1436. self._addChangeset()
  1437. if ftype & GV_AREA:
  1438. # len(fids) == 1 -> no new area
  1439. return (len(fids)-1, fids)
  1440. return (len(fids), fids)
  1441. def _ModifyLineVertex(self, coords, add = True):
  1442. """Add or remove vertex
  1443. Shape of line/boundary is not changed when adding new vertex.
  1444. :param coords: coordinates of point
  1445. :param add: True to add, False to remove
  1446. :type add: bool
  1447. :return: 1 on success
  1448. :return: 0 nothing changed
  1449. :return: -1 error
  1450. """
  1451. if not self._checkMap():
  1452. return -1
  1453. selected = self._display.selected
  1454. if len(selected['ids']) != 1:
  1455. return 0
  1456. poList = self._display.GetSelectedIList()
  1457. if self.emit_signals:
  1458. cList = poList.contents
  1459. old_bboxs = [self._getBbox(cList.value[0])]
  1460. old_areas_cats = [self._getLineAreasCategories(cList.value[0])]
  1461. Vect_set_updated(self.poMapInfo, 1)
  1462. n_up_lines_old = Vect_get_num_updated_lines(self.poMapInfo)
  1463. Vect_reset_line(self.poPoints)
  1464. Vect_append_point(self.poPoints, coords[0], coords[1], 0.0)
  1465. thresh = self._display.GetThreshold(type = 'selectThresh')
  1466. if add:
  1467. ret = Vedit_add_vertex(self.poMapInfo, poList,
  1468. self.poPoints, thresh)
  1469. else:
  1470. ret = Vedit_remove_vertex(self.poMapInfo, poList,
  1471. self.poPoints, thresh)
  1472. Vect_destroy_list(poList)
  1473. if ret > 0 and self.emit_signals:
  1474. new_bboxs = []
  1475. new_areas_cats = []
  1476. n_up_lines = Vect_get_num_updated_lines(self.poMapInfo)
  1477. for i in range(n_up_lines_old, n_up_lines):
  1478. new_id = Vect_get_updated_line(self.poMapInfo, i)
  1479. new_areas_cats.append(self._getLineAreasCategories(new_id))
  1480. new_bboxs.append(self._getBbox(new_id))
  1481. if not add and ret > 0 and self._settings['breakLines']:
  1482. self._breakLineAtIntersection(Vect_get_num_lines(self.poMapInfo),
  1483. None)
  1484. if ret > 0:
  1485. self._addChangeset()
  1486. if ret > 0 and self.emit_signals:
  1487. if add:
  1488. self.vertexAdded.emit(old_bboxs=old_bboxs, new_bboxs=new_bboxs)
  1489. else:
  1490. self.vertexRemoved.emit(old_bboxs=old_bboxs,
  1491. new_bboxs=new_bboxs,
  1492. old_areas_cats=old_areas_cats,
  1493. new_areas_cats=new_areas_cats)
  1494. return 1
  1495. def GetLineCats(self, line):
  1496. """Get list of layer/category(ies) for selected feature.
  1497. :param line: feature id (-1 for first selected feature)
  1498. :return: list of layer/cats
  1499. """
  1500. ret = dict()
  1501. if not self._checkMap():
  1502. return ret
  1503. if line == -1 and len(self._display.selected['ids']) < 1:
  1504. return ret
  1505. if line == -1:
  1506. line = self._display.selected['ids'][0]
  1507. if not Vect_line_alive(self.poMapInfo, line):
  1508. self._error.DeadLine(line)
  1509. return ret
  1510. if Vect_read_line(self.poMapInfo, None, self.poCats, line) < 0:
  1511. self._error.ReadLine(line)
  1512. return ret
  1513. cats = self.poCats.contents
  1514. for i in range(cats.n_cats):
  1515. field = cats.field[i]
  1516. if field not in ret:
  1517. ret[field] = list()
  1518. ret[field].append(cats.cat[i])
  1519. return ret
  1520. def GetLayers(self):
  1521. """Get list of layers
  1522. Requires self.InitCats() to be called.
  1523. :return: list of layers
  1524. """
  1525. return self.cats.keys()
  1526. def UpdateSettings(self):
  1527. """Update digit (and display) settings
  1528. """
  1529. self._display.UpdateSettings()
  1530. self._settings['breakLines'] = bool(UserSettings.Get(group = 'vdigit', key = "breakLines",
  1531. subkey = 'enabled'))
  1532. self._settings['closeBoundary'] = bool(UserSettings.Get(group = 'vdigit', key = "closeBoundary",
  1533. subkey = 'enabled'))
  1534. def SetCategory(self):
  1535. """Update self.cats based on settings"""
  1536. sel = UserSettings.Get(group = 'vdigit', key = 'categoryMode', subkey = 'selection')
  1537. cat = None
  1538. if sel == 0: # next to usep
  1539. cat = self._setCategoryNextToUse()
  1540. elif sel == 1:
  1541. cat = UserSettings.Get(group = 'vdigit', key = 'category', subkey = 'value')
  1542. if cat:
  1543. layer = UserSettings.Get(group = 'vdigit', key = 'layer', subkey = 'value')
  1544. self.cats[layer] = cat
  1545. return cat
  1546. def _setCategoryNextToUse(self):
  1547. """Find maximum category number for the given layer and
  1548. update the settings
  1549. :return: category to be used
  1550. """
  1551. # get max category number for given layer and update the settings
  1552. layer = UserSettings.Get(group = 'vdigit', key = 'layer', subkey = 'value')
  1553. cat = self.cats.get(layer, 0) + 1
  1554. UserSettings.Set(group = 'vdigit', key = 'category', subkey = 'value',
  1555. value = cat)
  1556. Debug.msg(1, "IVDigit._setCategoryNextToUse(): cat=%d", cat)
  1557. return cat
  1558. def SelectLinesFromBackgroundMap(self, bbox):
  1559. """Select features from background map
  1560. :param bbox: bounding box definition
  1561. :return: list of selected feature ids
  1562. """
  1563. # try select features by box first
  1564. if self._display.SelectLinesByBox(bbox, poMapInfo = self.poBgMapInfo) < 1:
  1565. self._display.SelectLineByPoint(bbox[0], poMapInfo = self.poBgMapInfo)['line']
  1566. return self._display.selected['ids']
  1567. def GetUndoLevel(self):
  1568. """Get undo level (number of active changesets)
  1569. Note: Changesets starts wiht 0
  1570. """
  1571. return self.changesetCurrent
  1572. def GetFeatureType(self):
  1573. """Get feature type for OGR layers
  1574. :return: feature type as string (point, linestring, polygon)
  1575. :return: None for native format
  1576. """
  1577. topoFormat = Vect_get_finfo_topology_info(self.poMapInfo, None, None, None)
  1578. if topoFormat == GV_TOPO_PSEUDO:
  1579. return Vect_get_finfo_geometry_type(self.poMapInfo)
  1580. return ''