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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
#!/usr/bin/env python
#
# Copyright 2015 <+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 time
import numpy
from gnuradio import gr
import sciscipy
class Roots(gr.sync_block):
"""
docstring for block add_python
"""
def __init__(self,order):
self.out = [0,0,0,0,0,0,0,0,0,0]
a = []
self.b = [0,0,0,0,0,0]
self.order=int(order)+1
self.number = int(order)*2
for i in range(0,self.number):
a.append(numpy.float32)
gr.sync_block.__init__(self,
name="Roots",
in_sig=[numpy.float32],
out_sig=a)
def find_roots(self,n):
string1 = "s=%s;"
string2 = "h2="+str(n[3])+"*s^3+"+str(n[2])+"*s^2+"+str(n[1])+"*s+"+str(n[0])+";[E1]=roots(h2);a11=real(E1(1));a12=imag(E1(1));b11 = real(E1(2));b12 = imag(E1(2));c11 = real(E1(3));c12 = imag(E1(3)); d11 = real(E1(4)); d12 = imag(E1(4)); e11 = real(E1(5)); e12 = imag(E1(5));"
string = string1+string2
print "This is string\n", string
sciscipy.eval(string)
try:
self.a11 = sciscipy.read("a11")
print "I am a11", self.a11
except TypeError:
self.a11 = 0
try:
self.a12 = sciscipy.read("a12")
print "I am a12", self.a12
except TypeError:
self.a12 = 0
try:
self.b11 = sciscipy.read("b11")
print "I am b11", self.b11
except TypeError:
self.b11 = 0
try:
self.b12 = sciscipy.read("b12")
except TypeError:
self.b12 = 0
try:
self.c11 = sciscipy.read("c11")
print "I am c11\n", self.c11
except TypeError:
self.c11 = 0
try:
self.c12 = sciscipy.read("c12")
print "I am c12\n", self.c12
except TypeError:
self.c12 = 0
try:
self.d11 = sciscipy.read("d11")
print "I am d11\n", self.d11
except TypeError:
self.d11 = 0
try:
self.d12 = sciscipy.read("d12")
except TypeError:
self.d12 = 0
try:
self.e11 = sciscipy.read("e11")
except TypeError:
self.e11 = 0
try:
self.e12 = sciscipy.read("e12")
except TypeError:
self.e12 = 0
def work(self, input_items, output_items):
# print "I am here\n", output_items
k = self.order
for i in range(0,k):
self.b[i] = input_items[0][i]
self.find_roots(self.b)
self.out[0] = numpy.float32(self.a11)
self.out[1] = numpy.float32(self.b11)
self.out[2] = numpy.float32(self.a12)
self.out[3] = numpy.float32(self.b12)
self.out[4] = numpy.float32(self.c11)
self.out[5] = numpy.float32(self.c12)
self.out[6] = numpy.float32(self.d11)
self.out[7] = numpy.float32(self.d12)
self.out[8] = numpy.float32(self.e11)
self.out[9] = numpy.float32(self.e12)
for i in range (0,self.number):
output_items[i][:]= self.out[i]
print "I am out item",output_items
return len(output_items[0])
|
