GDAL python教程基础篇（4）OGR拷贝方法

使用OGR拷贝创建新的矢量数据集的方法其实就是将数据读取的顺序再重复一遍。总体来说这个过程就是构建数据源->构建层->构建要素->构建几何形状->关闭数据源。

在OGR的矢量数据模型中，矢量数据分为datasource，layer，feature三个层次，在这三个层次上，OGR均有不同的复制数据的方法。

1.在datasource层次创建数据

>>> from osgeo import ogr
>>> import os,math
>>> inshp = '/gdata/GSHHS_c.shp'
>>> ds = ogr.Open(inshp)
>>> driver = ogr.GetDriverByName("ESRI Shapefile")
>>> outputfile = '/tmp/xx_world_borders_copy.shp'
>>> if os.access( outputfile, os.F_OK ):
>>>     driver.DeleteDataSource(outputfile)
>>> pt_cp = driver.CopyDataSource(ds, outputfile)
>>> pt_cp.Release()
>>> dir(pt_cp)[:6] + ['... ...'] + dir(pt_cp)[-3:]
['AbortSQL','CommitTransaction','CopyLayer','CreateLayer','DeleteLayer','Dereference','... ...','name','this','thisown']

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

2.在layer层次拷贝数据

>>> from osgeo import ogr
>>> import os,math
>>> inshp = '/gdata/GSHHS_c.shp'
>>> ds = ogr.Open(inshp)
>>> driver = ogr.GetDriverByName("ESRI Shapefile")
>>> outputfile ='/tmp/xx_world_borders_copy2.shp'
>>> if os.access( outputfile, os.F_OK ):
>>>          driver.DeleteDataSource( outputfile )
>>> layer = ds.GetLayer()
>>> newds = driver.CreateDataSource(outputfile)
>>> pt_layer  = newds.CopyLayer(layer, 'abcd')
>>> newds.Destroy()
>>> dir(pt_layer)[:6] + ['... ...'] + dir(pt_layer)[-3:]
['AlterFieldDefn','Clip','CommitTransaction','CreateFeature','CreateField','CreateFields','... ...','schema','this','thisown']

若想对layer进行拷贝，首先得有数据源。使用newds的CopyLayer()方法对图层进行拷贝，以创建newds数据源。值得注意的是，当图层拷贝完之后，需要对newds进行Destroy()操作，将数据写入磁盘。

CopyLayer()的第1个参数是OGR的Layer对象，第2个参数是要生成图层的名称。对于Shapefile来说，这个名称是没有用的，但必须给这个字符串赋变量值。

3.在feature层次拷贝数据

>>> from osgeo import ogr
>>> import os,math
>>> inshp = '/gdata/GSHHS_c.shp'
>>> ds = ogr.Open(inshp)
>>> driver = ogr.GetDriverByName("ESRI Shapefile")
>>> outputfile ='/tmp/xx_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()
>>> extent
(-180.0, 180.00000000000023, -89.99999999999994, 83.53036100000006)
>>> feature = layer.GetNextFeature()
>>> while feature is not None:
>>>     layernew.CreateFeature(feature)
>>>     feature = layer.GetNextFeature()
>>> newds.Destroy()
>>> dir(feature)[:6] + ['... ...'] + dir(feature)[-3:]
['__bool__','__class__','__delattr__','__dir__','__doc__','__eq__','... ...','__sizeof__','__str__','__subclasshook__']

特别注意：Destroy()不能省略，Destroy()除了销毁数据还有数据flush到磁盘的作用。如果没有此命令，那么刚才创建的一系列要素就不会被写入磁盘的文件中。