widgets.py 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023
  1. """!
  2. @package web_services.widgets
  3. @brief Widgets for web services (WMS, WMTS, NasaOnEarh)
  4. List of classes:
  5. - widgets::WSPanel
  6. - widgets::LayersList
  7. (C) 2012-2013 by the GRASS Development Team
  8. This program is free software under the GNU General Public License
  9. (>=v2). Read the file COPYING that comes with GRASS for details.
  10. @author Martin Landa <landa.martin gmail.com>
  11. @author Stepan Turek <stepan.turek seznam.cz>
  12. """
  13. import os
  14. import sys
  15. import shutil
  16. from copy import deepcopy
  17. try:
  18. from xml.etree.ElementTree import ParseError
  19. except ImportError: # < Python 2.7
  20. from xml.parsers.expat import ExpatError as ParseError
  21. import wx
  22. import wx.lib.flatnotebook as FN
  23. import wx.lib.colourselect as csel
  24. import wx.lib.mixins.listctrl as listmix
  25. from wx.lib.newevent import NewEvent
  26. from wx.gizmos import TreeListCtrl
  27. from core import globalvar
  28. from core.debug import Debug
  29. from core.gcmd import GWarning, GMessage
  30. from core.gconsole import CmdThread, GStderr, EVT_CMD_DONE, EVT_CMD_OUTPUT
  31. from web_services.cap_interface import WMSCapabilities, WMTSCapabilities, OnEarthCapabilities
  32. from gui_core.widgets import GNotebook
  33. import grass.script as grass
  34. sys.path.append(os.path.join(os.getenv("GISBASE"), "etc", "r.in.wms"))
  35. from wms_base import WMSDriversInfo
  36. wxOnCapParsed, EVT_CAP_PARSED = NewEvent()
  37. class WSPanel(wx.Panel):
  38. def __init__(self, parent, receiver, web_service, **kwargs):
  39. """!Show data from capabilities file.
  40. Events: EVT_CAP_PARSED - this event is released when capabilities file is downloaded
  41. (after ConnectToServer method was called)
  42. @param parent - parent widget
  43. @param receiver - receives EVT_CAP_PARSED event
  44. @param web_service - web service to be panel generated for
  45. """
  46. wx.Panel.__init__(self, parent = parent, id = wx.ID_ANY)
  47. self.parent = parent
  48. self.ws = web_service
  49. self.receiver = receiver
  50. # stores widgets, which represents parameters/flags of d.wms
  51. self.params = {}
  52. self.flags = {}
  53. self.o_layer_name = ''
  54. # stores err output from r.in.wms during getting capabilities
  55. self.cmd_err_str = ''
  56. # stores selected layer from layer list
  57. self.sel_layers = []
  58. # downloaded and parsed data from server successfully?
  59. self.is_connected = False
  60. # common part of command for r.in.wms -c and d.wms
  61. self.ws_cmdl = None
  62. # provides information about driver parameters
  63. self.drv_info = WMSDriversInfo()
  64. self.drv_props = self.drv_info.GetDrvProperties(self.ws)
  65. self.ws_drvs = {
  66. 'WMS_1.1.1' : {
  67. 'cmd' : ['wms_version=1.1.1',
  68. 'driver=WMS_GRASS'],
  69. 'cap_parser' : lambda temp_file : WMSCapabilities(temp_file, '1.1.1'),
  70. },
  71. 'WMS_1.3.0' : {
  72. 'cmd' : ['wms_version=1.3.0',
  73. 'driver=WMS_GRASS'],
  74. 'cap_parser' : lambda temp_file : WMSCapabilities(temp_file, '1.3.0'),
  75. },
  76. 'WMTS' : {
  77. 'cmd' : ['driver=WMTS_GRASS'],
  78. 'cap_parser' : WMTSCapabilities,
  79. },
  80. 'OnEarth' : {
  81. 'cmd' : ['driver=OnEarth_GRASS'],
  82. 'cap_parser' : OnEarthCapabilities,
  83. }
  84. }
  85. self.cmdStdErr = GStderr(self)
  86. self.cmd_thread = CmdThread(self)
  87. self.cap_file = grass.tempfile()
  88. self.notebook = GNotebook(parent = self,
  89. style = FN.FNB_FANCY_TABS | FN.FNB_NO_X_BUTTON)
  90. self._requestPage()
  91. self._advancedSettsPage()
  92. self._layout()
  93. self.Bind(EVT_CMD_DONE, self.OnCapDownloadDone)
  94. self.Bind(EVT_CMD_OUTPUT, self.OnCmdOutput)
  95. def __del__(self):
  96. self.cmd_thread.abort(abortall =True)
  97. grass.try_remove(self.cap_file)
  98. def _layout(self):
  99. reqDataBox = wx.StaticBox(parent = self, id = wx.ID_ANY,
  100. label = _(" Requested data settings "))
  101. sizer = wx.StaticBoxSizer(reqDataBox, wx.VERTICAL)
  102. sizer.Add(item = self.notebook, proportion = 1,
  103. flag = wx.EXPAND)
  104. self.SetSizer(sizer)
  105. def _requestPage(self):
  106. """!Create request page"""
  107. self.req_page_panel = wx.Panel(parent = self, id = wx.ID_ANY)
  108. self.notebook.AddPage(page = self.req_page_panel,
  109. text=_('Request'),
  110. name = 'request')
  111. # list of layers
  112. self.layersBox = wx.StaticBox(parent = self.req_page_panel, id = wx.ID_ANY,
  113. label=_("List of layers "))
  114. style = wx.TR_DEFAULT_STYLE | wx.TR_HAS_BUTTONS | wx.TR_FULL_ROW_HIGHLIGHT
  115. if self.drv_props['req_multiple_layers']:
  116. style = style | wx.TR_MULTIPLE
  117. if 'WMS' not in self.ws:
  118. style = style | wx.TR_HIDE_ROOT
  119. self.list = LayersList(parent = self.req_page_panel,
  120. web_service = self.ws,
  121. style = style)
  122. self.params['format'] = None
  123. self.params['srs'] = None
  124. if 'srs' not in self.drv_props['ignored_params']:
  125. projText = wx.StaticText(parent = self.req_page_panel, id = wx.ID_ANY, label = _("Source projection:"))
  126. self.params['srs'] = wx.Choice(parent = self.req_page_panel, id = wx.ID_ANY)
  127. self.list.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnListSelChanged)
  128. # layout
  129. self.req_page_sizer = wx.BoxSizer(wx.VERTICAL)
  130. layersSizer = wx.StaticBoxSizer(self.layersBox, wx.HORIZONTAL)
  131. layersSizer.Add(item = self.list, proportion = 1,
  132. flag = wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border = 5)
  133. self.req_page_sizer.Add(item = layersSizer, proportion = 1,
  134. flag = wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border = 5)
  135. self.source_sizer = wx.BoxSizer(wx.HORIZONTAL)
  136. if self.params['format'] is not None:
  137. self.source_sizer.Add(item = self.params['format'],
  138. flag = wx.LEFT | wx.RIGHT | wx.BOTTOM, border = 5)
  139. if self.params['srs'] is not None:
  140. self.source_sizer.Add(item = projText, flag = wx.ALIGN_CENTER_VERTICAL | wx.ALL, border = 5)
  141. self.source_sizer.Add(item = self.params['srs'],
  142. flag = wx.ALIGN_CENTER_VERTICAL | wx.RIGHT | wx.TOP | wx.BOTTOM, border = 5)
  143. self.req_page_sizer.Add(item = self.source_sizer,
  144. flag = wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border = 5)
  145. self.req_page_panel.SetSizer(self.req_page_sizer)
  146. def enableButtons(self, enable = True):
  147. """!Enable/disable up, down, buttons
  148. """
  149. self.btnUp.Enable(enable)
  150. self.btnDown.Enable(enable)
  151. def _advancedSettsPage(self):
  152. """!Create advanced settings page
  153. """
  154. #TODO parse maxcol, maxrow, settings from d.wms module?
  155. #TODO OnEarth driver - add selection of time
  156. adv_setts_panel = wx.Panel(parent = self, id = wx.ID_ANY)
  157. self.notebook.AddPage(page = adv_setts_panel,
  158. text=_('Advanced request settings'),
  159. name = 'adv_req_setts')
  160. labels = {}
  161. self.l_odrder_list = None
  162. if 'WMS' in self.ws:
  163. labels['l_order'] = wx.StaticBox(parent = adv_setts_panel, id = wx.ID_ANY,
  164. label = _("Order of layers in raster"))
  165. self.l_odrder_list = wx.ListBox(adv_setts_panel, id = wx.ID_ANY, choices = [],
  166. style = wx.LB_SINGLE|wx.LB_NEEDED_SB)
  167. self.btnUp = wx.Button(adv_setts_panel, id = wx.ID_ANY, label = _("Up"))
  168. self.btnDown = wx.Button(adv_setts_panel, id = wx.ID_ANY, label = _("Down"))
  169. self.btnUp.Bind(wx.EVT_BUTTON, self.OnUp)
  170. self.btnDown.Bind(wx.EVT_BUTTON, self.OnDown)
  171. labels['method'] = wx.StaticText(parent = adv_setts_panel, id = wx.ID_ANY,
  172. label = _("Reprojection method:"))
  173. self.reproj_methods = ['near', 'bilinear', 'cubic', 'cubicspline']
  174. self.params['method'] = wx.Choice(parent = adv_setts_panel, id = wx.ID_ANY,
  175. choices = [_('near'), _('bilinear'), _('cubic'), _('cubicspline')])
  176. labels['maxcols'] = wx.StaticText(parent = adv_setts_panel, id = wx.ID_ANY,
  177. label = _("Maximum columns to request from server at time:"))
  178. self.params['maxcols'] = wx.SpinCtrl(parent = adv_setts_panel, id = wx.ID_ANY, size = (100, -1))
  179. labels['maxrows'] = wx.StaticText(parent = adv_setts_panel, id = wx.ID_ANY,
  180. label = _("Maximum rows to request from server at time:"))
  181. self.params['maxrows'] = wx.SpinCtrl(parent = adv_setts_panel, id = wx.ID_ANY, size = (100, -1))
  182. min = 100
  183. max = 10000
  184. self.params['maxcols'].SetRange(min,max)
  185. self.params['maxrows'].SetRange(min,max)
  186. val = 500
  187. self.params['maxcols'].SetValue(val)
  188. self.params['maxrows'].SetValue(val)
  189. self.flags['o'] = self.params['bgcolor'] = None
  190. if not 'o' in self.drv_props['ignored_flags']:
  191. self.flags['o'] = wx.CheckBox(parent = adv_setts_panel, id = wx.ID_ANY,
  192. label = _("Do not request transparent data"))
  193. self.flags['o'].Bind(wx.EVT_CHECKBOX, self.OnTransparent)
  194. labels['bgcolor'] = wx.StaticText(parent = adv_setts_panel, id = wx.ID_ANY,
  195. label = _("Background color:"))
  196. self.params['bgcolor'] = csel.ColourSelect(parent = adv_setts_panel, id = wx.ID_ANY,
  197. colour = (255, 255, 255),
  198. size = globalvar.DIALOG_COLOR_SIZE)
  199. self.params['bgcolor'].Enable(False)
  200. self.params['urlparams'] = None
  201. if self.params['urlparams'] not in self.drv_props['ignored_params']:
  202. labels['urlparams'] = wx.StaticText(parent = adv_setts_panel, id = wx.ID_ANY,
  203. label = _("Additional query parameters for server:"))
  204. self.params['urlparams'] = wx.TextCtrl(parent = adv_setts_panel, id = wx.ID_ANY)
  205. # layout
  206. border = wx.BoxSizer(wx.VERTICAL)
  207. if 'WMS' in self.ws:
  208. boxSizer = wx.StaticBoxSizer(labels['l_order'], wx.VERTICAL)
  209. gridSizer = wx.GridBagSizer (hgap = 3, vgap = 3)
  210. gridSizer.AddGrowableCol(0)
  211. gridSizer.Add(self.l_odrder_list,
  212. pos = (0,0),
  213. span = (4, 1),
  214. flag = wx.ALIGN_CENTER_VERTICAL|wx.EXPAND,
  215. border = 0)
  216. gridSizer.Add(self.btnUp,
  217. pos = (0,1),
  218. flag = wx.ALIGN_CENTER_VERTICAL,
  219. border = 0)
  220. gridSizer.Add(self.btnDown,
  221. pos = (1,1),
  222. flag = wx.ALIGN_CENTER_VERTICAL,
  223. border = 0)
  224. boxSizer.Add(gridSizer,
  225. flag = wx.EXPAND | wx.ALL,
  226. border = 5)
  227. border.Add(item = boxSizer,
  228. flag = wx.LEFT | wx.RIGHT | wx.UP | wx.EXPAND,
  229. border = 5)
  230. gridSizer = wx.GridBagSizer (hgap = 3, vgap = 3)
  231. gridSizer.AddGrowableCol(0)
  232. row = 0
  233. for k in ['method', 'maxcols', 'maxrows', 'o', 'bgcolor']:
  234. if self.params.has_key(k):
  235. param = self.params[k]
  236. elif self.flags.has_key(k):
  237. param = self.flags[k]
  238. if param is None:
  239. continue
  240. if labels.has_key(k) or k == 'o':
  241. if k != 'o':
  242. label = labels[k]
  243. else:
  244. label = param
  245. gridSizer.Add(label,
  246. flag = wx.ALIGN_LEFT |
  247. wx.ALIGN_CENTER_VERTICAL,
  248. pos = (row, 0))
  249. if k != 'o':
  250. gridSizer.Add(item = param,
  251. flag = wx.ALIGN_RIGHT |
  252. wx.ALIGN_CENTER_VERTICAL,
  253. pos = (row, 1))
  254. row += 1
  255. border.Add(item = gridSizer,
  256. flag = wx.LEFT | wx.RIGHT | wx.TOP | wx.EXPAND,
  257. border = 5)
  258. if self.params['urlparams']:
  259. gridSizer = wx.GridBagSizer (hgap = 3, vgap = 3)
  260. gridSizer.AddGrowableCol(1)
  261. row = 0
  262. gridSizer.Add(labels['urlparams'],
  263. flag = wx.ALIGN_LEFT |
  264. wx.ALIGN_CENTER_VERTICAL,
  265. pos = (row, 0))
  266. gridSizer.Add(item = self.params['urlparams'],
  267. flag = wx.ALIGN_RIGHT |
  268. wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  269. pos = (row, 1))
  270. border.Add(item = gridSizer,
  271. flag = wx.LEFT | wx.RIGHT | wx.TOP | wx.EXPAND,
  272. border = 5)
  273. adv_setts_panel.SetSizer(border)
  274. def OnUp(self, event):
  275. """!Move selected layer up
  276. """
  277. if self.l_odrder_list.GetSelections():
  278. pos = self.l_odrder_list.GetSelection()
  279. if pos:
  280. self.sel_layers.insert(pos - 1, self.sel_layers.pop(pos))
  281. if pos > 0:
  282. self._updateLayerOrderList(selected = (pos - 1))
  283. else:
  284. self._updateLayerOrderList(selected = 0)
  285. def OnDown(self, event):
  286. """!Move selected to down
  287. """
  288. if self.l_odrder_list.GetSelections():
  289. pos = self.l_odrder_list.GetSelection()
  290. if pos != len(self.sel_layers) - 1:
  291. self.sel_layers.insert(pos + 1, self.sel_layers.pop(pos))
  292. if pos < len(self.sel_layers) -1:
  293. self._updateLayerOrderList(selected = (pos + 1))
  294. else:
  295. self._updateLayerOrderList(selected = len(self.sel_layers) -1)
  296. def _updateLayerOrderList(self, selected = None):
  297. """!Update order in list.
  298. """
  299. def getlayercaption(layer):
  300. if l['title']:
  301. cap = (l['title'])
  302. else:
  303. cap = (l['name'])
  304. if l['style']:
  305. if l['style']['title']:
  306. cap += ' / ' + l['style']['title']
  307. else:
  308. cap += ' / ' + l['style']['name']
  309. return cap
  310. layer_capts = [getlayercaption(l) for l in self.sel_layers]
  311. self.l_odrder_list.Set(layer_capts)
  312. if self.l_odrder_list.IsEmpty():
  313. self.enableButtons(False)
  314. else:
  315. self.enableButtons(True)
  316. if selected is not None:
  317. self.l_odrder_list.SetSelection(selected)
  318. self.l_odrder_list.EnsureVisible(selected)
  319. def OnTransparent(self, event):
  320. checked = event.IsChecked()
  321. if checked:
  322. self.params['bgcolor'].Enable(True)
  323. else:
  324. self.params['bgcolor'].Enable(False)
  325. def ConnectToServer(self, url, username, password):
  326. """!Download and parse data from capabilities file.
  327. @param url - server url
  328. @param username - username for connection
  329. @param password - password for connection
  330. """
  331. self._prepareForNewConn(url, username, password)
  332. cap_cmd = ['r.in.wms', '-c', ('capfile_output=%s' % self.cap_file), '--overwrite'] + self.ws_cmdl
  333. self.currentPid = self.cmd_thread.GetId()
  334. self.cmd_thread.RunCmd(cap_cmd, stderr = self.cmdStdErr)
  335. def OnCmdOutput(self, event):
  336. """!Manage cmd output.
  337. """
  338. if Debug.GetLevel() != 0:
  339. Debug.msg(1, event.text)
  340. elif event.type != 'message' and event.type != 'warning':
  341. self.cmd_err_str += event.text + os.linesep
  342. def _prepareForNewConn(self, url, username, password):
  343. """!Prepare panel for new connection
  344. """
  345. self.is_connected = False
  346. self.sel_layers = []
  347. self.formats_list = []
  348. self.projs_list = []
  349. self.conn = {
  350. 'url' : url,
  351. 'password' : password,
  352. 'username' : username
  353. }
  354. conn_cmd = []
  355. for k, v in self.conn.iteritems():
  356. if v:
  357. conn_cmd.append("%s=%s" % (k,v))
  358. self.ws_cmdl = self.ws_drvs[self.ws]['cmd'] + conn_cmd
  359. def OnCapDownloadDone(self, event):
  360. """!Process donwloaded capabilities file and emits EVT_CAP_PARSED (see class constructor).
  361. """
  362. if event.pid != self.currentPid:
  363. return
  364. if event.returncode != 0:
  365. if self.cmd_err_str:
  366. self.cmd_err_str = _("Unable to download %s capabilities file\nfrom <%s>:\n" % \
  367. (self.ws.replace('_', ' '), self.conn['url'])) + self.cmd_err_str
  368. self._postCapParsedEvt(error_msg = self.cmd_err_str)
  369. self.cmd_err_str = ''
  370. return
  371. self._parseCapFile(self.cap_file)
  372. def _parseCapFile(self, cap_file):
  373. """!Parse capabilities data and emits EVT_CAP_PARSED (see class constructor).
  374. """
  375. try:
  376. self.cap = self.ws_drvs[self.ws]['cap_parser'](cap_file)
  377. except (IOError, ParseError) as error:
  378. error_msg = _("%s web service was not found in fetched capabilities file from <%s>:\n%s\n" % \
  379. (self.ws, self.conn['url'], str(error)))
  380. if Debug.GetLevel() != 0:
  381. Debug.msg(1, error_msg)
  382. self._postCapParsedEvt(None)
  383. else:
  384. self._postCapParsedEvt(error_msg = error_msg)
  385. return
  386. self.is_connected = True
  387. # WMS standard has formats defined for all layers
  388. if 'WMS' in self.ws:
  389. self.formats_list = sorted(self._getFormats())
  390. self._updateFormatRadioBox(self.formats_list)
  391. self._setDefaultFormatVal()
  392. self.list.LoadData(self.cap)
  393. self.OnListSelChanged(event = None)
  394. self._postCapParsedEvt(None)
  395. def ParseCapFile(self, url, username, password, cap_file = None,):
  396. """!Parse capabilities data and emits EVT_CAP_PARSED (see class constructor).
  397. """
  398. self._prepareForNewConn(url, username, password)
  399. if cap_file is None or not url:
  400. self._postCapParsedEvt(None)
  401. return
  402. shutil.copyfile(cap_file, self.cap_file)
  403. self._parseCapFile(self.cap_file)
  404. def UpdateWidgetsByCmd(self, cmd):
  405. """!Update panel widgets accordnig to passed cmd tuple
  406. @param cmd - cmd in tuple
  407. """
  408. dcmd = cmd[1]
  409. layers = []
  410. if dcmd.has_key('layers'):
  411. layers = dcmd['layers']
  412. styles = []
  413. if dcmd.has_key('styles'):
  414. styles = dcmd['styles']
  415. if 'WMS' in self.ws:
  416. layers = layers.split(',')
  417. styles = styles.split(',')
  418. else:
  419. layers = [layers]
  420. styles = [styles]
  421. if len(layers) != len(styles):
  422. styles = [''] * len(layers)
  423. l_st_list = []
  424. for i in range(len(layers)):
  425. l_st_list.append({'style' : styles[i],
  426. 'layer' : layers[i]})
  427. # WMS standard - first layer in params is most bottom...
  428. # therefore layers order need to be reversed
  429. l_st_list = [l for l in reversed(l_st_list)]
  430. self.list.SelectLayers(l_st_list)
  431. params = {}
  432. if dcmd.has_key('format'):
  433. params['format'] = dcmd['format']
  434. if dcmd.has_key('srs'):
  435. params['srs'] = 'EPSG:' + dcmd['srs']
  436. if dcmd.has_key('method'):
  437. params['method'] = dcmd['method']
  438. for p, v in params.iteritems():
  439. if self.params[p]:
  440. self.params[p].SetStringSelection(v)
  441. for p, conv_f in [('urlparams', None), ('maxcols', int), ('maxrows', int)]:
  442. if dcmd.has_key(p):
  443. v = dcmd[p]
  444. if conv_f:
  445. v = conv_f(v)
  446. self.params[p].SetValue(v)
  447. if dcmd.has_key('flags') and \
  448. 'o' in dcmd['flags']:
  449. self.flags['o'].SetValue(1)
  450. self.params['bgcolor'].Enable(True)
  451. if dcmd.has_key('bgcolor') and \
  452. self.params['bgcolor']:
  453. bgcolor = dcmd['bgcolor'].strip().lower()
  454. if len(bgcolor) == 8 and \
  455. '0x' == bgcolor[:2]:
  456. colour= '#' + bgcolor[2:]
  457. self.params['bgcolor'].SetColour(colour)
  458. def IsConnected(self):
  459. """!Was successful in downloading and parsing capabilities data?
  460. """
  461. return self.is_connected
  462. def _postCapParsedEvt(self, error_msg):
  463. """!Helper function
  464. """
  465. cap_parsed_event = wxOnCapParsed(error_msg = error_msg)
  466. wx.PostEvent(self.receiver, cap_parsed_event)
  467. def CreateCmd(self):
  468. """!Create d.wms cmd from values of panels widgets
  469. @return cmd list
  470. @return None if required widgets do not have selected/filled values.
  471. """
  472. # check required widgets
  473. if not self._checkImportValues():
  474. return None
  475. # create d.wms command
  476. lcmd = self.ws_cmdl
  477. lcmd = ['d.wms'] + lcmd
  478. layers="layers="
  479. styles='styles='
  480. first = True
  481. # WMS standard - first layer in params is most bottom...
  482. # therefore layers order need to be reversed
  483. for layer in reversed(self.sel_layers):
  484. if not first:
  485. layers += ','
  486. styles += ','
  487. first = False
  488. layers += layer['name']
  489. if layer['style'] is not None:
  490. styles += layer['style']['name']
  491. lcmd.append(layers)
  492. lcmd.append(styles)
  493. if 'format' not in self.drv_props['ignored_params']:
  494. i_format = self.params['format'].GetSelection()
  495. lcmd.append("format=%s" % self.formats_list[i_format])
  496. if 'srs' not in self.drv_props['ignored_params']:
  497. i_srs = self.params['srs'].GetSelection()
  498. epsg_num = int(self.projs_list[i_srs].split(':')[1])
  499. lcmd.append("srs=%s" % epsg_num)
  500. for k in ['maxcols', 'maxrows', 'urlparams']:
  501. lcmd.append(k + '=' + str(self.params[k].GetValue()))
  502. i_method = self.params['method'].GetSelection()
  503. lcmd.append('method=' + self.reproj_methods[i_method])
  504. if not 'o' in self.drv_props['ignored_flags'] and \
  505. self.flags['o'].IsChecked():
  506. lcmd.append('-o')
  507. c = self.params['bgcolor'].GetColour()
  508. hex_color = wx.Colour(c[0], c[1], c[2]).GetAsString(wx.C2S_HTML_SYNTAX)
  509. lcmd.append("bgcolor=" + '0x' + hex_color[1:])
  510. lcmd.append("map=" + self.o_layer_name)
  511. return lcmd
  512. def OnListSelChanged(self, event):
  513. """!Update widgets according to selected layer in list.
  514. """
  515. curr_sel_ls = self.list.GetSelectedLayers()
  516. # update self.sel_layers (selected layer list)
  517. if 'WMS' in self.ws:
  518. for sel_l in self.sel_layers[:]:
  519. if sel_l not in curr_sel_ls:
  520. self.sel_layers.remove(sel_l)
  521. for l in curr_sel_ls:
  522. if l not in self.sel_layers:
  523. self.sel_layers.append(l)
  524. self._updateLayerOrderList()
  525. else:
  526. self.sel_layers = curr_sel_ls
  527. # update projection
  528. self.projs_list = []
  529. projs_list = []
  530. intersect_proj = []
  531. first = True
  532. for l in curr_sel_ls:
  533. layer_projs = l['cap_intf_l'].GetLayerData('srs')
  534. if first:
  535. projs_list = layer_projs
  536. first = False
  537. continue
  538. projs_list = set(projs_list).intersection(layer_projs)
  539. if 'srs' not in self.drv_props['ignored_params']:
  540. for proj in projs_list:
  541. proj_spl = proj.strip().split(':')
  542. if 'epsg' in proj_spl[0].strip().lower():
  543. try:
  544. int(proj_spl[1])
  545. self.projs_list.append(proj)
  546. except ValueError, IndexError:
  547. continue
  548. cur_sel = self.params['srs'].GetStringSelection()
  549. self.projs_list = sorted(self.projs_list)
  550. self.params['srs'].SetItems(self.projs_list)
  551. if cur_sel:
  552. self.params['srs'].SetStringSelection(cur_sel)
  553. else:
  554. try:
  555. i = self.projs_list.index('EPSG:4326')
  556. self.params['srs'].SetSelection(i)
  557. except ValueError:
  558. if len(self.projs_list) > 0:
  559. self.params['srs'].SetSelection(0)
  560. # update format
  561. if 'WMS' not in self.ws and \
  562. 'format' not in self.drv_props['ignored_params']:
  563. self.formats_list = []
  564. cur_sel = None
  565. if self.params['format'] is not None:
  566. cur_sel = self.params['format'].GetStringSelection()
  567. if len(curr_sel_ls) > 0:
  568. self.formats_list = sorted(self._getFormats(curr_sel_ls[0]['cap_intf_l']))
  569. self._updateFormatRadioBox(self.formats_list)
  570. if cur_sel:
  571. self.params['format'].SetStringSelection(cur_sel)
  572. else:
  573. self._setDefaultFormatVal()
  574. self.Layout()
  575. def _setDefaultFormatVal(self):
  576. """!Set default format value.
  577. """
  578. try:
  579. i = self.formats_list.index('png')
  580. self.params['format'].SetSelection(i)
  581. except ValueError:
  582. pass
  583. def _updateFormatRadioBox(self, formats_list):
  584. """!Helper function
  585. """
  586. if self.params['format'] is not None:
  587. self.req_page_sizer.Detach(self.params['format'])
  588. self.params['format'].Destroy()
  589. if len(self.formats_list) > 0:
  590. self.params['format'] = wx.RadioBox(parent = self.req_page_panel, id = wx.ID_ANY,
  591. label = _("Source image format"), pos = wx.DefaultPosition,
  592. choices = formats_list, majorDimension = 4,
  593. style = wx.RA_SPECIFY_COLS)
  594. self.source_sizer.Insert(item = self.params['format'], before = 2,
  595. flag = wx.LEFT | wx.RIGHT | wx.BOTTOM, border = 5)
  596. def _getFormats(self, layer = None):
  597. """!Get formats
  598. WMS has formats defined generally for whole cap.
  599. In WMTS and NASA OnEarh formats are defined for layer.
  600. """
  601. formats_label = []
  602. if layer is None:
  603. formats_list = self.cap.GetFormats()
  604. else:
  605. formats_list = layer.GetLayerData('format')
  606. for frmt in formats_list:
  607. frmt = frmt.strip()
  608. label = self.drv_info.GetFormatLabel(frmt)
  609. if label:
  610. formats_label.append(label)
  611. return formats_label
  612. def _checkImportValues(self,):
  613. """!Check if required widgets are selected/filled
  614. """
  615. warning_str = ""
  616. show_war = False
  617. if not self.list or not self.list.GetSelectedLayers():
  618. warning_str += _("Select layer in layer list.\n")
  619. show_war = True
  620. if self.params['format'] is not None and \
  621. self.params['format'].GetSelection() == -1:
  622. warning_str += _("Select source image format.\n")
  623. show_war = True
  624. if self.params['srs'] is not None and \
  625. self.params['srs'].GetSelection() == -1:
  626. warning_str += _("Select source projection.\n")
  627. show_war = True
  628. if not self.o_layer_name:
  629. warning_str += _("Choose output layer name.\n")
  630. show_war = True
  631. if show_war:
  632. GMessage(parent = self.parent,
  633. message = warning_str)
  634. return False
  635. return True
  636. def SetOutputLayerName(self, name):
  637. """!Set name of layer to be added to layer tree
  638. """
  639. self.o_layer_name = name
  640. def GetCapFile(self):
  641. """!Get path to file where capabilities are saved
  642. """
  643. return self.cap_file
  644. def GetWebService(self):
  645. """!Get web service
  646. """
  647. return self.ws
  648. class LayersList(TreeListCtrl, listmix.ListCtrlAutoWidthMixin):
  649. def __init__(self, parent, web_service, style, pos=wx.DefaultPosition):
  650. """!List of layers and styles available in capabilities file
  651. """
  652. self.parent = parent
  653. self.ws = web_service
  654. TreeListCtrl.__init__(self, parent = parent, id = wx.ID_ANY, style = style)
  655. # setup mixins
  656. listmix.ListCtrlAutoWidthMixin.__init__(self)
  657. if self.ws != 'OnEarth':
  658. self.AddColumn(_('Name'))
  659. self.AddColumn(_('Type'))
  660. else:
  661. self.AddColumn(_('Layer name'))
  662. self.SetMainColumn(0) # column with the tree
  663. self.setResizeColumn(0)
  664. self.root = None
  665. self.Bind(wx.EVT_TREE_SEL_CHANGING, self.OnListSelChanging)
  666. def LoadData(self, cap = None):
  667. """!Load data into list
  668. """
  669. # detete first all items
  670. self.DeleteAllItems()
  671. if not cap:
  672. return
  673. def AddLayerChildrenToTree(parent_layer, parent_item):
  674. """!Recursive function which adds all capabilities layers/styles to the LayersList.
  675. """
  676. def gettitle(layer):
  677. """!Helper function"""
  678. if layer.GetLayerData('title') is not None:
  679. layer_title = layer.GetLayerData('title')
  680. elif layer.GetLayerData('name') is not None:
  681. layer_title = layer.GetLayerData('name')
  682. else:
  683. layer_title = str(layer.GetId())
  684. return layer_title
  685. def addlayer(layer, item):
  686. if self.ws != 'OnEarth':
  687. self.SetItemText(item, _('layer'), 1)
  688. styles = layer.GetLayerData('styles')
  689. def_st = None
  690. for st in styles:
  691. if st['name']:
  692. style_name = st['name']
  693. else:
  694. continue
  695. if st['title']:
  696. style_name = st['title']
  697. if st['isDefault']:
  698. def_st = st
  699. style_item = self.AppendItem(item, style_name)
  700. if self.ws != 'OnEarth':
  701. self.SetItemText(style_item, _('style'), 1)
  702. self.SetPyData(style_item, {'type' : 'style',
  703. 'layer' : layer, # it is parent layer of style
  704. 'style' : st})
  705. self.SetPyData(item, {'type' : 'layer', # is it layer or style?
  706. 'layer' : layer, # *Layer instance from web_services.cap_interface
  707. 'style' : def_st}) # layer can have assigned default style
  708. if parent_layer is None:
  709. parent_layer = cap.GetRootLayer()
  710. layer_title = gettitle(parent_layer)
  711. parent_item = self.AddRoot(layer_title)
  712. addlayer(parent_layer, parent_item)
  713. for layer in parent_layer.GetChildren():
  714. item = self.AppendItem(parent_item, gettitle(layer))
  715. addlayer(layer, item)
  716. AddLayerChildrenToTree(layer, item)
  717. AddLayerChildrenToTree(None, None)
  718. self.ExpandAll(self.GetRootItem())
  719. def GetSelectedLayers(self):
  720. """!Get selected layers/styles in LayersList
  721. @return dict with these items:
  722. 'name' : layer name used for request
  723. if it is style, it is name of parent layer
  724. 'title' : layer title
  725. 'style' : {'name' : 'style name', title : 'style title'}
  726. 'cap_intf_l' : *Layer instance from web_services.cap_interface
  727. """
  728. sel_layers = self.GetSelections()
  729. sel_layers_dict = []
  730. for s in sel_layers:
  731. try:
  732. layer = self.GetPyData(s)['layer']
  733. except ValueError:
  734. continue
  735. sel_layers_dict.append({
  736. 'name' : layer.GetLayerData('name'),
  737. 'title' : layer.GetLayerData('title'),
  738. 'style' : self.GetPyData(s)['style'],
  739. 'cap_intf_l' : layer
  740. })
  741. return sel_layers_dict
  742. def OnListSelChanging(self, event):
  743. """!Do not allow to select items, which cannot be requested from server.
  744. """
  745. cur_item = event.GetItem ()
  746. if not self.GetPyData(cur_item)['layer'].IsRequestable():
  747. event.Veto()
  748. def GetItemCount(self):
  749. """!Required for listmix.ListCtrlAutoWidthMixin
  750. """
  751. return 0
  752. def GetCountPerPage(self):
  753. """!Required for listmix.ListCtrlAutoWidthMixin
  754. """
  755. return 0
  756. def SelectLayers(self, l_st_list):
  757. """!Select layers/styles in LayersList
  758. @param l_st_list - [{style : 'style_name', layer : 'layer_name'}, ...]
  759. @return items from l_st_list which were not found
  760. """
  761. def checknext(item, l_st_list, items_to_sel):
  762. def compare(item, l_name, st_name):
  763. it_l_name = self.GetPyData(item)['layer'].GetLayerData('name')
  764. it_st = self.GetPyData(item)['style']
  765. it_type = self.GetPyData(item)['type']
  766. if it_l_name == l_name and \
  767. ( (not it_st and not st_name) or \
  768. (it_st and it_st['name'] == st_name and it_type == 'style')):
  769. return True
  770. return False
  771. for i, l_st in enumerate(l_st_list):
  772. l_name = l_st['layer']
  773. st_name = l_st['style']
  774. if compare(item, l_name, st_name):
  775. items_to_sel[i] = [item, l_st]
  776. break
  777. if len(items_to_sel) == len(l_st_list):
  778. item = self.GetNext(item)
  779. if not item.IsOk():
  780. return
  781. checknext(item, l_st_list, items_to_sel)
  782. self.UnselectAll()
  783. l_st_list = deepcopy(l_st_list)
  784. root_item = self.GetRootItem()
  785. items_to_sel = [None] * len(l_st_list)
  786. checknext(root_item, l_st_list, items_to_sel)
  787. # items are selected according to position in l_st_list
  788. # to be added to Layers order list in right order
  789. for i in items_to_sel:
  790. if not i:
  791. continue
  792. item, l_st = i
  793. un_o = True
  794. if self.HasFlag(wx.TR_MULTIPLE):
  795. un_o = False
  796. self.SelectItem(item, unselect_others = un_o)
  797. l_st_list.remove(l_st)
  798. return l_st_list