manyi/venv/Lib/site-packages/scipy/fftpack/tests/fftw_dct.c

#include <stdlib.h>
#include <stdio.h>

#include <fftw3.h>

#ifdef DCT_TEST_USE_SINGLE
typedef float float_prec;
#define PF "%.7f"
#define FFTW_PLAN fftwf_plan
#define FFTW_MALLOC fftwf_malloc
#define FFTW_FREE fftwf_free
#define FFTW_PLAN_CREATE fftwf_plan_r2r_1d
#define FFTW_EXECUTE fftwf_execute
#define FFTW_DESTROY_PLAN fftwf_destroy_plan
#define FFTW_CLEANUP fftwf_cleanup
#else
typedef double float_prec;
#define PF "%.18f"
#define FFTW_PLAN fftw_plan
#define FFTW_MALLOC fftw_malloc
#define FFTW_FREE fftw_free
#define FFTW_PLAN_CREATE fftw_plan_r2r_1d
#define FFTW_EXECUTE fftw_execute
#define FFTW_DESTROY_PLAN fftw_destroy_plan
#define FFTW_CLEANUP fftw_cleanup
#endif


enum type {
        DCT_I = 1,
        DCT_II = 2,
        DCT_III = 3,
        DCT_IV = 4,
        DST_I = 5,
        DST_II = 6,
        DST_III = 7,
	    DST_IV = 8,
};

int gen(int type, int sz)
{
        float_prec *a, *b;
        FFTW_PLAN p;
        int i, tp;

        a = FFTW_MALLOC(sizeof(*a) * sz);
        if (a == NULL) {
                fprintf(stderr, "failure\n");
                exit(EXIT_FAILURE);
        }
        b = FFTW_MALLOC(sizeof(*b) * sz);
        if (b == NULL) {
                fprintf(stderr, "failure\n");
                exit(EXIT_FAILURE);
        }

        switch(type) {
                case DCT_I:
                        tp = FFTW_REDFT00;
                        break;
                case DCT_II:
                        tp = FFTW_REDFT10;
                        break;
                case DCT_III:
                        tp = FFTW_REDFT01;
                        break;
                case DCT_IV:
                        tp = FFTW_REDFT11;
                        break;
                case DST_I:
                        tp = FFTW_RODFT00;
                        break;
                case DST_II:
                        tp = FFTW_RODFT10;
                        break;
                case DST_III:
                        tp = FFTW_RODFT01;
                        break;
                case DST_IV:
                        tp = FFTW_RODFT11;
                        break;
                default:
                        fprintf(stderr, "unknown type\n");
                        exit(EXIT_FAILURE);
        }

        switch(type) {
            case DCT_I:
            case DCT_II:
            case DCT_III:
            case DCT_IV:
                for(i=0; i < sz; ++i) {
                    a[i] = i;
                }
                break;
            case DST_I:
            case DST_II:
            case DST_III:
            case DST_IV:
/*                TODO: what should we do for dst's?*/
                for(i=0; i < sz; ++i) {
                    a[i] = i;
                }
                break;
            default:
                fprintf(stderr, "unknown type\n");
                exit(EXIT_FAILURE);
        }

        p = FFTW_PLAN_CREATE(sz, a, b, tp, FFTW_ESTIMATE);
        FFTW_EXECUTE(p);
        FFTW_DESTROY_PLAN(p);

        for(i=0; i < sz; ++i) {
                printf(PF"\n", b[i]);
        }
        FFTW_FREE(b);
        FFTW_FREE(a);

        return 0;
}

int main(int argc, char* argv[])
{
        int n, tp;

        if (argc < 3) {
                fprintf(stderr, "missing argument: program type n\n");
                exit(EXIT_FAILURE);
        }
        tp = atoi(argv[1]);
        n = atoi(argv[2]);

        gen(tp, n);
        FFTW_CLEANUP();

        return 0;
}