Skip to content

waveform_generator

Waveform synthesis: sine, cosine, square, triangle, sawtooth, Zadoff-Chu sequences, and binary PRBS patterns.

rfsoc_rfdc.waveform_generator

Attributes

generator = WaveFormGenerator() module-attribute

sine_wave = generator.generate_sine_wave() module-attribute

square_wave = generator.generate_square_wave() module-attribute

triangle_wave = generator.generate_triangle_wave() module-attribute

sawtooth_wave = generator.generate_sawtooth_wave() module-attribute

zadoff_chu_wave = generator.generate_zadoff_chu_wave() module-attribute

Classes

WaveFormGenerator

A class that generates different types of waveforms.

Functions
generate_sine_wave(repeat_time=1, sample_pts=1000, scaling_factor=1.0, raw_np=False) staticmethod
Source code in rfsoc_rfdc/waveform_generator.py 
11
12
13
14
15
16
17
18
19
20
@staticmethod
def generate_sine_wave(repeat_time=1, sample_pts=1000, scaling_factor=1.0, raw_np=False):
    t = np.linspace(0, repeat_time, int(
        sample_pts * repeat_time), endpoint=False)
    wave = np.sin(2 * np.pi * t)
    if raw_np:
        return wave
    else:
        wave *= int(MyRFdcType.DAC_MAX_SCALE * scaling_factor)
    return wave.astype(MyRFdcType.DATA_PATH_DTYPE)
generate_square_wave(repeat_time=1, sample_pts=1000, scaling_factor=1.0) staticmethod
Source code in rfsoc_rfdc/waveform_generator.py 
22
23
24
25
26
27
28
@staticmethod
def generate_square_wave(repeat_time=1, sample_pts=1000, scaling_factor=1.0):
    t = np.linspace(0, repeat_time, int(
        sample_pts * repeat_time), endpoint=False)
    wave = np.sign(np.sin(2 * np.pi * t)) * \
        int(MyRFdcType.DAC_MAX_SCALE * scaling_factor)
    return wave.astype(MyRFdcType.DATA_PATH_DTYPE)
generate_triangle_wave(repeat_time=1, sample_pts=1000, scaling_factor=1.0) staticmethod
Source code in rfsoc_rfdc/waveform_generator.py 
30
31
32
33
34
35
36
@staticmethod
def generate_triangle_wave(repeat_time=1, sample_pts=1000, scaling_factor=1.0):
    t = np.linspace(0, repeat_time, int(
        sample_pts * repeat_time), endpoint=False)
    wave = 2 * np.abs(2 * (t - np.floor(t + 0.5))) - 1
    wave *= int(MyRFdcType.DAC_MAX_SCALE * scaling_factor)
    return wave.astype(MyRFdcType.DATA_PATH_DTYPE)
generate_sawtooth_wave(repeat_time=1, sample_pts=1000, scaling_factor=1.0) staticmethod
Source code in rfsoc_rfdc/waveform_generator.py 
38
39
40
41
42
43
44
@staticmethod
def generate_sawtooth_wave(repeat_time=1, sample_pts=1000, scaling_factor=1.0):
    t = np.linspace(0, repeat_time, int(
        sample_pts * repeat_time), endpoint=False)
    wave = 2 * (t - np.floor(0.5 + t))
    wave *= int(MyRFdcType.DAC_MAX_SCALE * scaling_factor)
    return wave.astype(MyRFdcType.DATA_PATH_DTYPE)
generate_cosine_wave(repeat_time=1, sample_pts=1000, scaling_factor=1.0, raw_np=False) staticmethod
Source code in rfsoc_rfdc/waveform_generator.py 
46
47
48
49
50
51
52
53
54
55
@staticmethod
def generate_cosine_wave(repeat_time=1, sample_pts=1000, scaling_factor=1.0, raw_np=False):
    t = np.linspace(0, repeat_time, int(
        sample_pts * repeat_time), endpoint=False)
    wave = np.cos(2 * np.pi * t)
    if raw_np:
        return wave
    else:
        wave *= int(MyRFdcType.DAC_MAX_SCALE * scaling_factor)
    return wave.astype(MyRFdcType.DATA_PATH_DTYPE)
generate_binary_seq(repeat_time=1, sample_pts=1000, scaling_factor=1.0) staticmethod
Source code in rfsoc_rfdc/waveform_generator.py 
57
58
59
60
61
62
@staticmethod
def generate_binary_seq(repeat_time=1, sample_pts=1000, scaling_factor=1.0):
    num_of_samples = repeat_time * sample_pts
    wave = np.tile([0, 1], num_of_samples // 2 + 1)[:num_of_samples]
    wave *= int(MyRFdcType.DAC_MAX_SCALE * scaling_factor)
    return wave.astype(MyRFdcType.DATA_PATH_DTYPE)
generate_ten_sine(scaling_factor=1.0) staticmethod
Source code in rfsoc_rfdc/waveform_generator.py 
64
65
66
67
68
69
70
71
@staticmethod
def generate_ten_sine(scaling_factor=1.0):
    ten_sine = WaveFormGenerator.generate_sine_wave(
        repeat_time=10, sample_pts=10, scaling_factor=scaling_factor)
    ten_sine = np.append(ten_sine, np.zeros(
        100, dtype=MyRFdcType.DATA_PATH_DTYPE))
    wave = np.append(ten_sine, ten_sine)
    return wave
generate_zadoff_chu_wave(repeat_time=1, sample_pts=1000, scaling_factor=1.0) staticmethod
Source code in rfsoc_rfdc/waveform_generator.py 
 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
@staticmethod
def generate_zadoff_chu_wave(repeat_time=1, sample_pts=1000, scaling_factor=1.0):
    u = 1  # Root index of the the ZC sequence
    seq_length = sample_pts * repeat_time
    q = 0  # Cyclic shift of the sequence

    for el in [u, seq_length, q]:
        if not float(el).is_integer():
            raise ValueError('{} is not an integer'.format(el))
    if u <= 0:
        raise ValueError('u is not stricly positive')
    if u >= seq_length:
        raise ValueError('u is not stricly smaller than seq_length')
    if np.gcd(u, seq_length) != 1:
        raise ValueError(
            'the greatest common denominator of u and seq_length is not 1')

    cf = seq_length % 2
    n = np.arange(seq_length)
    zcseq = np.exp(-1j * np.pi * u * n * (n+cf+2.*q) / seq_length)

    scale_to_max = int(MyRFdcType.DAC_MAX_SCALE * scaling_factor)

    zcseq_real = np.real(zcseq) * scale_to_max
    zcseq_real = zcseq_real.astype(MyRFdcType.DATA_PATH_DTYPE)
    zcseq_imag = np.imag(zcseq) * scale_to_max
    zcseq_imag = zcseq_imag.astype(MyRFdcType.DATA_PATH_DTYPE)

    return zcseq_real, zcseq_imag
generate_no_wave(repeat_time=1, sample_pts=1000, scaling_factor=1.0) staticmethod
Source code in rfsoc_rfdc/waveform_generator.py 
103
104
105
106
@staticmethod
def generate_no_wave(repeat_time=1, sample_pts=1000, scaling_factor=1.0):
    wave = np.zeros(sample_pts * repeat_time)
    return wave.astype(MyRFdcType.DATA_PATH_DTYPE)