#!/usr/bin/env python
#
# Copyright 2012 Free Software Foundation, Inc.
#
# This file is part of GNU Radio
#
# GNU Radio 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.
#
# GNU Radio 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 GNU Radio; see the file COPYING.  If not, write to
# the Free Software Foundation, Inc., 51 Franklin Street,
# Boston, MA 02110-1301, USA.
#

try:
    from grc.python.Platform import Platform
except ImportError:
    from gnuradio.grc.python.Platform import Platform

from optparse import OptionParser
import os, sys

class grcc:
    def __init__(self, grcfile, out_dir):
        self.out_dir = out_dir
        self.platform = Platform()
        data = self.platform.parse_flow_graph(grcfile)

        self.fg = self.platform.get_new_flow_graph()
        self.fg.import_data(data)
        self.fg.validate()

        self.gen = self.platform.get_generator()(self.fg, out_dir)
        self.gen.write()

    def exec_program(self):
        progname = self.fg.get_option('id')
        os.system("{0}/{1}.py".format(self.out_dir, progname))
        
if __name__ == "__main__":
    usage="%prog: [options] filename"
    description = "Compiles a GRC file (.grc) into a GNU Radio Python program. The program is stored in ~/.grc_gnuradio by default, but this location can be changed with the -d option."

    parser = OptionParser(conflict_handler="resolve", usage=usage, description=description)
    parser.add_option("-d", "--directory", type="string", default='{0}/.grc_gnuradio/'.format(os.environ["HOME"]),
                      help="Specify the directory to output the compile program [default=%default]")
    parser.add_option("-e", "--execute", action="store_true", default=False,
                      help="Run the program after compiling [default=%default]")
    (options, args) = parser.parse_args ()

    if(len(args) != 1):
        sys.stderr.write("Please specify a GRC file name to compile.\n")
        sys.exit(1)

    g = grcc(args[0], options.directory+"/")

    if(options.execute):
        g.exec_program()