/[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 686 by bh, Wed Apr 16 13:22:25 2003 UTC revision 1343 by jonathan, Tue Jul 1 16:11:08 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  from wxPython.wx import wxPoint, wxPen, wxBrush, wxFont, \  import cStringIO
13       wxTRANSPARENT_PEN, wxTRANSPARENT_BRUSH, \  
14       wxBLACK, wxSOLID, wxCROSS_HATCH, wxSWISS, wxNORMAL  from Thuban import _
15    
16    from wxPython.wx import wxMemoryDC, wxEmptyBitmap, \
17        wxPoint, wxRect, wxPen, wxBrush, wxFont, \
18        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 import _  from Thuban.UI.common import Color2wxColour
27  from Thuban.UI.common import *  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 Transparent
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 49  class MapRenderer: Line 62  class MapRenderer:
62          """          """
63          # resolution in pixel/inch          # resolution in pixel/inch
64    
         assert scale > 0  
   
65          self.dc = dc          self.dc = dc
66          self.scale = scale          self.scale = scale
67          self.offset = offset          self.offset = offset
# Line 62  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 82  class MapRenderer: Line 117  class MapRenderer:
117          pen   = wxTRANSPARENT_PEN          pen   = wxTRANSPARENT_PEN
118    
119          old_prop = None          old_prop = None
120            old_group = None
121          lc = layer.GetClassification()          lc = layer.GetClassification()
122          field = lc.GetField()          field = lc.GetField()
123            defaultGroup = lc.GetDefaultGroup()
124    
125    
126          if shapetype != SHAPETYPE_POINT:          if shapetype != SHAPETYPE_POINT:
127              polygon_render_param = self.polygon_render_param(layer)              polygon_render_param = self.polygon_render_param(layer)
128    
129          for i in self.layer_ids(layer):          if shapetype == SHAPETYPE_POINT:
130              value = None              draw_func = lambda i: \
131                       self.draw_point_shape(layer, i)
132              if field is not None:          else:
133                  try:              draw_func = lambda i: \
134                      record = layer.table.read_record(i)                     self.draw_polygon_shape(polygon_render_param, i, pen, brush)
                     if record is not None:  
                         value = record[field]  
                 except:  
                     pass  
   
                 #  
                 # if the above statements fail 'value' should  
                 # be null, at which point this call will  
                 # at least retreive the NullData  
                 #  
135    
136                  group = lc.FindGroup(value)          table = layer.ShapeStore().Table()
137            for i in self.layer_ids(layer):
138    
139                  #prop = lc.GetProperties(value)              if field is None:
140                    group = defaultGroup
141              else:              else:
142                  group = lc.GetDefaultGroup()                  record = table.ReadRowAsDict(i)
143                    assert record is not None
144                    group = lc.FindGroup(record[field])
145    
146    
147              if not group.IsVisible():              if not group.IsVisible():
148                  continue                  continue
149    
             prop = group.GetProperties()  
150    
151              # don't recreate new objects if they are the same as before              # don't recreate new objects if they are the same as before
152              if prop != old_prop:              if group is not old_group:
153                  old_prop = prop                  old_group = group
154    
155                  if shapetype == SHAPETYPE_ARC:                  prop = group.GetProperties()
                     fill = Color.Transparent  
                 else:  
                     fill = prop.GetFill()  
   
   
                 if fill is Color.Transparent:  
                     brush = wxTRANSPARENT_BRUSH  
                 else:  
                     color = Color2wxColour(fill)  
                     brush = wxBrush(color, wxSOLID)  
   
                 stroke = prop.GetLineColor()  
                 stroke_width = prop.GetLineWidth()  
                 if stroke is Color.Transparent:  
                     pen = wxTRANSPARENT_PEN  
                 else:  
                     color = Color2wxColour(stroke)  
                     pen = wxPen(color, stroke_width, wxSOLID)  
156    
157                    if prop != old_prop:
158                        old_prop = prop
159    
160              if shapetype == SHAPETYPE_POINT:                      if shapetype == SHAPETYPE_ARC:
161                  self.dc.SetBrush(brush)                          fill = Transparent
162                  self.dc.SetPen(pen)                      else:
163                  self.draw_point_shape(layer, i)                          fill = prop.GetFill()
164              else:  
165                  self.draw_polygon_shape(polygon_render_param, i, pen, brush)  
166                        if fill is Transparent:
167                            brush = wxTRANSPARENT_BRUSH
168                        else:
169                            color = Color2wxColour(fill)
170                            brush = wxBrush(color, wxSOLID)
171    
172                        stroke = prop.GetLineColor()
173                        stroke_width = prop.GetLineWidth()
174                        if stroke is Transparent:
175                            pen = wxTRANSPARENT_PEN
176                        else:
177                            color = Color2wxColour(stroke)
178                            pen = wxPen(color, stroke_width, wxSOLID)
179    
180                        if shapetype == SHAPETYPE_POINT:
181                            self.dc.SetBrush(brush)
182                            self.dc.SetPen(pen)
183    
184                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.
# Line 161  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 171  class MapRenderer: Line 246  class MapRenderer:
246          draw_polygon_shape(draw_polygon_info, index, pen, brush)          draw_polygon_shape(draw_polygon_info, index, pen, brush)
247    
248      def projected_points(self, layer, index):      def projected_points(self, layer, index):
249          proj = self.map.projection          proj = self.map.GetProjection()
250          if proj is not None:          if proj is not None:
251              forward = proj.Forward              forward = proj.Forward
252          else:          else:
253              forward = None              forward = None
254          proj = layer.projection          proj = layer.GetProjection()
255          if proj is not None:          if proj is not None:
256              inverse = proj.Inverse              inverse = proj.Inverse
257          else:          else:
# Line 331  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):
   
     # When printing we want to see all layers  
     honor_visibility = 0  
410    
411      RenderMap = MapRenderer.render_map      honor_visibility = 1
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.686  
changed lines
  Added in v.1343
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26