Merge pull request #3 from ashwini-patil/master

Python files Added

1 files changed, 126 insertions, 0 deletions

diff --git a/gr-input/python/Roots.py b/gr-input/python/Roots.py
new file mode 100644
index 000000000..1057c8d76
--- /dev/null
+++ b/gr-input/python/Roots.py
@@ -0,0 +1,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])
+    
+