#include "multiple_data_view_controller.h"
#include "../i18n.h"
#include <assert.h>

using namespace Shared;

namespace Statistics {

MultipleDataViewController::MultipleDataViewController(Responder * parentResponder, Store * store, int * selectedBarIndex) :
  ViewController(parentResponder),
  m_store(store),
  m_selectedSeries(-1),
  m_selectedBarIndex(selectedBarIndex)
{
}

MultipleDataView * MultipleDataViewController::multipleDataView() {
  return static_cast<MultipleDataView *>(view());
}

bool MultipleDataViewController::isEmpty() const {
  return m_store->isEmpty();
}

I18n::Message MultipleDataViewController::emptyMessage() {
  return I18n::Message::NoDataToPlot;
}

Responder * MultipleDataViewController::defaultController() {
  return tabController();
}

void MultipleDataViewController::viewWillAppear() {
  multipleDataView()->setDisplayBanner(true);
  if (m_selectedSeries < 0) {
    m_selectedSeries = multipleDataView()->seriesOfSubviewAtIndex(0);
    multipleDataView()->selectDataView(m_selectedSeries);
  }
  reloadBannerView();
  multipleDataView()->reload();
}

bool MultipleDataViewController::handleEvent(Ion::Events::Event event) {
  assert(m_selectedSeries >= 0);
  if (event == Ion::Events::Down) {
    int currentSelectedSubview = multipleDataView()->indexOfSubviewAtSeries(m_selectedSeries);
    if (currentSelectedSubview < multipleDataView()->numberOfSubviews() - 2) {
      multipleDataView()->deselectDataView(m_selectedSeries);
      m_selectedSeries = multipleDataView()->seriesOfSubviewAtIndex(currentSelectedSubview+1);
      *m_selectedBarIndex = 0;
      multipleDataView()->selectDataView(m_selectedSeries);
      reloadBannerView();
      multipleDataView()->reload();
      app()->setFirstResponder(this);
      return true;
    }
    return false;
  }
  if (event == Ion::Events::Up) {
    int currentSelectedSubview = multipleDataView()->indexOfSubviewAtSeries(m_selectedSeries);
    if (currentSelectedSubview > 0) {
      multipleDataView()->deselectDataView(m_selectedSeries);
      assert(currentSelectedSubview > 0);
      m_selectedSeries = multipleDataView()->seriesOfSubviewAtIndex(currentSelectedSubview-1);
      multipleDataView()->selectDataView(m_selectedSeries);
      *m_selectedBarIndex = 0;
      app()->setFirstResponder(this);
    } else {
      app()->setFirstResponder(tabController());
    }
    reloadBannerView();
    multipleDataView()->reload();
    return true;
  }
  if (m_selectedSeries >= 0 && (event == Ion::Events::Left || event == Ion::Events::Right)) {
    int direction = event == Ion::Events::Left ? -1 : 1;
    if (moveSelectionHorizontally(direction)) {
      reloadBannerView();
      multipleDataView()->reload();
    }
    return true;
  }
  return false;
}

void MultipleDataViewController::didBecomeFirstResponder() {
  multipleDataView()->setDisplayBanner(true);
  if (m_selectedSeries < 0 || m_store->sumOfOccurrences(m_selectedSeries) == 0) {
    if (m_selectedSeries >= 0) {
      multipleDataView()->deselectDataView(m_selectedSeries);
    }
    m_selectedSeries = multipleDataView()->seriesOfSubviewAtIndex(0);
    multipleDataView()->selectDataView(m_selectedSeries);
    multipleDataView()->reload();
  }
}

void MultipleDataViewController::willExitResponderChain(Responder * nextFirstResponder) {
  if (nextFirstResponder == nullptr || nextFirstResponder == tabController()) {
    if (m_selectedSeries >= 0) {
      multipleDataView()->dataViewAtIndex(m_selectedSeries)->selectMainView(false);
      m_selectedSeries = -1;
      multipleDataView()->setDisplayBanner(false);
    }
    multipleDataView()->reload();
  }
}

}