/[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 722 by jonathan, Thu Apr 24 10:12:29 2003 UTC
# 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.__parser = None
80            self.__directory = ""
81            self.__dispatchers = {}
82    
83        def read(self, file_or_filename):
84    
85            if hasattr(file_or_filename, "read"):
86                # it's a file object
87                self.__directory = ""
88                self.__file = file_or_filename
89            else:
90                filename = file_or_filename
91                self.__directory = os.path.dirname(filename)
92                self.__file = open(filename)
93    
94            if self.__parser is None:
95                self.__parser = make_parser()
96                self.__parser.setContentHandler(self)
97                self.__parser.setErrorHandler(ErrorHandler())
98                self.__parser.setFeature(xml.sax.handler.feature_namespaces, 1)
99    
100                #
101                # Well, this isn't pretty, but it appears that if you
102                # use Python 2.2 without the site-package _xmlplus then
103                # the following will fail, and without them it will work.
104                # However, if you do have the site-package and you don't
105                # call these functions, the reader raises an exception
106                #
107                # The reason we set these to 0 in the first place is
108                # because there is an unresolved issue with external
109                # entities causing an exception in the reader
110                #
111                try:
112                    self.__parser.setFeature(xml.sax.handler.feature_validation,0)
113                    self.__parser.setFeature(xml.sax.handler.feature_external_ges,0)
114                    self.__parser.setFeature(xml.sax.handler.feature_external_pes,0)
115                except SAXNotRecognizedException:
116                    pass
117    
118            self.__parser.parse(self.__file)
119    
120            self.close()
121    
122        def close(self):
123            self.__file.close()
124            
125        def GetFileName(self):
126            if hasattr(self.__file, "name"):
127                return self.__file.name
128    
129            return ""
130    
131        def GetDirectory(self):
132            return self.__directory
133    
134          The directory parameter should be the directory containing the  
135          session file. It's needed to interpret embedded relative      def AddDispatchers(self, dict):
136          filenames.          """Add the function names that should be used to process XML tags.
137    
138            dict -- a dictionary whose keys are XML tag strings and whose values
139                    are pairs of strings such that the first string is
140                    the name of the function that should be called when the
141                    XML tag opens and the second string is the name of the
142                    function that should be called when the XML tag closes.
143                    If a pair element is None, no function is called.
144          """          """
145          self.directory = directory  
146          self.chars = ''          self.__dispatchers.update(dict)
147    
148      def startElementNS(self, name, qname, attrs):      def startElementNS(self, name, qname, attrs):
149          """Call the method given for name in self.start_dispatcher          """Call the method given for name in self.start_dispatcher
150          """          """
151          if name[0] is None:          if name[0] is None:
152              method_name = self.start_dispatcher.get(name[1])              method_name = self.__dispatchers.get(name[1])
153          else:          else:
154              # Dispatch with namespace              # Dispatch with namespace
155              method_name = self.start_dispatcher.get(name)              method_name = self.__dispatchers.get(name)
156          if method_name is not None:          if method_name is not None and method_name[0] is not None:
157              getattr(self, method_name)(name, qname, attrs)              getattr(self, method_name[0])(name, qname, attrs)
158    
159      def endElementNS(self, name, qname):      def endElementNS(self, name, qname):
160          """Call the method given for name in self.end_dispatcher          """Call the method given for name in self.end_dispatcher
161          """          """
162          if name[0] is None:          if name[0] is None:
163              method_name = self.end_dispatcher.get(name[1])              method_name = self.__dispatchers.get(name[1])
164          else:          else:
165              # Dispatch with namespace              # Dispatch with namespace
166              method_name = self.end_dispatcher.get(name)              method_name = self.__dispatchers.get(name)
167          if method_name is not None:          if method_name is not None and method_name[1] is not None:
168              getattr(self, method_name)(name, qname)              getattr(self, method_name[1])(name, qname)
169    
170      def GetDirectory(self):  class SessionLoader(XMLReader):
         return self.directory  
   
 class ProcessSession(XMLProcessor):  
171    
172      def __init__(self, directory):      def __init__(self):
173          """Inititialize the Sax handler."""          """Inititialize the Sax handler."""
174          XMLProcessor.__init__(self, directory)          XMLReader.__init__(self)
175    
176          self.theSession = None          self.theSession = None
177          self.aMap = None          self.aMap = None
178          self.aLayer = None          self.aLayer = None
179    
180            XMLReader.AddDispatchers(self,
181                {'session'       : ("start_session",        "end_session"),
182                 'map'           : ("start_map",            "end_map"),
183                 'projection'    : ("start_projection",     "end_projection"),
184                 'parameter'     : ("start_parameter",      None),
185                 'layer'         : ("start_layer",          "end_layer"),
186                 'classification': ("start_classification", "end_classification"),
187                 'clnull'        : ("start_clnull",         "end_clnull"),
188                 'clpoint'       : ("start_clpoint",        "end_clpoint"),
189                 'clrange'       : ("start_clrange",        "end_clrange"),
190                 'cldata'        : ("start_cldata",         "end_cldata"),
191                 'table'         : ("start_table",          "end_table"),
192                 'labellayer'    : ("start_labellayer",     None),
193                 'label'         : ("start_label",          None)})
194    
195      def start_session(self, name, qname, attrs):      def start_session(self, name, qname, attrs):
196          self.theSession = Session(attrs.get((None, 'title'), None))          self.theSession = Session(attrs.get((None, 'title'), None))
     XMLProcessor.start_dispatcher['session'] = "start_session"  
197    
198      def end_session(self, name, qname):      def end_session(self, name, qname):
199          pass          pass
     XMLProcessor.end_dispatcher['session'] = "end_session"  
200    
201      def start_map(self, name, qname, attrs):      def start_map(self, name, qname, attrs):
202          """Start a map."""          """Start a map."""
203          self.aMap = Map(attrs.get((None, 'title'), None))          self.aMap = Map(attrs.get((None, 'title'), None))
     XMLProcessor.start_dispatcher['map'] = "start_map"  
204    
205      def end_map(self, name, qname):      def end_map(self, name, qname):
206          self.theSession.AddMap(self.aMap)          self.theSession.AddMap(self.aMap)
     XMLProcessor.end_dispatcher['map'] = "end_map"  
207    
208      def start_projection(self, name, qname, attrs):      def start_projection(self, name, qname, attrs):
209          self.ProjectionParams = [ ]          self.ProjectionParams = [ ]
     XMLProcessor.start_dispatcher['projection'] = "start_projection"  
210    
211      def end_projection(self, name, qname):      def end_projection(self, name, qname):
212          self.aMap.SetProjection(Projection(self.ProjectionParams))          self.aMap.SetProjection(Projection(self.ProjectionParams))
     XMLProcessor.end_dispatcher['projection'] = "end_projection"  
213    
214      def start_parameter(self, name, qname, attrs):      def start_parameter(self, name, qname, attrs):
215          s = attrs.get((None, 'value'))          s = attrs.get((None, 'value'))
216          s = str(s) # we can't handle unicode in proj          s = str(s) # we can't handle unicode in proj
217          self.ProjectionParams.append(s)          self.ProjectionParams.append(s)
     XMLProcessor.start_dispatcher['parameter'] = "start_parameter"  
218    
219      def start_layer(self, name, qname, attrs, layer_class = Layer):      def start_layer(self, name, qname, attrs, layer_class = Layer):
220          """Start a layer          """Start a layer
# Line 165  class ProcessSession(XMLProcessor): Line 231  class ProcessSession(XMLProcessor):
231          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))
232          self.aLayer = layer_class(title, filename, fill = fill,          self.aLayer = layer_class(title, filename, fill = fill,
233                                    stroke = stroke, lineWidth = stroke_width)                                    stroke = stroke, lineWidth = stroke_width)
     XMLProcessor.start_dispatcher['layer'] = "start_layer"  
234    
235      def end_layer(self, name, qname):      def end_layer(self, name, qname):
236          self.aMap.AddLayer(self.aLayer)          self.aMap.AddLayer(self.aLayer)
     XMLProcessor.end_dispatcher['layer'] = "end_layer"  
237    
238      def start_classification(self, name, qname, attrs):      def start_classification(self, name, qname, attrs):
239          field = attrs.get((None, 'field'), None)          field = attrs.get((None, 'field'), None)
# Line 191  class ProcessSession(XMLProcessor): Line 255  class ProcessSession(XMLProcessor):
255    
256          self.aLayer.GetClassification().SetField(field)          self.aLayer.GetClassification().SetField(field)
257    
     XMLProcessor.start_dispatcher['classification'] = "start_classification"  
258    
259      def end_classification(self, name, qname):      def end_classification(self, name, qname):
260          pass          pass
     XMLProcessor.end_dispatcher['classification'] = "end_classification"  
261    
262      def start_clnull(self, name, qname, attrs):      def start_clnull(self, name, qname, attrs):
263          self.cl_group = ClassGroupDefault()          self.cl_group = ClassGroupDefault()
264          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
265          self.cl_prop = ClassGroupProperties()          self.cl_prop = ClassGroupProperties()
     XMLProcessor.start_dispatcher['clnull'] = "start_clnull"  
266    
267      def end_clnull(self, name, qname):      def end_clnull(self, name, qname):
268          self.cl_group.SetProperties(self.cl_prop)          self.cl_group.SetProperties(self.cl_prop)
269          self.aLayer.GetClassification().SetDefaultGroup(self.cl_group)          self.aLayer.GetClassification().SetDefaultGroup(self.cl_group)
270          del self.cl_group, self.cl_prop          del self.cl_group, self.cl_prop
     XMLProcessor.end_dispatcher['clnull'] = "end_clnull"  
271    
272      def start_clpoint(self, name, qname, attrs):      def start_clpoint(self, name, qname, attrs):
273          attrib_value = attrs.get((None, 'value'), "0")          attrib_value = attrs.get((None, 'value'), "0")
# Line 223  class ProcessSession(XMLProcessor): Line 283  class ProcessSession(XMLProcessor):
283          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
284          self.cl_prop = ClassGroupProperties()          self.cl_prop = ClassGroupProperties()
285    
     XMLProcessor.start_dispatcher['clpoint'] = "start_clpoint"  
286    
287      def end_clpoint(self, name, qname):      def end_clpoint(self, name, qname):
288          self.cl_group.SetProperties(self.cl_prop)          self.cl_group.SetProperties(self.cl_prop)
289          self.aLayer.GetClassification().AppendGroup(self.cl_group)          self.aLayer.GetClassification().AppendGroup(self.cl_group)
290          del self.cl_group, self.cl_prop          del self.cl_group, self.cl_prop
     XMLProcessor.end_dispatcher['clpoint'] = "end_clpoint"  
291    
292      def start_clrange(self, name, qname, attrs):      def start_clrange(self, name, qname, attrs):
293    
# Line 245  class ProcessSession(XMLProcessor): Line 303  class ProcessSession(XMLProcessor):
303          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
304          self.cl_prop = ClassGroupProperties()          self.cl_prop = ClassGroupProperties()
305    
     XMLProcessor.start_dispatcher['clrange'] = "start_clrange"  
306    
307      def end_clrange(self, name, qname):      def end_clrange(self, name, qname):
308          self.cl_group.SetProperties(self.cl_prop)          self.cl_group.SetProperties(self.cl_prop)
309          self.aLayer.GetClassification().AppendGroup(self.cl_group)          self.aLayer.GetClassification().AppendGroup(self.cl_group)
310          del self.cl_group, self.cl_prop          del self.cl_group, self.cl_prop
     XMLProcessor.end_dispatcher['clrange'] = "end_clrange"  
311    
312      def start_cldata(self, name, qname, attrs):      def start_cldata(self, name, qname, attrs):
313          self.cl_prop.SetLineColor(          self.cl_prop.SetLineColor(
# Line 259  class ProcessSession(XMLProcessor): Line 315  class ProcessSession(XMLProcessor):
315          self.cl_prop.SetLineWidth(          self.cl_prop.SetLineWidth(
316              int(attrs.get((None, 'stroke_width'), "0")))              int(attrs.get((None, 'stroke_width'), "0")))
317          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"  
318    
319      def end_cldata(self, name, qname):      def end_cldata(self, name, qname):
320          pass          pass
     XMLProcessor.end_dispatcher['cldata'] = "end_cldata"  
321    
322      def start_table(self, name, qname, attrs):      def start_table(self, name, qname, attrs):
323          #print "table title: %s" % attrs.get('title', None)          #print "table title: %s" % attrs.get('title', None)
324          pass          pass
     XMLProcessor.start_dispatcher['table'] = "start_table"  
325    
326      def end_table(self, name, qname):      def end_table(self, name, qname):
327          pass          pass
     XMLProcessor.end_dispatcher['table'] = "end_table"  
328    
329      def start_labellayer(self, name, qname, attrs):      def start_labellayer(self, name, qname, attrs):
330          self.aLayer = self.aMap.LabelLayer()          self.aLayer = self.aMap.LabelLayer()
     XMLProcessor.start_dispatcher['labellayer'] = "start_labellayer"  
331    
332      def start_label(self, name, qname, attrs):      def start_label(self, name, qname, attrs):
333          x = float(attrs[(None, 'x')])          x = float(attrs[(None, 'x')])
# Line 285  class ProcessSession(XMLProcessor): Line 336  class ProcessSession(XMLProcessor):
336          halign = attrs[(None, 'halign')]          halign = attrs[(None, 'halign')]
337          valign = attrs[(None, 'valign')]          valign = attrs[(None, 'valign')]
338          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"  
339    
340      def characters(self, chars):      def characters(self, chars):
341          pass          pass
342    
343    
 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)  
   
344  def load_session(filename):  def load_session(filename):
345      """Load a Thuban session from the file object file"""      """Load a Thuban session from the file object file"""
346    
347      dir = os.path.dirname(filename)      handler = SessionLoader()
348      handler = ProcessSession(dir)      handler.read(filename)
   
     load_xmlfile(filename, handler)  
349    
350      session = handler.theSession      session = handler.theSession
351      # Newly loaded session aren't modified      # Newly loaded session aren't modified
Legend:
Removed from v.694  
changed lines
  Added in v.722
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26