从odo.resource(源)到odo.resource(目标)获取数据

Question

我正在尝试扩展odo库，使其具有将GDAL数据集(包含空间信息的栅格)转换为NetCDF文件的功能。

读取gdal数据集很好。但是在netcdf的创建阶段，我需要一些gdal数据集的元数据(调用odo.odo(源，目标)时还不知道的元数据)。我怎样才能做到这一点呢？

到目前为止，我的代码的简短版本如下：

import odo
from odo import resource, append
import gdal
import netCDF4 as nc4
import numpy as np

@resource.register('.+\.tif')
def resource_gdal(uri, **kwargs):
    ds = gdal.Open(uri)

    # metadata I need to transfer to netcdf
    b = ds.GetGeoTransform() #bbox, interval

   return ds


@resource.register('.+\.nc')
def resource_netcdf(uri, dshape=None, **kwargs):
    ds = nc4.Dataset(uri,'w')

    # create lat lon dimensions and variables
    ds.createDimension(lat, dshape[0].val)
    ds.createDimension(lon, dshape[1].val)
    lat = ds.createVariable('lat','f4', ('lat',))
    lon = ds.createVariable('lon','f4', ('lon',))

    # create a range from the **gdal metadata**
    lat_array = np.arange(dshape[0].val)*b[1]+b[0]
    lon_array = np.arange(dshape[1].val)*b[5]+b[3]

    # assign the range to the netcdf variable
    lat[:] = lat_array
    lon[:] = lon_array

    # create the variable which will hold the gdal data
    data = ds.createVariable('data', 'f4', ('lat', 'lon',))

   return data


@append.register(nc4.Variable, gdal.Dataset)
def append_gdal_to_nc4(tgt, src, **kwargs):
    arr = src.ReadAsArray()
    tgt[:] = arr

    return tgt

谢谢!

Answer 1

我没有太多使用odo的经验，但从源代码和文档的浏览来看，resource_netcdf()似乎不应该参与到将gdal数据转换为netcdf的工作中。翻译应该是由convert.register修饰的gdal_to_netcdf()函数的工作。在这种情况下，由resource_gdal返回的gdal.Dataset对象将具有所有足够的信息(地理参考、像素大小)来生成netcdf。