#include "tools/cabana/chartswidget.h"

#include <QGraphicsLayout>
#include <QGridLayout>
#include <QRubberBand>
#include <QTimer>
#include <QtCharts/QLineSeries>
#include <QtCharts/QValueAxis>

// ChartsWidget

ChartsWidget::ChartsWidget(QWidget *parent) : QWidget(parent) {
  QVBoxLayout *main_layout = new QVBoxLayout(this);
  main_layout->setContentsMargins(0, 0, 0, 0);

  // title bar
  title_bar = new QWidget(this);
  title_bar->setVisible(false);
  QHBoxLayout *title_layout = new QHBoxLayout(title_bar);
  title_layout->setContentsMargins(0, 0, 0, 0);
  title_label = new QLabel(tr("Charts"));

  title_layout->addWidget(title_label);
  title_layout->addStretch();

  range_label = new QLabel();
  title_layout->addWidget(range_label);

  reset_zoom_btn = new QPushButton("⟲", this);
  reset_zoom_btn->setFixedSize(30, 30);
  reset_zoom_btn->setToolTip(tr("Reset zoom (drag on chart to zoom X-Axis)"));
  title_layout->addWidget(reset_zoom_btn);

  remove_all_btn = new QPushButton("✖", this);
  remove_all_btn->setToolTip(tr("Remove all charts"));
  remove_all_btn->setFixedSize(30, 30);
  title_layout->addWidget(remove_all_btn);

  dock_btn = new QPushButton();
  dock_btn->setFixedSize(30, 30);
  title_layout->addWidget(dock_btn);

  main_layout->addWidget(title_bar, 0, Qt::AlignTop);

  // charts
  QWidget *charts_container = new QWidget(this);
  QVBoxLayout *charts_main = new QVBoxLayout(charts_container);
  charts_layout = new QVBoxLayout();
  charts_main->addLayout(charts_layout);
  charts_main->addStretch();

  QScrollArea *charts_scroll = new QScrollArea(this);
  charts_scroll->setWidgetResizable(true);
  charts_scroll->setWidget(charts_container);
  charts_scroll->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);

  main_layout->addWidget(charts_scroll);

  QObject::connect(dbc(), &DBCManager::DBCFileChanged, [this]() { removeAll(nullptr); });
  QObject::connect(dbc(), &DBCManager::signalRemoved, this, &ChartsWidget::removeAll);
  QObject::connect(dbc(), &DBCManager::signalUpdated, this, &ChartsWidget::signalUpdated);
  QObject::connect(dbc(), &DBCManager::msgUpdated, [this](const QString &msg_id) {
    for (auto c : charts) {
      if (c->id == msg_id) c->updateTitle();
    }
  });
  QObject::connect(can, &CANMessages::eventsMerged, this, &ChartsWidget::eventsMerged);
  QObject::connect(can, &CANMessages::updated, this, &ChartsWidget::updateState);
  QObject::connect(remove_all_btn, &QPushButton::clicked, [this]() { removeAll(); });
  QObject::connect(reset_zoom_btn, &QPushButton::clicked, this, &ChartsWidget::zoomReset);
  QObject::connect(dock_btn, &QPushButton::clicked, [this]() {
    emit dock(!docking);
    docking = !docking;
    updateTitleBar();
  });
}

void ChartsWidget::eventsMerged() {
  if (auto events = can->events(); events && !events->empty()) {
    auto it = std::find_if(events->begin(), events->end(), [=](const Event *e) { return e->which == cereal::Event::Which::CAN; });
    event_range.first = it == events->end() ? 0 : (*it)->mono_time / (double)1e9 - can->routeStartTime();
    event_range.second = it == events->end() ? 0 : events->back()->mono_time / (double)1e9 - can->routeStartTime();
    if (display_range.first == 0 && event_range.second == 0) {
      display_range.first = event_range.first;
      display_range.second = std::min(event_range.first + settings.max_chart_x_range, event_range.second);
    }
  }
}

void ChartsWidget::zoomIn(double min, double max) {
  zoomed_range = {min, max};
  is_zoomed = zoomed_range != display_range;
  updateTitleBar();
  emit rangeChanged(min, max, is_zoomed);
  updateState();
}

void ChartsWidget::zoomReset() {
  zoomIn(display_range.first, display_range.second);
}

void ChartsWidget::updateState() {
  if (charts.isEmpty()) return;

  const double current_sec = can->currentSec();
  if (is_zoomed) {
    if (current_sec < zoomed_range.first || current_sec >= zoomed_range.second) {
      can->seekTo(zoomed_range.first);
    }
  } else {
    auto prev_range = display_range;
    if (current_sec < display_range.first || current_sec >= (display_range.second - 5)) {
      // line marker reached the end, or seeked to a timestamp out of range.
      display_range.first = current_sec - 5;
    }
    display_range.first = std::max(display_range.first, event_range.first);
    display_range.second = std::min(display_range.first + settings.max_chart_x_range, event_range.second);
    if (prev_range != display_range) {
      for (auto c : charts)
        c->chart_view->updateSeries(display_range);
    }
  }

  const auto &range = is_zoomed ? zoomed_range : display_range;
  for (auto c : charts) {
    c->chart_view->setRange(range.first, range.second);
    c->chart_view->updateLineMarker(current_sec);
  }
}

void ChartsWidget::updateTitleBar() {
  title_bar->setVisible(!charts.isEmpty());
  if (charts.isEmpty()) return;

  range_label->setVisible(is_zoomed);
  reset_zoom_btn->setEnabled(is_zoomed);
  if (is_zoomed) {
    range_label->setText(tr("%1 - %2").arg(zoomed_range.first, 0, 'f', 2).arg(zoomed_range.second, 0, 'f', 2));
  }
  title_label->setText(tr("Charts (%1)").arg(charts.size()));
  dock_btn->setText(docking ? "⬈" : "⬋");
  dock_btn->setToolTip(docking ? tr("Undock charts") : tr("Dock charts"));
}

void ChartsWidget::showChart(const QString &id, const Signal *sig, bool show) {
  auto it = std::find_if(charts.begin(), charts.end(), [=](auto c) { return c->id == id && c->signal == sig; });
  if (it != charts.end()) {
    if (!show) removeChart((*it));
  } else if (show) {
    auto chart = new ChartWidget(id, sig, this);
    chart->chart_view->updateSeries(display_range);
    QObject::connect(chart, &ChartWidget::remove, [=]() { removeChart(chart); });
    QObject::connect(chart->chart_view, &ChartView::zoomIn, this, &ChartsWidget::zoomIn);
    QObject::connect(chart->chart_view, &ChartView::zoomReset, this, &ChartsWidget::zoomReset);
    charts_layout->insertWidget(0, chart);
    charts.push_back(chart);
    emit chartOpened(chart->id, chart->signal);
    updateState();
  }
  updateTitleBar();
}

bool ChartsWidget::isChartOpened(const QString &id, const Signal *sig) {
  auto it = std::find_if(charts.begin(), charts.end(), [=](auto c) { return c->id == id && c->signal == sig; });
  return it != charts.end();
}

void ChartsWidget::removeChart(ChartWidget *chart) {
  charts.removeOne(chart);
  chart->deleteLater();
  updateTitleBar();
  emit chartClosed(chart->id, chart->signal);
}

void ChartsWidget::removeAll(const Signal *sig) {
  QMutableListIterator<ChartWidget *> it(charts);
  while (it.hasNext()) {
    auto c = it.next();
    if (sig == nullptr || c->signal == sig) {
      c->deleteLater();
      emit chartClosed(c->id, c->signal);
      it.remove();
    }
  }
  updateTitleBar();
}

void ChartsWidget::signalUpdated(const Signal *sig) {
  for (auto c : charts) {
    if (c->signal == sig) {
      c->updateTitle();
      c->chart_view->updateSeries(display_range);
      c->chart_view->setRange(display_range.first, display_range.second, true);
    }
  }
}

bool ChartsWidget::eventFilter(QObject *obj, QEvent *event) {
  if (obj != this && event->type() == QEvent::Close) {
    emit dock_btn->clicked();
    return true;
  }
  return false;
}

// ChartWidget

ChartWidget::ChartWidget(const QString &id, const Signal *sig, QWidget *parent) : id(id), signal(sig), QWidget(parent) {
  QVBoxLayout *main_layout = new QVBoxLayout(this);
  main_layout->setSpacing(0);
  main_layout->setContentsMargins(0, 0, 0, 0);

  header = new QWidget(this);
  QGridLayout *header_layout = new QGridLayout(header);
  header_layout->setContentsMargins(11, 11, 11, 0);
  msg_name_label = new QLabel(this);
  msg_name_label->setTextFormat(Qt::RichText);
  header_layout->addWidget(msg_name_label, 0, 0, Qt::AlignLeft);
  sig_name_label = new QLabel(this);
  header_layout->addWidget(sig_name_label, 0, 1, Qt::AlignCenter);  //, 0, Qt::AlignCenter);

  remove_btn = new QPushButton("✖", this);
  remove_btn->setFixedSize(20, 20);
  remove_btn->setToolTip(tr("Remove chart"));
  header_layout->addWidget(remove_btn, 0, 2, Qt::AlignRight);
  main_layout->addWidget(header);

  chart_view = new ChartView(id, sig, this);
  main_layout->addWidget(chart_view);
  main_layout->addStretch();

  setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);
  updateTitle();
  updateFromSettings();

  QObject::connect(remove_btn, &QPushButton::clicked, [=]() { emit remove(id, sig); });
  QObject::connect(&settings, &Settings::changed, this, &ChartWidget::updateFromSettings);
}

void ChartWidget::updateTitle() {
  msg_name_label->setText(tr("%1 <font color=\"gray\">%2</font>").arg(dbc()->msg(id)->name.c_str()).arg(id));
  sig_name_label->setText(signal->name.c_str());
}

void ChartWidget::updateFromSettings() {
  header->setStyleSheet(settings.chart_theme == 0 ? "background-color:white" : "background-color:#23242c");
  QString color_style = settings.chart_theme == 0 ? "color:black" : "color:white";
  sig_name_label->setStyleSheet("font-weight:bold;" + color_style);
  msg_name_label->setStyleSheet(color_style);
  remove_btn->setStyleSheet(color_style);
  chart_view->updateFromSettings();
}

// ChartView

ChartView::ChartView(const QString &id, const Signal *sig, QWidget *parent)
    : id(id), signal(sig), QChartView(nullptr, parent) {
  QLineSeries *series = new QLineSeries();
  QChart *chart = new QChart();
  chart->setBackgroundRoundness(0);
  chart->addSeries(series);
  chart->createDefaultAxes();
  chart->legend()->hide();
  chart->setMargins({0, 0, 0, 0});
  chart->layout()->setContentsMargins(0, 0, 0, 0);

  track_line = new QGraphicsLineItem(chart);
  track_line->setZValue(chart->zValue() + 10);
  track_line->setPen(QPen(Qt::darkGray, 1, Qt::DashLine));
  track_ellipse = new QGraphicsEllipseItem(chart);
  track_ellipse->setZValue(chart->zValue() + 10);
  track_ellipse->setBrush(Qt::darkGray);
  value_text = new QGraphicsTextItem(chart);
  value_text->setZValue(chart->zValue() + 10);
  line_marker = new QGraphicsLineItem(chart);
  line_marker->setZValue(chart->zValue() + 10);

  setChart(chart);

  setRenderHint(QPainter::Antialiasing);
  setRubberBand(QChartView::HorizontalRubberBand);
  if (auto rubber = findChild<QRubberBand *>()) {
    QPalette pal;
    pal.setBrush(QPalette::Base, QColor(0, 0, 0, 80));
    rubber->setPalette(pal);
  }

  QTimer *timer = new QTimer(this);
  timer->setInterval(100);
  timer->setSingleShot(true);
  timer->callOnTimeout(this, &ChartView::adjustChartMargins);

  QObject::connect(chart, &QChart::plotAreaChanged, [=](const QRectF &plotArea) {
    // use a singleshot timer to avoid recursion call.
    timer->start();
  });
}

void ChartView::updateFromSettings() {
  setFixedHeight(settings.chart_height);
  chart()->setTheme(settings.chart_theme == 0 ? QChart::ChartThemeLight : QChart::QChart::ChartThemeDark);
  line_marker->setPen(QPen(settings.chart_theme == 0 ? Qt::black : Qt::white, 2));
}

void ChartView::setRange(double min, double max, bool force_update) {
  auto axis_x = dynamic_cast<QValueAxis *>(chart()->axisX());
  if (force_update || (min != axis_x->min() || max != axis_x->max())) {
    axis_x->setRange(min, max);
    updateAxisY();
  }
}

void ChartView::adjustChartMargins() {
  // TODO: Remove hardcoded aligned_pos
  const int aligned_pos = 60;
  if (chart()->plotArea().left() != aligned_pos) {
    const float left_margin = chart()->margins().left() + aligned_pos - chart()->plotArea().left();
    chart()->setMargins(QMargins(left_margin, 0, 0, 0));
  }
}

void ChartView::updateLineMarker(double current_sec) {
  auto axis_x = dynamic_cast<QValueAxis *>(chart()->axisX());
  int x = chart()->plotArea().left() +
          chart()->plotArea().width() * (current_sec - axis_x->min()) / (axis_x->max() - axis_x->min());
  if (int(line_marker->line().x1()) != x) {
    line_marker->setLine(x, 0, x, height());
    chart()->update();
  }
}

void ChartView::updateSeries(const std::pair<double, double> &range) {
  auto events = can->events();
  if (!events) return;

  auto l = id.split(':');
  int bus = l[0].toInt();
  uint32_t address = l[1].toUInt(nullptr, 16);

  vals.clear();
  vals.reserve((range.second - range.first) * 100);  // [n]minutes * 100hz
  double route_start_time = can->routeStartTime();
  Event begin_event(cereal::Event::Which::INIT_DATA, (route_start_time + range.first) * 1e9);
  auto begin = std::lower_bound(events->begin(), events->end(), &begin_event, Event::lessThan());
  double end_ns = (route_start_time + range.second) * 1e9;
  for (auto it = begin; it != events->end() && (*it)->mono_time <= end_ns; ++it) {
    if ((*it)->which == cereal::Event::Which::CAN) {
      for (auto c : (*it)->event.getCan()) {
        if (bus == c.getSrc() && address == c.getAddress()) {
          auto dat = c.getDat();
          double value = get_raw_value((uint8_t *)dat.begin(), dat.size(), *signal);
          double ts = ((*it)->mono_time / (double)1e9) - route_start_time;  // seconds
          vals.push_back({ts, value});
        }
      }
    }
  }
  QLineSeries *series = (QLineSeries *)chart()->series()[0];
  series->replace(vals);
}

// auto zoom on yaxis
void ChartView::updateAxisY() {
  const auto axis_x = dynamic_cast<QValueAxis *>(chart()->axisX());
  const auto axis_y = dynamic_cast<QValueAxis *>(chart()->axisY());
  auto begin = std::lower_bound(vals.begin(), vals.end(), axis_x->min(), [](auto &p, double x) { return p.x() < x; });
  if (begin == vals.end())
    return;

  auto end = std::upper_bound(vals.begin(), vals.end(), axis_x->max(), [](double x, auto &p) { return x < p.x(); });
  const auto [min, max] = std::minmax_element(begin, end, [](auto &p1, auto &p2) { return p1.y() < p2.y(); });
  (min->y() == max->y()) ? axis_y->setRange(min->y() - 1, max->y() + 1)
                         : axis_y->setRange(min->y(), max->y());
}

void ChartView::enterEvent(QEvent *event) {
  track_line->setVisible(true);
  value_text->setVisible(true);
  track_ellipse->setVisible(true);
  QChartView::enterEvent(event);
}

void ChartView::leaveEvent(QEvent *event) {
  track_line->setVisible(false);
  value_text->setVisible(false);
  track_ellipse->setVisible(false);
  QChartView::leaveEvent(event);
}

void ChartView::mouseReleaseEvent(QMouseEvent *event) {
  auto rubber = findChild<QRubberBand *>();
  if (event->button() == Qt::LeftButton && rubber && rubber->isVisible()) {
    rubber->hide();
    QRectF rect = rubber->geometry().normalized();
    rect.translate(-chart()->plotArea().topLeft());
    const auto axis_x = dynamic_cast<QValueAxis *>(chart()->axisX());
    double min = axis_x->min() + (rect.left() / chart()->plotArea().width()) * (axis_x->max() - axis_x->min());
    double max = axis_x->min() + (rect.right() / chart()->plotArea().width()) * (axis_x->max() - axis_x->min());
    if (rubber->width() <= 0) {
      // no rubber dragged, seek to mouse position
      can->seekTo(min);
    } else if ((max - min) >= 0.5) {
      // zoom in if selected range is greater than 0.5s
      emit zoomIn(min, max);
    }
    event->accept();
  } else if (event->button() == Qt::RightButton) {
    emit zoomReset();
    event->accept();
  } else {
    QGraphicsView::mouseReleaseEvent(event);
  }
}

void ChartView::mouseMoveEvent(QMouseEvent *ev) {
  auto rubber = findChild<QRubberBand *>();
  bool is_zooming = rubber && rubber->isVisible();
  if (!is_zooming) {
    const auto plot_area = chart()->plotArea();
    auto axis_x = dynamic_cast<QValueAxis *>(chart()->axisX());
    double x = std::clamp((double)ev->pos().x(), plot_area.left(), plot_area.right()-1);
    double sec = axis_x->min() + (axis_x->max() - axis_x->min()) * (x - plot_area.left()) / plot_area.width();
    auto value = std::upper_bound(vals.begin(), vals.end(), sec, [](double x, auto &p) { return x < p.x(); });
    if (value != vals.end()) {
      QPointF pos = chart()->mapToPosition((*value));
      track_line->setLine(pos.x(), plot_area.top(), pos.x(), plot_area.bottom());
      track_ellipse->setRect(pos.x() - 5, pos.y() - 5, 10, 10);
      value_text->setHtml(tr("<div style='background-color:darkGray'><font color='white'>%1, %2)</font></div>")
                              .arg(value->x(), 0, 'f', 3).arg(value->y()));
      int text_x = pos.x() + 8;
      if ((text_x + value_text->boundingRect().width()) > plot_area.right()) {
        text_x = pos.x() - value_text->boundingRect().width() - 8;
      }
      value_text->setPos(text_x, pos.y() - 10);
    }
    track_line->setVisible(value != vals.end());
    value_text->setVisible(value != vals.end());
    track_ellipse->setVisible(value != vals.end());
  }
  QChartView::mouseMoveEvent(ev);
}