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
|
#
# Copyright (c) 2024 Analog Devices Inc.
#
# This file is part of libm2k
# (see http://www.github.com/analogdevicesinc/libm2k).
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 2.1 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
from enum import Enum
import numpy as np
import math
# create multiple signal shapes and reference shapes
# signals that will be sent to output buffer
def shape_gen(n, amplitude: float = 1.0, offset: float = 0.0):
# Generates different signal shapes that will be sent to the output
# Arguments:
# n -- Number of samples in the output buffer
# Returns:
# shape -- list that contains arrays corresponding to each signal shape generated
shape = [[]]
# generate sine wave
sine = amplitude*(np.sin(np.linspace(-np.pi, np.pi, n))) + offset
# generate square wave
square = amplitude * np.append(np.linspace(-1, -1, int(n / 2)), np.linspace(1, 1, int(n / 2))) + offset
# generate triangle
triangle = amplitude * np.append(np.linspace(-1, 1, int(n / 2)), np.linspace(1, -1, int(n / 2))) + offset
# generate rising ramp
rising_ramp = amplitude * np.linspace(-1, 1, n) + offset
# generate falling ramp
falling_ramp = amplitude * np.linspace(1, -1, n) + offset
# shape and reference shape buffers
shape = [sine, square, triangle, rising_ramp, falling_ramp]
return shape
def ref_shape_gen(n):
# Generates reference shapes which will be compared with the input collected through ain
# Arguments:
# n -- Number of samples in the reference buffer
# Returns:
# ref_shape -- list that contains arrays corresponding to each reference shape generated
ref_shape = [[]]
ref_sine = np.sin(np.linspace(-np.pi, np.pi, n))
ref_square = np.append(np.linspace(-1, -1, int(n / 2)), np.linspace(1, 1, math.ceil(n / 2)))
ref_triangle = np.append(np.linspace(-1, 1, int(n / 2)), np.linspace(1, -1, math.ceil(n / 2)))
ref_rising_ramp = np.linspace(-1, 1, n)
ref_falling_ramp = np.linspace(1, -1, n)
ref_shape = [ref_sine, ref_square, ref_triangle, ref_rising_ramp, ref_falling_ramp]
return ref_shape
def shape_name():
# Returns:
# shape_name-- Array that holds the names of the generated signal shapes in string form
shape_name = ['Sine', 'Square', 'Triangle', 'Rising_ramp', 'Falling_ramp']
return shape_name
class Shape(Enum):
SINE = 0
SQUARE = 1
TRIANGLE = 2
RISING_RAMP = 3
FALLING_RAMP = 4
|
