InputController.cpp

Go to the documentation of this file.

// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2018 ISIS Rutherford Appleton Laboratory UKRI,
//   NScD Oak Ridge National Laboratory, European Spallation Source,
//   Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
// SPDX - License - Identifier: GPL - 3.0 +
#include "MantidQtWidgets/Common/InputController.h"
 
#include <QApplication>
#include <QCursor>
#include <QMouseEvent>
#include <QPainter>
#include <QPixmap>
 
#include <cmath>
 
namespace MantidQt::MantidWidgets {
 
//--------------------------------------------------------------------------------
 
InputController::InputController(QObject *parent, bool contextAllowed)
    : QObject(parent), m_canShowContextMenu(contextAllowed) {}
 
//--------------------------------------------------------------------------------
 
InputController3DMove::InputController3DMove(QObject *parent)
    : InputController(parent, false), m_isButtonPressed(false) {}
 
void InputController3DMove::mousePressEvent(QMouseEvent *event) {
  if (event->buttons() & Qt::MiddleButton) {
    emit initZoom(event->position().toPoint().x(), event->position().toPoint().y());
    m_isButtonPressed = true;
  } else if ((event->buttons() & Qt::LeftButton) && !m_isRotationFrozen) {
    emit initRotation(event->position().toPoint().x(), event->position().toPoint().y());
    m_isButtonPressed = true;
  } else if ((event->buttons() & Qt::RightButton) || ((event->buttons() & Qt::LeftButton) && m_isRotationFrozen)) {
    emit initTranslation(event->position().toPoint().x(), event->position().toPoint().y());
    m_isButtonPressed = true;
  }
}
 
void InputController3DMove::mouseMoveEvent(QMouseEvent *event) {
  if ((event->buttons() & Qt::LeftButton) && !m_isRotationFrozen) {
    emit rotate(event->position().toPoint().x(), event->position().toPoint().y());
  } else if ((event->buttons() & Qt::RightButton) || ((event->buttons() & Qt::LeftButton) && m_isRotationFrozen)) {
    emit translate(event->position().toPoint().x(), event->position().toPoint().y());
  } else if (event->buttons() & Qt::MiddleButton) {
    emit zoom(event->position().toPoint().x(), event->position().toPoint().y());
  }
  InputController::mouseMoveEvent(event);
}
 
void InputController3DMove::mouseReleaseEvent(QMouseEvent * /*unused*/) {
  m_isButtonPressed = false;
  emit finish();
}
 
void InputController3DMove::wheelEvent(QWheelEvent *event) {
  emit wheelZoom(static_cast<int>(event->position().x()), static_cast<int>(event->position().y()),
                 event->angleDelta().y());
}
 
void InputController3DMove::freezeRotation(bool freeze) { m_isRotationFrozen = freeze; }
 
//--------------------------------------------------------------------------------
 
InputControllerPick::InputControllerPick(QObject *parent) : InputController(parent), m_isButtonPressed(false) {}
 
void InputControllerPick::mousePressEvent(QMouseEvent *event) {
  if (event->button() == Qt::LeftButton) {
    m_isButtonPressed = true;
    m_rect.setRect(event->position().toPoint().x(), event->position().toPoint().y(), 1, 1);
    emit pickPointAt(event->position().toPoint().x(), event->position().toPoint().y());
  }
}
 
void InputControllerPick::mouseMoveEvent(QMouseEvent *event) {
  if (m_isButtonPressed) {
    m_rect.setBottomRight(QPoint(event->position().toPoint().x(), event->position().toPoint().y()));
    emit setSelection(m_rect);
  } else {
    emit touchPointAt(event->position().toPoint().x(), event->position().toPoint().y());
  }
}
 
void InputControllerPick::mouseReleaseEvent(QMouseEvent * /*unused*/) {
  m_isButtonPressed = false;
  emit finishSelection();
}
 
//--------------------------------------------------------------------------------
 
InputControllerDrawShape::InputControllerDrawShape(QObject *parent)
    : InputController(parent), m_creating(false), m_x(0), m_y(0), m_shapeType(), m_isButtonPressed(false) {}
 
void InputControllerDrawShape::mousePressEvent(QMouseEvent *event) {
  if (event->button() == Qt::LeftButton) {
    m_isButtonPressed = true;
    if (m_creating && !m_shapeType.isEmpty()) {
      emit addShape(m_shapeType, event->position().toPoint().x(), event->position().toPoint().y(), m_borderColor,
                    m_fillColor);
    } else if (event->modifiers() & Qt::ControlModifier) {
      emit selectCtrlAt(event->position().toPoint().x(), event->position().toPoint().y());
    } else {
      emit selectAt(event->position().toPoint().x(), event->position().toPoint().y());
    }
    m_x = event->position().toPoint().x();
    m_y = event->position().toPoint().y();
    m_rect.setRect(event->position().toPoint().x(), event->position().toPoint().y(), 1, 1);
  }
}
 
void InputControllerDrawShape::mouseMoveEvent(QMouseEvent *event) {
  if (m_isButtonPressed) {
    if (m_creating) {
      emit moveRightBottomTo(event->position().toPoint().x(), event->position().toPoint().y());
    } else {
      emit moveBy(event->position().toPoint().x() - m_x, event->position().toPoint().y() - m_y);
      m_rect.setBottomRight(QPoint(event->position().toPoint().x(), event->position().toPoint().y()));
      m_x = event->position().toPoint().x();
      m_y = event->position().toPoint().y();
      emit setSelection(m_rect);
    }
  } else {
    emit touchPointAt(event->position().toPoint().x(), event->position().toPoint().y());
  }
}
 
void InputControllerDrawShape::mouseReleaseEvent(QMouseEvent * /*unused*/) {
  m_isButtonPressed = false;
  m_creating = false;
  m_shapeType = "";
  emit finishSelection(m_rect);
}
 
void InputControllerDrawShape::keyPressEvent(QKeyEvent *event) {
  switch (event->key()) {
  case Qt::Key_Delete:
  case Qt::Key_Backspace:
    emit removeSelectedShapes();
    break;
  case Qt::Key_C:
    if (event->modifiers() == Qt::CTRL) {
      emit copySelectedShapes();
    }
    break;
  case Qt::Key_V:
    if (event->modifiers() == Qt::CTRL) {
      emit pasteCopiedShapes();
    }
    break;
  }
}
 
void InputControllerDrawShape::leaveEvent(QEvent * /*unused*/) { emit restoreOverrideCursor(); }
 
void InputControllerDrawShape::startCreatingShape2D(const QString &type, const QColor &borderColor,
                                                    const QColor &fillColor) {
  m_creating = true;
  m_shapeType = type;
  m_borderColor = borderColor;
  m_fillColor = fillColor;
}
 
void InputControllerDrawShape::onDisabled() {
  m_creating = false;
  emit disabled();
}
//--------------------------------------------------------------------------------
 
InputControllerMoveUnwrapped::InputControllerMoveUnwrapped(QObject *parent)
    : InputController(parent, false), m_isButtonPressed(false) {}
 
void InputControllerMoveUnwrapped::mousePressEvent(QMouseEvent *event) {
  if (event->button() == Qt::LeftButton || event->button() == Qt::RightButton) {
    m_isButtonPressed = true;
    m_rect.setTopLeft(QPoint(event->position().toPoint().x(), event->position().toPoint().y()));
  }
}
 
void InputControllerMoveUnwrapped::mouseMoveEvent(QMouseEvent *event) {
  if (m_isButtonPressed) {
    m_rect.setBottomRight(QPoint(event->position().toPoint().x(), event->position().toPoint().y()));
    emit setSelectionRect(m_rect);
  }
  InputController::mouseMoveEvent(event);
}
 
void InputControllerMoveUnwrapped::mouseReleaseEvent(QMouseEvent *event) {
  if (m_isButtonPressed && event->button() == Qt::LeftButton) {
    emit zoom();
  } else if (m_isButtonPressed && event->button() == Qt::RightButton) {
    if (m_rect.width() > 1 && m_rect.height() > 1) {
      emit unzoom();
    } else {
      emit resetZoom();
    }
  }
  m_rect = QRect(); // reset rect
  m_isButtonPressed = false;
}
 
//--------------------------------------------------------------------------------
 
InputControllerDraw::InputControllerDraw(QObject *parent)
    : InputController(parent), m_max_size(32), m_size(30), m_isLeftButtonPressed(false), m_isRightButtonPressed(false),
      m_isActive(false), m_cursor(nullptr) {}
 
InputControllerDraw::~InputControllerDraw() { delete m_cursor; }
 
void InputControllerDraw::mousePressEvent(QMouseEvent *event) {
  m_isActive = true;
  setPosition(QPoint(event->position().toPoint().x(), event->position().toPoint().y()));
  if (event->button() == Qt::LeftButton) {
    m_isLeftButtonPressed = true;
    signalLeftClick();
  } else if (event->button() == Qt::RightButton) {
    m_isRightButtonPressed = true;
    signalRightClick();
  }
}
 
void InputControllerDraw::mouseMoveEvent(QMouseEvent *event) {
  m_isActive = true;
  setPosition(QPoint(event->position().toPoint().x(), event->position().toPoint().y()));
  if (m_isLeftButtonPressed) {
    signalLeftClick();
  } else if (m_isRightButtonPressed) {
    signalRightClick();
  }
}
 
void InputControllerDraw::mouseReleaseEvent(QMouseEvent *event) {
  if (event->button() == Qt::LeftButton) {
    m_isLeftButtonPressed = false;
  } else if (event->button() == Qt::RightButton) {
    m_isRightButtonPressed = false;
  }
}
 
void InputControllerDraw::wheelEvent(QWheelEvent *event) {
  int d = m_size + (event->angleDelta().y() > 0 ? 4 : -4);
  if (d > 2 && d < m_max_size) {
    m_size = d;
    resize();
    redrawCursor();
    QApplication::restoreOverrideCursor();
    QApplication::setOverrideCursor(QCursor(*m_cursor, 0, 0));
  }
}
 
void InputControllerDraw::enterEvent(QEvent * /*unused*/) {
  redrawCursor();
  QApplication::setOverrideCursor(QCursor(*m_cursor, 0, 0));
  m_isActive = true;
}
 
void InputControllerDraw::leaveEvent(QEvent * /*unused*/) {
  QApplication::restoreOverrideCursor();
  m_isActive = false;
}
 
void InputControllerDraw::redrawCursor() {
  if (!m_cursor) {
    m_cursor = new QPixmap(m_max_size, m_max_size);
  }
  drawCursor(m_cursor);
}
 
void InputControllerDraw::signalRightClick() {}
 
//--------------------------------------------------------------------------------
 
InputControllerSelection::InputControllerSelection(QObject *parent, QPixmap *icon)
    : InputControllerDraw(parent), m_rect(0, 0, cursorSize(), cursorSize()) {
  m_image = icon;
}
 
InputControllerSelection::~InputControllerSelection() { delete m_image; }
 
void InputControllerSelection::onPaint(QPainter &painter) {
  if (isActive() && !isLeftButtonPressed()) {
    painter.drawPixmap(m_rect.bottomRight(), *m_image);
  }
}
 
void InputControllerSelection::drawCursor(QPixmap *cursor) {
  cursor->fill(QColor(255, 255, 255, 0));
  QPainter painter(cursor);
  auto size = cursorSize();
 
  auto pen = QPen(Qt::DashLine);
  QList<qreal> dashPattern;
  dashPattern << 4 << 4;
  pen.setDashPattern(dashPattern);
  pen.setColor(QColor(0, 0, 0));
  painter.setPen(pen);
  painter.drawRect(QRect(0, 0, size, size));
 
  pen.setColor(QColor(255, 255, 255));
  pen.setDashOffset(4);
  painter.setPen(pen);
  painter.drawRect(QRect(0, 0, size, size));
}
 
void InputControllerSelection::setPosition(const QPoint &pos) { m_rect.moveTopLeft(pos); }
 
void InputControllerSelection::resize() {
  auto size = cursorSize();
  m_rect.setSize(QSize(size, size));
}
 
void InputControllerSelection::signalLeftClick() { emit selection(m_rect); }
 
//--------------------------------------------------------------------------------
 
InputControllerDrawAndErase::InputControllerDrawAndErase(QObject *parent)
    : InputControllerDraw(parent), m_pos(0, 0), m_rect(8), m_creating(false) {
  makePolygon();
}
 
void InputControllerDrawAndErase::makePolygon() {
  auto r = double(cursorSize()) / 2.0;
  double a = 2.0 * M_PI / double(m_rect.size());
  for (int i = 0; i < m_rect.size(); ++i) {
    double ia = double(i) * a;
    auto x = r + static_cast<int>(r * cos(ia));
    auto y = r + static_cast<int>(r * sin(ia));
    m_rect[i] = QPointF(x, y);
  }
}
 
void InputControllerDrawAndErase::signalLeftClick() {
  auto poly = m_rect.translated(m_pos);
  if (m_creating) {
    m_creating = false;
    emit addShape(poly, m_borderColor, m_fillColor);
  } else {
    emit draw(poly);
  }
}
 
void InputControllerDrawAndErase::signalRightClick() {
  auto poly = m_rect.translated(m_pos);
  emit erase(poly);
}
 
void InputControllerDrawAndErase::drawCursor(QPixmap *cursor) {
  cursor->fill(QColor(255, 255, 255, 0));
  QPainter painter(cursor);
 
  auto bRect = m_rect.boundingRect();
  auto poly = m_rect.translated(-bRect.topLeft());
 
  auto pen = QPen(Qt::DashLine);
  QList<qreal> dashPattern;
  qreal dashLength = cursorSize() < 10 ? 1 : 2;
  dashPattern << dashLength << dashLength;
  pen.setDashPattern(dashPattern);
  pen.setColor(QColor(0, 0, 0));
  painter.setPen(pen);
  painter.drawPolygon(poly);
 
  pen.setColor(QColor(255, 255, 255));
  pen.setDashOffset(dashLength);
  painter.setPen(pen);
  painter.drawPolygon(poly);
}
 
void InputControllerDrawAndErase::setPosition(const QPoint &pos) { m_pos = pos; }
 
void InputControllerDrawAndErase::resize() { makePolygon(); }
 
void InputControllerDrawAndErase::startCreatingShape2D(const QColor &borderColor, const QColor &fillColor) {
  m_borderColor = borderColor;
  m_fillColor = fillColor;
  m_creating = true;
}
} // namespace MantidQt::MantidWidgets