uhd

#!/usr/bin/env python

#

# Copyright 2011 Ettus Research LLC

#

# This program 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 of the License, or

# (at your option) any later version.

#

# This program 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 program.  If not, see <http://www.gnu.org/licenses/>.

#

import usrp_n2xx_net_burner #import implementation

try:

    import tkinter, tkinter.filedialog, tkinter.font, tkinter.messagebox

except ImportError:

    import tkFileDialog, tkFont, tkMessageBox

    import Tkinter as tkinter

    tkinter.filedialog = tkFileDialog

    tkinter.font = tkFont

    tkinter.messagebox = tkMessageBox

import os

class BinFileEntry(tkinter.Frame):

    """

    Simple file entry widget for getting the file path of bin files.

    Combines a label, entry, and button with file dialog callback.

    """

    def __init__(self, root, what, def_path=''):

        self._what = what

        tkinter.Frame.__init__(self, root)

        tkinter.Label(self, text=what+":").pack(side=tkinter.LEFT)

        self._entry = tkinter.Entry(self, width=50)

        self._entry.insert(tkinter.END, def_path)

        self._entry.pack(side=tkinter.LEFT)

        tkinter.Button(self, text="...", command=self._button_cb).pack(side=tkinter.LEFT)

    def _button_cb(self):

        filename = tkinter.filedialog.askopenfilename(

            parent=self,

            filetypes=[('bin files', '*.bin'), ('all files', '*.*')],

            title="Select bin file for %s"%self._what,

            initialdir=os.path.dirname(self.get_filename()),

        )

        # open file on your own

        if filename:

            self._entry.delete(0, tkinter.END)

            self._entry.insert(0, filename)

    def get_filename(self):

        return self._entry.get()

class ProgressBar(tkinter.Canvas):

    """

    A simple implementation of a progress bar.

    Draws rectangle that fills from left to right.

    """

    def __init__(self, root, width=500, height=20):

        self._width = width

        self._height = height

        tkinter.Canvas.__init__(self, root, relief="sunken", borderwidth=2, width=self._width-2, height=self._height-2)

        self._last_fill_pixels = None

        self.set(0.0)

    def set(self, frac):

        """

        Update the progress where fraction is between 0.0 and 1.0

        """

        #determine the number of pixels to draw

        fill_pixels = int(round(self._width*frac))

        if fill_pixels == self._last_fill_pixels: return

        self._last_fill_pixels = fill_pixels

        #draw a rectangle representing the progress

        if frac: self.create_rectangle(0, 0, fill_pixels, self._height, fill="#357EC7")

        else:    self.create_rectangle(0, 0, self._width, self._height, fill="#E8E8E8")

class DeviceEntryWidget(tkinter.Frame):

    """

    Simple entry widget for getting the network device name.

    Combines a label, entry, and helpful text box with hints.

    """

    def __init__(self, root, text=''):

        tkinter.Frame.__init__(self, root)

        tkinter.Button(self, text="Rescan for Devices", command=self._reload_cb).pack()

        self._hints = tkinter.Listbox(self)

        self._hints.bind("<<ListboxSelect>>", self._listbox_cb)

        self._reload_cb()

        self._hints.pack(expand=tkinter.YES, fill=tkinter.X)

        frame = tkinter.Frame(self)

        frame.pack()

        tkinter.Label(frame, text="Network Address:").pack(side=tkinter.LEFT)

        self._entry = tkinter.Entry(frame, width=50)

        self._entry.insert(tkinter.END, text)

        self._entry.pack(side=tkinter.LEFT)

    def _reload_cb(self):

        self._hints.delete(0, tkinter.END)

        for hint in usrp_n2xx_net_burner.enumerate_devices():

            self._hints.insert(tkinter.END, hint)

    def _listbox_cb(self, event):

        try:

            sel = self._hints.get(self._hints.curselection()[0])

            self._entry.delete(0, tkinter.END)

            self._entry.insert(0, sel)

        except Exception as e: print(e)

    def get_devname(self):

        return self._entry.get()

class SectionLabel(tkinter.Label):

    """

    Make a text label with bold font.

    """

    def __init__(self, root, text):

        tkinter.Label.__init__(self, root, text=text)

        #set the font bold

        f = tkinter.font.Font(font=self['font'])

        f['weight'] = 'bold'

        self['font'] = f.name

class USRPN2XXNetBurnerApp(tkinter.Frame):

    """

    The top level gui application for the usrp-n2xx network burner.

    Creates entry widgets and button with callback to write images.

    """

    def __init__(self, root, addr, fw, fpga):

        tkinter.Frame.__init__(self, root)

        #pack the file entry widgets

        SectionLabel(self, text="Select Images").pack(pady=5)

        self._fw_img_entry = BinFileEntry(self, "Firmware Image", def_path=fw)

        self._fw_img_entry.pack()

        self._fpga_img_entry = BinFileEntry(self, "FPGA Image", def_path=fpga)

        self._fpga_img_entry.pack()

        #pack the destination entry widget

        SectionLabel(self, text="Select Device").pack(pady=5)

        self._net_dev_entry = DeviceEntryWidget(self, text=addr)

        self._net_dev_entry.pack()

        #the do it button

        SectionLabel(self, text="").pack(pady=5)

        button = tkinter.Button(self, text="Burn Images", command=self._burn)

        self._enable_input = lambda: button.configure(state=tkinter.NORMAL)

        self._disable_input = lambda: button.configure(state=tkinter.DISABLED)

        button.pack()

        #a progress bar to monitor the status

        progress_frame = tkinter.Frame(self)

        progress_frame.pack()

        self._status = tkinter.StringVar()

        tkinter.Label(progress_frame, textvariable=self._status).pack(side=tkinter.LEFT)

        self._pbar = ProgressBar(progress_frame)

        self._pbar.pack(side=tkinter.RIGHT, expand=True)

    def _burn(self):

        #grab strings from the gui

        fw = self._fw_img_entry.get_filename()

        fpga = self._fpga_img_entry.get_filename()

        addr = self._net_dev_entry.get_devname()

        #check input

        if not addr:

            tkinter.messagebox.showerror('Error:', 'No address specified!')

            return

        if not fw and not fpga:

            tkinter.messagebox.showerror('Error:', 'No images specified!')

            return

        if fw and not os.path.exists(fw):

            tkinter.messagebox.showerror('Error:', 'Firmware image not found!')

            return

        if fpga and not os.path.exists(fpga):

            tkinter.messagebox.showerror('Error:', 'FPGA image not found!')

            return

        self._disable_input()

        try:

            #make a new burner object and attempt the burner operation

            burner = usrp_n2xx_net_burner.burner_socket(addr=addr)

            for (image_type, fw_img, fpga_img) in (('FPGA', '', fpga), ('Firmware', fw, '')):

                #setup callbacks that update the gui

                def status_cb(status):

                    self._pbar.set(0.0) #status change, reset the progress

                    self._status.set("%s %s "%(status.title(), image_type))

                    self.update()

                def progress_cb(progress):

                    self._pbar.set(progress)

                    self.update()

                burner.set_callbacks(progress_cb=progress_cb, status_cb=status_cb)

                burner.burn_fw(fw=fw_img, fpga=fpga_img, reset=False, safe=False)

            if tkinter.messagebox.askyesno("Burn was successful!", "Reset the device?"):

                burner.reset_usrp()

        except Exception as e:

            tkinter.messagebox.showerror('Verbose:', 'Error: %s'%str(e))

        #reset the progress bar

        self._pbar.set(0.0)

        self._status.set("")

        self._enable_input()

########################################################################

# main

########################################################################

if __name__=='__main__':

    options = usrp_n2xx_net_burner.get_options()

    root = tkinter.Tk()

    root.title('USRP-N2XX Net Burner')

    USRPN2XXNetBurnerApp(root, addr=options.addr, fw=options.fw, fpga=options.fpga).pack()

    root.mainloop()

    exit()