dialogs.py 27 KB

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