C++使用GDAL进行栅格影像的掩膜处理

使用GDAL进行栅格影像的掩膜处理，可以通过以下步骤实现：

1. 导入GDAL库和头文件

#include "gdal_priv.h"
#include "cpl_conv.h" // for CPLMalloc()

2. 打开输入和输出影像

GDALAllRegister();

GDALDataset *poDataset = (GDALDataset *) GDALOpen("input.tif", GA_ReadOnly);
GDALDataset *poMask = (GDALDataset *) GDALOpen("mask.tif", GA_ReadOnly);
GDALDataset *poOutput = GetGDALDriverManager()->GetDriverByName("GTiff")->CreateCopy("output.tif", poDataset, FALSE, NULL, NULL, NULL);

3. 读取掩膜波段数据

int maskBand = 1;
int maskWidth = poMask->GetRasterXSize();
int maskHeight = poMask->GetRasterYSize();

float *maskData = new float[maskWidth * maskHeight];
GDALRasterBand *maskBandPtr = poMask->GetRasterBand(maskBand);
maskBandPtr->RasterIO(GF_Read, 0, 0, maskWidth, maskHeight, maskData, maskWidth, maskHeight, GDT_Float32, 0, 0);

4. 读取输入影像波段数据并进行掩膜处理

int inputWidth = poDataset->GetRasterXSize();
int inputHeight = poDataset->GetRasterYSize();

for (int band = 1; band <= poDataset->GetRasterCount(); band++) {
    GDALRasterBand *inputBandPtr = poDataset->GetRasterBand(band);
    GDALRasterBand *outputBandPtr = poOutput->GetRasterBand(band);

    float *inputData = new float[inputWidth * inputHeight];
    float *outputData = new float[inputWidth * inputHeight];

    inputBandPtr->RasterIO(GF_Read, 0, 0, inputWidth, inputHeight, inputData, inputWidth, inputHeight, GDT_Float32, 0, 0);

    for (int i = 0; i < inputWidth * inputHeight; i++) {
        if (maskData[i] == 0) {
            outputData[i] = inputData[i];
        } else {
            outputData[i] = 0;
        }
    }

    outputBandPtr->RasterIO(GF_Write, 0, 0, inputWidth, inputHeight, outputData, inputWidth, inputHeight, GDT_Float32, 0, 0);

    delete[] inputData;
    delete[] outputData;
}

5. 释放内存并关闭文件

delete[] maskData;

GDALClose((GDALDatasetH) poDataset);
GDALClose((GDALDatasetH) poMask);
GDALClose((GDALDatasetH) poOutput);

上述代码演示了如何使用GDAL库进行栅格影像的掩膜处理。你可以根据自己的需求进行修改和扩展。