fpc/tests/test/alglib/u_fht.pp
2009-12-10 22:25:34 +00:00

122 lines
3.1 KiB
ObjectPascal

(*************************************************************************
Copyright (c) 2009, Sergey Bochkanov (ALGLIB project).
>>> SOURCE LICENSE >>>
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 (www.fsf.org); 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.
A copy of the GNU General Public License is available at
http://www.fsf.org/licensing/licenses
>>> END OF LICENSE >>>
*************************************************************************)
unit u_fht;
interface
uses Math, Sysutils, u_ap, u_ftbase, u_fft;
procedure FHTR1D(var A : TReal1DArray; N : Integer);
procedure FHTR1DInv(var A : TReal1DArray; N : Integer);
implementation
(*************************************************************************
1-dimensional Fast Hartley Transform.
Algorithm has O(N*logN) complexity for any N (composite or prime).
INPUT PARAMETERS
A - array[0..N-1] - real function to be transformed
N - problem size
OUTPUT PARAMETERS
A - FHT of a input array, array[0..N-1],
A_out[k] = sum(A_in[j]*(cos(2*pi*j*k/N)+sin(2*pi*j*k/N)), j=0..N-1)
-- ALGLIB --
Copyright 04.06.2009 by Bochkanov Sergey
*************************************************************************)
procedure FHTR1D(var A : TReal1DArray; N : Integer);
var
Plan : FTPlan;
I : Integer;
FA : TComplex1DArray;
begin
Assert(N>0, 'FHTR1D: incorrect N!');
//
// Special case: N=1, FHT is just identity transform.
// After this block we assume that N is strictly greater than 1.
//
if N=1 then
begin
Exit;
end;
//
// Reduce FHt to real FFT
//
FFTR1D(A, N, FA);
I:=0;
while I<=N-1 do
begin
A[I] := FA[I].X-FA[I].Y;
Inc(I);
end;
end;
(*************************************************************************
1-dimensional inverse FHT.
Algorithm has O(N*logN) complexity for any N (composite or prime).
INPUT PARAMETERS
A - array[0..N-1] - complex array to be transformed
N - problem size
OUTPUT PARAMETERS
A - inverse FHT of a input array, array[0..N-1]
-- ALGLIB --
Copyright 29.05.2009 by Bochkanov Sergey
*************************************************************************)
procedure FHTR1DInv(var A : TReal1DArray; N : Integer);
var
I : Integer;
begin
Assert(N>0, 'FHTR1DInv: incorrect N!');
//
// Special case: N=1, iFHT is just identity transform.
// After this block we assume that N is strictly greater than 1.
//
if N=1 then
begin
Exit;
end;
//
// Inverse FHT can be expressed in terms of the FHT as
//
// invfht(x) = fht(x)/N
//
FHTR1D(A, N);
I:=0;
while I<=N-1 do
begin
A[I] := A[I]/N;
Inc(I);
end;
end;
end.