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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
|
/* ------------------------------------------------------------------
* Copyright (C) 2017 AVT GmbH - Fabien Vercasson
* Copyright (C) 2019 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.
* -------------------------------------------------------------------
*/
/*! \section AVT Input
*
* This input comunicate with AVT encoder
*
* The encoded frames and reassambled
* The PAD Frames are sent to the encoder for insertion.
* The encoder is remotely controled to set bitrate and audio mode.
*
*/
#pragma once
#include "Socket.h"
#include "OrderedQueue.h"
#include <cstdint>
#include <cstdio>
#include <string>
#include <queue>
#include <vector>
#define DEF_BR 64
// The enum values folown the AVT messages definitions.
enum {
AVT_Mono = 0,
AVT_Mono_SBR,
AVT_Stereo,
AVT_Stereo_SBR,
AVT_Stereo_SBR_PS
};
enum {
AVT_MonoMode_LR2 = 0,
AVT_MonoMode_L,
AVT_MonoMode_R
};
enum {
AVT_DAC_32 = 0,
AVT_DAC_48
};
class AVTInput
{
public:
AVTInput(const std::string& input_uri, const std::string& output_uri, uint32_t pad_port,
size_t jitterBufferSize = 40);
/*! Open the file and prepare the wav decoder.
*
* \return nonzero on error
*/
int prepare(void);
/*! Inform class and remove encoder about the bitrate and audio mode
*
* \return nonzero on error
*/
int setDabPlusParameters(int bitrate, int channels, int sample_rate, bool sbr, bool ps);
/*! Read incomming frames from the encoder, reorder and reassemble then into DAB+ superframes
*! Give the next reassembled audio frame (120ms for DAB+)
*
* \return the size of the frame or 0 if none are available yet
*/
ssize_t getNextFrame(std::vector<uint8_t> &buf);
/*! Store a new PAD frame.
*! Frames are sent to the encoder on request
*/
void pushPADFrame(const uint8_t* buf, size_t size);
/* \return true if PAD Frame queue is full */
bool padQueueFull();
private:
std::string _input_uri;
std::string _output_uri;
uint32_t _pad_port;
size_t _jitterBufferSize;
Socket::UDPSocket _input_socket;
Socket::UDPSocket _output_socket;
Socket::UDPPacket _output_packet;
Socket::UDPSocket _input_pad_socket;
Socket::UDPPacket _input_pad_packet;
OrderedQueue _ordered;
std::queue<std::vector<uint8_t> > _padFrameQueue;
int32_t _subChannelIndex = DEF_BR/8;
int32_t _bitRate = DEF_BR * 1000;
int32_t _audioMode = AVT_Mono;
int32_t _monoMode = AVT_MonoMode_LR2;
int32_t _dac = AVT_DAC_48;
size_t _dab24msFrameSize = DEF_BR*3;
uint32_t _dummyFrameNumber = 0;
bool _frameAligned = false;
std::vector<uint8_t> _currentFrame;
size_t _currentFrameSize = 0;
int32_t _nbFrames = 0;
uint8_t* _nextFrameIndex = 0;
bool _parseURI(const char* uri, std::string& address, long& port);
int _openSocketSrv(Socket::UDPSocket* socket, const char* uri);
int _openSocketCli();
void _sendCtrlMessage();
void _sendPADFrame();
void _interpretMessage(const uint8_t* data, size_t size);
bool _checkMessage();
void _purgeMessages();
/*! Test Bytes 1,2,3 for STI detection */
bool _isSTI(const uint8_t* buf);
/*! Find and extract the DAB frame from UDP/RTP/STI received frame
* \param frameNumber will contain the frameNumber
* \param dataSize will contain the actual DAB frame size
* \return Pointer to first byte of the DAB frame, or NULL if not found
*/
const uint8_t* _findDABFrameFromUDP(const uint8_t* buf, size_t size,
int32_t& frameNumber, size_t& dataSize);
/*! Read and store one frame from encoder
*
* \return true if a data has been received
*/
bool _readFrame();
/*! Output info about received frames*/
enum _frameType {
_typeSTI,
_typeSTIRTP,
_typeCantExtract
};
_frameType _lastInfoFrameType;
size_t _lastInfoSize = 0;
size_t _infoNbFrame = 0;
void _info(_frameType type, size_t size);
};
|
