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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
#include "PipeLine.h" 
 
PipeMaterial::PipeMaterial() : buffer(nullptr), buffSize(0), former(nullptr) 
{ 
} 
 
PipeLine::PipeLine() : global_params_map(), elem_create_func_map(), elems() 
{ 
} 
 
PipeLine::~PipeLine() 
{ 
    // pipe stop 
     
    for(elem_vec_t::iterator iter = elems.begin(); iter != elems.end(); ++iter) 
    { 
        PipeLineElem* elem = *iter; 
        if (elem != nullptr) 
        { 
            elem->finit(); 
            delete *iter; 
        } 
    } 
     
    elems.clear(); 
} 
 
bool PipeLine::register_elem_creator(const std::string& type, elem_create_func_t func) 
{ 
    if (type.empty() || func == nullptr) 
        return false; 
 
    elem_create_func_map_t::iterator iter = elem_create_func_map.find(type); 
    if (iter != elem_create_func_map.end()) 
        return false; 
     
    elem_create_func_map.insert(std::make_pair(type, func)); 
    return true; 
} 
 
void PipeLine::push_elem(PipeLineElem* elem) 
{ 
    if(elem != nullptr) 
    { 
        elem->manager = this; 
        elems.push_back(elem); 
    } 
} 
 
PipeLineElem* PipeLine::push_elem(const std::string& type) 
{ 
    elem_create_func_map_t::iterator iter = elem_create_func_map.find(type); 
    if (iter == elem_create_func_map.end()) 
        return nullptr; 
     
    elem_create_func_t func = iter->second; 
    if (func == nullptr) 
        return nullptr; 
     
    PipeLineElem* elem = func(); 
    if (elem == nullptr) 
        return nullptr; 
     
    elem->manager = this; 
    push_elem(elem); 
    return elem; 
} 
 
PipeLineElem* PipeLine::pipe(PipeMaterial* pm /*= nullptr*/) 
{ 
    PipeLineElem* elem_begin = *elems.begin(); 
    PipeLineElem* elem_last = *elems.rbegin(); 
 
    if (elems.empty() || elem_begin == nullptr || elem_last == nullptr) 
        return nullptr; 
 
    uint8_t pmPlacement[sizeof(PipeMaterial)]; 
    if (pm == nullptr) 
        pm = new (pmPlacement) PipeMaterial; 
     
    if (elems.size() == 1) 
    { 
        elem_begin->gain(*pm); 
        return elem_begin; 
    } 
    else if (elems.size() == 2) 
    { 
        elem_begin->gain(*pm); 
        elem_last->pay(*pm); 
        return elem_last; 
    } 
    else 
    { 
        elem_begin->gain(*pm); 
         
        elem_vec_t::iterator iter = elems.begin(); 
        while (elem_begin != elem_last) 
        { 
            ++iter; 
            elem_begin = *iter; 
            elem_begin->pay(*pm); 
            elem_begin->gain(*pm); 
        } 
     
        elem_last->pay(*pm); 
        return elem_last; 
    } 
     
    return nullptr; 
} 
 
void PipeLine::set_global_param(const std::string& name, const std::string& value) 
{ 
    if (name.empty()) 
        return; 
 
    global_params_map_t::iterator iter = global_params_map.find(name); 
    if (iter == global_params_map.end()) 
        global_params_map.insert(std::make_pair(name, value)); 
    else 
        iter->second = value; 
} 
 
std::string PipeLine::get_global_param(const std::string& name) const 
{ 
    global_params_map_t::const_iterator iter = global_params_map.find(name); 
    if (iter == global_params_map.end()) 
        return ""; 
    else 
        return iter->second; 
}