本过滤器将视频流和音频流写入AVI视频文件。
过滤器信息
过滤器名称:写AVI
过滤器GUID:{2EF49957-37DF-4356-A2A0-ECBC52D1984B}
DLL注册函数名:DllRegisterServer
删除注册函数名:DllUnregisterServer
过滤器有2个输入引脚。
输入引脚1标识:Video
输入引脚1媒体类型:
主要类型:MEDIATYPE_Video
子类型:MEDIASUBTYPE_NULL
格式类型:FORMAT_VideoInfo
输入引脚2标识:Audio
输入引脚2媒体类型:
主要类型:MEDIATYPE_Audio
子类型:MEDIASUBTYPE_MP3,MEDIASUBTYPE_PCM,MEDIASUBTYPE_ALAW
格式类型:FORMAT_WaveFormatEx
过滤器开发信息
本文介绍的方法完全从AVI文件结构入手,写RIFF文件头(12字节),创建hdrl_LIST,包括avih和1个视频strl_LIST,1个音频strl_LIST;在视频音频strl_LIST中各创建1个indx索引,indx索引中每个条目指向一个索引块,视频索引块条目指向1个视频帧的存储位置,音频索引块条目指向1个音频数据块的存储位置。没有创建idx1索引。视频音频数据都存储在RIFF_movi_LIST和AVIX_movi_LIST中,每写入一次,创建一个索引块条目,条目报告数据的位置和大小。包括索引块也存储在RIFF_movi_LIST和AVIX_movi_LIST中;每创建一个索引块,就增加一个indx索引条目,条目报告索引块的位置,大小和音频或视频的长度;视频长度就是帧数量,PCM音频长度是音频帧的数量,MP3长度直接使用数据的字节大小。
AVI文件没有限制视频音频的编码方式。本过滤器没有限制视频编码方式,但音频编码方式限制为MP3,PCM,ALAW。
下面是使用本过滤器创建的1个AVI文件的结构图片:
过滤器DLL的全部代码
DLL.h
cpp
#ifndef DLL_FILE
#define DLL_FILE
#include "strmbase10.h"//过滤器基础类定义文件
#if _DEBUG
#pragma comment(lib, "strmbasd10.lib")//过滤器基础类实现文件调试版本
#else
#pragma comment(lib, "strmbase10.lib")//过滤器基础类实现文件发布版本
#endif
// {2EF49957-37DF-4356-A2A0-ECBC52D1984B}
DEFINE_GUID(CLSID_AviWriter,//过滤器GUID
0x2ef49957, 0x37df, 0x4356, 0xa2, 0xa0, 0xec, 0xbc, 0x52, 0xd1, 0x98, 0x4b);
DEFINE_GUID(MEDIASUBTYPE_ALAW,//
0x00000006, 0x0000, 0x0010, 0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71);
DEFINE_GUID(MEDIASUBTYPE_MP3,
0x00000055, 0x0000, 0x0010, 0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71);
#include "strsafe.h"
#include "aviriff.h"
class AVI_HEADER
{
public:
BYTE p[4642];
char* RIFF_ch;//文件规范标识
DWORD* FileSize;//文件大小
char* AVI_ch;//文件类型
char* hdrl_LIST_ch;//hdrl_LIST
DWORD* hdrl_LIST_Size;//
char* hdrl_ch;
char* fcc; //avih块
DWORD* cb;//avih块大小
DWORD* dwMicroSecPerFrame;//帧持续时间,单位微秒
DWORD* dwMaxBytesPerSec;//传输率
DWORD* dwPaddingGranularity;//块对齐
DWORD* dwFlags;//文件属性
DWORD* dwTotalFrames;//帧总数
DWORD* dwInitialFrames;//初始帧
DWORD* dwStreams;//流数量
DWORD* dwSuggestedBufferSize;//建议的缓冲区的大小
DWORD* dwWidth;//图像的宽度
DWORD* dwHeight;//图像的高度
BYTE* dwReserved;//保留
char* strl_LIST;//视频strl_LIST
DWORD* strl_LIST_Size;//
char* strl_ch;//strl
char* strh_ch;//strh
DWORD* strh_Size;//
char* fccType;//
char* fccHandler;//首选处理器
DWORD* strhFlags;//标志
WORD* wPriority;//流的优先级
WORD* wLanguage;//语言
DWORD* strh_dwInitialFrames;//初始帧
DWORD* dwScale;//采样率分母
DWORD* dwRate;//采样率分子
DWORD* dwStart;//流的开始时间
DWORD* dwLength;//流的长度,以视频帧为单位
DWORD* strh_dwSuggestedBufferSize;//建议的缓冲区大小
DWORD* dwQuality;//质量参数
DWORD* dwSampleSize;//单个数据样本的大小
struct Frame {
short int left;
short int top;
short int right;
short int bottom;
};
Frame* rcFrame;//目标矩形
char* strf_ch;//strf
DWORD* strf_Size;//strf块大小
DWORD* biSize;//位图头大小
LONG* biWidth;//位图的宽度
LONG* biHeight;//位图的高度
WORD* biPlanes;//平面数
WORD* biBitCount;//每个像素的位数
DWORD* biCompression;//编码方式
DWORD* biSizeImage;//图像大小
LONG* biXPelsPerMeter;//目标设备的水平分辨率
LONG* biYPelsPerMeter;//目标设备的垂直分辨率
DWORD* biClrUsed;//颜色数量
DWORD* biClrImportant;//重要颜色数量
char* ApendID;//
DWORD* ApendSize;//
BYTE* ApendData;//
char* indx_ch; //视频indx块
DWORD* index_Size;
WORD* wLongsPerEntry;//每个条目所占DWORD个数
BYTE* bIndexSubType;//索引子类型,0指向索引块,1指向数据块,8条目是真实数据
BYTE* bIndexType;//索引类型,0标准索引,1场索引
DWORD* nEntriesInUse;//条目数量
char* dwChunkId;//流的FOURCC
DWORD* Reserved1, *Reserved2, *Reserved3;//保留
struct ENTRY {
LONGLONG Offset;//偏移量
DWORD Size;//大小
DWORD Dur;//长度
};
ENTRY* Entry;//indx条目数组
char* strl_LIST_A;//音频strl_LIST
DWORD* strl_LIST_A_Size;//
char* strl_A_ch;//strl
char* strh_A_ch;//strh
DWORD* strh_A_Size;//
char* fccType_A;//
char* fccHandler_A;//首选处理器
DWORD* strhFlags_A;//标志
WORD* wPriority_A;//流的优先级
WORD* wLanguage_A;//语言
DWORD* strh_A_dwInitialFrames;//初始帧
DWORD* dwScale_A;//采样率分母
DWORD* dwRate_A;//采样率分子
DWORD* dwStart_A;//流的开始时间
DWORD* dwLength_A;//流的长度,以音频帧为单位
DWORD* strh_A_dwSuggestedBufferSize;//建议的缓冲区大小
DWORD* dwQuality_A;//质量参数
DWORD* dwSampleSize_A;//单个数据样本的大小
Frame* rcFrame_A;//目标矩形
char* strf_A_ch;//strf
DWORD* strf_A_Size;//strf块大小
WORD* wFormatTag_A;//音频格式
WORD* nChannels_A;//声道数
DWORD* nSamplesPerSec_A;//采样率
DWORD* nAvgBytesPerSec_A;//传输率
WORD* nBlockAlign_A;//块对齐
WORD* wBitsPerSample_A;//样本位数
WORD* cbSize_A;//附加信息大小
BYTE* pApend;//大小64
char* indx_ch_A; //音频indx块
DWORD* index_Size_A;
WORD* wLongsPerEntry_A;//每个条目所占DWORD个数
BYTE* bIndexSubType_A;//索引子类型,0指向索引块,1指向数据块,8条目是真实数据
BYTE* bIndexType_A;//索引类型,0标准索引,1场索引
DWORD* nEntriesInUse_A;//条目数量
char* dwChunkId_A;//流的FOURCC
DWORD* Reserved_A1, *Reserved_A2, *Reserved_A3;//保留
ENTRY* EntryA;//indx条目数组
AVI_HEADER()
{
RIFF_ch = (char*)&p[0]; CopyMemory(RIFF_ch, "RIFF", 4);//文件规范标识
FileSize = (DWORD*)&p[4]; *FileSize = 0;//文件大小
AVI_ch = (char*)&p[8]; CopyMemory(AVI_ch, "AVI ", 4);//文件类型
hdrl_LIST_ch = (char*)&p[12]; CopyMemory(hdrl_LIST_ch, "LIST", 4);//hdrl_LIST
hdrl_LIST_Size = (DWORD*)&p[16]; *hdrl_LIST_Size = 4622;//
hdrl_ch = (char*)&p[20]; CopyMemory(hdrl_ch, "hdrl", 4);
fcc = (char*)&p[24]; CopyMemory(fcc, "avih", 4); //avih块
cb = (DWORD*)&p[28]; *cb = 56;//avih块大小
dwMicroSecPerFrame = (DWORD*)&p[32];//帧持续时间,单位微秒
dwMaxBytesPerSec = (DWORD*)&p[36]; *dwMaxBytesPerSec = 0; //传输率
dwPaddingGranularity = (DWORD*)&p[40]; *dwPaddingGranularity = 1;//块对齐
dwFlags = (DWORD*)&p[44]; *dwFlags = AVIF_ISINTERLEAVED | AVIF_TRUSTCKTYPE;//文件属性
/*AVIF_ISINTERLEAVED 表示AVI文件已交错
AVIF_TRUSTCKTYPE 使用CKType确定关键帧*/
dwTotalFrames = (DWORD*)&p[48];//帧总数
dwInitialFrames = (DWORD*)&p[52]; *dwInitialFrames = 0;//初始帧
dwStreams = (DWORD*)&p[56]; *dwStreams = 2;//流数量
dwSuggestedBufferSize = (DWORD*)&p[60]; *dwSuggestedBufferSize = 0;//建议的缓冲区的大小
dwWidth = (DWORD*)&p[64];//图像的宽度
dwHeight = (DWORD*)&p[68];//图像的高度
dwReserved = &p[72]; memset(dwReserved, 0, 16);//保留
strl_LIST = (char*)&p[88]; CopyMemory(strl_LIST, "LIST", 4);//视频strl_LIST
strl_LIST_Size = (DWORD*)&p[92]; *strl_LIST_Size = 2300;//
strl_ch = (char*)&p[96]; CopyMemory(strl_ch, "strl", 4);//strl
strh_ch = (char*)&p[100]; CopyMemory(strh_ch, "strh", 4);//strh
strh_Size = (DWORD*)&p[104]; *strh_Size = 56;//
fccType = (char*)&p[108]; CopyMemory(fccType, "vids", 4);
fccHandler = (char*)&p[112];//首选处理器
strhFlags = (DWORD*)&p[116]; *strhFlags = 0;//标志
wPriority = (WORD*)&p[120]; *wPriority = 0;//流的优先级
wLanguage = (WORD*)&p[122]; *wLanguage = 0;//语言
strh_dwInitialFrames = (DWORD*)&p[124]; *strh_dwInitialFrames = 0;//初始帧
dwScale = (DWORD*)&p[128]; //采样率分母
dwRate = (DWORD*)&p[132]; *dwRate = 10000000;//采样率分子
dwStart = (DWORD*)&p[136]; *dwStart = 0;//流的开始时间
dwLength = (DWORD*)&p[140];//流的长度,以视频帧为单位
strh_dwSuggestedBufferSize = (DWORD*)&p[144];//建议的缓冲区大小
dwQuality = (DWORD*)&p[148]; *dwQuality = 0;//质量参数
dwSampleSize = (DWORD*)&p[152]; *dwSampleSize = 0;//单个数据样本的大小
rcFrame = (Frame*)&p[156];//目标矩形
strf_ch = (char*)&p[164]; CopyMemory(strf_ch, "strf", 4);//strf
strf_Size = (DWORD*)&p[168]; *strf_Size = 40;//strf块大小
biSize = (DWORD*)&p[172]; *biSize = 40;//位图头大小
biWidth = (LONG*)&p[176];//位图的宽度
biHeight = (LONG*)&p[180];//位图的高度
biPlanes = (WORD*)&p[184]; *biPlanes = 1;//平面数
biBitCount = (WORD*)&p[186];//每个像素的位数
biCompression = (DWORD*)&p[188];//编码方式
biSizeImage = (DWORD*)&p[192];//图像大小
biXPelsPerMeter = (LONG*)&p[196];//目标设备的水平分辨率
biYPelsPerMeter = (LONG*)&p[200];//目标设备的垂直分辨率
biClrUsed = (DWORD*)&p[204];//颜色数量
biClrImportant = (DWORD*)&p[208];//重要颜色数量
ApendID = (char*)&p[212]; CopyMemory(ApendID, "JUNK", 4);//附加信息总大小104
ApendSize = (DWORD*)&p[216]; *ApendSize = 96;
ApendData = &p[220];
indx_ch = (char*)&p[316]; CopyMemory(indx_ch, "indx", 4);//视频indx块
index_Size = (DWORD*)&p[320]; *index_Size = 2072;
wLongsPerEntry = (WORD*)&p[324]; *wLongsPerEntry = 4;//每个条目所占DWORD个数
bIndexSubType = &p[326]; *bIndexSubType = 0;//索引子类型,0指向索引块,1指向数据块,8条目是真实数据
bIndexType = &p[327]; *bIndexType = 0;//索引类型,0标准索引,1场索引
nEntriesInUse = (DWORD*)&p[328]; *nEntriesInUse = 0;//条目数量
dwChunkId = (char*)&p[332]; CopyMemory(dwChunkId, "00db", 4);//流的FOURCC
Reserved1 = (DWORD*)&p[336]; *Reserved1 = 0;//保留
Reserved2 = (DWORD*)&p[340]; *Reserved2 = 0;
Reserved3 = (DWORD*)&p[344]; *Reserved3 = 0;
Entry = (ENTRY*)&p[348];
strl_LIST_A = (char*)&p[2396]; CopyMemory(strl_LIST_A, "LIST", 4);//音频strl_LIST
strl_LIST_A_Size = (DWORD*)&p[2400]; *strl_LIST_A_Size = 2238;//
strl_A_ch = (char*)&p[2404]; CopyMemory(strl_A_ch, "strl", 4);//strl
strh_A_ch = (char*)&p[2408]; CopyMemory(strh_A_ch, "strh", 4);//strh
strh_A_Size = (DWORD*)&p[2412]; *strh_A_Size = 56;//
fccType_A = (char*)&p[2416]; CopyMemory(fccType_A, "auds", 4);
fccHandler_A = (char*)&p[2420]; //首选处理器
strhFlags_A = (DWORD*)&p[2424]; *strhFlags_A = 0;//标志
wPriority_A = (WORD*)&p[2428]; *wPriority_A = 0;//流的优先级
wLanguage_A = (WORD*)&p[2430]; *wLanguage_A = 0;//语言
strh_A_dwInitialFrames = (DWORD*)&p[2432]; *strh_A_dwInitialFrames = 0;//初始帧
dwScale_A = (DWORD*)&p[2436]; *dwScale_A = 1;//采样率分母
dwRate_A = (DWORD*)&p[2440]; *dwRate_A;//采样率分子
dwStart_A = (DWORD*)&p[2444]; *dwStart_A = 0;//流的开始时间
dwLength_A = (DWORD*)&p[2448];//流的长度,以音频帧为单位
strh_A_dwSuggestedBufferSize = (DWORD*)&p[2452]; //建议的缓冲区大小
dwQuality_A = (DWORD*)&p[2456]; *dwQuality_A = 0;//质量参数
dwSampleSize_A = (DWORD*)&p[2460];//单个数据样本的大小
rcFrame_A = (Frame*)&p[2464];//目标矩形
rcFrame_A->left = 0; rcFrame_A->top = 0; rcFrame_A->right = 0; rcFrame_A->bottom = 0;
strf_A_ch = (char*)&p[2472]; CopyMemory(strf_A_ch, "strf", 4);//strf
strf_A_Size = (DWORD*)&p[2476]; *strf_A_Size = 18;//strf块大小
wFormatTag_A = (WORD*)&p[2480];//音频格式
nChannels_A = (WORD*)&p[2482];//声道数
nSamplesPerSec_A = (DWORD*)&p[2484];//采样率
nAvgBytesPerSec_A = (DWORD*)&p[2488];//传输率
nBlockAlign_A = (WORD*)&p[2492];//块对齐
wBitsPerSample_A = (WORD*)&p[2494];//样本位数
cbSize_A = (WORD*)&p[2496]; *cbSize_A = 0;//附加信息大小
pApend = &p[2498]; CopyMemory(pApend, "JUNK", 4);
DWORD* ApendSize_A = (DWORD*)&p[2502]; *ApendSize_A = 56;
indx_ch_A = (char*)&p[2562]; CopyMemory(indx_ch_A, "indx", 4); //音频indx块
index_Size_A = (DWORD*)&p[2566]; *index_Size_A = 2072;
wLongsPerEntry_A = (WORD*)&p[2570]; *wLongsPerEntry_A = 4;//每个条目所占DWORD个数
bIndexSubType_A = &p[2572]; *bIndexSubType_A = 0;//索引子类型,0指向索引块,1指向数据块,8条目是真实数据
bIndexType_A = &p[2573]; *bIndexType_A = 0;//索引类型,0标准索引,1场索引
nEntriesInUse_A = (DWORD*)&p[2574]; *nEntriesInUse_A = 0;//条目数量
dwChunkId_A = (char*)&p[2578]; CopyMemory(dwChunkId_A, "01wb", 4);//流的FOURCC
Reserved_A1 = (DWORD*)&p[2582]; *Reserved_A1 = 0;//保留
Reserved_A2 = (DWORD*)&p[2586]; *Reserved_A2 = 0;
Reserved_A3 = (DWORD*)&p[2590]; *Reserved_A3 = 0;
EntryA = (ENTRY*)&p[2594];
//以上总大小4642字节
}
};
class CFilter;
class CPin2;
class CPin1 : public CBaseInputPin
{
friend class CFilter;
friend class CPin2;
public:
CPin1(CFilter *pFilter, HRESULT *phr, LPCWSTR pPinName);
~CPin1();
HRESULT CheckMediaType(const CMediaType *pmt);
HRESULT SetMediaType(const CMediaType *pmt);
STDMETHODIMP Receive(IMediaSample *pSample);
HRESULT Active();
HRESULT Inactive();
HRESULT InitFile();//初始化AVI输出文件
HRESULT WriteEnd();//完善AVI输出文件
STDMETHODIMP EndOfStream();
CFilter *pCFilter;
HANDLE hFile = NULL;//输出文件句柄
AVI_HEADER header;//AVI文件头
LONGLONG movi_LIST_Size_Pos;//movi_LIST块大小的文件位置
int mIndex = 0;//视频主索引条目索引
BYTE pV[80000000]; //视频索引块内存数组
int pVIndex;//视频索引内存数组索引
DWORD* VideoSize = NULL;//记录视频"块大小"地址
DWORD* Entries = NULL;//视频"条目数"地址
LONGLONG* pOffset = NULL;//基本偏移量
DWORD nEntriesInUse = 0;//视频主索引条目数
void WriteVideo(IMediaSample *pSample);//写视频样本
void WriteVideoIndexEnd();//完善上一个视频索引块,写视频主索引条目,递增视频主索引条目索引
int mIndexA = 0;//音频主索引条目索引
BYTE pA[80000000]; //音频索引块内存数组
int pAIndex;//音频索引内存数组索引
DWORD* AudioSize = NULL;//记录音频"块大小"地址
DWORD* EntriesA = NULL;//音频"条目数"地址
LONGLONG* pOffsetA = NULL;//基本偏移量
DWORD nEntriesInUseA = 0;//音频索引条目数
void WriteAudio(IMediaSample *pSample);//写音频样本
LONGLONG RIFF_Size_Pos;//RIFF_AVIX,RIFF大小位置
LONGLONG movi_Size_Pos;//RIFF_AVIX,movi_LIST大小位置
void WriteAudioIndexEnd();//完善上一个音频索引块,写音频主索引条目,递增音频主索引条目索引
LONGLONG BaseOffset[128];//基本偏移量
LONGLONG BaseOffsetA[128];//基本偏移量
CMediaType VideoMediaType;//视频媒体类型
DWORD VideoCount = 0;//视频帧总数
DWORD AudioCount = 0;//音频长度
DWORD ChuckCount = 0;//音频块长度
long VideoBufferSize = 0, AudioBufferSize = 0;
HANDLE hStop;//"停止"事件句柄
LONGLONG CurTime = 0;//视频流当前时间
HANDLE WriteCompleted = NULL;//"写样本完成"事件句柄
BOOL END = FALSE;//为TRUE时,标记视频流结束
};
class CPin2 : public CBaseInputPin
{
friend class CFilter;
friend class CPin1;
public:
CPin2(CFilter *pFilter, HRESULT *phr, LPCWSTR pPinName);
~CPin2();
HRESULT CheckMediaType(const CMediaType *pmt);
HRESULT SetMediaType(const CMediaType *pmt);
STDMETHODIMP Receive(IMediaSample *pSample);
STDMETHODIMP EndOfStream();
CFilter *pCFilter;//过滤器指针
CMediaType AudioMediaType;//音频媒体类型
LONGLONG CurTime = 0;//音频流当前时间
HANDLE WriteCompleted = NULL;//"写样本完成"事件句柄
BOOL END = FALSE;//为TRUE时,标记音频流结束
};
class CFilter : public CBaseFilter, public CCritSec, public IFileSinkFilter
{
friend class CPin1;
friend class CPin2;
public:
CFilter(LPWSTR lpName, LPUNKNOWN pUnk, HRESULT *phr);
virtual ~CFilter();
DECLARE_IUNKNOWN
STDMETHODIMP NonDelegatingQueryInterface(REFIID riid, void ** ppv);
virtual HRESULT STDMETHODCALLTYPE SetFileName(LPCOLESTR pszFileName, const AM_MEDIA_TYPE *pmt);
virtual HRESULT STDMETHODCALLTYPE GetCurFile(LPOLESTR * ppszFileName, AM_MEDIA_TYPE *pmt);
static CUnknown * WINAPI CreateInstance(LPUNKNOWN, HRESULT *);
int GetPinCount();
CBasePin *GetPin(int n);
CPin1 *pCPin1 = NULL; //视频引脚指针
CPin2 *pCPin2 = NULL; //音频引脚指针
WCHAR* m_pFileName = NULL;//要创作的AVI文件路径
};
template <class T> void SafeRelease(T** ppT)
{
if (*ppT)
{
(*ppT)->Release();
*ppT = NULL;
}
}
#endif //DLL_FILE
DLL.cpp
cpp
#include "DLL.h"
const AMOVIESETUP_MEDIATYPE PinType1[] = //视频引脚媒体类型
{
{
&MEDIATYPE_Video, //主要类型
&MEDIASUBTYPE_NULL //子类型
}
};
const AMOVIESETUP_MEDIATYPE PinType2[] = //音频引脚媒体类型
{
{
&MEDIATYPE_Audio, //主要类型
&MEDIASUBTYPE_MP3 //子类型
},
{
&MEDIATYPE_Audio, //主要类型
&MEDIASUBTYPE_ALAW //子类型
},
{
&MEDIATYPE_Audio, //主要类型
&MEDIASUBTYPE_PCM //子类型
}
};
const AMOVIESETUP_PIN sudPins[] = // 引脚信息
{
{
L"Video", //引脚名称
FALSE, //渲染引脚
FALSE, //输出引脚
FALSE, //具有该引脚的零个实例
FALSE, //可以创建一个以上引脚的实例
&CLSID_NULL, //该引脚连接的过滤器的类标识
NULL, //该引脚连接的引脚名称
1, //引脚支持的媒体类型数
PinType1 //媒体类型信息
},
{
L"Audio", //引脚名称
FALSE, //渲染引脚
FALSE, //输出引脚
FALSE, //具有该引脚的零个实例
FALSE, //可以创建一个以上引脚的实例
&CLSID_NULL, //该引脚连接的过滤器的类标识
NULL, //该引脚连接的引脚名称
3, //引脚支持的媒体类型数
PinType2 //媒体类型信息
}
};
const AMOVIESETUP_FILTER FILTERINFO = //过滤器的注册信息
{
&CLSID_AviWriter, //过滤器的类标识
L"写AVI", //过滤器的名称
MERIT_DO_NOT_USE, //过滤器优先值
2, //引脚数量
sudPins //引脚信息
};
CFactoryTemplate g_Templates[] = //类工厂模板数组
{
{
L"写AVI", //对象(这里为过滤器)名称
&CLSID_AviWriter, //对象CLSID的指针
CFilter::CreateInstance, //创建对象实例的函数的指针
NULL, //指向从DLL入口点调用的函数的指针
&FILTERINFO //指向AMOVIESETUP_FILTER结构的指针
}
};
int g_cTemplates = 1;//模板数组大小
STDAPI DllRegisterServer()//注册DLL
{
return AMovieDllRegisterServer2(TRUE);
}
STDAPI DllUnregisterServer()//删除DLL注册
{
return AMovieDllRegisterServer2(FALSE);
}
extern "C" BOOL WINAPI DllEntryPoint(HINSTANCE, ULONG, LPVOID);
BOOL APIENTRY DllMain(HANDLE hModule, DWORD dwReason, LPVOID lpReserved)
{
return DllEntryPoint((HINSTANCE)(hModule), dwReason, lpReserved);
}
CFilter.cpp
cpp
#include "DLL.h"
CFilter::CFilter(LPWSTR lpName, LPUNKNOWN pUnk, HRESULT *phr) : CBaseFilter(lpName, pUnk, (CCritSec *) this, CLSID_AviWriter)
{
pCPin1 = new CPin1(this, phr, L"Video");//创建视频输入引脚
pCPin2 = new CPin2(this, phr, L"Audio");//创建音频输入引脚
}
CFilter::~CFilter()
{
if (m_pFileName)delete[] m_pFileName;
}
CUnknown * WINAPI CFilter::CreateInstance(LPUNKNOWN pUnk, HRESULT *phr)
{
return new CFilter(L"写AVI", pUnk, phr);//创建过滤器
}
int CFilter::GetPinCount()
{
return 2;
}
CBasePin *CFilter::GetPin(int n)
{
if (n == 0)
{
return pCPin1;
}
else if (n == 1)
{
return pCPin2;
}
return NULL;
}
STDMETHODIMP CFilter::NonDelegatingQueryInterface(REFIID iid, void ** ppv)
{
if (iid == IID_IFileSinkFilter)
{
return GetInterface(static_cast<IFileSinkFilter*>(this), ppv);
}
else
return CBaseFilter::NonDelegatingQueryInterface(iid, ppv);
}
HRESULT CFilter::GetCurFile(LPOLESTR *ppszFileName, AM_MEDIA_TYPE *pmt)
{
CheckPointer(ppszFileName, E_POINTER);
*ppszFileName = NULL;
if (m_pFileName != NULL)
{
size_t len = 1 + lstrlenW(m_pFileName);
*ppszFileName = (LPOLESTR)QzTaskMemAlloc(sizeof(WCHAR) * (len));
if (*ppszFileName != NULL)
{
HRESULT hr = StringCchCopyW(*ppszFileName, len, m_pFileName);
}
}
if (pmt)
{
ZeroMemory(pmt, sizeof(*pmt));
pmt->majortype = MEDIATYPE_NULL;
pmt->subtype = MEDIASUBTYPE_NULL;
}
return S_OK;
}
HRESULT CFilter::SetFileName(LPCOLESTR pszFileName, const AM_MEDIA_TYPE *pmt)
{
CheckPointer(pszFileName, E_POINTER);
if (wcslen(pszFileName) > MAX_PATH || wcslen(pszFileName)<4)
return ERROR_FILENAME_EXCED_RANGE;
size_t len = 1 + lstrlenW(pszFileName);
m_pFileName = new WCHAR[len];
if (m_pFileName == 0)
return E_OUTOFMEMORY;
HRESULT hr = StringCchCopyW(m_pFileName, len, pszFileName);
if (m_pFileName[len - 2] != 'i' || m_pFileName[len - 3] != 'v' || m_pFileName[len - 4] != 'a' || m_pFileName[len - 5] != '.')//如果不是AVI文件
{
delete[] m_pFileName; m_pFileName = NULL;
return VFW_E_INVALID_FILE_FORMAT;//设置文件名失败
}
return S_OK;
}
CPin1.cpp
cpp
#include "DLL.h"
#include "dvdmedia.h"
CPin1::CPin1(CFilter *pFilter, HRESULT *phr, LPCWSTR pPinName) : CBaseInputPin(NAME("Video"), pFilter, pFilter, phr, pPinName)
{
pCFilter = pFilter;
WriteCompleted = CreateEvent(NULL, TRUE, FALSE, NULL);//手动重置,初始无信号
hStop = CreateEvent(NULL, TRUE, FALSE, NULL);//手动重置,初始无信号
//写视频索引头
pVIndex = 0;
char chfcc[4] = { 'i', 'x','0','0' };//视频索引块标识
CopyMemory(&pV[pVIndex], chfcc, 4); pVIndex += 4;//写索引块标识
VideoSize = (DWORD*)&pV[pVIndex];//获取视频"块大小"地址
DWORD ixSize = 0;//块大小
CopyMemory(&pV[pVIndex], &ixSize, 4); pVIndex += 4;//写索引块大小
WORD wLongsPerEntry = 2;//条目的DWORD大小
CopyMemory(&pV[pVIndex], &wLongsPerEntry, 2); pVIndex += 2;//写条目大小
BYTE bIndexSubType = 0;//必须为0
CopyMemory(&pV[pVIndex], &bIndexSubType, 1); pVIndex += 1;//写索引子类型
BYTE bIndexType = AVI_INDEX_OF_CHUNKS;//必须为AVI_INDEX_OF_CHUNKS
CopyMemory(&pV[pVIndex], &bIndexType, 1); pVIndex += 1;//写索引类型
Entries = (DWORD*)&pV[pVIndex];//记录视频"条目数"地址
CopyMemory(&pV[pVIndex], &nEntriesInUse, 4); pVIndex += 4;//写条目数
char ChunkId[4] = { '0','0','d','b' };//流标识
CopyMemory(&pV[pVIndex], ChunkId, 4); pVIndex += 4;//写流标识
pOffset = (LONGLONG*)&pV[pVIndex];//获取基本偏移量地址
BaseOffset[0] = 0;
CopyMemory(&pV[pVIndex], &BaseOffset[0], 8); pVIndex += 8;//写基本偏移量
DWORD dwReserved = 0;//保留
CopyMemory(&pV[pVIndex], &dwReserved, 4); pVIndex += 4;//写保留
//写音频索引头
pAIndex = 0;
char chfccA[4] = { 'i', 'x','0','1' };//音频索引块标识
CopyMemory(&pA[pAIndex], chfccA, 4); pAIndex += 4;//写索引块标识
AudioSize = (DWORD*)&pA[pAIndex];//记录音频"块大小"地址
DWORD ixSizeA = 0;//块大小
CopyMemory(&pA[pAIndex], &ixSizeA, 4); pAIndex += 4;//写索引块大小
WORD wLongsPerEntryA = 2;//条目的DWORD大小
CopyMemory(&pA[pAIndex], &wLongsPerEntryA, 2); pAIndex += 2;//写条目大小
BYTE bIndexSubTypeA = 0;//必须为0
CopyMemory(&pA[pAIndex], &bIndexSubTypeA, 1); pAIndex += 1;//写索引子类型
BYTE bIndexTypeA = AVI_INDEX_OF_CHUNKS;//必须为AVI_INDEX_OF_CHUNKS
CopyMemory(&pA[pAIndex], &bIndexTypeA, 1); pAIndex += 1;//写索引类型
EntriesA = (DWORD*)&pA[pAIndex];//记录音频"条目数"地址
CopyMemory(&pA[pAIndex], &nEntriesInUseA, 4); pAIndex += 4;//写条目数
char ChunkIdA[4] = { '0','1','w','b' };//流标识
CopyMemory(&pA[pAIndex], ChunkIdA, 4); pAIndex += 4;//写流标识
pOffsetA = (LONGLONG*)&pA[pAIndex];//获取基本偏移量地址
BaseOffsetA[0] = 0;
CopyMemory(&pA[pAIndex], &BaseOffsetA[0], 8); pAIndex += 8;//写基本偏移量
DWORD dwReservedA = 0;//保留
CopyMemory(&pA[pAIndex], &dwReservedA, 4); pAIndex += 4;//写保留
}
CPin1::~CPin1()
{
}
LONGLONG GetFilePos(HANDLE hFile)//获取文件当前位置
{
LARGE_INTEGER move;
move.QuadPart = 0;
LARGE_INTEGER CUR;
SetFilePointerEx(hFile, move, &CUR, FILE_CURRENT);
return CUR.QuadPart;
}
HRESULT WriteFoure(HANDLE hFile, LONGLONG Pos, void* p)//在指定位置写入4字节,并将文件指针返回到原来的位置
{
LONGLONG Cur = GetFilePos(hFile);//获取当前位置
LARGE_INTEGER Move;
Move.QuadPart = Pos;
SetFilePointerEx(hFile, Move, NULL, FILE_BEGIN);//移动到指定位置
WriteFile(hFile, p, 4, NULL, NULL);//写入4字节
Move.QuadPart = Cur;
SetFilePointerEx(hFile, Move, NULL, FILE_BEGIN);//返回到返回原来的位置
return S_OK;
}
HRESULT CPin1::CheckMediaType(const CMediaType *pmt)
{
if (pmt->majortype == MEDIATYPE_Video && pmt->formattype == FORMAT_VideoInfo)
{
return S_OK;
}
return S_FALSE;
}
HRESULT CPin1::SetMediaType(const CMediaType *pmt)
{
VideoMediaType = *pmt;
return CBaseInputPin::SetMediaType(pmt);
}
HRESULT CPin1::Receive(IMediaSample * pSample)//接收函数
{
if (pCFilter->pCPin2->END)//如果音频流已结束
{
WriteVideo(pSample);
return S_OK;
}
Agan:
DWORD dw1 = WaitForSingleObject(WriteCompleted, 0);//检测"写样本完成"信号
DWORD dw2 = WaitForSingleObject(hStop, 0);//检测"停止"信号
if (dw2 == WAIT_OBJECT_0)
{
return S_FALSE;//如果有"停止"信号,返回S_FALSE以终止流
}
if (dw1 != WAIT_OBJECT_0)//没有"写样本完成"信号
{
goto Agan;//重新检测(阻塞)
}
WriteVideo(pSample);
if (CurTime <= pCFilter->pCPin2->CurTime)//如果视频当前时间小于或等于音频当前时间
{
ResetEvent(pCFilter->pCPin2->WriteCompleted);//设置音频"写样本完成"无信号
SetEvent(WriteCompleted);//设置视频"写样本完成"有信号
}
else//如果音频当前时间小
{
ResetEvent(WriteCompleted);//设置视频"写样本完成"无信号
SetEvent(pCFilter->pCPin2->WriteCompleted);//设置音频"写样本完成"有信号
}
return S_OK;
}
void CPin1::WriteVideo(IMediaSample *pSample)//写视频样本
{
BYTE* pBy = NULL;
HRESULT hr = pSample->GetPointer(&pBy);//获取视频引脚样本缓冲区指针
long len = pSample->GetActualDataLength();//获取有效数据长度
HRESULT SyncPoint = pSample->IsSyncPoint();//获取同步点标志
LONGLONG star, end;
hr = pSample->GetTime(&star, &end);
if (GetFilePos(hFile) - BaseOffset[mIndex]+len > 0x7FFFFFFF)//如果块大小大于 0x7FFFFFFF
{
WriteVideoIndexEnd();//完善上一个视频索引块,写视频主索引条目,递增视频主索引条目索引
WriteAudioIndexEnd();//完善上一个音频索引块,写音频主索引条目,递增音频主索引条目索引
if (mIndex == 1)
{
DWORD movi_LIST_Size = (DWORD)(GetFilePos(hFile) - movi_LIST_Size_Pos - 4);
WriteFoure(hFile, movi_LIST_Size_Pos, &movi_LIST_Size);//写movi_LIST大小
*header.FileSize = (DWORD)(GetFilePos(hFile) - 8);//赋值RIFF文件大小
}
else
{
DWORD RIFF_Size = (DWORD)(GetFilePos(hFile) - RIFF_Size_Pos - 4);
WriteFoure(hFile, RIFF_Size_Pos, &RIFF_Size);//写RIFF_AVIX,RIFF大小
DWORD movi_Size = (DWORD)(GetFilePos(hFile) - movi_Size_Pos - 4);
WriteFoure(hFile, movi_Size_Pos, &movi_Size);//写RIFF_AVIX,movi_LIST大小
}
char ch1[4] = { 'R','I','F','F' };//写RIFF_AVIX头
WriteFile(hFile, ch1, 4, NULL, NULL);//RIFF
RIFF_Size_Pos = GetFilePos(hFile);
DWORD RIFF_Size = 0;
WriteFile(hFile, &RIFF_Size, 4, NULL, NULL);//RIFF大小
char ch2[4] = { 'A','V','I','X' };
WriteFile(hFile, ch2, 4, NULL, NULL);//AVIX
char ch3[4] = { 'L','I','S','T' };
WriteFile(hFile, ch3, 4, NULL, NULL);//LIST
movi_Size_Pos = GetFilePos(hFile);
DWORD movi_Size = 0;
WriteFile(hFile, &movi_Size, 4, NULL, NULL);//movi_LIST大小
char ch4[4] = { 'm','o','v','i' };
WriteFile(hFile, ch4, 4, NULL, NULL);//movi
}
CurTime = end;
if (SyncPoint == S_OK)//如果有同步点标志
{
char chV[4] = { '0','0','d','b' };
WriteFile(hFile, chV, 4, NULL, NULL);//写视频数据标识
}
else
{
char chV[4] = { '0','0','d','c' };
WriteFile(hFile, chV, 4, NULL, NULL);//写视频数据标识
}
DWORD size = len;
WriteFile(hFile, &size, 4, NULL, NULL);//写数据大小
DWORD offset = (DWORD)(GetFilePos(hFile) - BaseOffset[mIndex]);
CopyMemory(&pV[pVIndex], &offset, 4); pVIndex += 4;//写视频索引块偏移量
CopyMemory(&pV[pVIndex], &size, 4); pVIndex += 4;//写视频索引块大小
WriteFile(hFile, pBy, len, NULL, NULL);//写数据
VideoBufferSize = max(VideoBufferSize, len);
nEntriesInUse++;//递增视频主索引条目数
VideoCount++;//递增帧数量
}
void CPin1::WriteAudio(IMediaSample *pSample)//写音频样本
{
BYTE* pBy = NULL;
HRESULT hr = pSample->GetPointer(&pBy);//获取视频引脚样本缓冲区指针
long len = pSample->GetActualDataLength();//获取有效数据长度
LONGLONG star, end;
if (pCFilter->pCPin2->AudioMediaType.subtype== MEDIASUBTYPE_MP3)//如果是MP3流
{
WAVEFORMATEX* p = (WAVEFORMATEX*)pCFilter->pCPin2->AudioMediaType.pbFormat;
DWORD TimePerFrame = (DWORD)((float)1152 / (float)p->nSamplesPerSec * (float)10000000);//计算每帧持续时间,单位100纳秒
pCFilter->pCPin2->CurTime += TimePerFrame;//计算当前时间
ChuckCount += len;
AudioCount += len;
}
else
{
hr = pSample->GetTime(&star, &end);
pCFilter->pCPin2->CurTime = end;
WAVEFORMATEX* p = (WAVEFORMATEX*)pCFilter->pCPin2->AudioMediaType.pbFormat;
ChuckCount += len / p->nBlockAlign;
AudioCount += len / p->nBlockAlign;
}
DWORD size = len;
char chA[4] = { '0', '1', 'w', 'b' };
WriteFile(hFile, chA, 4, NULL, NULL);//写音频数据标识
WriteFile(hFile, &size, 4, NULL, NULL);//写数据大小
DWORD offset = (DWORD)(GetFilePos(hFile) - BaseOffsetA[mIndexA]);
CopyMemory(&pA[pAIndex], &offset, 4); pAIndex += 4;//写音频索引偏移量
CopyMemory(&pA[pAIndex], &size, 4); pAIndex += 4;//写音频索引大小
WriteFile(hFile, pBy, len, NULL, NULL);//写数据
AudioBufferSize = max(AudioBufferSize, len);
nEntriesInUseA++;//递增音频索引条目数
}
HRESULT CPin1::Active()
{
if (InitFile() != S_OK)return S_FALSE;
return CBaseInputPin::Active();
}
HRESULT CPin1::Inactive()
{
SetEvent(hStop);
WriteVideoIndexEnd();//完善上一个视频索引块,写视频主索引条目,递增视频主索引条目索引
WriteAudioIndexEnd();//完善上一个音频索引块,写音频主索引条目,递增音频主索引条目索引
if (mIndex == 1)
{
DWORD movi_LIST_Size = (DWORD)(GetFilePos(hFile) - movi_LIST_Size_Pos - 4);
WriteFoure(hFile, movi_LIST_Size_Pos, &movi_LIST_Size);//写movi_LIST大小
*header.FileSize= (DWORD)(GetFilePos(hFile) - 8);//赋值RIFF文件大小
}
else
{
DWORD RIFF_Size = (DWORD)(GetFilePos(hFile) - RIFF_Size_Pos - 4);
WriteFoure(hFile, RIFF_Size_Pos, &RIFF_Size);//写RIFF_AVIX,RIFF大小
DWORD movi_Size = (DWORD)(GetFilePos(hFile) - movi_Size_Pos - 4);
WriteFoure(hFile, movi_Size_Pos, &movi_Size);//写RIFF_AVIX,movi_LIST大小
}
*header.nEntriesInUse = mIndex;
*header.nEntriesInUse_A = mIndexA;
*header.dwTotalFrames = VideoCount;//赋值帧总数,avih参数
WriteEnd();//完善输出文件
return CBaseInputPin::Inactive();
}
HRESULT CPin1::InitFile()// 初始化AVI输出文件
{
if (pCFilter->m_pFileName == NULL)
{
MessageBox(0, L"没有指定输出文件", L"写AVI", MB_OK); return S_FALSE;
}
hFile = CreateFile(pCFilter->m_pFileName, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);//创建输出文件
if (INVALID_HANDLE_VALUE == hFile)
{
MessageBox(0, L"创建输出文件失败", L"写AVI", MB_OK);
hFile = NULL;
return S_FALSE;
}
END = FALSE; pCFilter->pCPin2->END = FALSE; pCFilter->pCPin2->CurTime = 0;
VideoBufferSize = 0; AudioBufferSize = 0;
AudioCount = 0;//音频长度
ChuckCount = 0;//音频块长度
nEntriesInUse = 0;//视频主索引条目数
mIndex = 0;
header.Entry[0].Offset = BaseOffset[0] = 0;
nEntriesInUseA = 0;//音频主索引条目数
mIndexA = 0;
header.EntryA[0].Offset = BaseOffsetA[0] = 0;
WriteFile(hFile, &header, 4642, NULL, NULL);//写AVI文件头
char chLIST[4] = { 'L','I','S','T' };//写movi_LIST头
WriteFile(hFile, chLIST, 4, NULL, NULL);//LIST
movi_LIST_Size_Pos = GetFilePos(hFile);//获取movi_LIST大小地址
DWORD movi_LIST_Size = 0;
WriteFile(hFile, &movi_LIST_Size, 4, NULL, NULL);//movi_LIST大小
char chmovi[4] = { 'm','o','v','i' };
WriteFile(hFile, chmovi, 4, NULL, NULL);//movi
ResetEvent(hStop);//设置"停止"无信号
SetEvent(WriteCompleted);//设置视频"写样本完成"有信号
ResetEvent(pCFilter->pCPin2->WriteCompleted);//设置音频"写样本完成"无信号
return S_OK;
}
void CPin1::WriteVideoIndexEnd()//完善上一个视频索引块,写视频主索引条目,递增视频主索引条目索引
{
DWORD VideoIndexSize = pVIndex;//视频索引块大小
*VideoSize = VideoIndexSize - 8;//写视频索引块大小
*Entries = nEntriesInUse;//写条目数量
*pOffset = BaseOffset[mIndex];//重写基本偏移量
header.Entry[mIndex].Offset = GetFilePos(hFile);//视频主索引条目偏移量
header.Entry[mIndex].Size = VideoIndexSize;//视频主索引条目大小
header.Entry[mIndex].Dur = nEntriesInUse;//视频主索引条目长度
WriteFile(hFile, &pV[0], VideoIndexSize, NULL, NULL);//写视频索引块
mIndex++;//递增视频主索引条目索引
nEntriesInUse = 0;
BaseOffset[mIndex] = GetFilePos(hFile);//赋值下一个基本偏移量
pVIndex = 32;//移动指针到,第1个条目位置
}
void CPin1::WriteAudioIndexEnd()//完善上一个音频索引块,写音频主索引条目,递增音频主索引条目索引
{
DWORD AudioIndexSize = pAIndex;//音频索引块大小
*AudioSize = AudioIndexSize - 8;//写音频索引块大小
*EntriesA = nEntriesInUseA;//写条目数量
*pOffsetA = BaseOffsetA[mIndexA];//重写基本偏移量
header.EntryA[mIndexA].Offset = GetFilePos(hFile);//音频主索引条目偏移量
header.EntryA[mIndexA].Size = AudioIndexSize;//音频主索项条目大小
header.EntryA[mIndexA].Dur = ChuckCount;//音频主索条目长度
ChuckCount = 0;
WriteFile(hFile, &pA[0], AudioIndexSize, NULL, NULL);//写音频索引块
mIndexA++;//递增音频主索引条目索引
nEntriesInUseA = 0;
BaseOffsetA[mIndexA] = GetFilePos(hFile);//赋值下一个基本偏移量
pAIndex = 32;//移动指针到,第1个条目位置
}
HRESULT CPin1::WriteEnd()//完善AVI输出文件
{
VIDEOINFOHEADER* pH = (VIDEOINFOHEADER*)VideoMediaType.pbFormat;
*header.dwScale = (DWORD)pH->AvgTimePerFrame;//采样率分母,视频strh参数
*header.dwMicroSecPerFrame = (DWORD)(pH->AvgTimePerFrame / 10);//是视频帧持续时间,单位微秒,avih参数
*header.dwWidth = pH->bmiHeader.biWidth;//图像的宽度
*header.dwHeight = pH->bmiHeader.biHeight;//图像的高度
CopyMemory(header.fccHandler, &VideoMediaType.subtype.Data1, 4);//获取FOURCC
header.rcFrame->left = 0; header.rcFrame->top = 0;
header.rcFrame->right = (short)pH->bmiHeader.biWidth; header.rcFrame->bottom = (short)pH->bmiHeader.biHeight;//目标矩形
*header.biWidth = pH->bmiHeader.biWidth;//图像的宽度
*header.biHeight = pH->bmiHeader.biHeight;//图像的高度
*header.biBitCount = pH->bmiHeader.biBitCount;//每个像素的位数
*header.biCompression = pH->bmiHeader.biCompression;//编码方式
*header.biSizeImage = pH->bmiHeader.biSizeImage;//图像大小
*header.biXPelsPerMeter = pH->bmiHeader.biXPelsPerMeter;//目标设备的水平分辨率
*header.biYPelsPerMeter = pH->bmiHeader.biYPelsPerMeter;//目标设备的垂直分辨率
*header.biClrUsed = pH->bmiHeader.biClrUsed;//颜色数量
*header.biClrImportant = pH->bmiHeader.biClrImportant;//重要颜色数量
*header.dwLength = VideoCount;//赋值视频流的长度,单位视频帧,视频strh参数
WAVEFORMATEX* p = (WAVEFORMATEX*)pCFilter->pCPin2->AudioMediaType.pbFormat;
CopyMemory(header.fccHandler_A, &pCFilter->pCPin2->AudioMediaType.subtype.Data1, 4);//获取FOURCC,音频strh参数
if (pCFilter->pCPin2->AudioMediaType.subtype == MEDIASUBTYPE_MP3)//如果是MP3流
{
*pCFilter->pCPin1->header.dwRate_A = p->nAvgBytesPerSec;
*pCFilter->pCPin1->header.cbSize_A = p->cbSize;//附加信息大小
CopyMemory(pCFilter->pCPin1->header.pApend, (BYTE*)p + sizeof(WAVEFORMATEX), p->cbSize);//复制附加信息
BYTE* pP = pCFilter->pCPin1->header.pApend;
pP += p->cbSize;
CopyMemory(pP, "JUNK", 4); pP += 4;
DWORD JUNKSize = 56 - p->cbSize;
CopyMemory(pP, &JUNKSize, 4);
*pCFilter->pCPin1->header.strf_A_Size += p->cbSize;
*pCFilter->pCPin1->header.dwStart_A = 0;//流的开始时间
}
else//如果是PCM或ALAW流
{
*pCFilter->pCPin1->header.dwRate_A = p->nSamplesPerSec;//采样率分子,音频strh参数(采样率分母使用值1,故采样率分子直接等于采样率)
}
*pCFilter->pCPin1->header.dwSampleSize_A = p->nBlockAlign;//单个数据样本的大小,音频strh参数
*header.dwLength_A = AudioCount;//赋值音频流的长度,单位音频帧,音频strh参数
*pCFilter->pCPin1->header.wFormatTag_A = (WORD)pCFilter->pCPin2->AudioMediaType.subtype.Data1;//音频格式,音频strf参数
*pCFilter->pCPin1->header.nChannels_A = p->nChannels;//声道数,音频strf参数
*pCFilter->pCPin1->header.nSamplesPerSec_A = p->nSamplesPerSec;//采样率,音频strf参数
*pCFilter->pCPin1->header.nAvgBytesPerSec_A = p->nAvgBytesPerSec;//传输率,音频strf参数
*pCFilter->pCPin1->header.nBlockAlign_A = p->nBlockAlign;//块对齐,音频strf参数
*pCFilter->pCPin1->header.wBitsPerSample_A = p->wBitsPerSample;//样本位数,音频strf参数
if (VideoBufferSize % 4)VideoBufferSize = (VideoBufferSize / 4) * 4 + 4;//将缓冲区大小4倍取整
if (AudioBufferSize % 4)AudioBufferSize = (AudioBufferSize / 4) * 4 + 4;
*header.strh_dwSuggestedBufferSize = VideoBufferSize;//赋值视频建议的缓冲区大小,视频strh参数
*header.strh_A_dwSuggestedBufferSize = AudioBufferSize;//赋值音频建议的缓冲区大小,音频strh参数
LARGE_INTEGER Move;
Move.QuadPart = 0;
SetFilePointerEx(hFile, Move, NULL, FILE_BEGIN);//移动文件指针,到开始位置
WriteFile(hFile, &header, 4642, NULL, NULL);//重写AVI文件头,使赋值参数应用于文件
CloseHandle(hFile);//关闭输出文件
hFile = NULL;
return S_OK;
}
#define EC_VIDEOSTREAMEND EC_USER+1122//自定义"视频流结束"事件通知
STDMETHODIMP CPin1::EndOfStream()
{
pCFilter->NotifyEvent(EC_VIDEOSTREAMEND, NULL, NULL);//发送自定义"视频流结束"事件通知
END = TRUE;
return CBaseInputPin::EndOfStream();
}
CPin2.cpp
cpp
#include "DLL.h"
CPin2::CPin2(CFilter *pFilter, HRESULT *phr, LPCWSTR pPinName) : CBaseInputPin(NAME("Audio"), pFilter, pFilter, phr, pPinName)
{
pCFilter = pFilter;
WriteCompleted = CreateEvent(NULL, TRUE, FALSE, NULL);//手动重置,初始无信号
}
CPin2::~CPin2()//输入引脚析构函数
{
}
HRESULT CPin2::CheckMediaType(const CMediaType *pmt)
{
if (pmt->majortype == MEDIATYPE_Audio && (pmt->subtype == MEDIASUBTYPE_MP3 || pmt->subtype == MEDIASUBTYPE_PCM || pmt->subtype == MEDIASUBTYPE_ALAW)
&& pmt->formattype == FORMAT_WaveFormatEx)return S_OK;
return S_FALSE;
}
HRESULT CPin2::SetMediaType(const CMediaType *pmt)
{
AudioMediaType = *pmt;
return CBaseInputPin::SetMediaType(pmt);
}
HRESULT CPin2::Receive(IMediaSample * pSample)//接收函数
{
if (pCFilter->pCPin1->END)//如果视频流已结束
{
pCFilter->pCPin1->WriteAudio(pSample);//写音频样本
return S_OK;
}
Agan:
DWORD dw1 = WaitForSingleObject(WriteCompleted, 0);//检测"写样本完成"信号
DWORD dw2 = WaitForSingleObject(pCFilter->pCPin1->hStop, 0);//检测"停止"信号
if (dw2 == WAIT_OBJECT_0)
{
return S_FALSE;//如果有"停止"信号,返回S_FALSE以终止流
}
if (dw1 != WAIT_OBJECT_0)//没有"写样本完成"信号
{
goto Agan;//重新检测(阻塞)
}
pCFilter->pCPin1->WriteAudio(pSample);//写音频样本
if (CurTime < pCFilter->pCPin1->CurTime)//如果音频当前时间小
{
ResetEvent(pCFilter->pCPin1->WriteCompleted);//设置视频"写样本完成"无信号
SetEvent(WriteCompleted);//设置音频"写样本完成"有信号
}
else//如果视频当前时间小于或等于音频当前时间
{
ResetEvent(WriteCompleted);//设置音频"写样本完成"无信号
SetEvent(pCFilter->pCPin1->WriteCompleted);//设置视频"写样本完成"有信号
}
return S_OK;
}
#define EC_AUDIOSTREAMEND EC_USER+1121//自定义"音频流结束"事件通知
STDMETHODIMP CPin2::EndOfStream()
{
pCFilter->NotifyEvent(EC_AUDIOSTREAMEND, NULL, NULL);//发送自定义"音频流结束"事件通知
END = TRUE;
return CBaseInputPin::EndOfStream();
}