Python与开源GIS:使用OGR中拷贝方法创建新的Shapefile

矢量数据集创建也并没有非常难,就是把读取的时候的顺序重复以便就好了。总体来说这个过程就是构建数据源->构建层->构建要素->构建形状->关闭数据源。 我们首先来过看一下如何使用已有的数据来构建我们的数据,也就是复制/粘贴方法。在OGR的矢量数据模型中,矢量数据分为datasource,layer,与feature三个层次,在这三个层次上,OGR有不同的复制数据的方法。

在datasource层次创建数据

我们使用世界各国的边界数据作为示例,利用这个数据,来创建与之内容完全相同的另外一套数据。

    from osgeo import ogr
    import os,math
    inshp = '/gdata/world_borders.shp'
    ds = ogr.Open(inshp)
    driver = ogr.GetDriverByName("ESRI Shapefile")
    outputfile = '/gdata/world_borders_copy.shp'
    if os.access( outputfile, os.F_OK ):
        driver.DeleteDataSource( outputfile )
    pt_cp = driver.CopyDataSource(ds, outputfile)
    pt_cp.Release()

在这里,我们使用了CopyDataSource()的方法来对数据进行复制的操作。这个函数的第1个参数是要进行复制的数据源,第2个参数是要生成的数据的路径,并且返回一个指针。为了将数据写入到硬盘,必须释放这个指标,使用Release()方法。

在layer层次拷贝数据

下面我们在layer层次来完成同样的工作。

    from osgeo import ogr
    import os,math
    inshp = '/gdata/world_borders.shp'
    ds = ogr.Open(inshp)
    driver = ogr.GetDriverByName("ESRI Shapefile")
    outputfile = '/gdata/world_borders_copy2.shp'
    if os.access( outputfile, os.F_OK ):
        driver.DeleteDataSource( outputfile )
    layer = ds.GetLayer()
    newds = driver.CreateDataSource(outputfile)
    # layernew = newds.CreateLayer('linescopy',None,ogr.wkbLineString)
    pt_layer  = newds.CopyLayer(layer, 'abcd')
    newds.Destroy()

为了对layer进行拷贝操作,首先得有数据源。在这里首先建立了newds这一数据源,然后使用newds的CopyLayer()方法来对图层进行拷贝操作。值得注意的是,当图层拷贝完成之后,需要对newds使用Destroy()操作,才能将数据写入到磁盘。 CopyLayer()方法的第1个参数是OGR的Layer 对象,第2个参数是要生成的图层的名称。对于Shapefile来说,这个名称是没有用的,但是无论如何,都要给出一个字符串型的变量。

在feature层次拷贝数据

下面我们来看一下最复杂的在feature层次进行数据拷贝的使用方法。抽象来看的话,这个方法只是再多进行了一步操作。

    from osgeo import ogr
    import os,math
    inshp = '/gdata/world_borders.shp'
    ds = ogr.Open(inshp)
    driver = ogr.GetDriverByName("ESRI Shapefile")
    outputfile = '/gdata/world_borders_copy3.shp'
    if os.access( outputfile, os.F_OK ):
        driver.DeleteDataSource( outputfile )
    newds = driver.CreateDataSource(outputfile)
    layernew = newds.CreateLayer('worldcopy',None,ogr.wkbLineString)
    layer = ds.GetLayer()
    extent = layer.GetExtent()
    print (extent)
    feature = layer.GetNextFeature()
    while feature is not None:
        layernew.CreateFeature(feature)
        feature = layer.GetNextFeature()
    newds.Destroy()

特别注意:Destroy()不能省略,Destroy()除了销毁数据还担负了数据flush到磁盘的任务。如果没有这一句,那么刚才创建的一系列Feature都不会写入磁盘的文件中。