h264dmo.cpp

This the source release kit for the following system configuration(s): - AMD Alchemy(TM) DBAu1200(

//------------------------------------------------------------------------------
// File: H264DMO.cpp
//------------------------------------------------------------------------------


#include "stdafx.h"

#define FIX_LOCK_NAME
#include <dmo.h>
#include <dmoimpl.h>
#include <initguid.h>
#include <limits.h>
#include <uuids.h>      // DirectShow media type guids

#include "resource.h"
#include "h264.h"

CComModule _Module;

BEGIN_OBJECT_MAP(ObjectMap)
OBJECT_ENTRY(CLSID_H264DMO, CH264Dmo)
END_OBJECT_MAP()

#ifndef NUMELMS
   #define NUMELMS(aa) (sizeof(aa)/sizeof((aa)[0]))
#endif

#define FOURCC_H264    0x34363248
// {0x34363248-0000-0010-8000-00AA00389B71}
DEFINE_GUID(MEDIASUBTYPE_H264, 
0x34363248, 0x0000, 0x0010, 0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71);

#define FOURCC_h264    0x34363268
// {0x34363268-0000-0010-8000-00AA00389B71}
DEFINE_GUID(MEDIASUBTYPE_h264, 
0x34363268, 0x0000, 0x0010, 0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71);

// {05589f91-c356-11ce-bf01-00aa0055595a}
DEFINE_GUID(FORMAT_H264VideoInfo,
0x05589f91, 0xc356, 0x11ce, 0xbf, 0x01, 0x00, 0xaa, 0x00, 0x55, 0x59, 0x5a);

/////////////////////////////////////////////////////////////////////////////
// DLL Entry Point
extern "C"
BOOL WINAPI DllMain(HANDLE hDllHandle, DWORD dwReason, LPVOID /*lpReserved*/)
{
  if (dwReason == DLL_PROCESS_ATTACH)
  {
    _Module.Init(ObjectMap, reinterpret_cast<HINSTANCE>(hDllHandle));
    DisableThreadLibraryCalls(reinterpret_cast<HINSTANCE>(hDllHandle));
  } else if (dwReason == DLL_PROCESS_DETACH)
    _Module.Term();

  return TRUE;
}

/////////////////////////////////////////////////////////////////////////////
// Used to determine whether the DLL can be unloaded by OLE
STDAPI DllCanUnloadNow(void)
{
  return (_Module.GetLockCount()==0) ? S_OK : S_FALSE;
}

/////////////////////////////////////////////////////////////////////////////
// Returns a class factory to create an object of the requested type
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv)
{
  return _Module.GetClassObject(rclsid, riid, ppv);
}

/////////////////////////////////////////////////////////////////////////////
// DllRegisterServer - Adds entries to the system registry
STDAPI DllRegisterServer(void)
{
  //set the valid input types
  DMO_PARTIAL_MEDIATYPE InTypes[2];
  InTypes[0].type    = MEDIATYPE_Video;
  InTypes[0].subtype = MEDIASUBTYPE_H264;
  InTypes[1].type    = MEDIATYPE_Video;
  InTypes[1].subtype = MEDIASUBTYPE_h264;

  //set the valid output types
  DMO_PARTIAL_MEDIATYPE OutTypes[1];
  OutTypes[0].type    = MEDIATYPE_Video;
  OutTypes[0].subtype = MEDIASUBTYPE_Overlay;

  DMORegister(L"Alchemy H264 Video DMO Codec", 
              CLSID_H264DMO, 
              DMOCATEGORY_VIDEO_DECODER,
              0,
              NUMELMS(InTypes),
              InTypes,
              NUMELMS(OutTypes),
              OutTypes);

  // registers object
  return _Module.RegisterServer();
}

/////////////////////////////////////////////////////////////////////////////
// DllUnregisterServer - Removes entries from the system registry
STDAPI DllUnregisterServer(void)
{
  DMOUnregister(CLSID_H264DMO, DMOCATEGORY_VIDEO_DECODER);
  return _Module.UnregisterServer();
}