summaryrefslogtreecommitdiffstats
path: root/src/dabInputZmq.h
blob: cd8df6f138d2933b8c1a564bb25421f1a81aa142 (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
115
116
117
118
119
120
121
122
123
124
/*
   Copyright (C) 2009 Her Majesty the Queen in Right of Canada (Communications
   Research Center Canada)

   Copyright (C) 2013 Matthias P. Braendli
   http://mpb.li

   ZeroMQ input. see www.zeromq.org for more info

   For the AAC+ input, each zeromq message must contain one superframe.

   For the MPEG input, each zeromq message must contain one frame.

   From the ZeroMQ manpage 'zmq':

       The 0MQ lightweight messaging kernel is a library which extends the standard
       socket interfaces with features traditionally provided by specialised
       messaging middleware products. 0MQ sockets provide an abstraction of
       asynchronous message queues, multiple messaging patterns, message filtering
       (subscriptions), seamless access to multiple transport protocols and more.
   */
/*
   This file is part of ODR-DabMux.

   It defines a ZeroMQ input for dabplus data.

   ODR-DabMux is free software: you can redistribute it and/or modify
   it under the terms of the GNU General Public License as
   published by the Free Software Foundation, either version 3 of the
   License, or (at your option) any later version.

   ODR-DabMux is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with ODR-DabMux.  If not, see <http://www.gnu.org/licenses/>.
   */

#ifndef DAB_INPUT_ZMQ_H
#define DAB_INPUT_ZMQ_H

#ifdef HAVE_CONFIG_H
#   include "config.h"
#endif

#ifdef HAVE_INPUT_ZEROMQ

#include <list>
#include <string>
#include "zmq.hpp"
#include "dabInput.h"
#include "StatsServer.h"

/* The frame_buffer contains DAB logical frames as defined in
 * TS 102 563, section 6.
 * Five elements of this buffer make one AAC superframe (120ms audio)
 */

// Number of elements to prebuffer before starting the pipeline
#define INPUT_ZMQ_PREBUFFERING (5*4) // 480ms

// Maximum frame_buffer size in number of elements
#define INPUT_ZMQ_MAX_BUFFER_SIZE (5*8) // 960ms

class DabInputZmqBase : public DabInputBase, public RemoteControllable {
    public:
        DabInputZmqBase(const std::string name)
            : RemoteControllable(name),
            m_name(name), m_zmq_context(1),
            m_zmq_sock(m_zmq_context, ZMQ_SUB),
            m_bitrate(0), m_prebuffering(INPUT_ZMQ_PREBUFFERING) {
            }

        virtual int open(const std::string inputUri);
        virtual int readFrame(void* buffer, int size);
        virtual int setBitrate(int bitrate);
        virtual int close();

        /* Remote control */
        virtual void set_parameter(string parameter, string value);

        /* Getting a parameter always returns a string. */
        virtual string get_parameter(string parameter);

    protected:
        virtual int readFromSocket(int framesize) = 0;

        std::string m_name;
        zmq::context_t m_zmq_context;
        zmq::socket_t m_zmq_sock; // handle for the zmq socket
        int m_bitrate;
        int m_prebuffering;
        std::list<char*> m_frame_buffer; //stores elements of type char[<framesize>]
};

class DabInputZmqMPEG : public DabInputZmqBase {
    public:
        DabInputZmqMPEG(const std::string name)
            : DabInputZmqBase(name) {
                RC_ADD_PARAMETER(buffer,
                        "Size of the input buffer [mpeg frames]");
            }

    private:
        virtual int readFromSocket(int framesize);
};

class DabInputZmqAAC : public DabInputZmqBase {
    public:
        DabInputZmqAAC(const std::string name)
            : DabInputZmqBase(name) {
                RC_ADD_PARAMETER(buffer,
                        "Size of the input buffer [aac superframes]");
            }

    private:
        virtual int readFromSocket(int framesize);
};

#endif // HAVE_INPUT_ZMQ

#endif // DAB_INPUT_ZMQ_H