|
|
#include "stdafx.h"
|
|
|
|
|
|
#include <opencv2\opencv.hpp>
|
|
|
using namespace cv;
|
|
|
|
|
|
#include "GeoTIFFEdit.h"
|
|
|
|
|
|
// tif 图像
|
|
|
#include <ogr_srs_api.h>
|
|
|
#include <ogr_spatialref.h>
|
|
|
#include <gdal.h>
|
|
|
#include <gdal_priv.h>
|
|
|
|
|
|
// 几何校正
|
|
|
#include "SysGeoCorrect.h"
|
|
|
|
|
|
// 保存情报数据为tiff文件
|
|
|
bool SaveQBDataToGeoTIFF(QBStru *qbData, string savePath)
|
|
|
{
|
|
|
try
|
|
|
{
|
|
|
if (qbData == NULL || savePath == "")
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
// 图像数据有效性验证
|
|
|
if (qbData->image.bValid == false)
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
else
|
|
|
{
|
|
|
if ( false == SaveImgStruToGeoTIFF(&qbData->image.geoImg, savePath))
|
|
|
{
|
|
|
if (true == SysGeoCorrectImg(qbData, false)) // 几何校正
|
|
|
{
|
|
|
return SaveImgStruToGeoTIFF(&qbData->image.geoImg, savePath);
|
|
|
}
|
|
|
else
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
catch(cv::Exception &e)
|
|
|
{
|
|
|
string str = e.msg;
|
|
|
return false;
|
|
|
}
|
|
|
catch(...)
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
return true;
|
|
|
}
|
|
|
|
|
|
|
|
|
//功能：保存图像结构体至本地文件（GeoTiff格式）
|
|
|
//输入： 1. qbData 通用图像数据结构体
|
|
|
// 2. savePath Tiff文件保存路径
|
|
|
|
|
|
//输出： 1. 返回值：保存成功返回true，失败返回false
|
|
|
bool SaveImgStruToGeoTIFF(const ImgStru *Img, string savePath)
|
|
|
{
|
|
|
try
|
|
|
{
|
|
|
if (Img == NULL || savePath == "")
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
// 判断路径后缀的有效性
|
|
|
string postfix = savePath.substr(savePath.rfind('.') + 1, savePath.length());
|
|
|
if ("tif" != postfix)
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
// 图像数据有效性验证
|
|
|
if (Img->ImgWidth <= 0 ||
|
|
|
Img->ImgHeight <= 0 ||
|
|
|
Img->bitcount <= 0 ||
|
|
|
Img->buff == NULL)
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
// 经纬度边界有效性验证
|
|
|
double NorthLat = Img->BoundingBox.NorthLat;
|
|
|
double SouthLat = Img->BoundingBox.SouthLat;
|
|
|
double WestLon = Img->BoundingBox.WestLon;
|
|
|
double EastLon = Img->BoundingBox.EastLon;
|
|
|
|
|
|
// 纬度验证
|
|
|
if (NorthLat > 90 || NorthLat < -90 ||
|
|
|
SouthLat > 90 || SouthLat < -90 ||
|
|
|
NorthLat - SouthLat <= 0)
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
// 经度验证
|
|
|
if (WestLon > 180 || WestLon < -180 ||
|
|
|
EastLon > 180 || EastLon < -180 ||
|
|
|
EastLon - WestLon <= 0)
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
// 数据有效，执行下属操作
|
|
|
int ImgWidth = Img->ImgWidth;
|
|
|
int ImgHeight = Img->ImgHeight;
|
|
|
int Imgchannels = Img->bitcount / 8;
|
|
|
|
|
|
if (Img->bitcount != 8 && Img->bitcount != 24)
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
// 创建临时图像
|
|
|
cv::Mat geoImg;
|
|
|
|
|
|
if (Img->bitcount == 8)
|
|
|
{
|
|
|
geoImg = cv::Mat(ImgHeight, ImgWidth, CV_8UC1);
|
|
|
}
|
|
|
else
|
|
|
{
|
|
|
geoImg = cv::Mat(ImgHeight, ImgWidth, CV_8UC3);
|
|
|
}
|
|
|
|
|
|
|
|
|
// 创建失败，直接返回
|
|
|
if (geoImg.empty() == true)
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
else
|
|
|
{
|
|
|
// 复制数据
|
|
|
int lineByte = ImgWidth * Img->bitcount / 8;
|
|
|
unsigned int imgBufSize = static_cast<unsigned int>(ImgHeight * lineByte);
|
|
|
memcpy(geoImg.data, Img->buff, imgBufSize);
|
|
|
}
|
|
|
|
|
|
// 经纬度 分辨率
|
|
|
double Lon_resolution = (EastLon - WestLon) / ImgWidth;
|
|
|
double Lat_resolution = (SouthLat - NorthLat) / ImgHeight;
|
|
|
|
|
|
|
|
|
// 注册
|
|
|
GDALAllRegister();
|
|
|
CPLSetConfigOption("GDAL_FILENAME_IS_UTF8", "NO");
|
|
|
|
|
|
//创建数据集
|
|
|
GDALDataset* poDstDS = NULL;
|
|
|
char** papszOptions = NULL;
|
|
|
string fomat;
|
|
|
fomat = "GTiff";
|
|
|
GDALDriver *poDriver = GetGDALDriverManager()->GetDriverByName( fomat.c_str());
|
|
|
|
|
|
double adfGeoTransform[6];
|
|
|
adfGeoTransform[0] = WestLon; /* top left x */
|
|
|
adfGeoTransform[1] = Lon_resolution;/* w-e pixel resolution */
|
|
|
adfGeoTransform[2] = 0; /* rotation, 0 if image is "north up" */
|
|
|
adfGeoTransform[3] = NorthLat; /* top left y */
|
|
|
adfGeoTransform[4] = 0; /* rotation, 0 if image is "north up" */
|
|
|
adfGeoTransform[5] = Lat_resolution;/* n-s pixel resolution */
|
|
|
|
|
|
if (Imgchannels == 1)
|
|
|
{
|
|
|
// 创建图像
|
|
|
poDstDS = poDriver->Create( savePath.c_str(), ImgWidth, ImgHeight, 1, GDT_Byte, papszOptions );
|
|
|
|
|
|
//设置椭球体信息
|
|
|
poDstDS->SetGeoTransform( adfGeoTransform );
|
|
|
OGRSpatialReference oSRS;
|
|
|
char* pszWKT = NULL;
|
|
|
oSRS.SetWellKnownGeogCS( "WGS84" );
|
|
|
oSRS.exportToWkt( &pszWKT );
|
|
|
poDstDS->SetProjection( pszWKT );
|
|
|
|
|
|
//遥感的一个波段
|
|
|
GDALRasterBand *poBand = poDstDS->GetRasterBand( 1 );
|
|
|
|
|
|
//分波段输出数据到tiff文件
|
|
|
poBand->RasterIO( GF_Write, 0, 0, ImgWidth, ImgHeight, geoImg.data, ImgWidth, ImgHeight, GDT_Byte, 0, 0 );
|
|
|
|
|
|
GDALFlushCache( poDstDS );
|
|
|
GDALClose( poDstDS );
|
|
|
|
|
|
GetGDALDriverManager()->DeregisterDriver(poDriver);
|
|
|
GDALDestroyDriverManager();
|
|
|
|
|
|
poDstDS = NULL;
|
|
|
poDriver = NULL;
|
|
|
}
|
|
|
else
|
|
|
{
|
|
|
// 创建图像
|
|
|
poDstDS = poDriver->Create( savePath.c_str(), ImgWidth, ImgHeight, 3, GDT_Byte, papszOptions );
|
|
|
|
|
|
//设置椭球体信息
|
|
|
poDstDS->SetGeoTransform( adfGeoTransform );
|
|
|
OGRSpatialReference oSRS;
|
|
|
char* pszWKT = NULL;
|
|
|
oSRS.SetWellKnownGeogCS( "WGS84" );
|
|
|
oSRS.exportToWkt( &pszWKT );
|
|
|
poDstDS->SetProjection( pszWKT );
|
|
|
|
|
|
//遥感的一个波段
|
|
|
GDALRasterBand *poBand1 = poDstDS->GetRasterBand( 1 );
|
|
|
GDALRasterBand *poBand2 = poDstDS->GetRasterBand( 2 );
|
|
|
GDALRasterBand *poBand3 = poDstDS->GetRasterBand( 3 );
|
|
|
|
|
|
std::vector<cv::Mat> planes;
|
|
|
cv::split(geoImg, planes);
|
|
|
|
|
|
//分波段输出数据到tiff文件
|
|
|
poBand1->RasterIO( GF_Write, 0, 0, ImgWidth, ImgHeight, planes[2].data, ImgWidth, ImgHeight, GDT_Byte, 0, 0 );
|
|
|
poBand2->RasterIO( GF_Write, 0, 0, ImgWidth, ImgHeight, planes[1].data, ImgWidth, ImgHeight, GDT_Byte, 0, 0 );
|
|
|
poBand3->RasterIO( GF_Write, 0, 0, ImgWidth, ImgHeight, planes[0].data, ImgWidth, ImgHeight, GDT_Byte, 0, 0 );
|
|
|
|
|
|
GDALFlushCache( poDstDS );
|
|
|
GDALClose( poDstDS );
|
|
|
|
|
|
GetGDALDriverManager()->DeregisterDriver(poDriver);
|
|
|
GDALDestroyDriverManager();
|
|
|
|
|
|
poDstDS = NULL;
|
|
|
poDriver = NULL;
|
|
|
}
|
|
|
|
|
|
return true;
|
|
|
}
|
|
|
catch(cv::Exception &e)
|
|
|
{
|
|
|
string str = e.msg;
|
|
|
|
|
|
return false;
|
|
|
}
|
|
|
catch(...)
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
}
|
|
|
|
|
|
//功能：保存图像数据至本地文件（GeoTiff格式）
|
|
|
//输入： 1. Img OpenCV图像
|
|
|
// 2. LBbox 经纬度边界
|
|
|
// 3. savePath Tiff文件保存路径
|
|
|
|
|
|
|
|
|
//输出： 1. 返回值：保存成功返回true，失败返回false
|
|
|
bool SaveImgStruToGeoTIFF(cv::Mat& img, GeoBoundingBox& LBbox, string savePath)
|
|
|
{
|
|
|
try
|
|
|
{
|
|
|
// 图像有效性验证
|
|
|
if (img.empty() == true || savePath == "")
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
if (img.type() != CV_8UC1 && img.type() != CV_8UC3)
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
// 经纬度边界有效性验证
|
|
|
double NorthLat = LBbox.NorthLat;
|
|
|
double SouthLat = LBbox.SouthLat;
|
|
|
double WestLon = LBbox.WestLon;
|
|
|
double EastLon = LBbox.EastLon;
|
|
|
|
|
|
// 纬度验证
|
|
|
if (NorthLat > 90 || NorthLat < -90 ||
|
|
|
SouthLat > 90 || SouthLat < -90 ||
|
|
|
NorthLat - SouthLat <= 0)
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
// 经度验证
|
|
|
if (WestLon > 180 || WestLon < -180 ||
|
|
|
EastLon > 180 || EastLon < -180 ||
|
|
|
EastLon - WestLon <= 0)
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
// 经纬度 分辨率
|
|
|
double Lon_resolution = (EastLon - WestLon) / img.cols;
|
|
|
double Lat_resolution = (SouthLat - NorthLat) / img.rows;
|
|
|
|
|
|
// 判断路径后缀的有效性
|
|
|
if ("tif" != savePath.substr(savePath.find_first_of('.'), savePath.length()))
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
// 注册
|
|
|
GDALAllRegister();
|
|
|
CPLSetConfigOption("GDAL_FILENAME_IS_UTF8", "NO");
|
|
|
|
|
|
//创建数据集
|
|
|
GDALDataset* poDstDS = NULL;
|
|
|
char** papszOptions = NULL;
|
|
|
string fomat;
|
|
|
fomat = "GTiff";
|
|
|
GDALDriver *poDriver = GetGDALDriverManager()->GetDriverByName( fomat.c_str());
|
|
|
|
|
|
double adfGeoTransform[6];
|
|
|
adfGeoTransform[0] = WestLon; /* top left x */
|
|
|
adfGeoTransform[1] = Lon_resolution;/* w-e pixel resolution */
|
|
|
adfGeoTransform[2] = 0; /* rotation, 0 if image is "north up" */
|
|
|
adfGeoTransform[3] = NorthLat; /* top left y */
|
|
|
adfGeoTransform[4] = 0; /* rotation, 0 if image is "north up" */
|
|
|
adfGeoTransform[5] = Lat_resolution;/* n-s pixel resolution */
|
|
|
|
|
|
if (img.type() == 0)
|
|
|
{
|
|
|
// 创建图像
|
|
|
poDstDS = poDriver->Create( savePath.c_str(), img.cols, img.rows, 1, GDT_Byte, papszOptions );
|
|
|
|
|
|
//设置椭球体信息
|
|
|
poDstDS->SetGeoTransform( adfGeoTransform );
|
|
|
OGRSpatialReference oSRS;
|
|
|
char* pszWKT = NULL;
|
|
|
oSRS.SetWellKnownGeogCS( "WGS84" );
|
|
|
oSRS.exportToWkt( &pszWKT );
|
|
|
poDstDS->SetProjection( pszWKT );
|
|
|
|
|
|
//遥感的一个波段
|
|
|
GDALRasterBand *poBand = poDstDS->GetRasterBand( 1 );
|
|
|
|
|
|
//分波段输出数据到tiff文件
|
|
|
poBand->RasterIO( GF_Write, 0, 0, img.cols, img.rows, img.data, img.cols, img.rows, GDT_Byte, 0, 0 );
|
|
|
|
|
|
GDALFlushCache( poDstDS );
|
|
|
GDALClose( poDstDS );
|
|
|
|
|
|
GetGDALDriverManager()->DeregisterDriver(poDriver);
|
|
|
GDALDestroyDriverManager();
|
|
|
|
|
|
poDstDS = NULL;
|
|
|
poDriver = NULL;
|
|
|
}
|
|
|
else
|
|
|
{
|
|
|
// 创建图像
|
|
|
poDstDS = poDriver->Create( savePath.c_str(), img.cols, img.rows, 3, GDT_Byte, papszOptions );
|
|
|
|
|
|
//设置椭球体信息
|
|
|
poDstDS->SetGeoTransform( adfGeoTransform );
|
|
|
OGRSpatialReference oSRS;
|
|
|
char* pszWKT = NULL;
|
|
|
oSRS.SetWellKnownGeogCS( "WGS84" );
|
|
|
oSRS.exportToWkt( &pszWKT );
|
|
|
poDstDS->SetProjection( pszWKT );
|
|
|
|
|
|
//遥感的一个波段
|
|
|
GDALRasterBand *poBand1 = poDstDS->GetRasterBand( 1 );
|
|
|
GDALRasterBand *poBand2 = poDstDS->GetRasterBand( 2 );
|
|
|
GDALRasterBand *poBand3 = poDstDS->GetRasterBand( 3 );
|
|
|
|
|
|
std::vector<cv::Mat> planes;
|
|
|
cv::split(img, planes);
|
|
|
|
|
|
//分波段输出数据到tiff文件
|
|
|
poBand1->RasterIO( GF_Write, 0, 0, img.cols, img.rows, planes[2].data, img.cols, img.rows, GDT_Byte, 0, 0 );
|
|
|
poBand2->RasterIO( GF_Write, 0, 0, img.cols, img.rows, planes[1].data, img.cols, img.rows, GDT_Byte, 0, 0 );
|
|
|
poBand3->RasterIO( GF_Write, 0, 0, img.cols, img.rows, planes[0].data, img.cols, img.rows, GDT_Byte, 0, 0 );
|
|
|
|
|
|
GDALFlushCache( poDstDS );
|
|
|
GDALClose( poDstDS );
|
|
|
|
|
|
GetGDALDriverManager()->DeregisterDriver(poDriver);
|
|
|
GDALDestroyDriverManager();
|
|
|
|
|
|
poDstDS = NULL;
|
|
|
poDriver = NULL;
|
|
|
}
|
|
|
|
|
|
return true;
|
|
|
}
|
|
|
catch(cv::Exception &e)
|
|
|
{
|
|
|
string str = e.msg;
|
|
|
|
|
|
return false;
|
|
|
}
|
|
|
catch(...)
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
} |