/[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 6 by bh, Tue Aug 28 15:41:52 2001 UTC revision 784 by jonathan, Tue Apr 29 17:30:18 2003 UTC
# Line 1  Line 1 
1  # Copyright (C) 2001 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]>
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 11  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
19    import xml.sax.handler
20    from xml.sax import make_parser, ErrorHandler, SAXNotRecognizedException
21    
22    from Thuban import _
23    
24    from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \
25         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
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.classification import Classification, \
33  oldPython=0      ClassGroupDefault, ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \
34        ClassGroupProperties
 if not sys.__dict__.has_key("version_info"):  
     # We can assume to have python 1.5.2 or lower here now  
     oldPython=1  
   
 if oldPython:  
     try:  
         from xml.sax.saxexts import make_parser  
         from xml.sax.saxlib import HandlerBase  
         from xml.sax import saxutils  
     except ImportError:  
         sys.stdout.write(("You need to have Python-XML installed or"  
                           " a modern Python!\n"  
                           "Check www.python.org/sigs/xml-sig/\n\n"))  
         raise  
 else:  
     # Do the python 2.0 standard xml thing and map it on the old names  
     import xml.sax  
     import xml.sax.handler  
     HandlerBase=xml.sax.handler.ContentHandler  
     from xml.sax import make_parser  
   
 class testSAXContentHandler(HandlerBase):  
 # SAX compliant  
     def characters(self, ch, start, length):  
         pass  
       
 def test_for_broken_SAX():  
     ch=testSAXContentHandler()  
     try:  
         xml.sax.parseString("""<?xml version="1.0"?>  
             <child1 name="paul">Text goes here</child1>  
         """,ch)  
     except TypeError:  
         return 1  
     return 0  
35    
36    
37  def parse_color(color):  def parse_color(color):
38      """      """Return the color object for the string color.
39      Return the color object for the string color. Color may be either  
40      'None' or of the form '#RRGGBB' in the usual HTML color notation      Color may be either 'None' or of the form '#RRGGBB' in the usual
41        HTML color notation
42      """      """
43      color = string.strip(color)      color = string.strip(color)
44      if color == "None":      if color == "None":
45          result = None          result = Color.Transparent
46      elif color[0] == '#':      elif color[0] == '#':
47          if len(color) == 7:          if len(color) == 7:
48              r = string.atoi(color[1:3], 16) / 255.0              r = string.atoi(color[1:3], 16) / 255.0
# Line 72  def parse_color(color): Line 50  def parse_color(color):
50              b = string.atoi(color[5:7], 16) / 255.0              b = string.atoi(color[5:7], 16) / 255.0
51              result = Color(r, g, b)              result = Color(r, g, b)
52          else:          else:
53              raise ValueError("Invalid hexadecimal color specification %s"              raise ValueError(_("Invalid hexadecimal color specification %s")
54                               % color)                               % color)
55      else:      else:
56          raise ValueError("Invalid color specification %s" % color)          raise ValueError(_("Invalid color specification %s") % color)
57      return result      return result
58    
59    
60  class ProcessSession(HandlerBase):  class XMLReader(xml.sax.handler.ContentHandler):
61    
62      def __init__(self, directory):      # Dictionary mapping element names (or (URI, element name) pairs for
63          """Inititialize the Sax handler.      # documents using namespaces) to method names. The methods should
64        # accept the same parameters as the startElement (or startElementNS)
65        # methods. The start_dispatcher is used by the default startElement
66        # and startElementNS methods to call a method for the open tag of an
67        # element.
68        start_dispatcher = {}
69    
70        # end_dispatcher works just like start_dispatcher but it's used by
71        # endElement and endElementNS. The method whose names it maps to
72        # should accept the same parameters as endElement and endElementNS.
73        end_dispatcher = {}
74    
75          directory is the directory containing the session file. It's  
76          needed to interpret embedded relative filenames      def __init__(self):
         """  
         self.directory = directory  
77          self.chars = ''          self.chars = ''
78            self.__directory = ""
79            self.__dispatchers = {}
80    
81        def read(self, file_or_filename):
82    
83            if hasattr(file_or_filename, "read"):
84                # it's a file object
85                self.__directory = ""
86                self.__file = file_or_filename
87            else:
88                filename = file_or_filename
89                self.__directory = os.path.dirname(filename)
90                self.__file = open(filename)
91    
92            parser = make_parser()
93            parser.setContentHandler(self)
94            parser.setErrorHandler(ErrorHandler())
95            parser.setFeature(xml.sax.handler.feature_namespaces, 1)
96    
97            #
98            # Well, this isn't pretty, but it appears that if you
99            # use Python 2.2 without the site-package _xmlplus then
100            # the following will fail, and without them it will work.
101            # However, if you do have the site-package and you don't
102            # call these functions, the reader raises an exception
103            #
104            # The reason we set these to 0 in the first place is
105            # because there is an unresolved issue with external
106            # entities causing an exception in the reader
107            #
108            try:
109                parser.setFeature(xml.sax.handler.feature_validation,0)
110                parser.setFeature(xml.sax.handler.feature_external_ges,0)
111                parser.setFeature(xml.sax.handler.feature_external_pes,0)
112            except SAXNotRecognizedException:
113                pass
114    
115            parser.parse(self.__file)
116    
117            self.close()
118    
119        def close(self):
120            self.__file.close()
121            
122        def GetFilename(self):
123            if hasattr(self.__file, "name"):
124                return self.__file.name
125    
126            return ""
127    
128        def GetDirectory(self):
129            return self.__directory
130    
131    
132        def AddDispatchers(self, dict):
133            """Add the function names that should be used to process XML tags.
134    
135            dict -- a dictionary whose keys are XML tag strings and whose values
136                    are pairs of strings such that the first string is
137                    the name of the function that should be called when the
138                    XML tag opens and the second string is the name of the
139                    function that should be called when the XML tag closes.
140                    If a pair element is None, no function is called.
141            """
142    
143            self.__dispatchers.update(dict)
144    
145        def startElementNS(self, name, qname, attrs):
146            """Call the method given for name in self.start_dispatcher
147            """
148            if name[0] is None:
149                method_name = self.__dispatchers.get(name[1])
150            else:
151                # Dispatch with namespace
152                method_name = self.__dispatchers.get(name)
153            if method_name is not None and method_name[0] is not None:
154                getattr(self, method_name[0])(name, qname, attrs)
155    
156        def endElementNS(self, name, qname):
157            """Call the method given for name in self.end_dispatcher
158            """
159            if name[0] is None:
160                method_name = self.__dispatchers.get(name[1])
161            else:
162                # Dispatch with namespace
163                method_name = self.__dispatchers.get(name)
164            if method_name is not None and method_name[1] is not None:
165                getattr(self, method_name[1])(name, qname)
166    
167    class SessionLoader(XMLReader):
168    
169        def __init__(self):
170            """Inititialize the Sax handler."""
171            XMLReader.__init__(self)
172    
173          self.theSession = None          self.theSession = None
174          self.aMap = None          self.aMap = None
175          self.aLayer = None          self.aLayer = None
176    
177      def startElement(self, name, attrs):          XMLReader.AddDispatchers(self,
178          if name == 'session':              {'session'       : ("start_session",        "end_session"),
179              self.theSession = Session(attrs.get('title', None))               'map'           : ("start_map",            "end_map"),
180          elif name == 'map':               'projection'    : ("start_projection",     "end_projection"),
181              self.aMap = Map(attrs.get('title', None))               'parameter'     : ("start_parameter",      None),
182          elif name == 'projection':               'layer'         : ("start_layer",          "end_layer"),
183              self.ProjectionParams = [ ]               'classification': ("start_classification", "end_classification"),
184          elif name == 'parameter':               'clnull'        : ("start_clnull",         "end_clnull"),
185              self.ProjectionParams.append(attrs.get('value', None))               'clpoint'       : ("start_clpoint",        "end_clpoint"),
186          elif name == 'layer':               'clrange'       : ("start_clrange",        "end_clrange"),
187              title = attrs.get('title', "")               'cldata'        : ("start_cldata",         "end_cldata"),
188              filename = attrs.get('filename', "")               'table'         : ("start_table",          "end_table"),
189              filename = os.path.join(self.directory, filename)               'labellayer'    : ("start_labellayer",     None),
190              fill = parse_color(attrs.get('fill', "None"))               'label'         : ("start_label",          None)})
191              stroke = parse_color(attrs.get('stroke', "#000000"))  
192              self.aLayer = Layer(title, filename, fill = fill, stroke = stroke)      def start_session(self, name, qname, attrs):
193          elif name == 'table':          self.theSession = Session(attrs.get((None, 'title'), None))
194              print "table title: %s" % attrs.get('title', None)  
195          elif name == 'labellayer':      def end_session(self, name, qname):
196              self.aLayer = self.aMap.LabelLayer()          pass
197          elif name == 'label':  
198              x = float(attrs['x'])      def start_map(self, name, qname, attrs):
199              y = float(attrs['y'])          """Start a map."""
200              text = attrs['text']          self.aMap = Map(attrs.get((None, 'title'), None))
201              halign = attrs['halign']          self.__projReceiver = self.aMap
202              valign = attrs['valign']  
203              self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign)      def end_map(self, name, qname):
204                    self.theSession.AddMap(self.aMap)
205            self.__projReceiver = None
206    
207        def start_projection(self, name, qname, attrs):
208            self.ProjectionName = attrs.get((None, 'name'), None)
209            self.ProjectionParams = [ ]
210    
211        def end_projection(self, name, qname):
212            self.__projReceiver.SetProjection(
213                Projection(self.ProjectionParams, self.ProjectionName))
214    
215        def start_parameter(self, name, qname, attrs):
216            s = attrs.get((None, 'value'))
217            s = str(s) # we can't handle unicode in proj
218            self.ProjectionParams.append(s)
219    
220        def start_layer(self, name, qname, attrs, layer_class = Layer):
221            """Start a layer
222    
223            Instantiate a layer of class layer_class from the attributes in
224            attrs which may be a dictionary as well as the normal SAX attrs
225            object and bind it to self.aLayer.
226            """
227            title = attrs.get((None, 'title'), "")
228            filename = attrs.get((None, 'filename'), "")
229            filename = os.path.join(self.GetDirectory(), filename)
230            visible  = attrs.get((None, 'visible'), "true")
231            fill = parse_color(attrs.get((None, 'fill'), "None"))
232            stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))
233            stroke_width = int(attrs.get((None, 'stroke_width'), "1"))
234            self.aLayer = layer_class(title,
235                                      self.theSession.OpenShapefile(filename),
236                                      fill = fill, stroke = stroke,
237                                      lineWidth = stroke_width,
238                                      visible = visible != "false")
239    
240            self.__projReceiver = self.aLayer
241    
242        def end_layer(self, name, qname):
243            self.aMap.AddLayer(self.aLayer)
244            self.__projReceiver = None
245    
246        def start_classification(self, name, qname, attrs):
247            field = attrs.get((None, 'field'), None)
248    
249            fieldType = attrs.get((None, 'field_type'), None)
250            dbFieldType = self.aLayer.GetFieldType(field)
251    
252            if fieldType != dbFieldType:
253                raise ValueError(_("xml field type differs from database!"))
254    
255            # setup conversion routines depending on the kind of data
256            # we will be seeing later on
257            if fieldType == FIELDTYPE_STRING:
258                self.conv = str
259            elif fieldType == FIELDTYPE_INT:
260                self.conv = lambda p: int(float(p))
261            elif fieldType == FIELDTYPE_DOUBLE:
262                self.conv = float
263    
264            self.aLayer.GetClassification().SetField(field)
265    
266    
267        def end_classification(self, name, qname):
268            pass
269    
270        def start_clnull(self, name, qname, attrs):
271            self.cl_group = ClassGroupDefault()
272            self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
273            self.cl_prop = ClassGroupProperties()
274    
275        def end_clnull(self, name, qname):
276            self.cl_group.SetProperties(self.cl_prop)
277            self.aLayer.GetClassification().SetDefaultGroup(self.cl_group)
278            del self.cl_group, self.cl_prop
279    
280        def start_clpoint(self, name, qname, attrs):
281            attrib_value = attrs.get((None, 'value'), "0")
282    
283            value = self.conv(attrib_value)
284    
285            self.cl_group = ClassGroupSingleton(value)
286            self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
287            self.cl_prop = ClassGroupProperties()
288    
289    
290        def end_clpoint(self, name, qname):
291            self.cl_group.SetProperties(self.cl_prop)
292            self.aLayer.GetClassification().AppendGroup(self.cl_group)
293            del self.cl_group, self.cl_prop
294    
295        def start_clrange(self, name, qname, attrs):
296    
297            try:
298                min = self.conv(attrs.get((None, 'min'), "0"))
299                max = self.conv(attrs.get((None, 'max'), "0"))
300            except ValueError:
301                raise ValueError(_("Classification range is not a number!"))
302    
303            self.cl_group = ClassGroupRange(min, max)
304            self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
305            self.cl_prop = ClassGroupProperties()
306    
307    
308        def end_clrange(self, name, qname):
309            self.cl_group.SetProperties(self.cl_prop)
310            self.aLayer.GetClassification().AppendGroup(self.cl_group)
311            del self.cl_group, self.cl_prop
312    
313        def start_cldata(self, name, qname, attrs):
314            self.cl_prop.SetLineColor(
315                parse_color(attrs.get((None, 'stroke'), "None")))
316            self.cl_prop.SetLineWidth(
317                int(attrs.get((None, 'stroke_width'), "0")))
318            self.cl_prop.SetFill(parse_color(attrs.get((None, 'fill'), "None")))
319    
320        def end_cldata(self, name, qname):
321            pass
322    
323        def start_table(self, name, qname, attrs):
324            #print "table title: %s" % attrs.get('title', None)
325            pass
326    
327        def end_table(self, name, qname):
328            pass
329    
330        def start_labellayer(self, name, qname, attrs):
331            self.aLayer = self.aMap.LabelLayer()
332    
333        def start_label(self, name, qname, attrs):
334            x = float(attrs[(None, 'x')])
335            y = float(attrs[(None, 'y')])
336            text = attrs[(None, 'text')]
337            halign = attrs[(None, 'halign')]
338            valign = attrs[(None, 'valign')]
339            self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign)
340    
341        def characters(self, chars):
342            pass
343    
     if not oldPython and test_for_broken_SAX():  
         # works with python 2.0, but is not SAX compliant  
         def characters(self, ch):  
             self.my_characters(ch)  
     else:  
         # SAX compliant  
         def characters(self, ch, start, length):  
             self.my_characters(ch[start:start+length])  
   
     def my_characters(self, ch):  
         self.chars = self.chars + ch  
   
     def endElement(self, name):  
         # If it's not a parameter element, ignore it  
         if name == 'session':  
             #print "end of session"  
             pass  
         if name == 'map':  
             self.theSession.AddMap(self.aMap)  
         if name == 'projection':  
             self.aMap.SetProjection(Projection(self.ProjectionParams))  
         if name == 'layer':  
             self.aMap.AddLayer(self.aLayer)  
         if name == 'table':  
             #print "end of table"  
             pass  
344    
345  def load_session(filename):  def load_session(filename):
346      """Load a Thuban session from the file object file"""      """Load a Thuban session from the file object file"""
     dir = os.path.dirname(filename)  
     file = open(filename)  
     handler = ProcessSession(dir)  
347    
348      if oldPython:      handler = SessionLoader()
349          parser = make_parser()      handler.read(filename)
350          parser.setDocumentHandler(handler)  
         parser.setErrorHandler(saxutils.ErrorPrinter())  
         parser.parseFile(file)  
         parser.close()  
     else:  
         xml.sax.parse(file,handler)  
351      session = handler.theSession      session = handler.theSession
352      # Newly loaded session aren't modified      # Newly loaded session aren't modified
353      session.UnsetModified()      session.UnsetModified()
354    
355      return session      return session
356    
 if __name__ == "__main__":  
     # find out the command to run  
     if len(sys.argv) > 1:  
         print "usage: cat <file> | " + sys.argv[0]  
     else:  
         parseSession(sys.stdin)  
Legend:
Removed from v.6  
changed lines
  Added in v.784
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26