1 files changed, 40 insertions, 0 deletions

diff --git a/pacman-c++/mainwidget.cpp b/pacman-c++/mainwidget.cpp
index 9d7fb14..743b57e 100644
--- a/pacman-c++/mainwidget.cpp
+++ b/pacman-c++/mainwidget.cpp
@@ -6,7 +6,10 @@
 #include "constants.h"
 
 MainWidget::MainWidget()
+  : currentKey(0)
 {
+  setFocusPolicy(Qt::StrongFocus);
+  
   QVBoxLayout *layout = new QVBoxLayout(this);
 
   QHBoxLayout *m_scoreLayout = new QHBoxLayout();
@@ -172,3 +175,40 @@ QPoint MainWidget::mapPositionToCoord(unsigned int x, unsigned int y)
   return QPoint(x * field_size[0], y * field_size[1]);
 }
 
+transmission::field_t MainWidget::translateKey(int key)
+{
+  switch(key) {
+    case Qt::Key_Up:
+      return transmission::direction_up;
+      break;
+    case Qt::Key_Down:
+      return transmission::direction_down;
+      break;
+    case Qt::Key_Left:
+      return transmission::direction_left;
+      break;
+    case Qt::Key_Right:
+      return transmission::direction_right;
+      break;
+    default:
+      return 0;
+  }
+}
+
+void MainWidget::keyPressEvent(QKeyEvent* event)
+{
+  QWidget::keyPressEvent(event);
+  currentKey = translateKey( event->key() );
+}
+
+void MainWidget::keyReleaseEvent(QKeyEvent* event)
+{
+  QWidget::keyReleaseEvent(event);
+  transmission::field_t releasedKey = translateKey(event->key());
+  if (releasedKey == currentKey) {
+    // current key got released
+    // if this is false, a key got released which has already
+    // been replaced by a different key, so this case is disregarded
+    currentKey = 0; 
+  }
+}