/[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 365 by jonathan, Mon Jan 27 11:47:53 2003 UTC revision 930 by jonathan, Tue May 20 15:23:03 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 _
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  from Thuban.Model.range import Range
33    from Thuban.Model.classification import Classification, \
34        ClassGroupDefault, ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \
35        ClassGroupProperties
36    
37    
38  def parse_color(color):  def parse_color(color):
# Line 34  def parse_color(color): Line 43  def parse_color(color):
43      """      """
44      color = string.strip(color)      color = string.strip(color)
45      if color == "None":      if color == "None":
46          result = None          result = Color.Transparent
47      elif color[0] == '#':      elif color[0] == '#':
48          if len(color) == 7:          if len(color) == 7:
49              r = string.atoi(color[1:3], 16) / 255.0              r = string.atoi(color[1:3], 16) / 255.0
# Line 42  def parse_color(color): Line 51  def parse_color(color):
51              b = string.atoi(color[5:7], 16) / 255.0              b = string.atoi(color[5:7], 16) / 255.0
52              result = Color(r, g, b)              result = Color(r, g, b)
53          else:          else:
54              raise ValueError("Invalid hexadecimal color specification %s"              raise ValueError(_("Invalid hexadecimal color specification %s")
55                               % color)                               % color)
56      else:      else:
57          raise ValueError("Invalid color specification %s" % color)          raise ValueError(_("Invalid color specification %s") % color)
58      return result      return result
59    
60    
61  class ProcessSession(xml.sax.handler.ContentHandler):  class XMLReader(xml.sax.handler.ContentHandler):
62    
63      # Dictionary mapping element names (or (URI, element name) pairs for      # Dictionary mapping element names (or (URI, element name) pairs for
64      # documents using namespaces) to method names. The methods should      # documents using namespaces) to method names. The methods should
# Line 65  class ProcessSession(xml.sax.handler.Con Line 74  class ProcessSession(xml.sax.handler.Con
74      end_dispatcher = {}      end_dispatcher = {}
75    
76    
77      def __init__(self, directory):      def __init__(self):
78          """Inititialize the Sax handler.          self.chars = ''
79            self.__directory = ""
80            self.__dispatchers = {}
81    
82        def read(self, file_or_filename):
83    
84          The directory parameter should be the directory containing the          if hasattr(file_or_filename, "read"):
85          session file. It's needed to interpret embedded relative              # it's a file object
86          filenames.              self.__directory = ""
87                self.__file = file_or_filename
88            else:
89                filename = file_or_filename
90                self.__directory = os.path.dirname(filename)
91                self.__file = open(filename)
92    
93            parser = make_parser()
94            parser.setContentHandler(self)
95            parser.setErrorHandler(ErrorHandler())
96            parser.setFeature(xml.sax.handler.feature_namespaces, 1)
97    
98            #
99            # Well, this isn't pretty, but it appears that if you
100            # use Python 2.2 without the site-package _xmlplus then
101            # the following will fail, and without them it will work.
102            # However, if you do have the site-package and you don't
103            # call these functions, the reader raises an exception
104            #
105            # The reason we set these to 0 in the first place is
106            # because there is an unresolved issue with external
107            # entities causing an exception in the reader
108            #
109            try:
110                parser.setFeature(xml.sax.handler.feature_validation,0)
111                parser.setFeature(xml.sax.handler.feature_external_ges,0)
112                parser.setFeature(xml.sax.handler.feature_external_pes,0)
113            except SAXNotRecognizedException:
114                pass
115    
116            parser.parse(self.__file)
117    
118            self.close()
119    
120        def close(self):
121            self.__file.close()
122            
123        def GetFilename(self):
124            if hasattr(self.__file, "name"):
125                return self.__file.name
126    
127            return ""
128    
129        def GetDirectory(self):
130            return self.__directory
131    
132    
133        def AddDispatchers(self, dict):
134            """Add the function names that should be used to process XML tags.
135    
136            dict -- a dictionary whose keys are XML tag strings and whose values
137                    are pairs of strings such that the first string is
138                    the name of the function that should be called when the
139                    XML tag opens and the second string is the name of the
140                    function that should be called when the XML tag closes.
141                    If a pair element is None, no function is called.
142          """          """
143          self.directory = directory  
144          self.chars = ''          self.__dispatchers.update(dict)
         self.theSession = None  
         self.aMap = None  
         self.aLayer = None  
145    
146      def startElementNS(self, name, qname, attrs):      def startElementNS(self, name, qname, attrs):
147          """Call the method given for name in self.start_dispatcher          """Call the method given for name in self.start_dispatcher
148          """          """
149          if name[0] is None:          if name[0] is None:
150              method_name = self.start_dispatcher.get(name[1])              method_name = self.__dispatchers.get(name[1])
151          else:          else:
152              # Dispatch with namespace              # Dispatch with namespace
153              method_name = self.start_dispatcher.get(name)              method_name = self.__dispatchers.get(name)
154          if method_name is not None:          if method_name is not None and method_name[0] is not None:
155              getattr(self, method_name)(name, qname, attrs)              getattr(self, method_name[0])(name, qname, attrs)
156    
157      def endElementNS(self, name, qname):      def endElementNS(self, name, qname):
158          """Call the method given for name in self.end_dispatcher          """Call the method given for name in self.end_dispatcher
159          """          """
160          if name[0] is None:          if name[0] is None:
161              method_name = self.end_dispatcher.get(name[1])              method_name = self.__dispatchers.get(name[1])
162          else:          else:
163              # Dispatch with namespace              # Dispatch with namespace
164              method_name = self.end_dispatcher.get(name)              method_name = self.__dispatchers.get(name)
165          if method_name is not None:          if method_name is not None and method_name[1] is not None:
166              getattr(self, method_name)(name, qname)              getattr(self, method_name[1])(name, qname)
167    
168        def encode(self, str):
169            """Assume that str is in Unicode and encode it into Latin1.
170            
171            If str is None, return None
172            """
173    
174            if str is not None:
175                return str.encode("latin1")
176            else:
177                return None
178    
179    class SessionLoader(XMLReader):
180    
181        def __init__(self):
182            """Inititialize the Sax handler."""
183            XMLReader.__init__(self)
184    
185            self.theSession = None
186            self.aMap = None
187            self.aLayer = None
188    
189            XMLReader.AddDispatchers(self,
190                {'session'       : ("start_session",        "end_session"),
191                 'map'           : ("start_map",            "end_map"),
192                 'projection'    : ("start_projection",     "end_projection"),
193                 'parameter'     : ("start_parameter",      None),
194                 'layer'         : ("start_layer",          "end_layer"),
195                 'rasterlayer'   : ("start_rasterlayer",    "end_rasterlayer"),
196                 'classification': ("start_classification", "end_classification"),
197                 'clnull'        : ("start_clnull",         "end_clnull"),
198                 'clpoint'       : ("start_clpoint",        "end_clpoint"),
199                 'clrange'       : ("start_clrange",        "end_clrange"),
200                 'cldata'        : ("start_cldata",         "end_cldata"),
201                 'table'         : ("start_table",          "end_table"),
202                 'labellayer'    : ("start_labellayer",     None),
203                 'label'         : ("start_label",          None)})
204    
205      def start_session(self, name, qname, attrs):      def start_session(self, name, qname, attrs):
206          self.theSession = Session(attrs.get((None, 'title'), None))          self.theSession = Session(self.encode(attrs.get((None, 'title'), None)))
     start_dispatcher['session'] = "start_session"  
207    
208      def end_session(self, name, qname):      def end_session(self, name, qname):
209          pass          pass
     end_dispatcher['session'] = "end_session"  
210    
211      def start_map(self, name, qname, attrs):      def start_map(self, name, qname, attrs):
212          """Start a map."""          """Start a map."""
213          self.aMap = Map(attrs.get((None, 'title'), None))          self.aMap = Map(attrs.get((None, 'title'), None))
     start_dispatcher['map'] = "start_map"  
214    
215      def end_map(self, name, qname):      def end_map(self, name, qname):
216          self.theSession.AddMap(self.aMap)          self.theSession.AddMap(self.aMap)
217      end_dispatcher['map'] = "end_map"          self.aMap = None
218    
219      def start_projection(self, name, qname, attrs):      def start_projection(self, name, qname, attrs):
220            self.ProjectionName = self.encode(attrs.get((None, 'name'), None))
221          self.ProjectionParams = [ ]          self.ProjectionParams = [ ]
     start_dispatcher['projection'] = "start_projection"  
222    
223      def end_projection(self, name, qname):      def end_projection(self, name, qname):
224          self.aMap.SetProjection(Projection(self.ProjectionParams))          if self.aLayer is not None:
225      end_dispatcher['projection'] = "end_projection"              obj = self.aLayer
226            elif self.aMap is not None:
227                obj = self.aMap
228            else:
229                assert False, "projection tag out of context"
230                pass
231    
232            obj.SetProjection(
233                Projection(self.ProjectionParams, self.ProjectionName))
234    
235      def start_parameter(self, name, qname, attrs):      def start_parameter(self, name, qname, attrs):
236          s = attrs.get((None, 'value'))          s = attrs.get((None, 'value'))
237          s = str(s) # we can't handle unicode in proj          s = str(s) # we can't handle unicode in proj
238          self.ProjectionParams.append(s)          self.ProjectionParams.append(s)
     start_dispatcher['parameter'] = "start_parameter"  
239    
240      def start_layer(self, name, qname, attrs, layer_class = Layer):      def start_layer(self, name, qname, attrs, layer_class = Layer):
241          """Start a layer          """Start a layer
# Line 138  class ProcessSession(xml.sax.handler.Con Line 244  class ProcessSession(xml.sax.handler.Con
244          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
245          object and bind it to self.aLayer.          object and bind it to self.aLayer.
246          """          """
247          title = attrs.get((None, 'title'), "")          title = self.encode(attrs.get((None, 'title'), ""))
248          filename = attrs.get((None, 'filename'), "")          filename = attrs.get((None, 'filename'), "")
249          filename = os.path.join(self.directory, filename)          filename = os.path.join(self.GetDirectory(), filename)
250            filename = self.encode(filename)
251            visible  = self.encode(attrs.get((None, 'visible'), "true")) != "false"
252          fill = parse_color(attrs.get((None, 'fill'), "None"))          fill = parse_color(attrs.get((None, 'fill'), "None"))
253          stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))          stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))
254          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))
255          self.aLayer = layer_class(title, filename, fill = fill,          self.aLayer = layer_class(title,
256                                    stroke = stroke, stroke_width = stroke_width)                                    self.theSession.OpenShapefile(filename),
257      start_dispatcher['layer'] = "start_layer"                                    fill = fill, stroke = stroke,
258                                      lineWidth = stroke_width,
259                                      visible = visible)
260    
261      def end_layer(self, name, qname):      def end_layer(self, name, qname):
262          self.aMap.AddLayer(self.aLayer)          self.aMap.AddLayer(self.aLayer)
263      end_dispatcher['layer'] = "end_layer"          self.aLayer = None
264    
265        def start_rasterlayer(self, name, qname, attrs, layer_class = RasterLayer):
266            title = self.encode(attrs.get((None, 'title'), ""))
267            filename = attrs.get((None, 'filename'), "")
268            filename = os.path.join(self.GetDirectory(), filename)
269            filename = self.encode(filename)
270            visible  = self.encode(attrs.get((None, 'visible'), "true")) != "false"
271    
272            self.aLayer = layer_class(title, filename, visible = visible)
273    
274        def end_rasterlayer(self, name, qname):
275            self.aMap.AddLayer(self.aLayer)
276            self.aLayer = None
277    
278      def start_classification(self, name, qname, attrs):      def start_classification(self, name, qname, attrs):
279          self.aLayer.classification.setField(attrs.get((None, 'field'), None))          field = attrs.get((None, 'field'), None)
280      start_dispatcher['classification'] = "start_classification"  
281            fieldType = attrs.get((None, 'field_type'), None)
282            dbFieldType = self.aLayer.GetFieldType(field)
283    
284            if fieldType != dbFieldType:
285                raise ValueError(_("xml field type differs from database!"))
286    
287            # setup conversion routines depending on the kind of data
288            # we will be seeing later on
289            if fieldType == FIELDTYPE_STRING:
290                self.conv = str
291            elif fieldType == FIELDTYPE_INT:
292                self.conv = lambda p: int(float(p))
293            elif fieldType == FIELDTYPE_DOUBLE:
294                self.conv = float
295    
296            self.aLayer.GetClassification().SetField(field)
297    
298      def end_classification(self, name, qname):      def end_classification(self, name, qname):
299          pass          pass
     end_dispatcher['classification'] = "end_classification"  
300    
301      def start_clnull(self, name, qname, attrs):      def start_clnull(self, name, qname, attrs):
302          self.cl_data = {}          self.cl_group = ClassGroupDefault()
303      start_dispatcher['clnull'] = "start_clnull"          self.cl_group.SetLabel(self.encode(attrs.get((None, 'label'), "")))
304            self.cl_prop = ClassGroupProperties()
305    
306      def end_clnull(self, name, qname):      def end_clnull(self, name, qname):
307          self.aLayer.classification.setNull(self.cl_data)          self.cl_group.SetProperties(self.cl_prop)
308          del self.cl_data          self.aLayer.GetClassification().SetDefaultGroup(self.cl_group)
309      end_dispatcher['clnull'] = "end_clnull"          del self.cl_group, self.cl_prop
310    
311      def start_clpoint(self, name, qname, attrs):      def start_clpoint(self, name, qname, attrs):
312          attrib_value = attrs.get((None, 'value'), "0")          attrib_value = attrs.get((None, 'value'), "0")
313    
314          try:          value = self.conv(attrib_value)
315              self.cl_value  = int(attrib_value)  
316          except:          self.cl_group = ClassGroupSingleton(value)
317              self.cl_value  = attrib_value          self.cl_group.SetLabel(self.encode(attrs.get((None, 'label'), "")))
318            self.cl_prop = ClassGroupProperties()
319    
         self.cl_data = {}  
     start_dispatcher['clpoint'] = "start_clpoint"  
320    
321      def end_clpoint(self, name, qname):      def end_clpoint(self, name, qname):
322          self.aLayer.classification.addPoint(self.cl_value, self.cl_data)          self.cl_group.SetProperties(self.cl_prop)
323          del self.cl_value, self.cl_data          self.aLayer.GetClassification().AppendGroup(self.cl_group)
324      end_dispatcher['clpoint'] = "end_clpoint"          del self.cl_group, self.cl_prop
325    
326      def start_clrange(self, name, qname, attrs):      def start_clrange(self, name, qname, attrs):
327    
328            range = attrs.get((None, 'range'), None)
329            # for backward compatibility (min/max are not saved)
330            min   = attrs.get((None, 'min'), None)
331            max   = attrs.get((None, 'max'), None)
332    
333          try:          try:
334              self.cl_low = int(attrs.get((None, 'low'), "0"))              if range is not None:
335              self.cl_high = int(attrs.get((None, 'high'), "0"))                  self.cl_group = ClassGroupRange(Range(range))
336                elif min is not None and max is not None:
337                    self.cl_group = ClassGroupRange(self.conv(min), self.conv(max))
338                else:
339                    self.cl_group = ClassGroupRange(Range(None))
340    
341          except ValueError:          except ValueError:
342              raise ValueError("Classification range is not a number!")              raise ValueError(_("Classification range is not a number!"))
343    
344            self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
345            self.cl_prop = ClassGroupProperties()
346    
         self.cl_data = {}  
     start_dispatcher['clrange'] = "start_clrange"  
347    
348      def end_clrange(self, name, qname):      def end_clrange(self, name, qname):
349          self.aLayer.classification.addRange(          self.cl_group.SetProperties(self.cl_prop)
350              self.cl_low, self.cl_high, self.cl_data)          self.aLayer.GetClassification().AppendGroup(self.cl_group)
351          del self.cl_low, self.cl_high, self.cl_data          del self.cl_group, self.cl_prop
     end_dispatcher['clrange'] = "end_clrange"  
352    
353      def start_cldata(self, name, qname, attrs):      def start_cldata(self, name, qname, attrs):
354          self.cl_data['stroke'] = parse_color(          self.cl_prop.SetLineColor(
355                                   attrs.get((None, 'stroke'), "None"))              parse_color(attrs.get((None, 'stroke'), "None")))
356          self.cl_data['stroke_width'] = int(          self.cl_prop.SetLineWidth(
357                                   attrs.get((None, 'stroke_width'), "0"))              int(attrs.get((None, 'stroke_width'), "0")))
358          self.cl_data['fill'] = parse_color(          self.cl_prop.SetFill(parse_color(attrs.get((None, 'fill'), "None")))
                                  attrs.get((None, 'fill'), "None"))  
     start_dispatcher['cldata'] = "start_cldata"  
359    
360      def end_cldata(self, name, qname):      def end_cldata(self, name, qname):
361          pass          pass
     end_dispatcher['cldata'] = "end_cldata"  
362    
363      def start_table(self, name, qname, attrs):      def start_table(self, name, qname, attrs):
364          print "table title: %s" % attrs.get('title', None)          #print "table title: %s" % attrs.get('title', None)
365      start_dispatcher['table'] = "start_table"          pass
366    
367      def end_table(self, name, qname):      def end_table(self, name, qname):
368          pass          pass
     end_dispatcher['table'] = "end_table"  
369    
370      def start_labellayer(self, name, qname, attrs):      def start_labellayer(self, name, qname, attrs):
371          self.aLayer = self.aMap.LabelLayer()          self.aLayer = self.aMap.LabelLayer()
     start_dispatcher['labellayer'] = "start_labellayer"  
372    
373      def start_label(self, name, qname, attrs):      def start_label(self, name, qname, attrs):
374          x = float(attrs[(None, 'x')])          x = float(attrs[(None, 'x')])
375          y = float(attrs[(None, 'y')])          y = float(attrs[(None, 'y')])
376          text = attrs[(None, 'text')]          text = self.encode(attrs[(None, 'text')])
377          halign = attrs[(None, 'halign')]          halign = attrs[(None, 'halign')]
378          valign = attrs[(None, 'valign')]          valign = attrs[(None, 'valign')]
379          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"  
380    
381      def characters(self, chars):      def characters(self, chars):
382          pass          pass
# Line 242  class ProcessSession(xml.sax.handler.Con Line 384  class ProcessSession(xml.sax.handler.Con
384    
385  def load_session(filename):  def load_session(filename):
386      """Load a Thuban session from the file object file"""      """Load a Thuban session from the file object file"""
387      dir = os.path.dirname(filename)  
388      file = open(filename)      handler = SessionLoader()
389      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)  
390    
391      session = handler.theSession      session = handler.theSession
392      # Newly loaded session aren't modified      # Newly loaded session aren't modified
Legend:
Removed from v.365  
changed lines
  Added in v.930
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26