/[thuban]/branches/WIP-pyshapelib-bramz/Extensions/wms/wms.py
ViewVC logotype

Diff of /branches/WIP-pyshapelib-bramz/Extensions/wms/wms.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 2086 by bh, Fri Feb 20 20:09:25 2004 UTC revision 2172 by joey, Tue Apr 13 17:38:26 2004 UTC
# Line 37  from Thuban.Model.resource import get_sy Line 37  from Thuban.Model.resource import get_sy
37       EPSG_DEPRECATED_PROJ_FILE       EPSG_DEPRECATED_PROJ_FILE
38  from Thuban.UI.command import registry, Command  from Thuban.UI.command import registry, Command
39  import Thuban.UI.mainwindow  import Thuban.UI.mainwindow
40    from Thuban.UI.common import ThubanBeginBusyCursor, ThubanEndBusyCursor
41  from Thuban import _  from Thuban import _
42  import Thuban.UI.baserenderer  import Thuban.UI.baserenderer
43    
44    from capabilities import WMSCapabilities
45    
46  def epsg_code_to_projection(epsg):  def epsg_code_to_projection(epsg):
47      """Find the projection for the given epsg code.      """Find the projection for the given epsg code.
# Line 62  class WMSExtension(Extension): Line 64  class WMSExtension(Extension):
64          return (_("Extension: %s") % self.title,          return (_("Extension: %s") % self.title,
65                  [ object.TreeInfo() for object in self.objects ])                  [ object.TreeInfo() for object in self.objects ])
66    
67  class WMSLayer(BaseLayer, WMSClient):  class WMSLayer(BaseLayer):
68    
69      def __init__(self, title, url):      def __init__(self, title, url):
70          """Initializes the WMSLayer.          """Initializes the WMSLayer.
# Line 79  class WMSLayer(BaseLayer, WMSClient): Line 81  class WMSLayer(BaseLayer, WMSClient):
81          self.latlonbbox = None          self.latlonbbox = None
82          self.error_msg = None          self.error_msg = None
83          self.layer_name = None          self.layer_name = None
84            self.capabilities = None
85    
86          wms_response = self.getCapabilities(self.url, '1.0')          # Change the cursor to demonstrate that we're busy but working
87          self._capa_dom = xml.dom.minidom.parseString(wms_response)          ThubanBeginBusyCursor()
88            self.capabilities = WMSCapabilities(url)
89          root = self._capa_dom.documentElement          ThubanEndBusyCursor()
90          cap = root.getElementsByTagName('Capability')[0]  
91          layer = cap.getElementsByTagName('Layer')[0]          # name of the top layer of the remote map
92            foo = self.capabilities.getLayers()
93          # get projected bounding box and latlon bounding box          if len(foo) == 0:
94          for node in layer.childNodes:              self.error_msg = _('No layers found in remote resource:\n'\
95              if node.nodeName == 'BoundingBox':                                 '%s') % url
96                  minx = node.attributes.get('minx').nodeValue              return
97                  miny = node.attributes.get('miny').nodeValue          top_layer = foo[0]
98                  maxx = node.attributes.get('maxx').nodeValue          self.layer_name = top_layer
99                  maxy = node.attributes.get('maxy').nodeValue  
100                  self.bbox = (float(minx), float(miny), float(maxx), float(maxy))          # first projection of the top layer
101          bbox = layer.getElementsByTagName('LatLonBoundingBox')[0]          foo = self.capabilities.getLayerSRS(top_layer)
102          self.layer_name = layer.getElementsByTagName('Name')[0].childNodes[0].data          if len(foo) == 0:
103          minx = bbox.attributes.get('minx').nodeValue              self.error_msg = _('No LatLonBoundingBox found for top layer %s')\
104          miny = bbox.attributes.get('miny').nodeValue                               % top_layer
105          maxx = bbox.attributes.get('maxx').nodeValue              return
106          maxy = bbox.attributes.get('maxy').nodeValue          top_srs = foo[0]
107          self.latlonbbox = (float(minx), float(miny), float(maxx), float(maxy))  
108            # BoundingBox of the top layer
109            bbox = self.capabilities.getLayerBBox(top_layer, top_srs)
110            if len(bbox) == 0:
111                self.error_msg = _('No BoundingBox found for layer %s and EPSG:')\
112                                 % (top_layer, top_srs)
113                return
114            self.bbox = (float(bbox['minx']),
115                         float(bbox['miny']),
116                         float(bbox['maxx']),
117                         float(bbox['maxy']))
118    
119            # LatLonBox of the top layer
120            bbox = self.capabilities.getLayerLatLonBBox(top_layer)
121            self.latlonbbox = (float(bbox['minx']),
122                               float(bbox['miny']),
123                               float(bbox['maxx']),
124                               float(bbox['maxy']))
125    
126          # get projection          # get projection
127          srs = layer.getElementsByTagName('SRS')[0].childNodes[0].data          p = epsg_code_to_projection(top_srs)
         if len(srs.split(':')) == 1:  
             epsg_id = srs  
         else:  
             epsg_id = srs.split(':')[1]  
   
         p = epsg_code_to_projection(epsg_id)  
128          self.SetProjection(p)          self.SetProjection(p)
129    
130          if p is None:          if p is None:
131              self.error_msg = _('EPSG projection code %s not found!\n'\              self.error_msg = _('EPSG projection code %s not found!\n'\
132                                 'Setting projection to "None".\n'\                                 'Setting projection to "None".\n'\
133                                 'Please set an appropriate projection yourself.'\                                 'Please set an appropriate projection yourself.'\
134                                 % epsg_id)                                 % top_srs)
135    
136            # pre-determine the used format
137            self.wmsformat, self.format = \
138                self.calcFormat(self.capabilities.getFormats())
139            if self.wmsformat is None:
140                self.error_msg = \
141                    _('No supported image format found in remote resource')
142                return
143    
144          # get title          # get and set the title
145          title = layer.getElementsByTagName('Title')[0].childNodes[0].data          self.SetTitle(self.capabilities.getTitle().encode('latin1', 'replace'))
         self.SetTitle(title.encode('latin1', 'replace'))  
146    
         self._capa_dom.unlink()  
147    
148      def LatLongBoundingBox(self):      def LatLongBoundingBox(self):
149          """Return the layer's bounding box in lat-lon.          """Return the layer's bounding box in lat-lon.
# Line 135  class WMSLayer(BaseLayer, WMSClient): Line 155  class WMSLayer(BaseLayer, WMSClient):
155          """          """
156          return self.bbox          return self.bbox
157    
158    
159        def getFormat(self, format):
160            """
161            Return the image format for the render engine
162    
163            format -- format as returned by the WMS server
164    
165            If no mapping was found, None is returned
166    
167            An exception rule is implemented in order to not accept
168            image/wbmp or WBMP which refers to WAP bitmap format and is
169            not supported by the included render engine.
170            """
171            fmap = {'png' : "PNG",
172                    'jpeg': "JPEG",
173                    'jpg' : "JPEG",
174                    'tif' : "TIFF",
175                    'gif' : "GIF",
176                    'wbmp': None,
177                    'bmp' : "BMP"}
178    
179            for f in fmap.keys():
180                if format.lower().find(f) > -1:
181                        return fmap[f]
182            return None
183    
184            
185        def calcFormat(self, formats):
186            """
187            Calculate the preferred image format
188    
189            formats -- list of formates as returned by the WMS server
190    
191            The following priority is used:
192            - PNG
193            - JPEG
194            - TIFF
195            - GIF
196            - BMP
197    
198            If no matching format was found, None, None will be returned.
199    
200            An exception rule is implemented in order to not accept
201            image/wbmp or WBMP which refers to WAP bitmap format and is
202            not supported by the included render engine.
203            """
204            prio = ['jpeg', 'bmp']
205            for p in prio:
206                for f in formats:
207                    if f.lower().find(p) > -1:
208                        if f.lower().find('wbmp') == -1:
209                            return f, self.getFormat(f)
210            return None, None
211            
212    
213      def GetMapImg(self, width, height, bbox):      def GetMapImg(self, width, height, bbox):
214          bbox_dict = { 'minx': bbox[0], 'miny': bbox[1],          bbox_dict = { 'minx': bbox[0], 'miny': bbox[1],
215                        'maxx': bbox[2], 'maxy': bbox[3] }                        'maxx': bbox[2], 'maxy': bbox[3] }
216    
217            # Change the cursor to demonstrate that we're busy but working
218            ThubanBeginBusyCursor()
219    
220            wmsclient = WMSClient()
221    
222          epsg_id = int(self.GetProjection().EPSGCode())          epsg_id = int(self.GetProjection().EPSGCode())
         wms_response = self.getMap(self.url, 'JPEG', width, height,  
                                    epsg_id, bbox_dict,  
                                    [self.layer_name], version = '1.0')  
         return wms_response  
223    
224            wms_response = wmsclient.getMap(self.url, self.wmsformat, width, height,
225                                       epsg_id, bbox_dict,
226                                       [self.layer_name], version = self.capabilities.getVersion())
227            ThubanEndBusyCursor()
228            return wms_response, self.format
229    
230    
231  def render_wms_layer(renderer, layer):  def render_wms_layer(renderer, layer):
# Line 156  def render_wms_layer(renderer, layer): Line 238  def render_wms_layer(renderer, layer):
238      xmax = (width - offx) / scale      xmax = (width - offx) / scale
239      ymax = (offy - 0) / scale      ymax = (offy - 0) / scale
240    
241      img = layer.GetMapImg(width, height, (xmin, ymin, xmax, ymax))      img, format = layer.GetMapImg(width, height, (xmin, ymin, xmax, ymax))
242      renderer.draw_raster_data(img, "JPEG")      renderer.draw_raster_data(img, format)
243    
244      return ()      return ()
245    
# Line 219  def wms_dialog(context): Line 301  def wms_dialog(context):
301    
302      if dialog.ShowModal() == wxID_OK:      if dialog.ShowModal() == wxID_OK:
303          url = dialog.url          url = dialog.url
304            if len(url) == 0:
305                url = None
306      else:      else:
307          url = None          url = None
308      dialog.Destroy()      dialog.Destroy()
Legend:
Removed from v.2086  
changed lines
  Added in v.2172
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26