add MaterialBuffer

houxiao

2016-12-28 017b7eb736ccc53c52f81486be8196d7fafc3289

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
#include "PL_SensetimeFaceDetect.h" 
 
#include <opencv2/opencv.hpp> 
#include <cv_face.h> 
 
struct PL_SensetimeFaceDetect_Internal 
{ 
    uint8_t buffer[1920*1080*4]; 
    size_t buffSize; 
    size_t buffSizeMax; 
    MB_Frame lastFrame; 
 
    bool payError; 
     
    PL_SensetimeFaceDetect_Internal() :  
        buffSize(0), buffSizeMax(sizeof(buffer)), lastFrame(),  
        payError(true) 
    { 
    } 
     
    ~PL_SensetimeFaceDetect_Internal() 
    { 
    } 
     
    void reset() 
    { 
        buffSize = 0; 
        payError = true; 
         
        MB_Frame _lastFrame; 
        lastFrame = _lastFrame; 
    } 
}; 
 
PipeLineElem* create_PL_SensetimeFaceDetect() 
{ 
    return new PL_SensetimeFaceDetect; 
} 
 
PL_SensetimeFaceDetect::PL_SensetimeFaceDetect() : internal(new PL_SensetimeFaceDetect_Internal) 
{ 
} 
 
PL_SensetimeFaceDetect::~PL_SensetimeFaceDetect() 
{ 
    delete (PL_SensetimeFaceDetect_Internal*)internal; 
    internal= nullptr; 
} 
 
bool PL_SensetimeFaceDetect::init(void* args) 
{ 
    PL_SensetimeFaceDetect_Internal* in = (PL_SensetimeFaceDetect_Internal*)internal; 
    in->reset(); 
     
    return true; 
} 
 
void PL_SensetimeFaceDetect::finit() 
{ 
    PL_SensetimeFaceDetect_Internal* in = (PL_SensetimeFaceDetect_Internal*)internal; 
     
} 
 
bool PL_SensetimeFaceDetect::pay(const PipeMaterial& pm) 
{ 
    PL_SensetimeFaceDetect_Internal* in = (PL_SensetimeFaceDetect_Internal*)internal; 
 
    if (pm.type != PipeMaterial::PMT_FRAME) 
    { 
        printf("PL_H264Encoder::pay only support PMT_FRAME\n"); 
        return false; 
    } 
     
    if (pm.buffer == nullptr) 
        return false; 
     
    MB_Frame* frame = (MB_Frame*)pm.buffer; 
    if (frame->type != MB_Frame::MBFT_YUV420) 
    { 
        printf("PL_H264Encoder::pay only support MBFT_YUV420\n"); 
        return false; 
    } 
 
     
     
    //in->buffer readly 
 
    //static size_t f=0; 
    //char fname[50]; 
    //sprintf(fname, "%u.bgra", ++f); 
    //FILE * pFile = fopen (fname,"wb"); 
    //fwrite (in->buffer , sizeof(char), in->buffSize, pFile); 
    //fclose(pFile); 
 
    return true; 
} 
 
bool PL_SensetimeFaceDetect::gain(PipeMaterial& pm) 
{ 
    PL_SensetimeFaceDetect_Internal* in = (PL_SensetimeFaceDetect_Internal*)internal; 
 
    pm.type = PipeMaterial::PMT_FRAME; 
    pm.buffer = (uint8_t*)(&(in->lastFrame)); 
    pm.buffSize = sizeof(in->lastFrame); 
    pm.former = this; 
    return true; 
}