/[thuban]/branches/WIP-pyshapelib-bramz/Thuban/Model/load.py
ViewVC logotype

Diff of /branches/WIP-pyshapelib-bramz/Thuban/Model/load.py

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

revision 390 by jonathan, Mon Feb 10 15:25:49 2003 UTC revision 1159 by jonathan, Thu Jun 12 12:40:27 2003 UTC
# Line 1  Line 1 
1  # Copyright (C) 2001, 2002 by Intevation GmbH  # Copyright (C) 2001, 2002, 2003 by Intevation GmbH
2  # Authors:  # Authors:
3  # Jan-Oliver Wagner <[email protected]>  # Jan-Oliver Wagner <[email protected]>
4  # Bernhard Herzog <[email protected]>  # Bernhard Herzog <[email protected]>
5    # Jonathan Coles <[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 GRASS for details.  # Read the file COPYING coming with GRASS for details.
# Line 12  Parser for thuban session files. Line 13  Parser for thuban session files.
13    
14  __version__ = "$Revision$"  __version__ = "$Revision$"
15    
16  import sys, string, os  import string, os
17    
18  import xml.sax  import xml.sax
19  import xml.sax.handler  import xml.sax.handler
20  from xml.sax import make_parser, ErrorHandler  from xml.sax import make_parser, ErrorHandler, SAXNotRecognizedException
21    
22  from Thuban import _  from Thuban import _
23    
24    from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \
25         FIELDTYPE_STRING
26    
27  from Thuban.Model.session import Session  from Thuban.Model.session import Session
28  from Thuban.Model.map import Map  from Thuban.Model.map import Map
29  from Thuban.Model.layer import Layer  from Thuban.Model.layer import Layer, RasterLayer
30  from Thuban.Model.color import Color  from Thuban.Model.color import Color
31  from Thuban.Model.proj import Projection  from Thuban.Model.proj import Projection
32  from Thuban.Model.classification import Classification, ClassData  from Thuban.Model.range import Range
33    from Thuban.Model.classification import Classification, \
34        ClassGroupDefault, ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \
35        ClassGroupProperties
36    
37    from Thuban.Model.xmlreader import XMLReader
38    import resource
39    
40  def parse_color(color):  def parse_color(color):
41      """Return the color object for the string color.      """Return the color object for the string color.
# Line 35  def parse_color(color): Line 45  def parse_color(color):
45      """      """
46      color = string.strip(color)      color = string.strip(color)
47      if color == "None":      if color == "None":
48          result = None          result = Color.Transparent
49      elif color[0] == '#':      elif color[0] == '#':
50          if len(color) == 7:          if len(color) == 7:
51              r = string.atoi(color[1:3], 16) / 255.0              r = string.atoi(color[1:3], 16) / 255.0
# Line 49  def parse_color(color): Line 59  def parse_color(color):
59          raise ValueError(_("Invalid color specification %s") % color)          raise ValueError(_("Invalid color specification %s") % color)
60      return result      return result
61    
62    class SessionLoader(XMLReader):
63    
64  class ProcessSession(xml.sax.handler.ContentHandler):      def __init__(self):
65            """Inititialize the Sax handler."""
66            XMLReader.__init__(self)
67    
     # Dictionary mapping element names (or (URI, element name) pairs for  
     # documents using namespaces) to method names. The methods should  
     # accept the same parameters as the startElement (or startElementNS)  
     # methods. The start_dispatcher is used by the default startElement  
     # and startElementNS methods to call a method for the open tag of an  
     # element.  
     start_dispatcher = {}  
   
     # end_dispatcher works just like start_dispatcher but it's used by  
     # endElement and endElementNS. The method whose names it maps to  
     # should accept the same parameters as endElement and endElementNS.  
     end_dispatcher = {}  
   
   
     def __init__(self, directory):  
         """Inititialize the Sax handler.  
   
         The directory parameter should be the directory containing the  
         session file. It's needed to interpret embedded relative  
         filenames.  
         """  
         self.directory = directory  
         self.chars = ''  
68          self.theSession = None          self.theSession = None
69          self.aMap = None          self.aMap = None
70          self.aLayer = None          self.aLayer = None
71    
72      def startElementNS(self, name, qname, attrs):          XMLReader.AddDispatchers(self,
73          """Call the method given for name in self.start_dispatcher              {'session'       : ("start_session",        "end_session"),
74          """               'map'           : ("start_map",            "end_map"),
75          if name[0] is None:               'projection'    : ("start_projection",     "end_projection"),
76              method_name = self.start_dispatcher.get(name[1])               'parameter'     : ("start_parameter",      None),
77          else:               'layer'         : ("start_layer",          "end_layer"),
78              # Dispatch with namespace               'rasterlayer'   : ("start_rasterlayer",    "end_rasterlayer"),
79              method_name = self.start_dispatcher.get(name)               'classification': ("start_classification", "end_classification"),
80          if method_name is not None:               'clnull'        : ("start_clnull",         "end_clnull"),
81              getattr(self, method_name)(name, qname, attrs)               'clpoint'       : ("start_clpoint",        "end_clpoint"),
82                 'clrange'       : ("start_clrange",        "end_clrange"),
83      def endElementNS(self, name, qname):               'cldata'        : ("start_cldata",         "end_cldata"),
84          """Call the method given for name in self.end_dispatcher               'table'         : ("start_table",          "end_table"),
85          """               'labellayer'    : ("start_labellayer",     None),
86          if name[0] is None:               'label'         : ("start_label",          None)})
             method_name = self.end_dispatcher.get(name[1])  
         else:  
             # Dispatch with namespace  
             method_name = self.end_dispatcher.get(name)  
         if method_name is not None:  
             getattr(self, method_name)(name, qname)  
87    
88      def start_session(self, name, qname, attrs):      def start_session(self, name, qname, attrs):
89          self.theSession = Session(attrs.get((None, 'title'), None))          self.theSession = Session(self.encode(attrs.get((None, 'title'), None)))
     start_dispatcher['session'] = "start_session"  
90    
91      def end_session(self, name, qname):      def end_session(self, name, qname):
92          pass          pass
     end_dispatcher['session'] = "end_session"  
93    
94      def start_map(self, name, qname, attrs):      def start_map(self, name, qname, attrs):
95          """Start a map."""          """Start a map."""
96          self.aMap = Map(attrs.get((None, 'title'), None))          self.aMap = Map(attrs.get((None, 'title'), None))
     start_dispatcher['map'] = "start_map"  
97    
98      def end_map(self, name, qname):      def end_map(self, name, qname):
99          self.theSession.AddMap(self.aMap)          self.theSession.AddMap(self.aMap)
100      end_dispatcher['map'] = "end_map"          self.aMap = None
101    
102      def start_projection(self, name, qname, attrs):      def start_projection(self, name, qname, attrs):
103            self.ProjectionName = self.encode(attrs.get((None, 'name'), None))
104          self.ProjectionParams = [ ]          self.ProjectionParams = [ ]
     start_dispatcher['projection'] = "start_projection"  
105    
106      def end_projection(self, name, qname):      def end_projection(self, name, qname):
107          self.aMap.SetProjection(Projection(self.ProjectionParams))          if self.aLayer is not None:
108      end_dispatcher['projection'] = "end_projection"              obj = self.aLayer
109            elif self.aMap is not None:
110                obj = self.aMap
111            else:
112                assert False, "projection tag out of context"
113                pass
114    
115            obj.SetProjection(
116                Projection(self.ProjectionParams, self.ProjectionName))
117    
118      def start_parameter(self, name, qname, attrs):      def start_parameter(self, name, qname, attrs):
119          s = attrs.get((None, 'value'))          s = attrs.get((None, 'value'))
120          s = str(s) # we can't handle unicode in proj          s = str(s) # we can't handle unicode in proj
121          self.ProjectionParams.append(s)          self.ProjectionParams.append(s)
     start_dispatcher['parameter'] = "start_parameter"  
122    
123      def start_layer(self, name, qname, attrs, layer_class = Layer):      def start_layer(self, name, qname, attrs, layer_class = Layer):
124          """Start a layer          """Start a layer
# Line 139  class ProcessSession(xml.sax.handler.Con Line 127  class ProcessSession(xml.sax.handler.Con
127          attrs which may be a dictionary as well as the normal SAX attrs          attrs which may be a dictionary as well as the normal SAX attrs
128          object and bind it to self.aLayer.          object and bind it to self.aLayer.
129          """          """
130          title = attrs.get((None, 'title'), "")          title = self.encode(attrs.get((None, 'title'), ""))
131          filename = attrs.get((None, 'filename'), "")          filename = attrs.get((None, 'filename'), "")
132          filename = os.path.join(self.directory, filename)          filename = os.path.join(self.GetDirectory(), filename)
133            filename = self.encode(filename)
134            visible  = self.encode(attrs.get((None, 'visible'), "true")) != "false"
135          fill = parse_color(attrs.get((None, 'fill'), "None"))          fill = parse_color(attrs.get((None, 'fill'), "None"))
136          stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))          stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))
137          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))
138          self.aLayer = layer_class(title, filename, fill = fill,          self.aLayer = layer_class(title,
139                                    stroke = stroke, stroke_width = stroke_width)                                    self.theSession.OpenShapefile(filename),
140      start_dispatcher['layer'] = "start_layer"                                    fill = fill, stroke = stroke,
141                                      lineWidth = stroke_width,
142                                      visible = visible)
143    
144      def end_layer(self, name, qname):      def end_layer(self, name, qname):
145          self.aMap.AddLayer(self.aLayer)          self.aMap.AddLayer(self.aLayer)
146      end_dispatcher['layer'] = "end_layer"          self.aLayer = None
147    
148        def start_rasterlayer(self, name, qname, attrs, layer_class = RasterLayer):
149            title = self.encode(attrs.get((None, 'title'), ""))
150            filename = attrs.get((None, 'filename'), "")
151            filename = os.path.join(self.GetDirectory(), filename)
152            filename = self.encode(filename)
153            visible  = self.encode(attrs.get((None, 'visible'), "true")) != "false"
154    
155            self.aLayer = layer_class(title, filename, visible = visible)
156    
157        def end_rasterlayer(self, name, qname):
158            self.aMap.AddLayer(self.aLayer)
159            self.aLayer = None
160    
161      def start_classification(self, name, qname, attrs):      def start_classification(self, name, qname, attrs):
162          self.aLayer.classification.SetField(attrs.get((None, 'field'), None))          field = attrs.get((None, 'field'), None)
163      start_dispatcher['classification'] = "start_classification"  
164            fieldType = attrs.get((None, 'field_type'), None)
165            dbFieldType = self.aLayer.GetFieldType(field)
166    
167            if fieldType != dbFieldType:
168                raise ValueError(_("xml field type differs from database!"))
169    
170            # setup conversion routines depending on the kind of data
171            # we will be seeing later on
172            if fieldType == FIELDTYPE_STRING:
173                self.conv = str
174            elif fieldType == FIELDTYPE_INT:
175                self.conv = lambda p: int(float(p))
176            elif fieldType == FIELDTYPE_DOUBLE:
177                self.conv = float
178    
179            self.aLayer.GetClassification().SetField(field)
180    
181      def end_classification(self, name, qname):      def end_classification(self, name, qname):
182          pass          pass
     end_dispatcher['classification'] = "end_classification"  
183    
184      def start_clnull(self, name, qname, attrs):      def start_clnull(self, name, qname, attrs):
185          self.cl_data = ClassData()          self.cl_group = ClassGroupDefault()
186      start_dispatcher['clnull'] = "start_clnull"          self.cl_group.SetLabel(self.encode(attrs.get((None, 'label'), "")))
187            self.cl_prop = ClassGroupProperties()
188    
189      def end_clnull(self, name, qname):      def end_clnull(self, name, qname):
190          self.aLayer.classification.SetDefaultData(self.cl_data)          self.cl_group.SetProperties(self.cl_prop)
191          del self.cl_data          self.aLayer.GetClassification().SetDefaultGroup(self.cl_group)
192      end_dispatcher['clnull'] = "end_clnull"          del self.cl_group, self.cl_prop
193    
194      def start_clpoint(self, name, qname, attrs):      def start_clpoint(self, name, qname, attrs):
195          attrib_value = attrs.get((None, 'value'), "0")          attrib_value = attrs.get((None, 'value'), "0")
196    
197          try:          value = self.conv(attrib_value)
198              self.cl_value  = int(attrib_value)  
199          except:          self.cl_group = ClassGroupSingleton(value)
200              self.cl_value  = attrib_value          self.cl_group.SetLabel(self.encode(attrs.get((None, 'label'), "")))
201            self.cl_prop = ClassGroupProperties()
202    
         self.cl_data = ClassData()  
     start_dispatcher['clpoint'] = "start_clpoint"  
203    
204      def end_clpoint(self, name, qname):      def end_clpoint(self, name, qname):
205          self.aLayer.classification.AddPoint(self.cl_value, self.cl_data)          self.cl_group.SetProperties(self.cl_prop)
206          del self.cl_value, self.cl_data          self.aLayer.GetClassification().AppendGroup(self.cl_group)
207      end_dispatcher['clpoint'] = "end_clpoint"          del self.cl_group, self.cl_prop
208    
209      def start_clrange(self, name, qname, attrs):      def start_clrange(self, name, qname, attrs):
210    
211            range = attrs.get((None, 'range'), None)
212            # for backward compatibility (min/max are not saved)
213            min   = attrs.get((None, 'min'), None)
214            max   = attrs.get((None, 'max'), None)
215    
216          try:          try:
217              self.cl_low = int(attrs.get((None, 'low'), "0"))              if range is not None:
218              self.cl_high = int(attrs.get((None, 'high'), "0"))                  self.cl_group = ClassGroupRange(Range(range))
219                elif min is not None and max is not None:
220                    self.cl_group = ClassGroupRange(self.conv(min), self.conv(max))
221                else:
222                    self.cl_group = ClassGroupRange(Range(None))
223    
224          except ValueError:          except ValueError:
225              raise ValueError(_("Classification range is not a number!"))              raise ValueError(_("Classification range is not a number!"))
226    
227          self.cl_data = ClassData()          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
228      start_dispatcher['clrange'] = "start_clrange"          self.cl_prop = ClassGroupProperties()
229    
230    
231      def end_clrange(self, name, qname):      def end_clrange(self, name, qname):
232          self.aLayer.classification.AddRange(          self.cl_group.SetProperties(self.cl_prop)
233              self.cl_low, self.cl_high, self.cl_data)          self.aLayer.GetClassification().AppendGroup(self.cl_group)
234          del self.cl_low, self.cl_high, self.cl_data          del self.cl_group, self.cl_prop
     end_dispatcher['clrange'] = "end_clrange"  
235    
236      def start_cldata(self, name, qname, attrs):      def start_cldata(self, name, qname, attrs):
237          self.cl_data.SetStroke(parse_color(attrs.get((None, 'stroke'), "None")))          self.cl_prop.SetLineColor(
238          self.cl_data.SetStrokeWidth(              parse_color(attrs.get((None, 'stroke'), "None")))
239            self.cl_prop.SetLineWidth(
240              int(attrs.get((None, 'stroke_width'), "0")))              int(attrs.get((None, 'stroke_width'), "0")))
241          self.cl_data.SetFill(parse_color(attrs.get((None, 'fill'), "None")))          self.cl_prop.SetFill(parse_color(attrs.get((None, 'fill'), "None")))
     start_dispatcher['cldata'] = "start_cldata"  
242    
243      def end_cldata(self, name, qname):      def end_cldata(self, name, qname):
244          pass          pass
     end_dispatcher['cldata'] = "end_cldata"  
245    
246      def start_table(self, name, qname, attrs):      def start_table(self, name, qname, attrs):
247          print "table title: %s" % attrs.get('title', None)          #print "table title: %s" % attrs.get('title', None)
248      start_dispatcher['table'] = "start_table"          pass
249    
250      def end_table(self, name, qname):      def end_table(self, name, qname):
251          pass          pass
     end_dispatcher['table'] = "end_table"  
252    
253      def start_labellayer(self, name, qname, attrs):      def start_labellayer(self, name, qname, attrs):
254          self.aLayer = self.aMap.LabelLayer()          self.aLayer = self.aMap.LabelLayer()
     start_dispatcher['labellayer'] = "start_labellayer"  
255    
256      def start_label(self, name, qname, attrs):      def start_label(self, name, qname, attrs):
257          x = float(attrs[(None, 'x')])          x = float(attrs[(None, 'x')])
258          y = float(attrs[(None, 'y')])          y = float(attrs[(None, 'y')])
259          text = attrs[(None, 'text')]          text = self.encode(attrs[(None, 'text')])
260          halign = attrs[(None, 'halign')]          halign = attrs[(None, 'halign')]
261          valign = attrs[(None, 'valign')]          valign = attrs[(None, 'valign')]
262          self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign)          self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign)
     start_dispatcher['label'] = "start_label"  
263    
264      def characters(self, chars):      def characters(self, chars):
265          pass          pass
# Line 241  class ProcessSession(xml.sax.handler.Con Line 267  class ProcessSession(xml.sax.handler.Con
267    
268  def load_session(filename):  def load_session(filename):
269      """Load a Thuban session from the file object file"""      """Load a Thuban session from the file object file"""
270      dir = os.path.dirname(filename)  
271      file = open(filename)      handler = SessionLoader()
272      handler = ProcessSession(dir)      handler.read(filename)
   
     parser = make_parser()  
     parser.setContentHandler(handler)  
     parser.setErrorHandler(ErrorHandler())  
     parser.setFeature(xml.sax.handler.feature_namespaces, 1)  
     parser.parse(file)  
273    
274      session = handler.theSession      session = handler.theSession
275      # Newly loaded session aren't modified      # Newly loaded session aren't modified

Legend:
Removed from v.390  
changed lines
  Added in v.1159

[email protected]
ViewVC Help
Powered by ViewVC 1.1.26