import numpy as np
from matplotlib import pyplot as plt
from fpu_test_gen import reverse_endian, dtype_size

WEIGTHS = [
    [
        [0.7095146, -0.41895103, -0.08075078, -0.5218736],
        [-0.4351325, 1.0214638, 0.14494987, -0.78134096],
        [0.38553882, -1.0607314, 0.01327306, -0.28972188],
        [0.84955347, 0.32464203, 0.8879888, 0.00756884],
        [-0.8693629, 0.8418823, 0.60206324, -0.78290594],
        [0.1586302, 0.01737848, 0.75329006, -0.57819647],
        [-0.16126093, 0.5317601, 0.34316933, -0.7074082],
        [0.09219088, -0.624525, -0.61903083, -0.87057704]
    ],
    [[0.36770403, -0.78046024, 0.3979908, 0.5494289, -0.13859335, 0.40053025, 0.08249452, -0.32528356],
     [-0.17659009, 0.13901198, -0.45248222, -0.7894139, -0.81092286, -0.521815, 0.30632392, -0.3143816],
     [-0.04314173, 0.14361085, 0.6259473, 0.3571782, -0.38011226, 0.01378736, 0.05794358, 0.09667788],
     [-0.46864474, 0.36618456, -0.45595396, -0.39789405, 0.73964316, -0.30294785, 0.2482118, -0.2127953],
     [-0.37941265, 0.45330787, -0.12066315, 0.5636705, 0.68990386, 0.6543718, 0.86367106, -0.5707757],
     [-0.78606385, 0.24032554, -0.4472755, -0.24661142, -0.2698564, -0.8365823, -0.13674814, -0.39799848],
     [0.11138931, 0.48950365, 0.12998834, 0.4947537, 0.516593, 0.82281274, 0.04789656, 0.30206403],
     [0.23097174, 0.30290592, -0.596446, -0.40108407, 0.12246455, -0.47260976, -0.55030185, 0.44481543]
     ]
    [
        [0.5724262, 0.5853241, 0.3748752, -0.892384, -1.0270239, 0.2170913, -0.07271451, 0.14661156],
        [0.30391088, -0.92324615, 0.8088594, -1.0522624, 0.07374455, -0.550893, 0.8194236, -0.62796086]
    ]
]
BIAS = [
    # L1
    [0.01425434, -0.06219335, -0.0201127, -0.04791382, -0.04360008, -0.05311861, -0.01731363, -0.00014839],
    # L2
    [0.03480967, 0.06208326, -0.01576317, -0.00037753, -0.03940378, 0.05157978, -0.02775403, 0.04540931],
    # L3
    [0.03787775, -0.03655371],
]


def generate_nn_values(dtype=np.float32):
    dsize = dtype_size(dtype)


if __name__ == '__main__':
    generate_nn_values()