vdigit.py 66 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542
  1. """!
  2. @package vdigit
  3. @brief Dialogs for wxGUI vector digitizer
  4. Classes:
  5. - VDigit
  6. - VDigitSettingsDialog
  7. - VDigitCategoryDialog
  8. - CategoryListCtrl
  9. - VDigitZBulkDialog
  10. - VDigitDuplicatesDialog
  11. - CheckListFeature
  12. (C) 2007-2011 by the GRASS Development Team
  13. This program is free software under the GNU General Public License
  14. (>=v2). Read the file COPYING that comes with GRASS for details.
  15. @author Martin Landa <landa.martin gmail.com>
  16. """
  17. import os
  18. import sys
  19. import string
  20. import copy
  21. import textwrap
  22. import traceback
  23. from threading import Thread
  24. import wx
  25. import wx.lib.colourselect as csel
  26. import wx.lib.mixins.listctrl as listmix
  27. import gcmd
  28. import dbm
  29. from debug import Debug as Debug
  30. import gselect
  31. import globalvar
  32. from units import Units
  33. from preferences import globalSettings as UserSettings
  34. try:
  35. from wxvdigit import IVDigit, GV_LINES
  36. haveVDigit = True
  37. errorMsg = ''
  38. except ImportError, err:
  39. haveVDigit = False
  40. errorMsg = err
  41. GV_LINES = -1
  42. class IVDigit:
  43. def __init__(self):
  44. pass
  45. class VDigit(IVDigit):
  46. def __init__(self, mapwindow):
  47. """!Base class of vector digitizer
  48. @param mapwindow reference to mapwindow (MapFrame) instance
  49. """
  50. IVDigit.__init__(self, mapwindow)
  51. class VDigitSettingsDialog(wx.Dialog):
  52. def __init__(self, parent, title, style = wx.DEFAULT_DIALOG_STYLE):
  53. """!Standard settings dialog for digitization purposes
  54. """
  55. wx.Dialog.__init__(self, parent = parent, id = wx.ID_ANY, title = title, style = style)
  56. self.parent = parent # mapdisplay.BufferedWindow class instance
  57. # notebook
  58. notebook = wx.Notebook(parent = self, id = wx.ID_ANY, style = wx.BK_DEFAULT)
  59. self.__CreateSymbologyPage(notebook)
  60. if not UserSettings.Get(group = 'vdigit', key = 'categoryMode', subkey = 'selection'):
  61. self.parent.digit.SetCategoryNextToUse()
  62. self.__CreateGeneralPage(notebook)
  63. self.__CreateAttributesPage(notebook)
  64. self.__CreateQueryPage(notebook)
  65. # buttons
  66. btnApply = wx.Button(self, wx.ID_APPLY)
  67. btnCancel = wx.Button(self, wx.ID_CANCEL)
  68. btnSave = wx.Button(self, wx.ID_SAVE)
  69. btnSave.SetDefault()
  70. # bindigs
  71. btnApply.Bind(wx.EVT_BUTTON, self.OnApply)
  72. btnApply.SetToolTipString(_("Apply changes for this session"))
  73. btnApply.SetDefault()
  74. btnSave.Bind(wx.EVT_BUTTON, self.OnSave)
  75. btnSave.SetToolTipString(_("Close dialog and save changes to user settings file"))
  76. btnCancel.Bind(wx.EVT_BUTTON, self.OnCancel)
  77. btnCancel.SetToolTipString(_("Close dialog and ignore changes"))
  78. # sizers
  79. btnSizer = wx.StdDialogButtonSizer()
  80. btnSizer.AddButton(btnCancel)
  81. btnSizer.AddButton(btnApply)
  82. btnSizer.AddButton(btnSave)
  83. btnSizer.Realize()
  84. mainSizer = wx.BoxSizer(wx.VERTICAL)
  85. mainSizer.Add(item = notebook, proportion = 1, flag = wx.EXPAND | wx.ALL, border = 5)
  86. mainSizer.Add(item = btnSizer, proportion = 0,
  87. flag = wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border = 5)
  88. self.Bind(wx.EVT_CLOSE, self.OnCancel)
  89. self.SetSizer(mainSizer)
  90. mainSizer.Fit(self)
  91. def __CreateSymbologyPage(self, notebook):
  92. """!Create notebook page concerning with symbology settings"""
  93. panel = wx.Panel(parent = notebook, id = wx.ID_ANY)
  94. notebook.AddPage(page = panel, text = _("Symbology"))
  95. sizer = wx.BoxSizer(wx.VERTICAL)
  96. flexSizer = wx.FlexGridSizer (cols = 3, hgap = 5, vgap = 5)
  97. flexSizer.AddGrowableCol(0)
  98. self.symbology = {}
  99. for label, key in self.__SymbologyData():
  100. textLabel = wx.StaticText(panel, wx.ID_ANY, label)
  101. color = csel.ColourSelect(panel, id = wx.ID_ANY,
  102. colour = UserSettings.Get(group = 'vdigit', key = 'symbol',
  103. subkey = [key, 'color']), size = globalvar.DIALOG_COLOR_SIZE)
  104. isEnabled = UserSettings.Get(group = 'vdigit', key = 'symbol',
  105. subkey = [key, 'enabled'])
  106. if isEnabled is not None:
  107. enabled = wx.CheckBox(panel, id = wx.ID_ANY, label = "")
  108. enabled.SetValue(isEnabled)
  109. self.symbology[key] = (enabled, color)
  110. else:
  111. enabled = (1, 1)
  112. self.symbology[key] = (None, color)
  113. flexSizer.Add(textLabel, proportion = 0, flag = wx.ALIGN_CENTER_VERTICAL)
  114. flexSizer.Add(enabled, proportion = 0, flag = wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  115. flexSizer.Add(color, proportion = 0, flag = wx.ALIGN_RIGHT | wx.FIXED_MINSIZE)
  116. color.SetName("GetColour")
  117. sizer.Add(item = flexSizer, proportion = 1, flag = wx.ALL | wx.EXPAND, border = 10)
  118. panel.SetSizer(sizer)
  119. return panel
  120. def __CreateGeneralPage(self, notebook):
  121. """!Create notebook page concerning with symbology settings"""
  122. panel = wx.Panel(parent = notebook, id = wx.ID_ANY)
  123. notebook.AddPage(page = panel, text = _("General"))
  124. border = wx.BoxSizer(wx.VERTICAL)
  125. #
  126. # display section
  127. #
  128. box = wx.StaticBox (parent = panel, id = wx.ID_ANY, label = " %s " % _("Display"))
  129. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  130. flexSizer = wx.FlexGridSizer (cols = 3, hgap = 5, vgap = 5)
  131. flexSizer.AddGrowableCol(0)
  132. # line width
  133. text = wx.StaticText(parent = panel, id = wx.ID_ANY, label = _("Line width"))
  134. self.lineWidthValue = wx.SpinCtrl(parent = panel, id = wx.ID_ANY, size = (75, -1),
  135. initial = UserSettings.Get(group = 'vdigit', key = "lineWidth", subkey = 'value'),
  136. min = 1, max = 1e6)
  137. units = wx.StaticText(parent = panel, id = wx.ID_ANY, size = (115, -1),
  138. label = UserSettings.Get(group = 'vdigit', key = "lineWidth", subkey = 'units'),
  139. style = wx.ALIGN_LEFT)
  140. flexSizer.Add(text, proportion = 0, flag = wx.ALIGN_CENTER_VERTICAL)
  141. flexSizer.Add(self.lineWidthValue, proportion = 0, flag = wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  142. flexSizer.Add(units, proportion = 0, flag = wx.ALIGN_RIGHT | wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL | wx.LEFT,
  143. border = 10)
  144. sizer.Add(item = flexSizer, proportion = 1, flag = wx.ALL | wx.EXPAND, border = 1)
  145. border.Add(item = sizer, proportion = 0, flag = wx.ALL | wx.EXPAND, border = 5)
  146. #
  147. # snapping section
  148. #
  149. box = wx.StaticBox (parent = panel, id = wx.ID_ANY, label = " %s " % _("Snapping"))
  150. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  151. flexSizer = wx.FlexGridSizer(cols = 3, hgap = 5, vgap = 5)
  152. flexSizer.AddGrowableCol(0)
  153. # snapping
  154. text = wx.StaticText(parent = panel, id = wx.ID_ANY, label = _("Snapping threshold"))
  155. self.snappingValue = wx.SpinCtrl(parent = panel, id = wx.ID_ANY, size = (75, -1),
  156. initial = UserSettings.Get(group = 'vdigit', key = "snapping", subkey = 'value'),
  157. min = -1, max = 1e6)
  158. self.snappingValue.Bind(wx.EVT_SPINCTRL, self.OnChangeSnappingValue)
  159. self.snappingValue.Bind(wx.EVT_TEXT, self.OnChangeSnappingValue)
  160. self.snappingUnit = wx.Choice(parent = panel, id = wx.ID_ANY, size = (125, -1),
  161. choices = ["screen pixels", "map units"])
  162. self.snappingUnit.SetStringSelection(UserSettings.Get(group = 'vdigit', key = "snapping", subkey = 'units'))
  163. self.snappingUnit.Bind(wx.EVT_CHOICE, self.OnChangeSnappingUnits)
  164. flexSizer.Add(text, proportion = 0, flag = wx.ALIGN_CENTER_VERTICAL)
  165. flexSizer.Add(self.snappingValue, proportion = 0, flag = wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  166. flexSizer.Add(self.snappingUnit, proportion = 0, flag = wx.ALIGN_RIGHT | wx.FIXED_MINSIZE)
  167. vertexSizer = wx.BoxSizer(wx.VERTICAL)
  168. self.snapVertex = wx.CheckBox(parent = panel, id = wx.ID_ANY,
  169. label = _("Snap also to vertex"))
  170. self.snapVertex.SetValue(UserSettings.Get(group = 'vdigit', key = "snapToVertex", subkey = 'enabled'))
  171. vertexSizer.Add(item = self.snapVertex, proportion = 0, flag = wx.EXPAND)
  172. self.mapUnits = self.parent.MapWindow.Map.GetProjInfo()['units']
  173. self.snappingInfo = wx.StaticText(parent = panel, id = wx.ID_ANY,
  174. label = _("Snapping threshold is %(value).1f %(units)s") % \
  175. {'value' : self.parent.digit.GetDisplay().GetThreshold(),
  176. 'units' : self.mapUnits})
  177. vertexSizer.Add(item = self.snappingInfo, proportion = 0,
  178. flag = wx.ALL | wx.EXPAND, border = 1)
  179. sizer.Add(item = flexSizer, proportion = 1, flag = wx.EXPAND)
  180. sizer.Add(item = vertexSizer, proportion = 1, flag = wx.EXPAND)
  181. border.Add(item = sizer, proportion = 0, flag = wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border = 5)
  182. #
  183. # select box
  184. #
  185. box = wx.StaticBox (parent = panel, id = wx.ID_ANY, label = " %s " % _("Select vector features"))
  186. # feature type
  187. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  188. inSizer = wx.BoxSizer(wx.HORIZONTAL)
  189. self.selectFeature = {}
  190. for feature in ('point', 'line',
  191. 'centroid', 'boundary'):
  192. chkbox = wx.CheckBox(parent = panel, label = feature)
  193. self.selectFeature[feature] = chkbox.GetId()
  194. chkbox.SetValue(UserSettings.Get(group = 'vdigit', key = 'selectType',
  195. subkey = [feature, 'enabled']))
  196. inSizer.Add(item = chkbox, proportion = 0,
  197. flag = wx.EXPAND | wx.ALL, border = 5)
  198. sizer.Add(item = inSizer, proportion = 0, flag = wx.EXPAND)
  199. # threshold
  200. flexSizer = wx.FlexGridSizer (cols = 3, hgap = 5, vgap = 5)
  201. flexSizer.AddGrowableCol(0)
  202. text = wx.StaticText(parent = panel, id = wx.ID_ANY, label = _("Select threshold"))
  203. self.selectThreshValue = wx.SpinCtrl(parent = panel, id = wx.ID_ANY, size = (75, -1),
  204. initial = UserSettings.Get(group = 'vdigit', key = "selectThresh", subkey = 'value'),
  205. min = 1, max = 1e6)
  206. units = wx.StaticText(parent = panel, id = wx.ID_ANY, size = (115, -1),
  207. label = UserSettings.Get(group = 'vdigit', key = "lineWidth", subkey = 'units'),
  208. style = wx.ALIGN_LEFT)
  209. flexSizer.Add(text, proportion = 0, flag = wx.ALIGN_CENTER_VERTICAL)
  210. flexSizer.Add(self.selectThreshValue, proportion = 0, flag = wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  211. flexSizer.Add(units, proportion = 0, flag = wx.ALIGN_RIGHT | wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL | wx.LEFT,
  212. border = 10)
  213. self.selectIn = wx.CheckBox(parent = panel, id = wx.ID_ANY,
  214. label = _("Select only features inside of selection bounding box"))
  215. self.selectIn.SetValue(UserSettings.Get(group = 'vdigit', key = "selectInside", subkey = 'enabled'))
  216. self.selectIn.SetToolTipString(_("By default are selected all features overlapping selection bounding box "))
  217. self.checkForDupl = wx.CheckBox(parent = panel, id = wx.ID_ANY,
  218. label = _("Check for duplicates"))
  219. self.checkForDupl.SetValue(UserSettings.Get(group = 'vdigit', key = "checkForDupl", subkey = 'enabled'))
  220. sizer.Add(item = flexSizer, proportion = 0, flag = wx.EXPAND)
  221. sizer.Add(item = self.selectIn, proportion = 0, flag = wx.EXPAND | wx.ALL, border = 1)
  222. sizer.Add(item = self.checkForDupl, proportion = 0, flag = wx.EXPAND | wx.ALL, border = 1)
  223. border.Add(item = sizer, proportion = 0, flag = wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM, border = 5)
  224. #
  225. # digitize lines box
  226. #
  227. box = wx.StaticBox (parent = panel, id = wx.ID_ANY, label = " %s " % _("Digitize line features"))
  228. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  229. self.intersect = wx.CheckBox(parent = panel, label = _("Break lines at intersection"))
  230. self.intersect.SetValue(UserSettings.Get(group = 'vdigit', key = 'breakLines', subkey = 'enabled'))
  231. sizer.Add(item = self.intersect, proportion = 0, flag = wx.ALL | wx.EXPAND, border = 1)
  232. border.Add(item = sizer, proportion = 0, flag = wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM, border = 5)
  233. #
  234. # save-on-exit box
  235. #
  236. box = wx.StaticBox (parent = panel, id = wx.ID_ANY, label = " %s " % _("Save changes"))
  237. # save changes on exit?
  238. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  239. self.save = wx.CheckBox(parent = panel, label = _("Save changes on exit"))
  240. self.save.SetValue(UserSettings.Get(group = 'vdigit', key = 'saveOnExit', subkey = 'enabled'))
  241. sizer.Add(item = self.save, proportion = 0, flag = wx.ALL | wx.EXPAND, border = 1)
  242. border.Add(item = sizer, proportion = 0, flag = wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM, border = 5)
  243. panel.SetSizer(border)
  244. return panel
  245. def __CreateQueryPage(self, notebook):
  246. """!Create notebook page for query tool"""
  247. panel = wx.Panel(parent = notebook, id = wx.ID_ANY)
  248. notebook.AddPage(page = panel, text = _("Query tool"))
  249. border = wx.BoxSizer(wx.VERTICAL)
  250. #
  251. # query tool box
  252. #
  253. box = wx.StaticBox (parent = panel, id = wx.ID_ANY, label = " %s " % _("Choose query tool"))
  254. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  255. LocUnits = self.parent.MapWindow.Map.GetProjInfo()['units']
  256. self.queryBox = wx.CheckBox(parent = panel, id = wx.ID_ANY, label = _("Select by box"))
  257. self.queryBox.SetValue(UserSettings.Get(group = 'vdigit', key = "query", subkey = 'box'))
  258. sizer.Add(item = self.queryBox, proportion = 0, flag = wx.ALL | wx.EXPAND, border = 1)
  259. sizer.Add((0, 5))
  260. #
  261. # length
  262. #
  263. self.queryLength = wx.RadioButton(parent = panel, id = wx.ID_ANY, label = _("length"))
  264. self.queryLength.Bind(wx.EVT_RADIOBUTTON, self.OnChangeQuery)
  265. sizer.Add(item = self.queryLength, proportion = 0, flag = wx.ALL | wx.EXPAND, border = 1)
  266. flexSizer = wx.FlexGridSizer (cols = 4, hgap = 5, vgap = 5)
  267. flexSizer.AddGrowableCol(0)
  268. txt = wx.StaticText(parent = panel, id = wx.ID_ANY, label = _("Select lines"))
  269. self.queryLengthSL = wx.Choice (parent = panel, id = wx.ID_ANY,
  270. choices = [_("shorter than"), _("longer than")])
  271. self.queryLengthSL.SetSelection(UserSettings.Get(group = 'vdigit', key = "queryLength", subkey = 'than-selection'))
  272. self.queryLengthValue = wx.SpinCtrl(parent = panel, id = wx.ID_ANY, size = (100, -1),
  273. initial = 1,
  274. min = 0, max = 1e6)
  275. self.queryLengthValue.SetValue(UserSettings.Get(group = 'vdigit', key = "queryLength", subkey = 'thresh'))
  276. units = wx.StaticText(parent = panel, id = wx.ID_ANY, label = "%s" % LocUnits)
  277. flexSizer.Add(txt, proportion = 0, flag = wx.ALIGN_CENTER_VERTICAL)
  278. flexSizer.Add(self.queryLengthSL, proportion = 0, flag = wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  279. flexSizer.Add(self.queryLengthValue, proportion = 0, flag = wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  280. flexSizer.Add(units, proportion = 0, flag = wx.ALIGN_CENTER_VERTICAL)
  281. sizer.Add(item = flexSizer, proportion = 0, flag = wx.ALL | wx.EXPAND, border = 1)
  282. #
  283. # dangle
  284. #
  285. self.queryDangle = wx.RadioButton(parent = panel, id = wx.ID_ANY, label = _("dangle"))
  286. self.queryDangle.Bind(wx.EVT_RADIOBUTTON, self.OnChangeQuery)
  287. sizer.Add(item = self.queryDangle, proportion = 0, flag = wx.ALL | wx.EXPAND, border = 1)
  288. flexSizer = wx.FlexGridSizer (cols = 4, hgap = 5, vgap = 5)
  289. flexSizer.AddGrowableCol(0)
  290. txt = wx.StaticText(parent = panel, id = wx.ID_ANY, label = _("Select dangles"))
  291. self.queryDangleSL = wx.Choice (parent = panel, id = wx.ID_ANY,
  292. choices = [_("shorter than"), _("longer than")])
  293. self.queryDangleSL.SetSelection(UserSettings.Get(group = 'vdigit', key = "queryDangle", subkey = 'than-selection'))
  294. self.queryDangleValue = wx.SpinCtrl(parent = panel, id = wx.ID_ANY, size = (100, -1),
  295. initial = 1,
  296. min = 0, max = 1e6)
  297. self.queryDangleValue.SetValue(UserSettings.Get(group = 'vdigit', key = "queryDangle", subkey = 'thresh'))
  298. units = wx.StaticText(parent = panel, id = wx.ID_ANY, label = "%s" % LocUnits)
  299. flexSizer.Add(txt, proportion = 0, flag = wx.ALIGN_CENTER_VERTICAL)
  300. flexSizer.Add(self.queryDangleSL, proportion = 0, flag = wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  301. flexSizer.Add(self.queryDangleValue, proportion = 0, flag = wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  302. flexSizer.Add(units, proportion = 0, flag = wx.ALIGN_CENTER_VERTICAL)
  303. sizer.Add(item = flexSizer, proportion = 0, flag = wx.ALL | wx.EXPAND, border = 1)
  304. if UserSettings.Get(group = 'vdigit', key = "query", subkey = 'selection') == 0:
  305. self.queryLength.SetValue(True)
  306. else:
  307. self.queryDangle.SetValue(True)
  308. # enable & disable items
  309. self.OnChangeQuery(None)
  310. border.Add(item = sizer, proportion = 0, flag = wx.ALL | wx.EXPAND, border = 5)
  311. panel.SetSizer(border)
  312. return panel
  313. def __CreateAttributesPage(self, notebook):
  314. """!Create notebook page for query tool"""
  315. panel = wx.Panel(parent = notebook, id = wx.ID_ANY)
  316. notebook.AddPage(page = panel, text = _("Attributes"))
  317. border = wx.BoxSizer(wx.VERTICAL)
  318. #
  319. # add new record
  320. #
  321. box = wx.StaticBox (parent = panel, id = wx.ID_ANY, label = " %s " % _("Digitize new feature"))
  322. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  323. # checkbox
  324. self.addRecord = wx.CheckBox(parent = panel, id = wx.ID_ANY,
  325. label = _("Add new record into table"))
  326. self.addRecord.SetValue(UserSettings.Get(group = 'vdigit', key = "addRecord", subkey = 'enabled'))
  327. sizer.Add(item = self.addRecord, proportion = 0, flag = wx.ALL | wx.EXPAND, border = 1)
  328. # settings
  329. flexSizer = wx.FlexGridSizer(cols = 2, hgap = 3, vgap = 3)
  330. flexSizer.AddGrowableCol(0)
  331. settings = ((_("Layer"), 1), (_("Category"), 1), (_("Mode"), _("Next to use")))
  332. # layer
  333. text = wx.StaticText(parent = panel, id = wx.ID_ANY, label = _("Layer"))
  334. self.layer = wx.SpinCtrl(parent = panel, id = wx.ID_ANY, size = (125, -1),
  335. min = 1, max = 1e3)
  336. self.layer.SetValue(int(UserSettings.Get(group = 'vdigit', key = "layer", subkey = 'value')))
  337. flexSizer.Add(item = text, proportion = 0, flag = wx.ALIGN_CENTER_VERTICAL)
  338. flexSizer.Add(item = self.layer, proportion = 0,
  339. flag = wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
  340. # category number
  341. text = wx.StaticText(parent = panel, id = wx.ID_ANY, label = _("Category number"))
  342. self.category = wx.SpinCtrl(parent = panel, id = wx.ID_ANY, size = (125, -1),
  343. initial = UserSettings.Get(group = 'vdigit', key = "category", subkey = 'value'),
  344. min = -1e9, max = 1e9)
  345. if UserSettings.Get(group = 'vdigit', key = "categoryMode", subkey = 'selection') != 1:
  346. self.category.Enable(False)
  347. flexSizer.Add(item = text, proportion = 0, flag = wx.ALIGN_CENTER_VERTICAL)
  348. flexSizer.Add(item = self.category, proportion = 0,
  349. flag = wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
  350. # category mode
  351. text = wx.StaticText(parent = panel, id = wx.ID_ANY, label = _("Category mode"))
  352. self.categoryMode = wx.Choice(parent = panel, id = wx.ID_ANY, size = (125, -1),
  353. choices = [_("Next to use"), _("Manual entry"), _("No category")])
  354. self.categoryMode.SetSelection(UserSettings.Get(group = 'vdigit', key = "categoryMode", subkey = 'selection'))
  355. flexSizer.Add(item = text, proportion = 0, flag = wx.ALIGN_CENTER_VERTICAL)
  356. flexSizer.Add(item = self.categoryMode, proportion = 0,
  357. flag = wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
  358. sizer.Add(item = flexSizer, proportion = 1, flag = wx.ALL | wx.EXPAND, border = 1)
  359. border.Add(item = sizer, proportion = 0,
  360. flag = wx.ALL | wx.EXPAND, border = 5)
  361. #
  362. # digitize new area
  363. #
  364. box = wx.StaticBox (parent = panel, id = wx.ID_ANY, label = " %s " % _("Digitize new area"))
  365. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  366. # add centroid
  367. self.addCentroid = wx.CheckBox(parent = panel, id = wx.ID_ANY,
  368. label = _("Add centroid to left/right area"))
  369. self.addCentroid.SetValue(UserSettings.Get(group = 'vdigit', key = "addCentroid", subkey = 'enabled'))
  370. sizer.Add(item = self.addCentroid, proportion = 0, flag = wx.ALL | wx.EXPAND, border = 1)
  371. # attach category to boundary
  372. self.catBoundary = wx.CheckBox(parent = panel, id = wx.ID_ANY,
  373. label = _("Do not attach category to boundaries"))
  374. self.catBoundary.SetValue(UserSettings.Get(group = 'vdigit', key = "catBoundary", subkey = 'enabled'))
  375. sizer.Add(item = self.catBoundary, proportion = 0, flag = wx.ALL | wx.EXPAND, border = 1)
  376. border.Add(item = sizer, proportion = 0,
  377. flag = wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border = 5)
  378. #
  379. # delete existing record
  380. #
  381. box = wx.StaticBox (parent = panel, id = wx.ID_ANY, label = " %s " % _("Delete existing feature(s)"))
  382. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  383. # checkbox
  384. self.deleteRecord = wx.CheckBox(parent = panel, id = wx.ID_ANY,
  385. label = _("Delete record from table"))
  386. self.deleteRecord.SetValue(UserSettings.Get(group = 'vdigit', key = "delRecord", subkey = 'enabled'))
  387. sizer.Add(item = self.deleteRecord, proportion = 0, flag = wx.ALL | wx.EXPAND, border = 1)
  388. border.Add(item = sizer, proportion = 0,
  389. flag = wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border = 5)
  390. #
  391. # geometry attributes (currently only length and area are supported)
  392. #
  393. box = wx.StaticBox (parent = panel, id = wx.ID_ANY,
  394. label = " %s " % _("Geometry attributes"))
  395. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  396. gridSizer = wx.GridBagSizer(hgap = 3, vgap = 3)
  397. gridSizer.AddGrowableCol(0)
  398. self.geomAttrb = { 'length' : { 'label' : _('length') },
  399. 'area' : { 'label' : _('area') },
  400. 'perimeter' : { 'label' : _('perimeter') } }
  401. digitToolbar = self.parent.toolbars['vdigit']
  402. try:
  403. vectorName = digitToolbar.GetLayer().GetName()
  404. except AttributeError:
  405. vectorName = None # no vector selected for editing
  406. layer = UserSettings.Get(group = 'vdigit', key = "layer", subkey = 'value')
  407. mapLayer = self.parent.toolbars['vdigit'].GetLayer()
  408. tree = self.parent.tree
  409. item = tree.FindItemByData('maplayer', mapLayer)
  410. row = 0
  411. for attrb in ['length', 'area', 'perimeter']:
  412. # checkbox
  413. check = wx.CheckBox(parent = panel, id = wx.ID_ANY,
  414. label = self.geomAttrb[attrb]['label'])
  415. ### self.deleteRecord.SetValue(UserSettings.Get(group='vdigit', key="delRecord", subkey='enabled'))
  416. check.Bind(wx.EVT_CHECKBOX, self.OnGeomAttrb)
  417. # column (only numeric)
  418. column = gselect.ColumnSelect(parent = panel, size = (200, -1))
  419. column.InsertColumns(vector = vectorName,
  420. layer = layer, excludeKey = True,
  421. type = ['integer', 'double precision'])
  422. # units
  423. if attrb == 'area':
  424. choices = Units.GetUnitsList('area')
  425. else:
  426. choices = Units.GetUnitsList('length')
  427. win_units = wx.Choice(parent = panel, id = wx.ID_ANY,
  428. choices = choices, size = (120, -1))
  429. # default values
  430. check.SetValue(False)
  431. if item and tree.GetPyData(item)[0]['vdigit'] and \
  432. tree.GetPyData(item)[0]['vdigit'].has_key('geomAttr') and \
  433. tree.GetPyData(item)[0]['vdigit']['geomAttr'].has_key(attrb):
  434. check.SetValue(True)
  435. column.SetStringSelection(tree.GetPyData(item)[0]['vdigit']['geomAttr'][attrb]['column'])
  436. if attrb == 'area':
  437. type = 'area'
  438. else:
  439. type = 'length'
  440. unitsIdx = Units.GetUnitsIndex(type,
  441. tree.GetPyData(item)[0]['vdigit']['geomAttr'][attrb]['units'])
  442. win_units.SetSelection(unitsIdx)
  443. if not vectorName:
  444. check.Enable(False)
  445. column.Enable(False)
  446. if not check.IsChecked():
  447. column.Enable(False)
  448. self.geomAttrb[attrb]['check'] = check.GetId()
  449. self.geomAttrb[attrb]['column'] = column.GetId()
  450. self.geomAttrb[attrb]['units'] = win_units.GetId()
  451. gridSizer.Add(item = check,
  452. flag = wx.ALIGN_CENTER_VERTICAL,
  453. pos = (row, 0))
  454. gridSizer.Add(item = column,
  455. pos = (row, 1))
  456. gridSizer.Add(item = win_units,
  457. pos = (row, 2))
  458. row += 1
  459. note = '\n'.join(textwrap.wrap(_("Note: These settings are stored "
  460. " in the workspace not in the vector digitizer "
  461. "preferences."), 55))
  462. gridSizer.Add(item = wx.StaticText(parent = panel, id = wx.ID_ANY,
  463. label = note),
  464. pos = (3, 0), span = (1, 3))
  465. sizer.Add(item = gridSizer, proportion = 1,
  466. flag = wx.ALL | wx.EXPAND, border = 1)
  467. border.Add(item = sizer, proportion = 0,
  468. flag = wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border = 5)
  469. # bindings
  470. self.Bind(wx.EVT_CHECKBOX, self.OnChangeAddRecord, self.addRecord)
  471. self.Bind(wx.EVT_CHOICE, self.OnChangeCategoryMode, self.categoryMode)
  472. self.Bind(wx.EVT_SPINCTRL, self.OnChangeLayer, self.layer)
  473. panel.SetSizer(border)
  474. return panel
  475. def __SymbologyData(self):
  476. """!Data for __CreateSymbologyPage()
  477. label | checkbox | color
  478. """
  479. return (
  480. # ("Background", "symbolBackground"),
  481. (_("Highlight"), "highlight"),
  482. (_("Highlight (duplicates)"), "highlightDupl"),
  483. (_("Point"), "point"),
  484. (_("Line"), "line"),
  485. (_("Boundary (no area)"), "boundaryNo"),
  486. (_("Boundary (one area)"), "boundaryOne"),
  487. (_("Boundary (two areas)"), "boundaryTwo"),
  488. (_("Centroid (in area)"), "centroidIn"),
  489. (_("Centroid (outside area)"), "centroidOut"),
  490. (_("Centroid (duplicate in area)"), "centroidDup"),
  491. (_("Node (one line)"), "nodeOne"),
  492. (_("Node (two lines)"), "nodeTwo"),
  493. (_("Vertex"), "vertex"),
  494. (_("Area (closed boundary + centroid)"), "area"),
  495. (_("Direction"), "direction"),)
  496. def OnGeomAttrb(self, event):
  497. """!Register geometry attributes (enable/disable)"""
  498. checked = event.IsChecked()
  499. id = event.GetId()
  500. key = None
  501. for attrb, val in self.geomAttrb.iteritems():
  502. if val['check'] == id:
  503. key = attrb
  504. break
  505. column = self.FindWindowById(self.geomAttrb[key]['column'])
  506. if checked:
  507. column.Enable()
  508. else:
  509. column.Enable(False)
  510. def OnChangeCategoryMode(self, event):
  511. """!Change category mode"""
  512. mode = event.GetSelection()
  513. UserSettings.Set(group = 'vdigit', key = "categoryMode", subkey = 'selection', value = mode)
  514. if mode == 1: # manual entry
  515. self.category.Enable(True)
  516. elif self.category.IsEnabled(): # disable
  517. self.category.Enable(False)
  518. if mode == 2 and self.addRecord.IsChecked(): # no category
  519. self.addRecord.SetValue(False)
  520. self.parent.digit.SetCategory()
  521. self.category.SetValue(UserSettings.Get(group = 'vdigit', key = 'category', subkey = 'value'))
  522. def OnChangeLayer(self, event):
  523. """!Layer changed"""
  524. layer = event.GetInt()
  525. if layer > 0:
  526. UserSettings.Set(group = 'vdigit', key = 'layer', subkey = 'value', value = layer)
  527. self.parent.digit.SetCategory()
  528. self.category.SetValue(UserSettings.Get(group = 'vdigit', key = 'category', subkey = 'value'))
  529. event.Skip()
  530. def OnChangeAddRecord(self, event):
  531. """!Checkbox 'Add new record' status changed"""
  532. self.category.SetValue(self.parent.digit.SetCategory())
  533. def OnChangeSnappingValue(self, event):
  534. """!Change snapping value - update static text"""
  535. value = self.snappingValue.GetValue()
  536. if value < 0:
  537. region = self.parent.MapWindow.Map.GetRegion()
  538. res = (region['nsres'] + region['ewres']) / 2.
  539. threshold = self.parent.digit.GetDisplay().GetThreshold(value = res)
  540. else:
  541. if self.snappingUnit.GetStringSelection() == "map units":
  542. threshold = value
  543. else:
  544. threshold = self.parent.digit.GetDisplay().GetThreshold(value = value)
  545. if value == 0:
  546. self.snappingInfo.SetLabel(_("Snapping disabled"))
  547. elif value < 0:
  548. self.snappingInfo.SetLabel(_("Snapping threshold is %(value).1f %(units)s "
  549. "(based on comp. resolution)") %
  550. {'value' : threshold,
  551. 'units' : self.mapUnits.lower()})
  552. else:
  553. self.snappingInfo.SetLabel(_("Snapping threshold is %(value).1f %(units)s") %
  554. {'value' : threshold,
  555. 'units' : self.mapUnits.lower()})
  556. event.Skip()
  557. def OnChangeSnappingUnits(self, event):
  558. """!Snapping units change -> update static text"""
  559. value = self.snappingValue.GetValue()
  560. units = self.snappingUnit.GetStringSelection()
  561. threshold = self.parent.digit.GetDisplay().GetThreshold(value = value, units = units)
  562. if units == "map units":
  563. self.snappingInfo.SetLabel(_("Snapping threshold is %(value).1f %(units)s") %
  564. {'value' : value,
  565. 'units' : self.mapUnits})
  566. else:
  567. self.snappingInfo.SetLabel(_("Snapping threshold is %(value).1f %(units)s") %
  568. {'value' : threshold,
  569. 'units' : self.mapUnits})
  570. event.Skip()
  571. def OnChangeQuery(self, event):
  572. """!Change query"""
  573. if self.queryLength.GetValue():
  574. # length
  575. self.queryLengthSL.Enable(True)
  576. self.queryLengthValue.Enable(True)
  577. self.queryDangleSL.Enable(False)
  578. self.queryDangleValue.Enable(False)
  579. else:
  580. # dangle
  581. self.queryLengthSL.Enable(False)
  582. self.queryLengthValue.Enable(False)
  583. self.queryDangleSL.Enable(True)
  584. self.queryDangleValue.Enable(True)
  585. def OnSave(self, event):
  586. """!Button 'Save' clicked"""
  587. self.UpdateSettings()
  588. self.parent.toolbars['vdigit'].settingsDialog = None
  589. fileSettings = {}
  590. UserSettings.ReadSettingsFile(settings = fileSettings)
  591. fileSettings['vdigit'] = UserSettings.Get(group = 'vdigit')
  592. file = UserSettings.SaveToFile(fileSettings)
  593. self.parent.GetLayerManager().goutput.WriteLog(_('Vector digitizer settings saved to file <%s>.') % file)
  594. self.Destroy()
  595. event.Skip()
  596. def OnApply(self, event):
  597. """!Button 'Apply' clicked"""
  598. self.UpdateSettings()
  599. def OnCancel(self, event):
  600. """!Button 'Cancel' clicked"""
  601. self.parent.toolbars['vdigit'].settingsDialog = None
  602. self.Destroy()
  603. if event:
  604. event.Skip()
  605. def UpdateSettings(self):
  606. """!Update digitizer settings
  607. """
  608. # symbology
  609. for key, (enabled, color) in self.symbology.iteritems():
  610. if enabled:
  611. UserSettings.Set(group = 'vdigit', key = 'symbol',
  612. subkey = [key, 'enabled'],
  613. value = enabled.IsChecked())
  614. UserSettings.Set(group = 'vdigit', key = 'symbol',
  615. subkey = [key, 'color'],
  616. value = tuple(color.GetColour()))
  617. else:
  618. UserSettings.Set(group = 'vdigit', key = 'symbol',
  619. subkey = [key, 'color'],
  620. value = tuple(color.GetColour()))
  621. # display
  622. UserSettings.Set(group = 'vdigit', key = "lineWidth", subkey = 'value',
  623. value = int(self.lineWidthValue.GetValue()))
  624. # snapping
  625. UserSettings.Set(group = 'vdigit', key = "snapping", subkey = 'value',
  626. value = int(self.snappingValue.GetValue()))
  627. UserSettings.Set(group = 'vdigit', key = "snapping", subkey = 'units',
  628. value = self.snappingUnit.GetStringSelection())
  629. UserSettings.Set(group = 'vdigit', key = "snapToVertex", subkey = 'enabled',
  630. value = self.snapVertex.IsChecked())
  631. # digitize new feature
  632. UserSettings.Set(group = 'vdigit', key = "addRecord", subkey = 'enabled',
  633. value = self.addRecord.IsChecked())
  634. UserSettings.Set(group = 'vdigit', key = "layer", subkey = 'value',
  635. value = int(self.layer.GetValue()))
  636. UserSettings.Set(group = 'vdigit', key = "category", subkey = 'value',
  637. value = int(self.category.GetValue()))
  638. UserSettings.Set(group = 'vdigit', key = "categoryMode", subkey = 'selection',
  639. value = self.categoryMode.GetSelection())
  640. # digitize new area
  641. UserSettings.Set(group = 'vdigit', key = "addCentroid", subkey = 'enabled',
  642. value = self.addCentroid.IsChecked())
  643. UserSettings.Set(group = 'vdigit', key = "catBoundary", subkey = 'enabled',
  644. value = self.catBoundary.IsChecked())
  645. # delete existing feature
  646. UserSettings.Set(group = 'vdigit', key = "delRecord", subkey = 'enabled',
  647. value = self.deleteRecord.IsChecked())
  648. # geometry attributes (workspace)
  649. mapLayer = self.parent.toolbars['vdigit'].GetLayer()
  650. tree = self.parent.tree
  651. item = tree.FindItemByData('maplayer', mapLayer)
  652. for key, val in self.geomAttrb.iteritems():
  653. checked = self.FindWindowById(val['check']).IsChecked()
  654. column = self.FindWindowById(val['column']).GetValue()
  655. unitsIdx = self.FindWindowById(val['units']).GetSelection()
  656. if item and not tree.GetPyData(item)[0]['vdigit']:
  657. tree.GetPyData(item)[0]['vdigit'] = { 'geomAttr' : dict() }
  658. if checked: # enable
  659. if key == 'area':
  660. type = key
  661. else:
  662. type = 'length'
  663. unitsKey = Units.GetUnitsKey(type, unitsIdx)
  664. tree.GetPyData(item)[0]['vdigit']['geomAttr'][key] = { 'column' : column,
  665. 'units' : unitsKey }
  666. else:
  667. if item and tree.GetPyData(item)[0]['vdigit'] and \
  668. tree.GetPyData(item)[0]['vdigit']['geomAttr'].has_key(key):
  669. del tree.GetPyData(item)[0]['vdigit']['geomAttr'][key]
  670. # query tool
  671. if self.queryLength.GetValue():
  672. UserSettings.Set(group = 'vdigit', key = "query", subkey = 'selection',
  673. value = 0)
  674. else:
  675. UserSettings.Set(group = 'vdigit', key = "query", subkey = 'type',
  676. value = 1)
  677. UserSettings.Set(group = 'vdigit', key = "query", subkey = 'box',
  678. value = self.queryBox.IsChecked())
  679. UserSettings.Set(group = 'vdigit', key = "queryLength", subkey = 'than-selection',
  680. value = self.queryLengthSL.GetSelection())
  681. UserSettings.Set(group = 'vdigit', key = "queryLength", subkey = 'thresh',
  682. value = int(self.queryLengthValue.GetValue()))
  683. UserSettings.Set(group = 'vdigit', key = "queryDangle", subkey = 'than-selection',
  684. value = self.queryDangleSL.GetSelection())
  685. UserSettings.Set(group = 'vdigit', key = "queryDangle", subkey = 'thresh',
  686. value = int(self.queryDangleValue.GetValue()))
  687. # select features
  688. for feature in ('point', 'line',
  689. 'centroid', 'boundary'):
  690. UserSettings.Set(group = 'vdigit', key = 'selectType',
  691. subkey = [feature, 'enabled'],
  692. value = self.FindWindowById(self.selectFeature[feature]).IsChecked())
  693. UserSettings.Set(group = 'vdigit', key = "selectThresh", subkey = 'value',
  694. value = int(self.selectThreshValue.GetValue()))
  695. UserSettings.Set(group = 'vdigit', key = "checkForDupl", subkey = 'enabled',
  696. value = self.checkForDupl.IsChecked())
  697. UserSettings.Set(group = 'vdigit', key = "selectInside", subkey = 'enabled',
  698. value = self.selectIn.IsChecked())
  699. # on-exit
  700. UserSettings.Set(group = 'vdigit', key = "saveOnExit", subkey = 'enabled',
  701. value = self.save.IsChecked())
  702. # break lines
  703. UserSettings.Set(group = 'vdigit', key = "breakLines", subkey = 'enabled',
  704. value = self.intersect.IsChecked())
  705. self.parent.digit.GetDisplay().UpdateSettings()
  706. # redraw map if auto-rendering is enabled
  707. if self.parent.statusbarWin['render'].GetValue():
  708. self.parent.OnRender(None)
  709. class VDigitCategoryDialog(wx.Dialog, listmix.ColumnSorterMixin):
  710. def __init__(self, parent, title,
  711. map, query = None, cats = None,
  712. pos = wx.DefaultPosition,
  713. style = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER):
  714. """!Dialog used to display/modify categories of vector objects
  715. @param parent
  716. @param title dialog title
  717. @param query {coordinates, qdist} - used by v.edit/v.what
  718. @param cats directory of lines (layer/categories) - used by vdigit
  719. @param pos
  720. @param style
  721. """
  722. # parent
  723. self.parent = parent # mapdisplay.BufferedWindow class instance
  724. # map name
  725. self.map = map
  726. # line : {layer: [categories]}
  727. self.cats = {}
  728. # do not display dialog if no line is found (-> self.cats)
  729. if cats is None:
  730. if self.__GetCategories(query[0], query[1]) == 0 or not self.line:
  731. Debug.msg(3, "VDigitCategoryDialog(): nothing found!")
  732. else:
  733. self.cats = cats
  734. for line in cats.keys():
  735. for layer in cats[line].keys():
  736. self.cats[line][layer] = list(cats[line][layer])
  737. layers = []
  738. for layer in self.parent.parent.digit.GetLayers():
  739. layers.append(str(layer))
  740. # make copy of cats (used for 'reload')
  741. self.cats_orig = copy.deepcopy(self.cats)
  742. wx.Dialog.__init__(self, parent = self.parent, id = wx.ID_ANY, title = title,
  743. style = style, pos = pos)
  744. # list of categories
  745. box = wx.StaticBox(parent = self, id = wx.ID_ANY,
  746. label = " %s " % _("List of categories - right-click to delete"))
  747. listSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  748. self.list = CategoryListCtrl(parent = self, id = wx.ID_ANY,
  749. style = wx.LC_REPORT |
  750. wx.BORDER_NONE |
  751. wx.LC_SORT_ASCENDING |
  752. wx.LC_HRULES |
  753. wx.LC_VRULES)
  754. # sorter
  755. self.fid = self.cats.keys()[0]
  756. self.itemDataMap = self.list.Populate(self.cats[self.fid])
  757. listmix.ColumnSorterMixin.__init__(self, 2)
  758. self.fidMulti = wx.Choice(parent = self, id = wx.ID_ANY,
  759. size = (150, -1))
  760. self.fidMulti.Bind(wx.EVT_CHOICE, self.OnFeature)
  761. self.fidText = wx.StaticText(parent = self, id = wx.ID_ANY)
  762. if len(self.cats.keys()) == 1:
  763. self.fidMulti.Show(False)
  764. self.fidText.SetLabel(str(self.fid))
  765. else:
  766. self.fidText.Show(False)
  767. choices = []
  768. for fid in self.cats.keys():
  769. choices.append(str(fid))
  770. self.fidMulti.SetItems(choices)
  771. self.fidMulti.SetSelection(0)
  772. listSizer.Add(item = self.list, proportion = 1, flag = wx.EXPAND)
  773. # add new category
  774. box = wx.StaticBox(parent = self, id = wx.ID_ANY,
  775. label = " %s " % _("Add new category"))
  776. addSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  777. flexSizer = wx.FlexGridSizer (cols = 5, hgap = 5, vgap = 5)
  778. flexSizer.AddGrowableCol(3)
  779. layerNewTxt = wx.StaticText(parent = self, id = wx.ID_ANY,
  780. label = "%s:" % _("Layer"))
  781. self.layerNew = wx.Choice(parent = self, id = wx.ID_ANY, size = (75, -1),
  782. choices = layers)
  783. if len(layers) > 0:
  784. self.layerNew.SetSelection(0)
  785. catNewTxt = wx.StaticText(parent = self, id = wx.ID_ANY,
  786. label = "%s:" % _("Category"))
  787. try:
  788. newCat = max(self.cats[self.fid][1]) + 1
  789. except KeyError:
  790. newCat = 1
  791. self.catNew = wx.SpinCtrl(parent = self, id = wx.ID_ANY, size = (75, -1),
  792. initial = newCat, min = 0, max = 1e9)
  793. btnAddCat = wx.Button(self, wx.ID_ADD)
  794. flexSizer.Add(item = layerNewTxt, proportion = 0,
  795. flag = wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
  796. flexSizer.Add(item = self.layerNew, proportion = 0,
  797. flag = wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
  798. flexSizer.Add(item = catNewTxt, proportion = 0,
  799. flag = wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_RIGHT | wx.LEFT,
  800. border = 10)
  801. flexSizer.Add(item = self.catNew, proportion = 0,
  802. flag = wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
  803. flexSizer.Add(item = btnAddCat, proportion = 0,
  804. flag = wx.EXPAND | wx.ALIGN_RIGHT | wx.FIXED_MINSIZE)
  805. addSizer.Add(item = flexSizer, proportion = 1, flag = wx.ALL | wx.EXPAND, border = 5)
  806. # buttons
  807. btnApply = wx.Button(self, wx.ID_APPLY)
  808. btnApply.SetToolTipString(_("Apply changes"))
  809. btnCancel = wx.Button(self, wx.ID_CANCEL)
  810. btnCancel.SetToolTipString(_("Ignore changes and close dialog"))
  811. btnOk = wx.Button(self, wx.ID_OK)
  812. btnOk.SetToolTipString(_("Apply changes and close dialog"))
  813. btnOk.SetDefault()
  814. # sizers
  815. btnSizer = wx.StdDialogButtonSizer()
  816. btnSizer.AddButton(btnCancel)
  817. #btnSizer.AddButton(btnReload)
  818. #btnSizer.SetNegativeButton(btnReload)
  819. btnSizer.AddButton(btnApply)
  820. btnSizer.AddButton(btnOk)
  821. btnSizer.Realize()
  822. mainSizer = wx.BoxSizer(wx.VERTICAL)
  823. mainSizer.Add(item = listSizer, proportion = 1,
  824. flag = wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border = 5)
  825. mainSizer.Add(item = addSizer, proportion = 0,
  826. flag = wx.EXPAND | wx.ALIGN_CENTER |
  827. wx.LEFT | wx.RIGHT | wx.BOTTOM, border = 5)
  828. fidSizer = wx.BoxSizer(wx.HORIZONTAL)
  829. fidSizer.Add(item = wx.StaticText(parent = self, id = wx.ID_ANY,
  830. label = _("Feature id:")),
  831. proportion = 0, border = 5,
  832. flag = wx.ALIGN_CENTER_VERTICAL)
  833. fidSizer.Add(item = self.fidMulti, proportion = 0,
  834. flag = wx.EXPAND | wx.ALL, border = 5)
  835. fidSizer.Add(item = self.fidText, proportion = 0,
  836. flag = wx.EXPAND | wx.ALL, border = 5)
  837. mainSizer.Add(item = fidSizer, proportion = 0,
  838. flag = wx.EXPAND | wx.ALL, border = 5)
  839. mainSizer.Add(item = btnSizer, proportion = 0,
  840. flag = wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border = 5)
  841. self.SetSizer(mainSizer)
  842. mainSizer.Fit(self)
  843. self.SetAutoLayout(True)
  844. # set min size for dialog
  845. self.SetMinSize(self.GetBestSize())
  846. # bindings
  847. # buttons
  848. #btnReload.Bind(wx.EVT_BUTTON, self.OnReload)
  849. btnApply.Bind(wx.EVT_BUTTON, self.OnApply)
  850. btnOk.Bind(wx.EVT_BUTTON, self.OnOK)
  851. btnAddCat.Bind(wx.EVT_BUTTON, self.OnAddCat)
  852. btnCancel.Bind(wx.EVT_BUTTON, self.OnCancel)
  853. # list
  854. # self.Bind(wx.EVT_LIST_ITEM_SELECTED, self.OnItemSelected, self.list)
  855. # self.list.Bind(wx.EVT_RIGHT_DOWN, self.OnRightDown)
  856. self.list.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnRightUp) #wxMSW
  857. self.list.Bind(wx.EVT_RIGHT_UP, self.OnRightUp) #wxGTK
  858. self.Bind(wx.EVT_LIST_BEGIN_LABEL_EDIT, self.OnBeginEdit, self.list)
  859. self.Bind(wx.EVT_LIST_END_LABEL_EDIT, self.OnEndEdit, self.list)
  860. self.Bind(wx.EVT_LIST_COL_CLICK, self.OnColClick, self.list)
  861. def GetListCtrl(self):
  862. """!Used by ColumnSorterMixin"""
  863. return self.list
  864. def OnColClick(self, event):
  865. """!Click on column header (order by)"""
  866. event.Skip()
  867. def OnBeginEdit(self, event):
  868. """!Editing of item started"""
  869. event.Allow()
  870. def OnEndEdit(self, event):
  871. """!Finish editing of item"""
  872. itemIndex = event.GetIndex()
  873. layerOld = int (self.list.GetItem(itemIndex, 0).GetText())
  874. catOld = int (self.list.GetItem(itemIndex, 1).GetText())
  875. if event.GetColumn() == 0:
  876. layerNew = int(event.GetLabel())
  877. catNew = catOld
  878. else:
  879. layerNew = layerOld
  880. catNew = int(event.GetLabel())
  881. try:
  882. if layerNew not in self.cats[self.fid].keys():
  883. self.cats[self.fid][layerNew] = []
  884. self.cats[self.fid][layerNew].append(catNew)
  885. self.cats[self.fid][layerOld].remove(catOld)
  886. except:
  887. event.Veto()
  888. self.list.SetStringItem(itemIndex, 0, str(layerNew))
  889. self.list.SetStringItem(itemIndex, 1, str(catNew))
  890. dlg = wx.MessageDialog(self, _("Unable to add new layer/category <%(layer)s/%(category)s>.\n"
  891. "Layer and category number must be integer.\n"
  892. "Layer number must be greater then zero.") %
  893. { 'layer': self.layerNew.GetStringSelection(),
  894. 'category' : str(self.catNew.GetValue()) },
  895. _("Error"), wx.OK | wx.ICON_ERROR)
  896. dlg.ShowModal()
  897. dlg.Destroy()
  898. return False
  899. def OnRightDown(self, event):
  900. """!Mouse right button down"""
  901. x = event.GetX()
  902. y = event.GetY()
  903. item, flags = self.list.HitTest((x, y))
  904. if item != wx.NOT_FOUND and \
  905. flags & wx.LIST_HITTEST_ONITEM:
  906. self.list.Select(item)
  907. event.Skip()
  908. def OnRightUp(self, event):
  909. """!Mouse right button up"""
  910. if not hasattr(self, "popupID1"):
  911. self.popupID1 = wx.NewId()
  912. self.popupID2 = wx.NewId()
  913. self.popupID3 = wx.NewId()
  914. self.Bind(wx.EVT_MENU, self.OnItemDelete, id = self.popupID1)
  915. self.Bind(wx.EVT_MENU, self.OnItemDeleteAll, id = self.popupID2)
  916. self.Bind(wx.EVT_MENU, self.OnReload, id = self.popupID3)
  917. # generate popup-menu
  918. menu = wx.Menu()
  919. menu.Append(self.popupID1, _("Delete selected"))
  920. if self.list.GetFirstSelected() == -1:
  921. menu.Enable(self.popupID1, False)
  922. menu.Append(self.popupID2, _("Delete all"))
  923. menu.AppendSeparator()
  924. menu.Append(self.popupID3, _("Reload"))
  925. self.PopupMenu(menu)
  926. menu.Destroy()
  927. def OnItemSelected(self, event):
  928. """!Item selected"""
  929. event.Skip()
  930. def OnItemDelete(self, event):
  931. """!Delete selected item(s) from the list (layer/category pair)"""
  932. item = self.list.GetFirstSelected()
  933. while item != -1:
  934. layer = int (self.list.GetItem(item, 0).GetText())
  935. cat = int (self.list.GetItem(item, 1).GetText())
  936. self.list.DeleteItem(item)
  937. self.cats[self.fid][layer].remove(cat)
  938. item = self.list.GetFirstSelected()
  939. event.Skip()
  940. def OnItemDeleteAll(self, event):
  941. """!Delete all items from the list"""
  942. self.list.DeleteAllItems()
  943. self.cats[self.fid] = {}
  944. event.Skip()
  945. def OnFeature(self, event):
  946. """!Feature id changed (on duplicates)"""
  947. self.fid = int(event.GetString())
  948. self.itemDataMap = self.list.Populate(self.cats[self.fid],
  949. update = True)
  950. try:
  951. newCat = max(self.cats[self.fid][1]) + 1
  952. except KeyError:
  953. newCat = 1
  954. self.catNew.SetValue(newCat)
  955. event.Skip()
  956. def __GetCategories(self, coords, qdist):
  957. """!Get layer/category pairs for all available
  958. layers
  959. Return True line found or False if not found"""
  960. ret = gcmd.RunCommand('v.what',
  961. parent = self,
  962. quiet = True,
  963. map = self.map,
  964. east_north = '%f,%f' % \
  965. (float(coords[0]), float(coords[1])),
  966. distance = qdist)
  967. if not ret:
  968. return False
  969. for item in ret.splitlines():
  970. litem = item.lower()
  971. if "id:" in litem: # get line id
  972. self.line = int(item.split(':')[1].strip())
  973. elif "layer:" in litem: # add layer
  974. layer = int(item.split(':')[1].strip())
  975. if layer not in self.cats.keys():
  976. self.cats[layer] = []
  977. elif "category:" in litem: # add category
  978. self.cats[layer].append(int(item.split(':')[1].strip()))
  979. return True
  980. def OnReload(self, event):
  981. """!Reload button pressed"""
  982. # restore original list
  983. self.cats = copy.deepcopy(self.cats_orig)
  984. # polulate list
  985. self.itemDataMap = self.list.Populate(self.cats[self.fid],
  986. update = True)
  987. event.Skip()
  988. def OnCancel(self, event):
  989. """!Cancel button pressed"""
  990. self.parent.parent.dialogs['category'] = None
  991. if self.parent.parent.digit:
  992. self.parent.parent.digit.GetDisplay().SetSelected([])
  993. self.parent.UpdateMap(render = False)
  994. else:
  995. self.parent.parent.OnRender(None)
  996. self.Close()
  997. def OnApply(self, event):
  998. """!Apply button pressed"""
  999. for fid in self.cats.keys():
  1000. newfid = self.ApplyChanges(fid)
  1001. if fid == self.fid:
  1002. self.fid = newfid
  1003. def ApplyChanges(self, fid):
  1004. cats = self.cats[fid]
  1005. cats_orig = self.cats_orig[fid]
  1006. # action : (catsFrom, catsTo)
  1007. check = {'catadd': (cats, cats_orig),
  1008. 'catdel': (cats_orig, cats)}
  1009. newfid = -1
  1010. # add/delete new category
  1011. for action, catsCurr in check.iteritems():
  1012. for layer in catsCurr[0].keys():
  1013. catList = []
  1014. for cat in catsCurr[0][layer]:
  1015. if layer not in catsCurr[1].keys() or \
  1016. cat not in catsCurr[1][layer]:
  1017. catList.append(cat)
  1018. if catList != []:
  1019. if action == 'catadd':
  1020. add = True
  1021. else:
  1022. add = False
  1023. newfid = self.parent.parent.digit.SetLineCats(fid, layer,
  1024. catList, add)
  1025. if len(self.cats.keys()) == 1:
  1026. self.fidText.SetLabel("%d" % newfid)
  1027. else:
  1028. choices = self.fidMulti.GetItems()
  1029. choices[choices.index(str(fid))] = str(newfid)
  1030. self.fidMulti.SetItems(choices)
  1031. self.fidMulti.SetStringSelection(str(newfid))
  1032. self.cats[newfid] = self.cats[fid]
  1033. del self.cats[fid]
  1034. fid = newfid
  1035. if self.fid < 0:
  1036. wx.MessageBox(parent = self, message = _("Unable to update vector map."),
  1037. caption = _("Error"), style = wx.OK | wx.ICON_ERROR)
  1038. self.cats_orig[fid] = copy.deepcopy(cats)
  1039. return newfid
  1040. def OnOK(self, event):
  1041. """!OK button pressed"""
  1042. self.OnApply(event)
  1043. self.OnCancel(event)
  1044. def OnAddCat(self, event):
  1045. """!Button 'Add' new category pressed"""
  1046. try:
  1047. layer = int(self.layerNew.GetStringSelection())
  1048. cat = int(self.catNew.GetValue())
  1049. if layer <= 0:
  1050. raise ValueError
  1051. except ValueError:
  1052. dlg = wx.MessageDialog(self, _("Unable to add new layer/category <%(layer)s/%(category)s>.\n"
  1053. "Layer and category number must be integer.\n"
  1054. "Layer number must be greater then zero.") %
  1055. {'layer' : str(self.layerNew.GetValue()),
  1056. 'category' : str(self.catNew.GetValue())},
  1057. _("Error"), wx.OK | wx.ICON_ERROR)
  1058. dlg.ShowModal()
  1059. dlg.Destroy()
  1060. return False
  1061. if layer not in self.cats[self.fid].keys():
  1062. self.cats[self.fid][layer] = []
  1063. self.cats[self.fid][layer].append(cat)
  1064. # reload list
  1065. self.itemDataMap = self.list.Populate(self.cats[self.fid],
  1066. update = True)
  1067. # update category number for add
  1068. self.catNew.SetValue(cat + 1)
  1069. event.Skip()
  1070. return True
  1071. def GetLine(self):
  1072. """!Get id of selected line of 'None' if no line is selected"""
  1073. return self.cats.keys()
  1074. def UpdateDialog(self, query = None, cats = None):
  1075. """!Update dialog
  1076. @param query {coordinates, distance} - v.edit/v.what
  1077. @param cats directory layer/cats - vdigit
  1078. Return True if updated otherwise False
  1079. """
  1080. # line: {layer: [categories]}
  1081. self.cats = {}
  1082. # do not display dialog if no line is found (-> self.cats)
  1083. if cats is None:
  1084. ret = self.__GetCategories(query[0], query[1])
  1085. else:
  1086. self.cats = cats
  1087. for line in cats.keys():
  1088. for layer in cats[line].keys():
  1089. self.cats[line][layer] = list(cats[line][layer])
  1090. ret = 1
  1091. if ret == 0 or len(self.cats.keys()) < 1:
  1092. Debug.msg(3, "VDigitCategoryDialog(): nothing found!")
  1093. return False
  1094. # make copy of cats (used for 'reload')
  1095. self.cats_orig = copy.deepcopy(self.cats)
  1096. # polulate list
  1097. self.fid = self.cats.keys()[0]
  1098. self.itemDataMap = self.list.Populate(self.cats[self.fid],
  1099. update = True)
  1100. try:
  1101. newCat = max(self.cats[self.fid][1]) + 1
  1102. except KeyError:
  1103. newCat = 1
  1104. self.catNew.SetValue(newCat)
  1105. if len(self.cats.keys()) == 1:
  1106. self.fidText.Show(True)
  1107. self.fidMulti.Show(False)
  1108. self.fidText.SetLabel("%d" % self.fid)
  1109. else:
  1110. self.fidText.Show(False)
  1111. self.fidMulti.Show(True)
  1112. choices = []
  1113. for fid in self.cats.keys():
  1114. choices.append(str(fid))
  1115. self.fidMulti.SetItems(choices)
  1116. self.fidMulti.SetSelection(0)
  1117. self.Layout()
  1118. return True
  1119. class CategoryListCtrl(wx.ListCtrl,
  1120. listmix.ListCtrlAutoWidthMixin,
  1121. listmix.TextEditMixin):
  1122. def __init__(self, parent, id, pos = wx.DefaultPosition,
  1123. size = wx.DefaultSize, style = 0):
  1124. """!List of layers/categories"""
  1125. self.parent = parent
  1126. wx.ListCtrl.__init__(self, parent, id, pos, size, style)
  1127. listmix.ListCtrlAutoWidthMixin.__init__(self)
  1128. listmix.TextEditMixin.__init__(self)
  1129. def Populate(self, cats, update = False):
  1130. """!Populate the list"""
  1131. itemData = {} # requested by sorter
  1132. if not update:
  1133. self.InsertColumn(0, _("Layer"))
  1134. self.InsertColumn(1, _("Category"))
  1135. else:
  1136. self.DeleteAllItems()
  1137. i = 1
  1138. for layer in cats.keys():
  1139. catsList = cats[layer]
  1140. for cat in catsList:
  1141. index = self.InsertStringItem(sys.maxint, str(catsList[0]))
  1142. self.SetStringItem(index, 0, str(layer))
  1143. self.SetStringItem(index, 1, str(cat))
  1144. self.SetItemData(index, i)
  1145. itemData[i] = (str(layer), str(cat))
  1146. i = i + 1
  1147. if not update:
  1148. self.SetColumnWidth(0, 100)
  1149. self.SetColumnWidth(1, wx.LIST_AUTOSIZE)
  1150. self.currentItem = 0
  1151. return itemData
  1152. class VDigitZBulkDialog(wx.Dialog):
  1153. def __init__(self, parent, title, nselected, style = wx.DEFAULT_DIALOG_STYLE):
  1154. """!Dialog used for Z bulk-labeling tool
  1155. """
  1156. wx.Dialog.__init__(self, parent = parent, id = wx.ID_ANY, title = title, style = style)
  1157. self.parent = parent # mapdisplay.BufferedWindow class instance
  1158. # panel = wx.Panel(parent=self, id=wx.ID_ANY)
  1159. border = wx.BoxSizer(wx.VERTICAL)
  1160. txt = wx.StaticText(parent = self,
  1161. label = _("%d lines selected for z bulk-labeling") % nselected);
  1162. border.Add(item = txt, proportion = 0, flag = wx.ALL | wx.EXPAND, border = 5)
  1163. box = wx.StaticBox (parent = self, id = wx.ID_ANY, label = " %s " % _("Set value"))
  1164. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  1165. flexSizer = wx.FlexGridSizer (cols = 2, hgap = 5, vgap = 5)
  1166. flexSizer.AddGrowableCol(0)
  1167. # starting value
  1168. txt = wx.StaticText(parent = self,
  1169. label = _("Starting value"));
  1170. self.value = wx.SpinCtrl(parent = self, id = wx.ID_ANY, size = (150, -1),
  1171. initial = 0,
  1172. min = -1e6, max = 1e6)
  1173. flexSizer.Add(txt, proportion = 0, flag = wx.ALIGN_CENTER_VERTICAL)
  1174. flexSizer.Add(self.value, proportion = 0, flag = wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  1175. # step
  1176. txt = wx.StaticText(parent = self,
  1177. label = _("Step"))
  1178. self.step = wx.SpinCtrl(parent = self, id = wx.ID_ANY, size = (150, -1),
  1179. initial = 0,
  1180. min = 0, max = 1e6)
  1181. flexSizer.Add(txt, proportion = 0, flag = wx.ALIGN_CENTER_VERTICAL)
  1182. flexSizer.Add(self.step, proportion = 0, flag = wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  1183. sizer.Add(item = flexSizer, proportion = 1, flag = wx.ALL | wx.EXPAND, border = 1)
  1184. border.Add(item = sizer, proportion = 1, flag = wx.ALL | wx.EXPAND, border = 0)
  1185. # buttons
  1186. btnCancel = wx.Button(self, wx.ID_CANCEL)
  1187. btnOk = wx.Button(self, wx.ID_OK)
  1188. btnOk.SetDefault()
  1189. # sizers
  1190. btnSizer = wx.StdDialogButtonSizer()
  1191. btnSizer.AddButton(btnCancel)
  1192. btnSizer.AddButton(btnOk)
  1193. btnSizer.Realize()
  1194. mainSizer = wx.BoxSizer(wx.VERTICAL)
  1195. mainSizer.Add(item = border, proportion = 1, flag = wx.EXPAND | wx.ALL, border = 5)
  1196. mainSizer.Add(item = btnSizer, proportion = 0,
  1197. flag = wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border = 5)
  1198. self.SetSizer(mainSizer)
  1199. mainSizer.Fit(self)
  1200. class VDigitDuplicatesDialog(wx.Dialog):
  1201. def __init__(self, parent, data, title = _("List of duplicates"),
  1202. style = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER,
  1203. pos = wx.DefaultPosition):
  1204. """!Show duplicated feature ids
  1205. """
  1206. wx.Dialog.__init__(self, parent = parent, id = wx.ID_ANY, title = title, style = style,
  1207. pos = pos)
  1208. self.parent = parent # BufferedWindow
  1209. self.data = data
  1210. self.winList = []
  1211. # panel = wx.Panel(parent=self, id=wx.ID_ANY)
  1212. # notebook
  1213. self.notebook = wx.Notebook(parent = self, id = wx.ID_ANY, style = wx.BK_DEFAULT)
  1214. id = 1
  1215. for key in self.data.keys():
  1216. panel = wx.Panel(parent = self.notebook, id = wx.ID_ANY)
  1217. self.notebook.AddPage(page = panel, text = " %d " % (id))
  1218. # notebook body
  1219. border = wx.BoxSizer(wx.VERTICAL)
  1220. win = CheckListFeature(parent = panel, data = list(self.data[key]))
  1221. self.winList.append(win.GetId())
  1222. border.Add(item = win, proportion = 1,
  1223. flag = wx.ALL | wx.EXPAND, border = 5)
  1224. panel.SetSizer(border)
  1225. id += 1
  1226. # buttons
  1227. btnCancel = wx.Button(self, wx.ID_CANCEL)
  1228. btnOk = wx.Button(self, wx.ID_OK)
  1229. btnOk.SetDefault()
  1230. # sizers
  1231. btnSizer = wx.StdDialogButtonSizer()
  1232. btnSizer.AddButton(btnCancel)
  1233. btnSizer.AddButton(btnOk)
  1234. btnSizer.Realize()
  1235. mainSizer = wx.BoxSizer(wx.VERTICAL)
  1236. mainSizer.Add(item = self.notebook, proportion = 1, flag = wx.EXPAND | wx.ALL, border = 5)
  1237. mainSizer.Add(item = btnSizer, proportion = 0,
  1238. flag = wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border = 5)
  1239. self.SetSizer(mainSizer)
  1240. mainSizer.Fit(self)
  1241. self.SetAutoLayout(True)
  1242. # set min size for dialog
  1243. self.SetMinSize((250, 180))
  1244. def GetUnSelected(self):
  1245. """!Get unselected items (feature id)
  1246. @return list of ids
  1247. """
  1248. ids = []
  1249. for id in self.winList:
  1250. wlist = self.FindWindowById(id)
  1251. for item in range(wlist.GetItemCount()):
  1252. if not wlist.IsChecked(item):
  1253. ids.append(int(wlist.GetItem(item, 0).GetText()))
  1254. return ids
  1255. class CheckListFeature(wx.ListCtrl, listmix.ListCtrlAutoWidthMixin, listmix.CheckListCtrlMixin):
  1256. def __init__(self, parent, data,
  1257. pos = wx.DefaultPosition, log = None):
  1258. """!List of mapset/owner/group"""
  1259. self.parent = parent
  1260. self.data = data
  1261. wx.ListCtrl.__init__(self, parent, wx.ID_ANY,
  1262. style = wx.LC_REPORT)
  1263. listmix.CheckListCtrlMixin.__init__(self)
  1264. self.log = log
  1265. # setup mixins
  1266. listmix.ListCtrlAutoWidthMixin.__init__(self)
  1267. self.LoadData(self.data)
  1268. def LoadData(self, data):
  1269. """!Load data into list"""
  1270. self.InsertColumn(0, _('Feature id'))
  1271. self.InsertColumn(1, _('Layer (Categories)'))
  1272. for item in data:
  1273. index = self.InsertStringItem(sys.maxint, str(item[0]))
  1274. self.SetStringItem(index, 1, str(item[1]))
  1275. # enable all items by default
  1276. for item in range(self.GetItemCount()):
  1277. self.CheckItem(item, True)
  1278. self.SetColumnWidth(col = 0, width = wx.LIST_AUTOSIZE_USEHEADER)
  1279. self.SetColumnWidth(col = 1, width = wx.LIST_AUTOSIZE_USEHEADER)
  1280. def OnCheckItem(self, index, flag):
  1281. """!Mapset checked/unchecked"""
  1282. pass