lcserver/playback.py at master · IMC-Network/lcserver · 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
import modules
import sessions
import streaming

import vlc
import os

class Player(modules.Module):
    def __init__(self, input, log, sourceRoot, streamConnection = None):
        super().__init__(input, log)

        self.sourceRoot = sourceRoot

        if type(streamConnection) is streaming.StreamConnection:
            self.vlcInstance = vlc.Instance("\
--quiet --sout=#duplicate{{dst=display,dst=transcode{{vcodec=h264,scale=1,width={},height={},fps={},acodec=mp3,channels=2,samplerate={}:std{{access=rtmp,mux=ffmpeg{{mux=flv}},dst={}}}}}\
".format(
                streamConnection.transcodeWidth,
                streamConnection.transcodeHeight,
                streamConnection.transcodeFramerate,
                streamConnection.sampleRate,
                streamConnection.streamOutput
            ))
        else:
            self.vlcInstance = vlc.Instance("--quiet")

        self.vlcPlayer = self.vlcInstance.media_player_new()
        self.vlcEvents = self.vlcPlayer.event_manager()

        self.currentMRL = None
        self.playing = False
        self.cueHandled = False

        self.vlcEvents.event_attach(vlc.EventType.MediaPlayerEndReached, self._onMediaEndReached)

    def cueMRL(self, mrl):
        self.currentMRL = mrl

        if not self.cueHandled:
            self.vlcPlayer.set_mrl(self.currentMRL)

            self.cueHandled = True

    def cueFile(self, path):
        if os.path.isfile(self.sourceRoot + "/" + path):
            self.cueMRL("file:///" + self.sourceRoot + "/" + path)
        else:
            raise IOError("file not found")

    def play(self):
        if self.currentMRL != None:
            if not self.cueHandled:
                self.vlcPlayer.set_mrl(self.currentMRL)

            self.vlcPlayer.play()

            self.playing = True
            self.cueHandled = False
        else:
            raise TypeError("no media cued")

    def stop(self):
        self.vlcPlayer.stop()

        self.playing = False

    def _onMediaEndReached(self, event):
        self.playing = False

    def _runCommand(self, arguments, runningSession):
        if arguments[0] == "cueMRL":
            if len(arguments) > 1:
                self.cueMRL(arguments[1])
            else:
                runningSession.handleError(type(self).__name__, sessions.ReturnCode.ARGUMENT_ERROR, "argument `<mrl>` not specified")

                return sessions.ReturnCode.ARGUMENT_ERROR
        elif arguments[0] == "cueFile":
            if len(arguments) > 1:
                try:
                    self.cueFile(arguments[1])
                except IOError:
                    runningSession.handleError(type(self).__name__, sessions.ReturnCode.FILE_NOT_FOUND, "file not found")

                    return sessions.ReturnCode.FILE_NOT_FOUND
            else:
                runningSession.handleError(type(self).__name__, sessions.ReturnCode.ARGUMENT_ERROR, "argument `<path>` not specified")

                return sessions.ReturnCode.ARGUMENT_ERROR
        elif arguments[0] == "play":
            try:
                self.play()
            except TypeError:
                runningSession.handleError(type(self).__name__, sessions.ReturnCode.ERROR, "no media cued")

                return sessions.ReturnCode.ERROR
        elif arguments[0] == "stop":
            self.stop()
        else:
            return sessions.ReturnCode.MODULE_COMMAND_NOT_FOUND

        return sessions.ReturnCode.OK