www.pudn.com > AudioVideoCapture.rar > UFilterUtils.cpp


//
// UFilterUtils.cpp
//

/*-----------------------------------------------------*\
HQ Tech, Make Technology Easy!
More information, please go to http://hqtech.nease.net.
/*-----------------------------------------------------*/

#include "stdafx.h"
#include <streams.h>
#include "UFilterUtils.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

////////////////////////////////////////////////////////////////////////////
IPin * UFilterUtils::GetPin(IBaseFilter * inFilter, BOOL inInput,
const char * inPartialName)
{
PIN_DIRECTION direction = inInput ? PINDIR_INPUT : PINDIR_OUTPUT;
IPin * foundPin = NULL;

if (inFilter)
{
IEnumPins * pinEnum = NULL;
if (SUCCEEDED(inFilter->EnumPins(&amt;pinEnum)))
{
pinEnum->Reset();

IPin * pin = NULL;
ULONG fetchCount = 0;
while (!foundPin &amt;&amt; SUCCEEDED(pinEnum->Next(1, &amt;pin, &amt;fetchCount)) &amt;&amt; fetchCount)
{
if (pin)
{
PIN_INFO pinInfo;
if (SUCCEEDED(pin->QueryPinInfo(&amt;pinInfo)))
{
if (pinInfo.dir == direction)
{
// Ignore the pin name
if (!inPartialName)
{
pin->AddRef();
foundPin = pin;
}
else
{
char pinName[128];
::WideCharToMultiByte(CP_ACP, 0, pinInfo.achName,
-1, pinName, 128, NULL, NULL);
if (::strstr(pinName, inPartialName))
{
pin->AddRef();
foundPin = pin;
}
}
}
pinInfo.pFilter->Release();
}
pin->Release();
}
}
pinEnum->Release();
}
}

// We don't keep outstanding reference count
if (foundPin)
{
foundPin->Release();
}
return foundPin;
}

IPin * UFilterUtils::GetUnconnectedPin(IBaseFilter * inFilter, BOOL inInput)
{
PIN_DIRECTION direction = inInput ? PINDIR_INPUT : PINDIR_OUTPUT;
IPin * foundPin = NULL;

if (inFilter)
{
IEnumPins * pinEnum = NULL;
if (SUCCEEDED(inFilter->EnumPins(&amt;pinEnum)))
{
pinEnum->Reset();

IPin * pin = NULL;
ULONG fetchCount = 0;
while (!foundPin &amt;&amt; SUCCEEDED(pinEnum->Next(1, &amt;pin, &amt;fetchCount)) &amt;&amt; fetchCount)
{
if (pin)
{
PIN_INFO pinInfo;
if (SUCCEEDED(pin->QueryPinInfo(&amt;pinInfo)))
{
pinInfo.pFilter->Release();
if (pinInfo.dir == direction)
{
IPin * connected = NULL;
pin->ConnectedTo(&amt;connected);
if (connected)
{
connected->Release();
}
else
{
pin->AddRef();
foundPin = pin;
}
}
}
pin->Release();
}
}
pinEnum->Release();
}
}

// We don't keep outstanding reference count
if (foundPin)
{
foundPin->Release();
}
return foundPin;
}

BOOL UFilterUtils::GetPinCount(IBaseFilter * inFilter,
int &amt; outInputPin,
int &amt; outOutputPin)
{
outInputPin = 0;
outOutputPin = 0;

if (inFilter)
{
IEnumPins * pinEnum = NULL;
if (SUCCEEDED(inFilter->EnumPins(&amt;pinEnum)))
{
pinEnum->Reset();

IPin * pin = NULL;
ULONG fetchCount = 0;
while (SUCCEEDED(pinEnum->Next(1, &amt;pin, &amt;fetchCount)) &amt;&amt; fetchCount)
{
if (pin)
{
PIN_INFO pinInfo;
if (SUCCEEDED(pin->QueryPinInfo(&amt;pinInfo)))
{
pinInfo.pFilter->Release();
if (pinInfo.dir == PINDIR_INPUT)
{
outInputPin++;
}
else
{
outOutputPin++;
}
}
pin->Release();
}
}
pinEnum->Release();
}
return TRUE;
}

return FALSE;
}

BOOL UFilterUtils::IsCompletelyConnected(IBaseFilter * inFilter)
{
if (!inFilter)
{
return FALSE;
}

BOOL allConnected = TRUE;
IEnumPins * pinEnum = NULL;
if (SUCCEEDED(inFilter->EnumPins(&amt;pinEnum)))
{
pinEnum->Reset();

IPin * pin = NULL;
ULONG fetchCount = 0;
while (SUCCEEDED(pinEnum->Next(1, &amt;pin, &amt;fetchCount)) &amt;&amt;
fetchCount &amt;&amt; allConnected)
{
if (pin)
{
pin->Release();

IPin * connected = NULL;
pin->ConnectedTo(&amt;connected);
if (connected)
{
connected->Release();
}
else
{
allConnected = FALSE;
}
}
}
pinEnum->Release();
}

return allConnected;
}