bool CaptureDShow::init()
{
// Create the pipeline.
if (FAILED(CoCreateInstance(CLSID_FilterGraph,
NULL,
CLSCTX_INPROC_SERVER,
IID_IGraphBuilder,
reinterpret_cast<void **>(&this->m_graph))))
return false;
// Create the webcam filter.
IBaseFilter *webcamFilter = this->findFilterP(this->m_device);
if (!webcamFilter) {
this->m_graph->Release();
this->m_graph = NULL;
return false;
}
if (FAILED(this->m_graph->AddFilter(webcamFilter, SOURCE_FILTER_NAME))) {
this->m_graph->Release();
this->m_graph = NULL;
return false;
}
// Create the Sample Grabber filter.
IBaseFilter *grabberFilter = NULL;
if (FAILED(CoCreateInstance(CLSID_SampleGrabber,
NULL,
CLSCTX_INPROC_SERVER,
IID_IBaseFilter,
reinterpret_cast<void **>(&grabberFilter)))) {
this->m_graph->Release();
this->m_graph = NULL;
return false;
}
if (FAILED(this->m_graph->AddFilter(grabberFilter, L"Grabber"))) {
this->m_graph->Release();
this->m_graph = NULL;
return false;
}
ISampleGrabber *grabberPtr = NULL;
if (FAILED(grabberFilter->QueryInterface(IID_ISampleGrabber,
reinterpret_cast<void **>(&grabberPtr)))) {
this->m_graph->Release();
this->m_graph = NULL;
return false;
}
if (FAILED(grabberPtr->SetOneShot(FALSE))) {
this->m_graph->Release();
this->m_graph = NULL;
return false;
}
HRESULT hr = grabberPtr->SetBufferSamples(TRUE);
if (FAILED(hr)) {
this->m_graph->Release();
this->m_graph = NULL;
return false;
}
if (this->m_ioMethod != IoMethodDirectRead) {
int type = this->m_ioMethod == IoMethodGrabSample? 0: 1;
hr = grabberPtr->SetCallback(&this->m_frameGrabber, type);
}
this->m_grabber = SampleGrabberPtr(grabberPtr, this->deleteUnknown);
if (!this->connectFilters(this->m_graph, webcamFilter, grabberFilter)) {
this->m_graph->Release();
this->m_graph = NULL;
return false;
}
// Create null filter.
IBaseFilter *nullFilter = NULL;
if (FAILED(CoCreateInstance(CLSID_NullRenderer,
NULL,
CLSCTX_INPROC_SERVER,
IID_IBaseFilter,
reinterpret_cast<void **>(&nullFilter)))) {
this->m_graph->Release();
this->m_graph = NULL;
return false;
}
if (FAILED(this->m_graph->AddFilter(nullFilter, L"NullFilter"))) {
this->m_graph->Release();
this->m_graph = NULL;
return false;
}
if (!this->connectFilters(this->m_graph, grabberFilter, nullFilter)) {
this->m_graph->Release();
this->m_graph = NULL;
return false;
}
// Set capture format
QList<int> streams = this->streams();
if (streams.isEmpty()) {
this->m_graph->Release();
this->m_graph = NULL;
return false;
}
MediaTypesList mediaTypes = this->listMediaTypes(webcamFilter);
if (mediaTypes.isEmpty()) {
this->m_graph->Release();
this->m_graph = NULL;
return false;
}
MediaTypePtr mediaType = streams[0] < mediaTypes.size()?
mediaTypes[streams[0]]:
mediaTypes.first();
if (FAILED(grabberPtr->SetMediaType(mediaType.data()))) {
this->m_graph->Release();
this->m_graph = NULL;
return false;
}
PinList pins = this->enumPins(webcamFilter, PINDIR_OUTPUT);
for (const PinPtr &pin: pins) {
IAMStreamConfig *pStreamConfig = NULL;
HRESULT hr =
pin->QueryInterface(IID_IAMStreamConfig,
reinterpret_cast<void **>(&pStreamConfig));
if (SUCCEEDED(hr))
pStreamConfig->SetFormat(mediaType.data());
if (pStreamConfig)
pStreamConfig->Release();
}
// Run the pipeline
IMediaControl *control = NULL;
if (FAILED(this->m_graph->QueryInterface(IID_IMediaControl,
reinterpret_cast<void **>(&control)))) {
this->m_graph->Release();
this->m_graph = NULL;
return false;
}
this->m_id = Ak::id();
AkCaps caps = this->capsFromMediaType(mediaType);
this->m_timeBase = AkFrac(caps.property("fps").toString()).invert();
if (FAILED(control->Run())) {
control->Release();
this->m_graph->Release();
this->m_graph = NULL;
return false;
}
control->Release();
this->m_localImageControls.clear();
this->m_localImageControls.clear();
return true;
}
void
WebCam::HandleCaptureMessage(WebCam* cam, HWND hwndDlg)
{
long lEventCode, lParam1, lParam2;
IMediaEventEx* pEvent = NULL;
HRESULT hr = cam->pGraph->QueryInterface(IID_IMediaEventEx, (void**)&pEvent);
if (FAILED(hr))
return;
int i=0;
while (pEvent && SUCCEEDED(pEvent->GetEvent(&lEventCode, &lParam1, &lParam2, 0)))
{
if (i++ > 10) break;
switch(lEventCode)
{
case EC_COMPLETE:
{
if (!cam || !cam->pGraph) break;
// Get the captured frame
cam->UploadImage();
// Turn off further notifications
//
IMediaEventEx* pMediaEvent = NULL;
cam->pGraph->QueryInterface(IID_IMediaEventEx, (void**)(&pMediaEvent));
if (pMediaEvent)
{
pMediaEvent->SetNotifyFlags(AM_MEDIAEVENT_NONOTIFY);
pMediaEvent->Release();
pMediaEvent = NULL;
}
// Turn off sample grabber
//
ISampleGrabber *pGrabber = NULL;
cam->pSampleGrabber->QueryInterface(IID_ISampleGrabber, (void**)(&pGrabber));
if (pGrabber)
{
pGrabber->SetBufferSamples(FALSE);
pGrabber->SetOneShot(FALSE);
pGrabber->Release();
pGrabber = NULL;
}
// Re-start the graph if window is visible,
// otherwise stop it
//
if (IsWindowVisible(hwndDlg))
{
cam->StopCapture();
cam->StartCapture();
}
else
{
cam->StopCapture();
}
return;
}
default:
{
return;
}
}
}
if (pEvent)
pEvent->Release();
pEvent = NULL;
}