From 97b734d64cc1b7bb84d889a4b11665dde88adf7f Mon Sep 17 00:00:00 2001
From: Milton Wu <mingjuwu@google.com>
Date: Thu, 14 Sep 2023 21:56:17 +0800
Subject: [PATCH] Add pace for UdfpsEnrollHelper

Add mPace in UdfpsEnrollHelper to support different pace from
Fingerprint HAL.

Bug: 299939594
Test: Verify enroll complete animation for different pace
Change-Id: I8fb26a4c845f5f60f6f79405c9140c288540d761
---
 .../settings/biometrics/fingerprint/UdfpsEnrollHelper.java | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/com/android/settings/biometrics/fingerprint/UdfpsEnrollHelper.java b/src/com/android/settings/biometrics/fingerprint/UdfpsEnrollHelper.java
index 70fdbf08bc2..0bfb87d2074 100644
--- a/src/com/android/settings/biometrics/fingerprint/UdfpsEnrollHelper.java
+++ b/src/com/android/settings/biometrics/fingerprint/UdfpsEnrollHelper.java
@@ -76,6 +76,8 @@ public class UdfpsEnrollHelper extends InstrumentedFragment {
 
     private int mCenterTouchCount = 0;
 
+    private int mPace = 1;
+
     @Nullable
     UdfpsEnrollHelper.Listener mListener;
 
@@ -157,6 +159,9 @@ public class UdfpsEnrollHelper extends InstrumentedFragment {
             }
         }
 
+        if (mRemainingSteps > remaining) {
+            mPace = mRemainingSteps - remaining;
+        }
         mRemainingSteps = remaining;
 
         if (mListener != null && mTotalSteps != -1) {
@@ -258,7 +263,7 @@ public class UdfpsEnrollHelper extends InstrumentedFragment {
             return false;
         }
 
-        return mRemainingSteps <= 2 && mRemainingSteps >= 0;
+        return mRemainingSteps <= mPace && mRemainingSteps >= 0;
     }
 
     private int getStageThresholdSteps(int totalSteps, int stageIndex) {