/[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 1930 by bh, Tue Nov 11 13:24:41 2003 UTC revision 2642 by bh, Fri Jul 1 20:49:04 2005 UTC
# Line 1  Line 1 
1  # Copyright (C) 2001, 2002, 2003 by Intevation GmbH  # Copyright (C) 2001, 2002, 2003, 2004, 2005 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 100  class AttrDesc: Line 101  class AttrDesc:
101    
102  class SessionLoader(XMLReader):  class SessionLoader(XMLReader):
103    
104      def __init__(self, db_connection_callback = None):      def __init__(self, db_connection_callback = None,
105                           shapefile_callback = None):
106          """Inititialize the Sax handler."""          """Inititialize the Sax handler."""
107          XMLReader.__init__(self)          XMLReader.__init__(self)
108    
109          self.db_connection_callback = db_connection_callback          self.db_connection_callback = db_connection_callback
110            self.shapefile_callback = shapefile_callback
111          self.theSession = None          self.theSession = None
112          self.aMap = None          self.aMap = None
113          self.aLayer = None          self.aLayer = None
# Line 143  class SessionLoader(XMLReader): Line 145  class SessionLoader(XMLReader):
145          for xmlns in ("http://thuban.intevation.org/dtds/thuban-0.8.dtd",          for xmlns in ("http://thuban.intevation.org/dtds/thuban-0.8.dtd",
146                        "http://thuban.intevation.org/dtds/thuban-0.9-dev.dtd",                        "http://thuban.intevation.org/dtds/thuban-0.9-dev.dtd",
147                        "http://thuban.intevation.org/dtds/thuban-0.9.dtd",                        "http://thuban.intevation.org/dtds/thuban-0.9.dtd",
148                        "http://thuban.intevation.org/dtds/thuban-1.0-dev.dtd"):                        "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():              for key, value in dispatchers.items():
153                  dispatchers[(xmlns, key)] = value                  dispatchers[(xmlns, key)] = value
154    
# Line 186  class SessionLoader(XMLReader): Line 191  class SessionLoader(XMLReader):
191          If the attribute has a default value and it is not present in          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.          attrs, use that default value as the value in the returned dict.
193    
194          If a conversion is specified, convert the value before putting          The value is converted before putting it into the returned dict.
195          it into the returned dict. The following conversions are          The following conversions are available:
         available:  
196    
197             'filename' -- The attribute is a filename.             'filename' -- The attribute is a filename.
198    
# Line 212  class SessionLoader(XMLReader): Line 216  class SessionLoader(XMLReader):
216                        ascii encoding.                        ascii encoding.
217    
218             a callable -- The attribute value is passed to the callable             a callable -- The attribute value is passed to the callable
219                           and the return value is used a as the converted                           and the return value is used as the converted
220                           value                           value
221    
222            If no conversion is specified for an attribute it is converted
223            with self.encode.
224          """          """
225          normalized = {}          normalized = {}
226    
# Line 241  class SessionLoader(XMLReader): Line 248  class SessionLoader(XMLReader):
248                                      % (element, d.name))                                      % (element, d.name))
249              elif d.conversion == "filename":              elif d.conversion == "filename":
250                  value = os.path.abspath(os.path.join(self.GetDirectory(),                  value = os.path.abspath(os.path.join(self.GetDirectory(),
251                                                       value))                                                       self.encode(value)))
252              elif d.conversion == "ascii":              elif d.conversion == "ascii":
253                  value = value.encode("ascii")                  value = value.encode("ascii")
254              elif d.conversion:              elif d.conversion:
255                  # Assume it's a callable                  # Assume it's a callable
256                  value = d.conversion(value)                  value = d.conversion(value)
257                else:
258                   value = self.encode(value)
259    
260              normalized[d.name] = value              normalized[d.name] = value
261          return normalized          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):      def start_dbconnection(self, name, qname, attrs):
309          attrs = self.check_attrs(name, attrs,          attrs = self.check_attrs(name, attrs,
310                                   [AttrDesc("id", True),                                   [AttrDesc("id", True),
# Line 294  class SessionLoader(XMLReader): Line 348  class SessionLoader(XMLReader):
348                                    AttrDesc("dbconn", True,                                    AttrDesc("dbconn", True,
349                                             conversion = "idref"),                                             conversion = "idref"),
350                                    AttrDesc("tablename", True,                                    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")])                                             conversion = "ascii")])
361          ID = attrs["id"]          # The default value of geometry_column to use when instantiating
362          db = attrs["dbconn"]          # the db shapestore is None which we currently can't easily use
363          tablename = attrs["tablename"]          # in check_attrs
364          self.idmap[ID] = self.theSession.OpenDBShapeStore(db, tablename)          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):      def start_fileshapesource(self, name, qname, attrs):
373          attrs = self.check_attrs(name, attrs,          attrs = self.check_attrs(name, attrs,
# Line 311  class SessionLoader(XMLReader): Line 380  class SessionLoader(XMLReader):
380          filetype = attrs["filetype"]          filetype = attrs["filetype"]
381          if filetype != "shapefile":          if filetype != "shapefile":
382              raise LoadError("shapesource filetype %r not supported" % filetype)              raise LoadError("shapesource filetype %r not supported" % filetype)
383          self.idmap[ID] = self.theSession.OpenShapefile(filename)          self.idmap[ID] = self.open_shapefile(filename)
384    
385      def start_derivedshapesource(self, name, qname, attrs):      def start_derivedshapesource(self, name, qname, attrs):
386          attrs = self.check_attrs(name, attrs,          attrs = self.check_attrs(name, attrs,
# Line 422  class SessionLoader(XMLReader): Line 491  class SessionLoader(XMLReader):
491          if attrs.has_key((None, "shapestore")):          if attrs.has_key((None, "shapestore")):
492              store = self.idmap[attrs[(None, "shapestore")]]              store = self.idmap[attrs[(None, "shapestore")]]
493          else:          else:
494              store = self.theSession.OpenShapefile(filename)              store = self.open_shapefile(filename)
495    
496          self.aLayer = layer_class(title, store,          self.aLayer = layer_class(title, store,
497                                    fill = fill, stroke = stroke,                                    fill = fill, stroke = stroke,
498                                    lineWidth = stroke_width,                                    lineWidth = stroke_width,
# Line 438  class SessionLoader(XMLReader): Line 508  class SessionLoader(XMLReader):
508          filename = os.path.join(self.GetDirectory(), filename)          filename = os.path.join(self.GetDirectory(), filename)
509          filename = self.encode(filename)          filename = self.encode(filename)
510          visible  = self.encode(attrs.get((None, 'visible'), "true")) != "false"          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          self.aLayer = layer_class(title, filename, visible = visible)          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):      def end_rasterlayer(self, name, qname):
524          self.aMap.AddLayer(self.aLayer)          self.aMap.AddLayer(self.aLayer)
525          self.aLayer = None          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 497  class SessionLoader(XMLReader): Line 579  class SessionLoader(XMLReader):
579          del self.cl_group, self.cl_prop          del self.cl_group, self.cl_prop
580    
581      def start_clrange(self, name, qname, attrs):      def start_clrange(self, name, qname, attrs):
582            attrs = self.check_attrs(name, attrs,
583                                     [AttrDesc("range", False, None),
584                                      AttrDesc("min", False, None),
585                                      AttrDesc("max", False, None)])
586    
587          range = attrs.get((None, 'range'), None)          range = attrs['range']
588          # for backward compatibility (min/max are not saved)          # for backward compatibility (min/max are not saved)
589          min   = attrs.get((None, 'min'), None)          min   = attrs['min']
590          max   = attrs.get((None, 'max'), None)          max   = attrs['max']
591    
592          try:          try:
593              if range is not None:              if range is not None:
# Line 529  class SessionLoader(XMLReader): Line 615  class SessionLoader(XMLReader):
615              parse_color(attrs.get((None, 'stroke'), "None")))              parse_color(attrs.get((None, 'stroke'), "None")))
616          self.cl_prop.SetLineWidth(          self.cl_prop.SetLineWidth(
617              int(attrs.get((None, 'stroke_width'), "0")))              int(attrs.get((None, 'stroke_width'), "0")))
618            self.cl_prop.SetSize(int(attrs.get((None, 'size'), "5")))
619          self.cl_prop.SetFill(parse_color(attrs.get((None, 'fill'), "None")))          self.cl_prop.SetFill(parse_color(attrs.get((None, 'fill'), "None")))
620    
621      def end_cldata(self, name, qname):      def end_cldata(self, name, qname):
# Line 538  class SessionLoader(XMLReader): Line 625  class SessionLoader(XMLReader):
625          self.aLayer = self.aMap.LabelLayer()          self.aLayer = self.aMap.LabelLayer()
626    
627      def start_label(self, name, qname, attrs):      def start_label(self, name, qname, attrs):
628          x = float(attrs[(None, 'x')])          attrs = self.check_attrs(name, attrs,
629          y = float(attrs[(None, 'y')])                                   [AttrDesc("x", True, conversion = float),
630          text = self.encode(attrs[(None, 'text')])                                    AttrDesc("y", True, conversion = float),
631          halign = attrs[(None, 'halign')]                                    AttrDesc("text", True),
632          valign = attrs[(None, 'valign')]                                    AttrDesc("halign", True,
633                                               conversion = "ascii"),
634                                      AttrDesc("valign", True,
635                                               conversion = "ascii")])
636            x = attrs['x']
637            y = attrs['y']
638            text = attrs['text']
639            halign = attrs['halign']
640            valign = attrs['valign']
641            if halign not in ("left", "center", "right"):
642                raise LoadError("Unsupported halign value %r" % halign)
643            if valign not in ("top", "center", "bottom"):
644                raise LoadError("Unsupported valign value %r" % valign)
645          self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign)          self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign)
646    
647      def characters(self, chars):      def characters(self, chars):
648          pass          pass
649    
650    
651  def load_session(filename, db_connection_callback = None):  def load_session(filename, db_connection_callback = None,
652                               shapefile_callback = None):
653      """Load a Thuban session from the file object file      """Load a Thuban session from the file object file
654    
655      The db_connection_callback, if given should be a callable object      The db_connection_callback, if given should be a callable object
# Line 562  def load_session(filename, db_connection Line 662  def load_session(filename, db_connection
662      corrected and perhaps additional parameters like a password or None      corrected and perhaps additional parameters like a password or None
663      to indicate that the user cancelled.      to indicate that the user cancelled.
664      """      """
665      handler = SessionLoader(db_connection_callback)      handler = SessionLoader(db_connection_callback, shapefile_callback)
666      handler.read(filename)      handler.read(filename)
667    
668      session = handler.theSession      session = handler.theSession
Legend:
Removed from v.1930  
changed lines
  Added in v.2642
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26