aboutsummaryrefslogtreecommitdiffstats
path: root/doc/thesis/figures/data/qpsk_sim_constellations_static.py
blob: 61fdcb12797f864424694fedba07701ff0724f76 (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
#!/usr/bin/env python3

import os
import numpy as np

DATA_DIR = "./figures/data/flowgraphs"
SPS = 4

channel   = np.fromfile(f"{DATA_DIR}/qpsk_channel_static.dat", dtype=np.complex64)[::SPS]
sync      = np.fromfile(f"{DATA_DIR}/qpsk_sync_static.dat", dtype=np.complex64)
equalized = np.fromfile(f"{DATA_DIR}/qpsk_equalized_static.dat", dtype=np.complex64)
locked    = np.fromfile(f"{DATA_DIR}/qpsk_locked_static.dat", dtype=np.complex64)

samples = [channel, sync, equalized, locked]

# print(list(map(len, samples)))

start = 67.8e3
end = start + 200

get_range = lambda arr: arr[int(start):int(end)]
get_parts = lambda v: (np.real(v), np.imag(v))

values = map(get_range, samples)
parts = [p for v in map(get_parts, values) for p in v]
data = np.array(list(zip(*parts)))

# save to file
location = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__)))
name, _ = os.path.splitext(os.path.basename(__file__))
filename = os.path.join(location, name + ".dat")
np.savetxt(filename, data, fmt='%.6e')