/[thuban]/branches/WIP-pyshapelib-bramz/Thuban/UI/renderer.py
ViewVC logotype

Diff of /branches/WIP-pyshapelib-bramz/Thuban/UI/renderer.py

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

revision 882 by jonathan, Fri May 9 16:34:28 2003 UTC revision 1324 by frank, Tue Jul 1 09:28:47 2003 UTC
# Line 2  Line 2 
2  # Authors:  # Authors:
3  # Bernhard Herzog <[email protected]>  # Bernhard Herzog <[email protected]>
4  # Jonathan Coles <[email protected]>  # Jonathan Coles <[email protected]>
5    # Frank Koormann <[email protected]>
6  #  #
7  # This program is free software under the GPL (>=v2)  # This program is free software under the GPL (>=v2)
8  # Read the file COPYING coming with Thuban for details.  # Read the file COPYING coming with Thuban for details.
9    
10  __version__ = "$Revision$"  __version__ = "$Revision$"
11    
12    import cStringIO
13    
14  from Thuban import _  from Thuban import _
15    
16  from wxPython.wx import wxPoint, wxPen, wxBrush, wxFont, \  from wxPython.wx import wxMemoryDC, wxEmptyBitmap, \
17       wxTRANSPARENT_PEN, wxTRANSPARENT_BRUSH, \      wxPoint, wxRect, wxPen, wxBrush, wxFont, \
18       wxBLACK, wxSOLID, wxCROSS_HATCH, wxSWISS, wxNORMAL      wxTRANSPARENT_PEN, wxTRANSPARENT_BRUSH, \
19        wxBLACK_PEN, wxRED_PEN, wxBLACK, \
20        wxSOLID, wxCROSS_HATCH, wxSWISS, wxNORMAL, \
21        wxBitmap, wxImageFromBitmap, wxBitmapFromImage, \
22        wxImageFromStream, wxBITMAP_TYPE_BMP
23    
24  from wxproj import draw_polygon_shape, draw_polygon_init  from wxproj import draw_polygon_shape, draw_polygon_init
25    
26  from Thuban.UI.common import Color2wxColour  from Thuban.UI.common import Color2wxColour
27    from Thuban.UI.classifier import ClassDataPreviewer
28    from Thuban.UI.scalebar import ScaleBar
29    
30  from Thuban.Model.layer import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \  from Thuban.Model.layer import Layer, RasterLayer, \
31       SHAPETYPE_POINT       SHAPETYPE_POLYGON, SHAPETYPE_ARC, SHAPETYPE_POINT
32  from Thuban.Model.label import ALIGN_CENTER, ALIGN_TOP, ALIGN_BOTTOM, \  from Thuban.Model.label import ALIGN_CENTER, ALIGN_TOP, ALIGN_BOTTOM, \
33       ALIGN_LEFT, ALIGN_RIGHT, ALIGN_BASELINE       ALIGN_LEFT, ALIGN_RIGHT, ALIGN_BASELINE
34    
35  from Thuban.Model.classification import Classification  from Thuban.Model.classification import Classification
36  from Thuban.Model.color import Color  from Thuban.Model.color import Color
37    import Thuban.Model.resource
38    
39    if Thuban.Model.resource.has_gdal_support():
40        from gdalwarp import ProjectRasterFile
41    
42  class MapRenderer:  class MapRenderer:
43    
# Line 60  class MapRenderer: Line 73  class MapRenderer:
73    
74      def render_map(self, map):      def render_map(self, map):
75          self.map = map          self.map = map
76            seenRaster = True
77    
78            if self.scale == 0:
79                return
80    
81            #
82            # This is only a good optimization if there is only one
83            # raster layer and the image covers the entire window (as
84            # it currently does). We note if there is a raster layer
85            # and only begin drawing layers once we have drawn it.
86            # That way we avoid drawing layers that won't be seen.
87            #
88            if Thuban.Model.resource.has_gdal_support():
89                for layer in map.Layers():
90                    if isinstance(layer, RasterLayer) and layer.Visible():
91                        seenRaster = False
92                        break
93    
94          for layer in map.Layers():          for layer in map.Layers():
95              # if honor_visibility is true, only draw visible layers,              # if honor_visibility is true, only draw visible layers,
96              # otherwise draw all layers              # otherwise draw all layers
97              if not self.honor_visibility or layer.Visible():              if not self.honor_visibility or layer.Visible():
98                  self.draw_shape_layer(layer)                  if isinstance(layer, Layer) and seenRaster:
99                        self.draw_shape_layer(layer)
100                    elif isinstance(layer, RasterLayer) \
101                        and Thuban.Model.resource.has_gdal_support():
102                        self.draw_raster_layer(layer)
103                        seenRaster = True
104    
105          self.draw_label_layer(map.LabelLayer())          self.draw_label_layer(map.LabelLayer())
106    
107      def draw_shape_layer(self, layer):      def draw_shape_layer(self, layer):
# Line 95  class MapRenderer: Line 132  class MapRenderer:
132          else:          else:
133              draw_func = lambda i: \              draw_func = lambda i: \
134                     self.draw_polygon_shape(polygon_render_param, i, pen, brush)                     self.draw_polygon_shape(polygon_render_param, i, pen, brush)
135                
136            table = layer.ShapeStore().Table()
137          for i in self.layer_ids(layer):          for i in self.layer_ids(layer):
138    
139              if field is None:              if field is None:
140                  group = defaultGroup                  group = defaultGroup
141              else:              else:
142                  record = layer.table.ReadRowAsDict(i)                  record = table.ReadRowAsDict(i)
143                  assert record is not None                  assert record is not None
144                  group = lc.FindGroup(record[field])                  group = lc.FindGroup(record[field])
145    
# Line 145  class MapRenderer: Line 183  class MapRenderer:
183    
184              draw_func(i)              draw_func(i)
185    
186        def draw_raster_layer(self, layer):
187            data = None
188            offx, offy = self.offset
189            width, height = self.dc.GetSizeTuple()
190    
191            inProj = ""
192            proj = layer.GetProjection()
193            if proj is not None:
194                for p in proj.GetAllParameters():
195                    inProj += "+" + p + " "
196    
197            outProj = ""
198            proj = self.map.GetProjection()
199            if proj is not None:
200                for p in proj.GetAllParameters():
201                    outProj += "+" + p + " "
202    
203            xmin = (0 - offx) / self.scale
204            ymin = (offy - height) / self.scale
205            xmax = (width - offx) / self.scale
206            ymax = (offy - 0) / self.scale
207    
208            try:
209                data = ProjectRasterFile(
210                    layer.GetImageFilename(),
211                    inProj,
212                    outProj,
213                    (xmin, ymin, xmax, ymax),
214                    "", (width, height))
215            except IOError, (strerr):
216                print strerr
217            except (AttributeError, ValueError):
218                pass
219            else:
220                if data is not None:
221                    stream = cStringIO.StringIO(data)
222                    image = wxImageFromStream(stream, wxBITMAP_TYPE_BMP)
223                    bitmap = wxBitmapFromImage(image)
224                    self.dc.BeginDrawing()
225                    self.dc.DrawBitmap(bitmap, 0, 0)
226                    self.dc.EndDrawing()
227    
228      def layer_ids(self, layer):      def layer_ids(self, layer):
229          """Return the shape ids of the given layer that have to be drawn.          """Return the shape ids of the given layer that have to be drawn.
230    
# Line 156  class MapRenderer: Line 236  class MapRenderer:
236      def polygon_render_param(self, layer):      def polygon_render_param(self, layer):
237          """Return the low-lever render parameter for the layer"""          """Return the low-lever render parameter for the layer"""
238          offx, offy = self.offset          offx, offy = self.offset
239          return draw_polygon_init(layer.shapefile, self.dc,          return draw_polygon_init(layer.ShapeStore().Shapefile(), self.dc,
240                                   self.map.projection,                                   self.map.projection,
241                                   layer.projection,                                   layer.projection,
242                                   self.scale, -self.scale,                                   self.scale, -self.scale,
# Line 326  class ScreenRenderer(MapRenderer): Line 406  class ScreenRenderer(MapRenderer):
406          return layer.ShapesInRegion((left, bottom, right, top))          return layer.ShapesInRegion((left, bottom, right, top))
407    
408    
409  class PrinterRender(MapRenderer):  class ExportRenderer(ScreenRenderer):
410    
411      # When printing we want to see all layers      honor_visibility = 1
     honor_visibility = 0  
   
     RenderMap = MapRenderer.render_map  
412            
413        def RenderMap(self, map, region, mapregion,
414                      selected_layer, selected_shapes ):
415            """Render the map.
416    
417            The rendering device has been specified during initialisation.
418            The device border distance was set in Thuban.UI.view.OutputTranform().
419            
420            RenderMap renders a frame set (one page frame, one around
421            legend/scalebar and one around the map), the map, the legend and the
422            scalebar on the given DC. The map is rendered with the region displayed
423            in the canvas view, centered on the area available for map display.
424            """
425            
426            self.update_region = region
427            self.selected_layer = selected_layer
428            self.selected_shapes = selected_shapes
429    
430            # Get some dimensions
431            llx, lly, urx, ury = region
432            self.mapregion = mapregion
433            mminx, mminy, mmaxx, mmaxy = self.mapregion
434    
435            # Manipulate the offset to position the map
436            offx, offy = self.offset
437            # 1. Shift to corner of map drawing area
438            offx = offx + mminx
439            offy = offy + mminy
440    
441            # 2. Center the map on the map drawing area:
442            # region identifies the region on the canvas view:
443            # center of map drawing area - half the size of region: rendering origin
444            self.shiftx = (mmaxx - mminx)*0.5 - (urx - llx)*0.5
445            self.shifty = (mmaxy - mminy)*0.5 - (ury - lly)*0.5
446    
447            self.offset = (offx+self.shiftx, offy+self.shifty)
448    
449            # Draw the map
450            self.dc.BeginDrawing()
451            self.dc.DestroyClippingRegion()
452            self.dc.SetClippingRegion(mminx+self.shiftx, mminy+self.shifty,
453                                      urx, ury)
454            self.render_map(map)
455            self.dc.EndDrawing()
456    
457            # Draw the rest (frames, legend, scalebar)
458            self.dc.BeginDrawing()
459            self.dc.DestroyClippingRegion()
460    
461            # Force the font for Legend drawing
462            font = wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)
463            self.dc.SetFont(font)
464    
465            self.render_frame(region)
466            self.render_legend(map)
467            self.render_scalebar(map)
468            self.dc.EndDrawing()
469    
470        def render_frame(self, region):
471            """Render the frames for map and legend/scalebar."""
472    
473            dc = self.dc
474            dc.SetPen(wxBLACK_PEN)
475            dc.SetBrush(wxTRANSPARENT_BRUSH)
476    
477            # Dimension stuff
478            width, height = dc.GetSizeTuple()
479            mminx, mminy, mmaxx, mmaxy = self.mapregion
480    
481            # Page Frame
482            dc.DrawRectangle(15,15,width-30, (mmaxy-mminy)+10)
483            
484            # Map Frame
485            llx, lly, urx, ury = region
486            dc.DrawRectangle(mminx + self.shiftx, mminy + self.shifty, urx, ury)
487            
488            # Legend Frame
489            dc.DrawRectangle(mmaxx+10,mminy,(width-20) - (mmaxx+10), mmaxy-mminy)
490    
491            dc.DestroyClippingRegion()
492            dc.SetClippingRegion(mmaxx+10,mminy,
493                                 (width-20) - (mmaxx+10), mmaxy-mminy)
494    
495        def render_legend(self, map):
496            """Render the legend on the Map."""
497    
498            previewer = ClassDataPreviewer()
499            dc = self.dc
500            dc.SetPen(wxBLACK_PEN)
501            dc.SetBrush(wxTRANSPARENT_BRUSH)
502    
503            # Dimension stuff
504            width, height = dc.GetSizeTuple()
505            mminx, mminy, mmaxx, mmaxy = self.mapregion
506            textwidth, textheight = dc.GetTextExtent("0")
507            iconwidth  = textheight
508            iconheight = textheight
509            stepy = textheight+3
510            dx = 10
511            posx = mmaxx + 10 + 5   # 10 pix distance mapframe/legend frame,
512                                    # 5 pix inside legend frame
513            posy = mminy + 5        # 5 pix inside legend frame
514            
515            # Render the legend
516            dc.SetTextForeground(wxBLACK)
517            if map.HasLayers():
518                layers = map.Layers()
519                layers.reverse()
520                for l in layers:
521                    if l.Visible():
522                        # Render title
523                        dc.DrawText(l.Title(), posx, posy)
524                        posy+=stepy
525                        if l.HasClassification():
526                            # Render classification
527                            clazz = l.GetClassification()
528                            shapeType = l.ShapeType()
529                            for g in clazz:
530                                if g.IsVisible():
531                                    previewer.Draw(dc,
532                                        wxRect(posx+dx, posy,
533                                               iconwidth, iconheight),
534                                        g.GetProperties(), shapeType)
535                                    dc.DrawText(g.GetDisplayText(),
536                                                posx+2*dx+iconwidth, posy)
537                                    posy+=stepy
538            
539        def render_scalebar(self, map):
540            """Render the scalebar."""
541    
542            scalebar = ScaleBar(map)
543    
544            # Dimension stuff
545            width, height = self.dc.GetSizeTuple()
546            mminx, mminy, mmaxx, mmaxy = self.mapregion
547          
548            # Render the scalebar
549            scalebar.DrawScaleBar(self.scale, self.dc,
550                                 (mmaxx+10+5, mmaxy-25),
551                                 ((width-15-5) - (mmaxx+10+5),20)
552                                )
553            # 10 pix between map and legend frame, 5 pix inside legend frame
554            # 25 pix from the legend frame bottom line
555            # Width: 15 pix from DC border, 5 pix inside frame, 10, 5 as above
556            # Height: 20
557    
558    class PrinterRenderer(ExportRenderer):
559    
560        # Printing as well as Export / Screen display only the visible layer.
561        honor_visibility = 1
562    
Legend:
Removed from v.882  
changed lines
  Added in v.1324
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26