1 files changed, 14 insertions, 0 deletions
diff --git a/src/com/android/launcher3/Workspace.java b/src/com/android/launcher3/Workspace.java
index 54d153299..c9fe24748 100644
--- a/src/com/android/launcher3/Workspace.java
+++ b/src/com/android/launcher3/Workspace.java
@@ -78,6 +78,8 @@ import com.android.launcher3.widget.PendingAddWidgetInfo;
 import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
 import java.util.concurrent.atomic.AtomicInteger;
 
 /**
@@ -359,6 +361,13 @@ public class Workspace extends PagedView
         }
     }
 
+    /**
+     * @return A {@link List} of {@link Long}s representing ids of the workspace screens
+     */
+    public List<Long> getWorkspaceScreenIds() {
+        return mScreenOrder;
+    }
+
     // estimate the size of a widget with spans hSpan, vSpan. return MAX_VALUE for each
     // dimension if unsuccessful
     public int[] estimateItemSize(ItemInfo itemInfo, boolean springLoaded) {
@@ -3807,6 +3816,11 @@ public class Workspace extends PagedView
             return;
         }
 
+        // Drop is finished, so we should use our actual cell coordinates now.
+        if (mDragInfo != null) {
+            ((CellLayout.LayoutParams) mDragInfo.cell.getLayoutParams()).useTmpCoords = false;
+        }
+
         boolean beingCalledAfterUninstall = mDeferredAction != null;
 
         if (success && !(beingCalledAfterUninstall && !mUninstallSuccessful)) {