135 lines
5.1 KiB
C++
135 lines
5.1 KiB
C++
//==========================================================================;
|
|
//
|
|
// dat2xds.h : Declaration of the custom composition class for gluing sbe source to vmr
|
|
// TODO: Need to update header file and change classids
|
|
// Copyright (c) Microsoft Corporation 1999.
|
|
//
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
|
|
|
|
#ifndef DAT2XDS_H
|
|
#define DAT2XDS_H
|
|
|
|
#pragma once
|
|
#include <uuids.h>
|
|
#include "bdamedia.h"
|
|
#include "MSVidEncoder.h"
|
|
#include "resource.h" // main symbols
|
|
#include <winerror.h>
|
|
#include <algorithm>
|
|
#include <compimpl.h>
|
|
#include <seg.h>
|
|
#include <objectwithsiteimplsec.h>
|
|
|
|
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CDat2XDSComp
|
|
class ATL_NO_VTABLE __declspec(uuid("0429EC6E-1144-4bed-B88B-2FB9899A4A3D")) CDat2XDSComp :
|
|
public CComObjectRootEx<CComSingleThreadModel>,
|
|
public CComCoClass<CDat2XDSComp, &__uuidof(CDat2XDSComp)>,
|
|
public IObjectWithSiteImplSec<CDat2XDSComp>,
|
|
public IMSVidCompositionSegmentImpl<CDat2XDSComp>
|
|
{
|
|
public:
|
|
CDat2XDSComp() {}
|
|
virtual ~CDat2XDSComp() {}
|
|
|
|
REGISTER_NONAUTOMATION_OBJECT(IDS_PROJNAME,
|
|
IDS_REG_DAT2XDSCOMP_DESC,
|
|
LIBID_MSVidCtlLib,
|
|
__uuidof(CDat2XDSComp));
|
|
|
|
DECLARE_PROTECT_FINAL_CONSTRUCT()
|
|
|
|
BEGIN_COM_MAP(CDat2XDSComp)
|
|
COM_INTERFACE_ENTRY(IMSVidCompositionSegment)
|
|
COM_INTERFACE_ENTRY(IMSVidGraphSegment)
|
|
COM_INTERFACE_ENTRY(IObjectWithSite)
|
|
COM_INTERFACE_ENTRY(IPersist)
|
|
END_COM_MAP()
|
|
|
|
// IMSVidComposition
|
|
public:
|
|
// IMSVidGraphSegment
|
|
// IMSVidCompositionSegment
|
|
STDMETHOD(Compose)(IMSVidGraphSegment * upstream, IMSVidGraphSegment * downstream)
|
|
{
|
|
if (m_fComposed) {
|
|
return NOERROR;
|
|
}
|
|
ASSERT(m_pGraph);
|
|
try {
|
|
VWGraphSegment up(upstream);
|
|
ASSERT(up.Graph() == m_pGraph);
|
|
VWGraphSegment down(downstream);
|
|
ASSERT(down.Graph() == m_pGraph);
|
|
if (upstream == downstream) {
|
|
return Error(IDS_CANT_COMPOSE_WITH_SELF, __uuidof(IMSVidCompositionSegment), E_INVALIDARG);
|
|
}
|
|
if (up.begin() == up.end()) {
|
|
TRACELM(TRACE_ERROR, "CComposition::Compose() can't compose empty up segment");
|
|
return NOERROR;
|
|
}
|
|
if (down.begin() == down.end()) {
|
|
TRACELM(TRACE_ERROR, "CComposition::Compose() can't compose empty down segment");
|
|
// this is not an error, for example, CA is an empty segment.
|
|
return NOERROR;
|
|
}
|
|
bool vidFound = false;
|
|
// VMR has a bug so we need to connect the video before the cc or no cc will be displayed
|
|
DSMediaType mtVideo(MEDIATYPE_Video, MEDIASUBTYPE_MPEG2_VIDEO, FORMAT_MPEG2Video);
|
|
for (VWGraphSegment::iterator iStart = up.begin(); iStart != up.end(); ++iStart) {
|
|
ASSERT((*iStart).GetGraph() == m_pGraph);
|
|
if(!vidFound){
|
|
for(DSFilter::iterator i = (*iStart).begin(); i != (*iStart).end(); ++i){
|
|
if((*i).GetDirection() == DOWNSTREAM){
|
|
for(DSPin::iterator p = (*i).begin(); p != (*i).end(); ++p){
|
|
if((*p) == mtVideo){
|
|
for (VWGraphSegment::iterator iStop = down.begin(); iStop != down.end(); ++ iStop) {
|
|
ASSERT((*iStop).GetGraph() == m_pGraph);
|
|
DSFilter pStop(*iStop);
|
|
(*i).IntelligentConnect(pStop, m_Filters);
|
|
}
|
|
vidFound = true;
|
|
iStart = up.begin();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else{
|
|
for (VWGraphSegment::iterator iStop = down.begin(); iStop != down.end(); ++ iStop) {
|
|
ASSERT((*iStop).GetGraph() == m_pGraph);
|
|
DSFilter pStart(*iStart);
|
|
DSFilter pStop(*iStop);
|
|
HRESULT hr = m_pGraph.Connect(pStart, pStop, m_Filters);
|
|
if (SUCCEEDED(hr)) {
|
|
m_Segments.push_back(up);
|
|
m_Segments.push_back(down);
|
|
m_pDown = m_Segments.end();
|
|
--m_pDown;
|
|
m_pUp = m_pDown;
|
|
--m_pUp;
|
|
m_fComposed = true;
|
|
return NOERROR;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
TRACELM(TRACE_ERROR, "CComposition::Compose() compose didn't connect anything");
|
|
return VFW_E_NO_DECOMPRESSOR;
|
|
} catch (ComException &e) {
|
|
return e;
|
|
} catch (...) {
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
}
|
|
|
|
|
|
};
|
|
|
|
#endif // Dat2XDS_H
|
|
// end of file - Dat2XDS.h
|