diff --git a/beep.m4a b/beep.m4a new file mode 100644 index 0000000..62facdc Binary files /dev/null and b/beep.m4a differ diff --git a/photobooth.css b/photobooth.css new file mode 100644 index 0000000..8f04607 --- /dev/null +++ b/photobooth.css @@ -0,0 +1,8 @@ +.snapshot_label { +/* background-color: rgba (100, 100, 0, 0); */ + background: rgba (50, 100, 10, 50); + color: white; + font-weight: bold; + font-size: 60px; + padding: 0 20px 0 20px; +} diff --git a/shutter.mp3 b/shutter.mp3 new file mode 100644 index 0000000..16a2888 Binary files /dev/null and b/shutter.mp3 differ diff --git a/video-overlay-test.py b/video-overlay-test.py index 2ce45ec..467fbde 100755 --- a/video-overlay-test.py +++ b/video-overlay-test.py @@ -8,6 +8,11 @@ gi.require_version('GstVideo', '1.0') gi.require_version('GdkX11', '3.0') from gi.repository import Gst, GObject, Gtk, GdkX11, Gdk, GstVideo +COUNTDOWN_SECONDS = 4 +AUDIOFILE_COUNTDOWN = "beep.m4a" +AUDIOFILE_SHUTTER = "shutter.mp3" + + class GTK_Main: def __init__(self): window = Gtk.Window(Gtk.WindowType.TOPLEVEL) @@ -15,27 +20,46 @@ class GTK_Main: window.set_default_size(1024, 830) window.connect("destroy", Gtk.main_quit, "WM destroy") Gtk.Window.fullscreen(window) - vbox = Gtk.VBox() - window.add(vbox) + + style_provider = Gtk.CssProvider() + css = open('photobooth.css') + css_data = css.read() + css.close() + style_provider.load_from_data(css_data) + Gtk.StyleContext.add_provider_for_screen( + Gdk.Screen.get_default(), + style_provider, + Gtk.STYLE_PROVIDER_PRIORITY_APPLICATION + ) + + self.overlay = Gtk.Overlay() + window.add(self.overlay) + self.movie_window = Gtk.DrawingArea() self.movie_window.add_events(Gdk.EventMask.BUTTON_PRESS_MASK) self.movie_window.connect("button-press-event", self.snapshot) - vbox.add(self.movie_window) + self.overlay.add(self.movie_window) + vbox = Gtk.VBox() hbox = Gtk.HBox() vbox.pack_start(hbox, False, False, 0) hbox.set_border_width(10) hbox.pack_start(Gtk.Label(), False, False, 0) - self.button = Gtk.Button("Start") - self.button.connect("clicked", self.start_stop) - hbox.pack_start(self.button, False, False, 0) - self.button2 = Gtk.Button("Quit") - self.button2.connect("clicked", self.exit) - hbox.pack_start(self.button2, False, False, 0) - hbox.add(Gtk.Label()) + self.button_play = Gtk.Button("Start") + self.button_play.connect("clicked", self.start_stop) + hbox.pack_start(self.button_play, False, False, 0) + self.button_quit = Gtk.Button("Quit") + self.button_quit.connect("clicked", self.exit) + hbox.pack_start(self.button_quit, False, False, 0) + + vbox.set_valign(Gtk.Align.END) + vbox.set_halign(Gtk.Align.CENTER) + self.overlay.add_overlay(vbox) + + self.overlay.show_all() window.show_all() - self.player = Gst.Pipeline.new("player") + self.videoplayer = Gst.Pipeline.new("player") source = Gst.ElementFactory.make("videotestsrc", "file-source") source.set_property("pattern", 1) capsfilter = Gst.ElementFactory.make("capsfilter", "capsfilter") @@ -55,7 +79,7 @@ class GTK_Main: videosink = Gst.ElementFactory.make("xvimagesink", "video-output") for ele in (source, capsfilter, video_convert, png_source, png_decoder, alphacolor, png_convert, imagefreeze, mixer, mix_convert, videosink): - self.player.add(ele) + self.videoplayer.add(ele) source.link(capsfilter) capsfilter.link(video_convert) @@ -70,38 +94,71 @@ class GTK_Main: mixer.link(mix_convert) mix_convert.link(videosink) - bus = self.player.get_bus() + bus = self.videoplayer.get_bus() bus.add_signal_watch() bus.enable_sync_message_emission() bus.connect("message", self.on_message) bus.connect("sync-message::element", self.on_sync_message) + self.countdown_timer = None + self.soundplayer = Gst.Pipeline.new("soundplayer") + def start_stop(self, w): - if self.button.get_label() == "Start": + if self.button_play.get_label() == "Start": self.xid = self.movie_window.get_property('window').get_xid() - self.button.set_label("Stop") - self.player.set_state(Gst.State.PLAYING) + self.button_play.set_label("Stop") + self.videoplayer.set_state(Gst.State.PLAYING) else: - self.player.set_state(Gst.State.NULL) - self.button.set_label("Start") + self.videoplayer.set_state(Gst.State.NULL) + self.button_play.set_label("Start") def snapshot(self, widget, event): - if event.button == 1: - print "TAKE SNAPSHOT!" - + if event.button == 1 and self.videoplayer.get_state(0).state == Gst.State.PLAYING and not self.countdown_timer: + self.countdown_timer = GObject.timeout_add(1000, self.countdown_cb) + self.snapshot_label = Gtk.Label(label="SNAPSHOT!", angle=25, halign=Gtk.Align.END) + self.snapshot_label.set_valign(Gtk.Align.CENTER) + self.snapshot_label.set_halign(Gtk.Align.CENTER) + self.snapshot_label.set_size_request(500, 300) + self.snapshot_label.get_style_context().add_class("snapshot_label") + self.audiplaybin = Gst.ElementFactory.make("playbin", "soundfile") + uri = 'file://'+os.path.abspath(AUDIOFILE_COUNTDOWN) + self.audiplaybin.set_property('uri',uri) + self.soundplayer.add(self.audiplaybin) + self.soundplayer.set_state(Gst.State.PLAYING) + self.overlay.add_overlay(self.snapshot_label) + self.overlay.show_all() + self.counter = COUNTDOWN_SECONDS + + def countdown_cb(self): + self.counter -= 1 + if self.counter == 0: + self.snap() + return False + self.snapshot_label.set_text("SNAP IN %i" % self.counter) + return True + + def snap(self): + print "TAKE SNAPSHOT!" + self.countdown_timer = None + self.soundplayer.set_state(Gst.State.READY) + uri = 'file://'+os.path.abspath(AUDIOFILE_SHUTTER) + self.audiplaybin.set_property('uri',uri) + self.soundplayer.set_state(Gst.State.PLAYING) + self.overlay.remove(self.snapshot_label) + def exit(self, widget, data=None): Gtk.main_quit() def on_message(self, bus, message): t = message.type if t == Gst.MessageType.EOS: - self.player.set_state(Gst.State.NULL) - self.button.set_label("Start") + self.videoplayer.set_state(Gst.State.NULL) + self.button_play.set_label("Start") elif t == Gst.MessageType.ERROR: err, debug = message.parse_error() print "Error: %s" % err, debug - self.player.set_state(Gst.State.NULL) - self.button.set_label("Start") + self.videoplayer.set_state(Gst.State.NULL) + self.button_play.set_label("Start") def on_sync_message(self, bus, message): struct = message.get_structure()