avi_dialogs.cpp

Go to the documentation of this file.

#define _WIN32_DCOM 

#include <windows.h>
#include <atlbase.h>
#include <stdio.h>
#include <dshow.h>

#include "avi_write.h"
#include "avi_resource.h"

extern void CreateCompressionFilter(IMoniker *pMoniker, BOOL dvPal);
extern int bCompressed;

void CentreDialogOnScreen(HWND hWnd){
 RECT rc;
 GetWindowRect(hWnd, &rc);
 SetWindowPos(hWnd, NULL,
    (GetSystemMetrics(SM_CXSCREEN) - (rc.right - rc.left)) / 2,
    (GetSystemMetrics(SM_CYSCREEN) - (rc.bottom - rc.top)) / 2,
    0, 0, SWP_NOSIZE | SWP_NOZORDER);
}


void OnInitDialog(HWND hDlg){
    HRESULT hr;
    ICreateDevEnum *pSysDevEnum = NULL;
    IEnumMoniker *pEnum = NULL;
    IMoniker *pMoniker = NULL;
    hr = CoCreateInstance(CLSID_SystemDeviceEnum, NULL, 
        CLSCTX_INPROC_SERVER, IID_ICreateDevEnum, 
        (void**)&pSysDevEnum);
    if (FAILED(hr))return;
    hr = pSysDevEnum->CreateClassEnumerator(
             CLSID_VideoCompressorCategory, &pEnum, 0);
    if (hr == S_OK) { // S_FALSE means nothing in this category.    
        while (S_OK == pEnum->Next(1, &pMoniker, NULL))    {
            IPropertyBag *pPropBag = NULL;
            pMoniker->BindToStorage(0, 0, IID_IPropertyBag, 
                (void **)&pPropBag);
            VARIANT var;
            VariantInit(&var);
            hr = pPropBag->Read(L"FriendlyName", &var, 0);
            if (SUCCEEDED(hr)){
               USES_CONVERSION;
                LRESULT iSel=SendDlgItemMessage(hDlg,IDC_LIST1,LB_ADDSTRING,0,
                (LPARAM)OLE2T(var.bstrVal));
            }   
            VariantClear(&var); 
            pPropBag->Release();
            pMoniker->Release();
        }
    }
    SendDlgItemMessage(hDlg, IDC_LIST1, 
                       LB_SETCURSEL, 0, 0);
    pSysDevEnum->Release();
    pEnum->Release();
}

void OnSelectFilter(HWND hDlg){
    HRESULT hr;
    ICreateDevEnum *pSysDevEnum = NULL;
    IEnumMoniker *pEnum = NULL;
    IMoniker *pMoniker = NULL;
    int id;
    char item_list[256];
    BOOL bPal=FALSE;
    id=(int)SendDlgItemMessage(hDlg, IDC_LIST1, 
                       LB_GETCURSEL, 0, 0);
    SendDlgItemMessage(hDlg, IDC_LIST1,LB_GETTEXT,  
                       (WPARAM)id,(LPARAM)(LPCTSTR)item_list);
    hr = CoCreateInstance(CLSID_SystemDeviceEnum, NULL, 
        CLSCTX_INPROC_SERVER, IID_ICreateDevEnum, 
        (void**)&pSysDevEnum);
    if (FAILED(hr))return;
    hr = pSysDevEnum->CreateClassEnumerator(
             CLSID_VideoCompressorCategory, &pEnum, 0);
    if (hr == S_OK){  // S_FALSE means nothing in this category.
        while (S_OK == pEnum->Next(1, &pMoniker, NULL))        {
            IPropertyBag *pPropBag = NULL;
            pMoniker->BindToStorage(0, 0, IID_IPropertyBag, 
                (void **)&pPropBag);
            VARIANT var;
            VariantInit(&var);
            hr = pPropBag->Read(L"FriendlyName", &var, 0);
            if (SUCCEEDED(hr)){
               USES_CONVERSION;
               if(strcmp(OLE2T(var.bstrVal),item_list) == 0){
                  if(strcmp(item_list,"DV Video Encoder") == 0)bPal=TRUE;
                  //if(bPal)MessageBox(NULL,item_list,OLE2T(var.bstrVal),MB_OK);
                  CreateCompressionFilter(pMoniker,bPal);
                  VariantClear(&var); 
                  pPropBag->Release();
                  pMoniker->Release();
                  break;
               }
            }   
            VariantClear(&var); 
            pPropBag->Release();
            pMoniker->Release();
        }
    }
    SendDlgItemMessage(hDlg, IDC_LIST1, 
                       LB_SETCURSEL, 0, 0);
    pSysDevEnum->Release();
    pEnum->Release();
}


 
static BOOL CALLBACK DlgProc(HWND hwnd,UINT msg,
                                   WPARAM wparam,LPARAM lparam){
 switch( msg ) {
   case WM_INITDIALOG:
     OnInitDialog(hwnd);
     CentreDialogOnScreen(hwnd);
     if(bCompressed)SendDlgItemMessage(hwnd,IDC_CHECK1,BM_SETCHECK,TRUE,0);
     return (TRUE);
   case WM_DESTROY:
     break;
   case WM_DRAWITEM:
     return TRUE;
   case WM_PAINT:
     break;
   case WM_COMMAND:
      switch(LOWORD(wparam)){
        case IDCANCEL:
          EndDialog(hwnd,FALSE);
          return(TRUE);
        case IDOK:
          if(IsDlgButtonChecked(hwnd,IDC_CHECK1) == BST_CHECKED) bCompressed=TRUE;
          else bCompressed=FALSE; 
          OnSelectFilter(hwnd);
          EndDialog(hwnd,TRUE);
          return(TRUE);
        default:
          break;
      }
      break;
    default: break;
 }
 return(FALSE);
}

BOOL DoDialog(HINSTANCE hInstance, HWND hWnd){
 return (BOOL)DialogBoxParam(hInstance,MAKEINTRESOURCE(IDD_DIALOG1),hWnd,
            (DLGPROC)DlgProc,(LPARAM)1);
}