From 6c6dcb2921f70adc4b16e9769967f2e8699a476e Mon Sep 17 00:00:00 2001
From: dengwei1 <dengwei1@motorola.com>
Date: Thu, 30 Dec 2021 11:22:16 +0800
Subject: [PATCH] dubai: vibrate add short index

set
short index

Change-Id: I71e53f871946b152db8a0a68f8f167ced13cb702
Signed-off-by: dengwei1 <dengwei1@motorola.com>
Reviewed-on: https://gerrit.mot.com/2159319
SME-Granted: SME Approvals Granted
SLTApproved: Slta Waiver
Tested-by: Jira Key
Reviewed-by: Wang Wang <wangwang1@lenovo.com>
Submit-Approved: Jira Key
---
 drivers/misc/awinic/aw862x/aw8622x.c | 40 +++++++++++++++++++++++++---
 drivers/misc/awinic/aw862x/aw8622x.h | 11 ++++----
 2 files changed, 43 insertions(+), 8 deletions(-)

diff --git a/drivers/misc/awinic/aw862x/aw8622x.c b/drivers/misc/awinic/aw862x/aw8622x.c
index 8a5c3ce2f5ba..b99f9cf690e7 100644
--- a/drivers/misc/awinic/aw862x/aw8622x.c
+++ b/drivers/misc/awinic/aw862x/aw8622x.c
@@ -37,7 +37,7 @@
 #include "aw8622x.h"
 #include "haptic_nv.h"
 
-#define AW8622X_Driver		"V1.5.1.0"
+#define AW8622X_Driver		"V1.5.1.1"
 
 /******************************************************
  *
@@ -3721,13 +3721,46 @@ static void aw8622x_vibrator_work_routine(struct work_struct *work)
 				aw8622x->rtp_file_num = moto_mode - AW8622X_SEQ_NO_RTP_REPEAT;
 				aw8622x->activate_mode = AW8622X_HAPTIC_LOOP_RTP_MODE;
 			}
-		} else if ( aw8622x->duration < 100 || moto_mode > 2) {
-			aw8622x->index = 1; /*1  short */
+		} else if ((aw8622x->duration < aw8622x->dts_info.duration_time[2]) || moto_mode > 2) {
+			/* mode -> index:3 -> 1 : simulate Home key or other hardware key down*/
+			/* mode -> index:4 -> 2 : simulate Home key or other hardware key up */
+			/* mode -> index:5 -> 3 : AOSP VIRTUAL_KEY */
+			/* mode -> index:6 -> 4 : AOSP CLOCK_TICK/GESTURE_END/GESTURE_START */
+			if (moto_mode > 2) { /* moto custom */
+				if (moto_mode == 3) {
+					aw8622x->index = 1;
+				} else if (moto_mode == 4) {
+					aw8622x->index = 3;
+				} else if (moto_mode == 5) {
+					aw8622x->index = 1;
+				} else if (moto_mode == 6) {
+					aw8622x->index = 2;
+				} else {
+					aw8622x->index = 1;
+				}
+			} else { /* duration ctl */
+				if ((aw8622x->duration > 0) && (aw8622x->duration <
+							aw8622x->dts_info.duration_time[0])) {
+					aw8622x->index = 3;	/* 1 level shrot shake */
+				} else if ((aw8622x->duration >= aw8622x->dts_info.duration_time[0]) &&
+					(aw8622x->duration < aw8622x->dts_info.duration_time[1])) {
+					aw8622x->index = 2;	/*2 level shrot shake*/
+				} else if ((aw8622x->duration >= aw8622x->dts_info.duration_time[1]) &&
+					(aw8622x->duration < aw8622x->dts_info.duration_time[2])) {
+					aw8622x->index = 1;	/*3 level shrot shake*/
+				}
+			}
+			if (aw8622x->index > AW8622x_MAX_RAM_MODE_NUM) {
+				aw_dev_err("%s index %d out of range use default 1\n", __func__, aw8622x->index);
+				aw8622x->index = 1;
+			}
+
 			aw8622x->activate_mode = AW8622X_HAPTIC_RAM_MODE;
 		} else {
 			aw8622x->index = 4; /* long  */
 			aw8622x->activate_mode = AW8622X_HAPTIC_RAM_LOOP_MODE;
 		}
+		aw_dev_info("%s index = %d \n", __func__, aw8622x->index);
 		/* moto mode select end */
 
 		aw_dev_info("%s activate_mode = %d \n", __func__, aw8622x->activate_mode);
@@ -3794,6 +3827,7 @@ int aw8622x_vibrator_init(struct aw8622x *aw8622x)
 
 	aw_dev_info("%s enter\n", __func__);
 
+	aw_dev_info("%s Custom Driver version %s\n", __func__, AW8622X_Driver);
 #ifdef TIMED_OUTPUT
 	aw_dev_info("%s: TIMED_OUT FRAMEWORK!\n", __func__);
 	aw8622x->vib_dev.name = "awinic_vibrator";
diff --git a/drivers/misc/awinic/aw862x/aw8622x.h b/drivers/misc/awinic/aw862x/aw8622x.h
index 23a69bb81caa..3f365e0c2ec0 100644
--- a/drivers/misc/awinic/aw862x/aw8622x.h
+++ b/drivers/misc/awinic/aw862x/aw8622x.h
@@ -126,11 +126,12 @@ enum aw8622X_awrw_flag {
 	AW8622X_READ = 1,
 };
 
-#define AW8622X_WAV_SEQ_SIZE            4
-#define AW8622X_MAX_FIRMWARE_LOAD_CNT 	20
-#define AW8622X_SEQ_NO_RTP_BASE 		102
-#define AW8622X_SEQ_NO_RTP_REPEAT 		100
-#define AW8622X_SEQ_NO_RTP_STOP 		120000
+#define AW8622X_WAV_SEQ_SIZE            (4)
+#define AW8622X_MAX_FIRMWARE_LOAD_CNT 	(20)
+#define AW8622X_SEQ_NO_RTP_BASE 		(102)
+#define AW8622X_SEQ_NO_RTP_REPEAT 		(100)
+#define AW8622X_SEQ_NO_RTP_STOP 		(120000)
+#define AW8622x_MAX_RAM_MODE_NUM		(4)
 
 #define AW8622X_REPEAT_RTP_PLAYING