Exemple #1
0
class AudioTestGen(Device):
    suggested_name = "audiotestgen"
    suggested_readable_name = "Audio Test Generator"

    def __init__(self, name):
        Device.__init__(self, name)
        self.ControlPanelClass = AudioTestGenControlPanel

        self.freqs = ObservableVariable([1000, 1000])
        self.freqs.changed.connect(self.change_freqs)

        caps = Gst.caps_from_string(self.SINGLE_CHANNEL_AUDIO_CAPS)

        self.src0 = Gst.ElementFactory.make('audiotestsrc', None)
        self.bin.add(self.src0)
        self.src0.set_property('is-live', True)

        self.src1 = Gst.ElementFactory.make('audiotestsrc', None)
        self.bin.add(self.src1)
        self.src1.set_property('is-live', True)

        self.interleave = Gst.ElementFactory.make('interleave', None)
        self.bin.add(self.interleave)

        self.src0.link_filtered(self.interleave, caps)
        self.src1.link_filtered(self.interleave, caps)

        self.add_output_audio_port_on(self.interleave, "src")

        self.change_freqs(self.freqs.get_value())

    def change_freqs(self, data):
        self.src0.set_property('freq', data[0])
        self.src1.set_property('freq', data[1])
class Monitor(Device):
    suggested_name = "monitor"
    suggested_readable_name = "Monitor"

    def __init__(self, name, size, location):
        Device.__init__(self, name)
        self.ControlPanelClass = MonitorControlPanel

        self.size = ObservableVariable(size)
        self.size.changed.connect(self.change_size)
        self.location = ObservableVariable(location)
        self.location.changed.connect(self.change_location)

        self.convert = Gst.ElementFactory.make('videoconvert', None)
        self.bin.add(self.convert)

        self.scale = Gst.ElementFactory.make('videoscale', None)
        self.bin.add(self.scale)

        self.sink = Gst.ElementFactory.make('xvimagesink', None)
        self.bin.add(self.sink)

        caps = Gst.caps_from_string(Device.DEFAULT_VIDEO_CAPS)
        self.convert.link_filtered(self.scale, caps)
        self.scale.link(self.sink)

        self.add_input_video_port_on(self.convert)

    def get_sink(self):
        return self.sink

    def set_window_id(self, id):
        self.get_sink().set_window_handle(id)

    def change_size(self):
        self.display.set_size(*self.size.get_value())

    def change_location(self):
        self.display.set_location(*self.location.get_value())

    def prepare_for_removal(self):
        super(Monitor, self).prepare_for_removal()

        if self.display:
            self.display.close()
Exemple #3
0
class VideoTestGen(Device):
    suggested_name = "videotestgen"
    suggested_readable_name = "Video Test Generator"

    def __init__(self, name, pattern=0):
        Device.__init__(self, name)
        self.ControlPanelClass = VideoTestGenControlPanel

        self.pattern = ObservableVariable(pattern)
        self.pattern.changed.connect(self.change_pattern)

        self.src = Gst.ElementFactory.make('videotestsrc', None)
        self.bin.add(self.src)
        self.src.set_property('pattern', self.pattern.get_value())

        self.convert = Gst.ElementFactory.make('videoconvert', None)
        self.bin.add(self.convert)

        self.text_overlay = Gst.ElementFactory.make('textoverlay', None)
        self.bin.add(self.text_overlay)
        self.text_overlay.set_property("text", self.name)
        self.text_overlay.set_property("shaded-background", True)

        self.caps_filter = Gst.ElementFactory.make('capsfilter', None)
        self.bin.add(self.caps_filter)

        caps = Gst.caps_from_string(Device.DEFAULT_VIDEO_CAPS)
        self.caps_filter.set_property('caps', caps)

        self.src.link(self.text_overlay)
        self.text_overlay.link(self.convert)
        self.convert.link(self.caps_filter)

        self.add_output_video_port_on(self.caps_filter, "src")

    def change_pattern(self):
        self.src.set_property('pattern', self.pattern.get_value())
Exemple #4
0
class Switcher(Device):
    suggested_name = "switcher"
    suggested_readable_name = "Switcher"

    def __init__(self, name, inputs):
        Device.__init__(self, name)
        self.ControlPanelClass = SwitcherControlPanel

        self.program_active_id = ObservableVariable(0)
        self.preview_active_id = ObservableVariable(0)
        self.opacity = ObservableVariable(1.0)

        # Create elements that exist for all instances of the switcher
        self.prog_mixer = Gst.ElementFactory.make('videomixer', None)
        self.bin.add(self.prog_mixer)

        self.prev_mixer = Gst.ElementFactory.make('videomixer', None)
        self.bin.add(self.prev_mixer)

        # Add outputs for static mixer pads
        self.add_output_video_port_on(self.prog_mixer, "src", "prog_out")
        self.add_output_video_port_on(self.prev_mixer, "src", "prev_out")

        for input_id in range(inputs):
            # Create tee and create input
            tee = Gst.ElementFactory.make('tee', None)
            self.bin.add(tee)

            name = "in" + str(input_id + 1)

            self.add_input_video_port_on(tee, name, name)

            #pad = Gst.GhostPad.new(name, tee.get_static_pad("sink"))
            #self.bin.add_pad(pad)

            # Create queues
            queue1 = Gst.ElementFactory.make('queue', None)
            self.bin.add(queue1)
            queue2 = Gst.ElementFactory.make('queue', None)
            self.bin.add(queue2)

            # Link tee to queues
            pad = tee.get_request_pad("src_%u")
            toPad = queue1.get_static_pad("sink")
            pad.link(toPad)

            pad = tee.get_request_pad("src_%u")
            toPad = queue2.get_static_pad("sink")
            pad.link(toPad)

            # Link queues to videomixers
            pad = queue1.get_static_pad("src")
            toPad = self.prog_mixer.get_request_pad("sink_%u")
            pad.link(toPad)

            pad = queue2.get_static_pad("src")
            toPad = self.prev_mixer.get_request_pad("sink_%u")
            pad.link(toPad)

        if inputs >= 2:
            self.set_bus_input("preview", 1)
            self.set_bus_input("program", 0)
        else:
            self.set_bus_input("preview", 0)
            self.set_bus_input("program", 0)

        self.add_action("take", self.take, "Swap program and preview or set program")
        self.add_action("preview", self.preview, "Set preview")
        self.add_action("fade", self.fade, "Set fade position")

    def set_bus_input(self, bus, input_id):
        print "Setting bus input: " + bus + ", " + str(input_id)

        high_pad_name = "sink_" + str(input_id)

        if bus == "program":
            mixer = self.prog_mixer
            low_pad_name = "sink_" + str(self.preview_active_id.get_value())
            prev_high_pad_name = "sink_" + str(self.program_active_id.get_value())
        elif bus == "preview":
            mixer = self.prev_mixer
            low_pad_name = "sink_" + str(self.program_active_id.get_value())
            prev_high_pad_name = "sink_" + str(self.preview_active_id.get_value())
        else:
            return

        # Arrange input z-orders
        pad = mixer.get_static_pad(high_pad_name)
        pad.set_property('zorder', 2)

        pad = mixer.get_static_pad(low_pad_name)
        pad.set_property('zorder', 1)

        # Iterate over all sink pads and set below
        it = mixer.iterate_sink_pads()

        while True:
            status, pad = it.next()
            if status != Gst.IteratorResult.OK:
                break

            if (pad.get_name() != high_pad_name and pad.get_name() != low_pad_name):
                pad.set_property("zorder", 0)
                pad.set_property("alpha", 1.0)

        # Update
        if bus == "program":
            self.program_active_id.set_value(input_id)
        elif bus == "preview":
            self.preview_active_id.set_value(input_id)

        self.set_opacity(self.opacity.get_value())

        #self.debug_mixer(mixer)

    def debug_mixer(self, mixer):
        it = mixer.iterate_sink_pads()

        while True:
            status, pad = it.next()
            if status != Gst.IteratorResult.OK:
                break

            pad_info = "Pad: " + pad.get_name()
            pad_info = pad_info + " (zorder: " + str(pad.get_property("zorder"))
            pad_info = pad_info + ", alpha: " + str(pad.get_property('alpha')) + ")"
            print pad_info

    def take(self, program_id=None):
        if program_id is None:
            current_prog_id = self.program_active_id.get_value()
            current_prev_id = self.preview_active_id.get_value()

            self.set_bus_input("preview", current_prog_id)
            self.set_bus_input("program", current_prev_id)
        else:
            self.set_bus_input("program", int(program_id))

    def preview(self, preview_id):
        self.set_bus_input("preview", int(preview_id))

    def set_opacity(self, opacity):
        self.opacity.set_value(opacity)

        pad = self.prog_mixer.get_static_pad("sink_" + str(self.program_active_id.get_value()))

        if self.program_active_id.get_value() != self.preview_active_id.get_value():
            pad.set_property("alpha", opacity)
        else:
            pad.set_property("alpha", 1.0)

    def fade(self, opacity):
        self.set_opacity(float(opacity))
class Switcher(Device):
    suggested_name = "switcher"
    suggested_readable_name = "Switcher"

    def __init__(self, name, inputs):
        Device.__init__(self, name)
        self.ControlPanelClass = SwitcherControlPanel

        self.program_active_id = ObservableVariable(0)
        self.preview_active_id = ObservableVariable(0)
        self.opacity = ObservableVariable(1.0)

        # Create elements that exist for all instances of the switcher
        self.prog_mixer = Gst.ElementFactory.make('videomixer', None)
        self.bin.add(self.prog_mixer)

        self.prev_mixer = Gst.ElementFactory.make('videomixer', None)
        self.bin.add(self.prev_mixer)

        # Add outputs for static mixer pads
        self.add_output_video_port_on(self.prog_mixer, "src", "prog_out")
        self.add_output_video_port_on(self.prev_mixer, "src", "prev_out")

        for input_id in range(inputs):
            # Create tee and create input
            tee = Gst.ElementFactory.make('tee', None)
            self.bin.add(tee)

            name = "in" + str(input_id + 1)

            self.add_input_video_port_on(tee, name, name)

            #pad = Gst.GhostPad.new(name, tee.get_static_pad("sink"))
            #self.bin.add_pad(pad)

            # Create queues
            queue1 = Gst.ElementFactory.make('queue', None)
            self.bin.add(queue1)
            queue2 = Gst.ElementFactory.make('queue', None)
            self.bin.add(queue2)

            # Link tee to queues
            pad = tee.get_request_pad("src_%u")
            toPad = queue1.get_static_pad("sink")
            pad.link(toPad)

            pad = tee.get_request_pad("src_%u")
            toPad = queue2.get_static_pad("sink")
            pad.link(toPad)

            # Link queues to videomixers
            pad = queue1.get_static_pad("src")
            toPad = self.prog_mixer.get_request_pad("sink_%u")
            pad.link(toPad)

            pad = queue2.get_static_pad("src")
            toPad = self.prev_mixer.get_request_pad("sink_%u")
            pad.link(toPad)

        if inputs >= 2:
            self.set_bus_input("preview", 1)
            self.set_bus_input("program", 0)
        else:
            self.set_bus_input("preview", 0)
            self.set_bus_input("program", 0)

        self.add_action("take", self.take,
                        "Swap program and preview or set program")
        self.add_action("preview", self.preview, "Set preview")
        self.add_action("fade", self.fade, "Set fade position")

    def set_bus_input(self, bus, input_id):
        print "Setting bus input: " + bus + ", " + str(input_id)

        high_pad_name = "sink_" + str(input_id)

        if bus == "program":
            mixer = self.prog_mixer
            low_pad_name = "sink_" + str(self.preview_active_id.get_value())
            prev_high_pad_name = "sink_" + str(
                self.program_active_id.get_value())
        elif bus == "preview":
            mixer = self.prev_mixer
            low_pad_name = "sink_" + str(self.program_active_id.get_value())
            prev_high_pad_name = "sink_" + str(
                self.preview_active_id.get_value())
        else:
            return

        # Arrange input z-orders
        pad = mixer.get_static_pad(high_pad_name)
        pad.set_property('zorder', 2)

        pad = mixer.get_static_pad(low_pad_name)
        pad.set_property('zorder', 1)

        # Iterate over all sink pads and set below
        it = mixer.iterate_sink_pads()

        while True:
            status, pad = it.next()
            if status != Gst.IteratorResult.OK:
                break

            if (pad.get_name() != high_pad_name
                    and pad.get_name() != low_pad_name):
                pad.set_property("zorder", 0)
                pad.set_property("alpha", 1.0)

        # Update
        if bus == "program":
            self.program_active_id.set_value(input_id)
        elif bus == "preview":
            self.preview_active_id.set_value(input_id)

        self.set_opacity(self.opacity.get_value())

        #self.debug_mixer(mixer)

    def debug_mixer(self, mixer):
        it = mixer.iterate_sink_pads()

        while True:
            status, pad = it.next()
            if status != Gst.IteratorResult.OK:
                break

            pad_info = "Pad: " + pad.get_name()
            pad_info = pad_info + " (zorder: " + str(
                pad.get_property("zorder"))
            pad_info = pad_info + ", alpha: " + str(
                pad.get_property('alpha')) + ")"
            print pad_info

    def take(self, program_id=None):
        if program_id is None:
            current_prog_id = self.program_active_id.get_value()
            current_prev_id = self.preview_active_id.get_value()

            self.set_bus_input("preview", current_prog_id)
            self.set_bus_input("program", current_prev_id)
        else:
            self.set_bus_input("program", int(program_id))

    def preview(self, preview_id):
        self.set_bus_input("preview", int(preview_id))

    def set_opacity(self, opacity):
        self.opacity.set_value(opacity)

        pad = self.prog_mixer.get_static_pad(
            "sink_" + str(self.program_active_id.get_value()))

        if self.program_active_id.get_value(
        ) != self.preview_active_id.get_value():
            pad.set_property("alpha", opacity)
        else:
            pad.set_property("alpha", 1.0)

    def fade(self, opacity):
        self.set_opacity(float(opacity))