Make usage of BundleSavedState more robust

app/src/main/java/org/isoron/uhabits/activities/common/views/BundleSavedState.java

@@ -20,16 +20,19 @@
 package org.isoron.uhabits.activities.common.views;
 
 import android.os.*;
+import android.support.v4.os.*;
 
 public class BundleSavedState extends android.support.v4.view.AbsSavedState
 {
     public static final Parcelable.Creator<BundleSavedState> CREATOR =
-        new Parcelable.Creator<BundleSavedState>()
+        ParcelableCompat.newCreator(
+            new ParcelableCompatCreatorCallbacks<BundleSavedState>()
             {
                 @Override
-            public BundleSavedState createFromParcel(Parcel source)
+                public BundleSavedState createFromParcel(Parcel source,
+                                                         ClassLoader loader)
                 {
-                return new BundleSavedState(source, getClass().getClassLoader());
+                    return new BundleSavedState(source, loader);
                 }
 
                 @Override
@@ -37,7 +40,7 @@ public class BundleSavedState extends android.support.v4.view.AbsSavedState
                 {
                     return new BundleSavedState[size];
                 }
-        };
+            });
 
     public final Bundle bundle;
 
@@ -50,7 +53,7 @@ public class BundleSavedState extends android.support.v4.view.AbsSavedState
     public BundleSavedState(Parcel source, ClassLoader loader)
     {
         super(source, loader);
-        this.bundle = source.readBundle(getClass().getClassLoader());
+        this.bundle = source.readBundle(loader);
     }
 
     @Override

app/src/main/java/org/isoron/uhabits/activities/common/views/ScrollableChart.java

@@ -107,6 +107,12 @@ public abstract class ScrollableChart extends View
     @Override
     public void onRestoreInstanceState(Parcelable state)
     {
+        if(!(state instanceof BundleSavedState))
+        {
+            super.onRestoreInstanceState(state);
+            return;
+        }
+
         BundleSavedState bss = (BundleSavedState) state;
         int x = bss.bundle.getInt("x");
         int y = bss.bundle.getInt("y");

app/src/main/java/org/isoron/uhabits/activities/habits/list/views/HabitCardListView.java

@@ -153,6 +153,12 @@ public class HabitCardListView extends RecyclerView
     @Override
     protected void onRestoreInstanceState(Parcelable state)
     {
+        if(!(state instanceof BundleSavedState))
+        {
+            super.onRestoreInstanceState(state);
+            return;
+        }
+
         BundleSavedState bss = (BundleSavedState) state;
         dataOffset = bss.bundle.getInt("dataOffset");
         super.onRestoreInstanceState(bss.getSuperState());