3.多线程获取音频AI的PCM数据

目次

一RV1126多线程获取音频PCM数据的流程
1.1.初始化AI模块：
1.2.启动AI模块：
1.3.开启多线程获取PCM音频数据：
二代码实战
三ffplay播放pcm数据需要的命令：

---

一RV1126多线程获取音频PCM数据的流程

RV1126多线程采集AI模块的数据，一般分为三个步骤：分别是初始化AI模块、启动AI模块开启采集、开启多线程采集AI数据并保存到本地。
1.1.初始化AI模块：

AI模块的初始化实际上就是对AI_CHN_ATTR_S的参数进行设置、然后调用RK_MPI_AI_SetChnAttr设置AI模块并使能RK_MPI_AI_EnableChn，代码如下：
AI_CHN_ATTR_S ai_chn_s;
ai_chn_s.pcAudioNode = AUDIO_PATH;
ai_chn_s.u32Channels = 2;
ai_chn_s.u32NbSamples = 1024;
ai_chn_s.u32SampleRate = 48000;
ai_chn_s.enAiLayout = AI_LAYOUT_NORMAL;
ai_chn_s.enSampleFormat = RK_SAMPLE_FMT_S16;
ret = RK_MPI_AI_SetChnAttr(AI_CHN, &ai_chn_s);
if(ret)
{
   printf("RK_MPI_AI_SetChnAttr Failed...\n");
}

ret = RK_MPI_AI_EnableChn(AI_CHN);
if(ret)
{
     printf("RK_MPI_AI_EnableChn Failed...\n");
}

1.2.启动AI模块：

设置完上面的AI模块后，就要开启AI模块的工作，使用的API是RK_MPI_AI_StartStream。代码如下：

....................................................................

ret = RK_MPI_AI_StartStream(AI_CHN);

 if(ret)

 {

        printf("RK_MPI_AI_StartStream Failed...\n");

 }

.......................................................................

1.3.开启多线程获取PCM音频数据：

开启一个线程去采集每一帧VI模块的数据，使用的API是RK_MPI_SYS_GetMediaBuffer， 模块ID是RK_ID_AI，通道号ID是AI创建的通道ID号。这个API的详细作用已经在之前的获取AI数据：
........................................
while(1)
{
  .........................
  mb = RK_MPI_SYS_GetMediaBuffer(RK_ID_AI, s32_chn_id, -1);
  fwrite(RK_MPI_MB_GetPtr(mb), RK_MPI_MB_GetSize(mb), 1, pcm_file);
.......................
}

二代码实战

1. #include <assert.h>
2. #include <fcntl.h>
3. #include <getopt.h>
4. #include <pthread.h>
5. #include <signal.h>
6. #include <stdbool.h>
7. #include <stdio.h>
8. #include <stdlib.h>
9. #include <time.h>
10. #include <unistd.h>
12. #include "rkmedia_api.h"
13. #define AUDIO_PATH "default"
14. #define AI_CHN 0
16. //获取PCM数据的线程
17. void * get_ai_pcm_thread(void * args)
18. {
19.     pthread_detach(pthread_self());
20.     FILE * pcm_file = fopen("test_ai.pcm", "w+");
21.     MEDIA_BUFFER mb ;
23.     while (1)
24.     {
25.         //获取PCM数据的线程
26.         mb = RK_MPI_SYS_GetMediaBuffer(RK_ID_AI, AI_CHN, -1);
27.         if(!mb)
28.         {
29.             printf("RK_MPI_SYS_GetMediaBuffer Failed....\n");
30.             break;
31.         }
32.         
33.         printf("get_pcm_data success...\n");
34.         fwrite(RK_MPI_MB_GetPtr(mb), RK_MPI_MB_GetSize(mb), 1, pcm_file);
35.         RK_MPI_MB_ReleaseBuffer(mb);
36.     }
37. }
40. int main(int argc, char *argv[])
41. {
42.     int ret;
43.     AI_CHN_ATTR_S ai_chn_s;
44.     ai_chn_s.pcAudioNode = AUDIO_PATH; //音频路径，默认default
45.     ai_chn_s.u32SampleRate = 48000; //音频采样率
46.     ai_chn_s.u32Channels = 2;   //音频通道数
47.     ai_chn_s.u32NbSamples = 1024; //音频采样个数，这里固定1024
48.     ai_chn_s.enSampleFormat = RK_SAMPLE_FMT_S16;
49.     ai_chn_s.enAiLayout = AI_LAYOUT_NORMAL;  //NORMAL
50.     ret = RK_MPI_AI_SetChnAttr(AI_CHN, &ai_chn_s);  //设置AI模块
51.     if(ret)
52.     {
53.         printf("Set Ai_Chn_Attr Failed....\n");
54.     }
55.     else
56.     {
57.         printf("Set Ai_Chn_Attr Success....\n");
58.     }
60.     ret = RK_MPI_AI_EnableChn(AI_CHN);  //使能AI模块
61.     if(ret)
62.     {
63.         printf("Enable Ai_Chn_Attr Failed....\n");
64.     }
66.     ret = RK_MPI_AI_StartStream(AI_CHN); //启动音频AI模块
67.     if(ret)
68.     {
69.         printf("RK_MPI_AI_StartStream Failed....\n");
70.     }
72.     pthread_t pid;
73.     pthread_create(&pid, NULL, get_ai_pcm_thread, NULL);
75.     while (1)
76.     {
77.         sleep(2);
78.     }
80.     RK_MPI_AI_DisableChn(AI_CHN);
83.     return 0;
84. }

复制代码

三ffplay播放pcm数据需要的命令：

ffplay -ar 采样率 -channels 通道数 -f 采样格式(s16le) -i pcm文件

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。