#!/usr/bin/env python

# -*- coding: utf-8 -*-

# Copyright (C) 2012-2013 Hector Martin "marcan" <hector@marcansoft.com>

#

# 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 2 or version 3.

#

# 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, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

import OpenGL.GL as gl

import ffms

class BackgroundVideo(object):

def __init__(self, song):

self.offset = 0

self.fade_in = 1

self.fade_out = 1

if "video_offset" in song.song:

self.offset = float(song.song["video_offset"])

if "fade_in" in song.song:

self.fade_in = float(song.song["fade_in"])

if "fade_out" in song.song:

self.fade_out = float(song.song["fade_out"])

self.vsource = ffms.VideoSource(song.videofile)

self.vsource.set_output_format([ffms.get_pix_fmt("bgr32")])

self.frameno = 0

self.frame = None

self.timecodes = [i/1000.0 for i in self.vsource.track.timecodes]

self.texid = gl.glGenTextures(1)

frame = self.vsource.get_frame(0)

self.width = frame.ScaledWidth

self.height = frame.ScaledHeight

self.frame = frame.planes[0].copy()

self.sar = self.vsource.properties.SARNum / self.vsource.properties.SARDen

if self.sar == 0:

self.sar = 1

if "video_sar" in song.song:

self.sar = float(song.song["video_sar"])

self.aspect = self.sar * self.width / self.height

def advance(self, time):

if self.frameno >= len(self.timecodes)-1:

return

cur_frame = self.frameno

while time > self.timecodes[self.frameno+1]:

self.frameno += 1

if self.frameno >= len(self.timecodes)-1:

break

if self.frameno != cur_frame:

self.frame = self.vsource.get_frame(self.frameno)

def draw(self, time, display, song_length):

brightness = 1

last_frame = self.timecodes

if time > (song_length - self.fade_out) and self.fade_out:

brightness *= max(0, min(1, (song_length - time) / self.fade_out))

if self.offset < 0:

time += self.offset

if time < self.fade_in and self.fade_in:

brightness *= max(0, min(1, time/self.fade_in))

if self.offset > 0:

time += self.offset

self.advance(time)

data = self.frame.planes[0]

gl.glDisable(gl.GL_BLEND)

gl.glBindTexture(gl.GL_TEXTURE_2D, self.texid)

gl.glTexParameteri(gl.GL_TEXTURE_2D, gl.GL_TEXTURE_MIN_FILTER, gl.GL_LINEAR);

gl.glTexParameteri(gl.GL_TEXTURE_2D, gl.GL_TEXTURE_MAG_FILTER, gl.GL_LINEAR);

gl.glTexParameteri(gl.GL_TEXTURE_2D, gl.GL_TEXTURE_WRAP_S, gl.GL_CLAMP);

gl.glTexParameteri(gl.GL_TEXTURE_2D, gl.GL_TEXTURE_WRAP_T, gl.GL_CLAMP);

gl.glTexImage2D(gl.GL_TEXTURE_2D, 0, 3,

self.width, self.height, 0,

gl.GL_RGBA, gl.GL_UNSIGNED_BYTE, data)

gl.glActiveTexture(gl.GL_TEXTURE0)

gl.glBindTexture(gl.GL_TEXTURE_2D, self.texid)

gl.glEnable(gl.GL_TEXTURE_2D)

gl.glBegin(gl.GL_TRIANGLE_STRIP)

h = self.height / self.width / self.sar

dh = display.height / display.width

offset = (dh-h) / 2

gl.glColor4f(brightness,brightness,brightness,1)

gl.glTexCoord2f(0,1)

gl.glVertex2f(0,offset)

gl.glTexCoord2f(1,1)

gl.glVertex2f(1,offset)

gl.glTexCoord2f(0,0)

gl.glVertex2f(0,offset+h)

gl.glTexCoord2f(1,0)

gl.glVertex2f(1,offset+h)

gl.glEnd()

gl.glDisable(gl.GL_TEXTURE_2D)

def __del__(self):

gl.glDeleteTextures(self.texid)