/*	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;
}
