jan: thuban/Extensions/export_shapefile export_shapefile.py, NONE, 1.1

[cvs@intevation.de]

Author: jan

Update of /thubanrepository/thuban/Extensions/export_shapefile
In directory doto:/tmp/cvs-serv10122

Added Files:
	export_shapefile.py 
Log Message:
Exports a layer as a Shapefile.


--- NEW FILE: export_shapefile.py ---
# Copyright (C) 2005 by Intevation GmbH
# Authors:
# Jan-Oliver Wagner <jan at intevation.de> (2005)
#
# This program is free software under the GPL (>=v2)
# Read the file COPYING coming with Thuban for details.

"""
Extend Thuban with a routine to export a layer
as a Shapefile.
"""

__version__ = '$Revision: 1.1 $'
# $Source: /thubanrepository/thuban/Extensions/export_shapefile/export_shapefile.py,v $
# $Id: export_shapefile.py,v 1.1 2005/08/11 21:32:10 jan Exp $

from wxPython.wx import wxFileDialog, wxOPEN, wxOVERWRITE_PROMPT, wxID_OK, \
     wxProgressDialog

from Thuban.Model.data import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \
                              SHAPETYPE_POINT
from Thuban.Model.table import table_to_dbf
from Thuban.UI.command import registry, Command
from Thuban.UI.mainwindow import main_menu
from Thuban import _

import shapelib

def ExportLayerAsShapefile(context):
    """Request filename from user, 
    and create the Shapefile(s) .shp, .dbf and .shx.

    context -- The Thuban context.
    """
    # First, find the current layer
    layer = context.mainwindow.canvas.SelectedLayer()
    if layer is None:
       context.mainwindow.RunMessageBox(_('Export Shapefile'),
           _('No layer selected'))
       return

    # Second, get the basefilename for the shapefile.
    dlg = wxFileDialog(context.mainwindow,
                       _('Export Shapefile'), '.', '',
                       _('Shapefile  Files (*.shp)|*.shp|'),
                       wxOPEN|wxOVERWRITE_PROMPT)
    if dlg.ShowModal() == wxID_OK:
        filename = dlg.GetPath()[:-4]
        dlg.Destroy()
    else:
        return

    # Third, create the dbf file of the new Shapefile
    dbf_filename = filename + '.dbf'
    if hasattr(layer, "ShapeStore"):
        table = layer.ShapeStore().Table()
	table_to_dbf(table, dbf_filename)

    # Fourth, prepare the shp file of the new Shapefile
    shp_filename = filename + '.shp'
    shapetypes = { SHAPETYPE_POLYGON: shapelib.SHPT_POLYGON,
        SHAPETYPE_ARC: shapelib.SHPT_ARC,
        SHAPETYPE_POINT: shapelib.SHPT_POINT}
    shp = shapelib.create(shp_filename, shapetypes[layer.ShapeType()])

    # Now go through all shapes and store them to the file
    dlg= wxProgressDialog(_("Export Shapefile"),
                          _("Storing shapes ..."),
                          layer.ShapeStore().NumShapes(),
                          None)

    cnt = 0
    step =  int(layer.ShapeStore().NumShapes() / 100.0)
    if step == 0: step = 1

    for s in layer.ShapeStore().AllShapes():
        i = s.ShapeID()
	print s.Points()
        obj = shapelib.SHPObject(shapetypes[layer.ShapeType()], i, s.Points())
        shp.write_object(i, obj)
        if cnt % step == 0:
            dlg.Update(cnt)
        cnt = cnt + 1

    del shp
    dlg.Destroy()


# register the new command
registry.Add(Command('ExportShapefile', _('Export Layer as Shapefile ...'),
                     ExportLayerAsShapefile,
                     helptext = _('Export the active layer as a Shapefile')))

# find the experimental menu (create it anew if not found)
experimental_menu = main_menu.FindOrInsertMenu("experimental", \
                                              _("Experimenta&l"))

# finally add the new entry to the extensions menu
experimental_menu.InsertItem('ExportShapefile')