void Test::test_PolyMult_works() {
srand(time(0));
int n = 4;
int randMin = -100, randMax = 100;
Vector<Complex> P(n), Q(n);
for(int i = 0; i < n; ++i) {
double real = rand()%(randMax-randMin)+randMin;
double imaginary = rand()%(randMax-randMin)+randMin;
P[i] = Complex(real, imaginary);
Q[i] = Complex(real, imaginary);
}
FFT fft;
fft.preComputeOmega(n);
Vector<Complex> resultsR = fft.polyMultR(P, Q, n);
fft.preComputeRBS(n);
Vector<Complex> resultsD = fft.polyMultD(P, Q, n);
for(int i = 0; i < n; ++i) {
if(resultsD[i] != resultsR[i])
std::cout << "*** Failed ***" << std::endl;
std::cout << resultsR[i] << " " << resultsD[i] << std::endl;
}
}
void Test::test_PolyMult_recursive() {
int n = 2;
Vector<Complex> P(n),Q(n);
P[0] = Complex(2,0);
P[1] = Complex(2,0);
Q[0] = Complex(2,0);
Q[1] = Complex(2,0);
FFT fft;
fft.preComputeOmega(n);
Vector<Complex> results = fft.polyMultR(P, Q, n);
for(Complex c : results) {
std::cout << c << std::endl;
}
}
void Test::run_FFT_recursive() {
int minN = 32, maxN = 20000000;
std::cout << "FFT recursive" << std::endl
<< "N" << '\t' << "Time" << std::endl;
for(int n = minN; n <= maxN; n*=2) {
FFT fft;
fft.preComputeOmega(n);
int runTime = averageRuntime([&](){
Vector<Complex> P = fft.makeRandomVector(n, -1, 1);
Vector<Complex> Q = fft.makeRandomVector(n, -1, 1);
Vector<Complex> result = fft.polyMultR(P, Q, n);
}, RUNS);
std::cout << n << '\t' << runTime << std::endl;
}
}
void Test::test_FFT_vs_previous_approaches() {
std::cout << "Simple Test" << std::endl;
{
int n = 2;
double p[] = {3,5};
double q[] = {2,1};
Vector<double> P(p, p+n);
Vector<double> Q(q, q+n);
Vector<Complex> PFFT (P.begin(), P.end());
Vector<Complex> QFFT (Q.begin(), Q.end());
PolyMult polyMultSB;
Vector<double> pqActualSB = polyMultSB.schoolBook(n, P, Q);
PolyMult polyMultDnC4;
std::vector<double> pqActualDnC4 = polyMultDnC4.divideAndConquer4(n, P, Q);
PolyMult polyMultDnC3;
std::vector<double> pqActualDnC3 = polyMultDnC3.divideAndConquer3(n, P, Q);
FFT fft;
fft.preComputeOmega(n);
Vector<Complex> pqActualFFTR = fft.polyMultR(PFFT, QFFT, n);
fft.preComputeRBS(n);
Vector<Complex> pqActualFFTD = fft.polyMultD(PFFT, QFFT, n);
double pqExpected[] = {6,13,5};
int w1 = 10;
std::cout << std::setw(w1) << "Expected"
<< std::setw(w1) << "SB"
<< std::setw(w1) << "DnC 4"
<< std::setw(w1) << "DnC 3"
<< std::setw(w1) << "FFT R"
<< std::setw(w1) << "FFT D"
<< std::endl;
std::cout << std::right;
for(int index = 0; index < 2*n-1; ++index) {
std::cout << std::setw(w1) << pqExpected[index]
<< std::setw(w1) << pqActualSB[index]
<< std::setw(w1) << pqActualDnC4[index]
<< std::setw(w1) << pqActualDnC3[index]
<< std::setw(w1) << pqActualFFTR[index].real()
<< std::setw(w1) << pqActualFFTD[index].real()
<< std::endl;
}
}
std::cout << std::endl
<< "Random Test" << std::endl;
{
srand(time(0));
int n = 8;
int randMin = 0, randMax = 100;
Vector<double> P(n), Q(n);
Vector<Complex> PFFT(n), QFFT(n);
for(int i = 0; i < n; ++i) {
double randomP = rand()%(randMax-randMin)+randMin;
double randomQ = rand()%(randMax-randMin)+randMin;
P[i] = randomP;
Q[i] = randomQ;
PFFT[i] = Complex(randomP, 0);
QFFT[i] = Complex(randomQ, 0);
}
PolyMult polyMultSB;
Vector<double> pqActualSB = polyMultSB.schoolBook(n, P, Q);
PolyMult polyMultDnC4;
std::vector<double> pqActualDnC4 = polyMultDnC4.divideAndConquer4(n, P, Q);
PolyMult polyMultDnC3;
std::vector<double> pqActualDnC3 = polyMultDnC3.divideAndConquer3(n, P, Q);
FFT fft;
fft.preComputeOmega(n);
Vector<Complex> pqActualFFTR = fft.polyMultR(PFFT, QFFT, n);
fft.preComputeRBS(n);
Vector<Complex> pqActualFFTD = fft.polyMultD(PFFT, QFFT, n);
int w1 = 10;
std::cout << std::setw(w1) << "SB"
<< std::setw(w1) << "DnC 4"
<< std::setw(w1) << "DnC 3"
<< std::setw(w1) << "FFT R"
<< std::setw(w1) << "FFT D"
<< std::endl;
std::cout << std::right << std::fixed;
for(int index = 0; index < 2*n-1; ++index) {
std::cout << std::setprecision(0) << std::setw(w1) << pqActualSB[index]
<< std::setprecision(0) << std::setw(w1) << pqActualDnC4[index]
<< std::setprecision(0) << std::setw(w1) << pqActualDnC3[index]
<< std::setprecision(0) << std::setw(w1) << pqActualFFTR[index].real()
<< std::setprecision(0) << std::setw(w1) << pqActualFFTD[index].real()
<< std::endl;
}
}
}