You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
3.3 KiB
87 lines
3.3 KiB
#!/usr/bin/env python |
|
# -*- coding: utf-8 -*- |
|
# |
|
# Copyright 2017 <+YOU OR YOUR COMPANY+>. |
|
# |
|
# This is free software; you can redistribute it and/or modify |
|
# it under the terms of the GNU General Public License as published by |
|
# the Free Software Foundation; either version 3, or (at your option) |
|
# any later version. |
|
# |
|
# This software 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 General Public License for more details. |
|
# |
|
# You should have received a copy of the GNU General Public License |
|
# along with this software; see the file COPYING. If not, write to |
|
# the Free Software Foundation, Inc., 51 Franklin Street, |
|
# Boston, MA 02110-1301, USA. |
|
# |
|
|
|
import numpy as np |
|
from gnuradio import gr |
|
import pmt,time,functools |
|
class RingBuffer(): |
|
"A 1D ring buffer using numpy arrays" |
|
def __init__(self, length): |
|
self.data = np.zeros(length, dtype='f') |
|
self.index = 0 |
|
def extend(self, x): |
|
"adds array x to ring buffer" |
|
x_index = (self.index + np.arange(x.size)) % self.data.size |
|
self.data[x_index] = x |
|
self.index = x_index[-1] + 1 |
|
def get(self): |
|
"Returns the first-in-first-out data in the ring buffer" |
|
idx = (self.index + np.arange(self.data.size)) %self.data.size |
|
return self.data[idx] |
|
class stream_selector(gr.sync_block): |
|
""" |
|
docstring for block stream_selector |
|
""" |
|
def __init__(self): |
|
gr.sync_block.__init__(self, |
|
name="stream_selector", |
|
in_sig=[np.float32,np.float32], |
|
out_sig=None |
|
) |
|
#self.repeat_time=repeat_time |
|
self.repeat_time=1 |
|
moving_avg_len=50 |
|
self.message_port_register_out(pmt.intern('rds_out')) |
|
self.message_port_register_in(pmt.intern('rds_arg')) |
|
self.set_msg_handler(pmt.intern('rds_arg'), functools.partial(self.handle_msg, port="arg")) |
|
self.message_port_register_in(pmt.intern('rds_re')) |
|
self.set_msg_handler(pmt.intern('rds_re'), functools.partial(self.handle_msg, port="re")) |
|
|
|
#self.port_of_recent_messages=RingBuffer(10)#change to more fitting ringbuffer for ints |
|
self.port_of_recent_messages=0#re as default |
|
self.real_abs=RingBuffer(moving_avg_len) |
|
self.arg_diff_abs=RingBuffer(moving_avg_len) |
|
self.msg_time=time.time() |
|
def handle_msg(self,msg,port): |
|
msg_py=pmt.to_python(msg) |
|
if port=="re": |
|
#self.port_of_recent_messages.extend(np.array([0],dtype='f')) |
|
self.port_of_recent_messages=(self.port_of_recent_messages*19+0)/20 |
|
if port=="arg": |
|
self.port_of_recent_messages=(self.port_of_recent_messages*19+100)/20 |
|
#self.port_of_recent_messages.extend(np.array([100],dtype='f')) |
|
#print(np.average(self.port_of_recent_messages.get())) |
|
print(self.port_of_recent_messages) |
|
print("msg: %s, port:%s"%(msg_py,port)) |
|
|
|
def work(self, input_items, output_items): |
|
self.real_abs.extend(np.absolute(input_items[0])) |
|
self.arg_diff_abs.extend(np.absolute(input_items[1])) |
|
|
|
if(time.time()-self.msg_time > self.repeat_time): |
|
self.msg_time=time.time() |
|
#print("real: ") |
|
##print(real_abs) |
|
#print(np.average(self.real_abs.get())) |
|
#print("arg: ") |
|
##print(arg_diff_abs) |
|
#print(np.average(self.arg_diff_abs.get())) |
|
return len(input_items[0])
|
|
|