From e17e19c0bd78348b0452f5b00846b2a63a749d33 Mon Sep 17 00:00:00 2001
From: Michael Jurka <mikejurka@google.com>
Date: Tue, 28 Sep 2010 11:01:39 -0700
Subject: [PATCH] Making "tap to add" in customization mode automatically add
 to the current screen

---
 .../android/launcher2/CustomizePagedView.java | 19 +++++++++++++++++--
 src/com/android/launcher2/PagedView.java      |  4 ++--
 2 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/src/com/android/launcher2/CustomizePagedView.java b/src/com/android/launcher2/CustomizePagedView.java
index f55b46e26c..03c4bce7ee 100644
--- a/src/com/android/launcher2/CustomizePagedView.java
+++ b/src/com/android/launcher2/CustomizePagedView.java
@@ -368,12 +368,27 @@ public class CustomizePagedView extends PagedView
                 c.setChecked(!wasChecked);
 
                 // End the current choice mode when we have no items selected
-                if (!c.isChecked()) {
+                /*if (!c.isChecked()) {
                     endChoiceMode();
                 } else if (isChoiceMode(CHOICE_MODE_NONE)) {
                     endChoiceMode();
                     startChoiceMode(CHOICE_MODE_SINGLE, this);
-                }
+                }*/
+                mChoiceMode = CHOICE_MODE_SINGLE;
+
+                Workspace w = mLauncher.getWorkspace();
+                int currentWorkspaceScreen = mLauncher.getCurrentWorkspaceScreen();
+                final CellLayout cl = (CellLayout)w.getChildAt(currentWorkspaceScreen);
+                cl.setHover(true);
+
+                animateClickFeedback(v, new Runnable() {
+                    @Override
+                    public void run() {
+                        cl.setHover(false);
+                        mLauncher.onWorkspaceClick(cl);
+                        mChoiceMode = CHOICE_MODE_NONE;
+                    }
+                });
             }
             return;
         }
diff --git a/src/com/android/launcher2/PagedView.java b/src/com/android/launcher2/PagedView.java
index 17d18f8bf2..5aec48ebec 100644
--- a/src/com/android/launcher2/PagedView.java
+++ b/src/com/android/launcher2/PagedView.java
@@ -104,7 +104,7 @@ public abstract class PagedView extends ViewGroup {
     // Multiple selection mode is not supported by all Launcher actions atm
     protected static final int CHOICE_MODE_MULTIPLE = 2;
 
-    private int mChoiceMode;
+    protected int mChoiceMode;
     private ActionMode mActionMode;
 
     protected PagedViewIconCache mPageViewIconCache;
@@ -1084,7 +1084,7 @@ public abstract class PagedView extends ViewGroup {
         if (!isChoiceMode(CHOICE_MODE_NONE)) {
             mChoiceMode = CHOICE_MODE_NONE;
             resetCheckedGrandchildren();
-            mActionMode.finish();
+            if (mActionMode != null) mActionMode.finish();
             mActionMode = null;
         }
     }