#!/usr/bin/env python # -*- coding: utf-8 -*- # # libavg - Media Playback Engine. # Copyright (C) 2003-2008 Ulrich von Zadow # # This library is free software; you can redistribute it and/or # modify it under the terms of the GNU Lesser General Public # License as published by the Free Software Foundation; either # version 2 of the License, or (at your option) any later version. # # This library 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 # Lesser General Public License for more details. # # You should have received a copy of the GNU Lesser General Public # License along with this library; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA # # Current versions can be found at www.libavg.de # import sys from libavg import avg, AVGApp from libavg.ui import button #from libavg import parsecamargs GUI_SIZE=(300, 200) g_Player = avg.Player.get() class Slider(avg.DivNode): def __init__(self, width, min, max, onChange, **kwargs): avg.DivNode.__init__(self, **kwargs) self.__onChange = onChange self.size = (width, 20) self.__min = min self.__max = max self.__val = min avg.LineNode(pos1=(7,14), pos2=(width-7,14), color="FFFFFF", strokewidth=2, parent=self) self.__slider = avg.DivNode(pos=(0,0), size=(14,20), parent=self) avg.PolygonNode(pos=((1,0), (13,0), (7,18)), fillopacity=1, fillcolor="FFFFFF", color="808080", parent=self.__slider) self.__slider.setEventHandler(avg.CURSORDOWN, avg.MOUSE, self.__onSliderDown) self.setEventHandler(avg.CURSORDOWN, avg.MOUSE, self.__onBarDown) self.__isDragging = False def getVal(self): return self.__val def setVal(self, val): self.__val = val self.__positionSlider() val = property(getVal, setVal) def __onSliderDown(self, event): self.__slider.setEventCapture() self.__slider.setEventHandler(avg.CURSORMOTION, avg.MOUSE, self.__onSliderMove) self.__slider.setEventHandler(avg.CURSORUP, avg.MOUSE, self.__onSliderUp) self.__sliderDownPos = event.pos self.__isDragging = True self.__dragStartVal = self.__val def __onSliderMove(self, event): numPixelsMoved = float(event.pos.x-self.__sliderDownPos.x) self.__val = (self.__dragStartVal+numPixelsMoved/(self.size.x-14) *(self.__max-self.__min)) self.__positionSlider() def __onSliderUp(self, event): self.__onSliderMove(event) self.__slider.releaseEventCapture() self.__slider.setEventHandler(avg.CURSORMOTION, avg.MOUSE, None) self.__slider.setEventHandler(avg.CURSORUP, avg.MOUSE, None) self.__isDragging = False def __onBarDown(self, event): if not(self.__isDragging): localPos = self.getRelPos(event.pos) ratio = (localPos.x-7)/(self.size.x-14) self.__val = self.__min+ratio*(self.__max-self.__min) print localPos, ", ", ratio, ", ", self.__val self.__positionSlider() def __positionSlider(self): if self.__val < self.__min: self.__val = self.__min elif self.__val > self.__max: self.__val = self.__max ratio = ((self.__val-self.__min)/(self.__max-self.__min)) self.__slider.pos = (ratio*(self.size.x-14), 0) self.__onChange() class FXSlider(avg.DivNode): def __init__(self, row, min, max, fxNode, fxAttrName, caption, isInt, **kwargs): avg.DivNode.__init__(self, **kwargs) avg.RectNode(pos=(0,8), size=(280,38), color="808080", strokewidth=2, parent=self) textBgRect = avg.RectNode(pos=(8,2), fillcolor="000000", fillopacity=1, strokewidth=0, parent=self) caption = avg.WordsNode(pos=(10,0), text=caption, parent=self) textBgRect.size = caption.getMediaSize()+(4, 2) self.__words = avg.WordsNode(pos=(240,23), parent=self) self.__slider = Slider(220, min, max, self.__onSliderMove, pos=(15,20), parent=self) self.pos = (0, row*46) self.__fxNode = fxNode self.__fxAttrName = fxAttrName self.__caption = caption self.__isInt = isInt self.__slider.val = getattr(self.__fxNode, fxAttrName) def __onSliderMove(self): if self.__isInt: setattr(self.__fxNode, self.__fxAttrName, int(self.__slider.val)) self.__words.text = "%i"%self.__slider.val else: setattr(self.__fxNode, self.__fxAttrName, self.__slider.val) self.__words.text = "%.2f"%self.__slider.val class TextButton(button.Button): def __init__(self, text, **kwargs): size = kwargs["size"] upNode = avg.DivNode() avg.RectNode(size=size, fillcolor="FFFFFF", fillopacity=1, color="FFFFFF", parent=upNode) avg.WordsNode(pos=(4,3), text=text, color="000000", parent=upNode) downNode = avg.DivNode() avg.RectNode(size=size, fillcolor="000000", fillopacity=1, color="FFFFFF", parent=downNode) avg.WordsNode(pos=(4,3), text=text, color="FFFFFF", parent=downNode) kwargs["upNode"] = upNode kwargs["downNode"] = downNode button.Button.__init__(self, **kwargs) def colorToString(colorTuple): s = "%02X%02X%02X"%colorTuple[:-1] return s class Chromakey(AVGApp): def init(self): avg.RectNode(size=(800,534), fillcolor="FF0000", fillopacity=1, strokewidth=0, parent=self._parentNode) self.__fileNode = avg.VideoNode(href=sys.argv[1], loop=True) self.__fileNode.play() self.__filter = avg.ChromaKeyFXNode() self.__fileNode.setEffect(Chromakey.__filter) self.__filter.color = "0000FF" self.__filter.htolerance = 0.05 self.__filter.stolerance = 1.0 self.__filter.ltolerance = 1.0 self.__filter.softness = 0.0 self.__createGUI() def __createGUI(self): self.__guiDiv = avg.DivNode(pos=(node.getMediaSize().x+10,10), parent=self._parentNode) self.__colorWords = avg.WordsNode(pos=(0,14), parent=self.__guiDiv) self.__colorWords.text = "Key Color: "+self.__filter.color self.__colorRect = avg.RectNode(pos=(200,12), size=(20, 20), fillcolor=self.__filter.color, fillopacity=1, color="FFFFFF", parent=self.__guiDiv) node.setEventHandler(avg.CURSORDOWN, avg.MOUSE, self.__onColorDown) FXSlider(1, 0.0, 1.0, self.__filter, "htolerance", "Hue Tolerance", False, parent=self.__guiDiv) FXSlider(2, 0.0, 1.0, self.__filter, "stolerance", "Saturation Tolerance", False, parent=self.__guiDiv) FXSlider(3, 0.0, 1.0, self.__filter, "ltolerance", "Lightness Tolerance", False, parent=self.__guiDiv) FXSlider(4, 0.0, 1.0, self.__filter, "softness", "Softness", False, parent=self.__guiDiv) FXSlider(5, 0, 8, self.__filter, "erosion", "Erosion", True, parent=self.__guiDiv) FXSlider(6, 0.0, 1.0, self.__filter, "spillthreshold", "Spill Suppression", False, parent=self.__guiDiv) TextButton(pos=(0,332), text="Whitebalance", size=(100,22), clickHandler=self.__onWhitebalance, parent=self.__guiDiv) TextButton(pos=(110,332), text="Dump Config", size=(100,22), clickHandler=self.__dumpConfig, parent=self.__guiDiv) def __onColorDown(self, event): pos = node.getRelPos(event.pos) bmp = node.getBitmap() color = bmp.getPixel(pos) colorString = colorToString(color) self.__filter.color = colorString self.__colorWords.text = "Key Color: "+colorString self.__colorRect.fillcolor = colorString def __onWhitebalance(self, event): node.setWhitebalance( node.getWhitebalanceU(), node.getWhitebalanceV()) node.doOneShotWhitebalance() def __dumpConfig(self, event): print "Chromakey:" print " color=", self.__filter.color print " htolerance=", self.__filter.htolerance print " stolerance=", self.__filter.stolerance print " ltolerance=", self.__filter.ltolerance print " softness=", self.__filter.softness print " erosion=", self.__filter.erosion print " spillthreshold=", self.__filter.spillthreshold resolution=(GUI_SIZE[0]+800, max(GUI_SIZE[1],534)) Chromakey.start(resolution=resolution)