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.
94 lines
3.8 KiB
94 lines
3.8 KiB
#!/usr/bin/env python3 |
|
# -*- 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 code |
|
|
|
class vector_cutter(gr.sync_block): |
|
""" |
|
docstring for block vector_cutter |
|
""" |
|
def __init__(self, insize=2048,outsize=1024,cutpoint=512,pad_out=False,zero_len=0): |
|
if pad_out: |
|
gr.sync_block.__init__(self, |
|
name="vector_cutter", |
|
in_sig=[(np.complex64,insize)], |
|
out_sig=[(np.complex64,outsize),(np.complex64,insize)]) |
|
else: |
|
gr.sync_block.__init__(self, |
|
name="vector_cutter", |
|
in_sig=[(np.complex64,insize)], |
|
out_sig=[(np.complex64,outsize)]) |
|
self.cutpoint=0 |
|
self.set_cutpoint(cutpoint) |
|
self.insize=insize |
|
self.outsize=outsize |
|
#print(pad_out) |
|
self.pad_out=pad_out |
|
self.zero_len=zero_len |
|
self.mask=np.concatenate((np.zeros(zero_len),np.ones(outsize-zero_len*2),np.zeros(zero_len))) |
|
def set_zero_len(self,zero_len=None): |
|
if zero_len is not None: |
|
if isinstance(zero_len, float) or isinstance(zero_len, int): |
|
self.zero_len=zero_len |
|
else: |
|
self.zero_len = int(zero_len) |
|
self.mask=np.concatenate((np.zeros(self.zero_len),np.ones(self.outsize-self.zero_len*2),np.zeros(self.zero_len))) |
|
def set_cutpoint(self, cutpoint=None): |
|
#print("cutpoint set to %i"%cutpoint) |
|
if cutpoint is not None and cutpoint>=0: |
|
self.cutpoint = int(cutpoint) |
|
#if isinstance(cutpoint, float) or isinstance(cutpoint, int): |
|
#self.cutpoint=cutpoint |
|
#else: |
|
#self.cutpoint = int(cutpoint) |
|
def work(self, input_items, output_items): |
|
in0 = input_items[0] |
|
out = output_items[0] |
|
if self.pad_out: |
|
out_padded = output_items[1] |
|
# <+signal processing here+> |
|
#out[:] = in0[512:1536] |
|
cutpoint=self.cutpoint |
|
attenuation=1e-2#40db (power) |
|
for i,in_vec in enumerate(in0): |
|
if 0<=cutpoint<self.insize-self.outsize: |
|
out_cut=in_vec[cutpoint:cutpoint+self.outsize] |
|
if self.pad_out: |
|
out_padded[i]=in_vec*np.concatenate((attenuation*np.ones(cutpoint),np.ones(self.outsize),attenuation*np.ones(self.insize-self.outsize-cutpoint))) |
|
elif cutpoint <=self.insize: |
|
out_cut=np.append(in_vec[cutpoint:self.insize],in_vec[0:self.outsize-self.insize+cutpoint]) |
|
if self.pad_out: |
|
out_padded[i]=in_vec*np.concatenate((np.ones(cutpoint+self.outsize-self.insize),attenuation*np.ones(self.insize-self.outsize),np.ones(self.insize-cutpoint))) |
|
else: |
|
out_cut=in_vec[0:self.outsize]#dummy values |
|
#out[i]=np.append(out_cut[self.outsize/2:self.outsize],out_cut[0:self.outsize/2]) |
|
if self.zero_len==0: |
|
out[i]=out_cut |
|
else: |
|
out[i]=out_cut*self.mask |
|
#out = in0[512:1536] |
|
#code.interact(local=locals()) |
|
#out[0] = in0[0][512:1536] |
|
#out[1] = in0[1][512:1536] |
|
return len(output_items[0]) |
|
|
|
|