summaryrefslogtreecommitdiff
path: root/gr-input/python/gr_step_source.py
blob: 904f89292f71e9da89fbc212c633f22959c50065 (plain)
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
#!/usr/bin/python

import gras
import numpy
# Serial is imported in __init__
class step(gras.Block):
	

        def __init__(self):
                gras.Block.__init__(self,
                        name="ser",
                        in_sig=[numpy.float32],
                        out_sig=[numpy.float32])
		self.flag=True

        def set_parameters(self, step_size, offset, width):
		self.step_size = step_size	
		self.width = width
		self.offset = offset

        def work(self, input_items, output_items):
		
                out = output_items[0][0:1]
		input_stream = input_items[0][0]

		if self.flag:
			for i in range(self.width):
				out[:1] = self.offset	
				print "OUT", out
					
				self.produce(0,1) # Produce from port 0 output_items
				self.consume(0,1) # Consume from port 0 input_items

			self.flag = False

		else:	
			out[:1] = self.offset + input_stream*self.step_size
			
			print "OUT", out
				
			self.produce(0,1) # Produce from port 0 output_items
			self.consume(0,1) # Consume from port 0 input_items