#include "ADSREditor.hpp"
#include "MainWindow.hpp"
#include <QVBoxLayout>
#include <QPainter>
#include <QMouseEvent>

static QColor Red = QColor(255, 127, 127);
static QColor Green = QColor(127, 255, 127);
static QColor Blue = QColor(150, 150, 255);

ADSREditor* ADSRView::getEditor() const { return qobject_cast<ADSREditor*>(parentWidget()); }

void ADSRView::loadData(ProjectModel::ADSRNode* node) { m_node = node; }

void ADSRView::unloadData() { m_node.reset(); }

ProjectModel::INode* ADSRView::currentNode() const { return m_node.get(); }

void ADSRView::paintEvent(QPaintEvent* ev) {
  if (!m_node)
    return;

  QPainter painter(this);
  painter.setRenderHint(QPainter::Antialiasing);

  amuse::ITable& table = **m_node->m_obj;
  qreal adTime = 0.0;
  qreal aTime = 0.0;
  qreal relTime = 0.0;
  qreal sustain = 0.0;
  if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
    amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
    adTime += adsr.getAttack();
    aTime = adsr.getAttack();
    adTime += adsr.getDecay();
    sustain = adsr.getSustain();
    relTime = adsr.getRelease();
  } else if (table.Isa() == amuse::ITable::Type::ADSR) {
    amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
    adTime += adsr.getAttack();
    aTime = adsr.getAttack();
    adTime += adsr.getDecay();
    sustain = adsr.getSustain();
    relTime = adsr.getRelease();
  }
  qreal totalTime = adTime + 1.0 + relTime;

  qreal deviceRatio = devicePixelRatioF();
  qreal penWidth = std::max(std::floor(deviceRatio), 1.0) / deviceRatio;

  painter.setPen(QPen(QColor(127, 127, 127), penWidth));
  painter.setFont(m_gridFont);
  qreal yIncrement = (height() - 16.0) / 10.0;
  for (int i = 0; i < 11; ++i) {
    qreal thisY = i * yIncrement;
    qreal textY = thisY - (i == 0 ? 2.0 : (i == 10 ? 16.0 : 8.0));
    painter.drawStaticText(QPointF(0.0, textY), m_percentTexts[i]);
    painter.drawLine(QPointF(30.0, thisY), QPointF(width(), thisY));
  }

  size_t maxTime = size_t(std::max(adTime, relTime)) + 1;
  if (m_timeTexts.size() < maxTime) {
    m_timeTexts.reserve(maxTime);
    for (size_t i = m_timeTexts.size(); i < maxTime; ++i) {
      m_timeTexts.emplace_back();
      QStaticText& text = m_timeTexts.back();
      text.setText(QStringLiteral("%1.0s").arg(i));
      text.setTextWidth(32.0);
      text.setTextOption(QTextOption(Qt::AlignHCenter | Qt::AlignBaseline));
    }
  }
  qreal xBase = 30.0;
  qreal xIncrement = (width() - xBase) / totalTime;
  for (size_t i = 0; i <= size_t(adTime); ++i) {
    qreal thisX = i * xIncrement + xBase;
    qreal textX = thisX - 15.0;
    painter.drawStaticText(QPointF(textX, height() - 16.0), m_timeTexts[i]);
    painter.drawLine(QPointF(thisX, 0.0), QPointF(thisX, height() - 16.0));
  }
  xBase = (width() - 30.0) * ((adTime + 1.0) / totalTime) + 30.0;
  for (size_t i = 0; i <= size_t(relTime); ++i) {
    qreal thisX = i * xIncrement + xBase;
    qreal textX = thisX - 15.0;
    painter.drawStaticText(QPointF(textX, height() - 16.0), m_timeTexts[i]);
    painter.drawLine(QPointF(thisX, 0.0), QPointF(thisX, height() - 16.0));
  }

  qreal sustainY = (height() - 16.0) - (height() - 16.0) * sustain;
  QPointF pt0 = QPointF(30.0, height() - 16.0);
  QPointF pt1 = QPointF((width() - 30.0) * (aTime / totalTime) + 30.0, 0.0);
  QPointF pt2 = QPointF((width() - 30.0) * (adTime / totalTime) + 30.0, sustainY);
  QPointF pt3 = QPointF((width() - 30.0) * ((adTime + 1.0) / totalTime) + 30.0, sustainY);
  QPointF pt4 = QPointF(width(), height() - 16.0);

  painter.setPen(QPen(Red, penWidth * 3.0));
  painter.drawLine(pt0, pt1);
  painter.setPen(QPen(Green, penWidth * 3.0));
  painter.drawLine(pt1, pt2);
  painter.setPen(QPen(Qt::white, penWidth * 3.0));
  painter.drawLine(pt2, pt3);
  painter.setPen(QPen(Blue, penWidth * 3.0));
  painter.drawLine(pt3, pt4);
  painter.setPen(Qt::NoPen);
  painter.setBrush(Green);
  painter.drawEllipse(pt1, 8.0, 8.0);
  painter.setBrush(Qt::white);
  painter.drawEllipse(pt2, 8.0, 8.0);
  painter.setBrush(Blue);
  painter.drawEllipse(pt3, 8.0, 8.0);
}

static qreal PointDistanceSq(const QPointF& p1, const QPointF& p2) {
  QPointF delta = p1 - p2;
  return QPointF::dotProduct(delta, delta);
}

static qreal PointDistance(const QPointF& p1, const QPointF& p2) { return std::sqrt(PointDistanceSq(p1, p2)); }

void ADSRView::mousePressEvent(QMouseEvent* ev) {
  if (!m_node)
    return;

  amuse::ITable& table = **m_node->m_obj;
  qreal adTime = 0.0;
  qreal aTime = 0.0;
  qreal relTime = 0.0;
  qreal sustain = 0.0;
  if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
    amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
    adTime += adsr.getAttack();
    aTime = adsr.getAttack();
    adTime += adsr.getDecay();
    sustain = adsr.getSustain();
    relTime = adsr.getRelease();
  } else if (table.Isa() == amuse::ITable::Type::ADSR) {
    amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
    adTime += adsr.getAttack();
    aTime = adsr.getAttack();
    adTime += adsr.getDecay();
    sustain = adsr.getSustain();
    relTime = adsr.getRelease();
  }
  qreal totalTime = adTime + 1.0 + relTime;

  qreal sustainY = (height() - 16.0) - (height() - 16.0) * sustain;
  QPointF points[] = {QPointF((width() - 30.0) * (aTime / totalTime) + 30.0, 0.0),
                      QPointF((width() - 30.0) * (adTime / totalTime) + 30.0, sustainY),
                      QPointF((width() - 30.0) * ((adTime + 1.0) / totalTime) + 30.0, sustainY)};
  qreal dists[] = {PointDistance(ev->localPos(), points[0]), PointDistance(ev->localPos(), points[1]),
                   PointDistance(ev->localPos(), points[2])};

  int minDist = 0;
  if (dists[1] < dists[minDist] + 8.0) /* pt1 overlaps pt0, so include radius in test */
    minDist = 1;
  if (dists[2] < dists[minDist])
    minDist = 2;

  if (dists[minDist] < 10.0) {
    ++m_cycleIdx;
    m_dragPoint = minDist;
    mouseMoveEvent(ev);
  }
}

void ADSRView::mouseReleaseEvent(QMouseEvent* ev) { m_dragPoint = -1; }

void ADSRView::mouseMoveEvent(QMouseEvent* ev) {
  amuse::ITable& table = **m_node->m_obj;
  ADSRControls* ctrls = getEditor()->m_controls;

  qreal adTime = 0.0;
  qreal aTime = 0.0;
  qreal relTime = 0.0;
  if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
    amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
    adTime += adsr.getAttack();
    aTime = adsr.getAttack();
    adTime += adsr.getDecay();
    relTime = adsr.getRelease();
  } else if (table.Isa() == amuse::ITable::Type::ADSR) {
    amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
    adTime += adsr.getAttack();
    aTime = adsr.getAttack();
    adTime += adsr.getDecay();
    relTime = adsr.getRelease();
  }
  qreal totalTime = adTime + 1.0 + relTime;

  if (m_dragPoint == 0) {
    qreal newAttack = std::max(0.0, (ev->localPos().x() - 30.0) / (width() - 30.0) * totalTime);
    qreal delta = newAttack - aTime;
    ctrls->setAttackAndDecay(newAttack, std::max(0.0, ctrls->m_decay->value() - delta), m_cycleIdx);
  } else if (m_dragPoint == 1) {
    qreal newDecay = std::max(0.0, (ev->localPos().x() - 30.0) * totalTime / (width() - 30.0) - aTime);
    qreal newSustain = (-ev->localPos().y() + (height() - 16.0)) / (height() - 16.0);
    ctrls->setDecayAndSustain(newDecay, newSustain * 100.0, m_cycleIdx);
  } else if (m_dragPoint == 2) {
    qreal newRelease = std::max(0.0, (width() - 30.0) * (adTime + 1.0) / (ev->localPos().x() - 30.0) - (adTime + 1.0));
    ctrls->setRelease(newRelease, m_cycleIdx);
    ctrls->m_release->setValue(newRelease);
  }
}

ADSRView::ADSRView(QWidget* parent) : QWidget(parent) {
  for (int i = 0; i < 11; ++i) {
    m_percentTexts[i].setText(QStringLiteral("%1%").arg(100 - i * 10));
    m_percentTexts[i].setTextOption(QTextOption(Qt::AlignVCenter | Qt::AlignRight));
    m_percentTexts[i].setTextWidth(28.0);
  }
  m_gridFont.setPointSize(8);
}

ADSREditor* ADSRControls::getEditor() const { return qobject_cast<ADSREditor*>(parentWidget()); }

void ADSRControls::loadData() {
  m_enableUpdate = false;

  amuse::ITable& table = **getEditor()->m_adsrView->m_node->m_obj;
  m_attack->setDisabled(false);
  m_decay->setDisabled(false);
  m_sustain->setDisabled(false);
  m_release->setDisabled(false);
  m_dls->setDisabled(false);
  if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
    amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
    m_attack->setValue(adsr.getAttack());
    m_decay->setValue(adsr.getDecay());
    m_sustain->setValue(adsr.getSustain() * 100.0);
    m_release->setValue(adsr.getRelease());
    m_dls->setChecked(true);
    m_velToAttackLab->setVisible(true);
    m_velToAttack->setVisible(true);
    m_velToAttack->setValue(adsr._getVelToAttack());
    m_keyToDecayLab->setVisible(true);
    m_keyToDecay->setVisible(true);
    m_keyToDecay->setValue(adsr._getKeyToDecay());
  } else if (table.Isa() == amuse::ITable::Type::ADSR) {
    amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
    m_attack->setValue(adsr.getAttack());
    m_decay->setValue(adsr.getDecay());
    m_sustain->setValue(adsr.getSustain() * 100.0);
    m_release->setValue(adsr.getRelease());
    m_dls->setChecked(false);
    m_velToAttackLab->setVisible(false);
    m_velToAttack->setVisible(false);
    m_keyToDecayLab->setVisible(false);
    m_keyToDecay->setVisible(false);
  } else {
    unloadData();
  }

  m_enableUpdate = true;
}

void ADSRControls::unloadData() {
  m_attack->setDisabled(true);
  m_decay->setDisabled(true);
  m_sustain->setDisabled(true);
  m_release->setDisabled(true);
  m_dls->setDisabled(true);
  m_velToAttackLab->setVisible(false);
  m_velToAttack->setVisible(false);
  m_keyToDecayLab->setVisible(false);
  m_keyToDecay->setVisible(false);
}

class ADSRAttackUndoCommand : public EditorUndoCommand {
  double m_redoVal, m_undoVal;
  bool m_undid = false;

public:
  ADSRAttackUndoCommand(double redoVal, amuse::ObjToken<ProjectModel::ADSRNode> node)
  : EditorUndoCommand(node.get(), ADSRControls::tr("Change Attack")), m_redoVal(redoVal) {}
  void undo() {
    m_undid = true;
    amuse::ITable& table = **m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
      amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
      adsr.setAttack(m_undoVal);
    } else if (table.Isa() == amuse::ITable::Type::ADSR) {
      amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
      adsr.setAttack(m_undoVal);
    }
    EditorUndoCommand::undo();
  }
  void redo() {
    amuse::ITable& table = **m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
      amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
      m_undoVal = adsr.getAttack();
      adsr.setAttack(m_redoVal);
    } else if (table.Isa() == amuse::ITable::Type::ADSR) {
      amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
      m_undoVal = adsr.getAttack();
      adsr.setAttack(m_redoVal);
    }
    if (m_undid)
      EditorUndoCommand::redo();
  }
  bool mergeWith(const QUndoCommand* other) {
    if (other->id() == id()) {
      m_redoVal = static_cast<const ADSRAttackUndoCommand*>(other)->m_redoVal;
      return true;
    }
    return false;
  }
  int id() const { return int(Id::ADSRAttack); }
};

void ADSRControls::attackChanged(double val) {
  if (m_enableUpdate) {
    ADSRView* view = getEditor()->m_adsrView;
    g_MainWindow->pushUndoCommand(new ADSRAttackUndoCommand(val, view->m_node));
    view->update();
  }
}

class ADSRDecayUndoCommand : public EditorUndoCommand {
  double m_redoVal, m_undoVal;
  bool m_undid = false;

public:
  ADSRDecayUndoCommand(double redoVal, amuse::ObjToken<ProjectModel::ADSRNode> node)
  : EditorUndoCommand(node.get(), ADSRControls::tr("Change Decay")), m_redoVal(redoVal) {}
  void undo() {
    m_undid = true;
    amuse::ITable& table = **m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
      amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
      adsr.setDecay(m_undoVal);
    } else if (table.Isa() == amuse::ITable::Type::ADSR) {
      amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
      adsr.setDecay(m_undoVal);
    }
    EditorUndoCommand::undo();
  }
  void redo() {
    amuse::ITable& table = **m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
      amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
      m_undoVal = adsr.getDecay();
      adsr.setDecay(m_redoVal);
    } else if (table.Isa() == amuse::ITable::Type::ADSR) {
      amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
      m_undoVal = adsr.getDecay();
      adsr.setDecay(m_redoVal);
    }
    if (m_undid)
      EditorUndoCommand::redo();
  }
  bool mergeWith(const QUndoCommand* other) {
    if (other->id() == id()) {
      m_redoVal = static_cast<const ADSRDecayUndoCommand*>(other)->m_redoVal;
      return true;
    }
    return false;
  }
  int id() const { return int(Id::ADSRDecay); }
};

void ADSRControls::decayChanged(double val) {
  if (m_enableUpdate) {
    ADSRView* view = getEditor()->m_adsrView;
    g_MainWindow->pushUndoCommand(new ADSRDecayUndoCommand(val, view->m_node));
    view->update();
  }
}

class ADSRSustainUndoCommand : public EditorUndoCommand {
  double m_redoVal, m_undoVal;
  bool m_undid = false;

public:
  ADSRSustainUndoCommand(double redoVal, amuse::ObjToken<ProjectModel::ADSRNode> node)
  : EditorUndoCommand(node.get(), ADSRControls::tr("Change Sustain")), m_redoVal(redoVal) {}
  void undo() {
    m_undid = true;
    amuse::ITable& table = **m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
      amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
      adsr.setSustain(m_undoVal);
    } else if (table.Isa() == amuse::ITable::Type::ADSR) {
      amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
      adsr.setSustain(m_undoVal);
    }
    EditorUndoCommand::undo();
  }
  void redo() {
    amuse::ITable& table = **m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
      amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
      m_undoVal = adsr.getSustain();
      adsr.setSustain(m_redoVal);
    } else if (table.Isa() == amuse::ITable::Type::ADSR) {
      amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
      m_undoVal = adsr.getSustain();
      adsr.setSustain(m_redoVal);
    }
    if (m_undid)
      EditorUndoCommand::redo();
  }
  bool mergeWith(const QUndoCommand* other) {
    if (other->id() == id()) {
      m_redoVal = static_cast<const ADSRSustainUndoCommand*>(other)->m_redoVal;
      return true;
    }
    return false;
  }
  int id() const { return int(Id::ADSRSustain); }
};

void ADSRControls::sustainChanged(double val) {
  if (m_enableUpdate) {
    ADSRView* view = getEditor()->m_adsrView;
    g_MainWindow->pushUndoCommand(new ADSRSustainUndoCommand(val / 100.0, view->m_node));
    view->update();
  }
}

class ADSRAttackAndDecayUndoCommand : public EditorUndoCommand {
  double m_redoAttack, m_redoDecay;
  double m_undoAttack, m_undoDecay;
  uint64_t m_cycleCount;
  bool m_undid = false;

public:
  ADSRAttackAndDecayUndoCommand(double redoAttack, double redoDecay, uint64_t cycleCount,
                                amuse::ObjToken<ProjectModel::ADSRNode> node)
  : EditorUndoCommand(node.get(), ADSRControls::tr("Change Attack/Decay"))
  , m_redoAttack(redoAttack)
  , m_redoDecay(redoDecay)
  , m_cycleCount(cycleCount) {}
  void undo() {
    m_undid = true;
    amuse::ITable& table = **m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
      amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
      adsr.setAttack(m_undoAttack);
      adsr.setDecay(m_undoDecay);
    } else if (table.Isa() == amuse::ITable::Type::ADSR) {
      amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
      adsr.setAttack(m_undoAttack);
      adsr.setDecay(m_undoDecay);
    }
    EditorUndoCommand::undo();
  }
  void redo() {
    amuse::ITable& table = **m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
      amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
      m_undoAttack = adsr.getAttack();
      m_undoDecay = adsr.getDecay();
      adsr.setAttack(m_redoAttack);
      adsr.setDecay(m_redoDecay);
    } else if (table.Isa() == amuse::ITable::Type::ADSR) {
      amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
      m_undoAttack = adsr.getAttack();
      m_undoDecay = adsr.getDecay();
      adsr.setAttack(m_redoAttack);
      adsr.setDecay(m_redoDecay);
    }
    if (m_undid)
      EditorUndoCommand::redo();
  }
  bool mergeWith(const QUndoCommand* other) {
    if (other->id() == id() && m_cycleCount == static_cast<const ADSRAttackAndDecayUndoCommand*>(other)->m_cycleCount) {
      m_redoAttack = static_cast<const ADSRAttackAndDecayUndoCommand*>(other)->m_redoAttack;
      m_redoDecay = static_cast<const ADSRAttackAndDecayUndoCommand*>(other)->m_redoDecay;
      return true;
    }
    return false;
  }
  int id() const { return int(Id::ADSRAttackAndDecay); }
};

void ADSRControls::setAttackAndDecay(double attack, double decay, uint64_t cycleCount) {
  m_enableUpdate = false;
  m_attack->setValue(attack);
  m_decay->setValue(decay);
  m_enableUpdate = true;

  ADSRView* view = getEditor()->m_adsrView;
  g_MainWindow->pushUndoCommand(
      new ADSRAttackAndDecayUndoCommand(m_attack->value(), m_decay->value(), cycleCount, view->m_node));
  view->update();
}

class ADSRDecayAndSustainUndoCommand : public EditorUndoCommand {
  double m_redoDecay, m_redoSustain;
  double m_undoDecay, m_undoSustain;
  uint64_t m_cycleCount;
  bool m_undid = false;

public:
  ADSRDecayAndSustainUndoCommand(double redoDecay, double redoSustain, uint64_t cycleCount,
                                 amuse::ObjToken<ProjectModel::ADSRNode> node)
  : EditorUndoCommand(node.get(), ADSRControls::tr("Change Decay/Sustain"))
  , m_redoDecay(redoDecay)
  , m_redoSustain(redoSustain)
  , m_cycleCount(cycleCount) {}
  void undo() {
    m_undid = true;
    amuse::ITable& table = **m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
      amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
      adsr.setDecay(m_undoDecay);
      adsr.setSustain(m_undoSustain);
    } else if (table.Isa() == amuse::ITable::Type::ADSR) {
      amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
      adsr.setDecay(m_undoDecay);
      adsr.setSustain(m_undoSustain);
    }
    EditorUndoCommand::undo();
  }
  void redo() {
    amuse::ITable& table = **m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
      amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
      m_undoDecay = adsr.getDecay();
      m_undoSustain = adsr.getSustain();
      adsr.setDecay(m_redoDecay);
      adsr.setSustain(m_redoSustain);
    } else if (table.Isa() == amuse::ITable::Type::ADSR) {
      amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
      m_undoDecay = adsr.getDecay();
      m_undoSustain = adsr.getSustain();
      adsr.setDecay(m_redoDecay);
      adsr.setSustain(m_redoSustain);
    }
    if (m_undid)
      EditorUndoCommand::redo();
  }
  bool mergeWith(const QUndoCommand* other) {
    if (other->id() == id() &&
        m_cycleCount == static_cast<const ADSRDecayAndSustainUndoCommand*>(other)->m_cycleCount) {
      m_redoDecay = static_cast<const ADSRDecayAndSustainUndoCommand*>(other)->m_redoDecay;
      m_redoSustain = static_cast<const ADSRDecayAndSustainUndoCommand*>(other)->m_redoSustain;
      return true;
    }
    return false;
  }
  int id() const { return int(Id::ADSRDecayAndSustain); }
};

void ADSRControls::setDecayAndSustain(double decay, double sustain, uint64_t cycleCount) {
  m_enableUpdate = false;
  m_decay->setValue(decay);
  m_sustain->setValue(sustain);
  m_enableUpdate = true;

  ADSRView* view = getEditor()->m_adsrView;
  g_MainWindow->pushUndoCommand(
      new ADSRDecayAndSustainUndoCommand(m_decay->value(), m_sustain->value() / 100.0, cycleCount, view->m_node));
  view->update();
}

class ADSRReleaseUndoCommand : public EditorUndoCommand {
  double m_redoVal, m_undoVal;
  uint64_t m_cycleCount;
  bool m_undid = false;

public:
  ADSRReleaseUndoCommand(double redoVal, uint64_t cycleCount, amuse::ObjToken<ProjectModel::ADSRNode> node)
  : EditorUndoCommand(node.get(), ADSRControls::tr("Change Release")), m_redoVal(redoVal), m_cycleCount(cycleCount) {}
  void undo() {
    m_undid = true;
    amuse::ITable& table = **m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
      amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
      adsr.setRelease(m_undoVal);
    } else if (table.Isa() == amuse::ITable::Type::ADSR) {
      amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
      adsr.setRelease(m_undoVal);
    }
    EditorUndoCommand::undo();
  }
  void redo() {
    amuse::ITable& table = **m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
      amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
      m_undoVal = adsr.getRelease();
      adsr.setRelease(m_redoVal);
    } else if (table.Isa() == amuse::ITable::Type::ADSR) {
      amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
      m_undoVal = adsr.getRelease();
      adsr.setRelease(m_redoVal);
    }
    if (m_undid)
      EditorUndoCommand::redo();
  }
  bool mergeWith(const QUndoCommand* other) {
    if (other->id() == id() && m_cycleCount == static_cast<const ADSRReleaseUndoCommand*>(other)->m_cycleCount) {
      m_redoVal = static_cast<const ADSRReleaseUndoCommand*>(other)->m_redoVal;
      return true;
    }
    return false;
  }
  int id() const { return int(Id::ADSRRelease); }
};

void ADSRControls::setRelease(double release, uint64_t cycleCount) {
  m_enableUpdate = false;
  m_release->setValue(release);
  m_enableUpdate = true;

  ADSRView* view = getEditor()->m_adsrView;
  g_MainWindow->pushUndoCommand(new ADSRReleaseUndoCommand(m_release->value(), cycleCount, view->m_node));
  view->update();
}

void ADSRControls::releaseChanged(double val) {
  if (m_enableUpdate) {
    ADSRView* view = getEditor()->m_adsrView;
    g_MainWindow->pushUndoCommand(new ADSRReleaseUndoCommand(val, ~0ull, view->m_node));
    view->update();
  }
}

template <class T>
static std::unique_ptr<T> MakeAlternateVersion(amuse::ITable& table) {
  std::unique_ptr<T> ret = std::make_unique<T>();
  if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
    amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
    ret->setAttack(adsr.getAttack());
    ret->setDecay(adsr.getDecay());
    ret->setSustain(adsr.getSustain());
    ret->setRelease(adsr.getRelease());
  } else if (table.Isa() == amuse::ITable::Type::ADSR) {
    amuse::ADSR& adsr = static_cast<amuse::ADSR&>(table);
    ret->setAttack(adsr.getAttack());
    ret->setDecay(adsr.getDecay());
    ret->setSustain(adsr.getSustain());
    ret->setRelease(adsr.getRelease());
  }
  return ret;
}

class ADSRDLSUndoCommand : public EditorUndoCommand {
  bool m_redoVal;
  double m_redoVelToAttack, m_redoKeyToDecay;
  double m_undoVelToAttack, m_undoKeyToDecay;
  bool m_undid = false;

public:
  ADSRDLSUndoCommand(bool redoVal, double redoVelToAttack, double redoKeyToDecay,
                     amuse::ObjToken<ProjectModel::ADSRNode> node)
  : EditorUndoCommand(node.get(), ADSRControls::tr("Change DLS"))
  , m_redoVal(redoVal)
  , m_redoVelToAttack(redoVelToAttack)
  , m_redoKeyToDecay(redoKeyToDecay) {}
  void undo() {
    m_undid = true;
    std::unique_ptr<amuse::ITable>& table = *m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if ((table->Isa() == amuse::ITable::Type::ADSRDLS && !m_redoVal) ||
        (table->Isa() == amuse::ITable::Type::ADSR && m_redoVal))
      return;

    std::unique_ptr<amuse::ITable> oldTable = std::move(table);
    if (!m_redoVal) {
      table = MakeAlternateVersion<amuse::ADSRDLS>(*oldTable);
      static_cast<amuse::ADSRDLS&>(*table)._setVelToAttack(m_undoVelToAttack);
      static_cast<amuse::ADSRDLS&>(*table)._setKeyToDecay(m_undoKeyToDecay);
    } else {
      table = MakeAlternateVersion<amuse::ADSR>(*oldTable);
    }

    EditorUndoCommand::undo();
  }
  void redo() {
    std::unique_ptr<amuse::ITable>& table = *m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if ((table->Isa() == amuse::ITable::Type::ADSRDLS && m_redoVal) ||
        (table->Isa() == amuse::ITable::Type::ADSR && !m_redoVal))
      return;

    std::unique_ptr<amuse::ITable> oldTable = std::move(table);
    if (m_redoVal) {
      table = MakeAlternateVersion<amuse::ADSRDLS>(*oldTable);
      static_cast<amuse::ADSRDLS&>(*table)._setVelToAttack(m_redoVelToAttack);
      static_cast<amuse::ADSRDLS&>(*table)._setKeyToDecay(m_redoKeyToDecay);
    } else {
      if (oldTable->Isa() == amuse::ITable::Type::ADSRDLS) {
        m_undoVelToAttack = static_cast<amuse::ADSRDLS&>(*oldTable)._getVelToAttack();
        m_undoKeyToDecay = static_cast<amuse::ADSRDLS&>(*oldTable)._getKeyToDecay();
      }
      table = MakeAlternateVersion<amuse::ADSR>(*oldTable);
    }

    if (m_undid)
      EditorUndoCommand::redo();
  }
  bool mergeWith(const QUndoCommand* other) {
    if (other->id() == id()) {
      m_redoVal = static_cast<const ADSRDLSUndoCommand*>(other)->m_redoVal;
      m_redoVelToAttack = static_cast<const ADSRDLSUndoCommand*>(other)->m_redoVelToAttack;
      m_redoKeyToDecay = static_cast<const ADSRDLSUndoCommand*>(other)->m_redoKeyToDecay;
      return true;
    }
    return false;
  }
  int id() const { return int(Id::ADSRDLS); }
};

void ADSRControls::dlsStateChanged(int state) {
  if (m_enableUpdate) {
    m_velToAttackLab->setVisible(state == Qt::Checked);
    m_velToAttack->setVisible(state == Qt::Checked);
    m_keyToDecayLab->setVisible(state == Qt::Checked);
    m_keyToDecay->setVisible(state == Qt::Checked);

    ADSRView* view = getEditor()->m_adsrView;
    g_MainWindow->pushUndoCommand(
        new ADSRDLSUndoCommand(state == Qt::Checked, m_velToAttack->value(), m_keyToDecay->value(), view->m_node));
    view->update();
  }
}

class ADSRVelToAttackUndoCommand : public EditorUndoCommand {
  double m_redoVal, m_undoVal;
  bool m_undid = false;

public:
  ADSRVelToAttackUndoCommand(double redoVal, amuse::ObjToken<ProjectModel::ADSRNode> node)
  : EditorUndoCommand(node.get(), ADSRControls::tr("Change Vel To Attack")), m_redoVal(redoVal) {}
  void undo() {
    m_undid = true;
    amuse::ITable& table = **m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
      amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
      adsr._setVelToAttack(m_undoVal);
    }
    EditorUndoCommand::undo();
  }
  void redo() {
    amuse::ITable& table = **m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
      amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
      m_undoVal = adsr._getVelToAttack();
      adsr._setVelToAttack(m_redoVal);
    }
    if (m_undid)
      EditorUndoCommand::redo();
  }
  bool mergeWith(const QUndoCommand* other) {
    if (other->id() == id()) {
      m_redoVal = static_cast<const ADSRVelToAttackUndoCommand*>(other)->m_redoVal;
      return true;
    }
    return false;
  }
  int id() const { return int(Id::ADSRVelToAttack); }
};

void ADSRControls::velToAttackChanged(double val) {
  if (m_enableUpdate) {
    ADSRView* view = getEditor()->m_adsrView;
    g_MainWindow->pushUndoCommand(new ADSRVelToAttackUndoCommand(val, view->m_node));
    view->update();
  }
}

class ADSRKeyToDecayUndoCommand : public EditorUndoCommand {
  double m_redoVal, m_undoVal;
  bool m_undid = false;

public:
  ADSRKeyToDecayUndoCommand(double redoVal, amuse::ObjToken<ProjectModel::ADSRNode> node)
  : EditorUndoCommand(node.get(), ADSRControls::tr("Change Key To Decay")), m_redoVal(redoVal) {}
  void undo() {
    m_undid = true;
    amuse::ITable& table = **m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
      amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
      adsr._setKeyToDecay(m_undoVal);
    }
    EditorUndoCommand::undo();
  }
  void redo() {
    amuse::ITable& table = **m_node.cast<ProjectModel::ADSRNode>()->m_obj;
    if (table.Isa() == amuse::ITable::Type::ADSRDLS) {
      amuse::ADSRDLS& adsr = static_cast<amuse::ADSRDLS&>(table);
      m_undoVal = adsr._getKeyToDecay();
      adsr._setKeyToDecay(m_redoVal);
    }
    if (m_undid)
      EditorUndoCommand::redo();
  }
  bool mergeWith(const QUndoCommand* other) {
    if (other->id() == id()) {
      m_redoVal = static_cast<const ADSRKeyToDecayUndoCommand*>(other)->m_redoVal;
      return true;
    }
    return false;
  }
  int id() const { return int(Id::ADSRKeyToDecay); }
};

void ADSRControls::keyToDecayChanged(double val) {
  if (m_enableUpdate) {
    ADSRView* view = getEditor()->m_adsrView;
    g_MainWindow->pushUndoCommand(new ADSRKeyToDecayUndoCommand(val, view->m_node));
    view->update();
  }
}

ADSRControls::ADSRControls(QWidget* parent) : QFrame(parent) {
  setSizePolicy(QSizePolicy::Preferred, QSizePolicy::Fixed);
  setFixedHeight(100);
  setFrameShape(QFrame::StyledPanel);
  setFrameShadow(QFrame::Sunken);
  setBackgroundRole(QPalette::Base);
  setAutoFillBackground(true);

  QHBoxLayout* mainLayout = new QHBoxLayout;

  QGridLayout* leftLayout = new QGridLayout;

  QPalette palette = QWidget::palette();
  palette.setColor(QPalette::Base, palette.color(QPalette::Window));

  palette.setColor(QPalette::Text, Red);
  QLabel* lab = new QLabel(tr("Attack"));
  lab->setPalette(palette);
  leftLayout->addWidget(lab, 0, 0);
  m_attack = new QDoubleSpinBox;
  m_attack->setDisabled(true);
  m_attack->setRange(0, 65.535);
  m_attack->setDecimals(3);
  m_attack->setSingleStep(0.1);
  m_attack->setSuffix(tr(" sec"));
  m_attack->setPalette(palette);
  connect(m_attack, SIGNAL(valueChanged(double)), this, SLOT(attackChanged(double)));
  leftLayout->addWidget(m_attack, 1, 0);

  palette.setColor(QPalette::Text, Green);
  lab = new QLabel(tr("Decay"));
  lab->setPalette(palette);
  leftLayout->addWidget(lab, 0, 1);
  m_decay = new QDoubleSpinBox;
  m_decay->setDisabled(true);
  m_decay->setRange(0, 65.535);
  m_decay->setDecimals(3);
  m_decay->setSingleStep(0.1);
  m_decay->setSuffix(tr(" sec"));
  m_decay->setPalette(palette);
  connect(m_decay, SIGNAL(valueChanged(double)), this, SLOT(decayChanged(double)));
  leftLayout->addWidget(m_decay, 1, 1);

  palette.setColor(QPalette::Text, Qt::white);
  leftLayout->addWidget(new QLabel(tr("Sustain")), 0, 2);
  m_sustain = new QDoubleSpinBox;
  m_sustain->setDisabled(true);
  m_sustain->setRange(0.0, 100.0);
  m_sustain->setDecimals(3);
  m_sustain->setSuffix(tr(" %"));
  m_sustain->setPalette(palette);
  connect(m_sustain, SIGNAL(valueChanged(double)), this, SLOT(sustainChanged(double)));
  leftLayout->addWidget(m_sustain, 1, 2);

  palette.setColor(QPalette::Text, Blue);
  lab = new QLabel(tr("Release"));
  lab->setPalette(palette);
  leftLayout->addWidget(lab, 0, 3);
  m_release = new QDoubleSpinBox;
  m_release->setDisabled(true);
  m_release->setRange(0.0, 65.535);
  m_release->setDecimals(3);
  m_release->setSingleStep(0.1);
  m_release->setSuffix(tr(" sec"));
  m_release->setPalette(palette);
  connect(m_release, SIGNAL(valueChanged(double)), this, SLOT(releaseChanged(double)));
  leftLayout->addWidget(m_release, 1, 3);

  palette.setColor(QPalette::Text, Qt::white);
  leftLayout->addWidget(new QLabel(tr("DLS")), 0, 4);
  m_dls = new QCheckBox;
  m_dls->setPalette(palette);
  m_dls->setDisabled(true);
  m_dls->setChecked(false);
  connect(m_dls, SIGNAL(stateChanged(int)), this, SLOT(dlsStateChanged(int)));
  leftLayout->addWidget(m_dls, 1, 4);

  m_velToAttackLab = new QLabel(tr("Vel To Attack"));
  m_velToAttackLab->setVisible(false);
  leftLayout->addWidget(m_velToAttackLab, 0, 5);
  m_velToAttack = new QDoubleSpinBox;
  m_velToAttack->setVisible(false);
  m_velToAttack->setRange(0.0, 9999.999);
  m_velToAttack->setDecimals(3);
  m_velToAttack->setSingleStep(1.0);
  m_velToAttack->setPalette(palette);
  connect(m_velToAttack, SIGNAL(valueChanged(double)), this, SLOT(velToAttackChanged(double)));
  leftLayout->addWidget(m_velToAttack, 1, 5);

  m_keyToDecayLab = new QLabel(tr("Key To Decay"));
  m_keyToDecayLab->setVisible(false);
  leftLayout->addWidget(m_keyToDecayLab, 0, 6);
  m_keyToDecay = new QDoubleSpinBox;
  m_keyToDecay->setVisible(false);
  m_keyToDecay->setRange(0.0, 9999.999);
  m_keyToDecay->setDecimals(3);
  m_keyToDecay->setSingleStep(1.0);
  m_keyToDecay->setPalette(palette);
  connect(m_keyToDecay, SIGNAL(valueChanged(double)), this, SLOT(keyToDecayChanged(double)));
  leftLayout->addWidget(m_keyToDecay, 1, 6);

  leftLayout->setColumnMinimumWidth(0, 75);
  leftLayout->setColumnStretch(0, 1);
  leftLayout->setColumnMinimumWidth(1, 75);
  leftLayout->setColumnStretch(1, 1);
  leftLayout->setColumnMinimumWidth(2, 75);
  leftLayout->setColumnStretch(2, 1);
  leftLayout->setColumnMinimumWidth(3, 75);
  leftLayout->setColumnStretch(3, 1);
  leftLayout->setColumnMinimumWidth(4, 50);
  leftLayout->setColumnStretch(4, 0);
  leftLayout->setColumnMinimumWidth(5, 75);
  leftLayout->setColumnStretch(5, 1);
  leftLayout->setColumnMinimumWidth(6, 75);
  leftLayout->setColumnStretch(6, 1);
  leftLayout->setRowMinimumHeight(0, 22);
  leftLayout->setRowMinimumHeight(1, 37);
  leftLayout->setContentsMargins(10, 6, 0, 14);

  mainLayout->addLayout(leftLayout);
  mainLayout->addStretch();
  setLayout(mainLayout);
}

bool ADSREditor::loadData(ProjectModel::ADSRNode* node) {
  m_adsrView->loadData(node);
  m_controls->loadData();
  return true;
}

void ADSREditor::unloadData() {
  m_adsrView->unloadData();
  m_controls->unloadData();
}

ProjectModel::INode* ADSREditor::currentNode() const { return m_adsrView->currentNode(); }

ADSREditor::ADSREditor(QWidget* parent) : EditorWidget(parent), m_adsrView(new ADSRView), m_controls(new ADSRControls) {
  QVBoxLayout* layout = new QVBoxLayout;
  layout->setContentsMargins(QMargins());
  layout->setSpacing(1);
  layout->addWidget(m_adsrView);
  layout->addWidget(m_controls);
  setLayout(layout);
}