HLSLive/OutputContext.cpp at master · zzuzhp/HLSLive · 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
#include "OutputContext.h"
#include <cassert>

/////////////////////////////////////////////////////////////////////////////////
////

int
g_write(void * opaque, unsigned char * buf, int buf_size)
{
    OutputContext * inst = static_cast<OutputContext *>(opaque);
    return inst->write(buf, buf_size);
}

OutputContext::OutputContext() : m_pfnSink(NULL)
{

}

OutputContext::~OutputContext()
{
    clean();
}

void
OutputContext::clean()
{
    if (m_ctx)
    {
        /* note: the internal buffer could have changed internally by ffmpeg. */
        ::av_freep(&m_ctx->buffer);
        m_buffer = NULL;

        ::av_freep(m_ctx);
        m_ctx = NULL;
    }
}

bool
OutputContext::setparam(IOutputContextSink * pfnSink)
{
    if (!m_buffer)
    {
        return false;
    }

    m_ctx = avio_alloc_context((unsigned char *)m_buffer,
                               AVIO_BUFFER_SIZE,
                               1,
                               this,
                               NULL,
                               &g_write,
                               NULL);

    assert(m_ctx);
    if (!m_ctx)
    {
        return false;
    }

    m_pfnSink = pfnSink;

    return true;
}

int
OutputContext::write(unsigned char * buf, int buf_size)
{
    if (m_pfnSink)
    {
        (m_pfnSink->OCtxOutput)(buf, buf_size);
        return buf_size;
    }

    return 0;
}

/////////////////////////////////////////////////////////////////////////////////
////