// Copyright (C) by Josh Blum. See LICENSE.txt for licensing information.

#include <gras_impl/block_actor.hpp>

using namespace gras;

template <typename V, typename Sig>
void fill_item_sizes_from_sig(V &v, const Sig &s, const size_t size)
{
    //default item size of 1 in case we cant set
    v.resize(size, 1);

    //empty signature? maybe it was a message port
    if (s.empty()) return;

    //fill v by copying signature (with back extend mode)
    for (size_t i = 0; i < v.size(); i++) v[i] = s[i];
}

template <typename V, typename T>
void resize_fill_grow(V &v, const size_t new_len, const T &fill)
{
    if (v.size() >= new_len) return; //dont ever shrink it
    v.resize(new_len, fill);
}

template <typename V>
void resize_fill_back(V &v, const size_t new_len)
{
    if (v.empty()) v.resize(1);
    resize_fill_grow(v, new_len, v.back());
}

void BlockActor::handle_topology(
    const Apology::WorkerTopologyMessage &,
    const Theron::Address from
){
    MESSAGE_TRACER();

    const size_t num_inputs = this->get_num_inputs();
    const size_t num_outputs = this->get_num_outputs();

    //call notify_topology on block before committing settings
    this->block_ptr->notify_topology(num_inputs, num_outputs);

    //fill the item sizes from the IO signatures
    fill_item_sizes_from_sig(this->input_items_sizes, block_ptr->input_signature(), num_inputs);
    fill_item_sizes_from_sig(this->output_items_sizes, block_ptr->output_signature(), num_outputs);

    //resize and fill port properties
    resize_fill_back(this->input_configs, num_inputs);
    resize_fill_back(this->output_configs, num_outputs);

    //resize the bytes consumed/produced
    resize_fill_grow(this->items_consumed, num_inputs, 0);
    resize_fill_grow(this->items_produced, num_outputs, 0);

    //resize all work buffers to match current connections
    this->input_items.resize(num_inputs);
    this->output_items.resize(num_outputs);
    this->input_queues.resize(num_inputs);
    this->output_queues.resize(num_outputs);
    this->inputs_available.resize(num_inputs);
    if (num_inputs == 0) this->inputs_available.resize(1, true); //so its always "available"

    this->input_tokens.resize(num_inputs);
    this->output_tokens.resize(num_outputs);
    this->inputs_done.resize(num_inputs);
    this->outputs_done.resize(num_outputs);
    this->output_allocation_hints.resize(num_outputs);

    //resize tags vector to match sizes
    this->input_tags_changed.resize(num_inputs);
    this->input_tags.resize(num_inputs);

    //a block looses all connections, allow it to free
    if (num_inputs == 0 and num_outputs == 0)
    {
        this->mark_done();
    }

    for (size_t i = 0; i < num_inputs; i++)
    {
        InputUpdateMessage message;
        message.index = i;
        this->handle_input_update(message, Theron::Address());
    }
    for (size_t i = 0; i < num_outputs; i++)
    {
        OutputUpdateMessage message;
        message.index = i;
        this->handle_output_update(message, Theron::Address());
    }

    this->Send(0, from); //ACK
}