summaryrefslogtreecommitdiffstats
path: root/plugins/codecs/opus_dec/opusdecode.c
blob: 97398bb5cea42fb90b47d42a09d9379fa3ac0a88 (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
/* opusdecode.c
 * opus codec
 *
 * Wireshark - Network traffic analyzer
 * By Gerald Combs <gerald@wireshark.org>
 * Copyright 1998 Gerald Combs
 *
 * SPDX-License-Identifier: GPL-2.0-or-later
 */

#include "config.h"
#include <stdlib.h>
#include "opus/opus.h"

#include "wsutil/codecs.h"
#include "ws_attributes.h"

void codec_register_opus(void);

static void *
codec_opus_init(codec_context_t *ctx _U_)
{
    OpusDecoder *state;
    int err = OPUS_INTERNAL_ERROR;
    /* Opus has in-band signaling and can convert what is sent to our
     * desired output.
     * always use maximum 48000 to cover all 8k/12k/16k/24k/48k
     * always downmix to mono because RTP Player only supports mono now
     */
    state = opus_decoder_create(48000, 1, &err);
    return state;
}

static void
codec_opus_release(codec_context_t *ctx)
{
    OpusDecoder* state = (OpusDecoder*)ctx->priv;
    if (!state) {
      return; /* out-of-memory; */
    }
    opus_decoder_destroy(state);
}

static unsigned
codec_opus_get_channels(codec_context_t *ctx _U_)
{
    return 1;
}

static unsigned
codec_opus_get_frequency(codec_context_t *ctx _U_)
{
    /* although can set kinds of fs, but we set 48K now */
    return 48000;
}

static size_t
codec_opus_decode(codec_context_t *ctx,
                  const void *input, size_t inputSizeBytes,
                  void *output, size_t *outputSizeBytes)
{
    OpusDecoder *state = (OpusDecoder *)ctx->priv;

    if (!state) {
        return 0;  /* out-of-memory */
    }

    const unsigned char *data = (const unsigned char *)input;
    opus_int32 len = (opus_int32)inputSizeBytes;
    int frame_samples = opus_decoder_get_nb_samples(state, data, len);
    if (frame_samples < 0) { // OPUS_INVALID_PACKET
        return 0;
    }

    // reserve space for the first time
    if (!output || !outputSizeBytes) {
        return frame_samples*2;
    }
    opus_int16 *pcm = (opus_int16*)(output);
    int ret = opus_decode(state, data, len, pcm, frame_samples, 0);

    if (ret < 0) {
        return 0;
    }
    *outputSizeBytes = ret * 2;
    return *outputSizeBytes;
}

void
codec_register_opus(void)
{
    register_codec("opus", codec_opus_init, codec_opus_release,
                   codec_opus_get_channels, codec_opus_get_frequency, codec_opus_decode);
}

/*
 * Editor modelines  -  https://www.wireshark.org/tools/modelines.html
 *
 * Local variables:
 * c-basic-offset: 4
 * tab-width: 8
 * indent-tabs-mode: nil
 * End:
 *
 * vi: set shiftwidth=4 tabstop=8 expandtab:
 * :indentSize=4:tabSize=8:noTabs=true:
 */