diff --git a/common/include/rtg_interface.h b/common/include/rtg_interface.h index 71c472b63e2f95f28ccbca7893fb9cf9d483dcbf..1d52022b12653e037932fc48ea04219b4042ac68 100644 --- a/common/include/rtg_interface.h +++ b/common/include/rtg_interface.h @@ -86,9 +86,10 @@ enum rtg_sched_cmdid { extern "C" { int EnableRtg(bool flag); - int AddThreadToRtg(int tid, int grpId, int prioType = 0); - int AddThreadsToRtg(vector tids, int grpId, int prioType = 0); - int RemoveRtgThread(int tid); + int AddThreadToRtg(int tid, int grpId, int prioType = 0, bool isBlue = false); + int AddThreadsToRtg(vector tids, int grpId, int prioType = 0, bool isBlue = false); + int RemoveRtgThread(int tid, bool isBlue = false); + int RemoveRtgThreads(vector tids, bool isBlue = false); int DestroyRtgGrp(int grpId); int SetFrameRateAndPrioType(int rtgId, int rate, int rtgType); int BeginFrameFreq(int stateParam); diff --git a/interfaces/innerkits/frameintf/rtg_interface.cpp b/interfaces/innerkits/frameintf/rtg_interface.cpp index 804183e8e890e05ad71513526533d565674e1afd..d35684d112c0d1ec8cc8abe2d0c8ba7bb71056bb 100644 --- a/interfaces/innerkits/frameintf/rtg_interface.cpp +++ b/interfaces/innerkits/frameintf/rtg_interface.cpp @@ -108,7 +108,7 @@ int EnableRtg(bool flag) return 0; }; -int AddThreadToRtg(int tid, int grpId, int prioType) +int AddThreadToRtg(int tid, int grpId, int prioType, [[maybe_unused]] bool isBlue) { if (g_fd < 0) { return g_fd; @@ -134,7 +134,7 @@ int AddThreadToRtg(int tid, int grpId, int prioType) return ret; } -int AddThreadsToRtg(vector tids, int grpId, int prioType) +int AddThreadsToRtg(vector tids, int grpId, int prioType, [[maybe_unused]] bool isBlue) { struct rtg_grp_data grp_data; int ret; @@ -169,7 +169,7 @@ int AddThreadsToRtg(vector tids, int grpId, int prioType) return ret; }; -int RemoveRtgThread(int tid) +int RemoveRtgThread(int tid, [[maybe_unused]] bool isBlue) { if (g_fd < 0) { return g_fd; @@ -189,6 +189,35 @@ int RemoveRtgThread(int tid) return ret; }; +int RemoveRtgThreads(vector tids, [[maybe_unused]] bool isBlue) +{ + struct rtg_grp_data grp_data; + int ret; + if (g_fd < 0) { + return g_fd; + } + (void)memset_s(&grp_data, sizeof(struct rtg_grp_data), 0, sizeof(struct rtg_grp_data)); + int num = static_cast(tids.size()); + if (num > MAX_TID_NUM) { + return -1; + } + grp_data.tid_num = num; + grp_data.rtg_cmd = CMD_REMOVE_RTG_THREAD; + for (int i = 0; i < num; i++) { + if (tids[i] < 0) { + return -1; + } + grp_data.tids[i] = tids[i]; + } + ret = ioctl(g_fd, CMD_ID_SET_RTG, &grp_data); + if (ret < 0) { + RME_LOGE("remove grp threads failed, errno = %{public}d (%{public}s)", errno, strerror(errno)); + } else { + RME_LOGI("remove grp threads success, get rtg id %{public}d.", ret); + } + return ret; +} + int DestroyRtgGrp(int grpId) { if (g_fd < 0) {