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
|
// Copyright (C) by Josh Blum. See LICENSE.txt for licensing information.
#include <gras_impl/block_actor.hpp>
#include <boost/foreach.hpp>
using namespace gras;
void BlockActor::handle_input_tag(const InputTagMessage &message, const Theron::Address)
{
TimerAccumulate ta(data->stats.total_time_input);
MESSAGE_TRACER();
const size_t index = message.index;
//handle incoming stream tag, push into the tag storage
if GRAS_UNLIKELY(data->block_state == BLOCK_STATE_DONE) return;
data->input_tags[index].push_back(message.tag);
data->input_tags_changed[index] = true;
}
void BlockActor::handle_input_msg(const InputMsgMessage &message, const Theron::Address)
{
TimerAccumulate ta(data->stats.total_time_input);
MESSAGE_TRACER();
const size_t index = message.index;
//handle incoming async message, push into the msg storage
if GRAS_UNLIKELY(data->block_state == BLOCK_STATE_DONE) return;
data->input_msgs[index].push_back(message.msg);
this->update_input_avail(index);
ta.done();
this->task_main();
}
void BlockActor::handle_input_buffer(const InputBufferMessage &message, const Theron::Address)
{
TimerAccumulate ta(data->stats.total_time_input);
MESSAGE_TRACER();
const size_t index = message.index;
//handle incoming stream buffer, push into the queue
if GRAS_UNLIKELY(data->block_state == BLOCK_STATE_DONE) return;
data->input_queues.push(index, message.buffer);
this->update_input_avail(index);
ta.done();
this->task_main();
}
void BlockActor::handle_input_token(const InputTokenMessage &message, const Theron::Address)
{
TimerAccumulate ta(data->stats.total_time_input);
MESSAGE_TRACER();
ASSERT(message.index < worker->get_num_inputs());
//store the token of the upstream producer
data->token_pool.insert(message.token);
}
void BlockActor::handle_input_check(const InputCheckMessage &message, const Theron::Address)
{
TimerAccumulate ta(data->stats.total_time_input);
MESSAGE_TRACER();
const size_t index = message.index;
//record time of the first input declared done
if (not data->inputs_done.any())
{
data->first_input_done_time = boost::get_system_time();
}
//an upstream block declared itself done, recheck the token
data->inputs_done.set(index, data->input_tokens[index].unique());
//upstream done, give it one more attempt at task handling
ta.done();
this->task_main();
//Now check the status, mark block done if the input is done:
//When reserve_items is non zero, this ports is a sync input;
//mark the block done if the sync input will never be ready again.
//Otherwise, only mark done if all inputs are done with no data.
const size_t reserve_items = data->input_configs[index].reserve_items;
if (
(reserve_items != 0 and data->inputs_done[index] and not data->inputs_available[index])
or (reserve_items == 0 and data->inputs_done.all() and data->inputs_available.none())
) this->mark_done();
}
void BlockActor::handle_input_alloc(const InputAllocMessage &message, const Theron::Address)
{
TimerAccumulate ta(data->stats.total_time_input);
MESSAGE_TRACER();
const size_t index = message.index;
//new token for this downstream allocator
//so if the downstream overrides, it has a unique token
SBufferDeleter deleter = *(message.token);
SBufferToken token = SBufferToken(new SBufferDeleter(deleter));
SBufferConfig config = message.config;
config.token = token;
//handle the upstream block allocation request
OutputAllocMessage new_msg;
new_msg.queue = data->block->input_buffer_allocator(index, config);
if (new_msg.queue) worker->post_upstream(index, new_msg);
}
void BlockActor::handle_input_update(const InputUpdateMessage &message, const Theron::Address)
{
TimerAccumulate ta(data->stats.total_time_input);
MESSAGE_TRACER();
const size_t i = message.index;
//update buffer queue configuration
if (i >= data->input_queues.size()) return;
const size_t preload_bytes = data->input_configs[i].item_size*data->input_configs[i].preload_items;
const size_t reserve_bytes = data->input_configs[i].item_size*data->input_configs[i].reserve_items;
const size_t maximum_bytes = data->input_configs[i].item_size*data->input_configs[i].maximum_items;
data->input_queues.update_config(i, data->input_configs[i].item_size, preload_bytes, reserve_bytes, maximum_bytes);
this->update_input_avail(i);
}
|
