/***

  Olive - Non-Linear Video Editor
  Copyright (C) 2019 Olive Team

  This program is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program.  If not, see <http://www.gnu.org/licenses/>.

***/

#include "block.h"

#include <QDebug>

Block::Block() :
  next_(nullptr)
{
  previous_input_ = new NodeInput("prev_in");
  previous_input_->set_data_type(NodeParam::kBlock);
  previous_input_->set_dependent(false);
  AddParameter(previous_input_);

  block_output_ = new NodeOutput("this_out");
  AddParameter(block_output_);

  buffer_output_ = new NodeOutput("buffer_out");
  AddParameter(buffer_output_);

  connect(this, SIGNAL(EdgeAdded(NodeEdgePtr)), this, SLOT(EdgeAddedSlot(NodeEdgePtr)), Qt::DirectConnection);
  connect(this, SIGNAL(EdgeRemoved(NodeEdgePtr)), this, SLOT(EdgeRemovedSlot(NodeEdgePtr)), Qt::DirectConnection);
}

QString Block::Category()
{
  return tr("Block");
}

const rational &Block::in()
{
  return in_point_;
}

const rational &Block::out()
{
  return out_point_;
}

const rational& Block::length()
{
  return length_;
}

void Block::set_length(const rational &length)
{
  Q_ASSERT(length > 0);

  Lock();

  length_ = length;

  Unlock();

  Refresh();
}

void Block::set_length_and_media_in(const rational &length)
{
  // Calculate media_in adjustment
  set_media_in(media_in_ + (length_ - length));

  // Set the length
  set_length(length);
}

Block *Block::previous()
{
  return ValueToPtr<Block>(previous_input_->get_value(0));
}

Block *Block::next()
{
  return next_;
}

NodeInput *Block::previous_input()
{
  return previous_input_;
}

QVariant Block::Value(NodeOutput *output, const rational &in, const rational &out)
{
  Q_UNUSED(in)
  Q_UNUSED(out)

  if (output == block_output_) {
    // Simply set the output value to a pointer to this Block
    return PtrToValue(this);
  }

  return 0;
}

void Block::EdgeAddedSlot(NodeEdgePtr edge)
{
  if (edge->input() == previous_input()) {
    static_cast<Block*>(edge->output()->parent())->next_ = this;

    // The blocks surrounding this one have changed, we need to Refresh()
    Refresh();

    // Entire track will have shifted, so the whole cache needs to be re-validated
    SendInvalidateCache(0, RATIONAL_MAX);
  }
}

void Block::EdgeRemovedSlot(NodeEdgePtr edge)
{
  if (edge->input() == previous_input()) {
    static_cast<Block*>(edge->output()->parent())->next_ = nullptr;

    // The blocks surrounding this one have changed, we need to Refresh()
    Refresh();
  }
}

void Block::Refresh()
{
  // Set in point to the out point of the previous Node
  if (previous() != nullptr) {
    in_point_ = previous()->out();
  } else {
    in_point_ = 0;
  }

  // Update out point by adding this clip's length to the just calculated in point
  out_point_ = in_point_ + length();

  emit Refreshed();

  if (next() != nullptr) {
    next()->Refresh();
  }
}

NodeOutput *Block::buffer_output()
{
  return buffer_output_;
}

NodeOutput *Block::block_output()
{
  return block_output_;
}

void Block::ConnectBlocks(Block *previous, Block *next)
{
  NodeParam::ConnectEdge(previous->block_output(), next->previous_input());
}

void Block::DisconnectBlocks(Block *previous, Block *next)
{
  NodeParam::DisconnectEdge(previous->block_output(), next->previous_input());
}

const rational &Block::media_in()
{
  return media_in_;
}

void Block::set_media_in(const rational &media_in)
{
  Lock();

  if (media_in_ != media_in) {
    media_in_ = media_in;

    // Signal that this clips contents have changed
    SendInvalidateCache(in(), out());
  }

  Unlock();
}

const QString &Block::block_name()
{
  return block_name_;
}

void Block::set_block_name(const QString &name)
{
  block_name_ = name;
}

rational Block::SequenceToMediaTime(const rational &sequence_time)
{
  // These constants are not considered "values" per se, so we don't modify them
  if (sequence_time == RATIONAL_MIN || sequence_time == RATIONAL_MAX) {
    return sequence_time;
  }

  return sequence_time - in() + media_in();
}

rational Block::MediaToSequenceTime(const rational &media_time)
{
  // These constants are not considered "values" per se, so we don't modify them
  if (media_time == RATIONAL_MIN || media_time == RATIONAL_MAX) {
    return media_time;
  }

  return media_time - media_in() + in();
}

void Block::CopyParameters(Block *source, Block *dest)
{
  dest->set_block_name(source->block_name());
  dest->set_length(source->length());
  dest->set_media_in(source->media_in());
}

void Block::Link(Block *a, Block *b)
{
  if (a == b) {
    return;
  }

  if (a == nullptr || b == nullptr) {
    return;
  }

  // Assume both clips are already linked since Link() and Unlink() should be the only entry points to this array
  if (a->linked_clips_.contains(b)) {
    return;
  }

  a->linked_clips_.append(b);
  b->linked_clips_.append(a);
}

void Block::Link(QList<Block *> blocks)
{
  foreach (Block* a, blocks) {
    foreach (Block* b, blocks) {
      Link(a, b);
    }
  }
}

void Block::Unlink(Block *a, Block *b)
{
  a->linked_clips_.removeOne(b);
  b->linked_clips_.removeOne(a);
}

bool Block::AreLinked(Block *a, Block *b)
{
  return a->linked_clips_.contains(b);
}

const QVector<Block*> &Block::linked_clips()
{
  return linked_clips_;
}

bool Block::HasLinks()
{
  return !linked_clips_.isEmpty();
}

bool Block::IsBlock()
{
  return true;
}