256 lines
8.5 KiB
C
256 lines
8.5 KiB
C
|
#include "basic_include.h"
|
|||
|
|
#include "csi_kernel.h"
|
|||
|
|
#include "lib/multimedia/msi.h"
|
|||
|
|
#include "lib/multimedia/framebuff.h"
|
|||
|
|
#include "osal_file.h"
|
|||
|
|
#include "wave_code.h"
|
|||
|
|
|
|||
|
|
#define MAX_WAVE_ENCODE_RXBUF 4
|
|||
|
|
|
|||
|
|
struct wave_encode_struct {
|
|||
|
|
struct os_event event;
|
|||
|
|
struct msi *msi;
|
|||
|
|
struct msi *src_msi;
|
|||
|
|
void *task_hdl;
|
|||
|
|
void *wave_fp;
|
|||
|
|
uint8_t destroy_self;
|
|||
|
|
uint8_t next_status;
|
|||
|
|
uint8_t current_status;
|
|||
|
|
uint32_t samplerate;
|
|||
|
|
uint32_t data_size;
|
|||
|
|
TYPE_WAVE_HEAD wave_head;
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
const unsigned char wav_header[] = {
|
|||
|
|
'R', 'I', 'F', 'F', // "RIFF" 标志
|
|||
|
|
0, 0, 0, 0, // 文件长度
|
|||
|
|
'W', 'A', 'V', 'E', // "WAVE" 标志
|
|||
|
|
'f', 'm', 't', ' ', // "fmt" 标志
|
|||
|
|
16, 0, 0, 0, // 过渡字节（不定）
|
|||
|
|
0x01, 0x00, // 格式类别
|
|||
|
|
0x01, 0x00, // 声道数
|
|||
|
|
0, 0, 0, 0, // 采样率
|
|||
|
|
0, 0, 0, 0, // 位速
|
|||
|
|
0x01, 0x00, // 一个采样多声道数据块大小
|
|||
|
|
0x10, 0x00, // 一个采样占的 bit 数
|
|||
|
|
'd', 'a', 't', 'a', // 数据标记符＂data ＂
|
|||
|
|
0, 0, 0, 0 // 语音数据的长度，比文件长度小42一般。这个是计算音频播放时长的关键参数~
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
static void wave_encode_thread(void *d)
|
|||
|
|
{
|
|||
|
|
uint8_t *data = NULL;
|
|||
|
|
uint32_t data_len = 0;
|
|||
|
|
struct wave_encode_struct *s = (struct wave_encode_struct *)d;
|
|||
|
|
struct framebuff *frame_buf = NULL;
|
|||
|
|
|
|||
|
|
s->msi->enable = 1;
|
|||
|
|
msi_get(s->msi);
|
|||
|
|
|
|||
|
|
while(1)
|
|||
|
|
{
|
|||
|
|
frame_buf = msi_get_fb(s->msi, 0);
|
|||
|
|
if(frame_buf)
|
|||
|
|
{
|
|||
|
|
if(s->next_status == AUCODEC_PAUSE)
|
|||
|
|
s->current_status = AUCODEC_PAUSE;
|
|||
|
|
else {
|
|||
|
|
s->current_status = AUCODEC_RUN;
|
|||
|
|
data = frame_buf->data;
|
|||
|
|
data_len = frame_buf->len;
|
|||
|
|
osal_fwrite(data, 2, data_len/2, s->wave_fp);
|
|||
|
|
s->data_size += data_len;
|
|||
|
|
}
|
|||
|
|
msi_delete_fb(s->msi, frame_buf);
|
|||
|
|
WAVE_DEBUG("wave encode delete framebuff:%p,len:%d,\r\n",frame_buf,frame_buf->len);
|
|||
|
|
frame_buf = NULL;
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
os_sleep_ms(1);
|
|||
|
|
|
|||
|
|
if(s->next_status == AUCODEC_EXIT) {
|
|||
|
|
s->current_status = AUCODEC_EXIT;
|
|||
|
|
s->wave_head.riff_chunk.ChunkSize = s->data_size + sizeof(TYPE_WAVE_HEAD) - 8;
|
|||
|
|
s->wave_head.fmt_chunk.SampleRate = s->samplerate;
|
|||
|
|
s->wave_head.fmt_chunk.ByteRate = s->samplerate*2;
|
|||
|
|
s->wave_head.data_chunk.DataSize = s->data_size;
|
|||
|
|
osal_fseek(s->wave_fp, 0);
|
|||
|
|
osal_fwrite(&(s->wave_head), 1, sizeof(TYPE_WAVE_HEAD), s->wave_fp);
|
|||
|
|
goto wave_encode_thread_end;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
wave_encode_thread_end:
|
|||
|
|
os_event_set(&s->event, coder_exit_event, NULL);
|
|||
|
|
|
|||
|
|
while((s->next_status != AUCODEC_EXIT) && (s->destroy_self == 0)) {
|
|||
|
|
s->current_status = AUCODEC_END;
|
|||
|
|
os_sleep_ms(5);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if(s->src_msi) {
|
|||
|
|
msi_del_output(s->src_msi, NULL, s->msi->name);
|
|||
|
|
s->src_msi = NULL;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if(s->next_status != AUCODEC_EXIT)
|
|||
|
|
msi_destroy(s->msi);
|
|||
|
|
|
|||
|
|
msi_put(s->msi);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
static int32_t wave_encode_msi_action(struct msi *msi, uint32_t cmd_id, uint32_t param1, uint32_t param2)
|
|||
|
|
{
|
|||
|
|
int32_t ret = RET_OK;
|
|||
|
|
struct wave_encode_struct *wave_encode_s = (struct wave_encode_struct*)(msi->priv);
|
|||
|
|
switch(cmd_id) {
|
|||
|
|
case MSI_CMD_AUCODER:
|
|||
|
|
{
|
|||
|
|
ret = RET_ERR;
|
|||
|
|
if(wave_encode_s) {
|
|||
|
|
uint32_t cmd_self = (uint32_t)param1;
|
|||
|
|
switch(cmd_self) {
|
|||
|
|
case MSI_AUCODER_PAUSE:
|
|||
|
|
{
|
|||
|
|
if(wave_encode_s->current_status == AUCODEC_RUN) {
|
|||
|
|
wave_encode_s->next_status = AUCODEC_PAUSE;
|
|||
|
|
}
|
|||
|
|
ret = RET_OK;
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
case MSI_AUCODER_CONTINUE:
|
|||
|
|
{
|
|||
|
|
if(wave_encode_s->current_status == AUCODEC_PAUSE) {
|
|||
|
|
wave_encode_s->next_status = AUCODEC_RUN;
|
|||
|
|
}
|
|||
|
|
ret = RET_OK;
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
case MSI_AUCODER_GET_STATUS:
|
|||
|
|
{
|
|||
|
|
*((uint32_t*)param2) = (uint32_t)(wave_encode_s->current_status);
|
|||
|
|
ret = RET_OK;
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
case MSI_AUCODER_SET_SRCMSI:
|
|||
|
|
{
|
|||
|
|
if(wave_encode_s->src_msi) {
|
|||
|
|
msi_del_output(wave_encode_s->src_msi, NULL, msi->name);
|
|||
|
|
}
|
|||
|
|
wave_encode_s->src_msi = NULL;
|
|||
|
|
ret = msi_add_output((struct msi*)param2, NULL, msi->name);
|
|||
|
|
if(ret == RET_OK) {
|
|||
|
|
wave_encode_s->src_msi = (struct msi*)param2;
|
|||
|
|
}
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
case MSI_AUCODER_DEINIT:
|
|||
|
|
{
|
|||
|
|
msi_destroy(msi);
|
|||
|
|
ret = RET_OK;
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
default:
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
case MSI_CMD_TRANS_FB:
|
|||
|
|
{
|
|||
|
|
ret = RET_ERR;
|
|||
|
|
struct framebuff *frame_buf = (struct framebuff *)param1;
|
|||
|
|
if(frame_buf->mtype == F_AUDIO) {
|
|||
|
|
ret = RET_OK;
|
|||
|
|
}
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
case MSI_CMD_PRE_DESTROY:
|
|||
|
|
{
|
|||
|
|
if(wave_encode_s && wave_encode_s->task_hdl) {
|
|||
|
|
wave_encode_s->next_status = AUCODEC_EXIT;
|
|||
|
|
}
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
case MSI_CMD_POST_DESTROY:
|
|||
|
|
{
|
|||
|
|
if(wave_encode_s) {
|
|||
|
|
if(wave_encode_s->task_hdl) {
|
|||
|
|
os_event_wait(&wave_encode_s->event, coder_exit_event, NULL, OS_EVENT_WMODE_OR | OS_EVENT_WMODE_CLEAR, osWaitForever);
|
|||
|
|
}
|
|||
|
|
if(wave_encode_s->event.hdl) {
|
|||
|
|
os_event_del(&wave_encode_s->event);
|
|||
|
|
}
|
|||
|
|
if(wave_encode_s->wave_fp) {
|
|||
|
|
osal_fclose(wave_encode_s->wave_fp);
|
|||
|
|
wave_encode_s->wave_fp = NULL;
|
|||
|
|
}
|
|||
|
|
WAVE_CODE_FREE(wave_encode_s);
|
|||
|
|
wave_encode_s = NULL;
|
|||
|
|
}
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
default:
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
return ret;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
struct msi *wave_encode_init(char *filename, uint32_t samplerate, AUENC_INIT *auenc_init)
|
|||
|
|
{
|
|||
|
|
#if AUDIO_EN
|
|||
|
|
uint8_t msi_isnew = 0;
|
|||
|
|
|
|||
|
|
struct msi *msi = msi_new("R_WAVE_ENCODE", MAX_WAVE_ENCODE_RXBUF, &msi_isnew);
|
|||
|
|
if(!msi) {
|
|||
|
|
WAVE_INFO("create wave encode msi fail!\r\n");
|
|||
|
|
return NULL;
|
|||
|
|
}
|
|||
|
|
else if(!msi_isnew) {
|
|||
|
|
WAVE_INFO("wave encode msi has been create!\r\n");
|
|||
|
|
goto wave_encode_init_err;
|
|||
|
|
}
|
|||
|
|
struct wave_encode_struct *wave_encode_s = (struct wave_encode_struct*)WAVE_CODE_ZALLOC(sizeof(struct wave_encode_struct));
|
|||
|
|
if(!wave_encode_s) {
|
|||
|
|
WAVE_INFO("wave_encode_s malloc fail!\r\n");
|
|||
|
|
goto wave_encode_init_err;
|
|||
|
|
}
|
|||
|
|
msi->priv = wave_encode_s;
|
|||
|
|
msi->action = (msi_action)wave_encode_msi_action;
|
|||
|
|
if(os_event_init(&wave_encode_s->event) != RET_OK) {
|
|||
|
|
WAVE_INFO("create wave encode event fail!\r\n");
|
|||
|
|
goto wave_encode_init_err;
|
|||
|
|
}
|
|||
|
|
if(filename) {
|
|||
|
|
wave_encode_s->wave_fp = osal_fopen((const char*)filename, "wb+");
|
|||
|
|
if(wave_encode_s->wave_fp == NULL) {
|
|||
|
|
WAVE_INFO("open wave record file %s fail!\r\n", filename);
|
|||
|
|
goto wave_encode_init_err;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else {
|
|||
|
|
WAVE_INFO("wave record filename is null!\r\n");
|
|||
|
|
goto wave_encode_init_err;
|
|||
|
|
}
|
|||
|
|
os_memcpy(&(wave_encode_s->wave_head), wav_header, sizeof(TYPE_WAVE_HEAD));
|
|||
|
|
osal_fseek(wave_encode_s->wave_fp, sizeof(TYPE_WAVE_HEAD));
|
|||
|
|
if(auenc_init->src_msi && (msi_add_output(auenc_init->src_msi, NULL, msi->name) != RET_OK)) {
|
|||
|
|
goto wave_encode_init_err;
|
|||
|
|
}
|
|||
|
|
wave_encode_s->msi = msi;
|
|||
|
|
wave_encode_s->src_msi = auenc_init->src_msi;
|
|||
|
|
wave_encode_s->samplerate = samplerate;
|
|||
|
|
wave_encode_s->destroy_self = auenc_init->destroy_self;
|
|||
|
|
wave_encode_s->next_status = AUCODEC_RUN;
|
|||
|
|
wave_encode_s->current_status = AUCODEC_RUN;
|
|||
|
|
wave_encode_s->task_hdl = os_task_create("wave_encode_thread", wave_encode_thread, (void*)wave_encode_s, OS_TASK_PRIORITY_NORMAL, 0, NULL, 1024);
|
|||
|
|
if(wave_encode_s->task_hdl == NULL) {
|
|||
|
|
WAVE_INFO("create wave encode task fail!\r\n");
|
|||
|
|
goto wave_encode_init_err;
|
|||
|
|
}
|
|||
|
|
return msi;
|
|||
|
|
|
|||
|
|
wave_encode_init_err:
|
|||
|
|
msi_destroy(msi);
|
|||
|
|
#endif
|
|||
|
|
return NULL;
|
|||
|
|
}
|