wdsp/wisdom.c
Uladzimir Karpenka 89c8a0e2b5 first commit
2026-06-01 15:58:45 +03:00

120 lines
4.0 KiB
C

/* wisdom.c
This file is part of a program that implements a Software-Defined Radio.
Copyright (C) 2013-2025 Warren Pratt, NR0V
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
The author can be reached by email at
warren@wpratt.com
*/
#define _CRT_SECURE_NO_WARNINGS
#include "comm.h"
static char status[128];
PORT
char* wisdom_get_status()
{
return status;
}
PORT
int WDSPwisdom (char* directory)
{
int wisdom_return = 0; // 0 from existing, 1 rebuilt
fftw_plan tplan;
int psize;
#ifdef _WIN32
FILE *stream;
#endif
double* fftin;
double* fftout;
char wisdom_file[1024];
const int maxsize = max (MAX_WISDOM_SIZE_DISPLAY, MAX_WISDOM_SIZE_FILTER + 1);
strcpy (wisdom_file, directory);
strncat (wisdom_file, "wdspWisdom00", 16);
if(!fftw_import_wisdom_from_filename(wisdom_file))
{
fftin = (double *) malloc0 (maxsize * sizeof (complex));
fftout = (double *) malloc0 (maxsize * sizeof (complex));
#ifdef _WIN32
AllocConsole(); // create console
freopen_s(&stream, "conout$", "w", stdout); // redirect output to console
#endif
fprintf(stdout, "Optimizing FFT sizes through %d\n\n", maxsize);
fprintf(stdout, "Please do not close this window until wisdom plans are completed.\n\n");
sprintf(status, "Optimizing FFT sizes through %d", maxsize);
psize = 64;
while (psize <= MAX_WISDOM_SIZE_FILTER)
{
fprintf(stdout, "Planning COMPLEX FORWARD FFT size %d\n", psize);
fflush(stdout);
sprintf(status, "Planning COMPLEX FORWARD FFT size %d\n", psize);
tplan = fftw_plan_dft_1d(psize, (fftw_complex *)fftin, (fftw_complex *)fftout, FFTW_FORWARD, FFTW_PATIENT);
fftw_execute (tplan);
fftw_destroy_plan (tplan);
fprintf(stdout, "Planning COMPLEX BACKWARD FFT size %d\n", psize);
fflush(stdout);
sprintf(status, "Planning COMPLEX BACKWARD FFT size %d\n", psize);
tplan = fftw_plan_dft_1d(psize, (fftw_complex *)fftin, (fftw_complex *)fftout, FFTW_BACKWARD, FFTW_PATIENT);
fftw_execute (tplan);
fftw_destroy_plan (tplan);
fprintf(stdout, "Planning COMPLEX BACKWARD FFT size %d\n", psize + 1);
fflush(stdout);
sprintf(status, "Planning COMPLEX BACKWARD FFT size %d\n", psize + 1);
tplan = fftw_plan_dft_1d(psize + 1, (fftw_complex *)fftin, (fftw_complex *)fftout, FFTW_BACKWARD, FFTW_PATIENT);
fftw_execute (tplan);
fftw_destroy_plan (tplan);
psize *= 2;
}
psize = 64;
while (psize <= MAX_WISDOM_SIZE_DISPLAY)
{
if (psize > MAX_WISDOM_SIZE_FILTER)
{
fprintf(stdout, "Planning COMPLEX FORWARD FFT size %d\n", psize);
fflush(stdout);
sprintf(status, "Planning COMPLEX FORWARD FFT size %d\n", psize);
tplan = fftw_plan_dft_1d(psize, (fftw_complex *)fftin, (fftw_complex *)fftout, FFTW_FORWARD, FFTW_PATIENT);
fftw_execute (tplan);
fftw_destroy_plan (tplan);
}
fprintf(stdout, "Planning REAL FORWARD FFT size %d\n", psize);
fflush(stdout);
sprintf(status, "Planning REAL FORWARD FFT size %d\n", psize);
tplan = fftw_plan_dft_r2c_1d(psize, fftin, (fftw_complex *)fftout, FFTW_PATIENT);
fftw_execute (tplan);
fftw_destroy_plan (tplan);
psize *= 2;
}
fprintf(stdout, "\nFFTW planning complete.\n");
fflush(stdout);
sprintf(status, "\nFFTW planning complete.\n");
fftw_export_wisdom_to_filename(wisdom_file);
_aligned_free (fftout);
_aligned_free (fftin);
#ifdef _WIN32
FreeConsole(); // dismiss console
#endif
wisdom_return = 1;
}
return wisdom_return;
}