ScreenTranslator/SelectionDialog.cpp

#include "SelectionDialog.h"
#include "ui_SelectionDialog.h"
#include "LanguageHelper.h"
#include "StAssert.h"

#include <QMouseEvent>
#include <QPainter>
#include <QDebug>
#include <QMenu>

SelectionDialog::SelectionDialog (const LanguageHelper &dictionary, QWidget *parent) :
  QDialog (parent),
  ui (new Ui::SelectionDialog), dictionary_ (dictionary),
  languageMenu_ (new QMenu) {
  ui->setupUi (this);
  setWindowFlags (Qt::FramelessWindowHint | Qt::NoDropShadowWindowHint |
                  Qt::WindowStaysOnTopHint);

  ui->label->setAutoFillBackground (false);
  ui->label->installEventFilter (this);

  updateMenu ();
}

SelectionDialog::~SelectionDialog () {
  delete ui;
}

void SelectionDialog::updateMenu () {
  Q_CHECK_PTR (languageMenu_);
  languageMenu_->clear ();
  QStringList languages = dictionary_.availableOcrLanguagesUi ();
  if (languages.isEmpty ()) {
    return;
  }

  const int max = 10;

  if (languages.size () <= max) {
    foreach (const QString &language, languages) {
      languageMenu_->addAction (language);
    }
  }
  else {
    int subIndex = max;
    QMenu *subMenu = NULL;
    QString prevLetter;
    foreach (const QString &language, languages) {
      QString curLetter = language.left (1);
      if (++subIndex >= max && prevLetter != curLetter) {
        if (subMenu != NULL) {
          subMenu->setTitle (subMenu->title () + " - " + prevLetter);
        }
        subMenu = languageMenu_->addMenu (curLetter);
        subIndex = 0;
      }
      prevLetter = curLetter;
      subMenu->addAction (language);
    }
    subMenu->setTitle (subMenu->title () + " - " + prevLetter);
  }
}

bool SelectionDialog::eventFilter (QObject *object, QEvent *event) {
  if (object != ui->label) {
    return QDialog::eventFilter (object, event);
  }

  if (event->type () == QEvent::Show) {
    startSelectPos_ = currentSelectPos_ = QPoint ();
  }
  else if (event->type () == QEvent::MouseButtonPress) {
    QMouseEvent *mouseEvent = static_cast <QMouseEvent *> (event);
    if ((mouseEvent->button () == Qt::LeftButton ||
         mouseEvent->button () == Qt::RightButton) && startSelectPos_.isNull ()) {
      startSelectPos_ = mouseEvent->pos ();
    }
  }
  else if (event->type () == QEvent::MouseMove) {
    QMouseEvent *mouseEvent = static_cast <QMouseEvent *> (event);
    if ((mouseEvent->buttons () & Qt::LeftButton ||
         mouseEvent->buttons () & Qt::RightButton) && !startSelectPos_.isNull ()) {
      currentSelectPos_ = mouseEvent->pos ();
      ui->label->repaint ();
    }
  }
  else if (event->type () == QEvent::Paint) {
    QRect selection = QRect (startSelectPos_, currentSelectPos_).normalized ();
    if (selection.isValid ()) {
      QPainter painter (ui->label);
      painter.setPen (Qt::red);
      painter.drawRect (selection);
    }
  }
  else if (event->type () == QEvent::MouseButtonRelease) {
    QMouseEvent *mouseEvent = static_cast <QMouseEvent *> (event);
    if (mouseEvent->button () == Qt::LeftButton ||
        mouseEvent->button () == Qt::RightButton) {
      if (startSelectPos_.isNull () || currentPixmap_.isNull ()) {
        return QDialog::eventFilter (object, event);
      }
      QPoint endPos = mouseEvent->pos ();
      QRect selection = QRect (startSelectPos_, endPos).normalized ();
      QPixmap selectedPixmap = currentPixmap_.copy (selection);
      if (!selectedPixmap.isNull ()) {
        ProcessingItem item;
        item.source = selectedPixmap;
        item.screenPos = selection.topLeft ();

        if (mouseEvent->button () == Qt::RightButton &&
            !languageMenu_->children ().isEmpty ()) {
          QAction *action = languageMenu_->exec (QCursor::pos ());
          if (action == NULL) {
            reject ();
            return QDialog::eventFilter (object, event);
          }
          item.ocrLanguage = dictionary_.ocrUiToCode (action->text ());
          ST_ASSERT (!item.ocrLanguage.isEmpty ());
          item.sourceLanguage = dictionary_.translateForOcrCode (item.ocrLanguage);
          ST_ASSERT (!item.sourceLanguage.isEmpty ());
        }
        emit selected (item);
        accept ();
      }
    }
  }
  return QDialog::eventFilter (object, event);
}

void SelectionDialog::setPixmap (QPixmap pixmap) {
  ST_ASSERT (!pixmap.isNull ());
  currentPixmap_ = pixmap;
  QPalette palette = this->palette ();
  palette.setBrush (this->backgroundRole (), pixmap);
  this->setPalette (palette);
  this->resize (pixmap.size ());

  show ();
  setFocus ();
}