gstreamer-py-tuts/basic-tut-7.py at main · meerasndr/gstreamer-py-tuts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
#!/usr/bin/env python3
import sys
import logging
import gi
import signal

gi.require_version("GLib", "2.0")
gi.require_version("GObject", "2.0")
gi.require_version("Gst", "1.0")
from gi.repository import GLib, GObject, Gst

logging.basicConfig(
    level=logging.DEBUG, format="[%(name)s] [%(levelname)8s] - %(message)s"
)
logger = logging.getLogger(__name__)


def main():
    Gst.init(None)
    signal.signal(signal.SIGINT, signal.SIG_DFL)
    audio_source = Gst.ElementFactory.make("audiotestsrc", "audio_source")
    tee = Gst.ElementFactory.make("tee", "tee")
    audio_queue = Gst.ElementFactory.make("queue", "audio_queue")
    audio_convert = Gst.ElementFactory.make("audioconvert", "audio_convert")
    audio_resample = Gst.ElementFactory.make("audioresample", "audio_resample")
    audio_sink = Gst.ElementFactory.make("autoaudiosink", "audio_sink")
    video_queue = Gst.ElementFactory.make("queue", "video_queue")
    visual = Gst.ElementFactory.make("wavescope", "visual")
    video_convert = Gst.ElementFactory.make("videoconvert", "video_convert")
    video_sink = Gst.ElementFactory.make("autovideosink", "video_sink")

    # Create empty pipeline
    pipeline = Gst.Pipeline.new("test-pipeline")
    if (
        not pipeline
        or not audio_source
        or not tee
        or not audio_queue
        or not audio_convert
        or not audio_resample
        or not audio_sink
        or not video_queue
        or not visual
        or not video_convert
        or not video_sink
    ):
        logger.error("Not all elements could be created \n")
        sys.exit(1)

    audio_source.set_property("freq", 215.0)
    visual.set_property("shader", 0)
    visual.set_property("style", 1)

    # Add elements to pipeline and link those that can be automatically linked
    # because they have "Always" pads
    pipeline.add(
        audio_source,
        tee,
        audio_queue,
        audio_convert,
        audio_resample,
        audio_sink,
        video_queue,
        visual,
        video_convert,
        video_sink,
    )
    ret_first = audio_source.link(tee)
    ret_second = audio_queue.link(audio_convert)
    ret_second = ret_second and audio_convert.link(audio_resample)
    ret_second = ret_second and audio_resample.link(audio_sink)
    ret_third = video_queue.link(visual)
    ret_third = ret_third and visual.link(video_convert)
    ret_third = ret_third and video_convert.link(video_sink)

    if not ret_first or not ret_second or not ret_third:
        logger.error("Elements could not be linked\n")
        sys.exit(1)
    # Manually link the Tee, which has request pads
    tee_audio_pad = tee.get_request_pad("src_%u")
    # print(f"Obtained request pad {tee_audio_pad} for audio branch")
    queue_audio_pad = audio_queue.get_static_pad("sink")

    tee_video_pad = tee.get_request_pad("src_%u")
    # print(f"Obtained request pad {tee_video_pad} for video branch")
    queue_video_pad = video_queue.get_static_pad("sink")

    ret_audio = tee_audio_pad.link(queue_audio_pad)
    ret_video = tee_video_pad.link(queue_video_pad)

    if ret_audio != 0 or ret_video != 0:
        logger.error("Tee could not be linked\n")
        sys.exit(1)

    # Start playing Pipeline
    pipeline.set_state(Gst.State.PLAYING)

    bus = pipeline.get_bus()
    msg = bus.timed_pop_filtered(
        Gst.CLOCK_TIME_NONE, Gst.MessageType.ERROR or Gst.MessageType.EOS
    )


if __name__ == "__main__":
    main()