houxiao
2016-12-22 9d9849175b11b3ba9918ad4f980aa4a1c7c2afb0
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
#include "PL_AVFrameYUV420.h"
 
extern "C"
{
    #include <libavcodec/avcodec.h>
    #include <libavutil/frame.h>
    #include <libavformat/avformat.h>
}
 
struct AVFrameYUV420_Internal
{
    uint8_t buffer[1920*1080*3];
    size_t buffSize;
    size_t buffSizeMax;
    bool payError;
    
    AVFrameYUV420_Internal() : 
        buffSize(0), buffSizeMax(sizeof(buffer)), 
        payError(false)
    {
    }
    
    ~AVFrameYUV420_Internal()
    {
    }
    
    void reset()
    {
        buffSize = 0;
        payError = false;
    }
};
 
PipeLineElem* create_PL_AVFrameYUV420()
{
    return new PL_AVFrameYUV420;
}
 
PL_AVFrameYUV420::PL_AVFrameYUV420() : internal(new AVFrameYUV420_Internal)
{
}
 
PL_AVFrameYUV420::~PL_AVFrameYUV420()
{
    delete (AVFrameYUV420_Internal*)internal;
    internal= nullptr;
}
 
bool PL_AVFrameYUV420::init(void* args)
{
    AVFrameYUV420_Internal* in = (AVFrameYUV420_Internal*)internal;
    in->reset();
    
    return true;
}
 
void PL_AVFrameYUV420::finit()
{
    AVFrameYUV420_Internal* in = (AVFrameYUV420_Internal*)internal;
    
}
 
bool PL_AVFrameYUV420::pay(const PipeMaterial& pm)
{
    AVFrameYUV420_Internal* in = (AVFrameYUV420_Internal*)internal;
    
        int picSize = in->pAVCodecContext->height * in->pAVCodecContext->width;
        in->buffSize = picSize * 1.5;
 
        int height = in->pAVFrame->height;
        int width = in->pAVFrame->width;
 
        // write yuv420
        int a=0;
        for (int i = 0; i < height; i++)
        {
            memcpy(in->buffer + a, in->pAVFrame->data[0] + i * in->pAVFrame->linesize[0], width);
            a += width;
        }
        for (int i=0; i<height/2; i++)
        {
            memcpy(in->buffer + a, in->pAVFrame->data[1] + i * in->pAVFrame->linesize[1], width / 2);
            a += width / 2;
        }
        for (int i=0; i<height/2; i++)
        {
            memcpy(in->buffer + a, in->pAVFrame->data[2] + i * in->pAVFrame->linesize[2], width / 2);
            a += width / 2;
        }
 
        //in->buffer readly
 
        //static size_t f=0;
        //char fname[50];
        //sprintf(fname, "%u.yuv420", ++f);
        //FILE * pFile = fopen (fname,"wb");
        //fwrite (in->buffer , sizeof(char), in->buffSize, pFile);
        //fclose(pFile);
    
    
    return in->payError;
}
 
bool PL_AVFrameYUV420::gain(PipeMaterial& pm)
{
    AVFrameYUV420_Internal* in = (AVFrameYUV420_Internal*)internal;
 
    if (!in->payError)
    {
        pm.buffer = in->buffer;
        pm.buffSize = in->buffSize;
    }
    pm.former = this;
    return in->payError;
}