diff options
Diffstat (limited to 'gr-input/python/Transfer_Function.py')
-rw-r--r-- | gr-input/python/Transfer_Function.py | 100 |
1 files changed, 100 insertions, 0 deletions
diff --git a/gr-input/python/Transfer_Function.py b/gr-input/python/Transfer_Function.py new file mode 100644 index 000000000..d53c55e48 --- /dev/null +++ b/gr-input/python/Transfer_Function.py @@ -0,0 +1,100 @@ +#!/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 Transfer_Function(gr.sync_block): + """ + docstring for block add_python + """ + def __init__(self,order,itype,tstart,tstop,tstep): + a = [] + self.b = [0,0,0,0,0,0] + self.c = [0,0,0,0,0,0] + self.i = 0 + self.order=int(order)+1 + self.itype = itype + self.tstart = tstart + self.tstop = tstop + self.tstep = tstep + gr.sync_block.__init__(self, + name="tf_rises", + in_sig=[numpy.float32,numpy.float32], + out_sig=[numpy.float32]) + + #print "I am over slept" + #print len(self.ret_array) + + def find_resp(self,b,c): + if (self.itype == 11): + typo = "'imp'" + elif (self.itype == 12): + typo = "'step'" + else: + typo = "t" +# print "self b",self.b + string1 = "s=%s; h=syslin('c'," + string2 = str(b[4])+"*s^4+"+str(b[3])+"*s^3+"+str(b[2])+"*s^2+"+str(b[1])+"*s+"+str(b[0])+"," + string3 = str(c[4])+"*s^4+"+str(c[3])+"*s^3+"+str(c[2])+"*s^2+"+str(c[1])+"*s+"+str(c[0])+");" + string4 = "t="+str(self.tstart)+":"+str(self.tstep)+":"+str(self.tstop)+";" + string5 = "deff('u=input(t)','u=50');" + string6 = "r=tf2ss(h); a=csim("+typo+",t,r);" + string = string1+string2+string3+string4+string5+string6 + print "I am strin g",string + sciscipy.eval(string) + self.a = sciscipy.read("a") + print "value of a\n",self.a + + + + + + def work(self, input_items, output_items): + + print "input value:\n",input_items + for i in range(0,5): + self.b[i] = input_items[0][i] + # print "I am b[i]", b[i] + #print "I am BBBBB", self.b + + for j in range(0,5): + self.c[j] = input_items[1][j] + #print "I am c[j]", c[j] + #print "I am CCCCC", self.c + self.find_resp(self.b,self.c) + + + out = output_items[0] + self.i+=1 + print "I am self i\n",self.i + if self.i >= len(self.a): + self.i = 0 + print "len of out\n",len(out) + #time.sleep(0.125) + out[:] = numpy.float32(self.a[self.i]) + #print + print "i am out\n",out + + return len(output_items[0]) + + |