/[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 874 by jonathan, Fri May 9 16:31:12 2003 UTC revision 2615 by jonathan, Fri May 6 14:17:03 2005 UTC
# Line 1  Line 1 
1  # Copyright (C) 2001, 2002, 2003 by Intevation GmbH  # Copyright (C) 2001, 2002, 2003, 2004 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]>  # Jonathan Coles <[email protected]>
6    # Frank Koormann <[email protected]>
7  #  #
8  # This program is free software under the GPL (>=v2)  # This program is free software under the GPL (>=v2)
9  # Read the file COPYING coming with GRASS for details.  # Read the file COPYING coming with GRASS for details.
# Line 24  from Thuban import _ Line 25  from Thuban import _
25  from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \  from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \
26       FIELDTYPE_STRING       FIELDTYPE_STRING
27    
28    from Thuban.Model.color import Color, Transparent
29    
30  from Thuban.Model.session import Session  from Thuban.Model.session import Session
31  from Thuban.Model.map import Map  from Thuban.Model.map import Map
32  from Thuban.Model.layer import Layer  from Thuban.Model.layer import Layer, RasterLayer
 from Thuban.Model.color import Color  
33  from Thuban.Model.proj import Projection  from Thuban.Model.proj import Projection
34  from Thuban.Model.range import Range  from Thuban.Model.range import Range
35  from Thuban.Model.classification import Classification, \  from Thuban.Model.classification import Classification, \
36      ClassGroupDefault, ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \      ClassGroupDefault, ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \
37      ClassGroupProperties      ClassGroupProperties
38    from Thuban.Model.data import DerivedShapeStore, ShapefileStore
39    from Thuban.Model.table import DBFTable
40    from Thuban.Model.transientdb import TransientJoinedTable
41    
42    from Thuban.Model.xmlreader import XMLReader
43    import resource
44    
45    import postgisdb
46    
47    class LoadError(Exception):
48    
49        """Exception raised when the thuban file is corrupted
50    
51        Not all cases of corrupted thuban files will lead to this exception
52        but those that are found by checks in the loading code itself are.
53        """
54    
55    
56    class LoadCancelled(Exception):
57    
58        """Exception raised to indicate that loading was interrupted by the user"""
59    
60    
61  def parse_color(color):  def parse_color(color):
# Line 43  def parse_color(color): Line 66  def parse_color(color):
66      """      """
67      color = string.strip(color)      color = string.strip(color)
68      if color == "None":      if color == "None":
69          result = Color.Transparent          result = Transparent
70      elif color[0] == '#':      elif color[0] == '#':
71          if len(color) == 7:          if len(color) == 7:
72              r = string.atoi(color[1:3], 16) / 255.0              r = string.atoi(color[1:3], 16) / 255.0
# Line 57  def parse_color(color): Line 80  def parse_color(color):
80          raise ValueError(_("Invalid color specification %s") % color)          raise ValueError(_("Invalid color specification %s") % color)
81      return result      return result
82    
83    class AttrDesc:
84    
85  class XMLReader(xml.sax.handler.ContentHandler):      def __init__(self, name, required = False, default = "",
86                     conversion = None):
87      # Dictionary mapping element names (or (URI, element name) pairs for          if not isinstance(name, tuple):
88      # documents using namespaces) to method names. The methods should              fullname = (None, name)
     # accept the same parameters as the startElement (or startElementNS)  
     # methods. The start_dispatcher is used by the default startElement  
     # and startElementNS methods to call a method for the open tag of an  
     # element.  
     start_dispatcher = {}  
   
     # end_dispatcher works just like start_dispatcher but it's used by  
     # endElement and endElementNS. The method whose names it maps to  
     # should accept the same parameters as endElement and endElementNS.  
     end_dispatcher = {}  
   
   
     def __init__(self):  
         self.chars = ''  
         self.__directory = ""  
         self.__dispatchers = {}  
   
     def read(self, file_or_filename):  
   
         if hasattr(file_or_filename, "read"):  
             # it's a file object  
             self.__directory = ""  
             self.__file = file_or_filename  
89          else:          else:
90              filename = file_or_filename              fullname = name
91              self.__directory = os.path.dirname(filename)              name = name[1]
92              self.__file = open(filename)          self.name = name
93            self.fullname = fullname
94          parser = make_parser()          self.required = required
95          parser.setContentHandler(self)          self.default = default
96          parser.setErrorHandler(ErrorHandler())          self.conversion = conversion
         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  
97    
98          parser.parse(self.__file)          # set by the SessionLoader's check_attrs method
99            self.value = None
100    
         self.close()  
101    
102      def close(self):  class SessionLoader(XMLReader):
         self.__file.close()  
           
     def GetFilename(self):  
         if hasattr(self.__file, "name"):  
             return self.__file.name  
103    
104          return ""      def __init__(self, db_connection_callback = None,
105                           shapefile_callback = None):
106            """Inititialize the Sax handler."""
107            XMLReader.__init__(self)
108    
109      def GetDirectory(self):          self.db_connection_callback = db_connection_callback
110          return self.__directory          self.shapefile_callback = shapefile_callback
111            self.theSession = None
112            self.aMap = None
113            self.aLayer = None
114    
115            # Map ids used in the thuban file to the corresponding objects
116            # in the session
117            self.idmap = {}
118    
119            dispatchers = {
120                'session'       : ("start_session",        "end_session"),
121    
122                'dbconnection': ("start_dbconnection", None),
123    
124                'dbshapesource': ("start_dbshapesource", None),
125                'fileshapesource': ("start_fileshapesource", None),
126                'derivedshapesource': ("start_derivedshapesource", None),
127                'filetable': ("start_filetable", None),
128                'jointable': ("start_jointable", None),
129    
130                'map'           : ("start_map",            "end_map"),
131                'projection'    : ("start_projection",     "end_projection"),
132                'parameter'     : ("start_parameter",      None),
133                'layer'         : ("start_layer",          "end_layer"),
134                'rasterlayer'   : ("start_rasterlayer",    "end_rasterlayer"),
135                'classification': ("start_classification", "end_classification"),
136                'clnull'        : ("start_clnull",         "end_clnull"),
137                'clpoint'       : ("start_clpoint",        "end_clpoint"),
138                'clrange'       : ("start_clrange",        "end_clrange"),
139                'cldata'        : ("start_cldata",         "end_cldata"),
140                'table'         : ("start_table",          "end_table"),
141                'labellayer'    : ("start_labellayer",     None),
142                'label'         : ("start_label",          None)}
143    
144            # all dispatchers should be used for the 0.8 and 0.9 namespaces too
145            for xmlns in ("http://thuban.intevation.org/dtds/thuban-0.8.dtd",
146                          "http://thuban.intevation.org/dtds/thuban-0.9-dev.dtd",
147                          "http://thuban.intevation.org/dtds/thuban-0.9.dtd",
148                          "http://thuban.intevation.org/dtds/thuban-1.0-dev.dtd",
149                          "http://thuban.intevation.org/dtds/thuban-1.0rc1.dtd",
150                          "http://thuban.intevation.org/dtds/thuban-1.0.0.dtd",
151                          "http://thuban.intevation.org/dtds/thuban-1.1-dev.dtd"):
152                for key, value in dispatchers.items():
153                    dispatchers[(xmlns, key)] = value
154    
155            XMLReader.AddDispatchers(self, dispatchers)
156    
157        def Destroy(self):
158            """Clear all instance variables to cut cyclic references.
159    
160            The GC would have collected the loader eventually but it can
161            happen that it doesn't run at all until Thuban is closed (2.3
162            but not 2.2 tries a bit harder and forces a collection when the
163            interpreter terminates)
164            """
165            self.__dict__.clear()
166    
167      def AddDispatchers(self, dict):      def start_session(self, name, qname, attrs):
168          """Add the function names that should be used to process XML tags.          self.theSession = Session(self.encode(attrs.get((None, 'title'),
169                                                            None)))
170    
171          dict -- a dictionary whose keys are XML tag strings and whose values      def end_session(self, name, qname):
172                  are pairs of strings such that the first string is          pass
                 the name of the function that should be called when the  
                 XML tag opens and the second string is the name of the  
                 function that should be called when the XML tag closes.  
                 If a pair element is None, no function is called.  
         """  
173    
174          self.__dispatchers.update(dict)      def check_attrs(self, element, attrs, descr):
175            """Check and convert some of the attributes of an element
176    
177      def startElementNS(self, name, qname, attrs):          Parameters:
178          """Call the method given for name in self.start_dispatcher             element -- The element name
179          """             attrs -- The attrs mapping as passed to the start_* methods
180          if name[0] is None:             descr -- Sequence of attribute descriptions (AttrDesc instances)
             method_name = self.__dispatchers.get(name[1])  
         else:  
             # Dispatch with namespace  
             method_name = self.__dispatchers.get(name)  
         if method_name is not None and method_name[0] is not None:  
             getattr(self, method_name[0])(name, qname, attrs)  
181    
182      def endElementNS(self, name, qname):          Return a dictionary containig normalized versions of the
183          """Call the method given for name in self.end_dispatcher          attributes described in descr. The keys of that dictionary are
184          """          the name attributes of the attribute descriptions. The attrs
185          if name[0] is None:          dictionary will not be modified.
             method_name = self.__dispatchers.get(name[1])  
         else:  
             # Dispatch with namespace  
             method_name = self.__dispatchers.get(name)  
         if method_name is not None and method_name[1] is not None:  
             getattr(self, method_name[1])(name, qname)  
   
     def encode(self, str):  
         """Assume that str is in Unicode and encode it into Latin1.  
           
         If str is None, return None  
         """  
186    
187          if str is not None:          If the attribute is required, i.e. the 'required' attribute of
188              return str.encode("latin1")          the descrtiption is true, but it is not in attrs, raise a
189          else:          LoadError.
             return None  
190    
191  class SessionLoader(XMLReader):          If the attribute has a default value and it is not present in
192            attrs, use that default value as the value in the returned dict.
193    
194      def __init__(self):          The value is converted before putting it into the returned dict.
195          """Inititialize the Sax handler."""          The following conversions are available:
         XMLReader.__init__(self)  
196    
197          self.theSession = None             'filename' -- The attribute is a filename.
         self.aMap = None  
         self.aLayer = None  
198    
199          XMLReader.AddDispatchers(self,                           If the filename is a relative name, interpret
200              {'session'       : ("start_session",        "end_session"),                           it relative to the directory containing the
201               'map'           : ("start_map",            "end_map"),                           .thuban file and make it an absolute name
              'projection'    : ("start_projection",     "end_projection"),  
              'parameter'     : ("start_parameter",      None),  
              'layer'         : ("start_layer",          "end_layer"),  
              'classification': ("start_classification", "end_classification"),  
              'clnull'        : ("start_clnull",         "end_clnull"),  
              'clpoint'       : ("start_clpoint",        "end_clpoint"),  
              'clrange'       : ("start_clrange",        "end_clrange"),  
              'cldata'        : ("start_cldata",         "end_cldata"),  
              'table'         : ("start_table",          "end_table"),  
              'labellayer'    : ("start_labellayer",     None),  
              'label'         : ("start_label",          None)})  
202    
203      def start_session(self, name, qname, attrs):             'shapestore' -- The attribute is the ID of a shapestore
204          self.theSession = Session(self.encode(attrs.get((None, 'title'), None)))                             defined earlier in the .thuban file. Look it
205                               up self.idmap
206    
207      def end_session(self, name, qname):             'table' -- The attribute is the ID of a table or shapestore
208          pass                        defined earlier in the .thuban file. Look it up
209                          self.idmap. If it's the ID of a shapestore the
210                          value will be the table of the shapestore.
211    
212               'idref' -- The attribute is the id of an object defined
213                          earlier in the .thuban file. Look it up self.idmap
214    
215               'ascii' -- The attribute is converted to a bytestring with
216                          ascii encoding.
217    
218               a callable -- The attribute value is passed to the callable
219                             and the return value is used as the converted
220                             value
221    
222            If no conversion is specified for an attribute it is converted
223            with self.encode.
224            """
225            normalized = {}
226    
227            for d in descr:
228                if d.required and not attrs.has_key(d.fullname):
229                    raise LoadError("Element %s requires an attribute %r"
230                                    % (element, d.name))
231                value = attrs.get(d.fullname, d.default)
232    
233                if d.conversion in ("idref", "shapesource"):
234                    if value in self.idmap:
235                        value = self.idmap[value]
236                    else:
237                        raise LoadError("Element %s requires an already defined ID"
238                                        " in attribute %r"
239                                        % (element, d.name))
240                elif d.conversion == "table":
241                    if value in self.idmap:
242                        value = self.idmap[value]
243                        if isinstance(value, ShapefileStore):
244                            value = value.Table()
245                    else:
246                        raise LoadError("Element %s requires an already defined ID"
247                                        " in attribute %r"
248                                        % (element, d.name))
249                elif d.conversion == "filename":
250                    value = os.path.abspath(os.path.join(self.GetDirectory(),
251                                                         value))
252                elif d.conversion == "ascii":
253                    value = value.encode("ascii")
254                elif d.conversion:
255                    # Assume it's a callable
256                    value = d.conversion(value)
257                else:
258                   value = self.encode(value)
259    
260                normalized[d.name] = value
261            return normalized
262    
263        def open_shapefile(self, filename):
264            """Open shapefile, with alternative path handling.
265            
266               If a shapefile cannot be opened and an IOError is raised, check for
267               an alternative. This alternative can be specified interactively by
268               the user or taken from a list of (potential) locations, depending on
269               the callback implementation.
270                
271               The alternative is rechecked. If taken from a list the user
272               has to confirm the alternative.
273            """
274    
275            # Flag if the alternative path was specified interactively / from list.
276            from_list = 0
277            while 1:
278                try:
279                    store = self.theSession.OpenShapefile(filename)
280                    if from_list:
281                        # A valid path has been guessed from a list
282                        # Let the user confirm - or select an alternative.
283                        filename, from_list = self.shapefile_callback(
284                                                filename, "check")
285                        if filename is None:
286                            # Selection cancelled
287                            raise LoadCancelled
288                        elif store.FileName() == filename:
289                            # Proposed file has been accepted
290                            break
291                        else:
292                            # the filename has been changed, try the new file
293                            pass
294                    else:
295                        break
296                except IOError:
297                    if self.shapefile_callback is not None:
298                        filename, from_list = self.shapefile_callback(
299                                                filename,
300                                                mode = "search",
301                                                second_try = from_list)
302                        if filename is None:
303                            raise LoadCancelled
304                    else:
305                        raise
306            return store
307    
308        def start_dbconnection(self, name, qname, attrs):
309            attrs = self.check_attrs(name, attrs,
310                                     [AttrDesc("id", True),
311                                      AttrDesc("dbtype", True),
312                                      AttrDesc("host", False, ""),
313                                      AttrDesc("port", False, ""),
314                                      AttrDesc("user", False, ""),
315                                      AttrDesc("dbname", True)])
316            ID = attrs["id"]
317            dbtype = attrs["dbtype"]
318            if dbtype != "postgis":
319                raise LoadError("dbtype %r not supported" % filetype)
320    
321            del attrs["id"]
322            del attrs["dbtype"]
323    
324            # Try to open the connection and if it fails ask the user for
325            # the correct parameters repeatedly.
326            # FIXME: it would be better not to insist on getting a
327            # connection here. We should handle this more like the raster
328            # images where the layers etc still are created but are not
329            # drawn in case Thuban can't use the data for various reasons
330            while 1:
331                try:
332                    conn = postgisdb.PostGISConnection(**attrs)
333                    break
334                except postgisdb.ConnectionError, val:
335                    if self.db_connection_callback is not None:
336                        attrs = self.db_connection_callback(attrs, str(val))
337                        if attrs is None:
338                            raise LoadCancelled
339                    else:
340                        raise
341    
342            self.idmap[ID] = conn
343            self.theSession.AddDBConnection(conn)
344    
345        def start_dbshapesource(self, name, qname, attrs):
346            attrs = self.check_attrs(name, attrs,
347                                     [AttrDesc("id", True),
348                                      AttrDesc("dbconn", True,
349                                               conversion = "idref"),
350                                      AttrDesc("tablename", True,
351                                               conversion = "ascii"),
352                                      # id_column and geometry_column were
353                                      # newly introduced with thuban-1.1.dtd
354                                      # where they're required.  Since we
355                                      # support the older formats too we
356                                      # have them optional here.
357                                      AttrDesc("id_column", False, "gid",
358                                               conversion = "ascii"),
359                                      AttrDesc("geometry_column", False,
360                                               conversion = "ascii")])
361            # The default value of geometry_column to use when instantiating
362            # the db shapestore is None which we currently can't easily use
363            # in check_attrs
364            geometry_column = attrs["geometry_column"]
365            if not geometry_column:
366                geometry_column = None
367            dbopen = self.theSession.OpenDBShapeStore
368            self.idmap[attrs["id"]] = dbopen(attrs["dbconn"], attrs["tablename"],
369                                             id_column = attrs["id_column"],
370                                             geometry_column=geometry_column)
371    
372        def start_fileshapesource(self, name, qname, attrs):
373            attrs = self.check_attrs(name, attrs,
374                                      [AttrDesc("id", True),
375                                       AttrDesc("filename", True,
376                                                conversion = "filename"),
377                                       AttrDesc("filetype", True)])
378            ID = attrs["id"]
379            filename = attrs["filename"]
380            filetype = attrs["filetype"]
381            if filetype != "shapefile":
382                raise LoadError("shapesource filetype %r not supported" % filetype)
383            self.idmap[ID] = self.open_shapefile(filename)
384    
385        def start_derivedshapesource(self, name, qname, attrs):
386            attrs = self.check_attrs(name, attrs,
387                                     [AttrDesc("id", True),
388                                      AttrDesc("shapesource", True,
389                                               conversion = "shapesource"),
390                                      AttrDesc("table", True, conversion="table")])
391            store = DerivedShapeStore(attrs["shapesource"], attrs["table"])
392            self.theSession.AddShapeStore(store)
393            self.idmap[attrs["id"]] = store
394    
395        def start_filetable(self, name, qname, attrs):
396            attrs = self.check_attrs(name, attrs,
397                                     [AttrDesc("id", True),
398                                      AttrDesc("title", True),
399                                      AttrDesc("filename", True,
400                                               conversion = "filename"),
401                                      AttrDesc("filetype")])
402            filetype = attrs["filetype"]
403            if filetype != "DBF":
404                raise LoadError("shapesource filetype %r not supported" % filetype)
405            table = DBFTable(attrs["filename"])
406            table.SetTitle(attrs["title"])
407            self.idmap[attrs["id"]] = self.theSession.AddTable(table)
408    
409        def start_jointable(self, name, qname, attrs):
410            attrs = self.check_attrs(name, attrs,
411                                     [AttrDesc("id", True),
412                                      AttrDesc("title", True),
413                                      AttrDesc("left", True, conversion="table"),
414                                      AttrDesc("leftcolumn", True),
415                                      AttrDesc("right", True, conversion="table"),
416                                      AttrDesc("rightcolumn", True),
417    
418                                      # jointype is required for file
419                                      # version 0.9 but this attribute
420                                      # wasn't in the 0.8 version because of
421                                      # an oversight so we assume it's
422                                      # optional since we want to handle
423                                      # both file format versions here.
424                                      AttrDesc("jointype", False,
425                                               default="INNER")])
426    
427            jointype = attrs["jointype"]
428            if jointype == "LEFT OUTER":
429                outer_join = True
430            elif jointype == "INNER":
431                outer_join = False
432            else:
433                raise LoadError("jointype %r not supported" % jointype )
434            table = TransientJoinedTable(self.theSession.TransientDB(),
435                                         attrs["left"], attrs["leftcolumn"],
436                                         attrs["right"], attrs["rightcolumn"],
437                                         outer_join = outer_join)
438            table.SetTitle(attrs["title"])
439            self.idmap[attrs["id"]] = self.theSession.AddTable(table)
440    
441      def start_map(self, name, qname, attrs):      def start_map(self, name, qname, attrs):
442          """Start a map."""          """Start a map."""
443          self.aMap = Map(attrs.get((None, 'title'), None))          self.aMap = Map(self.encode(attrs.get((None, 'title'), None)))
444    
445      def end_map(self, name, qname):      def end_map(self, name, qname):
446          self.theSession.AddMap(self.aMap)          self.theSession.AddMap(self.aMap)
447          self.aMap = None          self.aMap = None
448    
449      def start_projection(self, name, qname, attrs):      def start_projection(self, name, qname, attrs):
450          self.ProjectionName = self.encode(attrs.get((None, 'name'), None))          attrs = self.check_attrs(name, attrs,
451          self.ProjectionParams = [ ]                                   [AttrDesc("name", conversion=self.encode),
452                                      AttrDesc("epsg", default=None,
453                                               conversion=self.encode)])
454            self.projection_name = attrs["name"]
455            self.projection_epsg = attrs["epsg"]
456            self.projection_params = [ ]
457    
458      def end_projection(self, name, qname):      def end_projection(self, name, qname):
459          if self.aLayer is not None:          if self.aLayer is not None:
# Line 228  class SessionLoader(XMLReader): Line 464  class SessionLoader(XMLReader):
464              assert False, "projection tag out of context"              assert False, "projection tag out of context"
465              pass              pass
466    
467          obj.SetProjection(          obj.SetProjection(Projection(self.projection_params,
468              Projection(self.ProjectionParams, self.ProjectionName))                                       self.projection_name,
469                                         epsg = self.projection_epsg))
470    
471      def start_parameter(self, name, qname, attrs):      def start_parameter(self, name, qname, attrs):
472          s = attrs.get((None, 'value'))          s = attrs.get((None, 'value'))
473          s = str(s) # we can't handle unicode in proj          s = str(s) # we can't handle unicode in proj
474          self.ProjectionParams.append(s)          self.projection_params.append(s)
475    
476      def start_layer(self, name, qname, attrs, layer_class = Layer):      def start_layer(self, name, qname, attrs, layer_class = Layer):
477          """Start a layer          """Start a layer
# Line 246  class SessionLoader(XMLReader): Line 483  class SessionLoader(XMLReader):
483          title = self.encode(attrs.get((None, 'title'), ""))          title = self.encode(attrs.get((None, 'title'), ""))
484          filename = attrs.get((None, 'filename'), "")          filename = attrs.get((None, 'filename'), "")
485          filename = os.path.join(self.GetDirectory(), filename)          filename = os.path.join(self.GetDirectory(), filename)
486          visible  = attrs.get((None, 'visible'), "true")          filename = self.encode(filename)
487            visible  = self.encode(attrs.get((None, 'visible'), "true")) != "false"
488          fill = parse_color(attrs.get((None, 'fill'), "None"))          fill = parse_color(attrs.get((None, 'fill'), "None"))
489          stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))          stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))
490          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))
491          self.aLayer = layer_class(title,          if attrs.has_key((None, "shapestore")):
492                                    self.theSession.OpenShapefile(filename),              store = self.idmap[attrs[(None, "shapestore")]]
493            else:
494                store = self.open_shapefile(filename)
495    
496            self.aLayer = layer_class(title, store,
497                                    fill = fill, stroke = stroke,                                    fill = fill, stroke = stroke,
498                                    lineWidth = stroke_width,                                    lineWidth = stroke_width,
499                                    visible = visible != "false")                                    visible = visible)
500    
501      def end_layer(self, name, qname):      def end_layer(self, name, qname):
502          self.aMap.AddLayer(self.aLayer)          self.aMap.AddLayer(self.aLayer)
503          self.aLayer = None          self.aLayer = None
504    
505        def start_rasterlayer(self, name, qname, attrs, layer_class = RasterLayer):
506            title = self.encode(attrs.get((None, 'title'), ""))
507            filename = attrs.get((None, 'filename'), "")
508            filename = os.path.join(self.GetDirectory(), filename)
509            filename = self.encode(filename)
510            visible  = self.encode(attrs.get((None, 'visible'), "true")) != "false"
511            opacity  = float(attrs.get((None, 'opacity'), "1"))
512            masktype = str(attrs.get((None, 'masktype'), "bit"))
513    
514            masktypes = {"none": layer_class.MASK_NONE,
515                         "bit":  layer_class.MASK_BIT,
516                         "alpha": layer_class.MASK_ALPHA}
517    
518            self.aLayer = layer_class(title, filename,
519                                      visible = visible,
520                                      opacity = opacity,
521                                      masktype = masktypes[masktype])
522    
523        def end_rasterlayer(self, name, qname):
524            self.aMap.AddLayer(self.aLayer)
525            self.aLayer = None
526    
527      def start_classification(self, name, qname, attrs):      def start_classification(self, name, qname, attrs):
528          field = attrs.get((None, 'field'), None)          attrs = self.check_attrs(name, attrs,
529                                     [AttrDesc("field", True),
530                                      AttrDesc("field_type", True)])
531            field = attrs["field"]
532            fieldType = attrs["field_type"]
533    
         fieldType = attrs.get((None, 'field_type'), None)  
534          dbFieldType = self.aLayer.GetFieldType(field)          dbFieldType = self.aLayer.GetFieldType(field)
535    
536          if fieldType != dbFieldType:          if fieldType != dbFieldType:
# Line 278  class SessionLoader(XMLReader): Line 545  class SessionLoader(XMLReader):
545          elif fieldType == FIELDTYPE_DOUBLE:          elif fieldType == FIELDTYPE_DOUBLE:
546              self.conv = float              self.conv = float
547    
548          self.aLayer.GetClassification().SetField(field)          self.aLayer.SetClassificationColumn(field)
549    
550      def end_classification(self, name, qname):      def end_classification(self, name, qname):
551          pass          pass
# Line 296  class SessionLoader(XMLReader): Line 563  class SessionLoader(XMLReader):
563      def start_clpoint(self, name, qname, attrs):      def start_clpoint(self, name, qname, attrs):
564          attrib_value = attrs.get((None, 'value'), "0")          attrib_value = attrs.get((None, 'value'), "0")
565    
566          value = self.conv(attrib_value)          field = self.aLayer.GetClassificationColumn()
567            if self.aLayer.GetFieldType(field) == FIELDTYPE_STRING:
568                value = self.encode(attrib_value)
569            else:
570                value = self.conv(attrib_value)
571          self.cl_group = ClassGroupSingleton(value)          self.cl_group = ClassGroupSingleton(value)
572          self.cl_group.SetLabel(self.encode(attrs.get((None, 'label'), "")))          self.cl_group.SetLabel(self.encode(attrs.get((None, 'label'), "")))
573          self.cl_prop = ClassGroupProperties()          self.cl_prop = ClassGroupProperties()
# Line 319  class SessionLoader(XMLReader): Line 589  class SessionLoader(XMLReader):
589              if range is not None:              if range is not None:
590                  self.cl_group = ClassGroupRange(Range(range))                  self.cl_group = ClassGroupRange(Range(range))
591              elif min is not None and max is not None:              elif min is not None and max is not None:
592                  self.cl_group = ClassGroupRange(self.conv(min), self.conv(max))                  self.cl_group = ClassGroupRange((self.conv(min),
593                                                     self.conv(max)))
594              else:              else:
595                  self.cl_group = ClassGroupRange(Range(None))                  self.cl_group = ClassGroupRange(Range(None))
596    
# Line 340  class SessionLoader(XMLReader): Line 611  class SessionLoader(XMLReader):
611              parse_color(attrs.get((None, 'stroke'), "None")))              parse_color(attrs.get((None, 'stroke'), "None")))
612          self.cl_prop.SetLineWidth(          self.cl_prop.SetLineWidth(
613              int(attrs.get((None, 'stroke_width'), "0")))              int(attrs.get((None, 'stroke_width'), "0")))
614            self.cl_prop.SetSize(int(attrs.get((None, 'size'), "5")))
615          self.cl_prop.SetFill(parse_color(attrs.get((None, 'fill'), "None")))          self.cl_prop.SetFill(parse_color(attrs.get((None, 'fill'), "None")))
616    
617      def end_cldata(self, name, qname):      def end_cldata(self, name, qname):
618          pass          pass
619    
     def start_table(self, name, qname, attrs):  
         #print "table title: %s" % attrs.get('title', None)  
         pass  
   
     def end_table(self, name, qname):  
         pass  
   
620      def start_labellayer(self, name, qname, attrs):      def start_labellayer(self, name, qname, attrs):
621          self.aLayer = self.aMap.LabelLayer()          self.aLayer = self.aMap.LabelLayer()
622    
623      def start_label(self, name, qname, attrs):      def start_label(self, name, qname, attrs):
624          x = float(attrs[(None, 'x')])          attrs = self.check_attrs(name, attrs,
625          y = float(attrs[(None, 'y')])                                   [AttrDesc("x", True, conversion = float),
626          text = self.encode(attrs[(None, 'text')])                                    AttrDesc("y", True, conversion = float),
627          halign = attrs[(None, 'halign')]                                    AttrDesc("text", True),
628          valign = attrs[(None, 'valign')]                                    AttrDesc("halign", True,
629                                               conversion = "ascii"),
630                                      AttrDesc("valign", True,
631                                               conversion = "ascii")])
632            x = attrs['x']
633            y = attrs['y']
634            text = attrs['text']
635            halign = attrs['halign']
636            valign = attrs['valign']
637            if halign not in ("left", "center", "right"):
638                raise LoadError("Unsupported halign value %r" % halign)
639            if valign not in ("top", "center", "bottom"):
640                raise LoadError("Unsupported valign value %r" % valign)
641          self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign)          self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign)
642    
643      def characters(self, chars):      def characters(self, chars):
644          pass          pass
645    
646    
647  def load_session(filename):  def load_session(filename, db_connection_callback = None,
648      """Load a Thuban session from the file object file"""                             shapefile_callback = None):
649        """Load a Thuban session from the file object file
650      handler = SessionLoader()  
651        The db_connection_callback, if given should be a callable object
652        that can be called like this:
653           db_connection_callback(params, message)
654    
655        where params is a dictionary containing the known connection
656        parameters and message is a string with a message why the connection
657        failed. db_connection_callback should return a new dictionary with
658        corrected and perhaps additional parameters like a password or None
659        to indicate that the user cancelled.
660        """
661        handler = SessionLoader(db_connection_callback, shapefile_callback)
662      handler.read(filename)      handler.read(filename)
663    
664      session = handler.theSession      session = handler.theSession
665      # Newly loaded session aren't modified      # Newly loaded session aren't modified
666      session.UnsetModified()      session.UnsetModified()
667    
668        handler.Destroy()
669    
670      return session      return session
671    
Legend:
Removed from v.874  
changed lines
  Added in v.2615
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26