nviz_tools.py 109 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701
  1. """
  2. @package nviz_tools.py
  3. @brief Nviz tools window
  4. Classes:
  5. - NvizToolWindow
  6. - ViewPositionWindow
  7. (C) 2008 by the GRASS Development Team
  8. This program is free software under the GNU General Public
  9. License (>=v2). Read the file COPYING that comes with GRASS
  10. for details.
  11. @author Martin Landa <landa.martin gmail.com> (Google SoC 2008)
  12. """
  13. import os
  14. import sys
  15. import wx
  16. import wx.lib.colourselect as csel
  17. import globalvar
  18. import gselect
  19. import gcmd
  20. from preferences import globalSettings as UserSettings
  21. from nviz_mapdisp import wxUpdateView as wxUpdateView
  22. from nviz_mapdisp import wxUpdateProperties as wxUpdateProperties
  23. sys.path.append(os.path.join(globalvar.ETCWXDIR, "nviz"))
  24. try:
  25. import grass7_wxnviz as wxnviz
  26. except ImportError:
  27. pass
  28. class NvizToolWindow(wx.Frame):
  29. """Experimental window for Nviz tools
  30. @todo integrate with Map display
  31. """
  32. def __init__(self, parent=None, id=wx.ID_ANY, title=_("Nviz tools"),
  33. pos=wx.DefaultPosition, size=wx.DefaultSize,
  34. style=wx.DEFAULT_FRAME_STYLE, mapWindow=None):
  35. self.parent = parent # MapFrame
  36. self.lmgr = self.parent.gismanager # GMFrame
  37. self.mapWindow = mapWindow
  38. wx.Frame.__init__(self, parent, id, title, pos, size, style)
  39. #
  40. # icon
  41. #
  42. self.SetIcon(wx.Icon(os.path.join(globalvar.ETCICONDIR, 'grass_nviz.ico'), wx.BITMAP_TYPE_ICO))
  43. #
  44. # dialog body
  45. #
  46. mainSizer = wx.BoxSizer(wx.VERTICAL)
  47. self.win = {} # window ids
  48. #
  49. # notebook
  50. #
  51. self.notebook = wx.Notebook(parent=self, id=wx.ID_ANY, style=wx.BK_DEFAULT)
  52. self.page = {}
  53. # view page
  54. self.__createViewPage()
  55. self.page['view'] = { 'id' : 0 }
  56. # surface page
  57. size = self.__createSurfacePage()
  58. size = (size[0] + 25, size[0] + 20)
  59. # vector page
  60. self.__createVectorPage()
  61. # volume page
  62. self.__createVolumePage()
  63. # settings page
  64. self.__createSettingsPage()
  65. self.page['settings'] = { 'id' : 1 }
  66. self.UpdatePage('settings')
  67. self.pageChanging = False
  68. mainSizer.Add(item=self.notebook, proportion=1,
  69. flag=wx.EXPAND | wx.ALL, border=5)
  70. #
  71. # bindings
  72. #
  73. self.Bind(wx.EVT_CLOSE, self.OnClose)
  74. #
  75. # layout
  76. #
  77. self.SetSizer(mainSizer)
  78. mainSizer.Fit(self)
  79. self.SetSize(size)
  80. def PostViewEvent(self, zExag=False):
  81. """Change view settings"""
  82. event = wxUpdateView(zExag=zExag)
  83. wx.PostEvent(self.mapWindow, event)
  84. def __createViewPage(self):
  85. """Create view settings page"""
  86. panel = wx.Panel(parent=self.notebook, id=wx.ID_ANY)
  87. self.notebook.AddPage(page=panel,
  88. text=" %s " % _("View"))
  89. pageSizer = wx.BoxSizer(wx.VERTICAL)
  90. gridSizer = wx.GridBagSizer(vgap=3, hgap=3)
  91. self.win['view'] = {}
  92. # position
  93. posSizer = wx.GridBagSizer(vgap=3, hgap=3)
  94. posSizer.Add(item=wx.StaticText(panel, id=wx.ID_ANY, label=_("W")),
  95. pos=(1, 0), flag=wx.ALIGN_CENTER)
  96. posSizer.Add(item=wx.StaticText(panel, id=wx.ID_ANY, label=_("N")),
  97. pos=(0, 1), flag=wx.ALIGN_CENTER | wx.ALIGN_BOTTOM)
  98. viewPos = ViewPositionWindow(panel, id=wx.ID_ANY, size=(175, 175),
  99. mapwindow=self.mapWindow)
  100. self.win['view']['pos'] = viewPos.GetId()
  101. posSizer.Add(item=viewPos,
  102. pos=(1, 1), flag=wx.ALIGN_CENTER | wx.ALIGN_CENTER_VERTICAL)
  103. posSizer.Add(item=wx.StaticText(panel, id=wx.ID_ANY, label=_("S")),
  104. pos=(2, 1), flag=wx.ALIGN_CENTER | wx.ALIGN_TOP)
  105. posSizer.Add(item=wx.StaticText(panel, id=wx.ID_ANY, label=_("E")),
  106. pos=(1, 2), flag=wx.ALIGN_CENTER)
  107. gridSizer.Add(item=posSizer, pos=(0, 0))
  108. # perspective
  109. range = UserSettings.Get(group='nviz', key='view', subkey='persp', internal=True)
  110. self.CreateControl(panel, dict=self.win['view'], name='persp',
  111. range=(range['min'], range['max']),
  112. bind=(self.OnViewChange, self.OnViewChanged, self.OnViewChangedSpin))
  113. gridSizer.Add(item=wx.StaticText(panel, id=wx.ID_ANY, label=_("Perspective:")),
  114. pos=(1, 0), flag=wx.ALIGN_CENTER)
  115. gridSizer.Add(item=self.FindWindowById(self.win['view']['persp']['slider']), pos=(2, 0))
  116. gridSizer.Add(item=self.FindWindowById(self.win['view']['persp']['spin']), pos=(3, 0),
  117. flag=wx.ALIGN_CENTER)
  118. # twist
  119. range = UserSettings.Get(group='nviz', key='view', subkey='twist', internal=True)
  120. self.CreateControl(panel, dict=self.win['view'], name='twist',
  121. range=(range['min'], range['max']),
  122. bind=(self.OnViewChange, self.OnViewChanged, self.OnViewChangedSpin))
  123. gridSizer.Add(item=wx.StaticText(panel, id=wx.ID_ANY, label=_("Twist:")),
  124. pos=(1, 1), flag=wx.ALIGN_CENTER)
  125. gridSizer.Add(item=self.FindWindowById(self.win['view']['twist']['slider']), pos=(2, 1))
  126. gridSizer.Add(item=self.FindWindowById(self.win['view']['twist']['spin']), pos=(3, 1),
  127. flag=wx.ALIGN_CENTER)
  128. # height + z-exag
  129. self.CreateControl(panel, dict=self.win['view'], name='height', sliderHor=False,
  130. range=(0, 1),
  131. bind=(self.OnViewChange, self.OnViewChanged, self.OnViewChangedSpin))
  132. self.CreateControl(panel, dict=self.win['view'], name='z-exag', sliderHor=False,
  133. range=(0, 1),
  134. bind=(self.OnViewChange, self.OnViewChanged, self.OnViewChangedSpin))
  135. heightSizer = wx.GridBagSizer(vgap=3, hgap=3)
  136. heightSizer.Add(item=wx.StaticText(panel, id=wx.ID_ANY, label=_("Height:")),
  137. pos=(0, 0), flag=wx.ALIGN_LEFT, span=(1, 2))
  138. heightSizer.Add(item=self.FindWindowById(self.win['view']['height']['slider']),
  139. flag=wx.ALIGN_RIGHT, pos=(1, 0))
  140. heightSizer.Add(item=self.FindWindowById(self.win['view']['height']['spin']),
  141. flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_LEFT | wx.TOP |
  142. wx.BOTTOM | wx.RIGHT, pos=(1, 1))
  143. heightSizer.Add(item=wx.StaticText(panel, id=wx.ID_ANY, label=_("Z-exag:")),
  144. pos=(0, 2), flag=wx.ALIGN_LEFT, span=(1, 2))
  145. heightSizer.Add(item=self.FindWindowById(self.win['view']['z-exag']['slider']),
  146. flag=wx.ALIGN_RIGHT, pos=(1, 2))
  147. heightSizer.Add(item=self.FindWindowById(self.win['view']['z-exag']['spin']),
  148. flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_LEFT | wx.TOP |
  149. wx.BOTTOM | wx.RIGHT, pos=(1, 3))
  150. gridSizer.Add(item=heightSizer, pos=(0, 1), flag=wx.ALIGN_RIGHT)
  151. # view setup + reset
  152. viewSizer = wx.BoxSizer(wx.HORIZONTAL)
  153. viewSizer.Add(item=wx.StaticText(panel, id=wx.ID_ANY,
  154. label=_("Look at:")),
  155. flag=wx.ALL | wx.ALIGN_CENTER_VERTICAL, border=5)
  156. viewType = wx.Choice (parent=panel, id=wx.ID_ANY, size=(125, -1),
  157. choices = [_("top"),
  158. _("north"),
  159. _("south"),
  160. _("east"),
  161. _("west"),
  162. _("north-west"),
  163. _("north-east"),
  164. _("south-east"),
  165. _("south-west")])
  166. viewType.SetSelection(0)
  167. viewType.Bind(wx.EVT_CHOICE, self.OnLookAt)
  168. # self.win['lookAt'] = viewType.GetId()
  169. viewSizer.Add(item=viewType, flag=wx.ALL | wx.ALIGN_CENTER_VERTICAL,
  170. border=5)
  171. reset = wx.Button(panel, id=wx.ID_ANY, label=_("Reset"))
  172. reset.SetToolTipString(_("Reset to default view"))
  173. # self.win['reset'] = reset.GetId()
  174. reset.Bind(wx.EVT_BUTTON, self.OnResetView)
  175. viewSizer.Add(item=reset, proportion=1,
  176. flag=wx.EXPAND | wx.ALL | wx.ALIGN_RIGHT,
  177. border=5)
  178. gridSizer.AddGrowableCol(3)
  179. gridSizer.Add(item=viewSizer, pos=(4, 0), span=(1, 2),
  180. flag=wx.EXPAND)
  181. # body
  182. pageSizer.Add(item=gridSizer, proportion=1,
  183. flag=wx.EXPAND | wx.ALL,
  184. border=5)
  185. panel.SetSizer(pageSizer)
  186. return panel.GetBestSize()
  187. def __createSurfacePage(self):
  188. """Create view settings page"""
  189. panel = wx.Panel(parent=self.notebook, id=wx.ID_ANY)
  190. self.page['surface'] = {}
  191. self.page['surface']['id'] = -1
  192. self.page['surface']['panel'] = panel.GetId()
  193. # panel = scrolled.ScrolledPanel(parent=self.notebook, id=wx.ID_ANY)
  194. # panel.SetupScrolling(scroll_x=True, scroll_y=True)
  195. pageSizer = wx.BoxSizer(wx.VERTICAL)
  196. self.win['surface'] = {}
  197. #
  198. # surface attributes
  199. #
  200. box = wx.StaticBox (parent=panel, id=wx.ID_ANY,
  201. label=" %s " % (_("Surface attributes")))
  202. boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  203. gridSizer = wx.GridBagSizer(vgap=3, hgap=3)
  204. # type
  205. self.win['surface']['attr'] = {}
  206. row = 0
  207. for code, attrb in (('topo', _("Topography")),
  208. ('color', _("Color")),
  209. ('mask', _("Mask")),
  210. ('transp', _("Transparency")),
  211. ('shine', _("Shininess")),
  212. ('emit', _("Emission"))):
  213. self.win['surface'][code] = {}
  214. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  215. label=attrb + ':'),
  216. pos=(row, 0), flag=wx.ALIGN_CENTER_VERTICAL)
  217. use = wx.Choice (parent=panel, id=wx.ID_ANY, size=(100, -1),
  218. choices = [_("map")])
  219. if code not in ('topo', 'color', 'shine'):
  220. use.Insert(item=_("unset"), pos=0)
  221. self.win['surface'][code]['required'] = False
  222. else:
  223. self.win['surface'][code]['required'] = True
  224. if code != 'mask':
  225. use.Append(item=_('constant'))
  226. self.win['surface'][code]['use'] = use.GetId()
  227. use.Bind(wx.EVT_CHOICE, self.OnMapObjUse)
  228. gridSizer.Add(item=use, flag=wx.ALIGN_CENTER_VERTICAL,
  229. pos=(row, 1))
  230. map = gselect.Select(parent=panel, id=wx.ID_ANY,
  231. # size=globalvar.DIALOG_GSELECT_SIZE,
  232. size=(200, -1),
  233. type="raster")
  234. self.win['surface'][code]['map'] = map.GetId() - 1 # FIXME
  235. map.Bind(wx.EVT_TEXT, self.OnSurfaceMap)
  236. # changing map topography not allowed
  237. if code == 'topo':
  238. map.Enable(False)
  239. gridSizer.Add(item=map, flag=wx.ALIGN_CENTER_VERTICAL,
  240. pos=(row, 2))
  241. if code == 'color':
  242. value = csel.ColourSelect(panel, id=wx.ID_ANY,
  243. colour=UserSettings.Get(group='nviz', key='surface',
  244. subkey=['color', 'value']))
  245. value.Bind(csel.EVT_COLOURSELECT, self.OnSurfaceMap)
  246. elif code == 'mask':
  247. value = None
  248. else:
  249. value = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  250. initial=0)
  251. if code == 'topo':
  252. value.SetRange(minVal=-1e9, maxVal=1e9)
  253. elif code in ('shine', 'transp', 'emit'):
  254. value.SetRange(minVal=0, maxVal=255)
  255. else:
  256. value.SetRange(minVal=0, maxVal=100)
  257. value.Bind(wx.EVT_TEXT, self.OnSurfaceMap)
  258. if value:
  259. self.win['surface'][code]['const'] = value.GetId()
  260. value.Enable(False)
  261. gridSizer.Add(item=value, flag=wx.ALIGN_CENTER_VERTICAL,
  262. pos=(row, 3))
  263. else:
  264. self.win['surface'][code]['const'] = None
  265. self.SetMapObjUseMap(nvizType='surface',
  266. attrb=code) # -> enable map / disable constant
  267. row += 1
  268. boxSizer.Add(item=gridSizer, proportion=1,
  269. flag=wx.ALL | wx.EXPAND, border=3)
  270. pageSizer.Add(item=boxSizer, proportion=0,
  271. flag=wx.EXPAND | wx.ALL,
  272. border=5)
  273. #
  274. # draw
  275. #
  276. self.win['surface']['draw'] = {}
  277. box = wx.StaticBox (parent=panel, id=wx.ID_ANY,
  278. label=" %s " % (_("Draw")))
  279. boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  280. gridSizer = wx.GridBagSizer(vgap=5, hgap=5)
  281. gridSizer.AddGrowableCol(4)
  282. # mode
  283. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  284. label=_("Mode:")),
  285. pos=(0, 0), flag=wx.ALIGN_CENTER_VERTICAL)
  286. mode = wx.Choice (parent=panel, id=wx.ID_ANY, size=(100, -1),
  287. choices = [_("coarse"),
  288. _("fine"),
  289. _("both")])
  290. mode.SetSelection(0)
  291. mode.SetName("selection")
  292. mode.Bind(wx.EVT_CHOICE, self.OnSurfaceMode)
  293. self.win['surface']['draw']['mode'] = mode.GetId()
  294. gridSizer.Add(item=mode, flag=wx.ALIGN_CENTER_VERTICAL,
  295. pos=(0, 1))
  296. # resolution (mode)
  297. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  298. label=_("Resolution:")),
  299. pos=(0, 2), flag=wx.ALIGN_CENTER_VERTICAL)
  300. resSizer = wx.BoxSizer(wx.HORIZONTAL)
  301. resSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  302. label=_("coarse:")),
  303. flag=wx.ALL | wx.ALIGN_CENTER_VERTICAL, border=3)
  304. resC = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  305. initial=1,
  306. min=1,
  307. max=100)
  308. resC.SetName("value")
  309. self.win['surface']['draw']['res-coarse'] = resC.GetId()
  310. resC.Bind(wx.EVT_SPINCTRL, self.OnSurfaceResolution)
  311. resSizer.Add(item=resC, flag=wx.ALL, border=3)
  312. resSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  313. label=_("fine:")),
  314. flag=wx.ALL | wx.ALIGN_CENTER_VERTICAL, border=3)
  315. resF = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  316. initial=1,
  317. min=1,
  318. max=100)
  319. resF.SetName("value")
  320. self.win['surface']['draw']['res-fine'] = resF.GetId()
  321. resF.Bind(wx.EVT_SPINCTRL, self.OnSurfaceResolution)
  322. resSizer.Add(item=resF, flag=wx.ALL, border=3)
  323. gridSizer.Add(item=resSizer, pos=(0, 3), span=(1, 2))
  324. # style
  325. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  326. label=_("Coarse style:")),
  327. pos=(1, 0), flag=wx.ALIGN_CENTER_VERTICAL)
  328. style = wx.Choice (parent=panel, id=wx.ID_ANY, size=(100, -1),
  329. choices = [_("wire"),
  330. _("surface")])
  331. style.SetName("selection")
  332. self.win['surface']['draw']['style'] = style.GetId()
  333. style.Bind(wx.EVT_CHOICE, self.OnSurfaceMode)
  334. gridSizer.Add(item=style, flag=wx.ALIGN_CENTER_VERTICAL,
  335. pos=(1, 1))
  336. # shading
  337. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  338. label=_("Shading:")),
  339. pos=(1, 2), flag=wx.ALIGN_CENTER_VERTICAL)
  340. shade = wx.Choice (parent=panel, id=wx.ID_ANY, size=(100, -1),
  341. choices = [_("flat"),
  342. _("gouraud")])
  343. shade.SetName("selection")
  344. self.win['surface']['draw']['shading'] = shade.GetId()
  345. shade.Bind(wx.EVT_CHOICE, self.OnSurfaceMode)
  346. gridSizer.Add(item=shade, flag=wx.ALIGN_CENTER_VERTICAL,
  347. pos=(1, 3))
  348. # color
  349. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  350. label=_("Wire color:")),
  351. pos=(2, 0), flag=wx.ALIGN_CENTER_VERTICAL)
  352. color = csel.ColourSelect(panel, id=wx.ID_ANY)
  353. color.SetName("colour")
  354. color.Bind(csel.EVT_COLOURSELECT, self.OnSurfaceWireColor)
  355. self.win['surface']['draw']['wire-color'] = color.GetId()
  356. gridSizer.Add(item=color, flag=wx.ALIGN_CENTER_VERTICAL,
  357. pos=(2, 1))
  358. boxSizer.Add(item=gridSizer, proportion=1,
  359. flag=wx.ALL | wx.EXPAND, border=3)
  360. pageSizer.Add(item=boxSizer, proportion=0,
  361. flag=wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM,
  362. border=5)
  363. all = wx.Button(panel, id=wx.ID_ANY, label=_("Set to all"))
  364. all.SetToolTipString(_("Use draw settings for all loaded surfaces"))
  365. all.Bind(wx.EVT_BUTTON, self.OnSurfaceModeAll)
  366. gridSizer.Add(item=all, flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  367. pos=(2, 4))
  368. #
  369. # mask
  370. #
  371. box = wx.StaticBox (parent=panel, id=wx.ID_ANY,
  372. label=" %s " % (_("Mask")))
  373. boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  374. gridSizer = wx.GridBagSizer(vgap=5, hgap=5)
  375. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  376. label=_("Mask zeros:")),
  377. pos=(0, 0), flag=wx.ALIGN_CENTER_VERTICAL)
  378. elev = wx.CheckBox(parent=panel, id=wx.ID_ANY,
  379. label=_("by elevation"))
  380. elev.Enable(False) # TODO: not implemented yet
  381. gridSizer.Add(item=elev, pos=(0, 1))
  382. color = wx.CheckBox(parent=panel, id=wx.ID_ANY,
  383. label=_("by color"))
  384. color.Enable(False) # TODO: not implemented yet
  385. gridSizer.Add(item=color, pos=(0, 2))
  386. boxSizer.Add(item=gridSizer, proportion=1,
  387. flag=wx.ALL | wx.EXPAND, border=3)
  388. pageSizer.Add(item=boxSizer, proportion=0,
  389. flag=wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM,
  390. border=5)
  391. #
  392. # position
  393. #
  394. self.win['surface']['position'] = {}
  395. box = wx.StaticBox (parent=panel, id=wx.ID_ANY,
  396. label=" %s " % (_("Position")))
  397. boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  398. gridSizer = wx.GridBagSizer(vgap=5, hgap=5)
  399. # position
  400. axis = wx.Choice (parent=panel, id=wx.ID_ANY, size=(75, -1),
  401. choices = ["X",
  402. "Y",
  403. "Z"])
  404. axis.SetSelection(0)
  405. self.win['surface']['position']['axis'] = axis.GetId()
  406. axis.Bind(wx.EVT_CHOICE, self.OnSurfaceAxis)
  407. gridSizer.Add(item=axis, flag=wx.ALIGN_CENTER_VERTICAL,
  408. pos=(0, 0))
  409. value = wx.Slider(parent=panel, id=wx.ID_ANY,
  410. value=0,
  411. minValue=-1e4,
  412. maxValue=1e4,
  413. style=wx.SL_HORIZONTAL | wx.SL_AUTOTICKS | \
  414. wx.SL_TOP | wx.SL_LABELS,
  415. size=(350, -1))
  416. self.win['surface']['position']['pos'] = value.GetId()
  417. value.Bind(wx.EVT_SCROLL, self.OnSurfacePosition)
  418. gridSizer.Add(item=value, flag=wx.ALIGN_CENTER_VERTICAL,
  419. pos=(0, 1))
  420. boxSizer.Add(item=gridSizer, proportion=1,
  421. flag=wx.ALL | wx.EXPAND, border=3)
  422. pageSizer.Add(item=boxSizer, proportion=0,
  423. flag=wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM,
  424. border=5)
  425. panel.SetSizer(pageSizer)
  426. return panel.GetBestSize()
  427. def __createVectorPage(self):
  428. """Create view settings page"""
  429. panel = wx.Panel(parent=self.notebook, id=wx.ID_ANY)
  430. self.page['vector'] = {}
  431. self.page['vector']['id'] = -1
  432. self.page['vector']['panel'] = panel.GetId()
  433. pageSizer = wx.BoxSizer(wx.VERTICAL)
  434. self.win['vector'] = {}
  435. #
  436. # desc
  437. #
  438. desc = wx.StaticText(parent=panel, id=wx.ID_ANY,
  439. label="")
  440. self.win['vector']['desc'] = desc.GetId()
  441. pageSizer.Add(item=desc, proportion=0,
  442. flag=wx.EXPAND | wx.ALL,
  443. border=10)
  444. #
  445. # vector lines
  446. #
  447. self.win['vector']['lines'] = {}
  448. showLines = wx.CheckBox(parent=panel, id=wx.ID_ANY,
  449. label=_("Show vector lines"))
  450. self.win['vector']['lines']['show'] = showLines.GetId()
  451. showLines.Bind(wx.EVT_CHECKBOX, self.OnVectorShow)
  452. pageSizer.Add(item=showLines, proportion=0,
  453. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=5)
  454. box = wx.StaticBox (parent=panel, id=wx.ID_ANY,
  455. label=" %s " % (_("Vector lines")))
  456. boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  457. gridSizer = wx.GridBagSizer(vgap=5, hgap=5)
  458. # width
  459. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  460. label=_("Width:")),
  461. pos=(0, 0), flag=wx.ALIGN_CENTER_VERTICAL)
  462. width = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  463. initial=1,
  464. min=1,
  465. max=100)
  466. self.win['vector']['lines']['width'] = width.GetId()
  467. width.Bind(wx.EVT_SPINCTRL, self.OnVectorLines)
  468. gridSizer.Add(item=width, pos=(0, 1),
  469. flag=wx.ALIGN_CENTER_VERTICAL)
  470. gridSizer.AddGrowableCol(2)
  471. # color
  472. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  473. label=_("Color:")),
  474. pos=(0, 3), flag=wx.ALIGN_CENTER_VERTICAL)
  475. color = csel.ColourSelect(panel, id=wx.ID_ANY,
  476. colour=UserSettings.Get(group='nviz', key='vector',
  477. subkey=['lines', 'color']))
  478. self.win['vector']['lines']['color'] = color.GetId()
  479. color.Bind(csel.EVT_COLOURSELECT, self.OnVectorLines)
  480. gridSizer.Add(item=color, pos=(0, 4))
  481. gridSizer.AddGrowableCol(5)
  482. # display
  483. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  484. label=_("Display:")),
  485. pos=(0, 6), flag=wx.ALIGN_CENTER_VERTICAL)
  486. display = wx.Choice (parent=panel, id=wx.ID_ANY, size=(100, -1),
  487. choices = [_("on surface"),
  488. _("flat")])
  489. self.win['vector']['lines']['flat'] = display.GetId()
  490. display.Bind(wx.EVT_CHOICE, self.OnVectorDisplay)
  491. gridSizer.Add(item=display, flag=wx.ALIGN_CENTER_VERTICAL,
  492. pos=(0, 7))
  493. # height
  494. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  495. label=_("Height above surface:")),
  496. pos=(1, 0), flag=wx.ALIGN_CENTER_VERTICAL,
  497. span=(1, 2))
  498. surface = wx.ComboBox(parent=panel, id=wx.ID_ANY, size=(250, -1),
  499. style=wx.CB_SIMPLE | wx.CB_READONLY,
  500. choices=[])
  501. surface.Bind(wx.EVT_COMBOBOX, self.OnVectorSurface)
  502. self.win['vector']['lines']['surface'] = surface.GetId()
  503. gridSizer.Add(item=surface,
  504. pos=(1, 2), span=(1, 6),
  505. flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_RIGHT)
  506. self.CreateControl(panel, dict=self.win['vector']['lines'], name='height', size=300,
  507. range=(0, 1000),
  508. bind=(self.OnVectorHeight, self.OnVectorHeightFull, self.OnVectorHeightSpin))
  509. gridSizer.Add(item=self.FindWindowById(self.win['vector']['lines']['height']['slider']),
  510. pos=(2, 2), span=(1, 6))
  511. gridSizer.Add(item=self.FindWindowById(self.win['vector']['lines']['height']['spin']),
  512. pos=(3, 4),
  513. flag=wx.ALIGN_CENTER)
  514. boxSizer.Add(item=gridSizer, proportion=1,
  515. flag=wx.ALL | wx.EXPAND, border=3)
  516. pageSizer.Add(item=boxSizer, proportion=0,
  517. flag=wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM,
  518. border=5)
  519. #
  520. # vector points
  521. #
  522. self.win['vector']['points'] = {}
  523. showPoints = wx.CheckBox(parent=panel, id=wx.ID_ANY,
  524. label=_("Show vector points"))
  525. self.win['vector']['points']['show'] = showPoints.GetId()
  526. showPoints.Bind(wx.EVT_CHECKBOX, self.OnVectorShow)
  527. pageSizer.Add(item=showPoints, proportion=0,
  528. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=5)
  529. box = wx.StaticBox (parent=panel, id=wx.ID_ANY,
  530. label=" %s " % (_("Vector points")))
  531. boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  532. gridSizer = wx.GridBagSizer(vgap=5, hgap=5)
  533. # icon size
  534. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  535. label=_("Icon size:")),
  536. pos=(0, 0), flag=wx.ALIGN_CENTER_VERTICAL)
  537. isize = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  538. initial=1,
  539. min=1,
  540. max=1e6)
  541. isize.SetName('value')
  542. self.win['vector']['points']['size'] = isize.GetId()
  543. isize.Bind(wx.EVT_SPINCTRL, self.OnVectorPoints)
  544. isize.Bind(wx.EVT_TEXT, self.OnVectorPoints)
  545. gridSizer.Add(item=isize, pos=(0, 1),
  546. flag=wx.ALIGN_CENTER_VERTICAL)
  547. # icon width
  548. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  549. label=_("width:")),
  550. pos=(0, 2), flag=wx.ALIGN_CENTER_VERTICAL)
  551. iwidth = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  552. initial=1,
  553. min=1,
  554. max=1e6)
  555. iwidth.SetName('value')
  556. self.win['vector']['points']['width'] = iwidth.GetId()
  557. iwidth.Bind(wx.EVT_SPINCTRL, self.OnVectorPoints)
  558. iwidth.Bind(wx.EVT_TEXT, self.OnVectorPoints)
  559. gridSizer.Add(item=iwidth, pos=(0, 3),
  560. flag=wx.ALIGN_CENTER_VERTICAL)
  561. # icon symbol
  562. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  563. label=_("symbol:")),
  564. pos=(0, 4), flag=wx.ALIGN_CENTER_VERTICAL)
  565. isym = wx.Choice (parent=panel, id=wx.ID_ANY, size=(100, -1),
  566. choices=UserSettings.Get(group='nviz', key='vector',
  567. subkey=['points', 'marker'], internal=True))
  568. isym.SetName("selection")
  569. self.win['vector']['points']['marker'] = isym.GetId()
  570. isym.Bind(wx.EVT_CHOICE, self.OnVectorPoints)
  571. gridSizer.Add(item=isym, flag=wx.ALIGN_CENTER_VERTICAL,
  572. pos=(0, 5))
  573. # icon color
  574. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  575. label=_("color:")),
  576. pos=(0, 6), flag=wx.ALIGN_CENTER_VERTICAL)
  577. icolor = csel.ColourSelect(panel, id=wx.ID_ANY)
  578. icolor.SetName("color")
  579. self.win['vector']['points']['color'] = icolor.GetId()
  580. icolor.Bind(csel.EVT_COLOURSELECT, self.OnVectorPoints)
  581. gridSizer.Add(item=icolor, flag=wx.ALIGN_CENTER_VERTICAL,
  582. pos=(0, 7))
  583. # high
  584. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  585. label=_("Height above surface:")),
  586. pos=(1, 0), flag=wx.ALIGN_CENTER_VERTICAL,
  587. span=(1, 2))
  588. surface = wx.ComboBox(parent=panel, id=wx.ID_ANY, size=(250, -1),
  589. style=wx.CB_SIMPLE | wx.CB_READONLY,
  590. choices=[])
  591. surface.Bind(wx.EVT_COMBOBOX, self.OnVectorSurface)
  592. self.win['vector']['points']['surface'] = surface.GetId()
  593. gridSizer.Add(item=surface,
  594. pos=(1, 2), span=(1, 6),
  595. flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_RIGHT)
  596. self.CreateControl(panel, dict=self.win['vector']['points'], name='height', size=300,
  597. range=(0, 1000),
  598. bind=(self.OnVectorHeight, self.OnVectorHeightFull, self.OnVectorHeightSpin))
  599. gridSizer.Add(item=self.FindWindowById(self.win['vector']['points']['height']['slider']),
  600. pos=(2, 2), span=(1, 6))
  601. gridSizer.Add(item=self.FindWindowById(self.win['vector']['points']['height']['spin']),
  602. pos=(3, 4),
  603. flag=wx.ALIGN_CENTER)
  604. boxSizer.Add(item=gridSizer, proportion=1,
  605. flag=wx.ALL | wx.EXPAND, border=3)
  606. pageSizer.Add(item=boxSizer, proportion=0,
  607. flag=wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM,
  608. border=5)
  609. panel.SetSizer(pageSizer)
  610. return panel.GetBestSize()
  611. def __createVolumePage(self):
  612. """Create view settings page"""
  613. panel = wx.Panel(parent=self.notebook, id=wx.ID_ANY)
  614. self.page['volume'] = {}
  615. self.page['volume']['id'] = -1
  616. self.page['volume']['panel'] = panel.GetId()
  617. pageSizer = wx.BoxSizer(wx.VERTICAL)
  618. self.win['volume'] = {}
  619. #
  620. # draw
  621. #
  622. self.win['volume']['draw'] = {}
  623. box = wx.StaticBox (parent=panel, id=wx.ID_ANY,
  624. label=" %s " % (_("Draw")))
  625. boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  626. gridSizer = wx.GridBagSizer(vgap=5, hgap=5)
  627. gridSizer.AddGrowableCol(4)
  628. # mode
  629. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  630. label=_("Mode:")),
  631. pos=(0, 0), flag=wx.ALIGN_CENTER_VERTICAL)
  632. mode = wx.Choice (parent=panel, id=wx.ID_ANY, size=(150, -1),
  633. choices = [_("isosurfaces"),
  634. _("slides")])
  635. mode.SetSelection(0)
  636. mode.SetName("selection")
  637. # mode.Bind(wx.EVT_CHOICE, self.OnSurfaceMode)
  638. self.win['volume']['draw']['mode'] = mode.GetId()
  639. gridSizer.Add(item=mode, flag=wx.ALIGN_CENTER_VERTICAL,
  640. pos=(0, 1))
  641. # shading
  642. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  643. label=_("Shading:")),
  644. pos=(0, 2), flag=wx.ALIGN_CENTER_VERTICAL)
  645. shade = wx.Choice (parent=panel, id=wx.ID_ANY, size=(100, -1),
  646. choices = [_("flat"),
  647. _("gouraud")])
  648. shade.SetName("selection")
  649. self.win['volume']['draw']['shading'] = shade.GetId()
  650. shade.Bind(wx.EVT_CHOICE, self.OnVolumeIsosurfMode)
  651. gridSizer.Add(item=shade, flag=wx.ALIGN_CENTER_VERTICAL,
  652. pos=(0, 3))
  653. # resolution (mode)
  654. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  655. label=_("Resolution:")),
  656. pos=(0, 4), flag=wx.ALIGN_CENTER_VERTICAL)
  657. resol = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  658. initial=1,
  659. min=1,
  660. max=100)
  661. resol.SetName("value")
  662. self.win['volume']['draw']['resolution'] = resol.GetId()
  663. resol.Bind(wx.EVT_SPINCTRL, self.OnVolumeIsosurfResolution)
  664. resol.Bind(wx.EVT_TEXT, self.OnVolumeIsosurfResolution)
  665. gridSizer.Add(item=resol, pos=(0, 5))
  666. boxSizer.Add(item=gridSizer, proportion=1,
  667. flag=wx.ALL | wx.EXPAND, border=3)
  668. pageSizer.Add(item=boxSizer, proportion=0,
  669. flag=wx.EXPAND | wx.ALL,
  670. border=5)
  671. #
  672. # manage isosurfaces
  673. #
  674. box = wx.StaticBox (parent=panel, id=wx.ID_ANY,
  675. label=" %s " % (_("List of isosurfaces")))
  676. boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  677. gridSizer = wx.GridBagSizer(vgap=3, hgap=3)
  678. # list
  679. isolevel = wx.CheckListBox(parent=panel, id=wx.ID_ANY,
  680. size=(300, 150))
  681. self.Bind(wx.EVT_CHECKLISTBOX, self.OnVolumeIsosurfCheck, isolevel)
  682. self.Bind(wx.EVT_LISTBOX, self.OnVolumeIsosurfSelect, isolevel)
  683. self.win['volume']['isosurfs'] = isolevel.GetId()
  684. gridSizer.Add(item=isolevel, pos=(0, 0), span=(4, 1))
  685. # buttons (add, delete, move up, move down)
  686. btnAdd = wx.Button(parent=panel, id=wx.ID_ADD)
  687. self.win['volume']['btnIsosurfAdd'] = btnAdd.GetId()
  688. btnAdd.Bind(wx.EVT_BUTTON, self.OnVolumeIsosurfAdd)
  689. gridSizer.Add(item=btnAdd,
  690. pos=(0, 1))
  691. btnDelete = wx.Button(parent=panel, id=wx.ID_DELETE)
  692. self.win['volume']['btnIsosurfDelete'] = btnDelete.GetId()
  693. btnDelete.Bind(wx.EVT_BUTTON, self.OnVolumeIsosurfDelete)
  694. btnDelete.Enable(False)
  695. gridSizer.Add(item=btnDelete,
  696. pos=(1, 1))
  697. btnMoveUp = wx.Button(parent=panel, id=wx.ID_UP)
  698. self.win['volume']['btnIsosurfMoveUp'] = btnMoveUp.GetId()
  699. btnMoveUp.Bind(wx.EVT_BUTTON, self.OnVolumeIsosurfMoveUp)
  700. btnMoveUp.Enable(False)
  701. gridSizer.Add(item=btnMoveUp,
  702. pos=(2, 1))
  703. btnMoveDown = wx.Button(parent=panel, id=wx.ID_DOWN)
  704. self.win['volume']['btnIsosurfMoveDown'] = btnMoveDown.GetId()
  705. btnMoveDown.Bind(wx.EVT_BUTTON, self.OnVolumeIsosurfMoveDown)
  706. btnMoveDown.Enable(False)
  707. gridSizer.Add(item=btnMoveDown,
  708. pos=(3, 1))
  709. boxSizer.Add(item=gridSizer, proportion=1,
  710. flag=wx.ALL | wx.EXPAND, border=3)
  711. pageSizer.Add(item=boxSizer, proportion=0,
  712. flag=wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM,
  713. border=5)
  714. #
  715. # isosurface attributes
  716. #
  717. box = wx.StaticBox (parent=panel, id=wx.ID_ANY,
  718. label=" %s " % (_("Isosurface attributes")))
  719. boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  720. gridSizer = wx.GridBagSizer(vgap=3, hgap=3)
  721. self.win['volume']['attr'] = {}
  722. row = 0
  723. for code, attrb in (('topo', _("Topography level")),
  724. ('color', _("Color")),
  725. ('mask', _("Mask")),
  726. ('transp', _("Transparency")),
  727. ('shine', _("Shininess")),
  728. ('emit', _("Emission"))):
  729. self.win['volume'][code] = {}
  730. # label
  731. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  732. label=attrb + ':'),
  733. pos=(row, 0), flag=wx.ALIGN_CENTER_VERTICAL)
  734. if code != 'topo':
  735. use = wx.Choice (parent=panel, id=wx.ID_ANY, size=(100, -1),
  736. choices = [_("map")])
  737. else:
  738. use = None
  739. # check for required properties
  740. if code not in ('topo', 'color', 'shine'):
  741. use.Insert(item=_("unset"), pos=0)
  742. self.win['volume'][code]['required'] = False
  743. else:
  744. self.win['volume'][code]['required'] = True
  745. if use and code != 'mask':
  746. use.Append(item=_('constant'))
  747. if use:
  748. self.win['volume'][code]['use'] = use.GetId()
  749. use.Bind(wx.EVT_CHOICE, self.OnMapObjUse)
  750. gridSizer.Add(item=use, flag=wx.ALIGN_CENTER_VERTICAL,
  751. pos=(row, 1))
  752. if code != 'topo':
  753. map = gselect.Select(parent=panel, id=wx.ID_ANY,
  754. # size=globalvar.DIALOG_GSELECT_SIZE,
  755. size=(200, -1),
  756. type="grid3")
  757. self.win['volume'][code]['map'] = map.GetId() - 1 # FIXME
  758. map.Bind(wx.EVT_TEXT, self.OnVolumeIsosurfMap)
  759. gridSizer.Add(item=map, flag=wx.ALIGN_CENTER_VERTICAL,
  760. pos=(row, 2))
  761. else:
  762. map = None
  763. if code == 'color':
  764. value = csel.ColourSelect(panel, id=wx.ID_ANY,
  765. colour=UserSettings.Get(group='nviz', key='volume',
  766. subkey=['color', 'value']))
  767. value.Bind(csel.EVT_COLOURSELECT, self.OnVolumeIsosurfMap)
  768. elif code == 'mask':
  769. value = None
  770. else:
  771. if code == 'topo':
  772. size = (200, -1)
  773. else:
  774. size = (65, -1)
  775. value = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=size,
  776. initial=0)
  777. if code == 'topo':
  778. value.SetRange(minVal=-1e9, maxVal=1e9)
  779. elif code in ('shine', 'transp', 'emit'):
  780. value.SetRange(minVal=0, maxVal=255)
  781. else:
  782. value.SetRange(minVal=0, maxVal=100)
  783. value.Bind(wx.EVT_SPINCTRL, self.OnVolumeIsosurfMap)
  784. value.Bind(wx.EVT_TEXT, self.OnVolumeIsosurfMap)
  785. if value:
  786. self.win['volume'][code]['const'] = value.GetId()
  787. if code == 'topo':
  788. gridSizer.Add(item=value, flag=wx.ALIGN_CENTER_VERTICAL,
  789. pos=(row, 2))
  790. else:
  791. value.Enable(False)
  792. gridSizer.Add(item=value, flag=wx.ALIGN_CENTER_VERTICAL,
  793. pos=(row, 3))
  794. else:
  795. self.win['volume'][code]['const'] = None
  796. if code != 'topo':
  797. self.SetMapObjUseMap(nvizType='volume',
  798. attrb=code) # -> enable map / disable constant
  799. row += 1
  800. boxSizer.Add(item=gridSizer, proportion=1,
  801. flag=wx.ALL | wx.EXPAND, border=3)
  802. pageSizer.Add(item=boxSizer, proportion=0,
  803. flag=wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM,
  804. border=5)
  805. panel.SetSizer(pageSizer)
  806. return panel.GetBestSize()
  807. def __createSettingsPage(self):
  808. """Create settings page"""
  809. panel = wx.Panel(parent=self.notebook, id=wx.ID_ANY)
  810. self.notebook.AddPage(page=panel,
  811. text=" %s " % _("Settings"))
  812. pageSizer = wx.BoxSizer(wx.VERTICAL)
  813. self.win['settings'] = {}
  814. #
  815. # general
  816. #
  817. self.win['settings']['general'] = {}
  818. box = wx.StaticBox (parent=panel, id=wx.ID_ANY,
  819. label=" %s " % (_("General")))
  820. boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  821. gridSizer = wx.GridBagSizer(vgap=3, hgap=3)
  822. # background color
  823. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  824. label=_("Background color:")),
  825. pos=(0, 0), flag=wx.ALIGN_CENTER_VERTICAL)
  826. color = csel.ColourSelect(panel, id=wx.ID_ANY,
  827. colour=UserSettings.Get(group='nviz', key='settings',
  828. subkey=['general', 'bgcolor']))
  829. self.win['settings']['general']['bgcolor'] = color.GetId()
  830. color.Bind(csel.EVT_COLOURSELECT, self.OnBgColor)
  831. gridSizer.Add(item=color, pos=(0, 1))
  832. boxSizer.Add(item=gridSizer, proportion=1,
  833. flag=wx.ALL | wx.EXPAND, border=3)
  834. pageSizer.Add(item=boxSizer, proportion=0,
  835. flag=wx.EXPAND | wx.ALL,
  836. border=5)
  837. #
  838. # view
  839. #
  840. self.win['settings']['view'] = {}
  841. box = wx.StaticBox (parent=panel, id=wx.ID_ANY,
  842. label=" %s " % (_("View")))
  843. boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  844. gridSizer = wx.GridBagSizer(vgap=3, hgap=3)
  845. # perspective
  846. self.win['settings']['view']['persp'] = {}
  847. pvals = UserSettings.Get(group='nviz', key='view', subkey='persp')
  848. ipvals = UserSettings.Get(group='nviz', key='view', subkey='persp', internal=True)
  849. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  850. label=_("Perspective:")),
  851. pos=(0, 0), flag=wx.ALIGN_CENTER_VERTICAL)
  852. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  853. label=_("(value)")),
  854. pos=(0, 1), flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_RIGHT)
  855. pval = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  856. initial=pvals['value'],
  857. min=ipvals['min'],
  858. max=ipvals['max'])
  859. self.win['settings']['view']['persp']['value'] = pval.GetId()
  860. gridSizer.Add(item=pval, pos=(0, 2),
  861. flag=wx.ALIGN_CENTER_VERTICAL)
  862. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  863. label=_("(step)")),
  864. pos=(0, 3), flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_RIGHT)
  865. pstep = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  866. initial=pvals['step'],
  867. min=ipvals['min'],
  868. max=ipvals['max']-1)
  869. self.win['settings']['view']['persp']['step'] = pstep.GetId()
  870. gridSizer.Add(item=pstep, pos=(0, 4),
  871. flag=wx.ALIGN_CENTER_VERTICAL)
  872. # position
  873. self.win['settings']['view']['pos'] = {}
  874. posvals = UserSettings.Get(group='nviz', key='view', subkey='pos')
  875. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  876. label=_("Position:")),
  877. pos=(1, 0), flag=wx.ALIGN_CENTER_VERTICAL)
  878. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  879. label=_("(x)")),
  880. pos=(1, 1), flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_RIGHT)
  881. px = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  882. initial=posvals['x'] * 100,
  883. min=0,
  884. max=100)
  885. self.win['settings']['view']['pos']['x'] = px.GetId()
  886. gridSizer.Add(item=px, pos=(1, 2),
  887. flag=wx.ALIGN_CENTER_VERTICAL)
  888. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  889. label="(y)"),
  890. pos=(1, 3), flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_RIGHT)
  891. py = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  892. initial=posvals['y'] * 100,
  893. min=0,
  894. max=100)
  895. self.win['settings']['view']['pos']['y'] = py.GetId()
  896. gridSizer.Add(item=py, pos=(1, 4),
  897. flag=wx.ALIGN_CENTER_VERTICAL)
  898. # height
  899. self.win['settings']['view']['height'] = {}
  900. hvals = UserSettings.Get(group='nviz', key='view', subkey='height')
  901. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  902. label=_("Height")),
  903. pos=(2, 0), flag=wx.ALIGN_CENTER_VERTICAL)
  904. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  905. label=_("(step)")),
  906. pos=(2, 1), flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_RIGHT)
  907. hstep = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  908. initial=hvals['step'],
  909. min=1,
  910. max=1e6)
  911. self.win['settings']['view']['height']['step'] = hstep.GetId()
  912. gridSizer.Add(item=hstep, pos=(2, 2),
  913. flag=wx.ALIGN_CENTER_VERTICAL)
  914. # twist
  915. self.win['settings']['view']['twist'] = {}
  916. tvals = UserSettings.Get(group='nviz', key='view', subkey='twist')
  917. itvals = UserSettings.Get(group='nviz', key='view', subkey='twist', internal=True)
  918. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  919. label=_("Twist")),
  920. pos=(3, 0), flag=wx.ALIGN_CENTER_VERTICAL)
  921. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  922. label=_("(value)")),
  923. pos=(3, 1), flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_RIGHT)
  924. tval = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  925. initial=tvals['value'],
  926. min=itvals['min'],
  927. max=itvals['max'])
  928. self.win['settings']['view']['twist']['value'] = tval.GetId()
  929. gridSizer.Add(item=tval, pos=(3, 2),
  930. flag=wx.ALIGN_CENTER_VERTICAL)
  931. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  932. label=_("(step)")),
  933. pos=(3, 3), flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_RIGHT)
  934. tstep = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  935. initial=tvals['step'],
  936. min=itvals['min'],
  937. max=itvals['max']-1)
  938. self.win['settings']['view']['twist']['step'] = tstep.GetId()
  939. gridSizer.Add(item=tstep, pos=(3, 4),
  940. flag=wx.ALIGN_CENTER_VERTICAL)
  941. # z-exag
  942. self.win['settings']['view']['z-exag'] = {}
  943. zvals = UserSettings.Get(group='nviz', key='view', subkey='z-exag')
  944. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  945. label=_("Z-exag")),
  946. pos=(4, 0), flag=wx.ALIGN_CENTER_VERTICAL)
  947. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  948. label=_("(value)")),
  949. pos=(4, 1), flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_RIGHT)
  950. zval = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  951. initial=zvals['value'],
  952. min=-1e6,
  953. max=1e6)
  954. self.win['settings']['view']['z-exag']['value'] = zval.GetId()
  955. gridSizer.Add(item=zval, pos=(4, 2),
  956. flag=wx.ALIGN_CENTER_VERTICAL)
  957. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  958. label=_("(step):")),
  959. pos=(4, 3), flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_RIGHT)
  960. zstep = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  961. initial=zvals['step'],
  962. min=-1e6,
  963. max=1e6)
  964. self.win['settings']['view']['z-exag']['step'] = zstep.GetId()
  965. gridSizer.Add(item=zstep, pos=(4, 4),
  966. flag=wx.ALIGN_CENTER_VERTICAL)
  967. boxSizer.Add(item=gridSizer, proportion=1,
  968. flag=wx.ALL | wx.EXPAND, border=3)
  969. pageSizer.Add(item=boxSizer, proportion=0,
  970. flag=wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM,
  971. border=5)
  972. #
  973. # surface
  974. #
  975. self.win['settings']['surface'] = {}
  976. box = wx.StaticBox (parent=panel, id=wx.ID_ANY,
  977. label=" %s " % (_("Surface")))
  978. boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  979. gridSizer = wx.GridBagSizer(vgap=3, hgap=3)
  980. boxSizer.Add(item=gridSizer, proportion=1,
  981. flag=wx.ALL | wx.EXPAND, border=3)
  982. pageSizer.Add(item=boxSizer, proportion=0,
  983. flag=wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM,
  984. border=5)
  985. #
  986. # vector lines
  987. #
  988. self.win['settings']['vector'] = {}
  989. self.win['settings']['vector']['lines'] = {}
  990. box = wx.StaticBox (parent=panel, id=wx.ID_ANY,
  991. label=" %s " % (_("Vector lines")))
  992. boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  993. gridSizer = wx.GridBagSizer(vgap=3, hgap=3)
  994. # show
  995. row = 0
  996. showLines = wx.CheckBox(parent=panel, id=wx.ID_ANY,
  997. label=_("Show lines"))
  998. self.win['settings']['vector']['lines']['show'] = showLines.GetId()
  999. showLines.SetValue(UserSettings.Get(group='nviz', key='vector',
  1000. subkey=['lines', 'show']))
  1001. gridSizer.Add(item=showLines, pos=(row, 0))
  1002. boxSizer.Add(item=gridSizer, proportion=1,
  1003. flag=wx.ALL | wx.EXPAND, border=3)
  1004. pageSizer.Add(item=boxSizer, proportion=0,
  1005. flag=wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM,
  1006. border=5)
  1007. #
  1008. # vector points
  1009. #
  1010. self.win['settings']['vector']['points'] = {}
  1011. box = wx.StaticBox (parent=panel, id=wx.ID_ANY,
  1012. label=" %s " % (_("Vector points")))
  1013. boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  1014. gridSizer = wx.GridBagSizer(vgap=3, hgap=5)
  1015. # show
  1016. row = 0
  1017. showPoints = wx.CheckBox(parent=panel, id=wx.ID_ANY,
  1018. label=_("Show points"))
  1019. showPoints.SetValue(UserSettings.Get(group='nviz', key='vector',
  1020. subkey=['points', 'show']))
  1021. self.win['settings']['vector']['points']['show'] = showPoints.GetId()
  1022. gridSizer.Add(item=showPoints, pos=(row, 0), span=(1, 8))
  1023. # icon size
  1024. row += 1
  1025. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  1026. label=_("Size:")),
  1027. pos=(row, 0), flag=wx.ALIGN_CENTER_VERTICAL)
  1028. isize = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  1029. initial=100,
  1030. min=1,
  1031. max=1e6)
  1032. self.win['settings']['vector']['points']['size'] = isize.GetId()
  1033. isize.SetValue(UserSettings.Get(group='nviz', key='vector',
  1034. subkey=['points', 'size']))
  1035. gridSizer.Add(item=isize, pos=(row, 1),
  1036. flag=wx.ALIGN_CENTER_VERTICAL)
  1037. # icon width
  1038. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  1039. label=_("Width:")),
  1040. pos=(row, 2), flag=wx.ALIGN_CENTER_VERTICAL)
  1041. iwidth = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  1042. initial=2,
  1043. min=1,
  1044. max=1e6)
  1045. self.win['settings']['vector']['points']['width'] = isize.GetId()
  1046. iwidth.SetValue(UserSettings.Get(group='nviz', key='vector',
  1047. subkey=['points', 'width']))
  1048. gridSizer.Add(item=iwidth, pos=(row, 3),
  1049. flag=wx.ALIGN_CENTER_VERTICAL)
  1050. # icon symbol
  1051. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  1052. label=_("Marker:")),
  1053. pos=(row, 4), flag=wx.ALIGN_CENTER_VERTICAL)
  1054. isym = wx.Choice (parent=panel, id=wx.ID_ANY, size=(100, -1),
  1055. choices=UserSettings.Get(group='nviz', key='vector',
  1056. subkey=['points', 'marker'], internal=True))
  1057. isym.SetName("selection")
  1058. self.win['settings']['vector']['points']['marker'] = isym.GetId()
  1059. isym.SetSelection(UserSettings.Get(group='nviz', key='vector',
  1060. subkey=['points', 'marker']))
  1061. gridSizer.Add(item=isym, flag=wx.ALIGN_CENTER_VERTICAL,
  1062. pos=(row, 5))
  1063. # icon color
  1064. gridSizer.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  1065. label=_("Color:")),
  1066. pos=(row, 6), flag=wx.ALIGN_CENTER_VERTICAL)
  1067. icolor = csel.ColourSelect(panel, id=wx.ID_ANY)
  1068. icolor.SetName("color")
  1069. self.win['settings']['vector']['points']['color'] = icolor.GetId()
  1070. icolor.SetColour(UserSettings.Get(group='nviz', key='vector',
  1071. subkey=['points', 'color']))
  1072. gridSizer.Add(item=icolor, flag=wx.ALIGN_CENTER_VERTICAL,
  1073. pos=(row, 7))
  1074. boxSizer.Add(item=gridSizer, proportion=1,
  1075. flag=wx.ALL | wx.EXPAND, border=3)
  1076. pageSizer.Add(item=boxSizer, proportion=0,
  1077. flag=wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM,
  1078. border=5)
  1079. #
  1080. # buttons
  1081. #
  1082. btnDefault = wx.Button(panel, wx.ID_CANCEL, label=_("Default"))
  1083. btnSave = wx.Button(panel, wx.ID_SAVE)
  1084. btnApply = wx.Button(panel, wx.ID_APPLY)
  1085. btnDefault.Bind(wx.EVT_BUTTON, self.OnDefault)
  1086. btnDefault.SetToolTipString(_("Restore default settings"))
  1087. btnApply.Bind(wx.EVT_BUTTON, self.OnApply)
  1088. btnApply.SetToolTipString(_("Apply changes for the current session"))
  1089. btnSave.Bind(wx.EVT_BUTTON, self.OnSave)
  1090. btnSave.SetToolTipString(_("Apply and save changes to user settings file (default for next sessions)"))
  1091. btnSave.SetDefault()
  1092. btnSizer = wx.StdDialogButtonSizer()
  1093. btnSizer.AddButton(btnDefault)
  1094. btnSizer.AddButton(btnApply)
  1095. btnSizer.AddButton(btnSave)
  1096. btnSizer.Realize()
  1097. pageSizer.Add(item=btnSizer, proportion=1,
  1098. flag=wx.EXPAND | wx.ALL | wx.ALIGN_RIGHT | wx.ALIGN_BOTTOM,
  1099. border=5)
  1100. panel.SetSizer(pageSizer)
  1101. return panel.GetBestSize()
  1102. def CreateControl(self, parent, dict, name, range, bind, sliderHor=True, size=200):
  1103. """Add control (Slider + SpinCtrl)"""
  1104. dict[name] = {}
  1105. if sliderHor:
  1106. style = wx.SL_HORIZONTAL | wx.SL_AUTOTICKS | \
  1107. wx.SL_BOTTOM
  1108. sizeW = (size, -1)
  1109. else:
  1110. style = wx.SL_VERTICAL | wx.SL_AUTOTICKS | \
  1111. wx.SL_INVERSE
  1112. sizeW = (-1, size)
  1113. try:
  1114. val = self.mapWindow.view[name]['value']
  1115. except KeyError:
  1116. val=-1
  1117. slider = wx.Slider(parent=parent, id=wx.ID_ANY,
  1118. value=val,
  1119. minValue=range[0],
  1120. maxValue=range[1],
  1121. style=style,
  1122. size=sizeW)
  1123. slider.SetName('slider')
  1124. slider.Bind(wx.EVT_SCROLL, bind[0])
  1125. slider.Bind(wx.EVT_SCROLL_CHANGED, bind[1])
  1126. dict[name]['slider'] = slider.GetId()
  1127. spin = wx.SpinCtrl(parent=parent, id=wx.ID_ANY, size=(65, -1),
  1128. initial=val,
  1129. min=range[0],
  1130. max=range[1])
  1131. # spin = wx.SpinButton(parent=parent, id=wx.ID_ANY)
  1132. # spin.SetValue (self.mapWindow.view[name]['value'])
  1133. # spin.SetRange(self.mapWindow.view[name]['min'],
  1134. # self.mapWindow.view[name]['max'])
  1135. # no 'changed' event ... (FIXME)
  1136. spin.SetName('spin')
  1137. spin.Bind(wx.EVT_SPINCTRL, bind[2])
  1138. spin.Bind(wx.EVT_TEXT, bind[2])
  1139. dict[name]['spin'] = spin.GetId()
  1140. def UpdateSettings(self):
  1141. """Update dialog settings"""
  1142. for control in ('height',
  1143. 'persp',
  1144. 'twist',
  1145. 'z-exag'):
  1146. for win in self.win['view'][control].itervalues():
  1147. if control == 'height':
  1148. value = UserSettings.Get(group='nviz', key='view',
  1149. subkey=['height', 'value'], internal=True)
  1150. else:
  1151. value = self.mapWindow.view[control]['value']
  1152. self.FindWindowById(win).SetValue(value)
  1153. self.FindWindowById(self.win['view']['pos']).Draw()
  1154. self.FindWindowById(self.win['view']['pos']).Refresh(False)
  1155. self.Refresh(False)
  1156. def __GetWindowName(self, dict, id):
  1157. for name in dict.iterkeys():
  1158. if type(dict[name]) is type({}):
  1159. for win in dict[name].itervalues():
  1160. if win == id:
  1161. return name
  1162. else:
  1163. if dict[name] == id:
  1164. return name
  1165. return None
  1166. def OnViewChange(self, event):
  1167. """Change view, render in quick mode"""
  1168. # find control
  1169. winName = self.__GetWindowName(self.win['view'], event.GetId())
  1170. if not winName:
  1171. return
  1172. if winName == 'height':
  1173. view = self.mapWindow.iview # internal
  1174. else:
  1175. view = self.mapWindow.view
  1176. view[winName]['value'] = event.GetInt()
  1177. for win in self.win['view'][winName].itervalues():
  1178. self.FindWindowById(win).SetValue(view[winName]['value'])
  1179. if winName == 'z-exag':
  1180. zExag = True
  1181. else:
  1182. zExag = False
  1183. self.PostViewEvent(zExag)
  1184. self.mapWindow.render['quick'] = True
  1185. self.mapWindow.Refresh(False)
  1186. def OnViewChanged(self, event):
  1187. """View changed, render in full resolution"""
  1188. self.mapWindow.render['quick'] = False
  1189. self.mapWindow.Refresh(False)
  1190. def OnViewChangedSpin(self, event):
  1191. """View changed, render in full resolution"""
  1192. # TODO: use step value instead
  1193. self.OnViewChange(event)
  1194. self.OnViewChanged(None)
  1195. def OnResetView(self, event):
  1196. """Reset to default view (view page)"""
  1197. self.mapWindow.ResetView()
  1198. self.UpdateSettings()
  1199. self.mapWindow.Refresh(False)
  1200. def OnLookAt(self, event):
  1201. """Look at (view page)"""
  1202. sel = event.GetSelection()
  1203. if sel == 0: # top
  1204. self.mapWindow.view['pos']['x'] = 0.5
  1205. self.mapWindow.view['pos']['y'] = 0.5
  1206. elif sel == 1: # north
  1207. self.mapWindow.view['pos']['x'] = 0.5
  1208. self.mapWindow.view['pos']['y'] = 0.0
  1209. elif sel == 2: # south
  1210. self.mapWindow.view['pos']['x'] = 0.5
  1211. self.mapWindow.view['pos']['y'] = 1.0
  1212. elif sel == 3: # east
  1213. self.mapWindow.view['pos']['x'] = 1.0
  1214. self.mapWindow.view['pos']['y'] = 0.5
  1215. elif sel == 4: # west
  1216. self.mapWindow.view['pos']['x'] = 0.0
  1217. self.mapWindow.view['pos']['y'] = 0.5
  1218. elif sel == 5: # north-west
  1219. self.mapWindow.view['pos']['x'] = 0.0
  1220. self.mapWindow.view['pos']['y'] = 0.0
  1221. elif sel == 6: # north-east
  1222. self.mapWindow.view['pos']['x'] = 1.0
  1223. self.mapWindow.view['pos']['y'] = 0.0
  1224. elif sel == 7: # south-east
  1225. self.mapWindow.view['pos']['x'] = 1.0
  1226. self.mapWindow.view['pos']['y'] = 1.0
  1227. elif sel == 8: # south-west
  1228. self.mapWindow.view['pos']['x'] = 0.0
  1229. self.mapWindow.view['pos']['y'] = 1.0
  1230. self.PostViewEvent(zExag=True)
  1231. self.UpdateSettings()
  1232. self.mapWindow.Refresh(False)
  1233. def OnDefault(self, event):
  1234. """Restore default settings"""
  1235. settings = copy.deepcopy(UserSettings.GetDefaultSettings()['nviz'])
  1236. UserSettings.Set(group='nviz',
  1237. value=settings)
  1238. for subgroup, key in settings.iteritems(): # view, surface, vector...
  1239. if subgroup != 'view':
  1240. continue
  1241. for subkey, value in key.iteritems():
  1242. for subvalue in value.keys():
  1243. win = self.FindWindowById(self.win['settings'][subgroup][subkey][subvalue])
  1244. val = settings[subgroup][subkey][subvalue]
  1245. if subkey == 'pos':
  1246. val = int(val * 100)
  1247. win.SetValue(val)
  1248. event.Skip()
  1249. def OnApply(self, event):
  1250. """Apply button pressed"""
  1251. if self.notebook.GetSelection() == self.page['settings']['id']:
  1252. self.ApplySettings()
  1253. if event:
  1254. event.Skip()
  1255. def ApplySettings(self):
  1256. """Apply Nviz settings for current session"""
  1257. settings = UserSettings.Get(group='nviz')
  1258. for subgroup, key in settings.iteritems(): # view, surface, vector...
  1259. for subkey, value in key.iteritems():
  1260. for subvalue in value.keys():
  1261. try: # TODO
  1262. win = self.FindWindowById(self.win['settings'][subgroup][subkey][subvalue])
  1263. except:
  1264. # print 'e', subgroup, subkey, subvalue
  1265. continue
  1266. if win.GetName() == "selection":
  1267. value = win.GetSelection()
  1268. elif win.GetName() == "color":
  1269. value = tuple(win.GetColour())
  1270. else:
  1271. value = win.GetValue()
  1272. if subkey == 'pos':
  1273. value = float(value) / 100
  1274. settings[subgroup][subkey][subvalue] = value
  1275. def OnSave(self, event):
  1276. """OK button pressed
  1277. Apply changes, update map and save settings of selected layer
  1278. """
  1279. #
  1280. # apply changes
  1281. #
  1282. self.OnApply(None)
  1283. if self.notebook.GetSelection() == self.page['settings']['id']:
  1284. fileSettings = {}
  1285. UserSettings.ReadSettingsFile(settings=fileSettings)
  1286. fileSettings['nviz'] = UserSettings.Get(group='nviz')
  1287. file = UserSettings.SaveToFile(fileSettings)
  1288. self.lmgr.goutput.WriteLog(_('Nviz settings saved to file <%s>.') % file)
  1289. def OnBgColor(self, event):
  1290. """Background color changed"""
  1291. color = event.GetValue()
  1292. color = str(color[0]) + ':' + str(color[1]) + ':' + str(color[2])
  1293. self.mapWindow.nvizClass.SetBgColor(str(color))
  1294. if self.parent.autoRender.IsChecked():
  1295. self.mapWindow.Refresh(False)
  1296. def OnClose(self, event):
  1297. """Close button pressed
  1298. Close dialog
  1299. """
  1300. self.Hide()
  1301. def OnMapObjUse(self, event):
  1302. """Set surface attribute -- use -- map/constant"""
  1303. if not self.mapWindow.init:
  1304. return
  1305. wx.Yield()
  1306. # find attribute row
  1307. attrb = self.__GetWindowName(self.win['surface'], event.GetId())
  1308. if not attrb:
  1309. attrb = self.__GetWindowName(self.win['volume'], event.GetId())
  1310. nvizType = 'volume'
  1311. else:
  1312. nvizType = 'surface'
  1313. selection = event.GetSelection()
  1314. if self.win[nvizType][attrb]['required']: # no 'unset'
  1315. selection += 1
  1316. if selection == 0: # unset
  1317. useMap = None
  1318. value = ''
  1319. elif selection == 1: # map
  1320. useMap = True
  1321. value = self.FindWindowById(self.win[nvizType][attrb]['map']).GetValue()
  1322. elif selection == 2: # constant
  1323. useMap = False
  1324. if attrb == 'color':
  1325. value = self.FindWindowById(self.win[nvizType][attrb]['const']).GetColour()
  1326. value = self._getColorString(value)
  1327. else:
  1328. value = self.FindWindowById(self.win[nvizType][attrb]['const']).GetValue()
  1329. self.SetMapObjUseMap(nvizType=nvizType,
  1330. attrb=attrb, map=useMap)
  1331. data = self.mapWindow.GetSelectedLayer(type='nviz')
  1332. if nvizType == 'surface':
  1333. data[nvizType]['attribute'][attrb] = { 'map' : useMap,
  1334. 'value' : str(value),
  1335. 'update' : None, }
  1336. else: # volume / isosurface
  1337. list = self.FindWindowById(self.win['volume']['isosurfs'])
  1338. id = list.GetSelection()
  1339. data[nvizType]['isosurface'][id][attrb] = { 'map' : useMap,
  1340. 'value' : str(value),
  1341. 'update' : None, }
  1342. # update properties
  1343. event = wxUpdateProperties(data=data)
  1344. wx.PostEvent(self.mapWindow, event)
  1345. if self.parent.autoRender.IsChecked():
  1346. self.mapWindow.Refresh(False)
  1347. def SetMapObjUseMap(self, nvizType, attrb, map=None):
  1348. """Update dialog widgets when attribute type changed"""
  1349. if attrb in ('topo', 'color', 'shine'):
  1350. incSel = -1 # decrement selection (no 'unset')
  1351. else:
  1352. incSel = 0
  1353. if map is True: # map
  1354. if attrb != 'topo': # changing map topography not allowed
  1355. # not sure why, but here must be disabled both ids, should be fixed!
  1356. self.FindWindowById(self.win[nvizType][attrb]['map'] + 1).Enable(True)
  1357. if self.win[nvizType][attrb]['const']:
  1358. self.FindWindowById(self.win[nvizType][attrb]['const']).Enable(False)
  1359. self.FindWindowById(self.win[nvizType][attrb]['use']).SetSelection(1 + incSel)
  1360. elif map is False: # const
  1361. self.FindWindowById(self.win[nvizType][attrb]['map'] + 1).Enable(False)
  1362. if self.win[nvizType][attrb]['const']:
  1363. self.FindWindowById(self.win[nvizType][attrb]['const']).Enable(True)
  1364. self.FindWindowById(self.win[nvizType][attrb]['use']).SetSelection(2 + incSel)
  1365. else: # unset
  1366. self.FindWindowById(self.win[nvizType][attrb]['map'] + 1).Enable(False)
  1367. if self.win[nvizType][attrb]['const']:
  1368. self.FindWindowById(self.win[nvizType][attrb]['const']).Enable(False)
  1369. self.FindWindowById(self.win[nvizType][attrb]['use']).SetSelection(0)
  1370. def OnSurfaceMap(self, event):
  1371. """Set surface attribute"""
  1372. self.SetMapObjAttrb(nvizType='surface', winId=event.GetId())
  1373. def SetMapObjAttrb(self, nvizType, winId):
  1374. """Set map object (surface/isosurface) attribute (map/constant)"""
  1375. if not self.mapWindow.init:
  1376. return
  1377. attrb = self.__GetWindowName(self.win[nvizType], winId)
  1378. if not attrb:
  1379. return
  1380. if nvizType == 'volume' and attrb == 'topo':
  1381. return
  1382. selection = self.FindWindowById(self.win[nvizType][attrb]['use']).GetSelection()
  1383. if self.win[nvizType][attrb]['required']:
  1384. selection += 1
  1385. if selection == 0: # unset
  1386. useMap = None
  1387. value = ''
  1388. elif selection == 1: # map
  1389. value = self.FindWindowById(self.win[nvizType][attrb]['map']).GetValue()
  1390. useMap = True
  1391. else: # constant
  1392. if attrb == 'color':
  1393. value = self.FindWindowById(self.win[nvizType][attrb]['const']).GetColour()
  1394. # tuple to string
  1395. value = self._getColorString(value)
  1396. else:
  1397. value = self.FindWindowById(self.win[nvizType][attrb]['const']).GetValue()
  1398. useMap = False
  1399. if not self.pageChanging:
  1400. data = self.mapWindow.GetSelectedLayer(type='nviz')
  1401. if nvizType == 'surface':
  1402. data[nvizType]['attribute'][attrb] = { 'map' : useMap,
  1403. 'value' : str(value),
  1404. 'update' : None, }
  1405. else: # volume / isosurface
  1406. list = self.FindWindowById(self.win['volume']['isosurfs'])
  1407. id = list.GetSelection()
  1408. data[nvizType]['isosurface'][id][attrb] = { 'map' : useMap,
  1409. 'value' : str(value),
  1410. 'update' : None, }
  1411. # update properties
  1412. event = wxUpdateProperties(data=data)
  1413. wx.PostEvent(self.mapWindow, event)
  1414. if self.parent.autoRender.IsChecked():
  1415. self.mapWindow.Refresh(False)
  1416. def OnSurfaceResolution(self, event):
  1417. """Draw resolution changed"""
  1418. self.SetSurfaceResolution()
  1419. if apply and self.parent.autoRender.IsChecked():
  1420. self.mapWindow.Refresh(False)
  1421. def SetSurfaceResolution(self):
  1422. """Set draw resolution"""
  1423. coarse = self.FindWindowById(self.win['surface']['draw']['res-coarse']).GetValue()
  1424. fine = self.FindWindowById(self.win['surface']['draw']['res-fine']).GetValue()
  1425. data = self.mapWindow.GetSelectedLayer(type='nviz')
  1426. data['surface']['draw']['resolution'] = { 'coarse' : coarse,
  1427. 'fine' : fine,
  1428. 'update' : None, }
  1429. # update properties
  1430. event = wxUpdateProperties(data=data)
  1431. wx.PostEvent(self.mapWindow, event)
  1432. def SetSurfaceMode(self):
  1433. """Set draw mode
  1434. @param apply allow auto-rendering
  1435. """
  1436. mode = self.FindWindowById(self.win['surface']['draw']['mode']).GetSelection()
  1437. if mode == 0: # coarse
  1438. self.FindWindowById(self.win['surface']['draw']['res-coarse']).Enable(True)
  1439. self.FindWindowById(self.win['surface']['draw']['res-fine']).Enable(False)
  1440. elif mode == 1: # fine
  1441. self.FindWindowById(self.win['surface']['draw']['res-coarse']).Enable(False)
  1442. self.FindWindowById(self.win['surface']['draw']['res-fine']).Enable(True)
  1443. else: # both
  1444. self.FindWindowById(self.win['surface']['draw']['res-coarse']).Enable(True)
  1445. self.FindWindowById(self.win['surface']['draw']['res-fine']).Enable(True)
  1446. style = self.FindWindowById(self.win['surface']['draw']['style']).GetSelection()
  1447. shade = self.FindWindowById(self.win['surface']['draw']['shading']).GetSelection()
  1448. value, desc = self.mapWindow.GetDrawMode(mode, style, shade)
  1449. return value, desc
  1450. def OnSurfaceMode(self, event):
  1451. """Set draw mode"""
  1452. value, desc = self.SetSurfaceMode()
  1453. data = self.mapWindow.GetSelectedLayer(type='nviz')
  1454. data['surface']['draw']['mode'] = { 'value' : value,
  1455. 'desc' : desc,
  1456. 'update' : None, }
  1457. # update properties
  1458. event = wxUpdateProperties(data=data)
  1459. wx.PostEvent(self.mapWindow, event)
  1460. if apply and self.parent.autoRender.IsChecked():
  1461. self.mapWindow.Refresh(False)
  1462. def OnSurfaceModeAll(self, event):
  1463. """Set draw mode (including wire color) for all loaded surfaces"""
  1464. value, desc = self.SetSurfaceMode()
  1465. coarse = self.FindWindowById(self.win['surface']['draw']['res-coarse']).GetValue()
  1466. fine = self.FindWindowById(self.win['surface']['draw']['res-fine']).GetValue()
  1467. color = self.FindWindowById(self.win['surface']['draw']['wire-color']).GetColour()
  1468. cvalue = self._getColorString(color)
  1469. for name in self.mapWindow.GetLayerNames(type='raster'):
  1470. data = self.mapWindow.GetLayerData(type='raster', name=name)
  1471. if not data:
  1472. continue # shouldy no happen
  1473. data['surface']['draw']['mode'] = { 'value' : value,
  1474. 'desc' : desc,
  1475. 'update' : None, }
  1476. data['surface']['draw']['resolution'] = { 'coarse' : coarse,
  1477. 'fine' : fine,
  1478. 'update' : None, }
  1479. data['surface']['draw']['wire-color'] = { 'value' : cvalue,
  1480. 'update' : None }
  1481. # update properties
  1482. event = wxUpdateProperties(data=data)
  1483. wx.PostEvent(self.mapWindow, event)
  1484. if apply and self.parent.autoRender.IsChecked():
  1485. self.mapWindow.Refresh(False)
  1486. def _getColorString(self, color):
  1487. """Set wire color"""
  1488. return str(color[0]) + ':' + str(color[1]) + ':' + str(color[2])
  1489. def OnSurfaceWireColor(self, event):
  1490. """Set wire color"""
  1491. data = self.mapWindow.GetSelectedLayer(type='nviz')
  1492. value = self._getColorString(event.GetValue())
  1493. data['surface']['draw']['wire-color'] = { 'value' : value,
  1494. 'update' : None }
  1495. # update properties
  1496. event = wxUpdateProperties(data=data)
  1497. wx.PostEvent(self.mapWindow, event)
  1498. if self.parent.autoRender.IsChecked():
  1499. self.mapWindow.Refresh(False)
  1500. def OnSurfaceAxis(self, event):
  1501. """Surface position, axis changed"""
  1502. mapLayer = self.mapWindow.GetSelectedLayer()
  1503. data = self.mapWindow.GetSelectedLayer(type='nviz')
  1504. id = data['object']['id']
  1505. axis = self.FindWindowById(self.win['surface']['position']['axis']).GetSelection()
  1506. win = self.FindWindowById(self.win['surface']['position']['pos'])
  1507. x, y, z = self.mapWindow.nvizClass.GetSurfacePosition(id)
  1508. if axis == 0: # x
  1509. win.SetRange(-1e4, 1e4)
  1510. win.SetValue(x)
  1511. elif axis == 1: # y
  1512. win.SetRange(-1e4, 1e4)
  1513. win.SetValue(y)
  1514. else: # z
  1515. win.SetRange(-1e3, 1e3)
  1516. win.SetValue(z)
  1517. def OnSurfacePosition(self, event):
  1518. """Surface position"""
  1519. axis = self.FindWindowById(self.win['surface']['position']['axis']).GetSelection()
  1520. value = event.GetInt()
  1521. mapLayer = self.mapWindow.GetSelectedLayer()
  1522. data = self.mapWindow.GetSelectedLayer(type='nviz')
  1523. id = data['surface']['object']['id']
  1524. x, y, z = self.mapWindow.nvizClass.GetSurfacePosition(id)
  1525. if axis == 0: # x
  1526. x = value
  1527. elif axis == 1: # y
  1528. y = value
  1529. else: # z
  1530. z = value
  1531. data = self.mapWindow.GetSelectedLayer(type='nviz')
  1532. data['surface']['position']['x'] = x
  1533. data['surface']['position']['y'] = y
  1534. data['surface']['position']['z'] = z
  1535. data['surface']['position']['update'] = None
  1536. # update properties
  1537. event = wxUpdateProperties(data=data)
  1538. wx.PostEvent(self.mapWindow, event)
  1539. if self.parent.autoRender.IsChecked():
  1540. self.mapWindow.Refresh(False)
  1541. def UpdateVectorShow(self, vecType, enabled):
  1542. """Enable/disable lines/points widgets
  1543. @param vecType vector type (lines, points)
  1544. """
  1545. if vecType != 'lines' and vecType != 'points':
  1546. return False
  1547. for win in self.win['vector'][vecType].keys():
  1548. if win == 'show':
  1549. continue
  1550. if type(self.win['vector'][vecType][win]) == type({}):
  1551. for swin in self.win['vector'][vecType][win].keys():
  1552. if enabled:
  1553. self.FindWindowById(self.win['vector'][vecType][win][swin]).Enable(True)
  1554. else:
  1555. self.FindWindowById(self.win['vector'][vecType][win][swin]).Enable(False)
  1556. else:
  1557. if enabled:
  1558. self.FindWindowById(self.win['vector'][vecType][win]).Enable(True)
  1559. else:
  1560. self.FindWindowById(self.win['vector'][vecType][win]).Enable(False)
  1561. return True
  1562. def OnVectorShow(self, event):
  1563. """Show vector lines/points"""
  1564. winId = event.GetId()
  1565. if winId == self.win['vector']['lines']['show']:
  1566. vecType = 'lines'
  1567. else: # points
  1568. vecType = 'points'
  1569. checked = event.IsChecked()
  1570. item = self.mapWindow.GetSelectedLayer(type='item')
  1571. data = self.mapWindow.GetSelectedLayer(type='nviz')['vector']
  1572. if checked:
  1573. self.mapWindow.LoadVector(item, (vecType,))
  1574. else:
  1575. self.mapWindow.UnloadVector(item, (vecType,))
  1576. self.UpdateVectorShow(vecType, checked)
  1577. if checked:
  1578. try:
  1579. id = data[vecType]['object']['id']
  1580. except KeyError:
  1581. id = -1
  1582. if id > 0:
  1583. self.mapWindow.SetMapObjProperties(item, id, vecType)
  1584. # update properties
  1585. event = wxUpdateProperties(data=data)
  1586. wx.PostEvent(self.mapWindow, event)
  1587. if self.parent.autoRender.IsChecked():
  1588. self.mapWindow.Refresh(False)
  1589. event.Skip()
  1590. def OnVectorDisplay(self, event):
  1591. """Display vector lines on surface/flat"""
  1592. if event.GetSelection() == 0: # surface
  1593. if len(self.mapWindow.layers['raster']['name']) < 1:
  1594. event.Veto()
  1595. return
  1596. self.FindWindowById(self.win['vector']['lines']['surface']).Enable(True)
  1597. # set first found surface
  1598. data = self.mapWindow.GetSelectedLayer(type='nviz')
  1599. data['vector']['lines']['mode']['surface'] = self.mapWindow.layers['raster']['name'][0]
  1600. self.FindWindowById(self.win['vector']['lines']['surface']).SetStringSelection( \
  1601. self.mapWindow.layers['raster']['name'][0])
  1602. else: # flat
  1603. self.FindWindowById(self.win['vector']['lines']['surface']).Enable(False)
  1604. self.OnVectorLines(event)
  1605. event.Skip()
  1606. def OnVectorLines(self, event):
  1607. """Set vector lines mode, apply changes if auto-rendering is enabled"""
  1608. width = self.FindWindowById(self.win['vector']['lines']['width']).GetValue()
  1609. color = self.FindWindowById(self.win['vector']['lines']['color']).GetColour()
  1610. color = str(color[0]) + ':' + str(color[1]) + ':' + str(color[2])
  1611. mode = {}
  1612. if self.FindWindowById(self.win['vector']['lines']['flat']).GetSelection() == 0:
  1613. mode['type'] = 'surface'
  1614. mode['surface'] = self.FindWindowById(self.win['vector']['lines']['surface']).GetValue()
  1615. mode['update'] = None
  1616. else:
  1617. mode['type'] = 'flat'
  1618. data = self.mapWindow.GetSelectedLayer(type='nviz')
  1619. for attrb in ('width', 'color', 'mode'):
  1620. data['vector']['lines'][attrb]['update'] = None
  1621. data['vector']['lines']['width']['value'] = width
  1622. data['vector']['lines']['color']['value'] = color
  1623. data['vector']['lines']['mode']['value'] = mode
  1624. # update properties
  1625. event = wxUpdateProperties(data=data)
  1626. wx.PostEvent(self.mapWindow, event)
  1627. if self.parent.autoRender.IsChecked():
  1628. self.mapWindow.Refresh(False)
  1629. def OnVectorHeight(self, event):
  1630. value = event.GetInt()
  1631. id = event.GetId()
  1632. if id == self.win['vector']['lines']['height']['spin'] or \
  1633. id == self.win['vector']['lines']['height']['slider']:
  1634. vtype = 'lines'
  1635. else:
  1636. vtype = 'points'
  1637. if type(event) == type(wx.ScrollEvent()):
  1638. # slider
  1639. win = self.FindWindowById(self.win['vector'][vtype]['height']['spin'])
  1640. else:
  1641. # spin
  1642. win = self.FindWindowById(self.win['vector'][vtype]['height']['slider'])
  1643. win.SetValue(value)
  1644. data = self.mapWindow.GetSelectedLayer(type='nviz')['vector'][vtype]
  1645. data['height'] = { 'value' : value,
  1646. 'update' : None }
  1647. # update properties
  1648. event = wxUpdateProperties(data=data)
  1649. wx.PostEvent(self.mapWindow, event)
  1650. self.mapWindow.render['quick'] = True
  1651. self.mapWindow.render['v' + vtype] = True
  1652. self.mapWindow.Refresh(False)
  1653. def OnVectorHeightFull(self, event):
  1654. """Vector height changed, render in full resolution"""
  1655. id = event.GetId()
  1656. if id == self.win['vector']['lines']['height']['spin'] or \
  1657. id == self.win['vector']['lines']['height']['slider']:
  1658. vtype = 'lines'
  1659. else:
  1660. vtype = 'points'
  1661. self.mapWindow.render['quick'] = False
  1662. self.mapWindow.render['v' + vtype] = False
  1663. self.mapWindow.Refresh(False)
  1664. def OnVectorHeightSpin(self, event):
  1665. """Vector height changed, render in full resolution"""
  1666. # TODO: use step value instead
  1667. self.OnVectorHeight(event)
  1668. self.OnVectorHeightFull(event)
  1669. def OnVectorSurface(self, event):
  1670. """Reference surface for vector map (lines/points)"""
  1671. id = event.GetId()
  1672. if id == self.win['vector']['lines']['surface']:
  1673. vtype = 'lines'
  1674. else:
  1675. vtype = 'points'
  1676. data['vector'][vtype]['mode']['surface'] = { 'value' : event.GetValue(),
  1677. 'update' : None }
  1678. # update properties
  1679. event = wxUpdateProperties(data=data)
  1680. wx.PostEvent(self.mapWindow, event)
  1681. if self.parent.autoRender.IsChecked():
  1682. self.mapWindow.Refresh(False)
  1683. def OnVectorPoints(self, event):
  1684. """Set vector points mode, apply changes if auto-rendering is enabled"""
  1685. size = self.FindWindowById(self.win['vector']['points']['size']).GetValue()
  1686. width = self.FindWindowById(self.win['vector']['points']['width']).GetValue()
  1687. color = self.FindWindowById(self.win['vector']['points']['color']).GetColour()
  1688. color = str(color[0]) + ':' + str(color[1]) + ':' + str(color[2])
  1689. marker = self.FindWindowById(self.win['vector']['points']['marker']).GetSelection()
  1690. data = self.mapWindow.GetSelectedLayer(type='nviz')
  1691. for attrb in ('size', 'width', 'color', 'marker'):
  1692. data['vector']['points'][attrb]['update'] = None
  1693. data['vector']['points']['size']['value'] = size
  1694. data['vector']['points']['width']['value'] = width
  1695. data['vector']['points']['color']['value'] = color
  1696. data['vector']['points']['marker']['value'] = marker
  1697. # update properties
  1698. event = wxUpdateProperties(data=data)
  1699. wx.PostEvent(self.mapWindow, event)
  1700. if self.parent.autoRender.IsChecked():
  1701. self.mapWindow.Refresh(False)
  1702. def UpdateIsosurfButtons(self, list):
  1703. """Enable/disable buttons 'add', 'delete',
  1704. 'move up', 'move down'"""
  1705. nitems = list.GetCount()
  1706. add = self.parent.FindWindowById(self.win['volume']['btnIsosurfAdd'])
  1707. delete = self.parent.FindWindowById(self.win['volume']['btnIsosurfDelete'])
  1708. moveDown = self.parent.FindWindowById(self.win['volume']['btnIsosurfMoveDown'])
  1709. moveUp = self.parent.FindWindowById(self.win['volume']['btnIsosurfMoveUp'])
  1710. if nitems >= wxnviz.MAX_ISOSURFS:
  1711. # disable add button on max
  1712. add.Enable(False)
  1713. else:
  1714. add.Enable(True)
  1715. if nitems < 1:
  1716. # disable 'delete' if only one item in the lis
  1717. delete.Enable(False)
  1718. else:
  1719. delete.Enable(True)
  1720. if list.GetSelection() >= nitems - 1:
  1721. # disable 'move-down' if last
  1722. moveDown.Enable(False)
  1723. else:
  1724. moveDown.Enable(True)
  1725. if list.GetSelection() < 1:
  1726. # disable 'move-up' if first
  1727. moveUp.Enable(False)
  1728. else:
  1729. moveUp.Enable(True)
  1730. def OnVolumeIsosurfMode(self, event):
  1731. """Set isosurface draw mode"""
  1732. self.SetIsosurfaceMode(event.GetSelection())
  1733. def SetIsosurfaceMode(self, selection):
  1734. """Set isosurface draw mode"""
  1735. layer = self.mapWindow.GetSelectedLayer()
  1736. data = self.mapWindow.GetSelectedLayer(type='nviz')['volume']
  1737. id = data['object']['id']
  1738. mode = 0
  1739. if selection == 0:
  1740. mode |= wxnviz.DM_FLAT
  1741. else:
  1742. mode |= wxnviz.DM_GOURAUD
  1743. self.mapWindow.nvizClass.SetIsosurfaceMode(id, mode)
  1744. if self.parent.autoRender.IsChecked():
  1745. self.mapWindow.Refresh(False)
  1746. def OnVolumeIsosurfResolution(self, event):
  1747. """Set isosurface draw resolution"""
  1748. self.SetIsosurfaceResolution(event.GetInt())
  1749. def SetIsosurfaceResolution(self, res):
  1750. """Set isosurface draw resolution"""
  1751. layer = self.mapWindow.GetSelectedLayer()
  1752. data = self.mapWindow.GetSelectedLayer(type='nviz')['volume']
  1753. id = data['object']['id']
  1754. self.mapWindow.nvizClass.SetIsosurfaceRes(id, res)
  1755. if self.parent.autoRender.IsChecked():
  1756. self.mapWindow.Refresh(False)
  1757. def OnVolumeIsosurfMap(self, event):
  1758. """Set surface attribute"""
  1759. self.SetMapObjAttrb(nvizType='volume', winId=event.GetId())
  1760. def OnVolumeIsosurfCheck(self, event):
  1761. """Isosurface checked (->load) or unchecked (->unload)"""
  1762. index = event.GetSelection()
  1763. list = self.FindWindowById(self.win['volume']['isosurfs'])
  1764. data = self.mapWindow.GetSelectedLayer(type='nviz')['volume']
  1765. id = data['object']['id']
  1766. isosurfId = event.GetSelection()
  1767. if list.IsChecked(index):
  1768. self.mapWindow.nvizClass.SetIsosurfaceTransp(id, isosurfId, False, "0")
  1769. else:
  1770. # disable -> make transparent
  1771. self.mapWindow.nvizClass.SetIsosurfaceTransp(id, isosurfId, False, "255")
  1772. if self.parent.autoRender.IsChecked():
  1773. self.mapWindow.Refresh(False)
  1774. def OnVolumeIsosurfSelect(self, event):
  1775. """Isosurface item selected"""
  1776. winUp = self.FindWindowById(self.win['volume']['btnIsosurfMoveUp'])
  1777. winDown = self.FindWindowById(self.win['volume']['btnIsosurfMoveDown'])
  1778. selection = event.GetSelection()
  1779. if selection == 0:
  1780. winUp.Enable(False)
  1781. if not winDown.IsEnabled():
  1782. winDown.Enable()
  1783. elif selection == self.FindWindowById(event.GetId()).GetCount() - 1:
  1784. winDown.Enable(False)
  1785. if not winUp.IsEnabled():
  1786. winUp.Enable()
  1787. else:
  1788. if not winDown.IsEnabled():
  1789. winDown.Enable()
  1790. if not winUp.IsEnabled():
  1791. winUp.Enable()
  1792. # update dialog
  1793. layer = self.mapWindow.GetSelectedLayer()
  1794. data = self.mapWindow.GetSelectedLayer(type='nviz')['volume']['isosurface'][selection]
  1795. self.UpdateVolumeIsosurfPage(layer, data)
  1796. def OnVolumeIsosurfAdd(self, event):
  1797. """Add new isosurface to the list"""
  1798. list = self.FindWindowById(self.win['volume']['isosurfs'])
  1799. level = self.FindWindowById(self.win['volume']['topo']['const']).GetValue()
  1800. sel = list.GetSelection()
  1801. if sel < 0 or sel >= list.GetCount() - 1:
  1802. item = list.Append(item="%s %s" % (_("Level"), str(level)))
  1803. else:
  1804. list.Insert(item="%s %s" % (_("Level"), str(level)),
  1805. pos=sel+1) # append
  1806. item = sel + 1
  1807. list.Check(item)
  1808. list.SetSelection(item)
  1809. layer = self.mapWindow.GetSelectedLayer()
  1810. data = self.mapWindow.GetSelectedLayer(type='nviz')['volume']
  1811. id = data['object']['id']
  1812. # collect properties
  1813. isosurfData = {}
  1814. for attrb in ('topo', 'color', 'mask',
  1815. 'transp', 'shine', 'emit'):
  1816. if attrb == 'topo':
  1817. isosurfData[attrb] = {}
  1818. win = self.FindWindowById(self.win['volume'][attrb]['const'])
  1819. isosurfData[attrb]['value'] = win.GetValue()
  1820. else:
  1821. uwin = self.FindWindowById(self.win['volume'][attrb]['use'])
  1822. sel = uwin.GetSelection()
  1823. if self.win['volume'][attrb]['required']:
  1824. sel += 1
  1825. if sel == 0: # unset
  1826. continue
  1827. isosurfData[attrb] = {}
  1828. if sel == 1: # map
  1829. isosurfData[attrb]['map'] = True
  1830. vwin = self.FindWindowById(self.win['volume'][attrb]['map'])
  1831. value = vwin.GetValue()
  1832. else: # const
  1833. isosurfData[attrb]['map'] = False
  1834. vwin = self.FindWindowById(self.win['volume'][attrb]['const'])
  1835. if vwin.GetName() == "color":
  1836. value = self._getColorString(vwin.GetValue())
  1837. else:
  1838. value = vwin.GetValue()
  1839. isosurfData[attrb]['value'] = value
  1840. data['isosurface'].insert(item, isosurfData)
  1841. # add isosurface
  1842. self.mapWindow.nvizClass.AddIsosurface(id, level)
  1843. # use by default 3d raster map for color
  1844. self.mapWindow.nvizClass.SetIsosurfaceColor(id, item, True, str(layer.name))
  1845. # update buttons
  1846. self.UpdateIsosurfButtons(list)
  1847. if self.parent.autoRender.IsChecked():
  1848. self.mapWindow.Refresh(False)
  1849. event.Skip()
  1850. def OnVolumeIsosurfDelete(self, event):
  1851. """Remove isosurface from list"""
  1852. list = self.FindWindowById(self.win['volume']['isosurfs'])
  1853. # remove item from list
  1854. isosurfId = list.GetSelection()
  1855. list.Delete(isosurfId)
  1856. # select last item
  1857. if list.GetCount() > 0:
  1858. list.SetSelection(list.GetCount()-1)
  1859. layer = self.mapWindow.GetSelectedLayer()
  1860. data = self.mapWindow.GetSelectedLayer(type='nviz')['volume']
  1861. id = data['object']['id']
  1862. # delete isosurface
  1863. del data['isosurface'][isosurfId]
  1864. self.mapWindow.nvizClass.DeleteIsosurface(id, isosurfId)
  1865. # update buttons
  1866. self.UpdateIsosurfButtons(list)
  1867. if self.parent.autoRender.IsChecked():
  1868. self.mapWindow.Refresh(False)
  1869. event.Skip()
  1870. def OnVolumeIsosurfMoveUp(self, event):
  1871. """Move isosurface up in the list"""
  1872. list = self.FindWindowById(self.win['volume']['isosurfs'])
  1873. sel = list.GetSelection()
  1874. if sel < 1:
  1875. return # this should not happen
  1876. layer = self.mapWindow.GetSelectedLayer()
  1877. data = self.mapWindow.GetSelectedLayer(type='nviz')['volume']
  1878. id = data['object']['id']
  1879. # move item up
  1880. text = list.GetStringSelection()
  1881. list.Insert(item=text, pos=sel-1)
  1882. list.Check(sel-1)
  1883. list.SetSelection(sel-1)
  1884. list.Delete(sel+1)
  1885. data['isosurface'].insert(sel-1, data['isosurface'][sel])
  1886. del data['isosurface'][sel+1]
  1887. self.mapWindow.nvizClass.MoveIsosurface(id, sel, True)
  1888. # update buttons
  1889. self.UpdateIsosurfButtons(list)
  1890. if self.parent.autoRender.IsChecked():
  1891. self.mapWindow.Refresh(False)
  1892. event.Skip()
  1893. def OnVolumeIsosurfMoveDown(self, event):
  1894. """Move isosurface dowm in the list"""
  1895. list = self.FindWindowById(self.win['volume']['isosurfs'])
  1896. sel = list.GetSelection()
  1897. if sel >= list.GetCount() - 1:
  1898. return # this should not happen
  1899. layer = self.mapWindow.GetSelectedLayer()
  1900. data = self.mapWindow.GetSelectedLayer(type='nviz')['volume']
  1901. id = data['object']['id']
  1902. # move item up
  1903. text = list.GetStringSelection()
  1904. list.Insert(item=text, pos=sel+2)
  1905. list.Check(sel+2)
  1906. list.SetSelection(sel+2)
  1907. list.Delete(sel)
  1908. data['isosurface'].insert(sel+2, data['isosurface'][sel])
  1909. del data['isosurface'][sel]
  1910. self.mapWindow.nvizClass.MoveIsosurface(id, sel, False)
  1911. # update buttons
  1912. self.UpdateIsosurfButtons(list)
  1913. if self.parent.autoRender.IsChecked():
  1914. self.mapWindow.Refresh(False)
  1915. event.Skip()
  1916. def UpdatePage(self, pageId):
  1917. """Update dialog (selected page)"""
  1918. self.pageChanging = True
  1919. layer = self.mapWindow.GetSelectedLayer()
  1920. data = self.mapWindow.GetSelectedLayer(type='nviz')
  1921. if pageId == 'view':
  1922. max = self.mapWindow.view['z-exag']['value'] * 10
  1923. hmin = self.mapWindow.iview['height']['min']
  1924. hmax = self.mapWindow.iview['height']['max']
  1925. for control in ('spin', 'slider'):
  1926. self.FindWindowById(self.win['view']['z-exag'][control]).SetRange(1,
  1927. max)
  1928. self.FindWindowById(self.win['view']['height'][control]).SetRange(hmin,
  1929. hmax)
  1930. elif pageId == 'surface':
  1931. if self.notebook.GetSelection() != self.page['surface']['id']:
  1932. for page in ('vector', 'volume'):
  1933. if self.page[page]['id'] > -1:
  1934. self.notebook.RemovePage(self.page[page]['id'])
  1935. self.page[page]['id'] = -1
  1936. self.page['surface']['id'] = 1
  1937. self.page['settings']['id'] = 2
  1938. panel = wx.FindWindowById(self.page['surface']['panel'])
  1939. self.notebook.InsertPage(n=self.page['surface']['id'],
  1940. page=panel,
  1941. text=" %s " % _("Layer properties"),
  1942. select=True)
  1943. self.UpdateSurfacePage(layer, data['surface'])
  1944. elif pageId == 'vector':
  1945. if self.notebook.GetSelection() != self.page['vector']['id']:
  1946. for page in ('surface', 'volume'):
  1947. if self.page[page]['id'] > -1:
  1948. self.notebook.RemovePage(self.page[page]['id'])
  1949. self.page[page]['id'] = -1
  1950. self.page['vector']['id'] = 1
  1951. self.page['settings']['id'] = 2
  1952. panel = wx.FindWindowById(self.page['vector']['panel'])
  1953. self.notebook.InsertPage(n=self.page['vector']['id'],
  1954. page=panel,
  1955. text=" %s " % _("Layer properties"),
  1956. select=True)
  1957. self.UpdateVectorPage(layer, data['vector'])
  1958. elif pageId == 'volume':
  1959. if self.notebook.GetSelection() != self.page['volume']['id']:
  1960. for page in ('surface', 'vector'):
  1961. if self.page[page]['id'] > -1:
  1962. self.notebook.RemovePage(self.page[page]['id'])
  1963. self.page[page]['id'] = -1
  1964. self.page['volume']['id'] = 1
  1965. self.page['settings']['id'] = 2
  1966. panel = wx.FindWindowById(self.page['volume']['panel'])
  1967. self.notebook.InsertPage(n=self.page['volume']['id'],
  1968. page=panel,
  1969. text=" %s " % _("Layer properties"),
  1970. select=True)
  1971. self.UpdateVolumePage(layer, data['volume'])
  1972. self.pageChanging = False
  1973. def UpdateSurfacePage(self, layer, data):
  1974. #
  1975. # attributes
  1976. #
  1977. for attr in ('topo', 'color'): # required
  1978. if layer and layer.type == 'raster':
  1979. self.FindWindowById(self.win['surface'][attr]['map']).SetValue(layer.name)
  1980. else:
  1981. self.FindWindowById(self.win['surface'][attr]['map']).SetValue('')
  1982. self.SetMapObjUseMap(nvizType='surface',
  1983. attrb=attr, map=True) # -> map
  1984. if data['attribute'].has_key('color'):
  1985. value = data['attribute']['color']['value']
  1986. if data['attribute']['color']['map']:
  1987. self.FindWindowById(self.win['surface']['color']['map']).SetValue(value)
  1988. else: # constant
  1989. color = map(int, value.split(':'))
  1990. self.FindWindowById(self.win['surface']['color']['const']).SetColour(color)
  1991. self.SetMapObjUseMap(nvizType='surface',
  1992. attrb=attr, map=data['attribute']['color']['map'])
  1993. self.SetMapObjUseMap(nvizType='surface',
  1994. attrb='shine', map=data['attribute']['shine']['map'])
  1995. value = data['attribute']['shine']['value']
  1996. if data['attribute']['shine']['map']:
  1997. self.FindWindowById(self.win['surface']['shine']['map']).SetValue(value)
  1998. else:
  1999. self.FindWindowById(self.win['surface']['shine']['const']).SetValue(value)
  2000. #
  2001. # draw
  2002. #
  2003. for control, dict in data['draw'].iteritems():
  2004. if control == 'all': # skip 'all' property
  2005. continue
  2006. if control == 'resolution':
  2007. self.FindWindowById(self.win['surface']['draw']['res-coarse']).SetValue(dict['coarse'])
  2008. self.FindWindowById(self.win['surface']['draw']['res-fine']).SetValue(dict['fine'])
  2009. continue
  2010. if control == 'mode':
  2011. if dict['desc']['mode'] == 'coarse':
  2012. self.FindWindowById(self.win['surface']['draw']['mode']).SetSelection(0)
  2013. elif dict['desc']['mode'] == 'fine':
  2014. self.FindWindowById(self.win['surface']['draw']['mode']).SetSelection(1)
  2015. else: # both
  2016. self.FindWindowById(self.win['surface']['draw']['mode']).SetSelection(2)
  2017. if dict['desc']['style'] == 'wire':
  2018. self.FindWindowById(self.win['surface']['draw']['style']).SetSelection(0)
  2019. else: # surface
  2020. self.FindWindowById(self.win['surface']['draw']['style']).SetSelection(1)
  2021. if dict['desc']['shading'] == 'flat':
  2022. self.FindWindowById(self.win['surface']['draw']['shading']).SetSelection(0)
  2023. else: # gouraud
  2024. self.FindWindowById(self.win['surface']['draw']['shading']).SetSelection(1)
  2025. continue
  2026. value = dict['value']
  2027. win = self.FindWindowById(self.win['surface']['draw'][control])
  2028. name = win.GetName()
  2029. if name == "selection":
  2030. win.SetSelection(value)
  2031. elif name == "colour":
  2032. color = map(int, value.split(':'))
  2033. win.SetColour(color)
  2034. else:
  2035. win.SetValue(value)
  2036. # enable/disable res widget + set draw mode
  2037. self.SetSurfaceMode()
  2038. color = self.FindWindowById(self.win['surface']['draw']['wire-color'])
  2039. def UpdateVectorPage(self, layer, data):
  2040. vInfo = gcmd.RunCommand('v.info',
  2041. parent = self,
  2042. read = True,
  2043. flags = 't',
  2044. map = layer.name)
  2045. if not vInfo:
  2046. return
  2047. npoints = nprimitives = 0
  2048. for line in vInfo.splitlines():
  2049. key, value = line.split('=')
  2050. if key == 'map3d':
  2051. mapIs3D = int(value)
  2052. elif key == 'points':
  2053. npoints = int(value)
  2054. nprimitives = npoints
  2055. elif key in ('lines',
  2056. 'boundaries',
  2057. 'centroids',
  2058. 'faces',
  2059. 'kernels'):
  2060. nprimitives += int(value)
  2061. if mapIs3D:
  2062. desc = _("Vector map <%s> is 3D") % layer.name
  2063. enable = False
  2064. else:
  2065. desc = _("Vector map <%s> is 2D") % layer.name
  2066. enable = True
  2067. desc += " - " + _("%(primitives)d primitives (%(points)d points)") % \
  2068. { 'primitives' : nprimitives, 'points' : npoints }
  2069. self.FindWindowById(self.win['vector']['lines']['flat']).Enable(enable)
  2070. for v in ('lines', 'points'):
  2071. self.FindWindowById(self.win['vector'][v]['surface']).Enable(enable)
  2072. self.FindWindowById(self.win['vector'][v]['height']['slider']).Enable(enable)
  2073. self.FindWindowById(self.win['vector'][v]['height']['spin']).Enable(enable)
  2074. self.FindWindowById(self.win['vector']['desc']).SetLabel(desc)
  2075. #
  2076. # lines
  2077. #
  2078. showLines = self.FindWindowById(self.win['vector']['lines']['show'])
  2079. if data['lines'].has_key('object'):
  2080. showLines.SetValue(True)
  2081. else:
  2082. showLines.SetValue(False)
  2083. if nprimitives - npoints > 0:
  2084. showLines.Enable(True)
  2085. else:
  2086. showLines.Enable(False)
  2087. self.UpdateVectorShow('lines',
  2088. showLines.IsChecked())
  2089. width = self.FindWindowById(self.win['vector']['lines']['width'])
  2090. width.SetValue(data['lines']['width']['value'])
  2091. color = self.FindWindowById(self.win['vector']['lines']['color'])
  2092. color.SetValue(map(int, data['lines']['color']['value'].split(':')))
  2093. for vtype in ('lines', 'points'):
  2094. if vtype == 'lines':
  2095. display = self.FindWindowById(self.win['vector']['lines']['flat'])
  2096. if data[vtype]['mode']['type'] == 'flat':
  2097. display.SetSelection(1)
  2098. else:
  2099. display.SetSelection(0)
  2100. if data[vtype]['mode']['type'] == 'surface':
  2101. rasters = self.mapWindow.GetLayerNames('raster')
  2102. surface = self.FindWindowById(self.win['vector'][vtype]['surface'])
  2103. surface.SetItems(rasters)
  2104. if len(rasters) > 0:
  2105. surface.SetStringSelection(data[vtype]['mode']['surface'])
  2106. for type in ('slider', 'spin'):
  2107. win = self.FindWindowById(self.win['vector']['lines']['height'][type])
  2108. win.SetValue(data['lines']['height']['value'])
  2109. #
  2110. # points
  2111. #
  2112. showPoints = self.FindWindowById(self.win['vector']['points']['show'])
  2113. if data['points'].has_key('object'):
  2114. showPoints.SetValue(True)
  2115. else:
  2116. showPoints.SetValue(False)
  2117. if npoints > 0:
  2118. showPoints.Enable(True)
  2119. else:
  2120. showPoints.Enable(False)
  2121. self.UpdateVectorShow('points',
  2122. showPoints.IsChecked())
  2123. # size, width, marker, color
  2124. for prop in ('size', 'width', 'marker', 'color'):
  2125. win = self.FindWindowById(self.win['vector']['points'][prop])
  2126. name = win.GetName()
  2127. if name == 'selection':
  2128. win.SetSelection(data['points'][prop]['value'])
  2129. elif name == 'color':
  2130. color = map(int, data['points'][prop]['value'].split(':'))
  2131. win.SetValue(color)
  2132. else:
  2133. win.SetValue(data['points'][prop]['value'])
  2134. # height
  2135. for type in ('slider', 'spin'):
  2136. win = self.FindWindowById(self.win['vector']['points']['height'][type])
  2137. win.SetValue(data['points']['height']['value'])
  2138. def UpdateVolumePage(self, layer, data):
  2139. """Update volume layer properties page"""
  2140. list = self.FindWindowById(self.win['volume']['isosurfs'])
  2141. #
  2142. # draw
  2143. #
  2144. for control, dict in data['draw'].iteritems():
  2145. if control == 'all': # skip 'all' property
  2146. continue
  2147. win = self.FindWindowById(self.win['volume']['draw'][control])
  2148. if control == 'shading':
  2149. if data['draw']['shading']['desc'] == 'flat':
  2150. value = 0
  2151. else:
  2152. value = 1
  2153. else:
  2154. value = dict['value']
  2155. if win.GetName() == "selection":
  2156. win.SetSelection(value)
  2157. else:
  2158. win.SetValue(value)
  2159. self.SetIsosurfaceMode(data['draw']['shading']['value'])
  2160. self.SetIsosurfaceResolution(data['draw']['resolution']['value'])
  2161. self.UpdateVolumeIsosurfPage(layer, data['attribute'])
  2162. def UpdateVolumeIsosurfPage(self, layer, data):
  2163. """Update dialog -- isosurface attributes"""
  2164. #
  2165. # isosurface attributes
  2166. #
  2167. for attrb in ('topo', 'color', 'mask',
  2168. 'transp', 'shine', 'emit'):
  2169. # check required first
  2170. if attrb == 'topo':
  2171. self.FindWindowById(self.win['volume'][attrb]['const']).SetValue(0)
  2172. continue
  2173. if attrb == 'color':
  2174. if layer and layer.type == '3d-raster':
  2175. self.FindWindowById(self.win['volume'][attrb]['map']).SetValue(layer.name)
  2176. else:
  2177. self.FindWindowById(self.win['volume'][attrb]['map']).SetValue('')
  2178. self.SetMapObjUseMap(nvizType='volume',
  2179. attrb=attrb, map=True) # -> map
  2180. continue
  2181. # skip empty attributes
  2182. if not data.has_key(attrb):
  2183. continue
  2184. value = data[attrb]['value']
  2185. if attrb == 'color':
  2186. if data[attrb]['map']:
  2187. self.FindWindowById(self.win['volume'][attrb]['map']).SetValue(value)
  2188. else: # constant
  2189. color = map(int, value.split(':'))
  2190. self.FindWindowById(self.win['volume'][attrb]['const']).SetColour(color)
  2191. else:
  2192. if data[attrb]['map']:
  2193. win = self.FindWindowById(self.win['volume'][attrb]['map'])
  2194. else:
  2195. win = self.FindWindowById(self.win['volume'][attrb]['const'])
  2196. win.SetValue(value)
  2197. self.SetMapObjUseMap(nvizType='volume',
  2198. attrb=attrb, map=data[attrb]['map'])
  2199. def SetPage(self, name):
  2200. """Get named page"""
  2201. self.notebook.SetSelection(self.page[name]['id'])
  2202. class ViewPositionWindow(wx.Window):
  2203. """Position control window (for NvizToolWindow)"""
  2204. def __init__(self, parent, id, mapwindow,
  2205. pos=wx.DefaultPosition,
  2206. size=wx.DefaultSize):
  2207. self.mapWindow = mapwindow
  2208. wx.Window.__init__(self, parent, id, pos, size)
  2209. self.SetBackgroundColour("WHITE")
  2210. self.pdc = wx.PseudoDC()
  2211. self.pdc.SetBrush(wx.Brush(colour='dark green', style=wx.SOLID))
  2212. self.pdc.SetPen(wx.Pen(colour='dark green', width=2, style=wx.SOLID))
  2213. self.Draw()
  2214. self.Bind(wx.EVT_ERASE_BACKGROUND, lambda x: None)
  2215. self.Bind(wx.EVT_PAINT, self.OnPaint)
  2216. # self.Bind(wx.EVT_MOTION, self.OnMouse)
  2217. self.Bind(wx.EVT_MOUSE_EVENTS, self.OnMouse)
  2218. def Draw(self, pos=None):
  2219. w, h = self.GetClientSize()
  2220. if pos is None:
  2221. x = self.mapWindow.view['pos']['x']
  2222. y = self.mapWindow.view['pos']['y']
  2223. x = x * w
  2224. y = y * h
  2225. else:
  2226. x, y = pos
  2227. self.pdc.Clear()
  2228. self.pdc.BeginDrawing()
  2229. self.pdc.DrawLine(w / 2, h / 2, x, y)
  2230. self.pdc.DrawCircle(x, y, 5)
  2231. self.pdc.EndDrawing()
  2232. def OnPaint(self, event):
  2233. dc = wx.BufferedPaintDC(self)
  2234. dc.SetBackground(wx.Brush("White"))
  2235. dc.Clear()
  2236. self.PrepareDC(dc)
  2237. self.pdc.DrawToDC(dc)
  2238. def OnMouse(self, event):
  2239. if event.LeftIsDown():
  2240. x, y = event.GetPosition()
  2241. self.Draw(pos=(x, y))
  2242. self.Refresh(False)
  2243. w, h = self.GetClientSize()
  2244. x = float(x) / w
  2245. y = float(y) / h
  2246. if x >= 0 and x <= 1.0:
  2247. self.mapWindow.view['pos']['x'] = x
  2248. if y >= 0 and y <= 1.0:
  2249. self.mapWindow.view['pos']['y'] = y
  2250. event = wxUpdateView(zExag=False)
  2251. wx.PostEvent(self.mapWindow, event)
  2252. self.mapWindow.render['quick'] = True
  2253. self.mapWindow.Refresh(eraseBackground=False)
  2254. elif event.LeftUp():
  2255. self.mapWindow.render['quick'] = False
  2256. self.mapWindow.Refresh(eraseBackground=False)
  2257. event.Skip()