/[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 706 by jonathan, Wed Apr 23 08:44:21 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]>
# Line 13  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, SAXNotRecognizedException  from xml.sax import make_parser, ErrorHandler, SAXNotRecognizedException
21    
22  from Thuban import _  from Thuban import _
 from Thuban.common import *  
23    
24  from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \  from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \
25       FIELDTYPE_STRING       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.range import Range
33  from Thuban.Model.classification import Classification, \  from Thuban.Model.classification import Classification, \
34      ClassGroupDefault, ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \      ClassGroupDefault, ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \
35      ClassGroupProperties      ClassGroupProperties
# Line 76  class XMLReader(xml.sax.handler.ContentH Line 76  class XMLReader(xml.sax.handler.ContentH
76    
77      def __init__(self):      def __init__(self):
78          self.chars = ''          self.chars = ''
         self.__parser = None  
79          self.__directory = ""          self.__directory = ""
80          self.__dispatchers = {}          self.__dispatchers = {}
81    
82      def read(self, file_or_filename):      def read(self, file_or_filename):
83    
84          if hasattr(file_or_filename, "write"):          if hasattr(file_or_filename, "read"):
85              # it's a file object              # it's a file object
86              self.__directory = ""              self.__directory = ""
87              self.__file = file_or_filename              self.__file = file_or_filename
# Line 91  class XMLReader(xml.sax.handler.ContentH Line 90  class XMLReader(xml.sax.handler.ContentH
90              self.__directory = os.path.dirname(filename)              self.__directory = os.path.dirname(filename)
91              self.__file = open(filename)              self.__file = open(filename)
92    
93          if self.__parser is None:          parser = make_parser()
94              self.__parser = make_parser()          parser.setContentHandler(self)
95              self.__parser.setContentHandler(self)          parser.setErrorHandler(ErrorHandler())
96              self.__parser.setErrorHandler(ErrorHandler())          parser.setFeature(xml.sax.handler.feature_namespaces, 1)
97              self.__parser.setFeature(xml.sax.handler.feature_namespaces, 1)  
98            #
99              #          # Well, this isn't pretty, but it appears that if you
100              # Well, this isn't pretty, but it appears that if you          # use Python 2.2 without the site-package _xmlplus then
101              # use Python 2.2 without the site-package _xmlplus then          # the following will fail, and without them it will work.
102              # the following will fail, and without them it will work.          # However, if you do have the site-package and you don't
103              # However, if you do have the site-package and you don't          # call these functions, the reader raises an exception
104              # call these functions, the reader raises an exception          #
105              #          # The reason we set these to 0 in the first place is
106              # The reason we set these to 0 in the first place is          # because there is an unresolved issue with external
107              # because there is an unresolved issue with external          # entities causing an exception in the reader
108              # entities causing an exception in the reader          #
109              #          try:
110              try:              parser.setFeature(xml.sax.handler.feature_validation,0)
111                  self.__parser.setFeature(xml.sax.handler.feature_validation,0)              parser.setFeature(xml.sax.handler.feature_external_ges,0)
112                  self.__parser.setFeature(xml.sax.handler.feature_external_ges,0)              parser.setFeature(xml.sax.handler.feature_external_pes,0)
113                  self.__parser.setFeature(xml.sax.handler.feature_external_pes,0)          except SAXNotRecognizedException:
114              except SAXNotRecognizedException:              pass
                 pass  
115    
116          self.__parser.parse(self.__file)          parser.parse(self.__file)
117    
118          self.close()          self.close()
119    
120      def close(self):      def close(self):
121          self.__file.close()          self.__file.close()
122                    
123      def GetFileName(self):      def GetFilename(self):
124          if hasattr(self.__file, "name"):          if hasattr(self.__file, "name"):
125              return self.__file.name              return self.__file.name
126    
# Line 133  class XMLReader(xml.sax.handler.ContentH Line 131  class XMLReader(xml.sax.handler.ContentH
131    
132    
133      def AddDispatchers(self, dict):      def AddDispatchers(self, dict):
134          self.__dispatchers.update(dict)          """Add the function names that should be used to process XML tags.
135    
136          #for key, (start, end) in dict.iteritems():          dict -- a dictionary whose keys are XML tag strings and whose values
137              #if start is not None: self.start_dispatcher[key] = start                  are pairs of strings such that the first string is
138              #if end   is not None: self.end_dispatcher[key]   = end                  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    
144            self.__dispatchers.update(dict)
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.__dispatchers.get(name[1])              method_name = self.__dispatchers.get(name[1])
             #method_name = self.start_dispatcher.get(name[1])  
151          else:          else:
152              # Dispatch with namespace              # Dispatch with namespace
153              method_name = self.__dispatchers.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:
             and method_name[0] is not None:  
155              getattr(self, method_name[0])(name, qname, attrs)              getattr(self, method_name[0])(name, qname, attrs)
156    
157      def endElementNS(self, name, qname):      def endElementNS(self, name, qname):
# Line 157  class XMLReader(xml.sax.handler.ContentH Line 159  class XMLReader(xml.sax.handler.ContentH
159          """          """
160          if name[0] is None:          if name[0] is None:
161              method_name = self.__dispatchers.get(name[1])              method_name = self.__dispatchers.get(name[1])
             #method_name = self.end_dispatcher.get(name[1])  
162          else:          else:
163              # Dispatch with namespace              # Dispatch with namespace
164              method_name = self.__dispatchers.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:
             and method_name[1] is not None:  
166              getattr(self, method_name[1])(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):  class SessionLoader(XMLReader):
180    
181      def __init__(self):      def __init__(self):
# Line 181  class SessionLoader(XMLReader): Line 192  class SessionLoader(XMLReader):
192               'projection'    : ("start_projection",     "end_projection"),               'projection'    : ("start_projection",     "end_projection"),
193               'parameter'     : ("start_parameter",      None),               'parameter'     : ("start_parameter",      None),
194               'layer'         : ("start_layer",          "end_layer"),               'layer'         : ("start_layer",          "end_layer"),
195                 'rasterlayer'   : ("start_rasterlayer",    "end_rasterlayer"),
196               'classification': ("start_classification", "end_classification"),               'classification': ("start_classification", "end_classification"),
197               'clnull'        : ("start_clnull",         "end_clnull"),               'clnull'        : ("start_clnull",         "end_clnull"),
198               'clpoint'       : ("start_clpoint",        "end_clpoint"),               'clpoint'       : ("start_clpoint",        "end_clpoint"),
# Line 191  class SessionLoader(XMLReader): Line 203  class SessionLoader(XMLReader):
203               'label'         : ("start_label",          None)})               '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)))
207    
208      def end_session(self, name, qname):      def end_session(self, name, qname):
209          pass          pass
# Line 202  class SessionLoader(XMLReader): Line 214  class SessionLoader(XMLReader):
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            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 = [ ]
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                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'))
# Line 221  class SessionLoader(XMLReader): Line 244  class SessionLoader(XMLReader):
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.GetDirectory(), 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, lineWidth = stroke_width)                                    self.theSession.OpenShapefile(filename),
257                                      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            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          field = attrs.get((None, 'field'), None)          field = attrs.get((None, 'field'), None)
# Line 253  class SessionLoader(XMLReader): Line 295  class SessionLoader(XMLReader):
295    
296          self.aLayer.GetClassification().SetField(field)          self.aLayer.GetClassification().SetField(field)
297    
   
298      def end_classification(self, name, qname):      def end_classification(self, name, qname):
299          pass          pass
300    
301      def start_clnull(self, name, qname, attrs):      def start_clnull(self, name, qname, attrs):
302          self.cl_group = ClassGroupDefault()          self.cl_group = ClassGroupDefault()
303          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))          self.cl_group.SetLabel(self.encode(attrs.get((None, 'label'), "")))
304          self.cl_prop = ClassGroupProperties()          self.cl_prop = ClassGroupProperties()
305    
306      def end_clnull(self, name, qname):      def end_clnull(self, name, qname):
# Line 270  class SessionLoader(XMLReader): Line 311  class SessionLoader(XMLReader):
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    
         #try:  
             #value  = Str2Num(attrib_value)  
         #except:  
             #value  = attrib_value  
   
314          value = self.conv(attrib_value)          value = self.conv(attrib_value)
315    
316          self.cl_group = ClassGroupSingleton(value)          self.cl_group = ClassGroupSingleton(value)
317          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))          self.cl_group.SetLabel(self.encode(attrs.get((None, 'label'), "")))
318          self.cl_prop = ClassGroupProperties()          self.cl_prop = ClassGroupProperties()
319    
320    
# Line 289  class SessionLoader(XMLReader): Line 325  class SessionLoader(XMLReader):
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              min = self.conv(attrs.get((None, 'min'), "0"))              if range is not None:
335              max = self.conv(attrs.get((None, 'max'), "0"))                  self.cl_group = ClassGroupRange(Range(range))
336              #min = Str2Num(attrs.get((None, 'min'), "0"))              elif min is not None and max is not None:
337              #max = Str2Num(attrs.get((None, 'max'), "0"))                  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    
         self.cl_group = ClassGroupRange(min, max)  
344          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
345          self.cl_prop = ClassGroupProperties()          self.cl_prop = ClassGroupProperties()
346    
# Line 330  class SessionLoader(XMLReader): Line 373  class SessionLoader(XMLReader):
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)
Legend:
Removed from v.706  
changed lines
  Added in v.930
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26