2025-02-10 23:13:41 +08:00
|
|
|
#pragma once
|
|
|
|
|
|
|
|
|
|
#ifndef EXTRACT_H
|
|
|
|
|
#define EXTRACT_H
|
|
|
|
|
|
|
|
|
|
#include <vector>
|
|
|
|
|
#include <iostream>
|
|
|
|
|
#include <cmath>
|
|
|
|
|
#include <fstream>
|
|
|
|
|
#include <sndfile.h>
|
|
|
|
|
#include <opencv2/opencv.hpp>
|
|
|
|
|
#include <fftw3.h>
|
|
|
|
|
#include <string>
|
2025-03-19 19:50:20 +08:00
|
|
|
#include <Eigen/Dense>
|
2025-02-10 23:13:41 +08:00
|
|
|
|
|
|
|
|
using namespace std;
|
2025-03-19 19:50:20 +08:00
|
|
|
using namespace Eigen;
|
2025-02-10 23:13:41 +08:00
|
|
|
|
2025-03-19 19:50:20 +08:00
|
|
|
int wm_extract(const wchar_t* path, int* wm, int wm_size);
|
2025-02-10 23:13:41 +08:00
|
|
|
|
2025-03-19 19:50:20 +08:00
|
|
|
vector<double> PQIMfft_extract(MatrixXd Gp_w, vector<int> &b, int M, double DD, double EE, int pp);
|
2025-02-10 23:13:41 +08:00
|
|
|
|
|
|
|
|
double drzh(const vector<vector<int>>& ww, vector<vector<int>>& W1, int pp, int qq);
|
|
|
|
|
|
|
|
|
|
double psnrzh(const vector<double>& yw, const vector<double>& yo);
|
|
|
|
|
|
|
|
|
|
double getWsr(const vector<double>& yw, const vector<double>& yo);
|
|
|
|
|
|
|
|
|
|
vector<vector<int>> igeneral(vector<vector<int>> W1, int m, int n, int a, int b);
|
|
|
|
|
|
|
|
|
|
// std::vector<int> randperm(int length, unsigned int seed);
|
|
|
|
|
|
|
|
|
|
#endif
|
