// QB_ImageProcessFunc.cpp : 定义 DLL 的初始化例程。
//

#include "stdafx.h"

#define IMGPROCESS_EXPORTS 

// 包含头文件
#include "QB_ImgProcessFun.h"
#include "SysGeoCorrect.h"
#include "GeoTIFFEdit.h"
#include "IntensityControl.h"
#include "ContrastControl.h"
#include "DefinitionControl.h"
#include "GeoInfoCalculate.h"
#include "GCPGeoCorrect.h"
#include "DecreaseHazeControl.h"
#include "GetElevation.h"
#include "fuseImg.h"
#include "superResolutionImg.h"

//	功能：图像去雾处理

//1.1. 图像去雾控制 
IMGPROCESS_API bool DoDehazeCtrl(QBStru *qbData, int A_MAX, double degree, bool bSrcFirst)
{
	return QBStruDeHazeControl(qbData, A_MAX, degree, bSrcFirst);
}

//1.2 功能：图像去雾控制（函数重载）
IMGPROCESS_API bool DoDehazeCtrl(ImgStru* src, ImgStru* dst, int A_MAX, double degree)
{
	return ImgStruDeHazeControl(src, dst, A_MAX, degree);
}



//2.1 功能：图像亮度控制
IMGPROCESS_API bool DoIntensityCtrl(QBStru *qbData, int degree, bool bSrcFirst)
{
	return QBStruIntensityControl(qbData, degree, bSrcFirst);
}

//2.2 功能：图像亮度控制
IMGPROCESS_API bool DoIntensityCtrl(ImgStru* src, ImgStru* dst, int degree)
{
	return ImgStruIntensityControl(src, dst, degree);
}




//3.1 功能：图像对比度控制
IMGPROCESS_API bool DoContrastCtrl(QBStru *qbData, int degree, bool bSrcFirst)
{
	return QBStruContrastControl(qbData, degree, bSrcFirst);
}

//3.2 功能：图像对比度控制（函数重载）
IMGPROCESS_API bool DoContrastCtrl(ImgStru* src, ImgStru* dst, int degree)
{
	return ImgStruContrastControl(src, dst, degree);
}


//4.1 功能：图像清晰度控制
IMGPROCESS_API bool DoDefinitionCtrl(QBStru *qbData, int degree, bool bSrcFirst)
{
	return QBStruDefinitionControl(qbData, degree, bSrcFirst);
}

//4.2 功能：图像清晰度控制（函数重载）
IMGPROCESS_API bool DoDefinitionCtrl(ImgStru* src, ImgStru* dst, int degree)
{
	return ImgStruDefinitionControl(src, dst, degree);
}


//5.1 功能：计算图像中心点经纬度坐标、载荷等效俯仰角和方位角
IMGPROCESS_API bool DoCalImgCenterCoordinate(double &Lon, double &Lat, double &H, const struQB_FJ *frame)
{
	return CalImgCenterCoordinate(Lon, Lat, H, frame);
}

//5.2 功能：计算图像中任意点经纬度坐标
IMGPROCESS_API bool DoCalAnyPtCoordinate(double &Lon, double &Lat, double &H, const struQB_FJ *frame, int pt_x, int pt_y)
{
	return CalAnyPtCoordinate(Lon, Lat, H, frame, pt_x, pt_y);
}

//6.1 功能：基于复接数据的图像系统级几何校正
IMGPROCESS_API bool DoSysGeoCorrect(QBStru *qbData, bool bSrcFirst)
{
	return SysGeoCorrectImg(qbData, bSrcFirst);
}

//6.2 功能：基于复接数据的图像系统级几何校正（重载）
IMGPROCESS_API bool DoSysGeoCorrect(ImgStru *dstImg, const ImgStru *srcImg, const struQB_FJ *frameData)
{
	return SysGeoCorrectImg(dstImg, srcImg, frameData);
}

//6.3 功能：基于地面控制点的几何精校正
IMGPROCESS_API bool DoGCPGeoCorrectImg(ImgStru *dstImg, const ImgStru *srcImg, const std::vector<Point2d>& GCP_XY, const std::vector<Point2d>& GCP_LonLat, int type)
{
	return GCPGeoCorrectImg(dstImg, srcImg, GCP_XY, GCP_LonLat, type);
}

//6.4 功能： OpenCV图像格式基于地面控制点的几何精校正
IMGPROCESS_API bool DoGCPGeoCorrectImg( cv::Mat& dstImg, GeoBoundingBox& LBbox, const cv::Mat& srcImg, const std::vector<cv::Point2d>& GCP_XY, const std::vector<cv::Point2d>& GCP_LonLat, int type)
{
	return GCPGeoCorrectImg(dstImg, LBbox, srcImg, GCP_XY, GCP_LonLat, type);
}

//6.5 功能：基于控制点的几何校正
IMGPROCESS_API bool DoCPBasedImgCorrect( cv::Mat &dstImg, const cv::Mat &srcImg, const std::vector<cv::Point2d>& GCP_xy, const std::vector<cv::Point2d>& GCP_XY)
{
	return CPBasedImgCorrect(dstImg, srcImg, GCP_xy, GCP_XY);
}


//7.1 功能：保存情报数据至本地文件（GeoTiff格式）
IMGPROCESS_API bool SaveToGeoTIFF(QBStru *qbData, string savePath)
{
	return SaveQBDataToGeoTIFF(qbData, savePath);
}

//7.2 功能：保存图像数据至本地文件（GeoTiff格式）
IMGPROCESS_API bool SaveToGeoTIFF(const ImgStru *Img, string savePath)
{
	return SaveImgStruToGeoTIFF(Img, savePath);
}

//7.3 功能：保存图像数据至本地文件（GeoTiff格式）
IMGPROCESS_API bool SaveToGeoTIFF(cv::Mat& Img, GeoBoundingBox& LBbox, string savePath)
{
	return SaveImgStruToGeoTIFF(Img, LBbox, savePath);
}



//8.1 功能：计算图像（几何校正前）中两点间的距离和方位（重载）
IMGPROCESS_API bool CalculateTwoPtsDistanceAzimuth(double &distance, double &azAngle, const struQB_FJ *frameData, cv::Point pt1, cv::Point pt2)
{
	return CalTwoPointsDistanceAzimuth(distance, azAngle, frameData, pt1, pt2);
}

//8.2 功能：计算图像中两点间的距离和方位
IMGPROCESS_API bool CalculateTwoPtsDistanceAzimuth(double &distance, double &azAngle, const QBStru *qbData, cv::Point pt1, cv::Point pt2, int flags)
{
	return QBStruCalTwoPtsDistanceAzimuth(distance, azAngle, qbData, pt1, pt2, flags);
}



//9.1 功能：计算图像中多点组成的多边形面积
IMGPROCESS_API bool CalculatePloygonArea(double& area, const struQB_FJ *frameData, cv::Point* points, int count)
{
	return CalPolygonArea(area, frameData, points, count);
}

//9.2 功能：计算图像中多边形面积
IMGPROCESS_API bool CalculatePloygonArea(double &area, const QBStru *qbData, cv::Point* point, int count, int flags)
{
	return QBStruCalPolygonArea(area, qbData, point, count, flags);
}


//10 读取高程
//功能：基于经纬度读取地面高程
//输入：
//    1. fLonDeg: 经度（度）
//    2. fLatDeg: 纬度（度）
//    3. sDir: 高程数据文件存储路径， 默认设为："D:\\ElevationData"
// 
//输出：
//    1. fElevator:高程（米）
//
//返回值：若函数运行成功，返回true，否则返回false
bool GetElevation( float& fElevator, double dLonDeg, double dLatDeg, string sDir, bool bNearst)
{
	return GetDem(fElevator, dLonDeg, dLatDeg, sDir, bNearst);
}

// 清理DEM记忆
void ClearDem()
{
	ClearDemMem();
}


// 11.1 功能：函数 DoFastFuseImgRGBplusIR() 快速融合两幅输入图像
// 输入：    
//	        img1： 输入图像1：RGB：三通道
//	        img2： 输入图像2：IR：三通道或单通道
// 输出：  
//          dst: 融合图像
// 返回值：
//          运算有效返回true，否则返回false
bool DoFastFuseImgRGBplusIR(cv::Mat &dst, const cv::Mat RGBimg, const cv::Mat IRimg)
{
	return FastFuseImgRGBplusIR(dst, RGBimg, IRimg);
}


// 11.2 功能：函数 DoFuseImgRGBplusIR() 基于小波变换融合两幅输入图像
// 输入：    
//	        img1： 输入图像1：RGB：三通道
//	        img2： 输入图像2：IR：三通道或单通道
//          wname: 小波名称
//                 支持小波：
//							1. haar
//							2. symN  [1, 10]
//							3. dbN   [1, 10]
//							4. coifN [1, 5]
//          level：小波分解层数
// 输出：  
//          dst: 融合图像
// 返回值：
//          运算有效返回true，否则返回false
bool DoFuseImgRGBplusIR(cv::Mat &dst, const cv::Mat RGBimg, const cv::Mat IRimg, const string wname, const unsigned int level)
{
	return FuseImgRGBplusIR(dst, RGBimg, IRimg, wname, level);
}



// 11.3 功能：函数 FuseImg() 基于小波变换融合两幅输入图像
// 输入：    
//	        img1： 输入图像1
//	        img2： 输入图像2
//          wname: 小波名称
//                 支持小波：
//							1. haar
//							2. symN  [1, 10]
//							3. dbN   [1, 10]
//							4. coifN [1, 10]
//          level：小波分解层数
// 输出：  
//          dst: 融合图像
// 返回值：
//          运算有效返回true，否则返回false
bool DoFuseImg(cv::Mat &dst, const cv::Mat img1, const cv::Mat img2, const string wname, const unsigned int level)
{
	return FuseImg(dst, img1, img2, wname, level);
}



// 功能：灰度图像（红外图像）超分辨率构建
// 输入：    
//	        imgDeque： 输入图像序列，要求图像尺寸相同，类型为CV_8UC1
// 输出：  
//          dst:  输出超分图像
// 返回重建中有效的图像数目
int GetSuperResolutionImg(cv::Mat &dst, const std::deque<cv::Mat> &imgDeque, double multiple)
{
	return SuperResolutionImg(dst, imgDeque, multiple);
}