summaryrefslogtreecommitdiff
path: root/lib/input_handlers.cpp
blob: 4ca0bdf6b0b3f4d0d08ba49ce2b682d55c77d2ec (plain)
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
// 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)
{
    MESSAGE_TRACER();
    const size_t index = message.index;

    //handle incoming stream tag, push into the tag storage
    this->input_tags[index].push_back(message.tag);

    //Changed is a boolean to enable sorting of tags: If the offset is 0, there is nothing to sort,
    //because tags are being used for message passing, or this is just the first tag in a stream.
    this->input_tags_changed[index] = this->input_tags_changed[index] or message.tag.offset != 0;
    this->inputs_available.set(index);
    this->handle_task();
}

void BlockActor::handle_input_buffer(const InputBufferMessage &message, const Theron::Address)
{
    MESSAGE_TRACER();
    const size_t index = message.index;

    //handle incoming stream buffer, push into the queue
    if (this->block_state == BLOCK_STATE_DONE) return;
    this->input_queues.push(index, message.buffer);
    this->inputs_available.set(index);
    this->handle_task();
}

void BlockActor::handle_input_token(const InputTokenMessage &message, const Theron::Address)
{
    MESSAGE_TRACER();
    ASSERT(message.index < this->get_num_inputs());

    //store the token of the upstream producer
    this->token_pool.insert(message.token);
}

void BlockActor::handle_input_check(const InputCheckMessage &message, const Theron::Address)
{
    MESSAGE_TRACER();
    const size_t index = message.index;

    //an upstream block declared itself done, recheck the token
    this->inputs_done.set(index, this->input_tokens[index].unique());
    if (this->is_input_done(index)) //missing an upstream provider
    {
        this->mark_done();
    }
    //or re-enter handle task so fail logic can mark done
    else
    {
        this->handle_task();
    }
}

void BlockActor::handle_input_alloc(const InputAllocMessage &message, const Theron::Address)
{
    MESSAGE_TRACER();
    const size_t index = message.index;

    //handle the upstream block allocation request
    OutputAllocMessage new_msg;
    new_msg.queue = block_ptr->input_buffer_allocator(index, message.config);
    if (new_msg.queue) this->post_upstream(index, new_msg);
}

void BlockActor::handle_input_update(const InputUpdateMessage &message, const Theron::Address)
{
    MESSAGE_TRACER();
    const size_t i = message.index;

    //update buffer queue configuration
    if (i >= this->input_queues.size()) return;
    const size_t preload_bytes = this->input_items_sizes[i]*this->input_configs[i].preload_items;
    const size_t reserve_bytes = this->input_items_sizes[i]*this->input_configs[i].reserve_items;
    const size_t maximum_bytes = this->input_items_sizes[i]*this->input_configs[i].maximum_items;
    this->input_queues.update_config(i, this->input_items_sizes[i], preload_bytes, reserve_bytes, maximum_bytes);
}