r_e_c_u_r/video_centre.py

import time

try:
    from omxdriver import OMXDriver  # <== for deving only

    has_omx = True
except ImportError:
    has_omx = False
from Tkinter import Tk, Canvas
import data_centre

logger = data_centre.setup_logging()


# layer = 0


class video_driver(object):
    def __init__(self, widget=None):
        print 'has omx :{}'.format(has_omx)
        self.widget = widget
        self.delay = 5
        logger.info('the has_omx flag is {}'.format(has_omx))
        if has_omx:
            self.last_player = video_player(self.widget, 'a')
            self.current_player = video_player(self.widget, 'b')
            self.next_player = video_player(self.widget, 'c')
            self.manual_next = False

            self.widget.after(self.delay, self.begin_playing)

    def begin_playing(self):
        # TODO: the first clip will be a demo
        # first_context = '/home/pi/pp_home/media/01_trashpalaceintro.mp4'

        self.current_player.load_content()

        self.wait_for_first_load()

    def wait_for_first_load(self):
        if self.current_player.is_loaded():
            self.play_video()
        else:
            # load player states
            self.widget.after(self.delay, self.wait_for_first_load)

    def switch_players(self):
        self.temp_player = self.last_player
        self.last_player = self.current_player
        logger.info('switch: last_player is {}'.format(self.last_player.name))
        self.current_player = self.next_player
        logger.info('switch: current_player is {}'.format(
            self.current_player.name))
        self.next_player = self.temp_player
        logger.info('switch: next_player is {}'.format(self.next_player.name))
        self.last_player.exit()

    def play_video(self):
        logger.info('{} is about to play'.format(self.current_player.name))
        self.current_player.play_content()
        # self.last_player.exit()

        # next_context = '/home/pi/pp_home/media/samplerloop3s.mp4'
        self.next_player.load_content()

        self.wait_for_next_cycle()

    def wait_for_next_cycle(self):

        if (self.current_player.is_finished() or self.manual_next):
            logger.info('{} is finished'.format(self.current_player.name))
            self.manual_next = False
            if self.next_player.is_loaded():
                logger.info('{} is loaded on switchover'.format(
                    self.next_player.name))
                self.switch_players()
                self.play_video()
            else:
                logger.info('{} is not loaded yet!'.format(
                    self.next_player.name))
                self.current_player.pause_content()
                self.wait_for_next_load()
        else:
            self.widget.after(self.delay, self.wait_for_next_cycle)

    def wait_for_next_load(self):
        if (self.next_player.is_loaded()):
            self.switch_players()
            self.play_video()
        else:
            self.widget.after(self.delay, self.wait_for_next_load)

    def get_info_for_video_display(self):
        if has_omx:
            return self.current_player.bank_number, self.current_player.status, self.next_player.bank_number, \
                self.next_player.status, self.current_player.get_position(), self.current_player.length
        else:
            return 0, 'test', 1, 'test', 0, 10

    def exit_all_players(self):
        self.next_player.exit()
        self.current_player.exit()


class video_player(object):
    def __init__(self, widget, name):
        self.widget = widget
        self.name = name
        self.video_name = ''
        self.status = 'UNASSIGNED'
        self.bank_number = '-'
        self.position = 0
        self.start = ''
        self.end = ''
        self.length = 10
        self.location = ''

        self.omx = OMXDriver(self.widget, '')

    def is_loaded(self):
        return self.omx.start_play_signal

    def is_finished(self):
        return self.omx.end_play_signal

    def get_position(self):
        if self.is_loaded():
            return self.omx.video_position/1000000
        else:
            return 0

    def play_content(self):
        self.status = 'PLAYING'
        logger.info('{} is playing now'.format(self.name))
        self.omx.pause_before_play_required = 'no'
        self.omx.show(True, 0)

    def load_content(self):
        self.status = 'LOADING'
        self.get_context_for_this_player()
        logger.info('{} is loading now {}'.format(
            self.name,self.location ))
        self.omx.load(self.location, 'after-first-frame',
                      '--win 250,350,800,800 --no-osd --display 5', '')

    def get_context_for_this_player(self):
        next_context = data_centre.get_next_context()
        self.location = next_context['location']
        self.length = next_context['length']
        self.start = next_context['start']
        self.end = next_context['end']
        self.video_name = next_context['name']
        self.bank_number = next_context['bank_number']

    def reload_content(self):
        self.status = 'RELOADING'
        if self.is_loaded():
            self.exit()
        else:
            self.widget.after(50, self.reload_content)
            print("trying to reload")
        self.load_content()

    # layer = layer + 1

    def set_to_default(self):
        ##not used
        self.omx.kill()
        self.omx = OMXDriver(self.widget, '')

    def exit(self):
        if (self.omx.omx_loaded is not None):
            logger.info('{} is exiting omx'.format(self.name))
            self.omx.stop()
            self.omx = OMXDriver(self.widget, '')

    def toggle_pause(self):
        is_paused = self.omx.omxplayer_is_paused()
        print is_paused
        if is_paused == 'RUNNING':
            if self.omx.send_pause():
                self.status = 'PAUSED'
                return
        elif is_paused == 'PAUSED':
            if self.omx.send_unpause():
                self.status = 'PLAYING'
                return
        print 'failed to toggle pause (this needs to be in message)'


# tk = Tk()

# canvas = Canvas(tk,width=500,height=400, bd=0, highlightthickness=0)
# canvas.pack()

# driver = video_driver(canvas)

# while True:
#	tk.update()