#!/usr/bin/env python
##################################################
# Gnuradio Python Flow Graph
# Title: Top Block
# Generated: Tue Aug 23 11:49:21 2011
##################################################

from gnuradio import eng_notation
from gnuradio import gr
from gnuradio import window
from gnuradio.eng_option import eng_option
from gnuradio.gr import firdes
from gnuradio.wxgui import fftsink2
from gnuradio.wxgui import forms
from grc_gnuradio import blks2 as grc_blks2
from grc_gnuradio import wxgui as grc_wxgui
from optparse import OptionParser
import wx
import time #File Recording

class top_block(grc_wxgui.top_block_gui):

	def __init__(self):
		grc_wxgui.top_block_gui.__init__(self, title="Top Block")
		_icon_path = "/home/matlis/.local/share/icons/hicolor/32x32/apps/gnuradio-grc.png"
		self.SetIcon(wx.Icon(_icon_path, wx.BITMAP_TYPE_ANY))

		##################################################
		# Variables
		##################################################
		self.samp_rate = samp_rate = 32000
		self.file_switch = file_switch = 0 #File Recording

		##################################################
		# File Recording Additions: Definitions
		##################################################
                self.recording = gr.file_sink(gr.sizeof_float, "/dev/null")
                self.recording_state = False
                self.prefix = "/home/matlis/"
                self.recording.close()
		##################################################

		##################################################
		# File Recording Additions: Controls
		##################################################
		self._file_switch_chooser = forms.button(
			parent=self.GetWin(),
			value=self.file_switch,
			callback=self.toggle_recording,
			label="Data Acquisition",
			choices=[0,1],
			labels=['Recording Off', 'Recording On'],
		)
		self.Add(self._file_switch_chooser)
		##################################################


		##################################################
		# Blocks
		##################################################
		self.gr_sig_source_x_0 = gr.sig_source_f(samp_rate, gr.GR_COS_WAVE, 1000, 1, 0)
		self.gr_throttle_0 = gr.throttle(gr.sizeof_float*1, samp_rate)
		self.wxgui_fftsink2_0 = fftsink2.fft_sink_f(
			self.GetWin(),
			baseband_freq=0,
			y_per_div=10,
			y_divs=10,
			ref_level=50,
			ref_scale=2.0,
			sample_rate=samp_rate,
			fft_size=1024,
			fft_rate=30,
			average=False,
			avg_alpha=None,
			title="FFT Plot",
			peak_hold=False,
		)
		self.Add(self.wxgui_fftsink2_0.win)

		##################################################
		# Connections
		##################################################
		self.connect((self.gr_sig_source_x_0, 0), (self.gr_throttle_0, 0))
		self.connect((self.gr_throttle_0, 0), (self.wxgui_fftsink2_0, 0))
		self.connect((self.gr_throttle_0, 0), (self.recording, 0)) # File Recording

		##################################################
		# File Recording Additions: Definitions
		##################################################
        def toggle_recording(self, file_switch):
		self.file_switch = file_switch
                # Pick up localtime, for generating filenames
                timestamp = time.localtime()

                # Generate filenames for data file
                filename = "%04d_%02d_%02d_%02d:%02d:%02d" % (timestamp.tm_year, timestamp.tm_mon,
                 timestamp.tm_mday, timestamp.tm_hour, timestamp.tm_min,timestamp.tm_sec)

                # Current recording?  Flip state
                if (self.recording_state == True):
                    self.recording_state = False
                    self.recording.close()
                # Not recording?
                else:
                    self.recording_state = True
                    self.recording.open (self.prefix+filename+".bin")
		##################################################

	def set_samp_rate(self, samp_rate):
		self.samp_rate = samp_rate
		self.gr_sig_source_x_0.set_sampling_freq(self.samp_rate)
		self.wxgui_fftsink2_0.set_sample_rate(self.samp_rate)


if __name__ == '__main__':
	parser = OptionParser(option_class=eng_option, usage="%prog: [options]")
	(options, args) = parser.parse_args()
	tb = top_block()
	tb.Run(True)