From c5f09ae1b334ac142797ed35f424df2ca7e734bc Mon Sep 17 00:00:00 2001
From: maan4 <maan4@huawei.com>
Date: Thu, 15 May 2025 19:39:56 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=89=8B=E8=A1=A8=E9=9C=87?=
 =?UTF-8?q?=E5=8A=A8=E4=B8=A2=E5=A4=B1=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: maan4 <maan4@huawei.com>
---
 .../include/miscdevice_service.h              |  3 ++
 .../src/miscdevice_service.cpp                | 30 +++++++++++++++----
 2 files changed, 28 insertions(+), 5 deletions(-)

diff --git a/services/miscdevice_service/include/miscdevice_service.h b/services/miscdevice_service/include/miscdevice_service.h
index 2c15952..c73c4ad 100644
--- a/services/miscdevice_service/include/miscdevice_service.h
+++ b/services/miscdevice_service/include/miscdevice_service.h
@@ -75,6 +75,9 @@ private:
     bool InitInterface();
     bool InitLightInterface();
     std::string GetPackageName(AccessTokenID tokenId);
+#ifdef OHOS_BUILD_ENABLE_VIBRATOR_PRESET_INFO
+    int32_t FastVibratorEffect(const VibrateInfo &info);
+#endif // OHOS_BUILD_ENABLE_VIBRATOR_PRESET_INFO
     void StartVibrateThread(VibrateInfo info);
     int32_t StopVibratorService(int32_t vibratorId);
     void StopVibrateThread();
diff --git a/services/miscdevice_service/src/miscdevice_service.cpp b/services/miscdevice_service/src/miscdevice_service.cpp
index 41447db..03faa8d 100644
--- a/services/miscdevice_service/src/miscdevice_service.cpp
+++ b/services/miscdevice_service/src/miscdevice_service.cpp
@@ -73,7 +73,7 @@ VibratorCapacity g_capacity;
 const std::string PHONE_TYPE = "phone";
 #endif // OHOS_BUILD_ENABLE_VIBRATOR_CUSTOM
 #ifdef OHOS_BUILD_ENABLE_VIBRATOR_PRESET_INFO
-constexpr int32_t VIBRATOR_DURATION = 50;
+constexpr int32_t SHORT_VIBRATOR_DURATION = 50;
 #endif // OHOS_BUILD_ENABLE_VIBRATOR_PRESET_INFO
 }  // namespace
 
@@ -501,10 +501,10 @@ void MiscdeviceService::StartVibrateThread(VibrateInfo info)
     }
 #ifdef OHOS_BUILD_ENABLE_VIBRATOR_PRESET_INFO
     VibrateInfo currentVibrateInfo = vibratorThread_->GetCurrentVibrateInfo();
-    if (info.duration <= VIBRATOR_DURATION && currentVibrateInfo.duration <= VIBRATOR_DURATION &&
-        info.mode == VIBRATE_PRESET && currentVibrateInfo.mode == VIBRATE_PRESET) {
+    if (info.duration <= SHORT_VIBRATOR_DURATION  && currentVibrateInfo.duration <= SHORT_VIBRATOR_DURATION &&
+        info.mode == VIBRATE_PRESET && currentVibrateInfo.mode == VIBRATE_PRESET && info.count == 1) {
         vibratorThread_->UpdateVibratorEffect(info);
-        vibratorThread_->Start("VibratorThread");
+        FastVibratorEffect(info);
     } else {
 #endif // OHOS_BUILD_ENABLE_VIBRATOR_PRESET_INFO
         StopVibrateThread();
@@ -545,7 +545,11 @@ int32_t MiscdeviceService::StopVibratorByMode(int32_t vibratorId, const std::str
         MISC_HILOGE("CheckVibratePermission failed, ret:%{public}d", ret);
         return PERMISSION_DENIED;
     }
-    if ((vibratorThread_ == nullptr) || (!vibratorThread_->IsRunning())) {
+    if ((vibratorThread_ == nullptr) || (!vibratorThread_->IsRunning()
+#ifdef OHOS_BUILD_ENABLE_VIBRATOR_PRESET_INFO
+    && !vibratorHdiConnection_.IsVibratorRunning()
+#endif // OHOS_BUILD_ENABLE_VIBRATOR_PRESET_INFO
+    )) {
 #if defined (OHOS_BUILD_ENABLE_VIBRATOR_CUSTOM) && defined (HDF_DRIVERS_INTERFACE_VIBRATOR)
         if (vibratorHdiConnection_.IsVibratorRunning()) {
             vibratorHdiConnection_.Stop(HDF_VIBRATOR_MODE_PRESET);
@@ -563,6 +567,11 @@ int32_t MiscdeviceService::StopVibratorByMode(int32_t vibratorId, const std::str
         return ERROR;
     }
     StopVibrateThread();
+#ifdef OHOS_BUILD_ENABLE_VIBRATOR_PRESET_INFO
+    if (vibratorHdiConnection_.IsVibratorRunning) {
+        vibratorHdiConnection_.Stop(HDF_VIBRATOR_MODE_PRESET);
+    }
+#endif // OHOS_BUILD_ENABLE_VIBRATOR_PRESET_INFO
     std::string packageName = GetPackageName(GetCallingTokenID());
     std::string curVibrateTime = GetCurrentTime();
     MISC_HILOGI("Stop vibrator, currentTime:%{public}s, package:%{public}s, pid:%{public}d, vibratorId:%{public}d,"
@@ -1111,5 +1120,16 @@ int32_t MiscdeviceService::GetVibratorCapacity(VibratorCapacity &capacity)
     capacity = g_capacity;
     return ERR_OK;
 }
+
+#ifdef OHOS_BUILD_ENABLE_VIBRATOR_PRESET_INFO
+int32_t MiscdeviceService::FastVibratorEffect(const VibrateInfo &info)
+{
+    int32_t ret = VibratorDevice.StartByIntensity(info.effect, info.intensity);
+    if (ret != SUCCESS) {
+        MISC_HILOGE("Vibrate effect %{public}s failed.", info.effect.c_str());
+    }
+    return NO_ERROR;
+}
+#endif // OHOS_BUILD_ENABLE_VIBRATOR_PRESET_INFO
 }  // namespace Sensors
 }  // namespace OHOS
-- 
Gitee