-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.cpp
81 lines (60 loc) · 1.62 KB
/
main.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
#include <iostream>
#include <vector>
#include <complex>
#include <fstream>
#include <omp.h>
#include <chrono>
#include "AudioProcessing.h"
using namespace std;
void readInput(string filename, vector<complex<float>> &result)
{
ifstream f(filename);
if (!f)
{
cout << filename << " is not open!" << endl;
return;
}
while (!f.eof())
{
complex<float> sample;
f >> sample;
result.push_back(sample);
}
f.close();
}
void writeOutput(string filename, const vector<complex<float>> &output)
{
ofstream f(filename);
if (!f)
{
cout << filename << " is not open!" << endl;
return;
}
for (auto it=output.begin(); it != output.end(); it++)
{
f << fixed << it->real() << ((it->imag()>=0) ? "+" : "") << it->imag() << "j\n";
}
f.close();
}
int main()
{
vector<complex<float>> input;
readInput("input.txt", input);
cout << input.size() << " samples" << endl;
//
// vector<complex<float>> output = FFT(input);
//
// writeOutput("output.txt", output);
vector<complex<float>> output;
for (int i = 0; i < 4; i++)
{
omp_set_num_threads(pow(2, i));
chrono::time_point<chrono::system_clock> start = chrono::system_clock::now(), end;
FFT(input);
end = chrono::system_clock::now();
float elapsed = chrono::duration_cast<chrono::milliseconds>(end - start).count();
cout << pow(2, i) << " threads:" << elapsed << endl;
}
// writeOutput("output.txt", output);
return 0;
}