C++实现归一化植被指数(NDVI)计算 | 使用TM数据
C++实现归一化植被指数(NDVI)计算
这段代码实现了计算归一化植被指数(NDVI)的功能,用于评估植被的健康状况和密度。
void CBandMath::NDVICal(CString m_pathname, int Bands, int Width, int Height, int DataType) {
int fRed, fNearRed;
if (DataType == 1) { // TM 数据
fRed = 3;
fNearRed = 5;
}
CString m_OutFile;
float *NDVIWr = new float[Height * Width];
unsigned char *Red = new unsigned char[Height * Width];
unsigned char *NearRed = new unsigned char[Height * Width];
CFile fReadFile, fWriteFile;
if (!fReadFile.Open(m_pathname, CFile::modeReadWrite)) { // 打开图像文件(参数)
AfxMessageBox('不能打开所读文件!');
return;
}
// 打开写入文件(其它类的变量)
if (!fWriteFile.Open('NDVI', CFile::modeCreate | CFile::modeWrite)) {
AfxMessageBox('请选择输出文件!');
return;
}
fReadFile.Seek((fRed - 1) * Height * Width, CFile::begin);
fReadFile.Read(Red, Height * Width);
fReadFile.Seek((fNearRed - 1) * Height * Width, CFile::begin);
fReadFile.Read(NearRed, Height * Width);
for (int k = 0; k < Height * Width; k++)
NDVIWr[k] = ((float)NearRed[k] - Red[k]) / (NearRed[k] + Red[k]);
fWriteFile.Write(NDVIWr, sizeof(float) * Height * Width);
delete[] NDVIWr;
delete[] Red;
delete[] NearRed;
fReadFile.Close();
fWriteFile.Close();
}
函数说明:
- 函数名: NDVICal
- 参数:
m_pathname: 图像文件路径Bands: 波段数Width: 图像宽度Height: 图像高度DataType: 数据类型 (1 表示 TM 数据)
- 返回值: 无
功能:
- 确定红波段和近红外波段: 根据数据类型确定红波段 (fRed) 和近红外波段 (fNearRed) 的编号。
- 分配内存: 创建数组存储红波段、近红外波段和 NDVI 值。
- 读取图像数据: 打开输入图像文件,读取红波段和近红外波段的像素值。
- 计算 NDVI: 遍历每个像素,使用公式 ((近红外 - 红) / (近红外 + 红)) 计算 NDVI 值。
- 写入结果: 将计算得到的 NDVI 值写入输出文件。
- 释放内存: 释放动态分配的内存空间。
- 关闭文件: 关闭输入和输出文件。
注意:
- 这段代码假设输入图像是多波段图像,并且包含红波段和近红外波段。
- 输出文件名为 'NDVI',并将以浮点格式存储 NDVI 值。
- 使用前请确保已正确设置输入参数和输出文件路径。
原文地址: https://www.cveoy.top/t/topic/jnwA 著作权归作者所有。请勿转载和采集!