#include "aud_media.h"


/*创建音频文件*/
int pcmfile_create(struct t_pcm_file* pp_pcm_file, E_PCM_FORMAT vp_format)
{

	int ret = (pp_pcm_file->m_filename) ? 0 : -1;
	FILE* fp = fopen(pp_pcm_file->m_filename, "wb+");
	pp_pcm_file->m_fp = fp;
	if (ret < 0 || !fp)    return -1;

	switch (vp_format)
	{
	case AMR_FMT:
		ret = amr_header_set(fp);   //添加amr文件头
		break;
	case PCM_FMT:
		ret = wav_header_set(fp);   //添加wav文件头
		break;
	default:
		fclose(fp); ret = -2;
		break;
	}
	return ret;
}

/*设置wav音频文件头*/
int wav_header_set(FILE*  pp_fp)
{
	t_wav_header hdr = { 0 };
	memset(&hdr, 0, sizeof(t_wav_header));
	hdr.m_riff_id = ID_RIFF;
	hdr.m_riff_fmt = ID_WAVE;
	hdr.m_fmt_id = ID_FMT;
	hdr.m_fmt_sz = 16;
	hdr.m_audio_format = FORMAT_PCM;
	hdr.m_num_channels = 1;
	hdr.m_sample_rate = 8000;
	hdr.m_bits_per_sample = 16;
	hdr.m_byte_rate = (8000 * 1 * hdr.m_bits_per_sample) / 8;
	hdr.m_block_align = (hdr.m_bits_per_sample * 1) / 8;
	hdr.m_data_id = ID_DATA;
	hdr.m_data_sz = 0;
	hdr.m_riff_sz = hdr.m_data_sz + 44 - 8;
	return (fwrite(&hdr, 1, sizeof(hdr), pp_fp) > 0) ? 0 : -1;
}


/*设置amr音频文件头*/
int amr_header_set(FILE*  pp_fp)
{
	t_amr_header hdr = { 0 };
	hdr.m_amr_hdr[0] = 0x23;
	hdr.m_amr_hdr[1] = 0x21;
	hdr.m_amr_hdr[2] = 0x41;
	hdr.m_amr_hdr[3] = 0x4d;
	hdr.m_amr_hdr[4] = 0x52;
	hdr.m_amr_hdr[5] = 0x0A;
	return (fwrite(&hdr, 1, 6, pp_fp) > 0) ? 0 : -1;
}