/[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 439 by jonathan, Thu Feb 27 15:54:05 2003 UTC revision 744 by jonathan, Fri Apr 25 10:26:35 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  from xml.sax import make_parser, ErrorHandler, SAXNotRecognizedException
21    
22  from Thuban import _  from Thuban import _
23  from Thuban.common import *  from Thuban.common import *
24    
25    from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \
26         FIELDTYPE_STRING
27    
28  from Thuban.Model.session import Session  from Thuban.Model.session import Session
29  from Thuban.Model.map import Map  from Thuban.Model.map import Map
30  from Thuban.Model.layer import Layer  from Thuban.Model.layer import Layer
# Line 40  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 = Color.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 55  def parse_color(color): Line 58  def parse_color(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 71  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):
         """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  
78          self.chars = ''          self.chars = ''
79          self.theSession = None          self.__parser = None
80          self.aMap = None          self.__directory = ""
81          self.aLayer = None          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    
135        def AddDispatchers(self, dict):
136            """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    
146            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    class SessionLoader(XMLReader):
171    
172        def __init__(self):
173            """Inititialize the Sax handler."""
174            XMLReader.__init__(self)
175    
176            self.theSession = None
177            self.aMap = None
178            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))
     start_dispatcher['session'] = "start_session"  
197    
198      def end_session(self, name, qname):      def end_session(self, name, qname):
199          pass          pass
     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))
204      start_dispatcher['map'] = "start_map"          self.__projReceiver = self.aMap
205    
206      def end_map(self, name, qname):      def end_map(self, name, qname):
207          self.theSession.AddMap(self.aMap)          self.theSession.AddMap(self.aMap)
208      end_dispatcher['map'] = "end_map"          self.__projReceiver = None
209    
210      def start_projection(self, name, qname, attrs):      def start_projection(self, name, qname, attrs):
211            self.ProjectionName = attrs.get((None, 'name'), None)
212          self.ProjectionParams = [ ]          self.ProjectionParams = [ ]
     start_dispatcher['projection'] = "start_projection"  
213    
214      def end_projection(self, name, qname):      def end_projection(self, name, qname):
215          self.aMap.SetProjection(Projection(self.ProjectionParams))          self.__projReceiver.SetProjection(
216      end_dispatcher['projection'] = "end_projection"              Projection(self.ProjectionParams, self.ProjectionName))
217    
218      def start_parameter(self, name, qname, attrs):      def start_parameter(self, name, qname, attrs):
219          s = attrs.get((None, 'value'))          s = attrs.get((None, 'value'))
220          s = str(s) # we can't handle unicode in proj          s = str(s) # we can't handle unicode in proj
221          self.ProjectionParams.append(s)          self.ProjectionParams.append(s)
     start_dispatcher['parameter'] = "start_parameter"  
222    
223      def start_layer(self, name, qname, attrs, layer_class = Layer):      def start_layer(self, name, qname, attrs, layer_class = Layer):
224          """Start a layer          """Start a layer
# Line 146  class ProcessSession(xml.sax.handler.Con Line 229  class ProcessSession(xml.sax.handler.Con
229          """          """
230          title = attrs.get((None, 'title'), "")          title = attrs.get((None, 'title'), "")
231          filename = attrs.get((None, 'filename'), "")          filename = attrs.get((None, 'filename'), "")
232          filename = os.path.join(self.directory, filename)          filename = os.path.join(self.GetDirectory(), filename)
233          fill = parse_color(attrs.get((None, 'fill'), "None"))          fill = parse_color(attrs.get((None, 'fill'), "None"))
234          stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))          stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))
235          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))
236          self.aLayer = layer_class(title, filename, fill = fill,          self.aLayer = layer_class(title,
237                                    stroke = stroke, stroke_width = stroke_width)                                    self.theSession.OpenShapefile(filename),
238      start_dispatcher['layer'] = "start_layer"                                    fill = fill, stroke = stroke,
239                                      lineWidth = stroke_width)
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      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          self.aLayer.GetClassification().SetField(          field = attrs.get((None, 'field'), None)
249              attrs.get((None, 'field'), None))  
250      start_dispatcher['classification'] = "start_classification"          fieldType = attrs.get((None, 'field_type'), None)
251            dbFieldType = self.aLayer.GetFieldType(field)
252    
253            if fieldType != dbFieldType:
254                raise ValueError(_("xml field type differs from database!"))
255    
256            # setup conversion routines depending on the kind of data
257            # we will be seeing later on
258            if fieldType == FIELDTYPE_STRING:
259                self.conv = str
260            elif fieldType == FIELDTYPE_INT:
261                self.conv = lambda p: int(float(p))
262            elif fieldType == FIELDTYPE_DOUBLE:
263                self.conv = float
264    
265            self.aLayer.GetClassification().SetField(field)
266    
267    
268      def end_classification(self, name, qname):      def end_classification(self, name, qname):
269          pass          pass
     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()
     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
     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")
283    
284          try:          #try:
285              value  = Str2Num(attrib_value)              #value  = Str2Num(attrib_value)
286          except:          #except:
287              value  = attrib_value              #value  = attrib_value
288    
289            value = self.conv(attrib_value)
290    
291          self.cl_group = ClassGroupSingleton(value)          self.cl_group = ClassGroupSingleton(value)
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    
     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().AddGroup(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
     end_dispatcher['clpoint'] = "end_clpoint"  
300    
301      def start_clrange(self, name, qname, attrs):      def start_clrange(self, name, qname, attrs):
302    
303          try:          try:
304              min = Str2Num(attrs.get((None, 'min'), "0"))              min = self.conv(attrs.get((None, 'min'), "0"))
305              max = Str2Num(attrs.get((None, 'max'), "0"))              max = self.conv(attrs.get((None, 'max'), "0"))
306                #min = Str2Num(attrs.get((None, 'min'), "0"))
307                #max = Str2Num(attrs.get((None, 'max'), "0"))
308          except ValueError:          except ValueError:
309              raise ValueError(_("Classification range is not a number!"))              raise ValueError(_("Classification range is not a number!"))
310    
# Line 211  class ProcessSession(xml.sax.handler.Con Line 312  class ProcessSession(xml.sax.handler.Con
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    
     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().AddGroup(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
     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.SetStroke(parse_color(attrs.get((None, 'stroke'), "None")))          self.cl_prop.SetLineColor(
323          self.cl_prop.SetStrokeWidth(              parse_color(attrs.get((None, 'stroke'), "None")))
324            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")))
     start_dispatcher['cldata'] = "start_cldata"  
327    
328      def end_cldata(self, name, qname):      def end_cldata(self, name, qname):
329          pass          pass
     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      start_dispatcher['table'] = "start_table"          pass
334    
335      def end_table(self, name, qname):      def end_table(self, name, qname):
336          pass          pass
     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()
     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 249  class ProcessSession(xml.sax.handler.Con Line 345  class ProcessSession(xml.sax.handler.Con
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)
     start_dispatcher['label'] = "start_label"  
348    
349      def characters(self, chars):      def characters(self, chars):
350          pass          pass
# Line 257  class ProcessSession(xml.sax.handler.Con Line 352  class ProcessSession(xml.sax.handler.Con
352    
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      dir = os.path.dirname(filename)  
356      file = open(filename)      handler = SessionLoader()
357      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)  
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.439  
changed lines
  Added in v.744
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26