200 lines
6.7 KiB
C
200 lines
6.7 KiB
C
|
#pragma once
|
|||
|
|
|
|||
|
|
#include <QtWidgets/QMainWindow>
|
|||
|
|
#include <QDebug>
|
|||
|
|
#include <QDomDocument>
|
|||
|
|
#include <QFile>
|
|||
|
|
#include <QFileDialog>
|
|||
|
|
#include <QMessageBox>
|
|||
|
|
#include <QStack>
|
|||
|
|
#include <QTextStream>
|
|||
|
|
#include <QTime>
|
|||
|
|
|
|||
|
|
#include "ui_niirs.h"
|
|||
|
|
|
|||
|
|
#include <opencv2/opencv.hpp>
|
|||
|
|
#include <opencv2/imgproc/types_c.h>
|
|||
|
|
#include <opencv2/core/core.hpp>
|
|||
|
|
|
|||
|
|
#include <math.h>
|
|||
|
|
#include <omp.h>
|
|||
|
|
//#include <stdlib.h>
|
|||
|
|
//#include <iostream>
|
|||
|
|
|
|||
|
|
#include "gdal_priv.h"
|
|||
|
|
|
|||
|
|
using namespace std;
|
|||
|
|
|
|||
|
|
struct ImgInfo
|
|||
|
|
{
|
|||
|
|
GDALDataType dataType; //标示数据类型
|
|||
|
|
int bandCount; //数据波段数目
|
|||
|
|
int xSize; //数据在x方向上的尺寸
|
|||
|
|
int ySize; //数据在y方向上的尺寸
|
|||
|
|
int xOffset; //x方向上的偏移；
|
|||
|
|
int yOffset; //y方向上的偏移；
|
|||
|
|
char projWkt[1024]; //数据的投影方式
|
|||
|
|
double geoTransform[6]; //数据的仿射坐标
|
|||
|
|
double noData; //数据的背景值
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
class NIIRS : public QMainWindow
|
|||
|
|
{
|
|||
|
|
Q_OBJECT
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
NIIRS(QWidget* parent = Q_NULLPTR);
|
|||
|
|
|
|||
|
|
void setXmlInterface(QString strXmlInterface);
|
|||
|
|
|
|||
|
|
/* 计算NIIRS
|
|||
|
|
* @return NIIRS */
|
|||
|
|
double calculateNiirsScore();
|
|||
|
|
//计算10个客观参数
|
|||
|
|
void calculateObjectivePara();
|
|||
|
|
//void calculateEvaluationImage();
|
|||
|
|
double calculateRER();
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
* @brief 获取图像基础信息并保存至结构体
|
|||
|
|
* @param srcImgPath 输入文件路径
|
|||
|
|
* @param imgInfo 结构体，保存获取的tiff文件信息
|
|||
|
|
* @return 返回错误信息 */
|
|||
|
|
int getImageInfo(const char* srcImgPath, struct ImgInfo* imgInfo);
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
* @brief 按波段读取tiff
|
|||
|
|
* @param srcImgPath 输入文件路径
|
|||
|
|
* @param imgInfo 结构体，存储tiff文件信息
|
|||
|
|
* @param imgArray 读取的tiff数组
|
|||
|
|
* @param bandIndex 要读取的波段，bandIndex=0,则读取所有波段
|
|||
|
|
* @return 返回错误信息 */
|
|||
|
|
int readTiffFile(const char* srcImgPath, ImgInfo imgInfo, void* imgArray, int bandIndex);
|
|||
|
|
|
|||
|
|
void initTableWidget(int rowNum, int colNum);
|
|||
|
|
void setTableWidgetValue(double val, int row, int col);
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
* @brief 遍历文件夹下文件
|
|||
|
|
* @param path 输入文件路径
|
|||
|
|
* @param fileType 文件格式
|
|||
|
|
* @return 文件路径absoluteFilePath,返回空值时为存在错误 */
|
|||
|
|
QStringList getAllFiles(QString path, QString fileType);
|
|||
|
|
|
|||
|
|
public slots:
|
|||
|
|
void onRadioBtn_Visi();
|
|||
|
|
void onRadioBtn_Infra();
|
|||
|
|
|
|||
|
|
void on_pbtOpenImg_clicked();
|
|||
|
|
//void on_pbtOpenEdgeX_clicked();
|
|||
|
|
//void on_pbtOpenEdgeY_clicked();
|
|||
|
|
|
|||
|
|
private:
|
|||
|
|
Ui::NIIRSClass ui;
|
|||
|
|
|
|||
|
|
/* 计算噪声增益
|
|||
|
|
* @param kernel MTFC所使用的滤波器
|
|||
|
|
* @return MTFC_G噪声增益 */
|
|||
|
|
double calculateMTFC_G(cv::Mat kernel);
|
|||
|
|
////计算信噪比，mean/StdDev
|
|||
|
|
//double calculateSNR(cv::Mat img);
|
|||
|
|
|
|||
|
|
/* 计算数字图像亮度
|
|||
|
|
* @return 灰度图片平均亮度 */
|
|||
|
|
double calculateMeanStdDev(cv::Mat img);
|
|||
|
|
////计算图像标准差方差
|
|||
|
|
//float calculateStdDev(cv::Mat img);
|
|||
|
|
//计算图像对比度
|
|||
|
|
double calculateContrastRatio(cv::Mat img);
|
|||
|
|
////计算信息熵
|
|||
|
|
//double calculateEntropy(cv::Mat img);
|
|||
|
|
/*
|
|||
|
|
* @brief 计算单波段的图像熵
|
|||
|
|
* @param imgArray tiff数组
|
|||
|
|
* @param maxVal tiff波段中的最大值
|
|||
|
|
* @return 该波段的图像熵 */
|
|||
|
|
double calculateTiffEntropy(unsigned short* bandBuf, int bufSize, int maxVal);
|
|||
|
|
//计算相对边缘相应
|
|||
|
|
// double calculateERx(cv::Mat img);
|
|||
|
|
// double calculateERy(cv::Mat img);
|
|||
|
|
//return pair.first:ER, pair.second:H
|
|||
|
|
QPair<double, double> calculateERx(cv::Mat img);
|
|||
|
|
QPair<double, double> calculateERy(cv::Mat img);
|
|||
|
|
|
|||
|
|
QPair<double, double> calculate_rer_h(cv::Mat img, cv::Point pointA, cv::Point pointB);
|
|||
|
|
|
|||
|
|
cv::Point2d calculateFootPoint(array<cv::Point, 3>& ps);
|
|||
|
|
//float SFRCalculation(cv::Mat& ROI, double gamma);
|
|||
|
|
//void de_Gamma(cv::Mat& Src, double gamma);
|
|||
|
|
//std::vector<double> CentroidFind(cv::Mat& Src, std::vector<double>& y_shifts, double* CCoffset);
|
|||
|
|
//void SLR(std::vector<double>& Cen_Shifts, std::vector<double>& y_shifts, double* a, double* b);
|
|||
|
|
//void ReduceRows(double slope, int* ImgHeight);
|
|||
|
|
//std::vector<double> OverSampling(cv::Mat& Src, double slope, double CCoffset, int height, int width, int* SamplingLen);
|
|||
|
|
//std::vector<double> HammingWindows(std::vector<double>& deSampling, int SamplingLen);
|
|||
|
|
//void DFT(std::vector<double>& data, int size);
|
|||
|
|
|
|||
|
|
|
|||
|
|
//陡度
|
|||
|
|
//基于sobel算子，计算平均梯度
|
|||
|
|
double gradient_Sobel(cv::Mat img);
|
|||
|
|
//基于Laplacian算子，拉普拉斯导数，评价清晰度
|
|||
|
|
double definition_Laplacian(cv::Mat img);
|
|||
|
|
//四个方向的角二阶矩
|
|||
|
|
void get_GLCM_0deg(cv::Mat& input, cv::Mat& dst);
|
|||
|
|
void get_GLCM_90deg(cv::Mat& input, cv::Mat& dst);
|
|||
|
|
void get_GLCM_45deg(cv::Mat& input, cv::Mat& dst);
|
|||
|
|
void get_GLCM_135deg(cv::Mat& input, cv::Mat& dst);
|
|||
|
|
//计算二阶矩的特征值
|
|||
|
|
double calculateASM(cv::Mat& src);
|
|||
|
|
//PIQE
|
|||
|
|
double calculatePIQEScore(cv::Mat& img);
|
|||
|
|
bool noticeDistCriterion(cv::Mat Block, int nSegments, int blockSize_1,
|
|||
|
|
int windowSize, float blockImpairedThreshold, int blockSize);
|
|||
|
|
cv::Mat segmentEdge(cv::Mat blockEdge, int nSegments, int blockSize_1, int windowSize);
|
|||
|
|
float noiseCriterion(cv::Mat Block, int blockSize_1, float blockVar, float blockStdDev);
|
|||
|
|
float centerSurDev(cv::Mat Block, int blockSize_1);
|
|||
|
|
|
|||
|
|
// 计算垂足的坐标，array：点、直线两端点
|
|||
|
|
cv::Point calculate_foot_point(array<cv::Point, 3>& ps);
|
|||
|
|
// 计算两点距离
|
|||
|
|
float pointDistance(cv::Point pt1, cv::Point pt2);
|
|||
|
|
//// 计算两直线交点
|
|||
|
|
//cv::Point line_intersection(array<cv::Point, 2>& line1, array<cv::Point, 2>& line2);
|
|||
|
|
//// 根据一条线的点pt、斜率k、另一条水平或垂直的直线的端点，求交点
|
|||
|
|
//cv::Point point_k_edge_intersection(cv::Point pt, double k, array<cv::Point, 2>& line);
|
|||
|
|
//// 计算点到直线距离
|
|||
|
|
//int calculate_distance(array<cv::Point, 3>& ps);
|
|||
|
|
|
|||
|
|
bool jpegTest = false;
|
|||
|
|
//int mBandCount = 0;
|
|||
|
|
|
|||
|
|
QString mXmlInterface = "";
|
|||
|
|
|
|||
|
|
//QString mXmlImgPath = "/home/testdata/19data/VALID-ALIGACE/JB19-2_MSS_000811113_MMB04_001_01_001_003_L1/JB19-2_MSS_000811113_MMB04_001_01_001_003_L1.tiff";
|
|||
|
|
QString mXmlImgPath = "";//D:\\A_testdata\\niirs_test\\data\\GF2_mask1024.tif
|
|||
|
|
QString mXmlTopLeftX = "0";//311,336
|
|||
|
|
QString mXmlTopLeftY = "0";//324,293
|
|||
|
|
QString mXmlBottomRightX = "0";//335,348
|
|||
|
|
QString mXmlBottomRightY = "0";//337,311
|
|||
|
|
|
|||
|
|
QString mXmlProductDir = "";
|
|||
|
|
QString mResultXmlDir = "";
|
|||
|
|
|
|||
|
|
cv::Point mRerEdgePointA;
|
|||
|
|
cv::Point mRerEdgePointB;
|
|||
|
|
|
|||
|
|
|
|||
|
|
double imgLight = 0.0; //亮度
|
|||
|
|
double imgStdDev = 0.0; //标准差
|
|||
|
|
double imgContrastRadio = 0.0; //对比度
|
|||
|
|
double imgEntropy = 0.0; //熵
|
|||
|
|
double imgSNR = 0.0; //信噪比
|
|||
|
|
double imgGradient = 0.0; //陡度
|
|||
|
|
double imgDefinition = 0.0; //清晰度
|
|||
|
|
double imgASM = 0.0; //角二阶矩
|
|||
|
|
double imgPIQE = 0.0; //PIQE
|
|||
|
|
|
|||
|
|
double imgRER = 0.986; //相对边缘响应
|
|||
|
|
|
|||
|
|
};
|