DirectShow AmCap源码

//------------------------------------------------------------------------------ // File: AMCap.cpp // // Desc: Audio/Video Capture sample for DirectShow // // // Copyright (c) Microsoft Corporation. All rights reserved. //------------------------------------------------------------------------------ #include "stdafx.h" #include <dbt.h> #include <mmreg.h> #include <msacm.h> #include <fcntl.h> #include <io.h> #include <stdio.h> #include <commdlg.h> #include <strsafe.h> #include "stdafx.h" #include "amcap.h" #include "status.h" #include "crossbar.h" #include "SampleCGB.h" //------------------------------------------------------------------------------ // Macros //------------------------------------------------------------------------------ #define ABS(x) (((x) > 0) ? (x) : -(x)) // An application can advertise the existence of its filter graph // by registering the graph with a global Running Object Table (ROT). // The GraphEdit application can detect and remotely view the running // filter graph, allowing you to 'spy' on the graph with GraphEdit. // // To enable registration in this sample, define REGISTER_FILTERGRAPH. // #ifdef DEBUG #define REGISTER_FILTERGRAPH #endif //------------------------------------------------------------------------------ // Global data //------------------------------------------------------------------------------ HINSTANCE ghInstApp=0; HACCEL ghAccel=0; HFONT ghfontApp=0; TEXTMETRIC gtm={0}; TCHAR gszAppName[]=TEXT("AMCAP"); HWND ghwndApp=0, ghwndStatus=0; HDEVNOTIFY ghDevNotify=0; PUnregisterDeviceNotification gpUnregisterDeviceNotification=0; PRegisterDeviceNotification gpRegisterDeviceNotification=0; DWORD g_dwGraphRegister=0; struct _capstuff { WCHAR wszCaptureFile[_MAX_PATH]; WORD wCapFileSize; // size in Meg ISampleCaptureGraphBuilder *pBuilder; IVideoWindow *pVW; IMediaEventEx *pME; IAMDroppedFrames *pDF; IAMVideoCompression *pVC; IAMVfwCaptureDialogs *pDlg; IAMStreamConfig *pASC; // for audio cap IAMStreamConfig *pVSC; // for video cap IBaseFilter *pRender; IBaseFilter *pVCap, *pACap; //采集设备filter IGraphBuilder *pFg; IFileSinkFilter *pSink; IConfigAviMux *pConfigAviMux; int iMasterStream; BOOL fCaptureGraphBuilt; BOOL fPreviewGraphBuilt; BOOL fCapturing; BOOL fPreviewing; BOOL fMPEG2; BOOL fCapAudio; BOOL fCapCC; BOOL fCCAvail; BOOL fCapAudioIsRelevant; bool fDeviceMenuPopulated; IMoniker *rgpmVideoMenu[10]; IMoniker *rgpmAudioMenu[10]; IMoniker *pmVideo; //视频采集设备Moniker IMoniker *pmAudio; double FrameRate; BOOL fWantPreview; long lCapStartTime; long lCapStopTime; WCHAR wachFriendlyName[120]; //保存当前采集设备的FriendlyName BOOL fUseTimeLimit; BOOL fUseFrameRate; DWORD dwTimeLimit; int iFormatDialogPos; int iSourceDialogPos; int iDisplayDialogPos; int iVCapDialogPos; int iVCrossbarDialogPos; int iTVTunerDialogPos; int iACapDialogPos; int iACrossbarDialogPos; int iTVAudioDialogPos; int iVCapCapturePinDialogPos; int iVCapPreviewPinDialogPos; int iACapCapturePinDialogPos; long lDroppedBase; long lNotBase; BOOL fPreviewFaked; CCrossbar *pCrossbar; int iVideoInputMenuPos; LONG NumberOfVideoInputs; HMENU hMenuPopup; int iNumVCapDevices; } gcap; // implements IAMCopyCaptureFileProgress // class CProgress : public IAMCopyCaptureFileProgress { public: CProgress() { }; ~CProgress() { }; STDMETHODIMP_(ULONG) AddRef() { return 1; }; STDMETHODIMP_(ULONG) Release() { return 0; }; STDMETHODIMP QueryInterface(REFIID iid, void **p) { if (p == NULL) { return E_POINTER; } if (iid == IID_IUnknown) { *p = reinterpret_cast<IUnknown*>(this); AddRef(); return S_OK; } else if (iid == IID_IAMCopyCaptureFileProgress) { *p = reinterpret_cast<IAMCopyCaptureFileProgress*>(this); AddRef(); return S_OK; } else { return E_NOINTERFACE; } }; STDMETHODIMP Progress(int i) { TCHAR tach[80]; HRESULT hr = StringCchPrintf(tach, 80, TEXT("Save File Progress: %d%%\0"), i); statusUpdateStatus(ghwndStatus, tach); return S_OK; }; }; //------------------------------------------------------------------------------ // Function Prototypes //------------------------------------------------------------------------------ typedef LONG(PASCAL *LPWNDPROC)(HWND, UINT, WPARAM, LPARAM); // pointer to a window procedure LONG WINAPI AppWndProc(HWND hwnd, UINT uiMessage, WPARAM wParam, LPARAM lParam); LONG PASCAL AppCommand(HWND hwnd, unsigned msg, WPARAM wParam, LPARAM lParam); BOOL CALLBACK AboutDlgProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam); void ErrMsg(LPTSTR sz,...); BOOL SetCaptureFile(HWND hWnd); BOOL SaveCaptureFile(HWND hWnd); BOOL AllocCaptureFile(HWND hWnd); int DoDialog(HWND hwndParent, int DialogID, DLGPROC fnDialog, long lParam); int FAR PASCAL AllocCapFileProc(HWND hDlg, UINT Message, UINT wParam, LONG lParam); int FAR PASCAL FrameRateProc(HWND hDlg, UINT Message, UINT wParam, LONG lParam); int FAR PASCAL TimeLimitProc(HWND hDlg, UINT Message, UINT wParam, LONG lParam); int FAR PASCAL PressAKeyProc(HWND hDlg, UINT Message, UINT wParam, LONG lParam); void TearDownGraph(void); BOOL BuildCaptureGraph(); BOOL BuildPreviewGraph(); void UpdateStatus(BOOL fAllStats); void AddDevicesToMenu(); void ChooseDevices(TCHAR *szVideo, TCHAR *szAudio); void ChooseDevices(IMoniker *pmVideo, IMoniker *pmAudio); void ChooseFrameRate(); BOOL InitCapFilters(); void FreeCapFilters(); BOOL StopPreview(); BOOL StartPreview(); BOOL StopCapture(); DWORDLONG GetSize(LPCTSTR tach); void MakeMenuOptions(); void OnClose(); //------------------------------------------------------------------------------ // Name: SetAppCaption() // Desc: Set the caption to be the application name followed by the capture file //------------------------------------------------------------------------------ void SetAppCaption() { TCHAR tach[_MAX_PATH + 80]; StringCchCopy(tach, NUMELMS(tach), gszAppName); if(gcap.wszCaptureFile[0] != 0) { HRESULT hr = StringCchCat(tach, _MAX_PATH + 80, TEXT(" - ")); hr = StringCchCat(tach, _MAX_PATH + 80, gcap.wszCaptureFile); } SetWindowText(ghwndApp, tach); } /*----------------------------------------------------------------------------*\ | AppInit( hInst, hPrev) | | | | Description: | | This is called when the application is first loaded into | | memory. It performs all initialization that doesn't need to be done | | once per instance. | | | | Arguments: | | hInstance instance handle of current instance | | hPrev instance handle of previous instance | | | | Returns: | | TRUE if successful, FALSE if not | | | \*----------------------------------------------------------------------------*/ BOOL AppInit(HINSTANCE hInst, HINSTANCE hPrev, int sw) { WNDCLASS cls; HDC hdc; const DWORD dwExStyle = 0; CoInitializeEx(NULL, COINIT_APARTMENTTHREADED); /* Save instance handle for DialogBoxs */ ghInstApp = hInst; ghAccel = LoadAccelerators(hInst, MAKEI