aboutsummaryrefslogtreecommitdiffstats
path: root/src/VLCInput.h
blob: e4a0557f69d300a3a2bd9fced5c59e1033db6e0e (plain)
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
106
107
108
109
110
111
112
113
114
/* ------------------------------------------------------------------
 * Copyright (C) 2015 Matthias P. Braendli
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 * express or implied.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 * -------------------------------------------------------------------
 */

#ifndef __VLC_INPUT_H_
#define __VLC_INPUT_H_

#include "config.h"

#if HAVE_VLC

#include <cstdio>
#include <string>
#include <vector>
#include <deque>

#include <vlc/vlc.h>
#include <boost/thread/thread.hpp>

#include "SampleQueue.h"

// 16 bits per sample is fine for now
#define BYTES_PER_SAMPLE 2

class VLCInput
{
    public:
        VLCInput(const std::string& uri,
                 int rate,
                 unsigned channels,
                 unsigned verbosity) :
            m_uri(uri),
            m_verbosity(verbosity),
            m_channels(channels),
            m_rate(rate),
            m_vlc(NULL) { }

        ~VLCInput() { cleanup(); }

        /* Prepare the audio input */
        int prepare();

        /* Read exactly length bytes into buf.
         * Blocks if not enough data is available,
         * or returns zero if EOF reached.
         *
         * Returns the number of bytes written into
         * the buffer.
         */
        ssize_t read(uint8_t* buf, size_t length);


        // Callbacks for VLC

        /* Notification of VLC exit */
        void exit_cb(void);

        /* Prepare a buffer for VLC */
        void preRender_cb(
                uint8_t** pp_pcm_buffer,
                size_t size);

        /* Receive a buffer with audio samples
         * from VLC
         */
        void postRender_cb(
                uint8_t* p_pcm_buffer,
                size_t size);

        int getRate() { return m_rate; }

        int getChannels() { return m_channels; }

    protected:
        void cleanup(void);

        ssize_t m_read(uint8_t* buf, size_t length);

        std::vector<uint8_t> m_current_buf;

        boost::mutex m_queue_mutex;
        std::deque<uint8_t> m_queue;

        std::string m_uri;
        unsigned m_verbosity;
        unsigned m_channels;
        int m_rate;

        // VLC pointers
        libvlc_instance_t     *m_vlc;
        libvlc_media_player_t *m_mp;

    private:
        VLCInput(const VLCInput& other) {}
};

#endif // HAVE_VLC

#endif