summaryrefslogtreecommitdiff
path: root/macros/fht.sci
blob: 9bd6b388c3a3cc24a2fb6402f7d7aa9d7bd9910f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
/* Description
        This function calculates the Fast Hartley transform of real input D.
        If D is a matrix, the Hartley transform is calculated along the columns by default.
        If N is specified, the first N elements along the specified dimension are used for the transform.
        If DIM is specified, the transform is calculated along the specified dimension.
   Calling Sequence
        M = fht (D)
        M = fht (D, N)
        M = fht (D, N, DIM)
   Parameters 
        D: Input data (real matrix or vector).
        N: Number of elements of D to be used for the transform (optional).
        DIM: Dimension along which the transform is to be computed (optional).
   Examples
        fht(1:4)
        ans =
       10   -4   -2   0  
*/
function M = fht(D, N, DIM)
    funcprot(0);
    rhs = argn(2);
    if rhs < 1 | rhs > 3 then
        error("Wrong number of input arguments.")
    end
    // The fht will be calculated along the first non-singleton dimension of the array i.e along the columns by default.
    dimension = size(D);
    nsdim = 1;
    for i = 1:length(dimension)
        if dimension(i) ~= 1 then
            nsdim = i;
            break;
        end
    end
    // Process input arguments
    select(rhs)
    case 1 then
        M = fft(D, -1, nsdim);
    case 2 then
        if isempty(N) then
            n = size(D, nsdim);
        else
            n = N;
        end
        new_size = size(D);
        new_size(nsdim) = n;
        D = resize_matrix(D, new_size);
        M = fft(D, -1, nsdim);
    case 3 then
        if isempty(N) then
            n = size(D, DIM);
        else
            n = N;
        end
        new_size = size(D);
        new_size(DIM) = n;
        D = resize_matrix(D, new_size);
        M = fft(D, -1, DIM);
    end
    // Return real part of the result
    M = real(M) - imag(M);
endfunction