Pipelined-FFT/input_csv_gen.py at main · whimbree/Pipelined-FFT · GitHub

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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
import csv
import math
from typing import List
from util import get_input_order, get_output_order, format_fixed_point_hex
import numpy as np
import re


def main():
    input = generate_input(1024, complex(0, 0), complex(0.0000001, 0), False)
    fft = np.fft.fft(input)
    with np.printoptions(threshold=np.inf):
        print(fft)
    write_input("tb_input.csv", input)
    write_expected_output("tb_expectedOutput.csv", fft)


def generate_input(
    num_points: int,
    initial_amount: complex,
    increment_amount: complex,
    decreasing: bool,
) -> np.array:
    x = []

    if decreasing:
        for i in range(num_points, 0, -1):
            x.append(initial_amount + increment_amount * i)
    else:
        for i in range(1, num_points + 1):
            x.append(initial_amount + increment_amount * i)

    return np.asarray(x)


def write_input(filename: str, table: np.array):
    input = reorder_input(table)

    with open(filename, mode="w", newline="") as file:
        file_writer = csv.writer(
            file, delimiter=",", quotechar='"', quoting=csv.QUOTE_MINIMAL
        )

        for i in range(0, len(input[0])):
            file_writer.writerow(
                [
                    format_fixed_point_hex(i)
                    for i in [
                        input[0][i].real,
                        input[0][i].imag,
                        input[1][i].real,
                        input[1][i].imag,
                        input[2][i].real,
                        input[2][i].imag,
                        input[3][i].real,
                        input[3][i].imag,
                    ]
                ]
            )


def reorder_input(table: np.array) -> np.array:
    input_order = get_input_order(len(table))

    return np.asarray(
        [[table[i] for i in input_order[j]] for j in range(len(input_order))]
    )


def write_expected_output(filename: str, table: np.array):
    output = reorder_output(table)

    with open(filename, mode="w", newline="") as file:
        file_writer = csv.writer(
            file, delimiter=",", quotechar='"', quoting=csv.QUOTE_MINIMAL
        )

        for i in range(0, len(output[0])):
            file_writer.writerow(
                [
                    format_fixed_point_hex(i)
                    for i in [
                        output[0][i].real,
                        output[0][i].imag,
                        output[1][i].real,
                        output[1][i].imag,
                        output[2][i].real,
                        output[2][i].imag,
                        output[3][i].real,
                        output[3][i].imag,
                    ]
                ]
            )


def reorder_output(table: np.array) -> np.array:
    output_order = get_output_order(len(table))

    return np.asarray(
        [[table[i] for i in output_order[j]] for j in range(len(output_order))]
    )


if __name__ == "__main__":
    main()