1 /* SPDX-License-Identifier: Apache-2.0 OR MIT */
2 /*
3 * Copyright (c) 2025 Rockchip Electronics Co., Ltd.
4 */
5
6 #define MODULE_TAG "kmpp_meta_test"
7
8 #include <pthread.h>
9
10 #include "mpp_time.h"
11 #include "mpp_debug.h"
12 #include "kmpp_meta_impl.h"
13
14 #define TEST_MAX 1
15 #define LOOP_MAX 10
16
meta_test(void * param)17 void *meta_test(void *param)
18 {
19 RK_S32 loop_max = LOOP_MAX;
20 RK_S64 time_start;
21 RK_S64 time_end;
22 MppMeta meta[LOOP_MAX];
23 MPP_RET ret = MPP_OK;
24 KmppShmPtr frame;
25 KmppShmPtr packet;
26 KmppShmPtr buffer;
27 KmppShmPtr sptr;
28 void *ptr;
29 RK_S32 val;
30 RK_S32 i;
31
32 time_start = mpp_time();
33
34 frame.uaddr = 0;
35 frame.kaddr = 0;
36 packet.uaddr = 0;
37 packet.kaddr = 0;
38 buffer.uaddr = 0;
39 buffer.kaddr = 0;
40 sptr.uaddr = 0;
41 sptr.kaddr = 0;
42
43 for (i = 0; i < loop_max; i++) {
44 ret |= kmpp_meta_get_f(&meta[i]);
45 mpp_assert(meta[i]);
46 }
47
48 for (i = 0; i < loop_max; i++) {
49 /* set */
50 ret |= kmpp_meta_set_shm(meta[i], KEY_INPUT_FRAME, &frame);
51 ret |= kmpp_meta_set_shm(meta[i], KEY_INPUT_PACKET, &packet);
52 ret |= kmpp_meta_set_shm(meta[i], KEY_OUTPUT_FRAME, &frame);
53 ret |= kmpp_meta_set_shm(meta[i], KEY_OUTPUT_PACKET, &packet);
54
55 ret |= kmpp_meta_set_shm(meta[i], KEY_MOTION_INFO, &sptr);
56 ret |= kmpp_meta_set_shm(meta[i], KEY_HDR_INFO, &sptr);
57
58 ret |= kmpp_meta_set_s32(meta[i], KEY_INPUT_BLOCK, 0);
59 ret |= kmpp_meta_set_s32(meta[i], KEY_OUTPUT_BLOCK, 0);
60 ret |= kmpp_meta_set_s32(meta[i], KEY_INPUT_IDR_REQ, 0);
61 ret |= kmpp_meta_set_s32(meta[i], KEY_OUTPUT_INTRA, 0);
62
63 ret |= kmpp_meta_set_s32(meta[i], KEY_TEMPORAL_ID, 0);
64 ret |= kmpp_meta_set_s32(meta[i], KEY_LONG_REF_IDX, 0);
65 ret |= kmpp_meta_set_s32(meta[i], KEY_ENC_AVERAGE_QP, 0);
66
67 //ret |= kmpp_meta_set_shm(meta[i], KEY_ROI_DATA, NULL);
68 ret |= kmpp_meta_set_shm(meta[i], KEY_OSD_DATA, NULL);
69 ret |= kmpp_meta_set_shm(meta[i], KEY_OSD_DATA2, NULL);
70 ret |= kmpp_meta_set_shm(meta[i], KEY_USER_DATA, NULL);
71 ret |= kmpp_meta_set_shm(meta[i], KEY_USER_DATAS, NULL);
72
73 ret |= kmpp_meta_set_shm(meta[i], KEY_QPMAP0, NULL);
74 ret |= kmpp_meta_set_shm(meta[i], KEY_NPU_SOBJ_FLAG, NULL);
75 ret |= kmpp_meta_set_ptr(meta[i], KEY_NPU_UOBJ_FLAG, NULL);
76
77 ret |= kmpp_meta_set_s32(meta[i], KEY_ENC_MARK_LTR, 0);
78 ret |= kmpp_meta_set_s32(meta[i], KEY_ENC_USE_LTR, 0);
79 ret |= kmpp_meta_set_s32(meta[i], KEY_ENC_FRAME_QP, 0);
80 ret |= kmpp_meta_set_s32(meta[i], KEY_ENC_BASE_LAYER_PID, 0);
81
82 /* get */
83 ret |= kmpp_meta_get_shm(meta[i], KEY_INPUT_FRAME, &frame);
84 ret |= kmpp_meta_get_shm(meta[i], KEY_INPUT_PACKET, &packet);
85 ret |= kmpp_meta_get_shm(meta[i], KEY_OUTPUT_FRAME, &frame);
86 ret |= kmpp_meta_get_shm(meta[i], KEY_OUTPUT_PACKET, &packet);
87
88 ret |= kmpp_meta_get_shm(meta[i], KEY_MOTION_INFO, &buffer);
89 ret |= kmpp_meta_get_shm(meta[i], KEY_HDR_INFO, &buffer);
90
91 ret |= kmpp_meta_get_s32(meta[i], KEY_INPUT_BLOCK, &val);
92 ret |= kmpp_meta_get_s32(meta[i], KEY_OUTPUT_BLOCK, &val);
93 ret |= kmpp_meta_get_s32(meta[i], KEY_INPUT_IDR_REQ, &val);
94 ret |= kmpp_meta_get_s32(meta[i], KEY_OUTPUT_INTRA, &val);
95
96 ret |= kmpp_meta_get_s32(meta[i], KEY_TEMPORAL_ID, &val);
97 ret |= kmpp_meta_get_s32(meta[i], KEY_LONG_REF_IDX, &val);
98 ret |= kmpp_meta_get_s32(meta[i], KEY_ENC_AVERAGE_QP, &val);
99
100 //ret |= kmpp_meta_get_shm(meta[i], KEY_ROI_DATA, &sptr);
101 ret |= kmpp_meta_get_shm(meta[i], KEY_OSD_DATA, &sptr);
102 ret |= kmpp_meta_get_shm(meta[i], KEY_OSD_DATA2, &sptr);
103 ret |= kmpp_meta_get_shm(meta[i], KEY_USER_DATA, &sptr);
104 ret |= kmpp_meta_get_shm(meta[i], KEY_USER_DATAS, &sptr);
105
106 ret |= kmpp_meta_get_shm(meta[i], KEY_QPMAP0, &buffer);
107 ret |= kmpp_meta_get_shm(meta[i], KEY_NPU_SOBJ_FLAG, &sptr);
108 ret |= kmpp_meta_get_ptr(meta[i], KEY_NPU_UOBJ_FLAG, &ptr);
109
110 ret |= kmpp_meta_get_s32(meta[i], KEY_ENC_MARK_LTR, &val);
111 ret |= kmpp_meta_get_s32(meta[i], KEY_ENC_USE_LTR, &val);
112 ret |= kmpp_meta_get_s32(meta[i], KEY_ENC_FRAME_QP, &val);
113 ret |= kmpp_meta_get_s32(meta[i], KEY_ENC_BASE_LAYER_PID, &val);
114 }
115
116 for (i = 0; i < loop_max; i++) {
117 ret |= kmpp_meta_put_f(meta[i]);
118 }
119
120 time_end = mpp_time();
121
122 if (ret)
123 mpp_log("meta setting and getting, ret %d\n", ret);
124
125 *((RK_S64 *)param) = (time_end - time_start) / loop_max;
126
127 return NULL;
128 }
129
main()130 int main()
131 {
132 pthread_t thds[TEST_MAX];
133 RK_S64 times[TEST_MAX];
134 RK_S32 thd_cnt = TEST_MAX;
135 RK_S64 avg_time = 0;
136 pthread_attr_t attr;
137 RK_S32 i;
138
139 pthread_attr_init(&attr);
140 pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_JOINABLE);
141
142 mpp_log(MODULE_TAG " start\n");
143
144 for (i = 0; i < thd_cnt; i++)
145 pthread_create(&thds[i], &attr, meta_test, ×[i]);
146
147 for (i = 0; i < thd_cnt; i++)
148 pthread_join(thds[i], NULL);
149
150 for (i = 0; i < thd_cnt; i++)
151 avg_time += times[i];
152
153 mpp_log(MODULE_TAG " %d threads %d loop config avg %lld us",
154 thd_cnt, LOOP_MAX, avg_time / thd_cnt);
155
156 mpp_log(MODULE_TAG " done\n");
157
158 return 0;
159 }
160