vdigit.py 96 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538
  1. """!
  2. @package vdigit
  3. @brief Vector digitizer extension
  4. Import:
  5. @code
  6. from vdigit import VDigit as VDigit
  7. @endcode
  8. Classes:
  9. - AbstractDigit
  10. - VDigit
  11. - AbstractDisplayDriver
  12. - CDisplayDriver
  13. - VDigitSettingsDialog
  14. - VDigitCategoryDialog
  15. - VDigitZBulkDialog
  16. - VDigitDuplicatesDialog
  17. - VDigitVBuildDialog
  18. (C) 2007-2009 by the GRASS Development Team
  19. This program is free software under the GNU General Public
  20. License (>=v2). Read the file COPYING that comes with GRASS
  21. for details.
  22. @author Martin Landa <landa.martin gmail.com>
  23. """
  24. import os
  25. import sys
  26. import string
  27. import copy
  28. from threading import Thread
  29. import wx
  30. import wx.lib.colourselect as csel
  31. import wx.lib.mixins.listctrl as listmix
  32. import gcmd
  33. import dbm
  34. from debug import Debug as Debug
  35. import gselect
  36. import globalvar
  37. from preferences import globalSettings as UserSettings
  38. try:
  39. digitPath = os.path.join(globalvar.ETCWXDIR, "vdigit")
  40. sys.path.append(digitPath)
  41. import grass7_wxvdigit as wxvdigit
  42. GV_LINES = wxvdigit.GV_LINES
  43. PseudoDC = wxvdigit.PseudoDC
  44. haveVDigit = True
  45. errorMsg = ''
  46. except ImportError, err:
  47. haveVDigit = False
  48. GV_LINES = None
  49. PseudoDC = wx.PseudoDC
  50. errorMsg = err
  51. print >> sys.stderr, "\nWARNING: Vector digitizer is not available (%s). " % err
  52. class AbstractDigit:
  53. """
  54. Abstract digitization class
  55. """
  56. def __init__(self, mapwindow):
  57. """!Initialization
  58. @param mapwindow reference to mapwindow (MapFrame) instance
  59. @param settings initial settings of digitization tool
  60. """
  61. self.map = None
  62. self.mapWindow = mapwindow
  63. Debug.msg (3, "AbstractDigit.__init__(): map=%s" % \
  64. self.map)
  65. #self.SetCategory()
  66. self.driver = CDisplayDriver(self, mapwindow)
  67. def __del__(self):
  68. pass
  69. def SetCategoryNextToUse(self):
  70. """!Find maximum category number in the map layer
  71. and update Digit.settings['category']
  72. @return 'True' on success, 'False' on failure
  73. """
  74. # vector map layer without categories, reset to '1'
  75. UserSettings.Set(group='vdigit', key='category', subkey='value', value=1)
  76. if self.map:
  77. cat = self.digit.GetCategory(UserSettings.Get(group='vdigit', key='layer', subkey='value'))
  78. cat += 1
  79. UserSettings.Set(group='vdigit', key='category', subkey='value',
  80. value=cat)
  81. def SetCategory(self):
  82. """!Return category number to use (according Settings)"""
  83. if UserSettings.Get(group='vdigit', key="categoryMode", subkey='selection') == 0:
  84. self.SetCategoryNextToUse()
  85. return UserSettings.Get(group='vdigit', key="category", subkey='value')
  86. def SetMapName(self, map):
  87. """!Set map name
  88. @param map map name to be set up or None (will close currently edited map)
  89. """
  90. Debug.msg (3, "AbstractDigit.SetMapName map=%s" % map)
  91. self.map = map
  92. try:
  93. ret = self.driver.Reset(self.map)
  94. except StandardError, e:
  95. raise gcmd.DigitError(parent=self.mapWindow.parent,
  96. message="%s %s (%s)" % (_('Unable to initialize display driver, '
  97. 'see README file for more information.\n\n'
  98. 'Details:'), e, errorMsg))
  99. if map and ret == -1:
  100. raise gcmd.DigitError(parent=self.mapWindow.parent,
  101. message=_('Unable to open vector map <%s> for editing.\n\n'
  102. 'Data are probably corrupted, '
  103. 'try to run v.build to rebuild '
  104. 'the topology (Vector->Develop vector map->'
  105. 'Create/rebuild topology).') % map)
  106. if not map and ret != 0:
  107. raise gcmd.DigitError(parent=self.mapWindow.parent,
  108. message=_('Unable to open vector map <%s> for editing.\n\n'
  109. 'Data are probably corrupted, '
  110. 'try to run v.build to rebuild '
  111. 'the topology (Vector->Develop vector map->'
  112. 'Create/rebuild topology).') % map)
  113. self.digit.InitCats()
  114. def SelectLinesByQueryThresh(self):
  115. """!Generic method used for SelectLinesByQuery()
  116. -- to get threshold value"""
  117. thresh = 0.0
  118. if UserSettings.Get(group='vdigit', key='query', subkey='selection') == 0:
  119. thresh = UserSettings.Get(group='vdigit', key='queryLength', subkey='thresh')
  120. if UserSettings.Get(group='vdigit', key="queryLength", subkey='than-selection') == 0:
  121. thresh = -1 * thresh
  122. else:
  123. thresh = UserSettings.Get(group='vdigit', key='queryDangle', subkey='thresh')
  124. if UserSettings.Get(group='vdigit', key="queryDangle", subkey='than-selection') == 0:
  125. thresh = -1 * thresh
  126. return thresh
  127. def GetSelectType(self):
  128. """!Get type(s) to be selected
  129. Used by SelectLinesByBox() and SelectLinesByPoint()"""
  130. type = 0
  131. for feature in (('point', wxvdigit.GV_POINT),
  132. ('line', wxvdigit.GV_LINE),
  133. ('centroid', wxvdigit.GV_CENTROID),
  134. ('boundary', wxvdigit.GV_BOUNDARY)):
  135. if UserSettings.Get(group='vdigit', key='selectType',
  136. subkey=[feature[0], 'enabled']) is True:
  137. type |= feature[1]
  138. return type
  139. def SelectLinesFromBackgroundMap(self, pos1, pos2):
  140. """!Select features from background map
  141. @param pos1,pos2 bounding box
  142. """
  143. bgmap = str(UserSettings.Get(group='vdigit', key='bgmap', subkey='value',
  144. internal=True))
  145. if bgmap == '':
  146. Debug.msg(4, "VEdit.SelectLinesFromBackgroundMap(): []")
  147. return []
  148. x1, y1 = pos1
  149. x2, y2 = pos2
  150. ret = gcmd.RunCommand('v.edit',
  151. parent = self,
  152. quiet = True,
  153. read = True,
  154. map = bgmap,
  155. tool = 'select',
  156. bbox = '%f,%f,%f,%f' % (x1, y1, x2, y2))
  157. if not ret:
  158. x, y = pos1
  159. ret = gcmd.RunCommand('v.edit',
  160. parent = self,
  161. quiet = True,
  162. read = True,
  163. map = bgmap,
  164. tool = 'select',
  165. coords = '%f,%f' % (x, y),
  166. thresh = self.driver.GetThreshold(type='selectThresh'))
  167. if not ret:
  168. return []
  169. output = ret.splitlines()[0] # first line
  170. ids = output.split(',')
  171. ids = map(int, ids) # str -> int
  172. Debug.msg(4, "VEdit.SelectLinesFromBackgroundMap(): %s" % \
  173. ",".join(["%d" % v for v in ids]))
  174. return ids
  175. class VDigit(AbstractDigit):
  176. """
  177. Prototype of digitization class based on v.digit reimplementation
  178. Under development (wxWidgets C/C++ background)
  179. """
  180. def __init__(self, mapwindow):
  181. """!Initialization
  182. @param mapwindow reference to mapwindow (MapFrame) instance
  183. @param settings initial settings of digitization tool
  184. """
  185. AbstractDigit.__init__(self, mapwindow)
  186. try:
  187. self.digit = wxvdigit.Digit(self.driver.GetDevice(),
  188. mapwindow)
  189. except (ImportError, NameError):
  190. self.digit = None
  191. self.toolbar = mapwindow.parent.toolbars['vdigit']
  192. self.UpdateSettings()
  193. def __del__(self):
  194. del self.digit
  195. def AddPoint (self, map, point, x, y, z=None):
  196. """!Add new point/centroid
  197. @param map map name (unused, for compatability with VEdit)
  198. @param point feature type (if true point otherwise centroid)
  199. @param x,y,z coordinates
  200. """
  201. if UserSettings.Get(group='vdigit', key="categoryMode", subkey='selection') == 2:
  202. layer = -1 # -> no category
  203. cat = -1
  204. else:
  205. layer = UserSettings.Get(group='vdigit', key="layer", subkey='value')
  206. cat = self.SetCategory()
  207. if point:
  208. type = wxvdigit.GV_POINT
  209. else:
  210. type = wxvdigit.GV_CENTROID
  211. snap, thresh = self.__getSnapThreshold()
  212. bgmap = str(UserSettings.Get(group='vdigit', key="bgmap",
  213. subkey='value', internal=True))
  214. if z:
  215. ret = self.digit.AddLine(type, [x, y, z], layer, cat,
  216. bgmap, snap, thresh)
  217. else:
  218. ret = self.digit.AddLine(type, [x, y], layer, cat,
  219. bgmap, snap, thresh)
  220. self.toolbar.EnableUndo()
  221. return ret
  222. def AddLine (self, map, line, coords):
  223. """!Add line/boundary
  224. @param map map name (unused, for compatability with VEdit)
  225. @param line feature type (if True line, otherwise boundary)
  226. @param coords list of coordinates
  227. """
  228. if len(coords) < 2:
  229. return
  230. if UserSettings.Get(group='vdigit', key="categoryMode", subkey='selection') == 2:
  231. layer = -1 # -> no category
  232. cat = -1
  233. else:
  234. layer = UserSettings.Get(group='vdigit', key="layer", subkey='value')
  235. cat = self.SetCategory()
  236. if line:
  237. type = wxvdigit.GV_LINE
  238. else:
  239. type = wxvdigit.GV_BOUNDARY
  240. listCoords = []
  241. for c in coords:
  242. for x in c:
  243. listCoords.append(x)
  244. snap, thresh = self.__getSnapThreshold()
  245. bgmap = str(UserSettings.Get(group='vdigit', key="bgmap",
  246. subkey='value', internal=True))
  247. ret = self.digit.AddLine(type, listCoords, layer, cat,
  248. bgmap, snap, thresh)
  249. self.toolbar.EnableUndo()
  250. return ret
  251. def DeleteSelectedLines(self):
  252. """!Delete selected features
  253. @return number of deleted lines
  254. """
  255. nlines = self.digit.DeleteLines(UserSettings.Get(group='vdigit', key='delRecord', subkey='enabled'))
  256. if nlines > 0:
  257. self.toolbar.EnableUndo()
  258. return nlines
  259. def MoveSelectedLines(self, move):
  260. """!Move selected features
  261. @param move direction (x, y)
  262. """
  263. snap, thresh = self.__getSnapThreshold()
  264. bgmap = str(UserSettings.Get(group='vdigit', key="bgmap",
  265. subkey='value', internal=True))
  266. try:
  267. nlines = self.digit.MoveLines(move[0], move[1], 0.0, # TODO 3D
  268. bgmap, snap, thresh)
  269. except SystemExit:
  270. pass
  271. if nlines > 0:
  272. self.toolbar.EnableUndo()
  273. return nlines
  274. def MoveSelectedVertex(self, coords, move):
  275. """!Move selected vertex of the line
  276. @param coords click coordinates
  277. @param move X,Y direction
  278. @return 1 vertex moved
  279. @return 0 vertex not moved (not found, line is not selected)
  280. """
  281. snap, thresh = self.__getSnapThreshold()
  282. bgmap = str(UserSettings.Get(group='vdigit', key="bgmap",
  283. subkey='value', internal=True))
  284. moved = self.digit.MoveVertex(coords[0], coords[1], 0.0, # TODO 3D
  285. move[0], move[1], 0.0,
  286. bgmap, snap,
  287. self.driver.GetThreshold(type='selectThresh'), thresh)
  288. if moved:
  289. self.toolbar.EnableUndo()
  290. return moved
  291. def AddVertex(self, coords):
  292. """!Add new vertex to the selected line/boundary on position 'coords'
  293. @param coords coordinates to add vertex
  294. @return 1 vertex added
  295. @return 0 nothing changed
  296. @return -1 on failure
  297. """
  298. added = self.digit.ModifyLineVertex(1, coords[0], coords[1], 0.0, # TODO 3D
  299. self.driver.GetThreshold(type='selectThresh'))
  300. if added > 0:
  301. self.toolbar.EnableUndo()
  302. return added
  303. def RemoveVertex(self, coords):
  304. """!Remove vertex from the selected line/boundary on position 'coords'
  305. @param coords coordinates to remove vertex
  306. @return 1 vertex removed
  307. @return 0 nothing changed
  308. @return -1 on failure
  309. """
  310. deleted = self.digit.ModifyLineVertex(0, coords[0], coords[1], 0.0, # TODO 3D
  311. self.driver.GetThreshold(type='selectThresh'))
  312. if deleted > 0:
  313. self.toolbar.EnableUndo()
  314. return deleted
  315. def SplitLine(self, coords):
  316. """!Split selected line/boundary on position 'coords'
  317. @param coords coordinates to split line
  318. @return 1 line modified
  319. @return 0 nothing changed
  320. @return -1 error
  321. """
  322. ret = self.digit.SplitLine(coords[0], coords[1], 0.0, # TODO 3D
  323. self.driver.GetThreshold('selectThresh'))
  324. if ret > 0:
  325. self.toolbar.EnableUndo()
  326. return ret
  327. def EditLine(self, line, coords):
  328. """!Edit existing line/boundary
  329. @param line id of line to be modified
  330. @param coords list of coordinates of modified line
  331. @return feature id of new line
  332. @return -1 on error
  333. """
  334. try:
  335. lineid = line[0]
  336. except:
  337. lineid = -1
  338. if len(coords) < 2:
  339. self.DeleteSelectedLines()
  340. return 0
  341. listCoords = []
  342. for c in coords:
  343. for x in c:
  344. listCoords.append(x)
  345. snap, thresh = self.__getSnapThreshold()
  346. bgmap = str(UserSettings.Get(group='vdigit', key="bgmap",
  347. subkey='value', internal=True))
  348. try:
  349. ret = self.digit.RewriteLine(lineid, listCoords,
  350. bgmap, snap, thresh)
  351. except SystemExit:
  352. pass
  353. if ret > 0:
  354. self.toolbar.EnableUndo()
  355. return ret
  356. def FlipLine(self):
  357. """!Flip selected lines/boundaries
  358. @return number of modified lines
  359. @return -1 on error
  360. """
  361. ret = self.digit.FlipLines()
  362. if ret > 0:
  363. self.toolbar.EnableUndo()
  364. return ret
  365. def MergeLine(self):
  366. """!Merge selected lines/boundaries
  367. @return number of modified lines
  368. @return -1 on error
  369. """
  370. ret = self.digit.MergeLines()
  371. if ret > 0:
  372. self.toolbar.EnableUndo()
  373. return ret
  374. def BreakLine(self):
  375. """!Break selected lines/boundaries
  376. @return number of modified lines
  377. @return -1 on error
  378. """
  379. ret = self.digit.BreakLines()
  380. if ret > 0:
  381. self.toolbar.EnableUndo()
  382. return ret
  383. def SnapLine(self):
  384. """!Snap selected lines/boundaries
  385. @return on success
  386. @return -1 on error
  387. """
  388. snap, thresh = self.__getSnapThreshold()
  389. ret = self.digit.SnapLines(thresh)
  390. if ret == 0:
  391. self.toolbar.EnableUndo()
  392. return ret
  393. def ConnectLine(self):
  394. """!Connect selected lines/boundaries
  395. @return 1 lines connected
  396. @return 0 lines not connected
  397. @return -1 on error
  398. """
  399. snap, thresh = self.__getSnapThreshold()
  400. ret = self.digit.ConnectLines(thresh)
  401. if ret > 0:
  402. self.toolbar.EnableUndo()
  403. return ret
  404. def CopyLine(self, ids=[]):
  405. """!Copy features from (background) vector map
  406. @param ids list of line ids to be copied
  407. @return number of copied features
  408. @return -1 on error
  409. """
  410. bgmap = str(UserSettings.Get(group='vdigit', key='bgmap',
  411. subkey='value', internal=True))
  412. if len(bgmap) > 0:
  413. ret = self.digit.CopyLines(ids, bgmap)
  414. else:
  415. ret = self.digit.CopyLines(ids, None)
  416. if ret > 0:
  417. self.toolbar.EnableUndo()
  418. return ret
  419. def CopyCats(self, fromId, toId, copyAttrb=False):
  420. """!Copy given categories to objects with id listed in ids
  421. @param cats ids of 'from' feature
  422. @param ids ids of 'to' feature(s)
  423. @return number of modified features
  424. @return -1 on error
  425. """
  426. if len(fromId) == 0 or len(toId) == 0:
  427. return 0
  428. ret = self.digit.CopyCats(fromId, toId, copyAttrb)
  429. if ret > 0:
  430. self.toolbar.EnableUndo()
  431. return ret
  432. def SelectLinesByQuery(self, pos1, pos2):
  433. """!Select features by query
  434. @param pos1, pos2 bounding box definition
  435. """
  436. thresh = self.SelectLinesByQueryThresh()
  437. w, n = pos1
  438. e, s = pos2
  439. query = wxvdigit.QUERY_UNKNOWN
  440. if UserSettings.Get(group='vdigit', key='query', subkey='selection') == 0:
  441. query = wxvdigit.QUERY_LENGTH
  442. else:
  443. query = wxvdigit.QUERY_DANGLE
  444. type = wxvdigit.GV_POINTS | wxvdigit.GV_LINES # TODO: 3D
  445. ids = self.digit.SelectLinesByQuery(w, n, 0.0, e, s, 1000.0,
  446. UserSettings.Get(group='vdigit', key='query', subkey='box'),
  447. query, type, thresh)
  448. Debug.msg(4, "VDigit.SelectLinesByQuery(): %s" % \
  449. ",".join(["%d" % v for v in ids]))
  450. return ids
  451. def GetLineCats(self, line=-1):
  452. """!Get layer/category pairs from given (selected) line
  453. @param line feature id (-1 for first selected line)
  454. """
  455. return dict(self.digit.GetLineCats(line))
  456. def GetLineLength(self, line):
  457. """!Get line length
  458. @param line feature id
  459. @return line length
  460. @return 0 for non-linear features
  461. """
  462. return self.digit.GetLineLength(line)
  463. def SetLineCats(self, line, layer, cats, add=True):
  464. """!Set categories for given line and layer
  465. @param line feature id
  466. @param layer layer number (-1 for first selected line)
  467. @param cats list of categories
  468. @param add if True to add, otherwise do delete categories
  469. @return new feature id (feature need to be rewritten)
  470. @return -1 on error
  471. """
  472. ret = self.digit.SetLineCats(line, layer, cats, add)
  473. if ret > 0:
  474. self.toolbar.EnableUndo()
  475. return ret
  476. def GetLayers(self):
  477. """!Get list of layers"""
  478. return self.digit.GetLayers()
  479. def TypeConvForSelectedLines(self):
  480. """!Feature type conversion for selected objects.
  481. Supported conversions:
  482. - point <-> centroid
  483. - line <-> boundary
  484. @return number of modified features
  485. @return -1 on error
  486. """
  487. ret = self.digit.TypeConvLines()
  488. if ret > 0:
  489. self.toolbar.EnableUndo()
  490. return ret
  491. def Undo(self, level=-1):
  492. """!Undo action
  493. @param level levels to undo (0 to revert all)
  494. @return id of current changeset
  495. """
  496. try:
  497. ret = self.digit.Undo(level)
  498. except SystemExit:
  499. ret = -2
  500. if ret == -2:
  501. raise gcmd.DigitError, _("Undo failed, data corrupted.")
  502. self.mapWindow.UpdateMap(render=False)
  503. if ret < 0: # disable undo tool
  504. self.toolbar.EnableUndo(False)
  505. def GetUndoLevel(self):
  506. """!Get undo level (number of active changesets)"""
  507. return self.digit.GetUndoLevel()
  508. def UpdateSettings(self):
  509. """!Update digit settigs"""
  510. if not self.digit:
  511. return
  512. self.digit.UpdateSettings(UserSettings.Get(group='vdigit', key='breakLines',
  513. subkey='enabled'),
  514. UserSettings.Get(group='vdigit', key='addCentroid',
  515. subkey='enabled'),
  516. UserSettings.Get(group='vdigit', key='catBoundary',
  517. subkey='enabled'))
  518. def __getSnapThreshold(self):
  519. """!Get snap mode and threshold value
  520. @return (snap, thresh)
  521. """
  522. thresh = self.driver.GetThreshold()
  523. if thresh > 0.0:
  524. if UserSettings.Get(group='vdigit', key='snapToVertex', subkey='enabled') is True:
  525. snap = wxvdigit.SNAPVERTEX
  526. else:
  527. snap = wxvdigit.SNAP
  528. else:
  529. snap = wxvdigit.NO_SNAP
  530. return (snap, thresh)
  531. class Digit(VDigit):
  532. """!Default digit class"""
  533. def __init__(self, mapwindow):
  534. VDigit.__init__(self, mapwindow)
  535. self.type = 'vdigit'
  536. def __del__(self):
  537. VDigit.__del__(self)
  538. class AbstractDisplayDriver:
  539. """!Abstract classs for display driver"""
  540. def __init__(self, parent, mapwindow):
  541. """!Initialization
  542. @param parent
  543. @param mapwindow reference to mapwindow (MFrame)
  544. """
  545. self.parent = parent
  546. self.mapwindow = mapwindow
  547. self.ids = {} # dict[g6id] = [pdcId]
  548. self.selected = [] # list of selected objects (grassId!)
  549. def GetThreshold(self, type='snapping', value=None, units=None):
  550. """!Return threshold in map units
  551. @param value threshold to be set up
  552. @param units units (map, screen)
  553. """
  554. if value is None:
  555. value = UserSettings.Get(group='vdigit', key=type, subkey='value')
  556. if units is None:
  557. units = UserSettings.Get(group='vdigit', key=type, subkey='units')
  558. reg = self.mapwindow.Map.region
  559. if value < 0:
  560. value = (reg['nsres'] + reg['ewres']) / 2.
  561. if units == "screen pixels":
  562. # pixel -> cell
  563. if reg['nsres'] > reg['ewres']:
  564. res = reg['nsres']
  565. else:
  566. res = reg['ewres']
  567. threshold = value * res
  568. else:
  569. threshold = value
  570. Debug.msg(4, "AbstractDisplayDriver.GetThreshold(): type=%s, thresh=%f" % (type, threshold))
  571. return threshold
  572. class CDisplayDriver(AbstractDisplayDriver):
  573. """
  574. Display driver using grass7_wxdriver module
  575. """
  576. def __init__(self, parent, mapwindow):
  577. """!Initialization
  578. @param parent
  579. @param mapwindow reference to mapwindow (MFrame)
  580. """
  581. AbstractDisplayDriver.__init__(self, parent, mapwindow)
  582. self.mapWindow = mapwindow
  583. if not self.mapwindow.parent.IsStandalone():
  584. logerr = self.mapwindow.parent.GetLayerManager().goutput.cmd_stderr
  585. else:
  586. logerr = None
  587. # initialize wx display driver
  588. try:
  589. self.__display = wxvdigit.DisplayDriver(mapwindow.pdcVector,
  590. mapwindow.pdcTmp,
  591. logerr)
  592. except:
  593. self.__display = None
  594. self.UpdateSettings()
  595. def GetDevice(self):
  596. """!Get device"""
  597. return self.__display
  598. def SetDevice(self, pdc):
  599. """!Set device for driver
  600. @param pdc wx.PseudoDC instance
  601. """
  602. self.__display.SetDevice(pdc)
  603. def Reset(self, map):
  604. """!Reset map
  605. Open or close the vector map by driver.
  606. @param map map name or None to close the map
  607. @return 0 on success (close map)
  608. @return topo level on success (open map)
  609. @return non-zero (close map)
  610. @return -1 on error (open map)
  611. """
  612. if map:
  613. name, mapset = map.split('@')
  614. try:
  615. ret = self.__display.OpenMap(str(name), str(mapset), True)
  616. except SystemExit:
  617. ret = -1
  618. else:
  619. ret = self.__display.CloseMap()
  620. return ret
  621. def ReloadMap(self):
  622. """!Reload map (close and re-open).
  623. Needed for v.edit, TODO: get rid of that..."""
  624. Debug.msg(4, "CDisplayDriver.ReloadMap():")
  625. self.__display.ReloadMap()
  626. def DrawMap(self):
  627. """!Draw vector map layer content
  628. @return wx.Image instance
  629. """
  630. nlines = self.__display.DrawMap(True) # force
  631. Debug.msg(3, "CDisplayDriver.DrawMap(): nlines=%d" % nlines)
  632. return nlines
  633. def SelectLinesByBox(self, begin, end, type=0, drawSeg=False):
  634. """!Select vector features by given bounding box.
  635. If type is given, only vector features of given type are selected.
  636. @param begin,end bounding box definition
  637. @param type select only objects of given type
  638. """
  639. x1, y1 = begin
  640. x2, y2 = end
  641. inBox = UserSettings.Get(group='vdigit', key='selectInside', subkey='enabled')
  642. nselected = self.__display.SelectLinesByBox(x1, y1, -1.0 * wxvdigit.PORT_DOUBLE_MAX,
  643. x2, y2, wxvdigit.PORT_DOUBLE_MAX,
  644. type, inBox, drawSeg)
  645. Debug.msg(4, "CDisplayDriver.SelectLinesByBox(): selected=%d" % \
  646. nselected)
  647. return nselected
  648. def SelectLineByPoint(self, point, type=0):
  649. """!Select vector feature by coordinates of click point (in given threshold).
  650. If type is given, only vector features of given type are selected.
  651. @param point click coordinates (bounding box given by threshold)
  652. @param type select only objects of given type
  653. """
  654. pointOnLine = self.__display.SelectLineByPoint(point[0], point[1], 0.0,
  655. self.GetThreshold(type='selectThresh'),
  656. type, 0); # without_z
  657. if len(pointOnLine) > 0:
  658. Debug.msg(4, "CDisplayDriver.SelectLineByPoint(): pointOnLine=%f,%f" % \
  659. (pointOnLine[0], pointOnLine[1]))
  660. return pointOnLine
  661. else:
  662. Debug.msg(4, "CDisplayDriver.SelectLineByPoint(): no line found")
  663. return None
  664. def GetSelected(self, grassId=True):
  665. """!Return ids of selected vector features
  666. @param grassId if grassId is True returns GRASS ids, otherwise
  667. internal ids of objects drawn in PseudoDC"""
  668. if grassId:
  669. selected = self.__display.GetSelected(True)
  670. else:
  671. selected = self.__display.GetSelected(False)
  672. Debug.msg(4, "CDisplayDriver.GetSelected(): grassId=%d, ids=%s" % \
  673. (grassId, (",".join(["%d" % v for v in selected]))))
  674. return selected
  675. def GetSelectedCoord(self):
  676. """!Return ids of selected vector features and their coordinates"""
  677. return dict(self.__display.GetSelectedCoord())
  678. def GetRegionSelected(self):
  679. """!Get minimal region extent of selected features (ids/cats)"""
  680. return self.__display.GetRegionSelected()
  681. def GetDuplicates(self):
  682. """!Return ids of (selected) duplicated vector features
  683. """
  684. # -> id : (list of ids)
  685. dupl = dict(self.__display.GetDuplicates())
  686. # -> id : ((id, cat), ...)
  687. dupl_full = {}
  688. for key in dupl.keys():
  689. dupl_full[key] = []
  690. for id in dupl[key]:
  691. catStr = ''
  692. cats = self.parent.GetLineCats(line=id)
  693. for layer in cats.keys():
  694. if len(cats[layer]) > 0:
  695. catStr = "%d: (" % layer
  696. for cat in cats[layer]:
  697. catStr += "%d," % cat
  698. catStr = catStr.rstrip(',')
  699. catStr += ')'
  700. dupl_full[key].append([id, catStr])
  701. return dupl_full
  702. def GetSelectedVertex(self, coords):
  703. """!Get PseudoDC id(s) of vertex (of selected line)
  704. on position 'coords'
  705. @param coords click position
  706. """
  707. x, y = coords
  708. id = self.__display.GetSelectedVertex(x, y, self.GetThreshold(type='selectThresh'))
  709. Debug.msg(4, "CDisplayDriver.GetSelectedVertex(): id=%s" % \
  710. (",".join(["%d" % v for v in id])))
  711. return id
  712. def SetSelected(self, id, field=-1):
  713. """!Set selected vector features
  714. @param id list of feature ids/categories to be selected
  715. @param field field(layer) number, -1 for ids instead of cats
  716. """
  717. Debug.msg(4, "CDisplayDriver.SetSelected(): id=%s" % \
  718. id)
  719. self.__display.SetSelected(id, field)
  720. def UnSelect(self, id):
  721. """!Unselect vector features
  722. @param id list of feature id(s)
  723. """
  724. Debug.msg(4, "CDisplayDriver.UnSelect(): id=%s" % \
  725. ",".join(["%d" % v for v in id]))
  726. self.__display.UnSelect(id)
  727. def UpdateRegion(self):
  728. """!Set geographical region
  729. Needed for 'cell2pixel' conversion"""
  730. map = self.mapwindow.Map
  731. reg = map.region
  732. self.__display.SetRegion(reg['n'],
  733. reg['s'],
  734. reg['e'],
  735. reg['w'],
  736. reg['nsres'],
  737. reg['ewres'],
  738. reg['center_easting'],
  739. reg['center_northing'],
  740. map.width, map.height)
  741. def GetMapBoundingBox(self):
  742. """!Return bounding box of given vector map layer
  743. @return (w,s,b,e,n,t)
  744. """
  745. return self.__display.GetMapBoundingBox()
  746. def DrawSelected(self, draw=True):
  747. """!Show/hide selected features"""
  748. self.__display.DrawSelected(draw)
  749. def UpdateSettings(self, alpha=255):
  750. """!Update display driver settings"""
  751. # TODO map units
  752. if not self.__display:
  753. return
  754. color = {}
  755. for symbol in ("highlight",
  756. "highlightDupl",
  757. "point",
  758. "line",
  759. "boundaryNo",
  760. "boundaryOne",
  761. "boundaryTwo",
  762. "centroidIn",
  763. "centroidOut",
  764. "centroidDup",
  765. "nodeOne",
  766. "nodeTwo",
  767. "vertex",
  768. "area",
  769. "direction"):
  770. color[symbol] = wx.Color(UserSettings.Get(group='vdigit', key='symbol',
  771. subkey=[symbol, 'color'])[0],
  772. UserSettings.Get(group='vdigit', key='symbol',
  773. subkey=[symbol, 'color'])[1],
  774. UserSettings.Get(group='vdigit', key='symbol',
  775. subkey=[symbol, 'color'])[2]).GetRGB()
  776. self.__display.UpdateSettings (color['highlight'],
  777. UserSettings.Get(group='vdigit', key='checkForDupl',
  778. subkey='enabled'),
  779. color['highlightDupl'],
  780. UserSettings.Get(group='vdigit', key='symbol',
  781. subkey=['point', 'enabled']),
  782. color['point'],
  783. UserSettings.Get(group='vdigit', key='symbol',
  784. subkey=['line', 'enabled']),
  785. color['line'],
  786. UserSettings.Get(group='vdigit', key='symbol',
  787. subkey=['boundaryNo', 'enabled']),
  788. color['boundaryNo'],
  789. UserSettings.Get(group='vdigit', key='symbol',
  790. subkey=['boundaryOne', 'enabled']),
  791. color['boundaryOne'],
  792. UserSettings.Get(group='vdigit', key='symbol',
  793. subkey=['boundaryTwo', 'enabled']),
  794. color['boundaryTwo'],
  795. UserSettings.Get(group='vdigit', key='symbol',
  796. subkey=['centroidIn', 'enabled']),
  797. color['centroidIn'],
  798. UserSettings.Get(group='vdigit', key='symbol',
  799. subkey=['centroidOut', 'enabled']),
  800. color['centroidOut'],
  801. UserSettings.Get(group='vdigit', key='symbol',
  802. subkey=['centroidDup', 'enabled']),
  803. color['centroidDup'],
  804. UserSettings.Get(group='vdigit', key='symbol',
  805. subkey=['nodeOne', 'enabled']),
  806. color['nodeOne'],
  807. UserSettings.Get(group='vdigit', key='symbol',
  808. subkey=['nodeTwo', 'enabled']),
  809. color['nodeTwo'],
  810. UserSettings.Get(group='vdigit', key='symbol',
  811. subkey=['vertex', 'enabled']),
  812. color['vertex'],
  813. UserSettings.Get(group='vdigit', key='symbol',
  814. subkey=['area', 'enabled']),
  815. color['area'],
  816. UserSettings.Get(group='vdigit', key='symbol',
  817. subkey=['direction', 'enabled']),
  818. color['direction'],
  819. UserSettings.Get(group='vdigit', key='lineWidth',
  820. subkey='value'),
  821. alpha)
  822. class VDigitSettingsDialog(wx.Dialog):
  823. """
  824. Standard settings dialog for digitization purposes
  825. """
  826. def __init__(self, parent, title, style=wx.DEFAULT_DIALOG_STYLE):
  827. wx.Dialog.__init__(self, parent=parent, id=wx.ID_ANY, title=title, style=style)
  828. self.parent = parent # mapdisplay.BufferedWindow class instance
  829. # notebook
  830. notebook = wx.Notebook(parent=self, id=wx.ID_ANY, style=wx.BK_DEFAULT)
  831. self.__CreateSymbologyPage(notebook)
  832. parent.digit.SetCategory() # update category number (next to use)
  833. self.__CreateGeneralPage(notebook)
  834. self.__CreateAttributesPage(notebook)
  835. self.__CreateQueryPage(notebook)
  836. # buttons
  837. btnApply = wx.Button(self, wx.ID_APPLY)
  838. btnCancel = wx.Button(self, wx.ID_CANCEL)
  839. btnSave = wx.Button(self, wx.ID_SAVE)
  840. btnSave.SetDefault()
  841. # bindigs
  842. btnApply.Bind(wx.EVT_BUTTON, self.OnApply)
  843. btnApply.SetToolTipString(_("Apply changes for this session"))
  844. btnApply.SetDefault()
  845. btnSave.Bind(wx.EVT_BUTTON, self.OnSave)
  846. btnSave.SetToolTipString(_("Close dialog and save changes to user settings file"))
  847. btnCancel.Bind(wx.EVT_BUTTON, self.OnCancel)
  848. btnCancel.SetToolTipString(_("Close dialog and ignore changes"))
  849. # sizers
  850. btnSizer = wx.StdDialogButtonSizer()
  851. btnSizer.AddButton(btnCancel)
  852. btnSizer.AddButton(btnApply)
  853. btnSizer.AddButton(btnSave)
  854. btnSizer.Realize()
  855. mainSizer = wx.BoxSizer(wx.VERTICAL)
  856. mainSizer.Add(item=notebook, proportion=1, flag=wx.EXPAND | wx.ALL, border=5)
  857. mainSizer.Add(item=btnSizer, proportion=0,
  858. flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
  859. self.Bind(wx.EVT_CLOSE, self.OnCancel)
  860. self.SetSizer(mainSizer)
  861. mainSizer.Fit(self)
  862. def __CreateSymbologyPage(self, notebook):
  863. """!Create notebook page concerning with symbology settings"""
  864. panel = wx.Panel(parent=notebook, id=wx.ID_ANY)
  865. notebook.AddPage(page=panel, text=_("Symbology"))
  866. sizer = wx.BoxSizer(wx.VERTICAL)
  867. flexSizer = wx.FlexGridSizer (cols=3, hgap=5, vgap=5)
  868. flexSizer.AddGrowableCol(0)
  869. self.symbology = {}
  870. for label, key in self.__SymbologyData():
  871. textLabel = wx.StaticText(panel, wx.ID_ANY, label)
  872. color = csel.ColourSelect(panel, id=wx.ID_ANY,
  873. colour=UserSettings.Get(group='vdigit', key='symbol',
  874. subkey=[key, 'color']), size=(25, 25))
  875. isEnabled = UserSettings.Get(group='vdigit', key='symbol',
  876. subkey=[key, 'enabled'])
  877. if isEnabled is not None:
  878. enabled = wx.CheckBox(panel, id=wx.ID_ANY, label="")
  879. enabled.SetValue(isEnabled)
  880. self.symbology[key] = (enabled, color)
  881. else:
  882. enabled = (1, 1)
  883. self.symbology[key] = (None, color)
  884. flexSizer.Add(textLabel, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
  885. flexSizer.Add(enabled, proportion=0, flag=wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  886. flexSizer.Add(color, proportion=0, flag=wx.ALIGN_RIGHT | wx.FIXED_MINSIZE)
  887. color.SetName("GetColour")
  888. sizer.Add(item=flexSizer, proportion=1, flag=wx.ALL | wx.EXPAND, border=10)
  889. panel.SetSizer(sizer)
  890. return panel
  891. def __CreateGeneralPage(self, notebook):
  892. """!Create notebook page concerning with symbology settings"""
  893. panel = wx.Panel(parent=notebook, id=wx.ID_ANY)
  894. notebook.AddPage(page=panel, text=_("General"))
  895. border = wx.BoxSizer(wx.VERTICAL)
  896. #
  897. # display section
  898. #
  899. box = wx.StaticBox (parent=panel, id=wx.ID_ANY, label=" %s " % _("Display"))
  900. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  901. flexSizer = wx.FlexGridSizer (cols=3, hgap=5, vgap=5)
  902. flexSizer.AddGrowableCol(0)
  903. # line width
  904. text = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("Line width"))
  905. self.lineWidthValue = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(75, -1),
  906. initial=UserSettings.Get(group='vdigit', key="lineWidth", subkey='value'),
  907. min=1, max=1e6)
  908. units = wx.StaticText(parent=panel, id=wx.ID_ANY, size=(115, -1),
  909. label=UserSettings.Get(group='vdigit', key="lineWidth", subkey='units'),
  910. style=wx.ALIGN_LEFT)
  911. flexSizer.Add(text, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
  912. flexSizer.Add(self.lineWidthValue, proportion=0, flag=wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  913. flexSizer.Add(units, proportion=0, flag=wx.ALIGN_RIGHT | wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL | wx.LEFT,
  914. border=10)
  915. sizer.Add(item=flexSizer, proportion=1, flag=wx.ALL | wx.EXPAND, border=1)
  916. border.Add(item=sizer, proportion=0, flag=wx.ALL | wx.EXPAND, border=5)
  917. #
  918. # snapping section
  919. #
  920. box = wx.StaticBox (parent=panel, id=wx.ID_ANY, label=" %s " % _("Snapping"))
  921. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  922. flexSizer = wx.FlexGridSizer (cols=3, hgap=5, vgap=5)
  923. flexSizer.AddGrowableCol(0)
  924. # snapping
  925. text = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("Snapping threshold"))
  926. self.snappingValue = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(75, -1),
  927. initial=UserSettings.Get(group='vdigit', key="snapping", subkey='value'),
  928. min=-1, max=1e6)
  929. self.snappingValue.Bind(wx.EVT_SPINCTRL, self.OnChangeSnappingValue)
  930. self.snappingValue.Bind(wx.EVT_TEXT, self.OnChangeSnappingValue)
  931. self.snappingUnit = wx.Choice(parent=panel, id=wx.ID_ANY, size=(125, -1),
  932. choices=["screen pixels", "map units"])
  933. self.snappingUnit.SetStringSelection(UserSettings.Get(group='vdigit', key="snapping", subkey='units'))
  934. self.snappingUnit.Bind(wx.EVT_CHOICE, self.OnChangeSnappingUnits)
  935. flexSizer.Add(text, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
  936. flexSizer.Add(self.snappingValue, proportion=0, flag=wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  937. flexSizer.Add(self.snappingUnit, proportion=0, flag=wx.ALIGN_RIGHT | wx.FIXED_MINSIZE)
  938. vertexSizer = wx.BoxSizer(wx.VERTICAL)
  939. self.snapVertex = wx.CheckBox(parent=panel, id=wx.ID_ANY,
  940. label=_("Snap also to vertex"))
  941. self.snapVertex.SetValue(UserSettings.Get(group='vdigit', key="snapToVertex", subkey='enabled'))
  942. vertexSizer.Add(item=self.snapVertex, proportion=0, flag=wx.EXPAND)
  943. self.mapUnits = self.parent.MapWindow.Map.ProjInfo()['units']
  944. self.snappingInfo = wx.StaticText(parent=panel, id=wx.ID_ANY,
  945. label=_("Snapping threshold is %(value).1f %(units)s") % \
  946. {'value' : self.parent.digit.driver.GetThreshold(),
  947. 'units' : self.mapUnits})
  948. vertexSizer.Add(item=self.snappingInfo, proportion=0,
  949. flag=wx.ALL | wx.EXPAND, border=1)
  950. sizer.Add(item=flexSizer, proportion=1, flag=wx.EXPAND)
  951. sizer.Add(item=vertexSizer, proportion=1, flag=wx.EXPAND)
  952. border.Add(item=sizer, proportion=0, flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=5)
  953. #
  954. # select box
  955. #
  956. box = wx.StaticBox (parent=panel, id=wx.ID_ANY, label=" %s " % _("Select vector features"))
  957. # feature type
  958. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  959. inSizer = wx.BoxSizer(wx.HORIZONTAL)
  960. self.selectFeature = {}
  961. for feature in ('point', 'line',
  962. 'centroid', 'boundary'):
  963. chkbox = wx.CheckBox(parent=panel, label=feature)
  964. self.selectFeature[feature] = chkbox.GetId()
  965. chkbox.SetValue(UserSettings.Get(group='vdigit', key='selectType',
  966. subkey=[feature, 'enabled']))
  967. inSizer.Add(item=chkbox, proportion=0,
  968. flag=wx.EXPAND | wx.ALL, border=5)
  969. sizer.Add(item=inSizer, proportion=0, flag=wx.EXPAND)
  970. # threshold
  971. flexSizer = wx.FlexGridSizer (cols=3, hgap=5, vgap=5)
  972. flexSizer.AddGrowableCol(0)
  973. text = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("Select threshold"))
  974. self.selectThreshValue = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(75, -1),
  975. initial=UserSettings.Get(group='vdigit', key="selectThresh", subkey='value'),
  976. min=1, max=1e6)
  977. units = wx.StaticText(parent=panel, id=wx.ID_ANY, size=(115, -1),
  978. label=UserSettings.Get(group='vdigit', key="lineWidth", subkey='units'),
  979. style=wx.ALIGN_LEFT)
  980. flexSizer.Add(text, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
  981. flexSizer.Add(self.selectThreshValue, proportion=0, flag=wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  982. flexSizer.Add(units, proportion=0, flag=wx.ALIGN_RIGHT | wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL | wx.LEFT,
  983. border=10)
  984. self.selectIn = wx.CheckBox(parent=panel, id=wx.ID_ANY,
  985. label=_("Select only features inside of selection bounding box"))
  986. self.selectIn.SetValue(UserSettings.Get(group='vdigit', key="selectInside", subkey='enabled'))
  987. self.selectIn.SetToolTipString(_("By default are selected all features overlapping selection bounding box "))
  988. self.checkForDupl = wx.CheckBox(parent=panel, id=wx.ID_ANY,
  989. label=_("Check for duplicates"))
  990. self.checkForDupl.SetValue(UserSettings.Get(group='vdigit', key="checkForDupl", subkey='enabled'))
  991. sizer.Add(item=flexSizer, proportion=0, flag=wx.EXPAND)
  992. sizer.Add(item=self.selectIn, proportion=0, flag=wx.EXPAND | wx.ALL, border=1)
  993. sizer.Add(item=self.checkForDupl, proportion=0, flag=wx.EXPAND | wx.ALL, border=1)
  994. border.Add(item=sizer, proportion=0, flag=wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM, border=5)
  995. #
  996. # digitize lines box
  997. #
  998. box = wx.StaticBox (parent=panel, id=wx.ID_ANY, label=" %s " % _("Digitize line features"))
  999. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  1000. self.intersect = wx.CheckBox(parent=panel, label=_("Break lines at intersection"))
  1001. self.intersect.SetValue(UserSettings.Get(group='vdigit', key='breakLines', subkey='enabled'))
  1002. sizer.Add(item=self.intersect, proportion=0, flag=wx.ALL | wx.EXPAND, border=1)
  1003. border.Add(item=sizer, proportion=0, flag=wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM, border=5)
  1004. #
  1005. # save-on-exit box
  1006. #
  1007. box = wx.StaticBox (parent=panel, id=wx.ID_ANY, label=" %s " % _("Save changes"))
  1008. # save changes on exit?
  1009. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  1010. self.save = wx.CheckBox(parent=panel, label=_("Save changes on exit"))
  1011. self.save.SetValue(UserSettings.Get(group='vdigit', key='saveOnExit', subkey='enabled'))
  1012. sizer.Add(item=self.save, proportion=0, flag=wx.ALL | wx.EXPAND, border=1)
  1013. border.Add(item=sizer, proportion=0, flag=wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM, border=5)
  1014. panel.SetSizer(border)
  1015. return panel
  1016. def __CreateQueryPage(self, notebook):
  1017. """!Create notebook page for query tool"""
  1018. panel = wx.Panel(parent=notebook, id=wx.ID_ANY)
  1019. notebook.AddPage(page=panel, text=_("Query tool"))
  1020. border = wx.BoxSizer(wx.VERTICAL)
  1021. #
  1022. # query tool box
  1023. #
  1024. box = wx.StaticBox (parent=panel, id=wx.ID_ANY, label=" %s " % _("Choose query tool"))
  1025. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  1026. LocUnits = self.parent.MapWindow.Map.ProjInfo()['units']
  1027. self.queryBox = wx.CheckBox(parent=panel, id=wx.ID_ANY, label=_("Select by box"))
  1028. self.queryBox.SetValue(UserSettings.Get(group='vdigit', key="query", subkey='box'))
  1029. sizer.Add(item=self.queryBox, proportion=0, flag=wx.ALL | wx.EXPAND, border=1)
  1030. sizer.Add((0, 5))
  1031. #
  1032. # length
  1033. #
  1034. self.queryLength = wx.RadioButton(parent=panel, id=wx.ID_ANY, label=_("length"))
  1035. self.queryLength.Bind(wx.EVT_RADIOBUTTON, self.OnChangeQuery)
  1036. sizer.Add(item=self.queryLength, proportion=0, flag=wx.ALL | wx.EXPAND, border=1)
  1037. flexSizer = wx.FlexGridSizer (cols=4, hgap=5, vgap=5)
  1038. flexSizer.AddGrowableCol(0)
  1039. txt = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("Select lines"))
  1040. self.queryLengthSL = wx.Choice (parent=panel, id=wx.ID_ANY,
  1041. choices = [_("shorter than"), _("longer than")])
  1042. self.queryLengthSL.SetSelection(UserSettings.Get(group='vdigit', key="queryLength", subkey='than-selection'))
  1043. self.queryLengthValue = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(100, -1),
  1044. initial=1,
  1045. min=0, max=1e6)
  1046. self.queryLengthValue.SetValue(UserSettings.Get(group='vdigit', key="queryLength", subkey='thresh'))
  1047. units = wx.StaticText(parent=panel, id=wx.ID_ANY, label="%s" % LocUnits)
  1048. flexSizer.Add(txt, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
  1049. flexSizer.Add(self.queryLengthSL, proportion=0, flag=wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  1050. flexSizer.Add(self.queryLengthValue, proportion=0, flag=wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  1051. flexSizer.Add(units, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
  1052. sizer.Add(item=flexSizer, proportion=0, flag=wx.ALL | wx.EXPAND, border=1)
  1053. #
  1054. # dangle
  1055. #
  1056. self.queryDangle = wx.RadioButton(parent=panel, id=wx.ID_ANY, label=_("dangle"))
  1057. self.queryDangle.Bind(wx.EVT_RADIOBUTTON, self.OnChangeQuery)
  1058. sizer.Add(item=self.queryDangle, proportion=0, flag=wx.ALL | wx.EXPAND, border=1)
  1059. flexSizer = wx.FlexGridSizer (cols=4, hgap=5, vgap=5)
  1060. flexSizer.AddGrowableCol(0)
  1061. txt = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("Select dangles"))
  1062. self.queryDangleSL = wx.Choice (parent=panel, id=wx.ID_ANY,
  1063. choices = [_("shorter than"), _("longer than")])
  1064. self.queryDangleSL.SetSelection(UserSettings.Get(group='vdigit', key="queryDangle", subkey='than-selection'))
  1065. self.queryDangleValue = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(100, -1),
  1066. initial=1,
  1067. min=0, max=1e6)
  1068. self.queryDangleValue.SetValue(UserSettings.Get(group='vdigit', key="queryDangle", subkey='thresh'))
  1069. units = wx.StaticText(parent=panel, id=wx.ID_ANY, label="%s" % LocUnits)
  1070. flexSizer.Add(txt, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
  1071. flexSizer.Add(self.queryDangleSL, proportion=0, flag=wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  1072. flexSizer.Add(self.queryDangleValue, proportion=0, flag=wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  1073. flexSizer.Add(units, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
  1074. sizer.Add(item=flexSizer, proportion=0, flag=wx.ALL | wx.EXPAND, border=1)
  1075. if UserSettings.Get(group='vdigit', key="query", subkey='selection') == 0:
  1076. self.queryLength.SetValue(True)
  1077. else:
  1078. self.queryDangle.SetValue(True)
  1079. # enable & disable items
  1080. self.OnChangeQuery(None)
  1081. border.Add(item=sizer, proportion=0, flag=wx.ALL | wx.EXPAND, border=5)
  1082. panel.SetSizer(border)
  1083. return panel
  1084. def __CreateAttributesPage(self, notebook):
  1085. """!Create notebook page for query tool"""
  1086. panel = wx.Panel(parent=notebook, id=wx.ID_ANY)
  1087. notebook.AddPage(page=panel, text=_("Attributes"))
  1088. border = wx.BoxSizer(wx.VERTICAL)
  1089. #
  1090. # add new record
  1091. #
  1092. box = wx.StaticBox (parent=panel, id=wx.ID_ANY, label=" %s " % _("Digitize new feature"))
  1093. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  1094. # checkbox
  1095. self.addRecord = wx.CheckBox(parent=panel, id=wx.ID_ANY,
  1096. label=_("Add new record into table"))
  1097. self.addRecord.SetValue(UserSettings.Get(group='vdigit', key="addRecord", subkey='enabled'))
  1098. sizer.Add(item=self.addRecord, proportion=0, flag=wx.ALL | wx.EXPAND, border=1)
  1099. # settings
  1100. flexSizer = wx.FlexGridSizer(cols=2, hgap=3, vgap=3)
  1101. flexSizer.AddGrowableCol(0)
  1102. settings = ((_("Layer"), 1), (_("Category"), 1), (_("Mode"), _("Next to use")))
  1103. # layer
  1104. text = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("Layer"))
  1105. if self.parent.digit.map:
  1106. layers = map(str, self.parent.digit.GetLayers())
  1107. if len(layers) == 0:
  1108. layers = [str(UserSettings.Get(group='vdigit', key="layer", subkey='value')), ]
  1109. else:
  1110. layers = [str(UserSettings.Get(group='vdigit', key="layer", subkey='value')), ]
  1111. self.layer = wx.Choice(parent=panel, id=wx.ID_ANY, size=(125, -1),
  1112. choices=layers)
  1113. self.layer.SetStringSelection(str(UserSettings.Get(group='vdigit', key="layer", subkey='value')))
  1114. flexSizer.Add(item=text, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
  1115. flexSizer.Add(item=self.layer, proportion=0,
  1116. flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
  1117. # category number
  1118. text = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("Category number"))
  1119. self.category = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(125, -1),
  1120. initial=UserSettings.Get(group='vdigit', key="category", subkey='value'),
  1121. min=-1e9, max=1e9)
  1122. if UserSettings.Get(group='vdigit', key="categoryMode", subkey='selection') != 1:
  1123. self.category.Enable(False)
  1124. flexSizer.Add(item=text, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
  1125. flexSizer.Add(item=self.category, proportion=0,
  1126. flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
  1127. # category mode
  1128. text = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("Category mode"))
  1129. self.categoryMode = wx.Choice(parent=panel, id=wx.ID_ANY, size=(125, -1),
  1130. choices=[_("Next to use"), _("Manual entry"), _("No category")])
  1131. self.categoryMode.SetSelection(UserSettings.Get(group='vdigit', key="categoryMode", subkey='selection'))
  1132. flexSizer.Add(item=text, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
  1133. flexSizer.Add(item=self.categoryMode, proportion=0,
  1134. flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
  1135. sizer.Add(item=flexSizer, proportion=1, flag=wx.ALL | wx.EXPAND, border=1)
  1136. border.Add(item=sizer, proportion=0,
  1137. flag=wx.ALL | wx.EXPAND, border=5)
  1138. #
  1139. # digitize new area
  1140. #
  1141. box = wx.StaticBox (parent=panel, id=wx.ID_ANY, label=" %s " % _("Digitize new area"))
  1142. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  1143. # add centroid
  1144. self.addCentroid = wx.CheckBox(parent=panel, id=wx.ID_ANY,
  1145. label=_("Add centroid to left/right area"))
  1146. self.addCentroid.SetValue(UserSettings.Get(group='vdigit', key="addCentroid", subkey='enabled'))
  1147. sizer.Add(item=self.addCentroid, proportion=0, flag=wx.ALL | wx.EXPAND, border=1)
  1148. # attach category to boundary
  1149. self.catBoundary = wx.CheckBox(parent=panel, id=wx.ID_ANY,
  1150. label=_("Do not attach category to boundaries"))
  1151. self.catBoundary.SetValue(UserSettings.Get(group='vdigit', key="catBoundary", subkey='enabled'))
  1152. sizer.Add(item=self.catBoundary, proportion=0, flag=wx.ALL | wx.EXPAND, border=1)
  1153. border.Add(item=sizer, proportion=0,
  1154. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=5)
  1155. #
  1156. # delete existing record
  1157. #
  1158. box = wx.StaticBox (parent=panel, id=wx.ID_ANY, label=" %s " % _("Delete existing feature(s)"))
  1159. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  1160. # checkbox
  1161. self.deleteRecord = wx.CheckBox(parent=panel, id=wx.ID_ANY,
  1162. label=_("Delete record from table"))
  1163. self.deleteRecord.SetValue(UserSettings.Get(group='vdigit', key="delRecord", subkey='enabled'))
  1164. sizer.Add(item=self.deleteRecord, proportion=0, flag=wx.ALL | wx.EXPAND, border=1)
  1165. border.Add(item=sizer, proportion=0,
  1166. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=5)
  1167. #
  1168. # geometry attributes (currently only length and area are supported)
  1169. #
  1170. box = wx.StaticBox (parent = panel, id = wx.ID_ANY,
  1171. label = " %s " % _("Geometry attributes"))
  1172. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  1173. flexSizer = wx.FlexGridSizer(cols=2, hgap=3, vgap=3)
  1174. flexSizer.AddGrowableCol(0)
  1175. self.geomAttrb = { 'length' : { 'label' : _('length') },
  1176. 'area' : { 'label' : _('area') } }
  1177. digitToolbar = self.parent.toolbars['vdigit']
  1178. try:
  1179. vectorName = digitToolbar.GetLayer().GetName()
  1180. except AttributeError:
  1181. vectorName = None # no vector selected for editing
  1182. layer = UserSettings.Get(group='vdigit', key="layer", subkey='value')
  1183. for attrb in self.geomAttrb.keys():
  1184. # checkbox
  1185. check = wx.CheckBox(parent = panel, id = wx.ID_ANY,
  1186. label = self.geomAttrb[attrb]['label'])
  1187. ### self.deleteRecord.SetValue(UserSettings.Get(group='vdigit', key="delRecord", subkey='enabled'))
  1188. check.Bind(wx.EVT_CHECKBOX, self.OnGeomAttrb)
  1189. # column (only numeric)
  1190. column = gselect.ColumnSelect(parent = panel)
  1191. column.InsertColumns(vector = vectorName,
  1192. layer = layer, excludeKey = True,
  1193. type = ['integer', 'double precision'])
  1194. # default values
  1195. check.SetValue(UserSettings.Get(group='vdigit', key='geomAttrb',
  1196. subkey=[attrb, 'enabled'],
  1197. internal=True))
  1198. if not vectorName:
  1199. check.Enable(False)
  1200. column.Enable(False)
  1201. if not check.IsChecked():
  1202. column.Enable(False)
  1203. self.geomAttrb[attrb]['check'] = check.GetId()
  1204. self.geomAttrb[attrb]['column'] = column.GetId()
  1205. flexSizer.Add(item = check, proportion = 0,
  1206. flag = wx.ALIGN_CENTER_VERTICAL)
  1207. flexSizer.Add(item = column, proportion = 0)
  1208. sizer.Add(item=flexSizer, proportion=1,
  1209. flag=wx.ALL | wx.EXPAND, border=1)
  1210. border.Add(item=sizer, proportion=0,
  1211. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=5)
  1212. # bindings
  1213. self.Bind(wx.EVT_CHECKBOX, self.OnChangeAddRecord, self.addRecord)
  1214. self.Bind(wx.EVT_CHOICE, self.OnChangeCategoryMode, self.categoryMode)
  1215. self.Bind(wx.EVT_CHOICE, self.OnChangeLayer, self.layer)
  1216. panel.SetSizer(border)
  1217. return panel
  1218. def __SymbologyData(self):
  1219. """
  1220. Data for __CreateSymbologyPage()
  1221. label | checkbox | color
  1222. """
  1223. return (
  1224. # ("Background", "symbolBackground"),
  1225. (_("Highlight"), "highlight"),
  1226. (_("Highlight (duplicates)"), "highlightDupl"),
  1227. (_("Point"), "point"),
  1228. (_("Line"), "line"),
  1229. (_("Boundary (no area)"), "boundaryNo"),
  1230. (_("Boundary (one area)"), "boundaryOne"),
  1231. (_("Boundary (two areas)"), "boundaryTwo"),
  1232. (_("Centroid (in area)"), "centroidIn"),
  1233. (_("Centroid (outside area)"), "centroidOut"),
  1234. (_("Centroid (duplicate in area)"), "centroidDup"),
  1235. (_("Node (one line)"), "nodeOne"),
  1236. (_("Node (two lines)"), "nodeTwo"),
  1237. (_("Vertex"), "vertex"),
  1238. (_("Area (closed boundary + centroid)"), "area"),
  1239. (_("Direction"), "direction"),)
  1240. def OnGeomAttrb(self, event):
  1241. """!Register geometry attributes (enable/disable)"""
  1242. checked = event.IsChecked()
  1243. id = event.GetId()
  1244. key = None
  1245. for attrb, val in self.geomAttrb.iteritems():
  1246. if val['check'] == id:
  1247. key = attrb
  1248. break
  1249. column = self.FindWindowById(self.geomAttrb[key]['column'])
  1250. if checked:
  1251. column.Enable()
  1252. else:
  1253. column.Enable(False)
  1254. def OnChangeCategoryMode(self, event):
  1255. """!Change category mode"""
  1256. mode = event.GetSelection()
  1257. UserSettings.Set(group='vdigit', key="categoryMode", subkey='selection', value=mode)
  1258. if mode == 1: # manual entry
  1259. self.category.Enable(True)
  1260. elif self.category.IsEnabled(): # disable
  1261. self.category.Enable(False)
  1262. if mode == 2 and self.addRecord.IsChecked(): # no category
  1263. self.addRecord.SetValue(False)
  1264. self.parent.digit.SetCategory()
  1265. self.category.SetValue(UserSettings.Get(group='vdigit', key='category', subkey='value'))
  1266. def OnChangeLayer(self, event):
  1267. """!Layer changed"""
  1268. layer = int(event.GetString())
  1269. if layer > 0:
  1270. UserSettings.Set(group='vdigit', key='layer', subkey='value', value=layer)
  1271. self.parent.digit.SetCategory()
  1272. self.category.SetValue(UserSettings.Get(group='vdigit', key='category', subkey='value'))
  1273. event.Skip()
  1274. def OnChangeAddRecord(self, event):
  1275. """!Checkbox 'Add new record' status changed"""
  1276. self.category.SetValue(self.parent.digit.SetCategory())
  1277. def OnChangeSnappingValue(self, event):
  1278. """!Change snapping value - update static text"""
  1279. value = self.snappingValue.GetValue()
  1280. if value < 0:
  1281. region = self.parent.MapWindow.Map.GetRegion()
  1282. res = (region['nsres'] + region['ewres']) / 2.
  1283. threshold = self.parent.digit.driver.GetThreshold(value=res)
  1284. else:
  1285. if self.snappingUnit.GetStringSelection() == "map units":
  1286. threshold = value
  1287. else:
  1288. threshold = self.parent.digit.driver.GetThreshold(value=value)
  1289. if value == 0:
  1290. self.snappingInfo.SetLabel(_("Snapping disabled"))
  1291. elif value < 0:
  1292. self.snappingInfo.SetLabel(_("Snapping threshold is %(value).1f %(units)s "
  1293. "(based on comp. resolution)") %
  1294. {'value' : threshold,
  1295. 'units' : self.mapUnits.lower()})
  1296. else:
  1297. self.snappingInfo.SetLabel(_("Snapping threshold is %(value).1f %(units)s") %
  1298. {'value' : threshold,
  1299. 'units' : self.mapUnits.lower()})
  1300. event.Skip()
  1301. def OnChangeSnappingUnits(self, event):
  1302. """!Snapping units change -> update static text"""
  1303. value = self.snappingValue.GetValue()
  1304. units = self.snappingUnit.GetStringSelection()
  1305. threshold = self.parent.digit.driver.GetThreshold(value=value, units=units)
  1306. if units == "map units":
  1307. self.snappingInfo.SetLabel(_("Snapping threshold is %(value).1f %(units)s") %
  1308. {'value' : value,
  1309. 'units' : self.mapUnits})
  1310. else:
  1311. self.snappingInfo.SetLabel(_("Snapping threshold is %(value).1f %(units)s") %
  1312. {'value' : threshold,
  1313. 'units' : self.mapUnits})
  1314. event.Skip()
  1315. def OnChangeQuery(self, event):
  1316. """!Change query"""
  1317. if self.queryLength.GetValue():
  1318. # length
  1319. self.queryLengthSL.Enable(True)
  1320. self.queryLengthValue.Enable(True)
  1321. self.queryDangleSL.Enable(False)
  1322. self.queryDangleValue.Enable(False)
  1323. else:
  1324. # dangle
  1325. self.queryLengthSL.Enable(False)
  1326. self.queryLengthValue.Enable(False)
  1327. self.queryDangleSL.Enable(True)
  1328. self.queryDangleValue.Enable(True)
  1329. def OnSave(self, event):
  1330. """!Button 'Save' clicked"""
  1331. self.UpdateSettings()
  1332. self.parent.toolbars['vdigit'].settingsDialog = None
  1333. fileSettings = {}
  1334. UserSettings.ReadSettingsFile(settings=fileSettings)
  1335. fileSettings['vdigit'] = UserSettings.Get(group='vdigit')
  1336. file = UserSettings.SaveToFile(fileSettings)
  1337. self.parent.GetLayerManager().goutput.WriteLog(_('Vector digitizer settings saved to file <%s>.') % file)
  1338. self.Destroy()
  1339. event.Skip()
  1340. def OnApply(self, event):
  1341. """!Button 'Apply' clicked"""
  1342. self.UpdateSettings()
  1343. def OnCancel(self, event):
  1344. """!Button 'Cancel' clicked"""
  1345. self.parent.toolbars['vdigit'].settingsDialog = None
  1346. self.Destroy()
  1347. if event:
  1348. event.Skip()
  1349. def UpdateSettings(self):
  1350. """!Update UserSettings"""
  1351. # symbology
  1352. for key, (enabled, color) in self.symbology.iteritems():
  1353. if enabled:
  1354. UserSettings.Set(group='vdigit', key='symbol',
  1355. subkey=[key, 'enabled'],
  1356. value=enabled.IsChecked())
  1357. UserSettings.Set(group='vdigit', key='symbol',
  1358. subkey=[key, 'color'],
  1359. value=tuple(color.GetColour()))
  1360. else:
  1361. UserSettings.Set(group='vdigit', key='symbol',
  1362. subkey=[key, 'color'],
  1363. value=tuple(color.GetColour()))
  1364. # display
  1365. UserSettings.Set(group='vdigit', key="lineWidth", subkey='value',
  1366. value=int(self.lineWidthValue.GetValue()))
  1367. # snapping
  1368. UserSettings.Set(group='vdigit', key="snapping", subkey='value',
  1369. value=int(self.snappingValue.GetValue()))
  1370. UserSettings.Set(group='vdigit', key="snapping", subkey='units',
  1371. value=self.snappingUnit.GetStringSelection())
  1372. UserSettings.Set(group='vdigit', key="snapToVertex", subkey='enabled',
  1373. value=self.snapVertex.IsChecked())
  1374. # digitize new feature
  1375. UserSettings.Set(group='vdigit', key="addRecord", subkey='enabled',
  1376. value=self.addRecord.IsChecked())
  1377. UserSettings.Set(group='vdigit', key="layer", subkey='value',
  1378. value=int(self.layer.GetStringSelection()))
  1379. UserSettings.Set(group='vdigit', key="category", subkey='value',
  1380. value=int(self.category.GetValue()))
  1381. UserSettings.Set(group='vdigit', key="categoryMode", subkey='selection',
  1382. value=self.categoryMode.GetSelection())
  1383. # digitize new area
  1384. UserSettings.Set(group='vdigit', key="addCentroid", subkey='enabled',
  1385. value=self.addCentroid.IsChecked())
  1386. UserSettings.Set(group='vdigit', key="catBoundary", subkey='enabled',
  1387. value=self.catBoundary.IsChecked())
  1388. # delete existing feature
  1389. UserSettings.Set(group='vdigit', key="delRecord", subkey='enabled',
  1390. value=self.deleteRecord.IsChecked())
  1391. # geometry attributes
  1392. for key, val in self.geomAttrb.iteritems():
  1393. checked = self.FindWindowById(val['check']).IsChecked()
  1394. column = self.FindWindowById(val['column']).GetValue()
  1395. UserSettings.Set(group = 'vdigit', key = 'geomAttrb',
  1396. subkey = [key, 'enabled'], value = checked,
  1397. internal = True)
  1398. UserSettings.Set(group = 'vdigit', key = 'geomAttrb',
  1399. subkey = [key, 'column'], value = column,
  1400. internal = True)
  1401. # snapping threshold
  1402. self.parent.digit.threshold = self.parent.digit.driver.GetThreshold()
  1403. # query tool
  1404. if self.queryLength.GetValue():
  1405. UserSettings.Set(group='vdigit', key="query", subkey='selection',
  1406. value=0)
  1407. else:
  1408. UserSettings.Set(group='vdigit', key="query", subkey='type',
  1409. value=1)
  1410. UserSettings.Set(group='vdigit', key="query", subkey='box',
  1411. value=self.queryBox.IsChecked())
  1412. UserSettings.Set(group='vdigit', key="queryLength", subkey='than-selection',
  1413. value=self.queryLengthSL.GetSelection())
  1414. UserSettings.Set(group='vdigit', key="queryLength", subkey='thresh',
  1415. value=int(self.queryLengthValue.GetValue()))
  1416. UserSettings.Set(group='vdigit', key="queryDangle", subkey='than-selection',
  1417. value=self.queryDangleSL.GetSelection())
  1418. UserSettings.Set(group='vdigit', key="queryDangle", subkey='thresh',
  1419. value=int(self.queryDangleValue.GetValue()))
  1420. # select features
  1421. for feature in ('point', 'line',
  1422. 'centroid', 'boundary'):
  1423. UserSettings.Set(group='vdigit', key='selectType',
  1424. subkey=[feature, 'enabled'],
  1425. value=self.FindWindowById(self.selectFeature[feature]).IsChecked())
  1426. UserSettings.Set(group='vdigit', key="selectThresh", subkey='value',
  1427. value=int(self.selectThreshValue.GetValue()))
  1428. UserSettings.Set(group='vdigit', key="checkForDupl", subkey='enabled',
  1429. value=self.checkForDupl.IsChecked())
  1430. UserSettings.Set(group='vdigit', key="selectInside", subkey='enabled',
  1431. value=self.selectIn.IsChecked())
  1432. # on-exit
  1433. UserSettings.Set(group='vdigit', key="saveOnExit", subkey='enabled',
  1434. value=self.save.IsChecked())
  1435. # break lines
  1436. UserSettings.Set(group='vdigit', key="breakLines", subkey='enabled',
  1437. value=self.intersect.IsChecked())
  1438. # update driver settings
  1439. self.parent.digit.driver.UpdateSettings()
  1440. # update digit settings
  1441. self.parent.digit.UpdateSettings()
  1442. # redraw map if auto-rendering is enabled
  1443. if self.parent.autoRender.GetValue():
  1444. self.parent.OnRender(None)
  1445. class VDigitCategoryDialog(wx.Dialog, listmix.ColumnSorterMixin):
  1446. """
  1447. Dialog used to display/modify categories of vector objects
  1448. @param parent
  1449. @param title dialog title
  1450. @param query {coordinates, qdist} - used by v.edit/v.what
  1451. @param cats directory of lines (layer/categories) - used by vdigit
  1452. @param pos
  1453. @param style
  1454. """
  1455. def __init__(self, parent, title,
  1456. map, query=None, cats=None,
  1457. pos=wx.DefaultPosition,
  1458. style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER):
  1459. # parent
  1460. self.parent = parent # mapdisplay.BufferedWindow class instance
  1461. # map name
  1462. self.map = map
  1463. # line : {layer: [categories]}
  1464. self.cats = {}
  1465. # do not display dialog if no line is found (-> self.cats)
  1466. if cats is None:
  1467. if self.__GetCategories(query[0], query[1]) == 0 or not self.line:
  1468. Debug.msg(3, "VDigitCategoryDialog(): nothing found!")
  1469. else:
  1470. self.cats = cats
  1471. for line in cats.keys():
  1472. for layer in cats[line].keys():
  1473. self.cats[line][layer] = list(cats[line][layer])
  1474. layers = []
  1475. for layer in self.parent.parent.digit.GetLayers():
  1476. layers.append(str(layer))
  1477. # make copy of cats (used for 'reload')
  1478. self.cats_orig = copy.deepcopy(self.cats)
  1479. wx.Dialog.__init__(self, parent=self.parent, id=wx.ID_ANY, title=title,
  1480. style=style, pos=pos)
  1481. # list of categories
  1482. box = wx.StaticBox(parent=self, id=wx.ID_ANY,
  1483. label=" %s " % _("List of categories - right-click to delete"))
  1484. listSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  1485. self.list = CategoryListCtrl(parent=self, id=wx.ID_ANY,
  1486. style=wx.LC_REPORT |
  1487. wx.BORDER_NONE |
  1488. wx.LC_SORT_ASCENDING |
  1489. wx.LC_HRULES |
  1490. wx.LC_VRULES)
  1491. # sorter
  1492. self.fid = self.cats.keys()[0]
  1493. self.itemDataMap = self.list.Populate(self.cats[self.fid])
  1494. listmix.ColumnSorterMixin.__init__(self, 2)
  1495. self.fidMulti = wx.Choice(parent=self, id=wx.ID_ANY,
  1496. size=(150, -1))
  1497. self.fidMulti.Bind(wx.EVT_CHOICE, self.OnFeature)
  1498. self.fidText = wx.StaticText(parent=self, id=wx.ID_ANY)
  1499. if len(self.cats.keys()) == 1:
  1500. self.fidMulti.Show(False)
  1501. self.fidText.SetLabel(str(self.fid))
  1502. else:
  1503. self.fidText.Show(False)
  1504. choices = []
  1505. for fid in self.cats.keys():
  1506. choices.append(str(fid))
  1507. self.fidMulti.SetItems(choices)
  1508. self.fidMulti.SetSelection(0)
  1509. listSizer.Add(item=self.list, proportion=1, flag=wx.EXPAND)
  1510. # add new category
  1511. box = wx.StaticBox(parent=self, id=wx.ID_ANY,
  1512. label=" %s " % _("Add new category"))
  1513. addSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  1514. flexSizer = wx.FlexGridSizer (cols=5, hgap=5, vgap=5)
  1515. flexSizer.AddGrowableCol(3)
  1516. layerNewTxt = wx.StaticText(parent=self, id=wx.ID_ANY,
  1517. label="%s:" % _("Layer"))
  1518. self.layerNew = wx.Choice(parent=self, id=wx.ID_ANY, size=(75, -1),
  1519. choices=layers)
  1520. if len(layers) > 0:
  1521. self.layerNew.SetSelection(0)
  1522. catNewTxt = wx.StaticText(parent=self, id=wx.ID_ANY,
  1523. label="%s:" % _("Category"))
  1524. try:
  1525. newCat = max(self.cats[self.fid][1]) + 1
  1526. except KeyError:
  1527. newCat = 1
  1528. self.catNew = wx.SpinCtrl(parent=self, id=wx.ID_ANY, size=(75, -1),
  1529. initial=newCat, min=0, max=1e9)
  1530. btnAddCat = wx.Button(self, wx.ID_ADD)
  1531. flexSizer.Add(item=layerNewTxt, proportion=0,
  1532. flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
  1533. flexSizer.Add(item=self.layerNew, proportion=0,
  1534. flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
  1535. flexSizer.Add(item=catNewTxt, proportion=0,
  1536. flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_RIGHT | wx.LEFT,
  1537. border=10)
  1538. flexSizer.Add(item=self.catNew, proportion=0,
  1539. flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
  1540. flexSizer.Add(item=btnAddCat, proportion=0,
  1541. flag=wx.EXPAND | wx.ALIGN_RIGHT | wx.FIXED_MINSIZE)
  1542. addSizer.Add(item=flexSizer, proportion=1, flag=wx.ALL | wx.EXPAND, border=5)
  1543. # buttons
  1544. btnApply = wx.Button(self, wx.ID_APPLY)
  1545. btnApply.SetToolTipString(_("Apply changes"))
  1546. btnCancel = wx.Button(self, wx.ID_CANCEL)
  1547. btnCancel.SetToolTipString(_("Ignore changes and close dialog"))
  1548. btnOk = wx.Button(self, wx.ID_OK)
  1549. btnOk.SetToolTipString(_("Apply changes and close dialog"))
  1550. btnOk.SetDefault()
  1551. # sizers
  1552. btnSizer = wx.StdDialogButtonSizer()
  1553. btnSizer.AddButton(btnCancel)
  1554. #btnSizer.AddButton(btnReload)
  1555. #btnSizer.SetNegativeButton(btnReload)
  1556. btnSizer.AddButton(btnApply)
  1557. btnSizer.AddButton(btnOk)
  1558. btnSizer.Realize()
  1559. mainSizer = wx.BoxSizer(wx.VERTICAL)
  1560. mainSizer.Add(item=listSizer, proportion=1,
  1561. flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
  1562. mainSizer.Add(item=addSizer, proportion=0,
  1563. flag=wx.EXPAND | wx.ALIGN_CENTER |
  1564. wx.LEFT | wx.RIGHT | wx.BOTTOM, border=5)
  1565. fidSizer = wx.BoxSizer(wx.HORIZONTAL)
  1566. fidSizer.Add(item=wx.StaticText(parent=self, id=wx.ID_ANY,
  1567. label=_("Feature id:")),
  1568. proportion=0, border=5,
  1569. flag=wx.ALIGN_CENTER_VERTICAL)
  1570. fidSizer.Add(item=self.fidMulti, proportion=0,
  1571. flag=wx.EXPAND | wx.ALL, border=5)
  1572. fidSizer.Add(item=self.fidText, proportion=0,
  1573. flag=wx.EXPAND | wx.ALL, border=5)
  1574. mainSizer.Add(item=fidSizer, proportion=0,
  1575. flag=wx.EXPAND | wx.ALL, border=5)
  1576. mainSizer.Add(item=btnSizer, proportion=0,
  1577. flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
  1578. self.SetSizer(mainSizer)
  1579. mainSizer.Fit(self)
  1580. self.SetAutoLayout(True)
  1581. # set min size for dialog
  1582. self.SetMinSize(self.GetBestSize())
  1583. # bindings
  1584. # buttons
  1585. #btnReload.Bind(wx.EVT_BUTTON, self.OnReload)
  1586. btnApply.Bind(wx.EVT_BUTTON, self.OnApply)
  1587. btnOk.Bind(wx.EVT_BUTTON, self.OnOK)
  1588. btnAddCat.Bind(wx.EVT_BUTTON, self.OnAddCat)
  1589. btnCancel.Bind(wx.EVT_BUTTON, self.OnCancel)
  1590. # list
  1591. # self.Bind(wx.EVT_LIST_ITEM_SELECTED, self.OnItemSelected, self.list)
  1592. # self.list.Bind(wx.EVT_RIGHT_DOWN, self.OnRightDown)
  1593. self.list.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnRightUp) #wxMSW
  1594. self.list.Bind(wx.EVT_RIGHT_UP, self.OnRightUp) #wxGTK
  1595. self.Bind(wx.EVT_LIST_BEGIN_LABEL_EDIT, self.OnBeginEdit, self.list)
  1596. self.Bind(wx.EVT_LIST_END_LABEL_EDIT, self.OnEndEdit, self.list)
  1597. self.Bind(wx.EVT_LIST_COL_CLICK, self.OnColClick, self.list)
  1598. def GetListCtrl(self):
  1599. """!Used by ColumnSorterMixin"""
  1600. return self.list
  1601. def OnColClick(self, event):
  1602. """!Click on column header (order by)"""
  1603. event.Skip()
  1604. def OnBeginEdit(self, event):
  1605. """!Editing of item started"""
  1606. event.Allow()
  1607. def OnEndEdit(self, event):
  1608. """!Finish editing of item"""
  1609. itemIndex = event.GetIndex()
  1610. layerOld = int (self.list.GetItem(itemIndex, 0).GetText())
  1611. catOld = int (self.list.GetItem(itemIndex, 1).GetText())
  1612. if event.GetColumn() == 0:
  1613. layerNew = int(event.GetLabel())
  1614. catNew = catOld
  1615. else:
  1616. layerNew = layerOld
  1617. catNew = int(event.GetLabel())
  1618. try:
  1619. if layerNew not in self.cats[self.fid].keys():
  1620. self.cats[self.fid][layerNew] = []
  1621. self.cats[self.fid][layerNew].append(catNew)
  1622. self.cats[self.fid][layerOld].remove(catOld)
  1623. except:
  1624. event.Veto()
  1625. self.list.SetStringItem(itemIndex, 0, str(layerNew))
  1626. self.list.SetStringItem(itemIndex, 1, str(catNew))
  1627. dlg = wx.MessageDialog(self, _("Unable to add new layer/category <%(layer)s/%(category)s>.\n"
  1628. "Layer and category number must be integer.\n"
  1629. "Layer number must be greater then zero.") %
  1630. { 'layer': self.layerNew.GetStringSelection(),
  1631. 'category' : str(self.catNew.GetValue()) },
  1632. _("Error"), wx.OK | wx.ICON_ERROR)
  1633. dlg.ShowModal()
  1634. dlg.Destroy()
  1635. return False
  1636. def OnRightDown(self, event):
  1637. """!Mouse right button down"""
  1638. x = event.GetX()
  1639. y = event.GetY()
  1640. item, flags = self.list.HitTest((x, y))
  1641. if item != wx.NOT_FOUND and \
  1642. flags & wx.LIST_HITTEST_ONITEM:
  1643. self.list.Select(item)
  1644. event.Skip()
  1645. def OnRightUp(self, event):
  1646. """!Mouse right button up"""
  1647. if not hasattr(self, "popupID1"):
  1648. self.popupID1 = wx.NewId()
  1649. self.popupID2 = wx.NewId()
  1650. self.popupID3 = wx.NewId()
  1651. self.Bind(wx.EVT_MENU, self.OnItemDelete, id=self.popupID1)
  1652. self.Bind(wx.EVT_MENU, self.OnItemDeleteAll, id=self.popupID2)
  1653. self.Bind(wx.EVT_MENU, self.OnReload, id=self.popupID3)
  1654. # generate popup-menu
  1655. menu = wx.Menu()
  1656. menu.Append(self.popupID1, _("Delete selected"))
  1657. if self.list.GetFirstSelected() == -1:
  1658. menu.Enable(self.popupID1, False)
  1659. menu.Append(self.popupID2, _("Delete all"))
  1660. menu.AppendSeparator()
  1661. menu.Append(self.popupID3, _("Reload"))
  1662. self.PopupMenu(menu)
  1663. menu.Destroy()
  1664. def OnItemSelected(self, event):
  1665. """!Item selected"""
  1666. event.Skip()
  1667. def OnItemDelete(self, event):
  1668. """!Delete selected item(s) from the list (layer/category pair)"""
  1669. item = self.list.GetFirstSelected()
  1670. while item != -1:
  1671. layer = int (self.list.GetItem(item, 0).GetText())
  1672. cat = int (self.list.GetItem(item, 1).GetText())
  1673. self.list.DeleteItem(item)
  1674. self.cats[self.fid][layer].remove(cat)
  1675. item = self.list.GetFirstSelected()
  1676. event.Skip()
  1677. def OnItemDeleteAll(self, event):
  1678. """!Delete all items from the list"""
  1679. self.list.DeleteAllItems()
  1680. self.cats[self.fid] = {}
  1681. event.Skip()
  1682. def OnFeature(self, event):
  1683. """!Feature id changed (on duplicates)"""
  1684. self.fid = int(event.GetString())
  1685. self.itemDataMap = self.list.Populate(self.cats[self.fid],
  1686. update=True)
  1687. try:
  1688. newCat = max(self.cats[self.fid][1]) + 1
  1689. except KeyError:
  1690. newCat = 1
  1691. self.catNew.SetValue(newCat)
  1692. event.Skip()
  1693. def __GetCategories(self, coords, qdist):
  1694. """!Get layer/category pairs for all available
  1695. layers
  1696. Return True line found or False if not found"""
  1697. ret = gcmd.RunCommand('v.what',
  1698. parent = self,
  1699. quiet = True,
  1700. map = self.map,
  1701. east_north = '%f,%f' % \
  1702. (float(coords[0]), float(coords[1])),
  1703. distance = qdist)
  1704. if not ret:
  1705. return False
  1706. for item in ret.splitlines():
  1707. litem = item.lower()
  1708. if "id:" in litem: # get line id
  1709. self.line = int(item.split(':')[1].strip())
  1710. elif "layer:" in litem: # add layer
  1711. layer = int(item.split(':')[1].strip())
  1712. if layer not in self.cats.keys():
  1713. self.cats[layer] = []
  1714. elif "category:" in litem: # add category
  1715. self.cats[layer].append(int(item.split(':')[1].strip()))
  1716. return True
  1717. def OnReload(self, event):
  1718. """!Reload button pressed"""
  1719. # restore original list
  1720. self.cats = copy.deepcopy(self.cats_orig)
  1721. # polulate list
  1722. self.itemDataMap = self.list.Populate(self.cats[self.fid],
  1723. update=True)
  1724. event.Skip()
  1725. def OnCancel(self, event):
  1726. """!Cancel button pressed"""
  1727. self.parent.parent.dialogs['category'] = None
  1728. if self.parent.parent.digit:
  1729. self.parent.parent.digit.driver.SetSelected([])
  1730. self.parent.UpdateMap(render=False)
  1731. else:
  1732. self.parent.parent.OnRender(None)
  1733. self.Close()
  1734. def OnApply(self, event):
  1735. """!Apply button pressed"""
  1736. for fid in self.cats.keys():
  1737. newfid = self.ApplyChanges(fid)
  1738. if fid == self.fid:
  1739. self.fid = newfid
  1740. def ApplyChanges(self, fid):
  1741. cats = self.cats[fid]
  1742. cats_orig = self.cats_orig[fid]
  1743. # action : (catsFrom, catsTo)
  1744. check = {'catadd': (cats, cats_orig),
  1745. 'catdel': (cats_orig, cats)}
  1746. newfid = -1
  1747. # add/delete new category
  1748. for action, catsCurr in check.iteritems():
  1749. for layer in catsCurr[0].keys():
  1750. catList = []
  1751. for cat in catsCurr[0][layer]:
  1752. if layer not in catsCurr[1].keys() or \
  1753. cat not in catsCurr[1][layer]:
  1754. catList.append(cat)
  1755. if catList != []:
  1756. if action == 'catadd':
  1757. add = True
  1758. else:
  1759. add = False
  1760. newfid = self.parent.parent.digit.SetLineCats(fid, layer,
  1761. catList, add)
  1762. if len(self.cats.keys()) == 1:
  1763. self.fidText.SetLabel("%d" % newfid)
  1764. else:
  1765. choices = self.fidMulti.GetItems()
  1766. choices[choices.index(str(fid))] = str(newfid)
  1767. self.fidMulti.SetItems(choices)
  1768. self.fidMulti.SetStringSelection(str(newfid))
  1769. self.cats[newfid] = self.cats[fid]
  1770. del self.cats[fid]
  1771. fid = newfid
  1772. if self.fid < 0:
  1773. wx.MessageBox(parent=self, message=_("Unable to update vector map."),
  1774. caption=_("Error"), style=wx.OK | wx.ICON_ERROR)
  1775. self.cats_orig[fid] = copy.deepcopy(cats)
  1776. return newfid
  1777. def OnOK(self, event):
  1778. """!OK button pressed"""
  1779. self.OnApply(event)
  1780. self.OnCancel(event)
  1781. def OnAddCat(self, event):
  1782. """!Button 'Add' new category pressed"""
  1783. try:
  1784. layer = int(self.layerNew.GetStringSelection())
  1785. cat = int(self.catNew.GetValue())
  1786. if layer <= 0:
  1787. raise ValueError
  1788. except ValueError:
  1789. dlg = wx.MessageDialog(self, _("Unable to add new layer/category <%(layer)s/%(category)s>.\n"
  1790. "Layer and category number must be integer.\n"
  1791. "Layer number must be greater then zero.") %
  1792. {'layer' : str(self.layerNew.GetValue()),
  1793. 'category' : str(self.catNew.GetValue())},
  1794. _("Error"), wx.OK | wx.ICON_ERROR)
  1795. dlg.ShowModal()
  1796. dlg.Destroy()
  1797. return False
  1798. if layer not in self.cats[self.fid].keys():
  1799. self.cats[self.fid][layer] = []
  1800. self.cats[self.fid][layer].append(cat)
  1801. # reload list
  1802. self.itemDataMap = self.list.Populate(self.cats[self.fid],
  1803. update=True)
  1804. # update category number for add
  1805. self.catNew.SetValue(cat + 1)
  1806. event.Skip()
  1807. return True
  1808. def GetLine(self):
  1809. """!Get id of selected line of 'None' if no line is selected"""
  1810. return self.cats.keys()
  1811. def UpdateDialog(self, query=None, cats=None):
  1812. """!Update dialog
  1813. @param query {coordinates, distance} - v.edit/v.what
  1814. @param cats directory layer/cats - vdigit
  1815. Return True if updated otherwise False
  1816. """
  1817. # line: {layer: [categories]}
  1818. self.cats = {}
  1819. # do not display dialog if no line is found (-> self.cats)
  1820. if cats is None:
  1821. ret = self.__GetCategories(query[0], query[1])
  1822. else:
  1823. self.cats = cats
  1824. for line in cats.keys():
  1825. for layer in cats[line].keys():
  1826. self.cats[line][layer] = list(cats[line][layer])
  1827. ret = 1
  1828. if ret == 0 or len(self.cats.keys()) < 1:
  1829. Debug.msg(3, "VDigitCategoryDialog(): nothing found!")
  1830. return False
  1831. # make copy of cats (used for 'reload')
  1832. self.cats_orig = copy.deepcopy(self.cats)
  1833. # polulate list
  1834. self.fid = self.cats.keys()[0]
  1835. self.itemDataMap = self.list.Populate(self.cats[self.fid],
  1836. update=True)
  1837. try:
  1838. newCat = max(self.cats[self.fid][1]) + 1
  1839. except KeyError:
  1840. newCat = 1
  1841. self.catNew.SetValue(newCat)
  1842. if len(self.cats.keys()) == 1:
  1843. self.fidText.Show(True)
  1844. self.fidMulti.Show(False)
  1845. self.fidText.SetLabel("%d" % self.fid)
  1846. else:
  1847. self.fidText.Show(False)
  1848. self.fidMulti.Show(True)
  1849. choices = []
  1850. for fid in self.cats.keys():
  1851. choices.append(str(fid))
  1852. self.fidMulti.SetItems(choices)
  1853. self.fidMulti.SetSelection(0)
  1854. self.Layout()
  1855. return True
  1856. class CategoryListCtrl(wx.ListCtrl,
  1857. listmix.ListCtrlAutoWidthMixin,
  1858. listmix.TextEditMixin):
  1859. """!List of layers/categories"""
  1860. def __init__(self, parent, id, pos=wx.DefaultPosition,
  1861. size=wx.DefaultSize, style=0):
  1862. self.parent = parent
  1863. wx.ListCtrl.__init__(self, parent, id, pos, size, style)
  1864. listmix.ListCtrlAutoWidthMixin.__init__(self)
  1865. listmix.TextEditMixin.__init__(self)
  1866. def Populate(self, cats, update=False):
  1867. """!Populate the list"""
  1868. itemData = {} # requested by sorter
  1869. if not update:
  1870. self.InsertColumn(0, _("Layer"))
  1871. self.InsertColumn(1, _("Category"))
  1872. else:
  1873. self.DeleteAllItems()
  1874. i = 1
  1875. for layer in cats.keys():
  1876. catsList = cats[layer]
  1877. for cat in catsList:
  1878. index = self.InsertStringItem(sys.maxint, str(catsList[0]))
  1879. self.SetStringItem(index, 0, str(layer))
  1880. self.SetStringItem(index, 1, str(cat))
  1881. self.SetItemData(index, i)
  1882. itemData[i] = (str(layer), str(cat))
  1883. i = i + 1
  1884. if not update:
  1885. self.SetColumnWidth(0, 100)
  1886. self.SetColumnWidth(1, wx.LIST_AUTOSIZE)
  1887. self.currentItem = 0
  1888. return itemData
  1889. class VDigitZBulkDialog(wx.Dialog):
  1890. """
  1891. Dialog used for Z bulk-labeling tool
  1892. """
  1893. def __init__(self, parent, title, nselected, style=wx.DEFAULT_DIALOG_STYLE):
  1894. wx.Dialog.__init__(self, parent=parent, id=wx.ID_ANY, title=title, style=style)
  1895. self.parent = parent # mapdisplay.BufferedWindow class instance
  1896. # panel = wx.Panel(parent=self, id=wx.ID_ANY)
  1897. border = wx.BoxSizer(wx.VERTICAL)
  1898. txt = wx.StaticText(parent=self,
  1899. label=_("%d lines selected for z bulk-labeling") % nselected);
  1900. border.Add(item=txt, proportion=0, flag=wx.ALL | wx.EXPAND, border=5)
  1901. box = wx.StaticBox (parent=self, id=wx.ID_ANY, label=" %s " % _("Set value"))
  1902. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  1903. flexSizer = wx.FlexGridSizer (cols=2, hgap=5, vgap=5)
  1904. flexSizer.AddGrowableCol(0)
  1905. # starting value
  1906. txt = wx.StaticText(parent=self,
  1907. label=_("Starting value"));
  1908. self.value = wx.SpinCtrl(parent=self, id=wx.ID_ANY, size=(150, -1),
  1909. initial=0,
  1910. min=-1e6, max=1e6)
  1911. flexSizer.Add(txt, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
  1912. flexSizer.Add(self.value, proportion=0, flag=wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  1913. # step
  1914. txt = wx.StaticText(parent=self,
  1915. label=_("Step"))
  1916. self.step = wx.SpinCtrl(parent=self, id=wx.ID_ANY, size=(150, -1),
  1917. initial=0,
  1918. min=0, max=1e6)
  1919. flexSizer.Add(txt, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
  1920. flexSizer.Add(self.step, proportion=0, flag=wx.ALIGN_CENTER | wx.FIXED_MINSIZE)
  1921. sizer.Add(item=flexSizer, proportion=1, flag=wx.ALL | wx.EXPAND, border=1)
  1922. border.Add(item=sizer, proportion=1, flag=wx.ALL | wx.EXPAND, border=5)
  1923. # buttons
  1924. btnCancel = wx.Button(self, wx.ID_CANCEL)
  1925. btnOk = wx.Button(self, wx.ID_OK)
  1926. btnOk.SetDefault()
  1927. # sizers
  1928. btnSizer = wx.StdDialogButtonSizer()
  1929. btnSizer.AddButton(btnCancel)
  1930. btnSizer.AddButton(btnOk)
  1931. btnSizer.Realize()
  1932. mainSizer = wx.BoxSizer(wx.VERTICAL)
  1933. mainSizer.Add(item=border, proportion=1, flag=wx.EXPAND | wx.ALL, border=5)
  1934. mainSizer.Add(item=btnSizer, proportion=0,
  1935. flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
  1936. self.SetSizer(mainSizer)
  1937. mainSizer.Fit(self)
  1938. class VDigitDuplicatesDialog(wx.Dialog):
  1939. """
  1940. Show duplicated feature ids
  1941. """
  1942. def __init__(self, parent, data, title=_("List of duplicates"),
  1943. style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER,
  1944. pos=wx.DefaultPosition):
  1945. wx.Dialog.__init__(self, parent=parent, id=wx.ID_ANY, title=title, style=style,
  1946. pos=pos)
  1947. self.parent = parent # BufferedWindow
  1948. self.data = data
  1949. self.winList = []
  1950. # panel = wx.Panel(parent=self, id=wx.ID_ANY)
  1951. # notebook
  1952. self.notebook = wx.Notebook(parent=self, id=wx.ID_ANY, style=wx.BK_DEFAULT)
  1953. id = 1
  1954. for key in self.data.keys():
  1955. panel = wx.Panel(parent=self.notebook, id=wx.ID_ANY)
  1956. self.notebook.AddPage(page=panel, text=" %d " % (id))
  1957. # notebook body
  1958. border = wx.BoxSizer(wx.VERTICAL)
  1959. win = CheckListFeature(parent=panel, data=list(self.data[key]))
  1960. self.winList.append(win.GetId())
  1961. border.Add(item=win, proportion=1,
  1962. flag=wx.ALL | wx.EXPAND, border=5)
  1963. panel.SetSizer(border)
  1964. id += 1
  1965. # buttons
  1966. btnCancel = wx.Button(self, wx.ID_CANCEL)
  1967. btnOk = wx.Button(self, wx.ID_OK)
  1968. btnOk.SetDefault()
  1969. # sizers
  1970. btnSizer = wx.StdDialogButtonSizer()
  1971. btnSizer.AddButton(btnCancel)
  1972. btnSizer.AddButton(btnOk)
  1973. btnSizer.Realize()
  1974. mainSizer = wx.BoxSizer(wx.VERTICAL)
  1975. mainSizer.Add(item=self.notebook, proportion=1, flag=wx.EXPAND | wx.ALL, border=5)
  1976. mainSizer.Add(item=btnSizer, proportion=0,
  1977. flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
  1978. self.SetSizer(mainSizer)
  1979. mainSizer.Fit(self)
  1980. self.SetAutoLayout(True)
  1981. # set min size for dialog
  1982. self.SetMinSize((250, 180))
  1983. def GetUnSelected(self):
  1984. """!Get unselected items (feature id)
  1985. @return list of ids
  1986. """
  1987. ids = []
  1988. for id in self.winList:
  1989. wlist = self.FindWindowById(id)
  1990. for item in range(wlist.GetItemCount()):
  1991. if not wlist.IsChecked(item):
  1992. ids.append(int(wlist.GetItem(item, 0).GetText()))
  1993. return ids
  1994. class CheckListFeature(wx.ListCtrl, listmix.ListCtrlAutoWidthMixin, listmix.CheckListCtrlMixin):
  1995. """!List of mapset/owner/group"""
  1996. def __init__(self, parent, data,
  1997. pos=wx.DefaultPosition, log=None):
  1998. self.parent = parent
  1999. self.data = data
  2000. wx.ListCtrl.__init__(self, parent, wx.ID_ANY,
  2001. style=wx.LC_REPORT)
  2002. listmix.CheckListCtrlMixin.__init__(self)
  2003. self.log = log
  2004. # setup mixins
  2005. listmix.ListCtrlAutoWidthMixin.__init__(self)
  2006. self.LoadData(self.data)
  2007. def LoadData(self, data):
  2008. """!Load data into list"""
  2009. self.InsertColumn(0, _('Feature id'))
  2010. self.InsertColumn(1, _('Layer (Categories)'))
  2011. for item in data:
  2012. index = self.InsertStringItem(sys.maxint, str(item[0]))
  2013. self.SetStringItem(index, 1, str(item[1]))
  2014. # enable all items by default
  2015. for item in range(self.GetItemCount()):
  2016. self.CheckItem(item, True)
  2017. self.SetColumnWidth(col=0, width=wx.LIST_AUTOSIZE_USEHEADER)
  2018. self.SetColumnWidth(col=1, width=wx.LIST_AUTOSIZE_USEHEADER)
  2019. def OnCheckItem(self, index, flag):
  2020. """!Mapset checked/unchecked"""
  2021. pass