add pipeline

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; 
}