123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783 |
- """
- @package vdigit.dialogs
- @brief wxGUI vector digitizer dialogs
- Classes:
- - dialogs::VDigitCategoryDialog
- - dialogs::CategoryListCtrl
- - dialogs::VDigitZBulkDialog
- - dialogs::VDigitDuplicatesDialog
- - dialogs::CheckListFeature
- (C) 2007-2011 by the GRASS Development Team
- This program is free software under the GNU General Public License
- (>=v2). Read the file COPYING that comes with GRASS for details.
- @author Martin Landa <landa.martin gmail.com>
- """
- import copy
- import six
- import wx
- import wx.lib.mixins.listctrl as listmix
- from core.gcmd import RunCommand, GError
- from core.debug import Debug
- from gui_core.wrap import (
- SpinCtrl,
- Button,
- StaticText,
- StaticBox,
- Menu,
- ListCtrl,
- NewId,
- CheckListCtrlMixin,
- )
- class VDigitCategoryDialog(wx.Dialog, listmix.ColumnSorterMixin):
- def __init__(
- self,
- parent,
- title,
- vectorName,
- query=None,
- cats=None,
- style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER,
- **kwargs,
- ):
- """Dialog used to display/modify categories of vector objects
- :param parent:
- :param title: dialog title
- :param query: {coordinates, qdist} - used by v.edit/v.what
- :param cats: directory of lines (layer/categories) - used by vdigit
- :param style: dialog style
- """
- self.parent = parent # map window class instance
- self.digit = parent.digit
- # map name
- self.vectorName = vectorName
- # line : {layer: [categories]}
- self.cats = {}
- # do not display dialog if no line is found (-> self.cats)
- if cats is None:
- if self._getCategories(query[0], query[1]) == 0 or not self.line:
- Debug.msg(3, "VDigitCategoryDialog(): nothing found!")
- else:
- self.cats = cats
- for line in cats.keys():
- for layer in cats[line].keys():
- self.cats[line][layer] = list(cats[line][layer])
- layers = []
- for layer in self.digit.GetLayers():
- layers.append(str(layer))
- # make copy of cats (used for 'reload')
- self.cats_orig = copy.deepcopy(self.cats)
- wx.Dialog.__init__(
- self, parent=self.parent, id=wx.ID_ANY, title=title, style=style, **kwargs
- )
- # list of categories
- box = StaticBox(
- parent=self,
- id=wx.ID_ANY,
- label=" %s " % _("List of categories - right-click to delete"),
- )
- listSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
- self.list = CategoryListCtrl(
- parent=self,
- id=wx.ID_ANY,
- style=wx.LC_REPORT
- | wx.BORDER_NONE
- | wx.LC_SORT_ASCENDING
- | wx.LC_HRULES
- | wx.LC_VRULES,
- )
- # sorter
- self.fid = list(self.cats.keys())[0]
- self.itemDataMap = self.list.Populate(self.cats[self.fid])
- listmix.ColumnSorterMixin.__init__(self, 2)
- self.fidMulti = wx.Choice(parent=self, id=wx.ID_ANY, size=(150, -1))
- self.fidMulti.Bind(wx.EVT_CHOICE, self.OnFeature)
- self.fidText = StaticText(parent=self, id=wx.ID_ANY)
- if len(self.cats.keys()) == 1:
- self.fidMulti.Show(False)
- self.fidText.SetLabel(str(self.fid))
- else:
- self.fidText.Show(False)
- choices = []
- for fid in self.cats.keys():
- choices.append(str(fid))
- self.fidMulti.SetItems(choices)
- self.fidMulti.SetSelection(0)
- listSizer.Add(self.list, proportion=1, flag=wx.EXPAND)
- # add new category
- box = StaticBox(parent=self, id=wx.ID_ANY, label=" %s " % _("Add new category"))
- addSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
- flexSizer = wx.FlexGridSizer(cols=5, hgap=5, vgap=5)
- flexSizer.AddGrowableCol(3)
- layerNewTxt = StaticText(parent=self, id=wx.ID_ANY, label="%s:" % _("Layer"))
- self.layerNew = wx.Choice(
- parent=self, id=wx.ID_ANY, size=(75, -1), choices=layers
- )
- if len(layers) > 0:
- self.layerNew.SetSelection(0)
- catNewTxt = StaticText(parent=self, id=wx.ID_ANY, label="%s:" % _("Category"))
- try:
- newCat = max(self.cats[self.fid][1]) + 1
- except KeyError:
- newCat = 1
- self.catNew = SpinCtrl(
- parent=self, id=wx.ID_ANY, size=(75, -1), initial=newCat, min=0, max=1e9
- )
- btnAddCat = Button(self, wx.ID_ADD)
- flexSizer.Add(
- layerNewTxt, proportion=0, flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL
- )
- flexSizer.Add(
- self.layerNew,
- proportion=0,
- flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL,
- )
- flexSizer.Add(
- catNewTxt,
- proportion=0,
- flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_RIGHT | wx.LEFT,
- border=10,
- )
- flexSizer.Add(
- self.catNew, proportion=0, flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL
- )
- flexSizer.Add(
- btnAddCat, proportion=0, flag=wx.EXPAND | wx.ALIGN_RIGHT | wx.FIXED_MINSIZE
- )
- addSizer.Add(flexSizer, proportion=1, flag=wx.ALL | wx.EXPAND, border=5)
- # buttons
- btnApply = Button(self, wx.ID_APPLY)
- btnApply.SetToolTip(_("Apply changes"))
- btnCancel = Button(self, wx.ID_CANCEL)
- btnCancel.SetToolTip(_("Ignore changes and close dialog"))
- btnOk = Button(self, wx.ID_OK)
- btnOk.SetToolTip(_("Apply changes and close dialog"))
- btnOk.SetDefault()
- # sizers
- btnSizer = wx.StdDialogButtonSizer()
- btnSizer.AddButton(btnCancel)
- # btnSizer.AddButton(btnReload)
- # btnSizer.SetNegativeButton(btnReload)
- btnSizer.AddButton(btnApply)
- btnSizer.AddButton(btnOk)
- btnSizer.Realize()
- mainSizer = wx.BoxSizer(wx.VERTICAL)
- mainSizer.Add(listSizer, proportion=1, flag=wx.EXPAND | wx.ALL, border=5)
- mainSizer.Add(
- addSizer,
- proportion=0,
- flag=wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM,
- border=5,
- )
- fidSizer = wx.BoxSizer(wx.HORIZONTAL)
- fidSizer.Add(
- StaticText(parent=self, id=wx.ID_ANY, label=_("Feature id:")),
- proportion=0,
- border=5,
- flag=wx.ALIGN_CENTER_VERTICAL,
- )
- fidSizer.Add(self.fidMulti, proportion=0, flag=wx.EXPAND | wx.ALL, border=5)
- fidSizer.Add(self.fidText, proportion=0, flag=wx.EXPAND | wx.ALL, border=5)
- mainSizer.Add(fidSizer, proportion=0, flag=wx.EXPAND | wx.ALL, border=5)
- mainSizer.Add(btnSizer, proportion=0, flag=wx.EXPAND | wx.ALL, border=5)
- self.SetSizer(mainSizer)
- mainSizer.Fit(self)
- self.SetAutoLayout(True)
- # set min size for dialog
- self.SetMinSize(self.GetBestSize())
- # bindings
- btnApply.Bind(wx.EVT_BUTTON, self.OnApply)
- btnOk.Bind(wx.EVT_BUTTON, self.OnOK)
- btnAddCat.Bind(wx.EVT_BUTTON, self.OnAddCat)
- btnCancel.Bind(wx.EVT_BUTTON, self.OnCancel)
- self.Bind(wx.EVT_CLOSE, lambda evt: self.Hide())
- # list
- self.list.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnRightUp) # wxMSW
- self.list.Bind(wx.EVT_RIGHT_UP, self.OnRightUp) # wxGTK
- self.Bind(wx.EVT_LIST_BEGIN_LABEL_EDIT, self.OnBeginEdit, self.list)
- self.Bind(wx.EVT_LIST_END_LABEL_EDIT, self.OnEndEdit, self.list)
- self.Bind(wx.EVT_LIST_COL_CLICK, self.OnColClick, self.list)
- def GetListCtrl(self):
- """Used by ColumnSorterMixin"""
- return self.list
- def OnColClick(self, event):
- """Click on column header (order by)"""
- event.Skip()
- def OnBeginEdit(self, event):
- """Editing of item started"""
- event.Allow()
- def OnEndEdit(self, event):
- """Finish editing of item"""
- itemIndex = event.GetIndex()
- layerOld = int(self.list.GetItem(itemIndex, 0).GetText())
- catOld = int(self.list.GetItem(itemIndex, 1).GetText())
- if event.GetColumn() == 0:
- layerNew = int(event.GetLabel())
- catNew = catOld
- else:
- layerNew = layerOld
- catNew = int(event.GetLabel())
- try:
- if layerNew not in self.cats[self.fid].keys():
- self.cats[self.fid][layerNew] = []
- self.cats[self.fid][layerNew].append(catNew)
- self.cats[self.fid][layerOld].remove(catOld)
- except:
- event.Veto()
- self.list.SetItem(itemIndex, 0, str(layerNew))
- self.list.SetItem(itemIndex, 1, str(catNew))
- dlg = wx.MessageDialog(
- self,
- _(
- "Unable to add new layer/category <%(layer)s/%(category)s>.\n"
- "Layer and category number must be integer.\n"
- "Layer number must be greater than zero."
- )
- % {
- "layer": self.layerNew.GetStringSelection(),
- "category": str(self.catNew.GetValue()),
- },
- _("Error"),
- wx.OK | wx.ICON_ERROR,
- )
- dlg.ShowModal()
- dlg.Destroy()
- return False
- def OnRightDown(self, event):
- """Mouse right button down"""
- x = event.GetX()
- y = event.GetY()
- item, flags = self.list.HitTest((x, y))
- if item != wx.NOT_FOUND and flags & wx.LIST_HITTEST_ONITEM:
- self.list.Select(item)
- event.Skip()
- def OnRightUp(self, event):
- """Mouse right button up"""
- if not hasattr(self, "popupID1"):
- self.popupID1 = NewId()
- self.popupID2 = NewId()
- self.popupID3 = NewId()
- self.Bind(wx.EVT_MENU, self.OnItemDelete, id=self.popupID1)
- self.Bind(wx.EVT_MENU, self.OnItemDeleteAll, id=self.popupID2)
- self.Bind(wx.EVT_MENU, self.OnReload, id=self.popupID3)
- # generate popup-menu
- menu = Menu()
- menu.Append(self.popupID1, _("Delete selected"))
- if self.list.GetFirstSelected() == -1:
- menu.Enable(self.popupID1, False)
- menu.Append(self.popupID2, _("Delete all"))
- menu.AppendSeparator()
- menu.Append(self.popupID3, _("Reload"))
- self.PopupMenu(menu)
- menu.Destroy()
- def OnItemSelected(self, event):
- """Item selected"""
- event.Skip()
- def OnItemDelete(self, event):
- """Delete selected item(s) from the list (layer/category pair)"""
- item = self.list.GetFirstSelected()
- while item != -1:
- layer = int(self.list.GetItem(item, 0).GetText())
- cat = int(self.list.GetItem(item, 1).GetText())
- self.list.DeleteItem(item)
- self.cats[self.fid][layer].remove(cat)
- item = self.list.GetFirstSelected()
- event.Skip()
- def OnItemDeleteAll(self, event):
- """Delete all items from the list"""
- self.list.DeleteAllItems()
- self.cats[self.fid] = {}
- event.Skip()
- def OnFeature(self, event):
- """Feature id changed (on duplicates)"""
- self.fid = int(event.GetString())
- self.itemDataMap = self.list.Populate(self.cats[self.fid], update=True)
- try:
- newCat = max(self.cats[self.fid][1]) + 1
- except KeyError:
- newCat = 1
- self.catNew.SetValue(newCat)
- event.Skip()
- def _getCategories(self, coords, qdist):
- """Get layer/category pairs for all available
- layers
- :return: True line found or False if not found
- """
- ret = RunCommand(
- "v.what",
- parent=self,
- quiet=True,
- map=self.vectorName,
- east_north="%f,%f" % (float(coords[0]), float(coords[1])),
- distance=qdist,
- )
- if not ret:
- return False
- for item in ret.splitlines():
- litem = item.lower()
- if "id:" in litem: # get line id
- self.line = int(item.split(":")[1].strip())
- elif "layer:" in litem: # add layer
- layer = int(item.split(":")[1].strip())
- if layer not in self.cats.keys():
- self.cats[layer] = []
- elif "category:" in litem: # add category
- self.cats[layer].append(int(item.split(":")[1].strip()))
- return True
- def OnReload(self, event):
- """Reload button pressed"""
- # restore original list
- self.cats = copy.deepcopy(self.cats_orig)
- # polulate list
- self.itemDataMap = self.list.Populate(self.cats[self.fid], update=True)
- event.Skip()
- def OnCancel(self, event):
- """Cancel button pressed"""
- self.parent.parent.dialogs["category"] = None
- if self.digit:
- self.digit.GetDisplay().SetSelected([])
- self.parent.UpdateMap(render=False)
- else:
- self.parent.parent.OnRender(None)
- self.Close()
- def OnApply(self, event):
- """Apply button pressed"""
- for fid in self.cats.keys():
- newfid = self.ApplyChanges(fid)
- if fid == self.fid and newfid > 0:
- self.fid = newfid
- def ApplyChanges(self, fid):
- """Apply changes
- :param fid: feature id
- """
- cats = self.cats[fid]
- cats_orig = self.cats_orig[fid]
- # action : (catsFrom, catsTo)
- check = {"catadd": (cats, cats_orig), "catdel": (cats_orig, cats)}
- newfid = -1
- # add/delete new category
- for action, catsCurr in six.iteritems(check):
- for layer in catsCurr[0].keys():
- catList = []
- for cat in catsCurr[0][layer]:
- if layer not in catsCurr[1].keys() or cat not in catsCurr[1][layer]:
- catList.append(cat)
- if catList != []:
- if action == "catadd":
- add = True
- else:
- add = False
- newfid = self.digit.SetLineCats(fid, layer, catList, add)
- if len(self.cats.keys()) == 1:
- self.fidText.SetLabel("%d" % newfid)
- else:
- choices = self.fidMulti.GetItems()
- choices[choices.index(str(fid))] = str(newfid)
- self.fidMulti.SetItems(choices)
- self.fidMulti.SetStringSelection(str(newfid))
- self.cats[newfid] = self.cats[fid]
- del self.cats[fid]
- fid = newfid
- if self.fid < 0:
- wx.MessageBox(
- parent=self,
- message=_("Unable to update vector map."),
- caption=_("Error"),
- style=wx.OK | wx.ICON_ERROR,
- )
- self.cats_orig[fid] = copy.deepcopy(cats)
- return newfid
- def OnOK(self, event):
- """OK button pressed"""
- self.OnApply(event)
- self.OnCancel(event)
- def OnAddCat(self, event):
- """Button 'Add' new category pressed"""
- try:
- layer = int(self.layerNew.GetStringSelection())
- cat = int(self.catNew.GetValue())
- if layer <= 0:
- raise ValueError
- except ValueError:
- GError(
- parent=self,
- message=_(
- "Unable to add new layer/category <%(layer)s/%(category)s>.\n"
- "Layer and category number must be integer.\n"
- "Layer number must be greater than zero."
- )
- % {
- "layer": str(self.layerNew.GetValue()),
- "category": str(self.catNew.GetValue()),
- },
- )
- return False
- if layer not in self.cats[self.fid].keys():
- self.cats[self.fid][layer] = []
- self.cats[self.fid][layer].append(cat)
- # reload list
- self.itemDataMap = self.list.Populate(self.cats[self.fid], update=True)
- # update category number for add
- self.catNew.SetValue(cat + 1)
- event.Skip()
- return True
- def GetLine(self):
- """Get id of selected line of 'None' if no line is selected"""
- return self.cats.keys()
- def UpdateDialog(self, query=None, cats=None):
- """Update dialog
- :param query: {coordinates, distance} - v.what
- :param cats: directory layer/cats - vdigit
- :return: True if updated otherwise False
- """
- # line: {layer: [categories]}
- self.cats = {}
- # do not display dialog if no line is found (-> self.cats)
- if cats is None:
- ret = self._getCategories(query[0], query[1])
- else:
- self.cats = cats
- for line in cats.keys():
- for layer in cats[line].keys():
- self.cats[line][layer] = list(cats[line][layer])
- ret = 1
- if ret == 0 or len(self.cats.keys()) < 1:
- Debug.msg(3, "VDigitCategoryDialog(): nothing found!")
- return False
- # make copy of cats (used for 'reload')
- self.cats_orig = copy.deepcopy(self.cats)
- # polulate list
- self.fid = list(self.cats.keys())[0]
- self.itemDataMap = self.list.Populate(self.cats[self.fid], update=True)
- try:
- newCat = max(self.cats[self.fid][1]) + 1
- except KeyError:
- newCat = 1
- self.catNew.SetValue(newCat)
- if len(self.cats.keys()) == 1:
- self.fidText.Show(True)
- self.fidMulti.Show(False)
- self.fidText.SetLabel("%d" % self.fid)
- else:
- self.fidText.Show(False)
- self.fidMulti.Show(True)
- choices = []
- for fid in self.cats.keys():
- choices.append(str(fid))
- self.fidMulti.SetItems(choices)
- self.fidMulti.SetSelection(0)
- self.Layout()
- return True
- class CategoryListCtrl(ListCtrl, listmix.ListCtrlAutoWidthMixin, listmix.TextEditMixin):
- def __init__(
- self, parent, id, pos=wx.DefaultPosition, size=wx.DefaultSize, style=0
- ):
- """List of layers/categories"""
- self.parent = parent
- ListCtrl.__init__(self, parent, id, pos, size, style)
- listmix.ListCtrlAutoWidthMixin.__init__(self)
- listmix.TextEditMixin.__init__(self)
- def Populate(self, cats, update=False):
- """Populate the list"""
- itemData = {} # requested by sorter
- if not update:
- self.InsertColumn(0, _("Layer"))
- self.InsertColumn(1, _("Category"))
- else:
- self.DeleteAllItems()
- i = 1
- for layer in cats.keys():
- catsList = cats[layer]
- for cat in catsList:
- index = self.InsertItem(self.GetItemCount(), str(catsList[0]))
- self.SetItem(index, 0, str(layer))
- self.SetItem(index, 1, str(cat))
- self.SetItemData(index, i)
- itemData[i] = (str(layer), str(cat))
- i = i + 1
- if not update:
- self.SetColumnWidth(0, 100)
- self.SetColumnWidth(1, wx.LIST_AUTOSIZE)
- self.currentItem = 0
- return itemData
- class VDigitZBulkDialog(wx.Dialog):
- def __init__(self, parent, title, nselected, style=wx.DEFAULT_DIALOG_STYLE):
- """Dialog used for Z bulk-labeling tool"""
- wx.Dialog.__init__(self, parent=parent, id=wx.ID_ANY, title=title, style=style)
- self.parent = parent # map window class instance
- # panel = wx.Panel(parent=self, id=wx.ID_ANY)
- border = wx.BoxSizer(wx.VERTICAL)
- txt = StaticText(
- parent=self, label=_("%d lines selected for z bulk-labeling") % nselected
- )
- border.Add(txt, proportion=0, flag=wx.ALL | wx.EXPAND, border=5)
- box = StaticBox(parent=self, id=wx.ID_ANY, label=" %s " % _("Set value"))
- sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
- flexSizer = wx.FlexGridSizer(cols=2, hgap=5, vgap=5)
- flexSizer.AddGrowableCol(0)
- # starting value
- txt = StaticText(parent=self, label=_("Starting value"))
- self.value = SpinCtrl(
- parent=self, id=wx.ID_ANY, size=(150, -1), initial=0, min=-1e6, max=1e6
- )
- flexSizer.Add(txt, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
- flexSizer.Add(self.value, proportion=0, flag=wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
- # step
- txt = StaticText(parent=self, label=_("Step"))
- self.step = SpinCtrl(
- parent=self, id=wx.ID_ANY, size=(150, -1), initial=0, min=0, max=1e6
- )
- flexSizer.Add(txt, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
- flexSizer.Add(self.step, proportion=0, flag=wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
- sizer.Add(flexSizer, proportion=1, flag=wx.ALL | wx.EXPAND, border=1)
- border.Add(sizer, proportion=1, flag=wx.ALL | wx.EXPAND, border=0)
- # buttons
- btnCancel = Button(self, wx.ID_CANCEL)
- btnOk = Button(self, wx.ID_OK)
- btnOk.SetDefault()
- # sizers
- btnSizer = wx.StdDialogButtonSizer()
- btnSizer.AddButton(btnCancel)
- btnSizer.AddButton(btnOk)
- btnSizer.Realize()
- mainSizer = wx.BoxSizer(wx.VERTICAL)
- mainSizer.Add(border, proportion=1, flag=wx.EXPAND | wx.ALL, border=5)
- mainSizer.Add(
- btnSizer, proportion=0, flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5
- )
- self.SetSizer(mainSizer)
- mainSizer.Fit(self)
- class VDigitDuplicatesDialog(wx.Dialog):
- def __init__(
- self,
- parent,
- data,
- title=_("List of duplicates"),
- style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER,
- pos=wx.DefaultPosition,
- ):
- """Show duplicated feature ids"""
- wx.Dialog.__init__(
- self, parent=parent, id=wx.ID_ANY, title=title, style=style, pos=pos
- )
- self.parent = parent # map window instance
- self.data = data
- self.winList = []
- # panel = wx.Panel(parent=self, id=wx.ID_ANY)
- # notebook
- self.notebook = wx.Notebook(parent=self, id=wx.ID_ANY, style=wx.BK_DEFAULT)
- id = 1
- for key in self.data.keys():
- panel = wx.Panel(parent=self.notebook, id=wx.ID_ANY)
- self.notebook.AddPage(page=panel, text=" %d " % (id))
- # notebook body
- border = wx.BoxSizer(wx.VERTICAL)
- win = CheckListFeature(parent=panel, data=list(self.data[key]))
- self.winList.append(win.GetId())
- border.Add(win, proportion=1, flag=wx.ALL | wx.EXPAND, border=5)
- panel.SetSizer(border)
- id += 1
- # buttons
- btnCancel = Button(self, wx.ID_CANCEL)
- btnOk = Button(self, wx.ID_OK)
- btnOk.SetDefault()
- # sizers
- btnSizer = wx.StdDialogButtonSizer()
- btnSizer.AddButton(btnCancel)
- btnSizer.AddButton(btnOk)
- btnSizer.Realize()
- mainSizer = wx.BoxSizer(wx.VERTICAL)
- mainSizer.Add(self.notebook, proportion=1, flag=wx.EXPAND | wx.ALL, border=5)
- mainSizer.Add(
- btnSizer, proportion=0, flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5
- )
- self.SetSizer(mainSizer)
- mainSizer.Fit(self)
- self.SetAutoLayout(True)
- # set min size for dialog
- self.SetMinSize((250, 180))
- def GetUnSelected(self):
- """Get unselected items (feature id)
- :return: list of ids
- """
- ids = []
- for id in self.winList:
- wlist = self.FindWindowById(id)
- for item in range(wlist.GetItemCount()):
- if not wlist.IsItemChecked(item):
- ids.append(int(wlist.GetItem(item, 0).GetText()))
- return ids
- class CheckListFeature(ListCtrl, listmix.ListCtrlAutoWidthMixin, CheckListCtrlMixin):
- def __init__(self, parent, data, pos=wx.DefaultPosition, log=None):
- """List of mapset/owner/group"""
- self.parent = parent
- self.data = data
- ListCtrl.__init__(self, parent, wx.ID_ANY, style=wx.LC_REPORT)
- CheckListCtrlMixin.__init__(self)
- self.log = log
- # setup mixins
- listmix.ListCtrlAutoWidthMixin.__init__(self)
- self.LoadData(self.data)
- def LoadData(self, data):
- """Load data into list"""
- self.InsertColumn(0, _("Feature id"))
- self.InsertColumn(1, _("Layer (Categories)"))
- for item in data:
- index = self.InsertItem(self.GetItemCount(), str(item[0]))
- self.SetItem(index, 1, str(item[1]))
- # enable all items by default
- for item in range(self.GetItemCount()):
- self.CheckItem(item, True)
- self.SetColumnWidth(col=0, width=wx.LIST_AUTOSIZE_USEHEADER)
- self.SetColumnWidth(col=1, width=wx.LIST_AUTOSIZE_USEHEADER)
- def OnCheckItem(self, index, flag):
- """Mapset checked/unchecked"""
- pass
|