add internal state saving system to save view state in rotation,...

DanielMartinus · Aug 11, 2018 · d070872 · d070872
1 parent 531a451
commit d070872
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 2 deletions.
diff --git a/app/src/main/java/nl/dionsegijn/steppertouchdemo/MainActivity.kt b/app/src/main/java/nl/dionsegijn/steppertouchdemo/MainActivity.kt
@@ -1,9 +1,7 @@
 package nl.dionsegijn.steppertouchdemo
 
 import android.os.Bundle
-import android.support.v4.view.ViewCompat
 import android.support.v7.app.AppCompatActivity
-import android.util.Log
 import android.widget.Toast
 import nl.dionsegijn.steppertouch.OnStepCallback
 import nl.dionsegijn.steppertouch.StepperTouch

diff --git a/library/src/main/java/nl/dionsegijn/steppertouch/StepperTouch.kt b/library/src/main/java/nl/dionsegijn/steppertouch/StepperTouch.kt
@@ -6,6 +6,8 @@ import android.graphics.Canvas
 import android.graphics.Path
 import android.graphics.RectF
 import android.graphics.drawable.GradientDrawable
+import android.os.Bundle
+import android.os.Parcelable
 import android.support.animation.SpringAnimation
 import android.support.annotation.ColorRes
 import android.support.v4.content.ContextCompat
@@ -23,6 +25,12 @@ import android.widget.TextView
  * Created by dionsegijn on 3/19/17.
  */
 class StepperTouch : FrameLayout, OnStepCallback {
+
+    object VIEW_RESTORE {
+        val COUNT_KEY = "count"
+        val SUPER_STATE_KEY = "superState"
+    }
+
     constructor(context: Context) : super(context) {
         prepareElements()
     }
@@ -106,6 +114,23 @@ class StepperTouch : FrameLayout, OnStepCallback {
         addView(viewStepper)
     }
 
+    public override fun onSaveInstanceState(): Parcelable? {
+        val superState = super.onSaveInstanceState()
+        val bundle = Bundle()
+        bundle.putParcelable(VIEW_RESTORE.SUPER_STATE_KEY, superState)
+        bundle.putInt(VIEW_RESTORE.COUNT_KEY, stepper.getValue())
+        return bundle
+    }
+
+    public override fun onRestoreInstanceState(state: Parcelable) {
+        val bundle: Bundle = state as Bundle
+        val superState = bundle.getParcelable<Parcelable>(VIEW_RESTORE.SUPER_STATE_KEY)
+        super.onRestoreInstanceState(superState)
+
+        val count = bundle.getInt(VIEW_RESTORE.COUNT_KEY)
+        stepper.setValue(count)
+    }
+
     fun enableSideTapForView(textView: View) {
         textView.setOnTouchListener { v, event ->
             if (event.action == MotionEvent.ACTION_DOWN) {