#include "toonzqt/schematicnode.h"

#include <qgraphicsscenemouseevent></qgraphicsscenemouseevent>

#include <qstyleoptiongraphicsitem></qstyleoptiongraphicsitem>

#include <qkeyevent></qkeyevent>

#include <algorithm></algorithm>

#include <qapplication></qapplication>

#include <qtextdocument></qtextdocument>

#include <qtextcursor></qtextcursor>

#include <qtextblock></qtextblock>

#include <qmenubar></qmenubar>

#include <qpolygonf></qpolygonf>

#include <qdesktopwidget></qdesktopwidget>

#include "tundo.h"

#include "toonzqt/menubarcommand.h"

#include "toonzqt/gutil.h"

//========================================================

//

// StageSchematicName

//

//========================================================

SchematicName::SchematicName(QGraphicsItem *parent, double width, double height)

    : QGraphicsTextItem("", parent), m_width(width), m_height(height) {

  setFlag(QGraphicsItem::ItemIsSelectable, true);

  setFlag(QGraphicsItem::ItemIsFocusable, true);

  setTextInteractionFlags(Qt::TextEditorInteraction);

  connect(document(), SIGNAL(contentsChanged()), this,

          SLOT(onContentsChanged()));

}

//--------------------------------------------------------

SchematicName::~SchematicName() {}

//--------------------------------------------------------

void SchematicName::setName(const QString &name) { setPlainText(name); }

//--------------------------------------------------------

void SchematicName::onContentsChanged() {

  QString text       = document()->toPlainText();

  QTextCursor cursor = textCursor();

  int position       = cursor.position();

  if (position > 0 && text.at(position - 1) == '\n') {

    text.remove("\n");

    setPlainText(text);

    ;

    emit focusOut();

  }

}

//--------------------------------------------------------

void SchematicName::focusOutEvent(QFocusEvent *fe) {

  qApp->removeEventFilter(this);

  if (fe->reason() == Qt::MouseFocusReason) emit focusOut();

}

//--------------------------------------------------------

void SchematicName::keyPressEvent(QKeyEvent *ke) {

  if (ke->key() == Qt::Key_Left || ke->key() == Qt::Key_Right) {

    QTextCursor cursor = textCursor();

    int currentPos     = cursor.position();

    if (ke->key() == Qt::Key_Left)

      cursor.setPosition(currentPos - 1);

    else

      cursor.setPosition(currentPos + 1);

    setTextCursor(cursor);

  } else

    QGraphicsTextItem::keyPressEvent(ke);

}

//--------------------------------------------------------

bool SchematicName::eventFilter(QObject *object, QEvent *event) {

  if (event->type() == QEvent::Shortcut ||

      event->type() == QEvent::ShortcutOverride) {

    if (!object->inherits("QGraphicsView")) {

      event->accept();

      return true;

    }

  }

  return false;

}

//--------------------------------------------------------

void SchematicName::focusInEvent(QFocusEvent *fe) {

  QGraphicsTextItem::focusInEvent(fe);

  qApp->installEventFilter(this);

  QTextDocument *doc = document();

  QTextCursor cursor(doc->begin());

  cursor.select(QTextCursor::Document);

  setTextCursor(cursor);

}

//========================================================

//

// class SchematicThumbnailToggle

//

//========================================================

SchematicThumbnailToggle::SchematicThumbnailToggle(SchematicNode *parent,

                                                   bool isOpened)

    : QGraphicsItem(parent), m_isDown(!isOpened) {}

//--------------------------------------------------------

SchematicThumbnailToggle::~SchematicThumbnailToggle() {}

//--------------------------------------------------------

QRectF SchematicThumbnailToggle::boundingRect() const {

  return QRectF(0, 0, 14, 14);

}

//--------------------------------------------------------

void SchematicThumbnailToggle::paint(QPainter *painter,

                                     const QStyleOptionGraphicsItem *option,

                                     QWidget *widget) {

  QRect rect(3, 3, 8, 8);

  QRect sourceRect = scene()->views()[0]->matrix().mapRect(rect);

  static QIcon onIcon(":Resources/schematic_thumbtoggle_on.svg");

  static QIcon offIcon(":Resources/schematic_thumbtoggle_off.svg");

  QPixmap pixmap;

  if (m_isDown)

    pixmap = offIcon.pixmap(sourceRect.size());

  else

    pixmap   = onIcon.pixmap(sourceRect.size());

  sourceRect = QRect(0, 0, sourceRect.width() * getDevPixRatio(),

                     sourceRect.height() * getDevPixRatio());

  painter->drawPixmap(rect, pixmap, sourceRect);

}

//--------------------------------------------------------

void SchematicThumbnailToggle::setIsDown(bool value) {

  m_isDown = value;

  emit(toggled(!m_isDown));

}

//--------------------------------------------------------

void SchematicThumbnailToggle::mousePressEvent(QGraphicsSceneMouseEvent *me) {

  m_isDown = !m_isDown;

  emit(toggled(!m_isDown));

}

//========================================================

//

// SchematicToggle

//

//========================================================

SchematicToggle::SchematicToggle(SchematicNode *parent, const QPixmap &pixmap,

                                 int flags, bool isLargeScaled)

    : QGraphicsItem(parent)

    , m_pixmap1(pixmap)

    , m_pixmap2()

    , m_state(0)

    , m_flags(flags)

    , m_width(isLargeScaled ? 18 : 30)

    , m_height(isLargeScaled ? 7 : 5) {}

//--------------------------------------------------------

SchematicToggle::SchematicToggle(SchematicNode *parent, const QPixmap &pixmap1,

                                 const QPixmap &pixmap2, int flags,

                                 bool isLargeScaled)

    : QGraphicsItem(parent)

    , m_pixmap1(pixmap1)

    , m_pixmap2(pixmap2)

    , m_state(0)

    , m_flags(flags)

    , m_width(isLargeScaled ? 18 : 30)

    , m_height(isLargeScaled ? 7 : 5) {}

//--------------------------------------------------------

SchematicToggle::~SchematicToggle() {}

//--------------------------------------------------------

QRectF SchematicToggle::boundingRect() const {

  return QRectF(0, 0, m_width, m_height);

}

//--------------------------------------------------------

void SchematicToggle::paint(QPainter *painter,

                            const QStyleOptionGraphicsItem *option,

                            QWidget *widget) {

  if (m_state != 0) {

    QPixmap &pix =

        (m_state == 2 && !m_pixmap2.isNull()) ? m_pixmap2 : m_pixmap1;

    painter->drawPixmap(boundingRect().toRect(), pix);

  }

}

//--------------------------------------------------------

void SchematicToggle::mousePressEvent(QGraphicsSceneMouseEvent *me) {

  if (me->button() == Qt::LeftButton) {

    if (m_pixmap2.isNull()) {

      m_state = 1 - m_state;

      emit(toggled(m_state != 0));

    } else if (m_flags & eEnableNullState) {

      m_state = (m_state + 1) % 3;

      emit(stateChanged(m_state));

    } else {

      m_state = 3 - m_state;

      emit(stateChanged(m_state));

    }

  }

  if (me->button() == Qt::RightButton) {

    SchematicNode *parent = dynamic_cast<schematicnode *="">(this->parentItem());</schematicnode>

    if (parent) parent->onClicked();

  }

}

//--------------------------------------------------------

void SchematicToggle::contextMenuEvent(QGraphicsSceneContextMenuEvent *cme) {

  if (!(m_flags & eIsParentColumn)) return;

  if (m_pixmap2.isNull()) {

    QMenu *menu                = new QMenu(0);

    CommandManager *cmdManager = CommandManager::instance();

    menu->addAction(cmdManager->getAction("MI_EnableThisColumnOnly"));

    menu->addAction(cmdManager->getAction("MI_EnableSelectedColumns"));

    menu->addAction(cmdManager->getAction("MI_EnableAllColumns"));

    menu->addAction(cmdManager->getAction("MI_DisableAllColumns"));

    menu->addAction(cmdManager->getAction("MI_DisableSelectedColumns"));

    menu->addAction(cmdManager->getAction("MI_SwapEnabledColumns"));

    QAction *action = menu->exec(cme->screenPos());

  } else {

    QMenu *menu                = new QMenu(0);

    CommandManager *cmdManager = CommandManager::instance();

    menu->addAction(cmdManager->getAction("MI_ActivateThisColumnOnly"));

    menu->addAction(cmdManager->getAction("MI_ActivateSelectedColumns"));

    menu->addAction(cmdManager->getAction("MI_ActivateAllColumns"));

    menu->addAction(cmdManager->getAction("MI_DeactivateAllColumns"));

    menu->addAction(cmdManager->getAction("MI_DeactivateSelectedColumns"));

    menu->addAction(cmdManager->getAction("MI_ToggleColumnsActivation"));

    QAction *action = menu->exec(cme->screenPos());

  }

}

//========================================================

//--------------------------------------------------------

/*! for Spline Aim and CP toggles

*/

void SchematicToggle_SplineOptions::paint(

    QPainter *painter, const QStyleOptionGraphicsItem *option,

    QWidget *widget) {

  QRectF rect = boundingRect();

  painter->fillRect(rect, Qt::white);

  if (m_state != 0) {

    QPixmap &pix =

        (m_state == 2 && !m_pixmap2.isNull()) ? m_pixmap2 : m_pixmap1;

    painter->drawPixmap(boundingRect().toRect(), pix);

  }

  painter->setBrush(Qt::NoBrush);

  painter->setPen(QColor(180, 180, 180, 255));

  painter->drawRect(rect);

}

//--------------------------------------------------------

/*! for Spline Aim and CP toggles

*/

void SchematicToggle_SplineOptions::mousePressEvent(

    QGraphicsSceneMouseEvent *me) {

  SchematicToggle::mousePressEvent(me);

  update();

}

//========================================================

//

// SchematicHandleSpinBox

//

//========================================================

SchematicHandleSpinBox::SchematicHandleSpinBox(QGraphicsItem *parent)

    : QGraphicsItem(parent), m_buttonState(Qt::NoButton), m_delta(0) {

  setFlag(QGraphicsItem::ItemIsSelectable, false);

  setFlag(QGraphicsItem::ItemIsFocusable, false);

  m_pixmap = QPixmap(":Resources/schematic_spin_arrows.svg");

}

//--------------------------------------------------------

SchematicHandleSpinBox::~SchematicHandleSpinBox() {}

//--------------------------------------------------------

QRectF SchematicHandleSpinBox::boundingRect() const {

  return QRectF(0, 0, 10, 10);

}

//--------------------------------------------------------

void SchematicHandleSpinBox::paint(QPainter *painter,

                                   const QStyleOptionGraphicsItem *option,

                                   QWidget *widget) {

  QRectF rect = boundingRect();

  painter->drawPixmap(rect.toRect(), m_pixmap);

  painter->setBrush(QColor(0, 0, 0, 0));

  painter->setPen(QColor(128, 128, 128, 255));

  painter->drawRect(rect);

}

//--------------------------------------------------------

void SchematicHandleSpinBox::mouseMoveEvent(QGraphicsSceneMouseEvent *me) {

  if (m_buttonState == Qt::LeftButton) {

    bool increase           = false;

    int delta               = me->screenPos().y() - me->lastScreenPos().y();

    if (delta < 0) increase = true;

    m_delta += abs(delta);

    if (m_delta > 5) {

      if (increase)

        emit(modifyHandle(1));

      else

        emit(modifyHandle(-1));

      m_delta = 0;

      emit sceneChanged();

    }

  }

}

//--------------------------------------------------------

void SchematicHandleSpinBox::mousePressEvent(QGraphicsSceneMouseEvent *me) {

  m_buttonState = me->button();

  TUndoManager::manager()->beginBlock();

}

//--------------------------------------------------------

void SchematicHandleSpinBox::mouseReleaseEvent(QGraphicsSceneMouseEvent *me) {

  m_buttonState = Qt::NoButton;

  m_delta       = 0;

  TUndoManager::manager()->endBlock();

  emit handleReleased();

}

//========================================================

//

// class SchematicLink

//

//========================================================

SchematicLink::SchematicLink(QGraphicsItem *parent, QGraphicsScene *scene)

#if QT_VERSION >= 0x050000

    : QGraphicsItem(parent)

#else

    : QGraphicsItem(parent, scene)

#endif

    , m_startPort(0)

    , m_endPort(0)

    , m_path()

    , m_hitPath()

    , m_lineShaped(false)

    , m_highlighted(false) {

#if QT_VERSION >= 0x050000

  scene->addItem(this);

#endif

  setFlag(QGraphicsItem::ItemIsMovable, false);

  setFlag(QGraphicsItem::ItemIsSelectable, true);

  setFlag(QGraphicsItem::ItemIsFocusable, false);

  setZValue(0.0);

}

//--------------------------------------------------------

SchematicLink::~SchematicLink() { m_startPort = m_endPort = 0; }

//--------------------------------------------------------

QRectF SchematicLink::boundingRect() const {

  return m_hitPath.boundingRect().adjusted(-5, -5, 5, 5);

}

//--------------------------------------------------------

QPainterPath SchematicLink::shape() const { return m_hitPath; }

//--------------------------------------------------------

void SchematicLink::paint(QPainter *painter,

                          const QStyleOptionGraphicsItem *option,

                          QWidget *widget) {

  if (getStartPort() && (getStartPort()->getType() == 100  // eStageSplinePort

                         || getStartPort()->getType() == 202)) {  // eFxLinkPort

    if (isSelected() || isHighlighted())

      painter->setPen(QColor(255, 255, 10));

    else

      painter->setPen(QColor(50, 255, 50, 128));

  } else if (isSelected() || isHighlighted())

    painter->setPen(QPen(Qt::cyan));

  else if (!m_lineShaped)

    painter->setPen(QPen(Qt::white));

  else

    painter->setPen(QPen(QColor(170, 170, 10), 0, Qt::DashLine));

  painter->setRenderHint(QPainter::Antialiasing, true);

  painter->drawPath(m_path);

}

//--------------------------------------------------------

void SchematicLink::updatePath(const QPointF &startPos, const QPointF &endPos) {

  prepareGeometryChange();

  setPos(startPos);

  if (!m_lineShaped) {

    QPointF p0((endPos.x() - startPos.x()) * 0.5, 0);

    QPointF p1(p0.x(), endPos.y() - startPos.y());

    QPointF p2(endPos - startPos);

    m_path = QPainterPath(QPointF(0, 0));

    m_path.cubicTo(p0, p1, p2);

    QPointF h(0, 5);

    QPointF p = h;

    if (p2.y() > 0)

      p.setX(p2.x() > 0 ? -5 : 5);

    else

      p.setX(p2.x() > 0 ? 5 : -5);

    m_hitPath = QPainterPath(QPointF(0, 0));

    m_hitPath.lineTo(h);

    m_hitPath.cubicTo(p0 + p, p1 + p, p2 + h);

    m_hitPath.lineTo(p2 - h);

    m_hitPath.cubicTo(p1 - p, p0 - p, -h);

    m_hitPath.lineTo(0, 0);

  } else {

    m_path = QPainterPath(QPointF(0, 0));

    m_path.lineTo(endPos - startPos);

    m_hitPath = QPainterPath(QPointF(0, 0));

    m_hitPath.lineTo(endPos - startPos);

  }

}

//--------------------------------------------------------

void SchematicLink::updatePath(SchematicPort *startPort,

                               SchematicPort *endPort) {

  updatePath(startPort->getLinkEndPoint(), endPort->getLinkEndPoint());

}

//--------------------------------------------------------

SchematicPort *SchematicLink::getOtherPort(const SchematicPort *port) const {

  if (port == m_startPort)

    return m_endPort;

  else if (port == m_endPort)

    return m_startPort;

  else

    return 0;

}

//--------------------------------------------------------

SchematicNode *SchematicLink::getOtherNode(const SchematicNode *node) const {

  if (node == m_startPort->getNode())

    return m_endPort->getNode();

  else if (node == m_endPort->getNode())

    return m_startPort->getNode();

  else

    return 0;

}

//--------------------------------------------------------

void SchematicLink::mousePressEvent(QGraphicsSceneMouseEvent *me) {

  QPointF pos              = me->scenePos();

  SchematicPort *startPort = getStartPort();

  SchematicPort *endPort   = getEndPort();

  if (startPort && endPort) {

    QRectF startRect = startPort->boundingRect();

    startRect.moveTopLeft(startPort->scenePos());

    QRectF endRect = endPort->boundingRect();

    endRect.moveTopLeft(endPort->scenePos());

    if (startRect.contains(pos) || endRect.contains(pos)) {

      me->ignore();

      return;

    }

  }

  QMatrix matrix = scene()->views()[0]->matrix();

#if QT_VERSION >= 0x050000

  double scaleFactor = sqrt(matrix.determinant());

#else

  double scaleFactor = sqrt(matrix.det());

#endif

  QPointF startPos = getStartPort()->getLinkEndPoint();

  QPointF endPos   = getEndPort()->getLinkEndPoint();

  QPointF p0((endPos.x() - startPos.x()) * 0.5, 0);

  QPointF p1(p0.x(), endPos.y() - startPos.y());

  QPointF p2(endPos - startPos);

  double sensitivity = 5 / scaleFactor;

  QPointF h(0, sensitivity);

  QPointF p = h;

  if (p2.y() > 0)

    p.setX(p2.x() > 0 ? -sensitivity : sensitivity);

  else

    p.setX(p2.x() > 0 ? sensitivity : -sensitivity);

  QPainterPath path(QPointF(0, 0));

  path.lineTo(h);

  path.cubicTo(p0 + p, p1 + p, p2 + h);

  path.lineTo(p2 - h);

  path.cubicTo(p1 - p, p0 - p, -h);

  path.lineTo(0, 0);

  if (!path.contains(me->scenePos() - scenePos())) {

    me->ignore();

    return;

  }

  if (!isSelected()) {

    if (me->modifiers() != Qt::ControlModifier) scene()->clearSelection();

    if (me->button() == Qt::LeftButton || me->button() == Qt::RightButton)

      setSelected(true);

  } else {

    if (me->modifiers() == Qt::ControlModifier &&

        me->button() == Qt::LeftButton)

      setSelected(false);

  }

}

//--------------------------------------------------------

void SchematicLink::mouseReleaseEvent(QGraphicsSceneMouseEvent *me) {

  if (me->modifiers() != Qt::ControlModifier && me->button() != Qt::RightButton)

    QGraphicsItem::mouseReleaseEvent(me);

}

//========================================================

//

// class SchematicPort

//

//========================================================

SchematicPort::SchematicPort(QGraphicsItem *parent, SchematicNode *node,

                             int type)

    : QGraphicsItem(parent)

    , m_node(node)

    , m_buttonState(Qt::NoButton)

    , m_highlighted(false)

    , m_ghostLink(0)

    , m_linkingTo(0)

    , m_hook(0, 0)

    , m_type(type) {

#if QT_VERSION >= 0x050000

  setAcceptHoverEvents(false);

#else

  setAcceptsHoverEvents(false);

#endif

  setFlag(QGraphicsItem::ItemIsSelectable, false);

  setFlag(QGraphicsItem::ItemIsFocusable, false);

}

//--------------------------------------------------------

SchematicPort::~SchematicPort() { m_links.clear(); }

//--------------------------------------------------------

void SchematicPort::mouseMoveEvent(QGraphicsSceneMouseEvent *me) {

  if (m_buttonState != Qt::LeftButton) return;

  if (!m_ghostLink) return;

  if (m_linkingTo) {

    showSnappedLinks();

    m_linkingTo = 0;

  }

  // Snapping

  SchematicPort *linkingTo = searchPort(me->scenePos());

  if (!linkingTo) {

    if (m_linkingTo) {

      m_linkingTo->highLight(false);

      m_linkingTo->update();

    }

    m_ghostLink->updatePath(this->getLinkEndPoint(), me->scenePos());

    m_linkingTo = linkingTo;

  }

  // if to be connected something

  else if (linkingTo != this) {

    m_ghostLink->updatePath(this, linkingTo);

    m_linkingTo = linkingTo;

    hideSnappedLinks();

  }

  // autopan

  QGraphicsView *viewer = scene()->views()[0];

  viewer->setInteractive(false);

  viewer->ensureVisible(QRectF(me->scenePos(), QSizeF(1, 1)), 5, 5);

  viewer->setInteractive(true);

}

//--------------------------------------------------------

void SchematicPort::mousePressEvent(QGraphicsSceneMouseEvent *me) {

  if (!isSelected()) {

    if (me->modifiers() != Qt::ControlModifier) scene()->clearSelection();

    if (me->button() == Qt::LeftButton || me->button() == Qt::RightButton)

      getNode()->setSelected(true);

  } else {

    if (me->modifiers() == Qt::ControlModifier &&

        me->button() == Qt::LeftButton)

      getNode()->setSelected(false);

  }

  getNode()->onClicked();

  if (me->button() == Qt::LeftButton && getType() != 202  // eFxLinkPort

      && getType() != 203                                 // eFxGroupedInPort

      && getType() != 204                                 // eFxGroupedOutPort

      && getType() != 103   // eStageSplineGroupPort

      && getType() != 104   // eStageParentGroupPort

      && getType() != 105)  // eStageChildGroupPort

  {

    m_buttonState = Qt::LeftButton;

    QPointF endPos(me->pos());

    m_ghostLink = new SchematicLink(0, scene());

    m_ghostLink->setZValue(3.0);

    m_ghostLink->updatePath(this->getLinkEndPoint(), me->scenePos());

    emit(isClicked());

  }

}

//--------------------------------------------------------

void SchematicPort::mouseReleaseEvent(QGraphicsSceneMouseEvent *me) {

  if (me->modifiers() != Qt::ControlModifier && me->button() != Qt::RightButton)

    QGraphicsItem::mouseReleaseEvent(me);

  if (m_ghostLink) m_ghostLink->hide();

  if (m_buttonState == Qt::LeftButton) emit(isReleased(me->scenePos()));

  // The link is added to the scene only if the user released the left mouse

  // button over

  // a SchematicPort different from SchematicPort of the parent node.

  if (m_buttonState == Qt::LeftButton && m_linkingTo &&

      !isLinkedTo(m_linkingTo) && linkTo(m_linkingTo, true)) {

    linkTo(m_linkingTo);

    m_buttonState = Qt::NoButton;

    m_linkingTo   = 0;

    emit sceneChanged();

    emit xsheetChanged();

  } else

    showSnappedLinks();

}

//--------------------------------------------------------

void SchematicPort::removeLink(SchematicLink *link) { m_links.removeAll(link); }

//--------------------------------------------------------

void SchematicPort::eraseLink(SchematicLink *link) {

  SchematicPort *otherPort = link->getOtherPort(this);

  if (otherPort) otherPort->removeLink(link);

  removeLink(link);

  if (link->scene()) link->scene()->removeItem(link);

  delete link;

}

//--------------------------------------------------------

void SchematicPort::eraseAllLinks() {

  while (!m_links.empty()) eraseLink(m_links[0]);

}

//--------------------------------------------------------

SchematicLink *SchematicPort::makeLink(SchematicPort *port) {

  if (isLinkedTo(port) || !port) return 0;

  SchematicLink *link = new SchematicLink(0, scene());

  if (getType() == 202 && port->getType() == 202) link->setLineShaped(true);

  link->setStartPort(this);

  link->setEndPort(port);

  addLink(link);

  port->addLink(link);

  link->updatePath();

  return link;

}

//--------------------------------------------------------

bool SchematicPort::isLinkedTo(SchematicPort *port) const {

  if (m_links.size() == 0) return false;

  int i;

  for (i = 0; i < m_links.size(); i++) {

    SchematicLink *link = m_links[i];

    if (((link->getStartPort() == this && link->getEndPort() == port) ||

         (link->getEndPort() == this && link->getStartPort() == port)) &&

        link->isVisible())

      return true;

  }

  return false;

}

//--------------------------------------------------------

void SchematicPort::updateLinksGeometry() {

  int linkCount = getLinkCount();

  int i;

  for (i = 0; i < linkCount; i++) {

    SchematicLink *link      = getLink(i);

    SchematicPort *startPort = link->getStartPort();

    SchematicPort *endPort   = link->getEndPort();

    if (startPort && endPort) {

      link->updatePath(startPort, endPort);

      link->setPos(startPort->getLinkEndPoint());

    }

  }

}

//--------------------------------------------------------

QPointF SchematicPort::getLinkEndPoint() const { return scenePos() + m_hook; }

//========================================================

//

// class SchematicNode

//

//========================================================

/*! \class SchematicNode schematicnode.h "../inlcude/toonzqt/schematicnode.h"

        \brief The class provides methods to draw and handle a node item in the

   SchematicScene.

*/

/*! \fn SchematicPort *SchematicNode::getInputPort() const

        Returns the input SchematicPort of the node.

*/

/*! \fn SchematicPort *SchematicNode::getOutputPort() const

        Returns the output SchematicPort of the node.

*/

/*! \fn SchematicScene* SchematicNode::getScene() const

        Returns the scene where the node is placed.

*/

SchematicNode::SchematicNode(SchematicScene *scene)

#if QT_VERSION >= 0x050000

    : QGraphicsItem(0)

#else

    : QGraphicsItem(0, scene)

#endif

    , m_scene(scene)

    , m_width(0)

    , m_height(0)

    , m_buttonState(Qt::NoButton) {

#if QT_VERSION >= 0x050000

  scene->addItem(this);

#endif

  setFlag(QGraphicsItem::ItemIsMovable, false);

  setFlag(QGraphicsItem::ItemIsSelectable, true);

  setFlag(QGraphicsItem::ItemIsFocusable, false);

  setZValue(1.0);

}

//--------------------------------------------------------

SchematicNode::~SchematicNode() {}

//--------------------------------------------------------

/*!Reimplements the pure virtual QGraphicsItem::boundingRect() method.

*/

QRectF SchematicNode::boundingRect() const { return QRectF(0, 0, 1, 1); }

//--------------------------------------------------------

/*! Reimplements the pure virtual QGraphicsItem::paint() method.

*/

void SchematicNode::paint(QPainter *painter,

                          const QStyleOptionGraphicsItem *option,

                          QWidget *widget) {

  QPen pen;

  if (isSelected()) {

    painter->setBrush(QColor(0, 0, 0, 0));

    pen.setColor(QColor(255, 255, 255, 255));

    pen.setWidth(4.0);

    pen.setJoinStyle(Qt::RoundJoin);

    painter->setPen(pen);

    painter->drawRect(-2, -2, m_width + 4, m_height + 4);

  }

  pen.setColor(QColor(0, 0, 0, 255));

  pen.setWidth(0);

  painter->setPen(pen);

}

//--------------------------------------------------------

/*! Reimplements the QGraphicsItem::mouseMoveEvent() method.

*/

void SchematicNode::mouseMoveEvent(QGraphicsSceneMouseEvent *me) {

  QList<qgraphicsitem *=""> items = scene()->selectedItems();</qgraphicsitem>

  if (items.empty()) return;

  QPointF delta         = me->scenePos() - me->lastScenePos();

  QGraphicsView *viewer = scene()->views()[0];

  for (auto const &item : items) {

    SchematicNode *node = dynamic_cast<schematicnode *="">(item);</schematicnode>

    if (node) {

      node->setPosition(node->scenePos() + delta);

      node->setSchematicNodePos(node->scenePos());

      node->updateLinksGeometry();

    }

  }

  viewer->setInteractive(false);

  viewer->ensureVisible(QRectF(me->scenePos(), QSizeF(1, 1)), 5, 5);

  viewer->setInteractive(true);

}

//--------------------------------------------------------

void SchematicNode::mousePressEvent(QGraphicsSceneMouseEvent *me) {

  if (!isSelected()) {

    if (me->modifiers() != Qt::ControlModifier) scene()->clearSelection();

    if (me->button() == Qt::LeftButton || me->button() == Qt::RightButton)

      setSelected(true);

  } else {

    if (me->modifiers() == Qt::ControlModifier &&

        me->button() == Qt::LeftButton)

      setSelected(false);

  }

  onClicked();

}

//--------------------------------------------------------

void SchematicNode::mouseReleaseEvent(QGraphicsSceneMouseEvent *me) {

  if (me->modifiers() != Qt::ControlModifier && me->button() != Qt::RightButton)

    QGraphicsItem::mouseReleaseEvent(me);

}

//--------------------------------------------------------

/* Add a pair (portId, SchematicPort*port) in the mapping

*/

SchematicPort *SchematicNode::addPort(int portId, SchematicPort *port) {

  QMap<int, *="" schematicport="">::iterator it;</int,>

  it = m_ports.find(portId);

  if (it != m_ports.end() && m_ports[portId] != port) {

    SchematicPort *oldPort = m_ports[portId];

    m_ports.erase(it);

    scene()->removeItem(oldPort);

    delete oldPort;

  }

  m_ports[portId] = port;

  return port;

}

//--------------------------------------------------------

/*!Erase the pair (portId, SchematicPort*port) from the mapping*/

void SchematicNode::erasePort(int portId) {

  QMap<int, *="" schematicport="">::iterator it = m_ports.find(portId);</int,>

  if (it != m_ports.end()) {

    delete it.value();

    m_ports.erase(it);

  }

}

//--------------------------------------------------------

/*! Returns a pointer to the SchematicPort mapped from \b portId.\n

    Returns 0 if \b portId doesn't map no SchematicPort.