NDVI 计算函数 - 基于 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();
}
原文地址: https://www.cveoy.top/t/topic/jlM5 著作权归作者所有。请勿转载和采集!