Python Block with Vectors

From GNU Radio
Jump to navigation Jump to search
Beginner Tutorials

Introducing GNU Radio

  1. What is GNU Radio?
  2. Installing GNU Radio
  3. Your First Flowgraph

Flowgraph Fundamentals

  1. Python Variables in GRC
  2. Variables in Flowgraphs
  3. Runtime Updating Variables
  4. Signal Data Types
  5. Converting Data Types
  6. Packing Bits
  7. Streams and Vectors
  8. Hier Blocks and Parameters

Creating and Modifying Python Blocks

  1. Creating Your First Block
  2. Python Block With Vectors
  3. Python Block Message Passing
  4. Python Block Tags

DSP Blocks

  1. Low Pass Filter Example
  2. Designing Filter Taps
  3. Sample Rate Change

This tutorial describes how the Python Embedded Block can be modified to accept vector inputs and outputs, and how the input_items vector indexing is different between vectors and streams.

The previous tutorial, Creating Your First Block, demonstrates how to create a Python block using the Embedded Python Block. The next tutorial, Python Block Message Passing describes how to send and receive messages using the Embedded Python Block.

Starting the Flowgraph[edit]

This tutorial uses vectors. Please complete the Streams and Vectors tutorial before moving on.

Add the blocks to the flowgraph:

  • Signal Source
  • Throttle
  • Stream to Vector
  • Embedded Python Block
  • Vector to Stream
  • QT GUI Time Sink
  • Virtual Sink
  • Virtual Source
  • Variable

Change the block properties:

  • Signal Source
    • Output Type: float
    • Frequency: 100
  • Variable
    • Id: vectorLength
    • Value: 16
  • Stream to Vector, Num Items: vectorLength
  • Vector to Stream, Num Items: vectorLength
  • Virtual Sink, Stream Id: sinusoid
  • Virtual Source, Stream Id: sinusoid
  • QT GUI Time Sink
    • Autoscale: Yes
    • Number of Inputs: 2

Connect the blocks according to the following flowgraph:

PythonVectorStartingFlowgraph.png

Accepting Vector Inputs and Outputs[edit]

The Embedded Python Block needs to be modified to:

  • accept vector inputs
  • produce vector outputs
  • change the data types to float

With respect to the Streams and Vectors tutorial, from a higher level point of view, Streams are typically just a special case of Vectors, ones that have just one data element in parallel instead of multiple. This is why we can just change the in_sig and out_sig parameters to use vectors rather than streams.

Since the case of streams is so common it has its own simple syntax, whereas the syntax for vectors is just a bit more complex. In particular, when using vectors, we must specify the vector's length along with the data type of the elements in the vectors. We do so using a tuple in Python.

Double-click on the block to edit the source code.

Change example_param in the function definition to vectorSize:

def __init__(self, vectorSize=16):

Change the name:

name='Max Hold Block',

Define the input and output vector signals using a tuple:

in_sig=[(np.float32,vectorSize)],
out_sig=[(np.float32,vectorSize)]

Remove the self.example_param = example_param line.

Remove the multiplication by self.example_param:

output_items[0][:] = input_items[0]

The code should now look like the following:

PythonVectorDefineBlock.png


Save the code (CTRL + S). Connect the Max Hold Block to the rest of the flowgraph:

PythonVectorConnectMaxHold.png

Warning for Vector Length Mismatches[edit]

The Embedded Python Block has one difference that other Out of Tree Modules do not have. Before a flowgraph can be run GRC checks to ensure all of the connected data types and vector sizes match. During this process the default value of vectorSize in the __init()__ function,

def __init__(self, vectorSize=16):

is used to define the size of the vectors for the input and output,

in_sig=[(np.float32,vectorSize)],
out_sig=[(np.float32,vectorSize)]

when determining if the flowgraph is correct. In this case vectorSize=16. GRC assumes the input and output ports are vectors with length 16, even if a different parameter is passed in through the block properties!. The following image shows how a vector size of 128 is passed in as a parameter which GRC does not catch as an error, but the flowgraph will crash once it's run:

MismatchVectorSizeOption1.png


Traceback (most recent call last):
  File "/home/username/vectorinput.py", line 250, in <module>
    main()
  File "/home/username/vectorinput.py", line 226, in main
    tb = top_block_cls()
  File "/home/username/vectorinput.py", line 188, in __init__
    self.connect((self.blocks_stream_to_vector_0, 0), (self.epy_block_0, 0))
  File "/usr/lib/python3/dist-packages/gnuradio/gr/hier_block2.py", line 48, in wrapped
    func(self, src, src_port, dst, dst_port)
  File "/usr/lib/python3/dist-packages/gnuradio/gr/hier_block2.py", line 111, in connect
    self.primitive_connect(*args)
  File "/usr/lib/python3/dist-packages/gnuradio/gr/runtime_swig.py", line 4531, in primitive_connect
    return _runtime_swig.top_block_sptr_primitive_connect(self, *args)
RuntimeError: itemsize mismatch: stream_to_vector0:0 using 64, Embedded Python Block0:0 using 512

Alternatively, GRC will show an error if the default parameter does not match the vector size of the other blocks. In this case, the default vector length in the code is 128 but the passed-in parameter is 16:

MismatchVectorSizeOption2.png

Indexing Streams[edit]

For a stream, inputs and outputs can be indexed using both the port number and the sample index.

Indexing based on the port number returns all of the input samples for a specific port. For example,

input_items[0]

returns all input samples on port 0.

The following line returns the 4th input sample on port 0:

input_items[0][3]

The indexing for streams is generalized to:

input_items[portIndex][sampleIndex]
output_items[portIndex][sampleIndex]

The image shows how to visualize indexing streams:

IndexingStreams.png

Indexing Vectors[edit]

The input input_items and output output_items include an extra dimension when using vectors.

Vectors add an additional dimension, represented as vectorIndex below. The input_items and output_items are now three-dimensional arrays:

input_items[portIndex][vectorIndex][sampleIndex]
output_items[portIndex][vectorIndex][sampleIndex]

Indexing based on the portIndex returns a two-dimensional array of all vectors and samples, for example:

input_items[portIndex]
output_items[portIndex]

Indexing based on portIndex and vectorIndex returns an single-dimensional array of samples, for example:

input_items[portIndex][vectorIndex]
input_items[portIndex][vectorIndex]

Indexing based on portIndex, vectorIndex and sampleIndex returns a single sample.

input_items[portIndex][vectorIndex][sampleIndex]
input_items[portIndex][vectorIndex][sampleIndex]

A visual example vector indexing is given below:

VectorIndexing.png

Creating Max Hold Function[edit]

The work() function is modified to include the max hold function. Add a loop over all of the vectors in input_items[0]:

for vectorIndex in range(len(input_items[0])):

Calculate the max value of the vector:

maxValue = np.max(input_items[0][vectorIndex])

Loop over each of the input samples:

for sampleIndex in range(len(input_items[0][vectorIndex])):

Assign each output sample maxValue:

output_items[0][vectorIndex][sampleIndex] = maxValue

The code should look like the following:

MaxHoldWorkFunction.png


Save the code (CTRL + S). Run the flowgraph. The output will shows a sinusoid and a sinusoid with a max-hold applied every 16 samples:

MaxHoldOutput.png

Multiple Vector Ports[edit]

The Max Hold Block is modified to add a second vector input and output port.

Add the following blocks to the workspace:

  • Noise Source
  • Stream to Vector
  • Vector to Stream
  • Virtual Sink
  • Virtual Source
  • QT GUI Time Sink

Change the following block properties:

  • Noise Source, Output Type: float
  • Stream to Vector, Num Items: vectorLength
  • Vector to Stream, Num Items: vectorLength
  • Virtual Sink, Stream Id: noise
  • Virtual Source, Stream Id: noise
  • QT GUI Time Sink
    • Autoscale: Yes
    • Number of Inputs: 2

Connect the blocks:

PythonVectorNoiseSource.png


Edit the code for the Max Hold Block. Add a second vector input and output:

in_sig=[(np.float32,vectorSize),(np.float32,vectorSize)],
out_sig=[(np.float32,vectorSize),(np.float32,vectorSize)]

PythonBlockSecondVector.png

The work() function is modified to perform the max hold function over both input ports.

Include an outer loop over all input ports:

for portIndex in range(len(input_items)):

Change all indexing [0] to [portIndex]:

for portIndex in range(len(input_items)):
    for vectorIndex in range(len(input_items[portIndex])):
        maxValue = np.max(input_items[portIndex][vectorIndex])
            for sampleIndex in range(len(input_items[portIndex][vectorIndex])):
                output_items[portIndex][vectorIndex][sampleIndex] = maxValue

The code should now look like the following:

PythonVectorFinalCode.png


Save the code and connect the blocks:

PythonVectorFinalFlowgraph.png


Run the flowgraph. Two max-hold outputs will now be generated, one for the noise source and one for the sinusoid:

TwoMaxHoldOutputs.png


The next tutorial, Python Block Message Passing describes how to send and receive messages using the Embedded Python Block.