[ui] Make temporal controller's forward, backward and pause

nirvn · web-flow · commit ee9dadbd64f9 · 2020-05-23T09:35:18.000+07:00
buttons behave as animation state toggle
diff --git a/python/gui/auto_generated/qgstemporalcontrollerwidget.sip.in b/python/gui/auto_generated/qgstemporalcontrollerwidget.sip.in
@@ -37,6 +37,11 @@ The dock widget retains ownership of the returned object.
 %End
 
 
+  protected:
+
+    virtual void keyPressEvent( QKeyEvent *e );
+
+
 };
 
 /************************************************************************
diff --git a/src/gui/qgstemporalcontrollerwidget.cpp b/src/gui/qgstemporalcontrollerwidget.cpp
@@ -30,11 +30,11 @@ QgsTemporalControllerWidget::QgsTemporalControllerWidget( QWidget *parent )
 
   mNavigationObject = new QgsTemporalNavigationObject( this );
 
-  connect( mForwardButton, &QPushButton::clicked, mNavigationObject, &QgsTemporalNavigationObject::playForward );
-  connect( mBackButton, &QPushButton::clicked, mNavigationObject, &QgsTemporalNavigationObject::playBackward );
+  connect( mForwardButton, &QPushButton::clicked, this, &QgsTemporalControllerWidget::togglePlayForward );
+  connect( mBackButton, &QPushButton::clicked, this, &QgsTemporalControllerWidget::togglePlayBackward );
+  connect( mStopButton, &QPushButton::clicked, this, &QgsTemporalControllerWidget::togglePause );
   connect( mNextButton, &QPushButton::clicked, mNavigationObject, &QgsTemporalNavigationObject::next );
   connect( mPreviousButton, &QPushButton::clicked, mNavigationObject, &QgsTemporalNavigationObject::previous );
-  connect( mStopButton, &QPushButton::clicked, mNavigationObject, &QgsTemporalNavigationObject::pause );
   connect( mFastForwardButton, &QPushButton::clicked, mNavigationObject, &QgsTemporalNavigationObject::skipToEnd );
   connect( mRewindButton, &QPushButton::clicked, mNavigationObject, &QgsTemporalNavigationObject::rewindToStart );
   connect( mLoopingCheckBox, &QCheckBox::toggled, this, [ = ]( bool state ) { mNavigationObject->setLooping( state ); } );
@@ -136,6 +136,77 @@ QgsTemporalControllerWidget::QgsTemporalControllerWidget( QWidget *parent )
   connect( QgsProject::instance(), &QgsProject::cleared, this, &QgsTemporalControllerWidget::onProjectCleared );
 }
 
+void QgsTemporalControllerWidget::keyPressEvent( QKeyEvent *e )
+{
+  if ( mSlider->hasFocus() && e->key() == Qt::Key_Space )
+  {
+    togglePause();
+  }
+  QWidget::keyPressEvent( e );
+}
+
+void QgsTemporalControllerWidget::togglePlayForward()
+{
+  mPlayingForward = true;
+
+  if ( mNavigationObject->animationState() != QgsTemporalNavigationObject::Forward )
+  {
+    mStopButton->setChecked( false );
+    mBackButton->setChecked( false );
+    mForwardButton->setChecked( true );
+    mNavigationObject->playForward();
+  }
+  else
+  {
+    mBackButton->setChecked( true );
+    mForwardButton->setChecked( false );
+    mNavigationObject->pause();
+  }
+}
+
+void QgsTemporalControllerWidget::togglePlayBackward()
+{
+  mPlayingForward = false;
+
+  if ( mNavigationObject->animationState() != QgsTemporalNavigationObject::Reverse )
+  {
+    mStopButton->setChecked( false );
+    mBackButton->setChecked( true );
+    mForwardButton->setChecked( false );
+    mNavigationObject->playBackward();
+  }
+  else
+  {
+    mBackButton->setChecked( true );
+    mBackButton->setChecked( false );
+    mNavigationObject->pause();
+  }
+}
+
+void QgsTemporalControllerWidget::togglePause()
+{
+  if ( mNavigationObject->animationState() != QgsTemporalNavigationObject::Idle )
+  {
+    mStopButton->setChecked( true );
+    mBackButton->setChecked( false );
+    mForwardButton->setChecked( false );
+    mNavigationObject->pause();
+  }
+  else
+  {
+    mBackButton->setChecked( mPlayingForward ? false : true );
+    mForwardButton->setChecked( mPlayingForward ? false : true );
+    if ( mPlayingForward )
+    {
+      mNavigationObject->playForward();
+    }
+    else
+    {
+      mNavigationObject->playBackward();
+    }
+  }
+}
+
 void QgsTemporalControllerWidget::updateTemporalExtent()
 {
   QgsDateTimeRange temporalExtent = QgsDateTimeRange( mStartDateTime->dateTime(),
diff --git a/src/gui/qgstemporalcontrollerwidget.h b/src/gui/qgstemporalcontrollerwidget.h
@@ -63,6 +63,10 @@ class GUI_EXPORT QgsTemporalControllerWidget : public QgsPanelWidget, private Ui
 
 #endif
 
+  protected:
+
+    void keyPressEvent( QKeyEvent *e ) override;
+
   private:
 
     /**
@@ -82,6 +86,11 @@ class GUI_EXPORT QgsTemporalControllerWidget : public QgsPanelWidget, private Ui
 
     bool mHasTemporalLayersLoaded = false;
 
+    void togglePlayForward();
+    void togglePlayBackward();
+    void togglePause();
+    bool mPlayingForward = true;
+
   private slots:
 
     /**
