C++ 读取wav文件中的PCM数据

24 篇文章 3 订阅

前言

wav文件通常会使用PCM格式数据存储音频,这种格式的数据读取出来直接就可以播放,要在wav文件中读取数据,我们首先要获取头部信息,wav的文件结构里面分为多个chunk,我们要做的就是识别这些chunk的信息,获取音频的格式以及数据。


一、如何实现?

首先需要构造wav头部,wav文件音频信息全部保存在头部,我们要做的就是读取wav头部信息,并且记录PCM的相关参数。

1.定义头结构

只定义PCM格式的wav文件头,对于PCM格式的数据只需要下面3个结构体即可。

struct WaveRIFF;
struct WaveFormat;
struct WaveData;

2.读取头部信息

打开文件后需要读取头部信息,需要获取声音的格式以及数据长度。

WaveRIFF riff;
WaveFormat format;
WaveData data;
int userDataSize;
f= fopen(fileName.c_str(), "rb+");
//读取头部信息
fread(&riff, 1, sizeof(riff), f);
fread(&format, 1, sizeof(format),f);

3.读取数据

获取头部信息后,就知道数据在位置及长度了,只需要直接读文件即可。

//跳到数据起始位置
seek(f, _dataOffset, SEEK_SET);
//读取数据
fread(buf, 1, bufLength, f);

二、完整代码

完整代码总用有3部分,头结构、WavFileReader.h、WavFileReader.cpp。

1.头结构

#pragma pack(push,1)
	struct WaveRIFF {
		const	char id[4] = { 'R','I', 'F', 'F' };
		uint32_t fileLength;
		const	char waveFlag[4] = { 'W','A', 'V', 'E' };
	};
	struct WaveFormat
	{
		const	char id[4] = { 'f','m', 't', ' ' };
		uint32_t blockSize = 16;
		uint16_t formatTag;
		uint16_t channels;
		uint32_t samplesPerSec;
		uint32_t avgBytesPerSec;
		uint16_t blockAlign;
		uint16_t  bitsPerSample;
	};
	struct  WaveData
	{
		const	char id[4] = { 'd','a', 't', 'a' };
		uint32_t dataLength;
	};
#pragma pack(pop)

2.WavFileReader.h

#pragma once
#include<string>
/************************************************************************
* @Project:  	AC.WavFileWriter
* @Decription:  wav文件读取工具
* 本版本只支持pcm读取,且未处理字节顺序。	riff文件是小端,通常在intel的设备上是没问题的,在java虚拟机上则需要处理。
* @Verision:  	v1.0.0.0
* @Author:  	Xin Nie
* @Create:  	2019/4/10 11:10:17
* @LastUpdate:  2019/4/16 10:45:00
************************************************************************
* Copyright @ 2019. All rights reserved.
************************************************************************/
namespace AC {
	/// <summary>
	/// wav文件读取对象
	/// </summary>
	class WavFileReader {
	public:
		/// <summary>
		/// 构造方法
		/// </summary>
		WavFileReader();
		/// <summary>
	    /// 析构方法
	    /// </summary>
		~WavFileReader();
		/// <summary>
		/// 打开wav文件
		/// </summary>
		/// <param name="fileName">文件名</param>
		/// <returns>是否打开成功</returns>
		bool OpenWavFile(const std::string& fileName);
		/// <summary>
		/// 关闭文件
		/// </summary>
		void CloseFlie();
		/// <summary>
		/// 读取音频数据
		/// </summary>
		/// <param name="buf">外部缓存</param>
		/// <param name="bufLength">缓存长度</param>
		/// <returns>读取长度</returns>
		int ReadData(unsigned char* buf, int bufLength);
		/// <summary>
		/// 设置读取位置
		/// </summary>
		/// <param name="position"> 读取位置</param>
		void SetPosition(int position);
		/// <summary>
		/// 获取读取位置
		/// </summary>
		/// <returns>读取位置</returns>
		int GetPosition();
		/// <summary>
		/// 获取文件长度
		/// </summary>
		/// <returns>文件长度</returns>
		int GetFileLength();
		/// <summary>
		/// 获取音频数据长度
		/// </summary>
		/// <returns>音频数据长度</returns>
		int GetDataLength();
		/// <summary>
		/// 获取声道数
		/// </summary>
		/// <returns>声道数</returns>
		int GetChannels();
		/// <summary>
		/// 获取采样率
		/// </summary>
		/// <returns>采样率,单位:hz</returns>
		int GetSampleRate();
		/// <summary>
		/// 获取位深
		/// </summary>
		/// <returns>位深,单位:bits</returns>
		int GetBitsPerSample();
	private:
		void* _file = nullptr;
		uint32_t _fileLength = 0;
		uint32_t _dataLength = 0;
		int _channels = 0;
		int  _sampleRate = 0;
		int  _bitsPerSample = 0;
		int _dataOffset = 0;
	};
}

3.WavFileReader.cpp

#include"WavFileReader.h"
namespace AC {
	WavFileReader::WavFileReader()
	{
	}
	WavFileReader::~WavFileReader()
	{
		CloseFlie();
	}
	bool WavFileReader::OpenWavFile(const std::string& fileName)
	{
		if (_file)
		{
			printf("已经打开了文件!\n");
			return false;
		}
		WaveRIFF riff;
		WaveFormat format;
		WaveData data;
		int userDataSize;
		_file = fopen(fileName.c_str(), "rb+");
		if (!_file)
		{
			printf("打开文件失败!\n");
			return false;
		}
		//读取头部信息
		if (fread(&riff, 1, sizeof(riff), static_cast<FILE*>(_file)) != sizeof(riff))
		{
			printf("文件读取错误,读取riff失败!\n");
			goto error;
		}
		if (std::string(riff.id, 4) != "RIFF" || std::string(riff.waveFlag, 4) != "WAVE")
		{
			printf("头部信息不正确,不是wav文件!\n");
			goto error;
		}
		if (fread(&format, 1, sizeof(format), static_cast<FILE*>(_file)) != sizeof(format))
		{
			printf("文件读取错误,读取format失败!\n");
			goto error;
		}
		if (std::string(format.id, 4) != "fmt ")
		{
			printf("头部信息不正确,缺少fmt!\n");
			goto error;
		}
		if (format.formatTag != 1)
		{
			printf("程序不支持,数据格式非pcm,只支持pcm格式的数据!\n");
			goto error;
		}
		userDataSize = format.blockSize - sizeof(format) + 8;
		if (userDataSize < 0)
		{
			printf("头部信息不正确,blockSize大小异常!\n");
			goto error;
		}
		else if (userDataSize > 0)
		{
			if (fseek(static_cast<FILE*>(_file), userDataSize, SEEK_CUR) != 0)
			{
				printf("文件读取错误!\n");
				goto error;
			}
		}
		while (1)
		{
			if (fread(&data, 1, sizeof(data), static_cast<FILE*>(_file)) != sizeof(data))
			{
				printf("文件读取错误!\n");
				goto error;
			};
			if (std::string(data.id, 4) != "data")
			{
				if (fseek(static_cast<FILE*>(_file), data.dataLength, SEEK_CUR) != 0)
				{
					printf("文件读取错误!\n");
					goto error;
				}
				continue;
			}
			break;
		}
		_dataOffset = ftell(static_cast<FILE*>(_file));
		_fileLength = riff.fileLength+8;
		_dataLength = data.dataLength;
		_channels = format.channels;
		_sampleRate = format.samplesPerSec;
		_bitsPerSample = format.bitsPerSample;
		return true;
	error:
		if (fclose(static_cast<FILE*>(_file)) == EOF)
		{
			printf("文件关闭失败!\n");
		}
		_file = nullptr;
		return false;
	}
	void WavFileReader::CloseFlie()
	{
		if (_file)
		{
			if (fclose(static_cast<FILE*>(_file)) == EOF)
			{
				printf("文件关闭失败!\n");
			}
			_file = nullptr;
		}
	}
	int WavFileReader::ReadData(unsigned char* buf, int bufLength)
	{
		if (ftell(static_cast<FILE*>(_file)) >= _dataOffset + _dataLength)
			return 0;
		return fread(buf, 1, bufLength, static_cast<FILE*>(_file));
	}

	void WavFileReader::SetPosition(int postion)
	{
		if (fseek(static_cast<FILE*>(_file), _dataOffset + postion, SEEK_SET) != 0)
		{
			printf("定位失败!\n");
		}
	}
	int WavFileReader::GetPosition()
	{
		return ftell(static_cast<FILE*>(_file)) - _dataOffset;
	}

	int WavFileReader::GetFileLength()
	{
		return _fileLength;
	}

	int WavFileReader::GetDataLength()
	{
		return _dataLength;
	}

	int WavFileReader::GetChannels()
	{
		return _channels;
	}

	int WavFileReader::GetSampleRate()
	{
		return _sampleRate;
	}

	int WavFileReader::GetBitsPerSample()
	{
		return _bitsPerSample;
	}
}

三、使用示例

1、播放

#include "WavFileReader.h"
int main(int argc, char** argv) {
	AC::WavFileReader read;
	unsigned char buf[1024];
	if (read.OpenWavFile("test_music.wav"))
	{
		int channels, sampleRate, bitsPerSample;
		//获取声音格式
		channels = read.GetChannels();
		sampleRate = read.GetSampleRate();
		bitsPerSample = read.GetBitsPerSample();
		//打开声音设备(channels,sampleRate,bitsPerSample)
		int size;
		do 
		{
		    //读取音频数据
			size = read.ReadData(buf,1024);
			if (size > 0)
			{
				//播放(buf,1024)
			}
			
		} while (size);
		read.CloseFlie();
	}
	return 0;
}

2、循环播放

#include "WavFileReader.h"
int main(int argc, char** argv) {
	AC::WavFileReader read;
	unsigned char buf[1024];
    bool exitFlag = false;
	if (read.OpenWavFile("test_music.wav"))
	{
		int channels, sampleRate, bitsPerSample;
		//获取声音格式
		channels = read.GetChannels();
		sampleRate = read.GetSampleRate();
		bitsPerSample = read.GetBitsPerSample();
		//打开声音设备(channels,sampleRate,bitsPerSample)
		int size;
		while (!exitFlag)
		{
		    //读取音频数据
			size = read.ReadData(buf, 1024);
			if (size > 0)
			{
				//播放(buf,1024)
			}
			else
			{   
			    //回到数据起始位置
				read.SetPosition(0);		
			}
		} 
		read.CloseFlie();
	}
	return 0;
}

总结

以上就是今天要讲的内容,wav文件中读取PCM还是相对较简单的,只要了解wav头结构,然后自定义其头结构,读取头部信息,校验头部信息,然后再读取数据所在的chunk,就可以实现这样一个功能。

  • 7
    点赞
  • 25
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论
下面是一个简单的C++代码示例,用于将PCM二进制数据转换为WAV文件: ```c++ #include <fstream> #include <iostream> #include <cstring> #include <stdint.h> using namespace std; int main(int argc, char* argv[]) { // 检查输入参数 if (argc != 4) { cout << "Usage: " << argv[0] << " <input file> <output file> <sample rate>" << endl; return 1; } // 打开输入文件 ifstream input(argv[1], ios::binary); if (!input) { cout << "Failed to open input file: " << argv[1] << endl; return 1; } // 创建输出文件 ofstream output(argv[2], ios::binary); if (!output) { cout << "Failed to create output file: " << argv[2] << endl; return 1; } // 设置采样率 int sampleRate = atoi(argv[3]); int bitsPerSample = 16; int channels = 1; // 写入WAV文件头 char header[44]; memset(header, 0, sizeof(header)); memcpy(header, "RIFF", 4); uint32_t fileSize = 0; memcpy(header + 4, &fileSize, 4); memcpy(header + 8, "WAVE", 4); memcpy(header + 12, "fmt ", 4); uint32_t fmtSize = 16; memcpy(header + 16, &fmtSize, 4); uint16_t audioFormat = 1; memcpy(header + 20, &audioFormat, 2); uint16_t numChannels = channels; memcpy(header + 22, &numChannels, 2); uint32_t sampleRateHz = sampleRate; memcpy(header + 24, &sampleRateHz, 4); uint32_t byteRate = sampleRateHz * numChannels * bitsPerSample / 8; memcpy(header + 28, &byteRate, 4); uint16_t blockAlign = numChannels * bitsPerSample / 8; memcpy(header + 32, &blockAlign, 2); memcpy(header + 34, &bitsPerSample, 2); memcpy(header + 36, "data", 4); uint32_t dataSize = 0; memcpy(header + 40, &dataSize, 4); output.write(header, sizeof(header)); // 逐个读取PCM样本并将其写入WAV文件 char buffer[4096]; while (input) { int bytes = input.read(buffer, sizeof(buffer)).gcount(); if (bytes > 0) { output.write(buffer, bytes); fileSize += bytes; } } // 更新WAV文件头 dataSize = fileSize - sizeof(header); memcpy(header + 4, &fileSize, 4); memcpy(header + 40, &dataSize, 4); output.seekp(0); output.write(header, sizeof(header)); // 关闭文件并退出 input.close(); output.close(); cout << "Conversion completed" << endl; return 0; } ``` 在上面的代码,我们首先检查输入参数,然后打开输入文件。接下来,我们创建输出文件,并设置WAV文件的采样率、位深度和通道数。我们使用这些参数填写WAV文件头,并将其写入输出文件。然后,我们逐个读取输入文件PCM样本,并将它们写入输出文件。最后,我们更新WAV文件数据大小字段,并关闭文件。 请注意,上述代码只能用于单通道、16位深度的PCM数据,如果您需要处理其他类型的PCM数据,需要相应地修改代码。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

CodeOfCC

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值