88 lines
2.9 KiB
C++
88 lines
2.9 KiB
C++
|
#include "pch.h"
|
|||
|
|
#pragma comment(lib, "E:\\Lib\\libfftw3-3.lib")
|
|||
|
|
#pragma comment(lib, "E:\\Lib\\libfftw3f-3.lib")
|
|||
|
|
#pragma comment(lib, "E:\\Lib\\libfftw3l-3.lib")
|
|||
|
|
#include <stdio.h>
|
|||
|
|
#include <stdlib.h>
|
|||
|
|
#include "fftw3.h"
|
|||
|
|
#include <cmath>
|
|||
|
|
#include <algorithm>
|
|||
|
|
#include <iostream>
|
|||
|
|
//<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ƶ<EFBFBD><C6B5>Ϊ FR , <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ϊ1024 (NN),<2C><>ô<EFBFBD><C3B4><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>N<EFBFBD><4E><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ƶ<EFBFBD><C6B5> X(N) = (FR / NN) * N
|
|||
|
|
#define FR 44100
|
|||
|
|
class Get_RGB {
|
|||
|
|
public:
|
|||
|
|
static void fftw3(float *in,int NN, int &hz);
|
|||
|
|
double hztospectrun(double hz);
|
|||
|
|
static void spectral_color(double &r, double &g, double &b, double l);
|
|||
|
|
static void GetRGB( float* in,int NN, double &R, double &G,double &B){//<2F><><EFBFBD><EFBFBD>RGBֵ
|
|||
|
|
int hz;
|
|||
|
|
double l=0;
|
|||
|
|
fftw3(in,NN, hz);
|
|||
|
|
spectral_color(R, G, B, l);
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
void Get_RGB::fftw3(float *in ,int NN, int &frequency)
|
|||
|
|
{
|
|||
|
|
struct Max {
|
|||
|
|
int id=0;
|
|||
|
|
int value=0;
|
|||
|
|
}M;
|
|||
|
|
|
|||
|
|
int i;
|
|||
|
|
fftwf_complex *out = NULL; // fftwf_complex --> <20><>Ϊfloat<61>汾
|
|||
|
|
fftwf_plan p;
|
|||
|
|
in = (float *)fftwf_malloc(sizeof(float) * NN);
|
|||
|
|
out = (fftwf_complex *)fftwf_malloc(sizeof(fftwf_complex) *NN);
|
|||
|
|
// <20><><EFBFBD><EFBFBD>Ҷ<EFBFBD>任
|
|||
|
|
p = fftwf_plan_dft_r2c_1d(NN, in, out, FFTW_ESTIMATE);
|
|||
|
|
fftwf_execute(p);
|
|||
|
|
// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
|||
|
|
for (i = 0; i < NN; i++)
|
|||
|
|
{
|
|||
|
|
float len = sqrt(out[i][0] * out[i][0] + out[i][1] * out[i][1]);
|
|||
|
|
if (len > M.value) {
|
|||
|
|
M.id=i ;
|
|||
|
|
}
|
|||
|
|
// printf("%d ",i);
|
|||
|
|
// printf("%.2f ", len);
|
|||
|
|
}
|
|||
|
|
frequency = M.id*(FR / NN);
|
|||
|
|
//frequency = M.id*(FR / NN);
|
|||
|
|
// <20>ͷ<EFBFBD><CDB7><EFBFBD>Դ
|
|||
|
|
|
|||
|
|
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
int main() {
|
|||
|
|
float in[10000];
|
|||
|
|
double R, G, B;
|
|||
|
|
int NN;
|
|||
|
|
std::cin >> NN;
|
|||
|
|
for (int i=0;i < NN;i++) {
|
|||
|
|
std::cin >> in[i];
|
|||
|
|
}
|
|||
|
|
Get_RGB::GetRGB(in,NN, R, G, B);
|
|||
|
|
std::cout << R << " " << G << " " << B;
|
|||
|
|
std::cout << "\n"<< R << " " << G << " " << B;
|
|||
|
|
}
|
|||
|
|
double Get_RGB::hztospectrun(double hz) {
|
|||
|
|
double hbox;
|
|||
|
|
hbox = hz;
|
|||
|
|
for (;hbox > 800;hbox = hbox / 2);
|
|||
|
|
return hbox;
|
|||
|
|
}
|
|||
|
|
void Get_RGB::spectral_color(double &r, double &g, double &b, double l) // RGB <0,1> <- lambda l <400,700> [nm]
|
|||
|
|
{
|
|||
|
|
double t; r = 0.0; g = 0.0; b = 0.0;
|
|||
|
|
if ((l >= 400.0) && (l < 410.0)) { t = (l - 400.0) / (410.0 - 400.0); r = +(0.33*t) - (0.20*t*t); }
|
|||
|
|
else if ((l >= 410.0) && (l < 475.0)) { t = (l - 410.0) / (475.0 - 410.0); r = 0.14 - (0.13*t*t); }
|
|||
|
|
else if ((l >= 545.0) && (l < 595.0)) { t = (l - 545.0) / (595.0 - 545.0); r = +(1.98*t) - (t*t); }
|
|||
|
|
else if ((l >= 595.0) && (l < 650.0)) { t = (l - 595.0) / (650.0 - 595.0); r = 0.98 + (0.06*t) - (0.40*t*t); }
|
|||
|
|
else if ((l >= 650.0) && (l < 700.0)) { t = (l - 650.0) / (700.0 - 650.0); r = 0.65 - (0.84*t) + (0.20*t*t); }
|
|||
|
|
if ((l >= 415.0) && (l < 475.0)) { t = (l - 415.0) / (475.0 - 415.0); g = +(0.80*t*t); }
|
|||
|
|
else if ((l >= 475.0) && (l < 590.0)) { t = (l - 475.0) / (590.0 - 475.0); g = 0.8 + (0.76*t) - (0.80*t*t); }
|
|||
|
|
else if ((l >= 585.0) && (l < 639.0)) { t = (l - 585.0) / (639.0 - 585.0); g = 0.84 - (0.84*t); }
|
|||
|
|
if ((l >= 400.0) && (l < 475.0)) { t = (l - 400.0) / (475.0 - 400.0); b = +(2.20*t) - (1.50*t*t); }
|
|||
|
|
else if ((l >= 475.0) && (l < 560.0)) { t = (l - 475.0) / (560.0 - 475.0); b = 0.7 - (t)+(0.30*t*t); }
|
|||
|
|
}
|