/[thuban]/branches/WIP-pyshapelib-bramz/Thuban/Model/save.py
ViewVC logotype

Diff of /branches/WIP-pyshapelib-bramz/Thuban/Model/save.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1430 by jonathan, Wed Jul 16 13:23:28 2003 UTC revision 2383 by jan, Thu Oct 7 14:19:48 2004 UTC
# Line 1  Line 1 
1  # Copyright (c) 2001, 2002, 2003 by Intevation GmbH  # $Id$
2    # Copyright (c) 2001-2004 by Intevation GmbH
3  # Authors:  # Authors:
4  # Jan-Oliver Wagner <[email protected]>  # Jan-Oliver Wagner <[email protected]> (2004)
5  # Bernhard Herzog <[email protected]>  # Bernhard Herzog <[email protected]> (2001-2004)
6  # Jonathan Coles <[email protected]>  # Jonathan Coles <[email protected]> (2003)
7    # Frank Koormann <[email protected]> (2003)
8  #  #
9  # This program is free software under the GPL (>=v2)  # This program is free software under the GPL (>=v2)
10  # Read the file COPYING coming with Thuban for details.  # Read the file COPYING coming with Thuban for details.
# Line 23  from Thuban.Model.classification import Line 25  from Thuban.Model.classification import
25      ClassGroupDefault, ClassGroupSingleton, ClassGroupRange, ClassGroupMap      ClassGroupDefault, ClassGroupSingleton, ClassGroupRange, ClassGroupMap
26  from Thuban.Model.transientdb import AutoTransientTable, TransientJoinedTable  from Thuban.Model.transientdb import AutoTransientTable, TransientJoinedTable
27  from Thuban.Model.table import DBFTable, FIELDTYPE_STRING  from Thuban.Model.table import DBFTable, FIELDTYPE_STRING
28  from Thuban.Model.data import DerivedShapeStore, ShapefileStore  from Thuban.Model.data import DerivedShapeStore, ShapefileStore, \
29                                  SHAPETYPE_POINT
30    
31  from Thuban.Model.xmlwriter import XMLWriter  from Thuban.Model.xmlwriter import XMLWriter
32    from postgisdb import PostGISConnection, PostGISShapeStore
33    
34  def relative_filename(dir, filename):  def relative_filename(dir, filename):
35      """Return a filename relative to dir for the absolute file name absname.      """Return a filename relative to dir for the absolute file name absname.
# Line 40  def relative_filename(dir, filename): Line 44  def relative_filename(dir, filename):
44          return filename          return filename
45    
46    
47    def unify_filename(filename):
48        """Return a 'unified' version of filename
49    
50        The .thuban files should be as platform independent as possible.
51        Since they must contain filenames the filenames have to unified. We
52        unify on unix-like filenames for now, which means we do nothing on a
53        posix system and simply replace backslashes with slashes on windows
54        """
55        if os.name == "posix":
56            return filename
57        elif os.name == "nt":
58            return "/".join(filename.split("\\"))
59        else:
60            raise RuntimeError("Unsupported platform for unify_filename: %s"
61                               % os.name)
62    
63  def sort_data_stores(stores):  def sort_data_stores(stores):
64      """Return a topologically sorted version of the sequence of data containers      """Return a topologically sorted version of the sequence of data containers
65    
# Line 102  class SessionSaver(XMLWriter): Line 122  class SessionSaver(XMLWriter):
122      def has_id(self, obj):      def has_id(self, obj):
123          return self.idmap.has_key(id(obj))          return self.idmap.has_key(id(obj))
124    
125        def prepare_filename(self, filename):
126            """Return the string to use when writing filename to the thuban file
127    
128            The returned string is a unified version (only slashes as
129            directory separators, see unify_filename) of filename expressed
130            relative to the directory the .thuban file is written to.
131            """
132            return unify_filename(relative_filename(self.dir, filename))
133    
134      def write(self, file_or_filename):      def write(self, file_or_filename):
135          XMLWriter.write(self, file_or_filename)          XMLWriter.write(self, file_or_filename)
136    
137          self.write_header("session", "thuban-0.9.dtd")          self.write_header("session", "thuban-1.1.dtd")
138          self.write_session(self.session)          self.write_session(self.session)
139          self.close()          self.close()
140    
# Line 132  class SessionSaver(XMLWriter): Line 161  class SessionSaver(XMLWriter):
161              attrs["xmlns:" + name] = uri              attrs["xmlns:" + name] = uri
162          # default name space          # default name space
163          attrs["xmlns"] = \          attrs["xmlns"] = \
164                 "http://thuban.intevation.org/dtds/thuban-0.9-dev.dtd"                 "http://thuban.intevation.org/dtds/thuban-1.1-dev.dtd"
165          self.open_element("session", attrs)          self.open_element("session", attrs)
166            self.write_db_connections(session)
167          self.write_data_containers(session)          self.write_data_containers(session)
168          for map in session.Maps():          for map in session.Maps():
169              self.write_map(map)              self.write_map(map)
170          self.close_element("session")          self.close_element("session")
171    
172        def write_db_connections(self, session):
173            for conn in session.DBConnections():
174                if isinstance(conn, PostGISConnection):
175                    self.write_element("dbconnection",
176                                       {"id": self.define_id(conn),
177                                        "dbtype": "postgis",
178                                        "host": conn.host,
179                                        "port": conn.port,
180                                        "user": conn.user,
181                                        "dbname": conn.dbname})
182                else:
183                    raise ValueError("Can't handle db connection %r" % conn)
184    
185      def write_data_containers(self, session):      def write_data_containers(self, session):
186          containers = sort_data_stores(session.DataContainers())          containers = sort_data_stores(session.DataContainers())
187          for container in containers:          for container in containers:
# Line 156  class SessionSaver(XMLWriter): Line 199  class SessionSaver(XMLWriter):
199              idvalue = self.define_id(container)              idvalue = self.define_id(container)
200              if isinstance(container, ShapefileStore):              if isinstance(container, ShapefileStore):
201                  self.define_id(container.Table(), idvalue)                  self.define_id(container.Table(), idvalue)
202                  filename = relative_filename(self.dir, container.FileName())                  filename = self.prepare_filename(container.FileName())
203                  self.write_element("fileshapesource",                  self.write_element("fileshapesource",
204                                     {"id": idvalue, "filename": filename,                                     {"id": idvalue, "filename": filename,
205                                      "filetype": "shapefile"})                                      "filetype": "shapefile"})
# Line 166  class SessionSaver(XMLWriter): Line 209  class SessionSaver(XMLWriter):
209                                     {"id": idvalue,                                     {"id": idvalue,
210                                      "shapesource": self.get_id(shapesource),                                      "shapesource": self.get_id(shapesource),
211                                      "table": self.get_id(table)})                                      "table": self.get_id(table)})
212                elif isinstance(container, PostGISShapeStore):
213                    conn = container.DBConnection()
214                    self.write_element("dbshapesource",
215                                       {"id": idvalue,
216                                        "dbconn": self.get_id(conn),
217                                        "tablename": container.TableName(),
218                                        "id_column": container.IDColumn().name,
219                                        "geometry_column":
220                                          container.GeometryColumn().name,
221                                        })
222              elif isinstance(container, DBFTable):              elif isinstance(container, DBFTable):
223                  filename = relative_filename(self.dir, container.FileName())                  filename = self.prepare_filename(container.FileName())
224                  self.write_element("filetable",                  self.write_element("filetable",
225                                     {"id": idvalue,                                     {"id": idvalue,
226                                      "title": container.Title(),                                      "title": container.Title(),
# Line 208  class SessionSaver(XMLWriter): Line 261  class SessionSaver(XMLWriter):
261          """Write the projection.          """Write the projection.
262          """          """
263          if projection and len(projection.params) > 0:          if projection and len(projection.params) > 0:
264              self.open_element("projection", {"name": projection.GetName()})              attrs = {"name": projection.GetName()}
265                epsg = projection.EPSGCode()
266                if epsg is not None:
267                    attrs["epsg"] = epsg
268                self.open_element("projection", attrs)
269              for param in projection.params:              for param in projection.params:
270                  self.write_element('parameter value="%s"' %                  self.write_element('parameter value="%s"' %
271                                     self.encode(param))                                     self.encode(param))
# Line 241  class SessionSaver(XMLWriter): Line 298  class SessionSaver(XMLWriter):
298              self.write_classification(layer)              self.write_classification(layer)
299              self.close_element("layer")              self.close_element("layer")
300          elif isinstance(layer, RasterLayer):          elif isinstance(layer, RasterLayer):
301              attrs["filename"] = relative_filename(self.dir, layer.filename)              attrs["filename"] = self.prepare_filename(layer.filename)
302              self.open_element("rasterlayer", attrs)              self.open_element("rasterlayer", attrs)
303              self.write_projection(layer.GetProjection())              self.write_projection(layer.GetProjection())
304              self.close_element("rasterlayer")              self.close_element("rasterlayer")
# Line 254  class SessionSaver(XMLWriter): Line 311  class SessionSaver(XMLWriter):
311    
312          lc = layer.GetClassification()          lc = layer.GetClassification()
313    
314          field = layer.GetClassificationField()          field = layer.GetClassificationColumn()
315    
316          #          #
317          # there isn't a classification of anything so do nothing          # there isn't a classification of anything so do nothing
318          #          #
# Line 290  class SessionSaver(XMLWriter): Line 347  class SessionSaver(XMLWriter):
347                      'stroke_width': str(data.GetLineWidth()),                      'stroke_width': str(data.GetLineWidth()),
348                      'fill'        : data.GetFill().hex()}                      'fill'        : data.GetFill().hex()}
349    
350                # only for point layers write the size attribute
351                if layer.ShapeType() == SHAPETYPE_POINT:
352                    dict['size'] =  str(data.GetSize())
353    
354              self.open_element(open_el)              self.open_element(open_el)
355              self.write_element("cldata", dict)              self.write_element("cldata", dict)
356              self.close_element(close_el)              self.close_element(close_el)
Legend:
Removed from v.1430  
changed lines
  Added in v.2383
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26