#include "go_to_parameter_controller.h"
#include <assert.h>

namespace Regression {

GoToParameterController::GoToParameterController(Responder * parentResponder, Store * store, CurveViewCursor * cursor) :
  FloatParameterController(parentResponder),
  m_abscisseCell(EditableTextMenuListCell(&m_selectableTableView, this, m_draftTextBuffer)),
  m_store(store),
  m_cursor(cursor),
  m_xPrediction(true)
{
}

void GoToParameterController::setXPrediction(bool xPrediction) {
  m_xPrediction = xPrediction;
}

const char * GoToParameterController::title() const {
  if (m_xPrediction) {
    return "Prediction sachant x";
  }
  return "Prediction sachant y";
}

float GoToParameterController::parameterAtIndex(int index) {
  assert(index == 0);
  if (m_xPrediction) {
    return m_cursor->x();
  }
  return m_cursor->y();
}

void GoToParameterController::setParameterAtIndex(int parameterIndex, float f) {
  assert(parameterIndex == 0);
  if (m_xPrediction) {
    float y = m_store->yValueForXValue(f);
    m_store->centerAxisAround(CurveViewRange::Axis::X, f);
    m_store->centerAxisAround(CurveViewRange::Axis::Y, y);
    m_store->moveCursorTo(f, y);
  } else {
    float x = m_store->xValueForYValue(f);
    m_store->centerAxisAround(CurveViewRange::Axis::X, x);
    m_store->centerAxisAround(CurveViewRange::Axis::Y, f);
    m_store->moveCursorTo(x, f);
  }
}

int GoToParameterController::numberOfRows() {
  return 1;
};

TableViewCell * GoToParameterController::reusableCell(int index) {
  assert(index == 0);
  return &m_abscisseCell;
}

int GoToParameterController::reusableCellCount() {
  return 1;
}

void GoToParameterController::willDisplayCellForIndex(TableViewCell * cell, int index) {
  EditableTextMenuListCell * myCell = (EditableTextMenuListCell *) cell;
  if (m_xPrediction) {
    myCell->setText("x");
  } else {
    myCell->setText("y");
  }
  FloatParameterController::willDisplayCellForIndex(cell, index);
}


}