/[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 694 by jonathan, Wed Apr 16 16:39:18 2003 UTC revision 772 by jonathan, Tue Apr 29 14:34:11 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
# Line 58  def parse_color(color): Line 58  def parse_color(color):
58      return result      return result
59    
60    
61  class XMLProcessor(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 74  class XMLProcessor(xml.sax.handler.Conte Line 74  class XMLProcessor(xml.sax.handler.Conte
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            if hasattr(file_or_filename, "read"):
85                # it's a file object
86                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          The directory parameter should be the directory containing the      def GetDirectory(self):
130          session file. It's needed to interpret embedded relative          return self.__directory
131          filenames.  
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)
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)
   
     def GetDirectory(self):  
         return self.directory  
167    
168  class ProcessSession(XMLProcessor):  class SessionLoader(XMLReader):
169    
170      def __init__(self, directory):      def __init__(self):
171          """Inititialize the Sax handler."""          """Inititialize the Sax handler."""
172          XMLProcessor.__init__(self, directory)          XMLReader.__init__(self)
173    
174          self.theSession = None          self.theSession = None
175          self.aMap = None          self.aMap = None
176          self.aLayer = None          self.aLayer = None
177    
178            XMLReader.AddDispatchers(self,
179                {'session'       : ("start_session",        "end_session"),
180                 'map'           : ("start_map",            "end_map"),
181                 'projection'    : ("start_projection",     "end_projection"),
182                 'parameter'     : ("start_parameter",      None),
183                 'layer'         : ("start_layer",          "end_layer"),
184                 'classification': ("start_classification", "end_classification"),
185                 'clnull'        : ("start_clnull",         "end_clnull"),
186                 'clpoint'       : ("start_clpoint",        "end_clpoint"),
187                 'clrange'       : ("start_clrange",        "end_clrange"),
188                 'cldata'        : ("start_cldata",         "end_cldata"),
189                 'table'         : ("start_table",          "end_table"),
190                 'labellayer'    : ("start_labellayer",     None),
191                 'label'         : ("start_label",          None)})
192    
193      def start_session(self, name, qname, attrs):      def start_session(self, name, qname, attrs):
194          self.theSession = Session(attrs.get((None, 'title'), None))          self.theSession = Session(attrs.get((None, 'title'), None))
     XMLProcessor.start_dispatcher['session'] = "start_session"  
195    
196      def end_session(self, name, qname):      def end_session(self, name, qname):
197          pass          pass
     XMLProcessor.end_dispatcher['session'] = "end_session"  
198    
199      def start_map(self, name, qname, attrs):      def start_map(self, name, qname, attrs):
200          """Start a map."""          """Start a map."""
201          self.aMap = Map(attrs.get((None, 'title'), None))          self.aMap = Map(attrs.get((None, 'title'), None))
202      XMLProcessor.start_dispatcher['map'] = "start_map"          self.__projReceiver = self.aMap
203    
204      def end_map(self, name, qname):      def end_map(self, name, qname):
205          self.theSession.AddMap(self.aMap)          self.theSession.AddMap(self.aMap)
206      XMLProcessor.end_dispatcher['map'] = "end_map"          self.__projReceiver = None
207    
208      def start_projection(self, name, qname, attrs):      def start_projection(self, name, qname, attrs):
209            self.ProjectionName = attrs.get((None, 'name'), None)
210          self.ProjectionParams = [ ]          self.ProjectionParams = [ ]
     XMLProcessor.start_dispatcher['projection'] = "start_projection"  
211    
212      def end_projection(self, name, qname):      def end_projection(self, name, qname):
213          self.aMap.SetProjection(Projection(self.ProjectionParams))          self.__projReceiver.SetProjection(
214      XMLProcessor.end_dispatcher['projection'] = "end_projection"              Projection(self.ProjectionParams, self.ProjectionName))
215    
216      def start_parameter(self, name, qname, attrs):      def start_parameter(self, name, qname, attrs):
217          s = attrs.get((None, 'value'))          s = attrs.get((None, 'value'))
218          s = str(s) # we can't handle unicode in proj          s = str(s) # we can't handle unicode in proj
219          self.ProjectionParams.append(s)          self.ProjectionParams.append(s)
     XMLProcessor.start_dispatcher['parameter'] = "start_parameter"  
220    
221      def start_layer(self, name, qname, attrs, layer_class = Layer):      def start_layer(self, name, qname, attrs, layer_class = Layer):
222          """Start a layer          """Start a layer
# Line 160  class ProcessSession(XMLProcessor): Line 228  class ProcessSession(XMLProcessor):
228          title = attrs.get((None, 'title'), "")          title = attrs.get((None, 'title'), "")
229          filename = attrs.get((None, 'filename'), "")          filename = attrs.get((None, 'filename'), "")
230          filename = os.path.join(self.GetDirectory(), filename)          filename = os.path.join(self.GetDirectory(), filename)
231            visible  = attrs.get((None, 'visible'), "true")
232          fill = parse_color(attrs.get((None, 'fill'), "None"))          fill = parse_color(attrs.get((None, 'fill'), "None"))
233          stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))          stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))
234          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))
235          self.aLayer = layer_class(title, filename, fill = fill,          self.aLayer = layer_class(title,
236                                    stroke = stroke, lineWidth = stroke_width)                                    self.theSession.OpenShapefile(filename),
237      XMLProcessor.start_dispatcher['layer'] = "start_layer"                                    fill = fill, stroke = stroke,
238                                      lineWidth = stroke_width,
239                                      visible = visible != "false")
240    
241            self.__projReceiver = self.aLayer
242    
243      def end_layer(self, name, qname):      def end_layer(self, name, qname):
244          self.aMap.AddLayer(self.aLayer)          self.aMap.AddLayer(self.aLayer)
245      XMLProcessor.end_dispatcher['layer'] = "end_layer"          self.__projReceiver = None
246    
247      def start_classification(self, name, qname, attrs):      def start_classification(self, name, qname, attrs):
248          field = attrs.get((None, 'field'), None)          field = attrs.get((None, 'field'), None)
# Line 191  class ProcessSession(XMLProcessor): Line 264  class ProcessSession(XMLProcessor):
264    
265          self.aLayer.GetClassification().SetField(field)          self.aLayer.GetClassification().SetField(field)
266    
     XMLProcessor.start_dispatcher['classification'] = "start_classification"  
267    
268      def end_classification(self, name, qname):      def end_classification(self, name, qname):
269          pass          pass
     XMLProcessor.end_dispatcher['classification'] = "end_classification"  
270    
271      def start_clnull(self, name, qname, attrs):      def start_clnull(self, name, qname, attrs):
272          self.cl_group = ClassGroupDefault()          self.cl_group = ClassGroupDefault()
273          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
274          self.cl_prop = ClassGroupProperties()          self.cl_prop = ClassGroupProperties()
     XMLProcessor.start_dispatcher['clnull'] = "start_clnull"  
275    
276      def end_clnull(self, name, qname):      def end_clnull(self, name, qname):
277          self.cl_group.SetProperties(self.cl_prop)          self.cl_group.SetProperties(self.cl_prop)
278          self.aLayer.GetClassification().SetDefaultGroup(self.cl_group)          self.aLayer.GetClassification().SetDefaultGroup(self.cl_group)
279          del self.cl_group, self.cl_prop          del self.cl_group, self.cl_prop
     XMLProcessor.end_dispatcher['clnull'] = "end_clnull"  
280    
281      def start_clpoint(self, name, qname, attrs):      def start_clpoint(self, name, qname, attrs):
282          attrib_value = attrs.get((None, 'value'), "0")          attrib_value = attrs.get((None, 'value'), "0")
# Line 223  class ProcessSession(XMLProcessor): Line 292  class ProcessSession(XMLProcessor):
292          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
293          self.cl_prop = ClassGroupProperties()          self.cl_prop = ClassGroupProperties()
294    
     XMLProcessor.start_dispatcher['clpoint'] = "start_clpoint"  
295    
296      def end_clpoint(self, name, qname):      def end_clpoint(self, name, qname):
297          self.cl_group.SetProperties(self.cl_prop)          self.cl_group.SetProperties(self.cl_prop)
298          self.aLayer.GetClassification().AppendGroup(self.cl_group)          self.aLayer.GetClassification().AppendGroup(self.cl_group)
299          del self.cl_group, self.cl_prop          del self.cl_group, self.cl_prop
     XMLProcessor.end_dispatcher['clpoint'] = "end_clpoint"  
300    
301      def start_clrange(self, name, qname, attrs):      def start_clrange(self, name, qname, attrs):
302    
# Line 245  class ProcessSession(XMLProcessor): Line 312  class ProcessSession(XMLProcessor):
312          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
313          self.cl_prop = ClassGroupProperties()          self.cl_prop = ClassGroupProperties()
314    
     XMLProcessor.start_dispatcher['clrange'] = "start_clrange"  
315    
316      def end_clrange(self, name, qname):      def end_clrange(self, name, qname):
317          self.cl_group.SetProperties(self.cl_prop)          self.cl_group.SetProperties(self.cl_prop)
318          self.aLayer.GetClassification().AppendGroup(self.cl_group)          self.aLayer.GetClassification().AppendGroup(self.cl_group)
319          del self.cl_group, self.cl_prop          del self.cl_group, self.cl_prop
     XMLProcessor.end_dispatcher['clrange'] = "end_clrange"  
320    
321      def start_cldata(self, name, qname, attrs):      def start_cldata(self, name, qname, attrs):
322          self.cl_prop.SetLineColor(          self.cl_prop.SetLineColor(
# Line 259  class ProcessSession(XMLProcessor): Line 324  class ProcessSession(XMLProcessor):
324          self.cl_prop.SetLineWidth(          self.cl_prop.SetLineWidth(
325              int(attrs.get((None, 'stroke_width'), "0")))              int(attrs.get((None, 'stroke_width'), "0")))
326          self.cl_prop.SetFill(parse_color(attrs.get((None, 'fill'), "None")))          self.cl_prop.SetFill(parse_color(attrs.get((None, 'fill'), "None")))
     XMLProcessor.start_dispatcher['cldata'] = "start_cldata"  
327    
328      def end_cldata(self, name, qname):      def end_cldata(self, name, qname):
329          pass          pass
     XMLProcessor.end_dispatcher['cldata'] = "end_cldata"  
330    
331      def start_table(self, name, qname, attrs):      def start_table(self, name, qname, attrs):
332          #print "table title: %s" % attrs.get('title', None)          #print "table title: %s" % attrs.get('title', None)
333          pass          pass
     XMLProcessor.start_dispatcher['table'] = "start_table"  
334    
335      def end_table(self, name, qname):      def end_table(self, name, qname):
336          pass          pass
     XMLProcessor.end_dispatcher['table'] = "end_table"  
337    
338      def start_labellayer(self, name, qname, attrs):      def start_labellayer(self, name, qname, attrs):
339          self.aLayer = self.aMap.LabelLayer()          self.aLayer = self.aMap.LabelLayer()
     XMLProcessor.start_dispatcher['labellayer'] = "start_labellayer"  
340    
341      def start_label(self, name, qname, attrs):      def start_label(self, name, qname, attrs):
342          x = float(attrs[(None, 'x')])          x = float(attrs[(None, 'x')])
# Line 285  class ProcessSession(XMLProcessor): Line 345  class ProcessSession(XMLProcessor):
345          halign = attrs[(None, 'halign')]          halign = attrs[(None, 'halign')]
346          valign = attrs[(None, 'valign')]          valign = attrs[(None, 'valign')]
347          self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign)          self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign)
     XMLProcessor.start_dispatcher['label'] = "start_label"  
348    
349      def characters(self, chars):      def characters(self, chars):
350          pass          pass
351    
352    
 def load_xmlfile(filename, handler):  
     file = open(filename)  
   
     parser = make_parser()  
     parser.setContentHandler(handler)  
     parser.setErrorHandler(ErrorHandler())  
     parser.setFeature(xml.sax.handler.feature_namespaces, 1)  
   
     #  
     # Well, this isn't pretty, but it appears that if you  
     # use Python 2.2 without the site-package _xmlplus then  
     # the following will fail, and without them it will work.  
     # However, if you do have the site-package and you don't  
     # call these functions, the reader raises an exception  
     #  
     # The reason we set these to 0 in the first place is  
     # because there is an unresolved issue with external  
     # entities causing an exception in the reader  
     #  
     try:  
         parser.setFeature(xml.sax.handler.feature_validation, 0)  
         parser.setFeature(xml.sax.handler.feature_external_ges, 0)  
         parser.setFeature(xml.sax.handler.feature_external_pes, 0)  
     except SAXNotRecognizedException:  
         pass  
   
     parser.parse(file)  
   
353  def load_session(filename):  def load_session(filename):
354      """Load a Thuban session from the file object file"""      """Load a Thuban session from the file object file"""
355    
356      dir = os.path.dirname(filename)      handler = SessionLoader()
357      handler = ProcessSession(dir)      handler.read(filename)
   
     load_xmlfile(filename, handler)  
358    
359      session = handler.theSession      session = handler.theSession
360      # Newly loaded session aren't modified      # Newly loaded session aren't modified
Legend:
Removed from v.694  
changed lines
  Added in v.772
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26