To modify the existing code for the FFT function, you can use the following code:
function Xk = myfft(x,n)
N = length(x); % The fundamental period of the DFT
if N<n
x=[x,zeros(1,n-N)];
N=n;
end
n = 0:1:N-1; % row vector for n
k = 0:1:N-1; % row vecor for k
WN = exp(-1i2pi/N); % Wn factor
nk = n'*k; % creates a N by N matrix of nk values
WNnk = WN .^ nk; % DFT matrix
Xk = x * WNnk; % row vector for DFT coefficients
The modifications include passing an additional parameter 'n' to the function to specify the desired size of the output FFT, and checking if the length of input 'x' is less than 'n', in which case zero-padding is done to make the length equal to 'n'.
Similarly, to modify the existing code for the IFFT function, you can use the following code:
function x_hat = myifft(ftx,n)
N = length(ftx); % The fundamental period of the DFT
if N<n
ftx=[ftx,zeros(1,n-N)];
N=n;
end
n = 0:1:N-1; % row vector for n
k = 0:1:N-1; % row vecor for k
WN = exp(-1i2pi/N); % Wn factor
nk = n'*k; % creates a N by N matrix of nk values
WNnk = WN .^(-nk); % IDFS matrix
x_hat = (ftx * WNnk)/N; % row vector for IDFS values
x_hat=real(x_hat);
