/[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 763 by bh, Tue Apr 29 10:55:26 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    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
31  from Thuban.Model.color import Color  from Thuban.Model.color import Color
32  from Thuban.Model.proj import Projection  from Thuban.Model.proj import Projection
33  from Thuban.Model.classification import Classification  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    class SessionLoader(XMLReader):
169    
170        def __init__(self):
171            """Inititialize the Sax handler."""
172            XMLReader.__init__(self)
173    
174            self.theSession = None
175            self.aMap = None
176            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))
     start_dispatcher['session'] = "start_session"  
195    
196      def end_session(self, name, qname):      def end_session(self, name, qname):
197          pass          pass
     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      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      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 = [ ]
     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      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)
     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 140  class ProcessSession(xml.sax.handler.Con Line 227  class ProcessSession(xml.sax.handler.Con
227          """          """
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.directory, filename)          filename = os.path.join(self.GetDirectory(), filename)
231          fill = parse_color(attrs.get((None, 'fill'), "None"))          fill = parse_color(attrs.get((None, 'fill'), "None"))
232          stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))          stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))
233          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))
234          self.aLayer = layer_class(title, filename, fill = fill,          self.aLayer = layer_class(title,
235                                    stroke = stroke, stroke_width = stroke_width)                                    self.theSession.OpenShapefile(filename),
236      start_dispatcher['layer'] = "start_layer"                                    fill = fill, stroke = stroke,
237                                      lineWidth = stroke_width)
238    
239            self.__projReceiver = self.aLayer
240    
241      def end_layer(self, name, qname):      def end_layer(self, name, qname):
242          self.aMap.AddLayer(self.aLayer)          self.aMap.AddLayer(self.aLayer)
243      end_dispatcher['layer'] = "end_layer"          self.__projReceiver = None
244    
245      def start_classification(self, name, qname, attrs):      def start_classification(self, name, qname, attrs):
246          self.aLayer.classification.setField(attrs.get((None, 'field'), None))          field = attrs.get((None, 'field'), None)
247      start_dispatcher['classification'] = "start_classification"  
248            fieldType = attrs.get((None, 'field_type'), None)
249            dbFieldType = self.aLayer.GetFieldType(field)
250    
251            if fieldType != dbFieldType:
252                raise ValueError(_("xml field type differs from database!"))
253    
254            # setup conversion routines depending on the kind of data
255            # we will be seeing later on
256            if fieldType == FIELDTYPE_STRING:
257                self.conv = str
258            elif fieldType == FIELDTYPE_INT:
259                self.conv = lambda p: int(float(p))
260            elif fieldType == FIELDTYPE_DOUBLE:
261                self.conv = float
262    
263            self.aLayer.GetClassification().SetField(field)
264    
265    
266      def end_classification(self, name, qname):      def end_classification(self, name, qname):
267          pass          pass
     end_dispatcher['classification'] = "end_classification"  
268    
269      def start_clnull(self, name, qname, attrs):      def start_clnull(self, name, qname, attrs):
270          self.cl_data = {}          self.cl_group = ClassGroupDefault()
271      start_dispatcher['clnull'] = "start_clnull"          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
272            self.cl_prop = ClassGroupProperties()
273    
274      def end_clnull(self, name, qname):      def end_clnull(self, name, qname):
275          self.aLayer.classification.setNull(self.cl_data)          self.cl_group.SetProperties(self.cl_prop)
276          del self.cl_data          self.aLayer.GetClassification().SetDefaultGroup(self.cl_group)
277      end_dispatcher['clnull'] = "end_clnull"          del self.cl_group, self.cl_prop
278    
279      def start_clpoint(self, name, qname, attrs):      def start_clpoint(self, name, qname, attrs):
280          attrib_value = attrs.get((None, 'value'), "0")          attrib_value = attrs.get((None, 'value'), "0")
281    
282          try:          #try:
283              self.cl_value  = int(attrib_value)              #value  = Str2Num(attrib_value)
284          except:          #except:
285              self.cl_value  = attrib_value              #value  = attrib_value
286    
287            value = self.conv(attrib_value)
288    
289            self.cl_group = ClassGroupSingleton(value)
290            self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
291            self.cl_prop = ClassGroupProperties()
292    
         self.cl_data = {}  
     start_dispatcher['clpoint'] = "start_clpoint"  
293    
294      def end_clpoint(self, name, qname):      def end_clpoint(self, name, qname):
295          self.aLayer.classification.addPoint(self.cl_value, self.cl_data)          self.cl_group.SetProperties(self.cl_prop)
296          del self.cl_value, self.cl_data          self.aLayer.GetClassification().AppendGroup(self.cl_group)
297      end_dispatcher['clpoint'] = "end_clpoint"          del self.cl_group, self.cl_prop
298    
299      def start_clrange(self, name, qname, attrs):      def start_clrange(self, name, qname, attrs):
300    
301          try:          try:
302              self.cl_low = int(attrs.get((None, 'low'), "0"))              min = self.conv(attrs.get((None, 'min'), "0"))
303              self.cl_high = int(attrs.get((None, 'high'), "0"))              max = self.conv(attrs.get((None, 'max'), "0"))
304                #min = Str2Num(attrs.get((None, 'min'), "0"))
305                #max = Str2Num(attrs.get((None, 'max'), "0"))
306          except ValueError:          except ValueError:
307              raise ValueError("Classification range is not a number!")              raise ValueError(_("Classification range is not a number!"))
308    
309            self.cl_group = ClassGroupRange(min, max)
310            self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
311            self.cl_prop = ClassGroupProperties()
312    
         self.cl_data = {}  
     start_dispatcher['clrange'] = "start_clrange"  
313    
314      def end_clrange(self, name, qname):      def end_clrange(self, name, qname):
315          self.aLayer.classification.addRange(          self.cl_group.SetProperties(self.cl_prop)
316              self.cl_low, self.cl_high, self.cl_data)          self.aLayer.GetClassification().AppendGroup(self.cl_group)
317          del self.cl_low, self.cl_high, self.cl_data          del self.cl_group, self.cl_prop
     end_dispatcher['clrange'] = "end_clrange"  
318    
319      def start_cldata(self, name, qname, attrs):      def start_cldata(self, name, qname, attrs):
320          self.cl_data['stroke'] = parse_color(          self.cl_prop.SetLineColor(
321                                   attrs.get((None, 'stroke'), "None"))              parse_color(attrs.get((None, 'stroke'), "None")))
322          self.cl_data['stroke_width'] = int(          self.cl_prop.SetLineWidth(
323                                   attrs.get((None, 'stroke_width'), "0"))              int(attrs.get((None, 'stroke_width'), "0")))
324          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"  
325    
326      def end_cldata(self, name, qname):      def end_cldata(self, name, qname):
327          pass          pass
     end_dispatcher['cldata'] = "end_cldata"  
328    
329      def start_table(self, name, qname, attrs):      def start_table(self, name, qname, attrs):
330          print "table title: %s" % attrs.get('title', None)          #print "table title: %s" % attrs.get('title', None)
331      start_dispatcher['table'] = "start_table"          pass
332    
333      def end_table(self, name, qname):      def end_table(self, name, qname):
334          pass          pass
     end_dispatcher['table'] = "end_table"  
335    
336      def start_labellayer(self, name, qname, attrs):      def start_labellayer(self, name, qname, attrs):
337          self.aLayer = self.aMap.LabelLayer()          self.aLayer = self.aMap.LabelLayer()
     start_dispatcher['labellayer'] = "start_labellayer"  
338    
339      def start_label(self, name, qname, attrs):      def start_label(self, name, qname, attrs):
340          x = float(attrs[(None, 'x')])          x = float(attrs[(None, 'x')])
# Line 234  class ProcessSession(xml.sax.handler.Con Line 343  class ProcessSession(xml.sax.handler.Con
343          halign = attrs[(None, 'halign')]          halign = attrs[(None, 'halign')]
344          valign = attrs[(None, 'valign')]          valign = attrs[(None, 'valign')]
345          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"  
346    
347      def characters(self, chars):      def characters(self, chars):
348          pass          pass
# Line 242  class ProcessSession(xml.sax.handler.Con Line 350  class ProcessSession(xml.sax.handler.Con
350    
351  def load_session(filename):  def load_session(filename):
352      """Load a Thuban session from the file object file"""      """Load a Thuban session from the file object file"""
353      dir = os.path.dirname(filename)  
354      file = open(filename)      handler = SessionLoader()
355      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)  
356    
357      session = handler.theSession      session = handler.theSession
358      # Newly loaded session aren't modified      # Newly loaded session aren't modified
Legend:
Removed from v.365  
changed lines
  Added in v.763
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26