dialogs.py 27 KB

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