#include <string>
#include <list>
#include <vector>
#include <iostream>
#include "tbb/pipeline.h"

#include "pipeline.hh"
#include "compose.hh"
#include "filterSample/grey.hh"
#include "filterSample/cartoon.hh"
#include "filterSample/blur.hh"
#include "filterSample/reverse.hh"
#include "filterSample/merge.hh"
#include "filterSample/interlace.hh"
#include "filterSample/inlay.hh"
#include "filterSample/count.hh"

Pipeline::Pipeline(std::list<VideoReader> input, VideoWriter output)
  : input_(input), output_(output)
{
  this->input_it_ = this->input_.begin();
  this->pipeline_.add_filter(this->input_.front());
  this->input_it_++;
}

bool Pipeline::add_filter(std::string filter, std::vector<int> args)
{
  if (filter == "grey" || filter == "gray")
    this->pipeline_.add_filter(*(new Grey()));
  else if (filter == "reverse" || filter == "inverse")
    this->pipeline_.add_filter(*(new Reverse()));
  else if (filter == "cartoon")
    this->pipeline_.add_filter(*(new Cartoon()));
  else if (filter == "count")
    this->pipeline_.add_filter(*(new Count()));
  else if (filter == "blur")
  {
    if (args.size() >= 1)
      this->pipeline_.add_filter(*(new Blur(args[0])));
    else
      this->pipeline_.add_filter(*(new Blur()));
  }
  else if (filter == "merge" && this->input_it_ != this->input_.end())
  {
    this->pipeline_.add_filter(*(new Compose(*this->input_it_)));
    this->input_it_++;
    this->pipeline_.add_filter(*(new Merge()));
  }
  else if (filter == "interlace" && this->input_it_ != this->input_.end())
  {
    this->pipeline_.add_filter(*(new Compose(*this->input_it_)));
    this->input_it_++;
    this->pipeline_.add_filter(*(new Interlace()));
  }
  else if (filter == "inlay" && this->input_it_ != this->input_.end()
    && args.size() >= 3)
  {
    this->pipeline_.add_filter(*(new Compose(*this->input_it_)));
    this->input_it_++;
    this->pipeline_.add_filter(
      *(new Inlay(cv::Vec3b {
            (unsigned char) args[0],
            (unsigned char) args[1],
            (unsigned char) args[2]
          }, 0)));
  }
  else if (filter == "strictinlay" && this->input_it_ != this->input_.end()
    && args.size() >= 3)
  {
    this->pipeline_.add_filter(*(new Compose(*this->input_it_)));
    this->input_it_++;
    this->pipeline_.add_filter(
      *(new Inlay(cv::Vec3b {
            (unsigned char) args[0],
            (unsigned char) args[1],
            (unsigned char) args[2]
          }, 1)));
  }
  else
    return false;

  return true;
}

void Pipeline::run()
{
  std::cout << "Applying video filter chain: ";
  std::flush(std::cout);
  this->pipeline_.add_filter(this->output_);
  this->pipeline_.run(CHUNK_SIZE);
  std::cout << "Done!" << std::endl;
}