path: root/voctogui/lib/ui.py

#!/usr/bin/python3
import gi, logging
from gi.repository import Gtk, Gst, Gdk, GLib

from lib.config import Config
from lib.uibuilder import UiBuilder

from lib.videodisplay import VideoDisplay
from lib.audioleveldisplay import AudioLevelDisplay
from lib.warningoverlay import VideoWarningOverlay

from lib.videopreviews import VideoPreviewsController
from lib.audioselector import AudioSelectorController

from lib.toolbar.composition import CompositionToolbarController
from lib.toolbar.streamblank import StreamblankToolbarController
# from lib.toolbar.specialfunctions import SpecialFunctionsToolbarController

class Ui(UiBuilder):
    def __init__(self, uifile):
        self.log = logging.getLogger('Ui')
        super().__init__(uifile)

    def setup(self):
        self.log.info('Initializing Ui')

        # Aquire the Main-Window from the UI-File
        self.win = self.get_check_widget('window')

        # Connect Close-Handler
        self.win.connect('delete-event', Gtk.main_quit)


        # Create Audio-Level Display
        drawing_area = self.find_widget_recursive(self.win, 'audiolevel_main')
        self.audio_level_display = AudioLevelDisplay(drawing_area)


        # Create Main-Video Overlay Controller
        self.video_warning_overlay = VideoWarningOverlay()


        # Create Main-Video Display
        drawing_area = self.find_widget_recursive(self.win, 'video_main')
        self.main_video_display = VideoDisplay(drawing_area,
            port=11000,
            play_audio=True,
            draw_callback=self.video_warning_overlay.draw_callback,
            level_callback=self.audio_level_display.level_callback)


        # Setup Preview Controller
        drawing_area = self.find_widget_recursive(self.win, 'box_left')
        self.video_previews_controller = VideoPreviewsController(drawing_area,
            win=self.win,
            uibuilder=self)

        drawing_area = self.find_widget_recursive(self.win, 'combo_audio')
        self.audio_selector_controller = AudioSelectorController(drawing_area,
            win=self.win,
            uibuilder=self)


        # Setup Toolbar Controllers
        toolbar = self.find_widget_recursive(self.win, 'toolbar')
        self.composition_toolbar_controller = CompositionToolbarController(toolbar,
            win=self.win,
            uibuilder=self)

        self.streamblank_toolbar_controller = StreamblankToolbarController(toolbar,
            win=self.win,
            uibuilder=self,
            warning_overlay=self.video_warning_overlay)

        # self.special_functions_toolbar_controller = SpecialFunctionsToolbarController(toolbar,
        #   video_display=self.main_video_display)


    def show(self):
        self.log.info('Showing Main-Window')
        self.win.show_all()