From e117fd0b9a011f230e3954738e7e321724ccee09 Mon Sep 17 00:00:00 2001
From: Udam Saini <udam@google.com>
Date: Thu, 28 Jan 2016 18:12:53 -0800
Subject: [PATCH] Hides retry button in setup wizard during confirm stage.

bug:24803305
Change-Id: Ib5f47bbd266f50e74515594cc90f73953f8ec8ce
---
 .../android/settings/SetupChooseLockPattern.java   | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/com/android/settings/SetupChooseLockPattern.java b/src/com/android/settings/SetupChooseLockPattern.java
index 69e60be029e..3a548bc7a98 100644
--- a/src/com/android/settings/SetupChooseLockPattern.java
+++ b/src/com/android/settings/SetupChooseLockPattern.java
@@ -136,6 +136,20 @@ public class SetupChooseLockPattern extends ChooseLockPattern {
             super.updateStage(stage);
             // Only enable the button for retry
             mRetryButton.setEnabled(stage == Stage.FirstChoiceValid);
+
+            switch (stage) {
+                case Introduction:
+                case HelpScreen:
+                case ChoiceTooShort:
+                case FirstChoiceValid:
+                    mRetryButton.setVisibility(View.VISIBLE);
+                    break;
+                case NeedToConfirm:
+                case ConfirmWrong:
+                case ChoiceConfirmed:
+                    mRetryButton.setVisibility(View.INVISIBLE);
+                    break;
+            }
         }
 
         @Override