source: ammosreader/sample_scripts/ammos_viewer.py@ 577ce87

"""I provide a simple bit viewer specialized for AMMOS files."""

import tkinter as tk
from PIL import Image, ImageTk
from random import randbytes

from ammosreader.AmmosAudioReader import AmmosAudioReader

class MainApplication(tk.Frame):
    """I implement a simple bit viewer specialized for AMMOS files."""

    def __init__(self, parent):
        """I create the application window."""
        super().__init__(parent)

        self.parent = parent
        self.file_name = "../sample_data/audio_data_sample.bin"
        self.ammos_container = AmmosAudioReader(self.file_name).read_all_frames_left()
        self.file = open(self.file_name, 'rb')
        self.bytes = self.file.read(-1)
        self.cycle = self.ammos_container.unique_frame_sizes()[0]
        self.cache = 500000000
        self.canvas_size = (800, 600)
        self.offset = 0
        self.buildup()

    def buildup(self):
        """I build up the initial user interface."""
        self.parent.title("File name:" + self.file_name + " Size:" + str(len(self.bytes)))
        self.pack(fill="both", expand=True)
        self.default_button_width = 10
        image_frame = tk.Frame(self)
        image_frame.pack(side=tk.LEFT)
        self.canvas = tk.Canvas(image_frame, bg="#000000", width=self.canvas_size[0], height=self.canvas_size[1])
        self.canvas.pack()
        # display_size = self.canvas_size[0] * self.canvas_size[1]
        # bytes_to_display = randbytes(display_size)
        # if len(self.bytes) < display_size:
        #    bytes_to_display = self.bytes[self.offset:] + bytearray([0xff] * (display_size-len(self.bytes)-self.offset))
        current_bytes = b"".join([self.bytes[each: each+self.canvas_size[0]] for each in range(0, len(self.bytes),
                                                                                               self.cycle)])
        print("Bytes size", len(current_bytes))
        pil_image = Image.frombytes("L", self.canvas_size, current_bytes)
        self.image = ImageTk.PhotoImage(pil_image)
        # pil_image.show()
        button_frame = tk.Frame(self)
        self.canvas.create_image(0, 0, anchor='nw', image=self.image)
        button_frame.pack(side="right")

        tk.Label(button_frame, text='Cycle in bits').pack(side=tk.TOP)
        self.cycle_entry = tk.Entry(button_frame)
        # self.cycle_entry.set()
        self.cycle_entry.pack(side=tk.TOP)

        tk.Label(button_frame, text='Cache in bytes').pack(side=tk.TOP)
        self.cache_entry = tk.Entry(button_frame)
        self.cache_entry.pack(side=tk.TOP)

        tk.Label(button_frame, text='Offset in bits').pack(side=tk.TOP)
        self.offset_entry = tk.Entry(button_frame)
        self.offset_entry.pack(side=tk.TOP)

        tk.Button(
            button_frame,
            text="Load file",
            width=self.default_button_width,
            command=self.load_file).pack(side="top", fill="both", expand=True)

        tk.Button(
            button_frame,
            text="Quit",
            width=self.default_button_width,
            command=self.quit).pack(side="top", fill="both", expand=True)

    def load_file(self):
        pass

    def quit(self):
        self.parent.destroy()


if __name__ == '__main__':
    root = tk.Tk()
    root.geometry("1920x1200")

    app = MainApplication(root)

    root.mainloop()