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
|
// 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;
//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 recheck the status, mark block done if the input is done
if (this->is_input_done(index)) 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;
//handle the upstream block allocation request
OutputAllocMessage new_msg;
new_msg.queue = data->block->input_buffer_allocator(index, message.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);
}
|
