/[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 1103 by jonathan, Fri May 30 06:29:25 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    from gdalwarp import ProjectRasterFile
26    
27  from Thuban.UI.common import Color2wxColour  from Thuban.UI.common import Color2wxColour
28    from Thuban.UI.classifier import ClassDataPreviewer
29    from Thuban.UI.scalebar import ScaleBar
30    
31  from Thuban.Model.layer import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \  from Thuban.Model.layer import Layer, RasterLayer, \
32       SHAPETYPE_POINT       SHAPETYPE_POLYGON, SHAPETYPE_ARC, SHAPETYPE_POINT
33  from Thuban.Model.label import ALIGN_CENTER, ALIGN_TOP, ALIGN_BOTTOM, \  from Thuban.Model.label import ALIGN_CENTER, ALIGN_TOP, ALIGN_BOTTOM, \
34       ALIGN_LEFT, ALIGN_RIGHT, ALIGN_BASELINE       ALIGN_LEFT, ALIGN_RIGHT, ALIGN_BASELINE
35    
# Line 60  class MapRenderer: Line 70  class MapRenderer:
70    
71      def render_map(self, map):      def render_map(self, map):
72          self.map = map          self.map = map
73            seenRaster = True
74    
75            if self.scale == 0:
76                return
77    
78            #
79            # This is only a good optimization if there is only one
80            # raster layer and the image covers the entire window (as
81            # it currently does).
82            #
83            for layer in map.Layers():
84                if isinstance(layer, RasterLayer):
85                    seenRaster = False
86                    break
87    
88          for layer in map.Layers():          for layer in map.Layers():
89              # if honor_visibility is true, only draw visible layers,              # if honor_visibility is true, only draw visible layers,
90              # otherwise draw all layers              # otherwise draw all layers
91              if not self.honor_visibility or layer.Visible():              if not self.honor_visibility or layer.Visible():
92                  self.draw_shape_layer(layer)                  if isinstance(layer, Layer) and seenRaster:
93                        self.draw_shape_layer(layer)
94                    elif isinstance(layer, RasterLayer):
95                        self.draw_raster_layer(layer)
96                        seenRaster = True
97    
98          self.draw_label_layer(map.LabelLayer())          self.draw_label_layer(map.LabelLayer())
99    
100      def draw_shape_layer(self, layer):      def draw_shape_layer(self, layer):
# Line 145  class MapRenderer: Line 175  class MapRenderer:
175    
176              draw_func(i)              draw_func(i)
177    
178        def draw_raster_layer(self, layer):
179            data = None
180            offx, offy = self.offset
181            width, height = self.dc.GetSizeTuple()
182    
183            inProj = ""
184            proj = layer.GetProjection()
185            if proj is not None:
186                for p in proj.GetAllParameters():
187                    inProj += "+" + p + " "
188    
189            outProj = ""
190            proj = self.map.GetProjection()
191            if proj is not None:
192                for p in proj.GetAllParameters():
193                    outProj += "+" + p + " "
194    
195            xmin = (0 - offx) / self.scale
196            ymin = (offy - height) / self.scale
197            xmax = (width - offx) / self.scale
198            ymax = (offy - 0) / self.scale
199    
200            try:
201                data = ProjectRasterFile(
202                    layer.GetImageFilename(),
203                    inProj,
204                    outProj,
205                    (xmin, ymin, xmax, ymax),
206                    "", (width, height))
207            except IOError, (strerr):
208                print strerr
209            except (AttributeError, ValueError):
210                pass
211            else:
212                if data is not None:
213                    stream = cStringIO.StringIO(data)
214                    image = wxImageFromStream(stream, wxBITMAP_TYPE_BMP)
215                    bitmap = wxBitmapFromImage(image)
216                    self.dc.BeginDrawing()
217                    self.dc.DrawBitmap(bitmap, 0, 0)
218                    self.dc.EndDrawing()
219    
220      def layer_ids(self, layer):      def layer_ids(self, layer):
221          """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.
222    
# Line 326  class ScreenRenderer(MapRenderer): Line 398  class ScreenRenderer(MapRenderer):
398          return layer.ShapesInRegion((left, bottom, right, top))          return layer.ShapesInRegion((left, bottom, right, top))
399    
400    
401  class PrinterRender(MapRenderer):  class ExportRenderer(ScreenRenderer):
   
     # When printing we want to see all layers  
     honor_visibility = 0  
402    
403      RenderMap = MapRenderer.render_map      honor_visibility = 1
404            
405        def RenderMap(self, map, region, mapregion,
406                      selected_layer, selected_shapes ):
407            """Render the map.
408    
409            The rendering device has been specified during initialisation.
410            The device border distance was set in Thuban.UI.view.OutputTranform().
411            
412            RenderMap renders a frame set (one page frame, one around
413            legend/scalebar and one around the map), the map, the legend and the
414            scalebar on the given DC. The map is rendered with the region displayed
415            in the canvas view, centered on the area available for map display.
416            """
417            
418            self.update_region = region
419            self.selected_layer = selected_layer
420            self.selected_shapes = selected_shapes
421    
422            # Get some dimensions
423            llx, lly, urx, ury = region
424            self.mapregion = mapregion
425            mminx, mminy, mmaxx, mmaxy = self.mapregion
426    
427            # Manipulate the offset to position the map
428            offx, offy = self.offset
429            # 1. Shift to corner of map drawing area
430            offx = offx + mminx
431            offy = offy + mminy
432    
433            # 2. Center the map on the map drawing area:
434            # region identifies the region on the canvas view:
435            # center of map drawing area - half the size of region: rendering origin
436            self.shiftx = (mmaxx - mminx)*0.5 - (urx - llx)*0.5
437            self.shifty = (mmaxy - mminy)*0.5 - (ury - lly)*0.5
438    
439            self.offset = (offx+self.shiftx, offy+self.shifty)
440    
441            # Draw the map
442            self.dc.BeginDrawing()
443            self.dc.DestroyClippingRegion()
444            self.dc.SetClippingRegion(mminx+self.shiftx, mminy+self.shifty,
445                                      urx, ury)
446            self.render_map(map)
447            self.dc.EndDrawing()
448    
449            # Draw the rest (frames, legend, scalebar)
450            self.dc.BeginDrawing()
451            self.dc.DestroyClippingRegion()
452    
453            # Force the font for Legend drawing
454            font = wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)
455            self.dc.SetFont(font)
456    
457            self.render_frame(region)
458            self.render_legend(map)
459            self.render_scalebar(map)
460            self.dc.EndDrawing()
461    
462        def render_frame(self, region):
463            """Render the frames for map and legend/scalebar."""
464    
465            dc = self.dc
466            dc.SetPen(wxBLACK_PEN)
467            dc.SetBrush(wxTRANSPARENT_BRUSH)
468    
469            # Dimension stuff
470            width, height = dc.GetSizeTuple()
471            mminx, mminy, mmaxx, mmaxy = self.mapregion
472    
473            # Page Frame
474            dc.DrawRectangle(15,15,width-30, (mmaxy-mminy)+10)
475            
476            # Map Frame
477            llx, lly, urx, ury = region
478            dc.DrawRectangle(mminx + self.shiftx, mminy + self.shifty, urx, ury)
479            
480            # Legend Frame
481            dc.DrawRectangle(mmaxx+10,mminy,(width-20) - (mmaxx+10), mmaxy-mminy)
482    
483            dc.DestroyClippingRegion()
484            dc.SetClippingRegion(mmaxx+10,mminy,
485                                 (width-20) - (mmaxx+10), mmaxy-mminy)
486    
487        def render_legend(self, map):
488            """Render the legend on the Map."""
489    
490            previewer = ClassDataPreviewer()
491            dc = self.dc
492            dc.SetPen(wxBLACK_PEN)
493            dc.SetBrush(wxTRANSPARENT_BRUSH)
494    
495            # Dimension stuff
496            width, height = dc.GetSizeTuple()
497            mminx, mminy, mmaxx, mmaxy = self.mapregion
498            textwidth, textheight = dc.GetTextExtent("0")
499            iconwidth  = textheight
500            iconheight = textheight
501            stepy = textheight+3
502            dx = 10
503            posx = mmaxx + 10 + 5   # 10 pix distance mapframe/legend frame,
504                                    # 5 pix inside legend frame
505            posy = mminy + 5        # 5 pix inside legend frame
506            
507            # Render the legend
508            dc.SetTextForeground(wxBLACK)
509            if map.HasLayers():
510                for l in map.Layers():
511                    if l.Visible():
512                        # Render title
513                        dc.DrawText(l.Title(), posx, posy)
514                        posy+=stepy
515                        # Render classification
516                        clazz = l.GetClassification()
517                        shapeType = l.ShapeType()
518                        for g in clazz:
519                            if g.IsVisible():
520                                previewer.Draw(dc,
521                                    wxRect(posx+dx, posy, iconwidth, iconheight),
522                                    g.GetProperties(), shapeType)
523                                dc.DrawText(g.GetDisplayText(),
524                                            posx+2*dx+iconwidth, posy)
525                                posy+=stepy
526            
527        def render_scalebar(self, map):
528            """Render the scalebar."""
529    
530            scalebar = ScaleBar(map)
531    
532            # Dimension stuff
533            width, height = self.dc.GetSizeTuple()
534            mminx, mminy, mmaxx, mmaxy = self.mapregion
535          
536            # Render the scalebar
537            scalebar.DrawScaleBar(self.scale, self.dc,
538                                 (mmaxx+10+5, mmaxy-25),
539                                 ((width-15-5) - (mmaxx+10+5),20)
540                                )
541            # 10 pix between map and legend frame, 5 pix inside legend frame
542            # 25 pix from the legend frame bottom line
543            # Width: 15 pix from DC border, 5 pix inside frame, 10, 5 as above
544            # Height: 20
545    
546    class PrinterRenderer(ExportRenderer):
547    
548        # Printing as well as Export / Screen display only the visible layer.
549        honor_visibility = 1
550    
Legend:
Removed from v.882  
changed lines
  Added in v.1103
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26