dialogs.py 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783
  1. """
  2. @package vdigit.dialogs
  3. @brief wxGUI vector digitizer dialogs
  4. Classes:
  5. - dialogs::VDigitCategoryDialog
  6. - dialogs::CategoryListCtrl
  7. - dialogs::VDigitZBulkDialog
  8. - dialogs::VDigitDuplicatesDialog
  9. - dialogs::CheckListFeature
  10. (C) 2007-2011 by the GRASS Development Team
  11. This program is free software under the GNU General Public License
  12. (>=v2). Read the file COPYING that comes with GRASS for details.
  13. @author Martin Landa <landa.martin gmail.com>
  14. """
  15. import copy
  16. import six
  17. import wx
  18. import wx.lib.mixins.listctrl as listmix
  19. from core.gcmd import RunCommand, GError
  20. from core.debug import Debug
  21. from gui_core.wrap import (
  22. SpinCtrl,
  23. Button,
  24. StaticText,
  25. StaticBox,
  26. Menu,
  27. ListCtrl,
  28. NewId,
  29. CheckListCtrlMixin,
  30. )
  31. class VDigitCategoryDialog(wx.Dialog, listmix.ColumnSorterMixin):
  32. def __init__(
  33. self,
  34. parent,
  35. title,
  36. vectorName,
  37. query=None,
  38. cats=None,
  39. style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER,
  40. **kwargs,
  41. ):
  42. """Dialog used to display/modify categories of vector objects
  43. :param parent:
  44. :param title: dialog title
  45. :param query: {coordinates, qdist} - used by v.edit/v.what
  46. :param cats: directory of lines (layer/categories) - used by vdigit
  47. :param style: dialog style
  48. """
  49. self.parent = parent # map window class instance
  50. self.digit = parent.digit
  51. # map name
  52. self.vectorName = vectorName
  53. # line : {layer: [categories]}
  54. self.cats = {}
  55. # do not display dialog if no line is found (-> self.cats)
  56. if cats is None:
  57. if self._getCategories(query[0], query[1]) == 0 or not self.line:
  58. Debug.msg(3, "VDigitCategoryDialog(): nothing found!")
  59. else:
  60. self.cats = cats
  61. for line in cats.keys():
  62. for layer in cats[line].keys():
  63. self.cats[line][layer] = list(cats[line][layer])
  64. layers = []
  65. for layer in self.digit.GetLayers():
  66. layers.append(str(layer))
  67. # make copy of cats (used for 'reload')
  68. self.cats_orig = copy.deepcopy(self.cats)
  69. wx.Dialog.__init__(
  70. self, parent=self.parent, id=wx.ID_ANY, title=title, style=style, **kwargs
  71. )
  72. # list of categories
  73. box = StaticBox(
  74. parent=self,
  75. id=wx.ID_ANY,
  76. label=" %s " % _("List of categories - right-click to delete"),
  77. )
  78. listSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  79. self.list = CategoryListCtrl(
  80. parent=self,
  81. id=wx.ID_ANY,
  82. style=wx.LC_REPORT
  83. | wx.BORDER_NONE
  84. | wx.LC_SORT_ASCENDING
  85. | wx.LC_HRULES
  86. | wx.LC_VRULES,
  87. )
  88. # sorter
  89. self.fid = list(self.cats.keys())[0]
  90. self.itemDataMap = self.list.Populate(self.cats[self.fid])
  91. listmix.ColumnSorterMixin.__init__(self, 2)
  92. self.fidMulti = wx.Choice(parent=self, id=wx.ID_ANY, size=(150, -1))
  93. self.fidMulti.Bind(wx.EVT_CHOICE, self.OnFeature)
  94. self.fidText = StaticText(parent=self, id=wx.ID_ANY)
  95. if len(self.cats.keys()) == 1:
  96. self.fidMulti.Show(False)
  97. self.fidText.SetLabel(str(self.fid))
  98. else:
  99. self.fidText.Show(False)
  100. choices = []
  101. for fid in self.cats.keys():
  102. choices.append(str(fid))
  103. self.fidMulti.SetItems(choices)
  104. self.fidMulti.SetSelection(0)
  105. listSizer.Add(self.list, proportion=1, flag=wx.EXPAND)
  106. # add new category
  107. box = StaticBox(parent=self, id=wx.ID_ANY, label=" %s " % _("Add new category"))
  108. addSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  109. flexSizer = wx.FlexGridSizer(cols=5, hgap=5, vgap=5)
  110. flexSizer.AddGrowableCol(3)
  111. layerNewTxt = StaticText(parent=self, id=wx.ID_ANY, label="%s:" % _("Layer"))
  112. self.layerNew = wx.Choice(
  113. parent=self, id=wx.ID_ANY, size=(75, -1), choices=layers
  114. )
  115. if len(layers) > 0:
  116. self.layerNew.SetSelection(0)
  117. catNewTxt = StaticText(parent=self, id=wx.ID_ANY, label="%s:" % _("Category"))
  118. try:
  119. newCat = max(self.cats[self.fid][1]) + 1
  120. except KeyError:
  121. newCat = 1
  122. self.catNew = SpinCtrl(
  123. parent=self, id=wx.ID_ANY, size=(75, -1), initial=newCat, min=0, max=1e9
  124. )
  125. btnAddCat = Button(self, wx.ID_ADD)
  126. flexSizer.Add(
  127. layerNewTxt, proportion=0, flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL
  128. )
  129. flexSizer.Add(
  130. self.layerNew,
  131. proportion=0,
  132. flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL,
  133. )
  134. flexSizer.Add(
  135. catNewTxt,
  136. proportion=0,
  137. flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_RIGHT | wx.LEFT,
  138. border=10,
  139. )
  140. flexSizer.Add(
  141. self.catNew, proportion=0, flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL
  142. )
  143. flexSizer.Add(
  144. btnAddCat, proportion=0, flag=wx.EXPAND | wx.ALIGN_RIGHT | wx.FIXED_MINSIZE
  145. )
  146. addSizer.Add(flexSizer, proportion=1, flag=wx.ALL | wx.EXPAND, border=5)
  147. # buttons
  148. btnApply = Button(self, wx.ID_APPLY)
  149. btnApply.SetToolTip(_("Apply changes"))
  150. btnCancel = Button(self, wx.ID_CANCEL)
  151. btnCancel.SetToolTip(_("Ignore changes and close dialog"))
  152. btnOk = Button(self, wx.ID_OK)
  153. btnOk.SetToolTip(_("Apply changes and close dialog"))
  154. btnOk.SetDefault()
  155. # sizers
  156. btnSizer = wx.StdDialogButtonSizer()
  157. btnSizer.AddButton(btnCancel)
  158. # btnSizer.AddButton(btnReload)
  159. # btnSizer.SetNegativeButton(btnReload)
  160. btnSizer.AddButton(btnApply)
  161. btnSizer.AddButton(btnOk)
  162. btnSizer.Realize()
  163. mainSizer = wx.BoxSizer(wx.VERTICAL)
  164. mainSizer.Add(listSizer, proportion=1, flag=wx.EXPAND | wx.ALL, border=5)
  165. mainSizer.Add(
  166. addSizer,
  167. proportion=0,
  168. flag=wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM,
  169. border=5,
  170. )
  171. fidSizer = wx.BoxSizer(wx.HORIZONTAL)
  172. fidSizer.Add(
  173. StaticText(parent=self, id=wx.ID_ANY, label=_("Feature id:")),
  174. proportion=0,
  175. border=5,
  176. flag=wx.ALIGN_CENTER_VERTICAL,
  177. )
  178. fidSizer.Add(self.fidMulti, proportion=0, flag=wx.EXPAND | wx.ALL, border=5)
  179. fidSizer.Add(self.fidText, proportion=0, flag=wx.EXPAND | wx.ALL, border=5)
  180. mainSizer.Add(fidSizer, proportion=0, flag=wx.EXPAND | wx.ALL, border=5)
  181. mainSizer.Add(btnSizer, proportion=0, flag=wx.EXPAND | wx.ALL, border=5)
  182. self.SetSizer(mainSizer)
  183. mainSizer.Fit(self)
  184. self.SetAutoLayout(True)
  185. # set min size for dialog
  186. self.SetMinSize(self.GetBestSize())
  187. # bindings
  188. btnApply.Bind(wx.EVT_BUTTON, self.OnApply)
  189. btnOk.Bind(wx.EVT_BUTTON, self.OnOK)
  190. btnAddCat.Bind(wx.EVT_BUTTON, self.OnAddCat)
  191. btnCancel.Bind(wx.EVT_BUTTON, self.OnCancel)
  192. self.Bind(wx.EVT_CLOSE, lambda evt: self.Hide())
  193. # list
  194. self.list.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnRightUp) # wxMSW
  195. self.list.Bind(wx.EVT_RIGHT_UP, self.OnRightUp) # wxGTK
  196. self.Bind(wx.EVT_LIST_BEGIN_LABEL_EDIT, self.OnBeginEdit, self.list)
  197. self.Bind(wx.EVT_LIST_END_LABEL_EDIT, self.OnEndEdit, self.list)
  198. self.Bind(wx.EVT_LIST_COL_CLICK, self.OnColClick, self.list)
  199. def GetListCtrl(self):
  200. """Used by ColumnSorterMixin"""
  201. return self.list
  202. def OnColClick(self, event):
  203. """Click on column header (order by)"""
  204. event.Skip()
  205. def OnBeginEdit(self, event):
  206. """Editing of item started"""
  207. event.Allow()
  208. def OnEndEdit(self, event):
  209. """Finish editing of item"""
  210. itemIndex = event.GetIndex()
  211. layerOld = int(self.list.GetItem(itemIndex, 0).GetText())
  212. catOld = int(self.list.GetItem(itemIndex, 1).GetText())
  213. if event.GetColumn() == 0:
  214. layerNew = int(event.GetLabel())
  215. catNew = catOld
  216. else:
  217. layerNew = layerOld
  218. catNew = int(event.GetLabel())
  219. try:
  220. if layerNew not in self.cats[self.fid].keys():
  221. self.cats[self.fid][layerNew] = []
  222. self.cats[self.fid][layerNew].append(catNew)
  223. self.cats[self.fid][layerOld].remove(catOld)
  224. except:
  225. event.Veto()
  226. self.list.SetItem(itemIndex, 0, str(layerNew))
  227. self.list.SetItem(itemIndex, 1, str(catNew))
  228. dlg = wx.MessageDialog(
  229. self,
  230. _(
  231. "Unable to add new layer/category <%(layer)s/%(category)s>.\n"
  232. "Layer and category number must be integer.\n"
  233. "Layer number must be greater than zero."
  234. )
  235. % {
  236. "layer": self.layerNew.GetStringSelection(),
  237. "category": str(self.catNew.GetValue()),
  238. },
  239. _("Error"),
  240. wx.OK | wx.ICON_ERROR,
  241. )
  242. dlg.ShowModal()
  243. dlg.Destroy()
  244. return False
  245. def OnRightDown(self, event):
  246. """Mouse right button down"""
  247. x = event.GetX()
  248. y = event.GetY()
  249. item, flags = self.list.HitTest((x, y))
  250. if item != wx.NOT_FOUND and flags & wx.LIST_HITTEST_ONITEM:
  251. self.list.Select(item)
  252. event.Skip()
  253. def OnRightUp(self, event):
  254. """Mouse right button up"""
  255. if not hasattr(self, "popupID1"):
  256. self.popupID1 = NewId()
  257. self.popupID2 = NewId()
  258. self.popupID3 = NewId()
  259. self.Bind(wx.EVT_MENU, self.OnItemDelete, id=self.popupID1)
  260. self.Bind(wx.EVT_MENU, self.OnItemDeleteAll, id=self.popupID2)
  261. self.Bind(wx.EVT_MENU, self.OnReload, id=self.popupID3)
  262. # generate popup-menu
  263. menu = Menu()
  264. menu.Append(self.popupID1, _("Delete selected"))
  265. if self.list.GetFirstSelected() == -1:
  266. menu.Enable(self.popupID1, False)
  267. menu.Append(self.popupID2, _("Delete all"))
  268. menu.AppendSeparator()
  269. menu.Append(self.popupID3, _("Reload"))
  270. self.PopupMenu(menu)
  271. menu.Destroy()
  272. def OnItemSelected(self, event):
  273. """Item selected"""
  274. event.Skip()
  275. def OnItemDelete(self, event):
  276. """Delete selected item(s) from the list (layer/category pair)"""
  277. item = self.list.GetFirstSelected()
  278. while item != -1:
  279. layer = int(self.list.GetItem(item, 0).GetText())
  280. cat = int(self.list.GetItem(item, 1).GetText())
  281. self.list.DeleteItem(item)
  282. self.cats[self.fid][layer].remove(cat)
  283. item = self.list.GetFirstSelected()
  284. event.Skip()
  285. def OnItemDeleteAll(self, event):
  286. """Delete all items from the list"""
  287. self.list.DeleteAllItems()
  288. self.cats[self.fid] = {}
  289. event.Skip()
  290. def OnFeature(self, event):
  291. """Feature id changed (on duplicates)"""
  292. self.fid = int(event.GetString())
  293. self.itemDataMap = self.list.Populate(self.cats[self.fid], update=True)
  294. try:
  295. newCat = max(self.cats[self.fid][1]) + 1
  296. except KeyError:
  297. newCat = 1
  298. self.catNew.SetValue(newCat)
  299. event.Skip()
  300. def _getCategories(self, coords, qdist):
  301. """Get layer/category pairs for all available
  302. layers
  303. :return: True line found or False if not found
  304. """
  305. ret = RunCommand(
  306. "v.what",
  307. parent=self,
  308. quiet=True,
  309. map=self.vectorName,
  310. east_north="%f,%f" % (float(coords[0]), float(coords[1])),
  311. distance=qdist,
  312. )
  313. if not ret:
  314. return False
  315. for item in ret.splitlines():
  316. litem = item.lower()
  317. if "id:" in litem: # get line id
  318. self.line = int(item.split(":")[1].strip())
  319. elif "layer:" in litem: # add layer
  320. layer = int(item.split(":")[1].strip())
  321. if layer not in self.cats.keys():
  322. self.cats[layer] = []
  323. elif "category:" in litem: # add category
  324. self.cats[layer].append(int(item.split(":")[1].strip()))
  325. return True
  326. def OnReload(self, event):
  327. """Reload button pressed"""
  328. # restore original list
  329. self.cats = copy.deepcopy(self.cats_orig)
  330. # polulate list
  331. self.itemDataMap = self.list.Populate(self.cats[self.fid], update=True)
  332. event.Skip()
  333. def OnCancel(self, event):
  334. """Cancel button pressed"""
  335. self.parent.parent.dialogs["category"] = None
  336. if self.digit:
  337. self.digit.GetDisplay().SetSelected([])
  338. self.parent.UpdateMap(render=False)
  339. else:
  340. self.parent.parent.OnRender(None)
  341. self.Close()
  342. def OnApply(self, event):
  343. """Apply button pressed"""
  344. for fid in self.cats.keys():
  345. newfid = self.ApplyChanges(fid)
  346. if fid == self.fid and newfid > 0:
  347. self.fid = newfid
  348. def ApplyChanges(self, fid):
  349. """Apply changes
  350. :param fid: feature id
  351. """
  352. cats = self.cats[fid]
  353. cats_orig = self.cats_orig[fid]
  354. # action : (catsFrom, catsTo)
  355. check = {"catadd": (cats, cats_orig), "catdel": (cats_orig, cats)}
  356. newfid = -1
  357. # add/delete new category
  358. for action, catsCurr in six.iteritems(check):
  359. for layer in catsCurr[0].keys():
  360. catList = []
  361. for cat in catsCurr[0][layer]:
  362. if layer not in catsCurr[1].keys() or cat not in catsCurr[1][layer]:
  363. catList.append(cat)
  364. if catList != []:
  365. if action == "catadd":
  366. add = True
  367. else:
  368. add = False
  369. newfid = self.digit.SetLineCats(fid, layer, catList, add)
  370. if len(self.cats.keys()) == 1:
  371. self.fidText.SetLabel("%d" % newfid)
  372. else:
  373. choices = self.fidMulti.GetItems()
  374. choices[choices.index(str(fid))] = str(newfid)
  375. self.fidMulti.SetItems(choices)
  376. self.fidMulti.SetStringSelection(str(newfid))
  377. self.cats[newfid] = self.cats[fid]
  378. del self.cats[fid]
  379. fid = newfid
  380. if self.fid < 0:
  381. wx.MessageBox(
  382. parent=self,
  383. message=_("Unable to update vector map."),
  384. caption=_("Error"),
  385. style=wx.OK | wx.ICON_ERROR,
  386. )
  387. self.cats_orig[fid] = copy.deepcopy(cats)
  388. return newfid
  389. def OnOK(self, event):
  390. """OK button pressed"""
  391. self.OnApply(event)
  392. self.OnCancel(event)
  393. def OnAddCat(self, event):
  394. """Button 'Add' new category pressed"""
  395. try:
  396. layer = int(self.layerNew.GetStringSelection())
  397. cat = int(self.catNew.GetValue())
  398. if layer <= 0:
  399. raise ValueError
  400. except ValueError:
  401. GError(
  402. parent=self,
  403. message=_(
  404. "Unable to add new layer/category <%(layer)s/%(category)s>.\n"
  405. "Layer and category number must be integer.\n"
  406. "Layer number must be greater than zero."
  407. )
  408. % {
  409. "layer": str(self.layerNew.GetValue()),
  410. "category": str(self.catNew.GetValue()),
  411. },
  412. )
  413. return False
  414. if layer not in self.cats[self.fid].keys():
  415. self.cats[self.fid][layer] = []
  416. self.cats[self.fid][layer].append(cat)
  417. # reload list
  418. self.itemDataMap = self.list.Populate(self.cats[self.fid], update=True)
  419. # update category number for add
  420. self.catNew.SetValue(cat + 1)
  421. event.Skip()
  422. return True
  423. def GetLine(self):
  424. """Get id of selected line of 'None' if no line is selected"""
  425. return self.cats.keys()
  426. def UpdateDialog(self, query=None, cats=None):
  427. """Update dialog
  428. :param query: {coordinates, distance} - v.what
  429. :param cats: directory layer/cats - vdigit
  430. :return: True if updated otherwise False
  431. """
  432. # line: {layer: [categories]}
  433. self.cats = {}
  434. # do not display dialog if no line is found (-> self.cats)
  435. if cats is None:
  436. ret = self._getCategories(query[0], query[1])
  437. else:
  438. self.cats = cats
  439. for line in cats.keys():
  440. for layer in cats[line].keys():
  441. self.cats[line][layer] = list(cats[line][layer])
  442. ret = 1
  443. if ret == 0 or len(self.cats.keys()) < 1:
  444. Debug.msg(3, "VDigitCategoryDialog(): nothing found!")
  445. return False
  446. # make copy of cats (used for 'reload')
  447. self.cats_orig = copy.deepcopy(self.cats)
  448. # polulate list
  449. self.fid = list(self.cats.keys())[0]
  450. self.itemDataMap = self.list.Populate(self.cats[self.fid], update=True)
  451. try:
  452. newCat = max(self.cats[self.fid][1]) + 1
  453. except KeyError:
  454. newCat = 1
  455. self.catNew.SetValue(newCat)
  456. if len(self.cats.keys()) == 1:
  457. self.fidText.Show(True)
  458. self.fidMulti.Show(False)
  459. self.fidText.SetLabel("%d" % self.fid)
  460. else:
  461. self.fidText.Show(False)
  462. self.fidMulti.Show(True)
  463. choices = []
  464. for fid in self.cats.keys():
  465. choices.append(str(fid))
  466. self.fidMulti.SetItems(choices)
  467. self.fidMulti.SetSelection(0)
  468. self.Layout()
  469. return True
  470. class CategoryListCtrl(ListCtrl, listmix.ListCtrlAutoWidthMixin, listmix.TextEditMixin):
  471. def __init__(
  472. self, parent, id, pos=wx.DefaultPosition, size=wx.DefaultSize, style=0
  473. ):
  474. """List of layers/categories"""
  475. self.parent = parent
  476. ListCtrl.__init__(self, parent, id, pos, size, style)
  477. listmix.ListCtrlAutoWidthMixin.__init__(self)
  478. listmix.TextEditMixin.__init__(self)
  479. def Populate(self, cats, update=False):
  480. """Populate the list"""
  481. itemData = {} # requested by sorter
  482. if not update:
  483. self.InsertColumn(0, _("Layer"))
  484. self.InsertColumn(1, _("Category"))
  485. else:
  486. self.DeleteAllItems()
  487. i = 1
  488. for layer in cats.keys():
  489. catsList = cats[layer]
  490. for cat in catsList:
  491. index = self.InsertItem(self.GetItemCount(), str(catsList[0]))
  492. self.SetItem(index, 0, str(layer))
  493. self.SetItem(index, 1, str(cat))
  494. self.SetItemData(index, i)
  495. itemData[i] = (str(layer), str(cat))
  496. i = i + 1
  497. if not update:
  498. self.SetColumnWidth(0, 100)
  499. self.SetColumnWidth(1, wx.LIST_AUTOSIZE)
  500. self.currentItem = 0
  501. return itemData
  502. class VDigitZBulkDialog(wx.Dialog):
  503. def __init__(self, parent, title, nselected, style=wx.DEFAULT_DIALOG_STYLE):
  504. """Dialog used for Z bulk-labeling tool"""
  505. wx.Dialog.__init__(self, parent=parent, id=wx.ID_ANY, title=title, style=style)
  506. self.parent = parent # map window class instance
  507. # panel = wx.Panel(parent=self, id=wx.ID_ANY)
  508. border = wx.BoxSizer(wx.VERTICAL)
  509. txt = StaticText(
  510. parent=self, label=_("%d lines selected for z bulk-labeling") % nselected
  511. )
  512. border.Add(txt, proportion=0, flag=wx.ALL | wx.EXPAND, border=5)
  513. box = StaticBox(parent=self, id=wx.ID_ANY, label=" %s " % _("Set value"))
  514. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  515. flexSizer = wx.FlexGridSizer(cols=2, hgap=5, vgap=5)
  516. flexSizer.AddGrowableCol(0)
  517. # starting value
  518. txt = StaticText(parent=self, label=_("Starting value"))
  519. self.value = SpinCtrl(
  520. parent=self, id=wx.ID_ANY, size=(150, -1), initial=0, min=-1e6, max=1e6
  521. )
  522. flexSizer.Add(txt, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
  523. flexSizer.Add(self.value, proportion=0, flag=wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  524. # step
  525. txt = StaticText(parent=self, label=_("Step"))
  526. self.step = SpinCtrl(
  527. parent=self, id=wx.ID_ANY, size=(150, -1), initial=0, min=0, max=1e6
  528. )
  529. flexSizer.Add(txt, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
  530. flexSizer.Add(self.step, proportion=0, flag=wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  531. sizer.Add(flexSizer, proportion=1, flag=wx.ALL | wx.EXPAND, border=1)
  532. border.Add(sizer, proportion=1, flag=wx.ALL | wx.EXPAND, border=0)
  533. # buttons
  534. btnCancel = Button(self, wx.ID_CANCEL)
  535. btnOk = Button(self, wx.ID_OK)
  536. btnOk.SetDefault()
  537. # sizers
  538. btnSizer = wx.StdDialogButtonSizer()
  539. btnSizer.AddButton(btnCancel)
  540. btnSizer.AddButton(btnOk)
  541. btnSizer.Realize()
  542. mainSizer = wx.BoxSizer(wx.VERTICAL)
  543. mainSizer.Add(border, proportion=1, flag=wx.EXPAND | wx.ALL, border=5)
  544. mainSizer.Add(
  545. btnSizer, proportion=0, flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5
  546. )
  547. self.SetSizer(mainSizer)
  548. mainSizer.Fit(self)
  549. class VDigitDuplicatesDialog(wx.Dialog):
  550. def __init__(
  551. self,
  552. parent,
  553. data,
  554. title=_("List of duplicates"),
  555. style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER,
  556. pos=wx.DefaultPosition,
  557. ):
  558. """Show duplicated feature ids"""
  559. wx.Dialog.__init__(
  560. self, parent=parent, id=wx.ID_ANY, title=title, style=style, pos=pos
  561. )
  562. self.parent = parent # map window instance
  563. self.data = data
  564. self.winList = []
  565. # panel = wx.Panel(parent=self, id=wx.ID_ANY)
  566. # notebook
  567. self.notebook = wx.Notebook(parent=self, id=wx.ID_ANY, style=wx.BK_DEFAULT)
  568. id = 1
  569. for key in self.data.keys():
  570. panel = wx.Panel(parent=self.notebook, id=wx.ID_ANY)
  571. self.notebook.AddPage(page=panel, text=" %d " % (id))
  572. # notebook body
  573. border = wx.BoxSizer(wx.VERTICAL)
  574. win = CheckListFeature(parent=panel, data=list(self.data[key]))
  575. self.winList.append(win.GetId())
  576. border.Add(win, proportion=1, flag=wx.ALL | wx.EXPAND, border=5)
  577. panel.SetSizer(border)
  578. id += 1
  579. # buttons
  580. btnCancel = Button(self, wx.ID_CANCEL)
  581. btnOk = Button(self, wx.ID_OK)
  582. btnOk.SetDefault()
  583. # sizers
  584. btnSizer = wx.StdDialogButtonSizer()
  585. btnSizer.AddButton(btnCancel)
  586. btnSizer.AddButton(btnOk)
  587. btnSizer.Realize()
  588. mainSizer = wx.BoxSizer(wx.VERTICAL)
  589. mainSizer.Add(self.notebook, proportion=1, flag=wx.EXPAND | wx.ALL, border=5)
  590. mainSizer.Add(
  591. btnSizer, proportion=0, flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5
  592. )
  593. self.SetSizer(mainSizer)
  594. mainSizer.Fit(self)
  595. self.SetAutoLayout(True)
  596. # set min size for dialog
  597. self.SetMinSize((250, 180))
  598. def GetUnSelected(self):
  599. """Get unselected items (feature id)
  600. :return: list of ids
  601. """
  602. ids = []
  603. for id in self.winList:
  604. wlist = self.FindWindowById(id)
  605. for item in range(wlist.GetItemCount()):
  606. if not wlist.IsItemChecked(item):
  607. ids.append(int(wlist.GetItem(item, 0).GetText()))
  608. return ids
  609. class CheckListFeature(ListCtrl, listmix.ListCtrlAutoWidthMixin, CheckListCtrlMixin):
  610. def __init__(self, parent, data, pos=wx.DefaultPosition, log=None):
  611. """List of mapset/owner/group"""
  612. self.parent = parent
  613. self.data = data
  614. ListCtrl.__init__(self, parent, wx.ID_ANY, style=wx.LC_REPORT)
  615. CheckListCtrlMixin.__init__(self)
  616. self.log = log
  617. # setup mixins
  618. listmix.ListCtrlAutoWidthMixin.__init__(self)
  619. self.LoadData(self.data)
  620. def LoadData(self, data):
  621. """Load data into list"""
  622. self.InsertColumn(0, _("Feature id"))
  623. self.InsertColumn(1, _("Layer (Categories)"))
  624. for item in data:
  625. index = self.InsertItem(self.GetItemCount(), str(item[0]))
  626. self.SetItem(index, 1, str(item[1]))
  627. # enable all items by default
  628. for item in range(self.GetItemCount()):
  629. self.CheckItem(item, True)
  630. self.SetColumnWidth(col=0, width=wx.LIST_AUTOSIZE_USEHEADER)
  631. self.SetColumnWidth(col=1, width=wx.LIST_AUTOSIZE_USEHEADER)
  632. def OnCheckItem(self, index, flag):
  633. """Mapset checked/unchecked"""
  634. pass