recordctl.cpp
上传用户:zhongxx05
上传日期:2007-06-06
资源大小:33641k
文件大小:8k
- /* ***** BEGIN LICENSE BLOCK *****
- * Version: RCSL 1.0/RPSL 1.0
- *
- * Portions Copyright (c) 1995-2002 RealNetworks, Inc. All Rights Reserved.
- *
- * The contents of this file, and the files included with this file, are
- * subject to the current version of the RealNetworks Public Source License
- * Version 1.0 (the "RPSL") available at
- * http://www.helixcommunity.org/content/rpsl unless you have licensed
- * the file under the RealNetworks Community Source License Version 1.0
- * (the "RCSL") available at http://www.helixcommunity.org/content/rcsl,
- * in which case the RCSL will apply. You may also obtain the license terms
- * directly from RealNetworks. You may not use this file except in
- * compliance with the RPSL or, if you have a valid RCSL with RealNetworks
- * applicable to this file, the RCSL. Please see the applicable RPSL or
- * RCSL for the rights, obligations and limitations governing use of the
- * contents of the file.
- *
- * This file is part of the Helix DNA Technology. RealNetworks is the
- * developer of the Original Code and owns the copyrights in the portions
- * it created.
- *
- * This file, and the files included with this file, is distributed and made
- * available on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
- * EXPRESS OR IMPLIED, AND REALNETWORKS HEREBY DISCLAIMS ALL SUCH WARRANTIES,
- * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, FITNESS
- * FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
- *
- * Technology Compatibility Kit Test Suite(s) Location:
- * http://www.helixcommunity.org/content/tck
- *
- * Contributor(s):
- *
- * ***** END LICENSE BLOCK ***** */
- #include "recordctl.h"
- #include "ihxpckts.h"
- #include "hxrecord.h"
- #include "hxheap.h"
- #ifdef _DEBUG
- #undef HX_THIS_FILE
- static const char HX_THIS_FILE[] = __FILE__;
- #endif
- HXRecordControl::HXRecordControl(IUnknown* pUnkPlayer, IUnknown* pUnkSource) :
- m_lRefCount(0),
- m_pRecordSource(NULL),
- m_pRecordService(NULL),
- m_bCanGetPackets(FALSE)
- {
- SPIHXRecordManager spRecordManager = pUnkPlayer;
- if(spRecordManager.IsValid())
- spRecordManager->GetRecordService(m_pRecordService);
- if(m_pRecordService)
- m_pRecordService->CreateRecordSource(pUnkSource, m_pRecordSource);
- if(m_pRecordSource)
- {
- if(m_pRecordSource->SetFormatResponse(this) == HXR_OK)
- m_bCanGetPackets = TRUE;
- }
- }
- HXRecordControl::~HXRecordControl()
- {
- Cleanup();
- }
- STDMETHODIMP
- HXRecordControl::QueryInterface(REFIID riid, void** ppvObj)
- {
- QInterfaceList qiList[] =
- {
- { GET_IIDHANDLE(IID_IHXFormatResponse), (IHXFormatResponse*)this },
- { GET_IIDHANDLE(IID_IUnknown), (IUnknown*)(IHXFormatResponse*)this },
- };
-
- return ::QIFind(qiList, QILISTSIZE(qiList), riid, ppvObj);
- }
- STDMETHODIMP_(ULONG32)
- HXRecordControl::AddRef()
- {
- return InterlockedIncrement(&m_lRefCount);
- }
- STDMETHODIMP_(ULONG32)
- HXRecordControl::Release()
- {
- if (InterlockedDecrement(&m_lRefCount) > 0)
- {
- return m_lRefCount;
- }
- delete this;
- return 0;
- }
- void
- HXRecordControl::Cleanup()
- {
- for(UINT16 nStream = 0; nStream < m_PendingGetPackets.GetSize(); nStream++)
- {
- IHXPacket* pPacket = (IHXPacket*)m_PendingGetPackets.GetAt(nStream);
- m_PendingGetPackets.SetAt(nStream, NULL);
- HX_RELEASE(pPacket);
- }
- if(m_pRecordService)
- m_pRecordService->CloseRecordSource(m_pRecordSource);
- HX_RELEASE(m_pRecordService);
- HX_RELEASE(m_pRecordSource);
- while(!m_PendingPutPackets.IsEmpty())
- {
- PendingPutPacket* pPutPacket = (PendingPutPacket*)m_PendingPutPackets.RemoveHead();
- HX_ASSERT(pPutPacket);
- HX_RELEASE(pPutPacket->pPacket);
- HX_DELETE(pPutPacket);
- }
- }
- STDMETHODIMP
- HXRecordControl::PacketReady(HX_RESULT status, IHXPacket* pPacket)
- {
- if(pPacket)
- {
- pPacket->AddRef();
- if(pPacket->GetStreamNumber() < m_PendingGetPackets.GetSize())
- HX_ASSERT(!m_PendingGetPackets.GetAt(pPacket->GetStreamNumber()));
- m_PendingGetPackets.SetAt(pPacket->GetStreamNumber(), pPacket);
- }
- return HXR_OK;
- }
- HX_RESULT
- HXRecordControl::Seek(ULONG32 seekTime)
- {
- if(!m_pRecordSource)
- return HXR_NOT_INITIALIZED;
- HX_RESULT nResult = HXR_FAILED;
- if(m_bCanGetPackets)
- {
- nResult = m_pRecordSource->Seek(seekTime);
- if(nResult == HXR_OK)
- {
- for(UINT16 nStream = 0; nStream < m_PendingGetPackets.GetSize(); nStream++)
- {
- IHXPacket* pPacket = (IHXPacket*)m_PendingGetPackets.GetAt(nStream);
- m_PendingGetPackets.SetAt(nStream, NULL);
- HX_RELEASE(pPacket);
-
- // If seek succeeded the first GetPacket() for each stream should not fail.
- // In case of metafiles though no packets should be send to renderers again.
- // Therefore GetPacket() fails in case of metafiles. SB.
- m_pRecordSource->GetPacket(nStream);
- }
- }
- }
- if(nResult != HXR_OK)
- {
- m_pRecordSource->Flush();
- for(UINT16 nStream = 0; nStream < m_PendingGetPackets.GetSize(); nStream++)
- {
- IHXPacket* pPacket = (IHXPacket*)m_PendingGetPackets.GetAt(nStream);
- m_PendingGetPackets.SetAt(nStream, NULL);
- HX_RELEASE(pPacket);
- }
- }
- return nResult;
- }
- HX_RESULT
- HXRecordControl::GetPacket(UINT16 usStreamNumber, IHXPacket*& pPacket)
- {
- pPacket = NULL;
- if(!m_pRecordSource || !m_bCanGetPackets)
- return HXR_FAILED;
- if (m_PendingGetPackets.IsEmpty())
- return HXR_NO_DATA;
-
- HX_RESULT nResult = HXR_OK;
- pPacket = (IHXPacket*)m_PendingGetPackets.GetAt(usStreamNumber);
- if(!pPacket)
- {
- nResult = m_pRecordSource->GetPacket(usStreamNumber);
- if(nResult == HXR_OK)
- {
- pPacket = (IHXPacket*)m_PendingGetPackets.GetAt(usStreamNumber);
- if(!pPacket)
- nResult = HXR_WOULD_BLOCK;
- }
- }
- m_PendingGetPackets.SetAt(usStreamNumber, NULL);
- return nResult;
- }
- HX_RESULT
- HXRecordControl::OnFileHeader(IHXValues* pValues)
- {
- UINT32 nStreamCount = 0;
- if(pValues)
- pValues->GetPropertyULONG32("StreamCount", nStreamCount);
- if(nStreamCount)
- {
- m_PendingGetPackets.SetSize(nStreamCount);
- for(UINT16 nStream = 0; nStream < nStreamCount; nStream++)
- m_PendingGetPackets.SetAt(nStream, NULL);
- }
- HX_RESULT nResult = HXR_FAILED;
- if(m_pRecordSource)
- nResult = m_pRecordSource->OnFileHeader(pValues);
- if(nResult != HXR_OK && nResult != HXR_RECORD)
- Cleanup();
- return nResult;
- }
- HX_RESULT
- HXRecordControl::OnStreamHeader(IHXValues* pValues)
- {
- HX_RESULT nResult = HXR_FAILED;
- if(m_pRecordSource)
- nResult = m_pRecordSource->OnStreamHeader(pValues);
- if(nResult != HXR_OK && nResult != HXR_RECORD)
- Cleanup();
- return nResult;
- }
- HX_RESULT
- HXRecordControl::OnPacket(IHXPacket* pPacket, INT32 nTimeOffset)
- {
- HX_RESULT nResult = HXR_FAILED;
- if(pPacket && m_pRecordSource)
- {
- if(m_PendingPutPackets.GetCount())
- nResult = HXR_RETRY;
- else
- nResult = m_pRecordSource->OnPacket(pPacket, nTimeOffset);
- if(nResult == HXR_RETRY)
- {
- PendingPutPacket* pPutPacket = new PendingPutPacket;
- if(!pPutPacket)
- return HXR_OUTOFMEMORY;
- pPutPacket->pPacket = pPacket;
- pPutPacket->pPacket->AddRef();
- pPutPacket->nTimeOffset = nTimeOffset;
- m_PendingPutPackets.AddTail(pPutPacket);
- nResult = HXR_OK;
- }
- }
- if(nResult != HXR_OK)
- Cleanup();
- return nResult;
- }
- BOOL
- HXRecordControl::CanAcceptPackets()
- {
- if(!m_pRecordSource)
- return FALSE;
- while(!m_PendingPutPackets.IsEmpty())
- {
- PendingPutPacket* pPutPacket = (PendingPutPacket*)m_PendingPutPackets.GetHead();
- HX_ASSERT(pPutPacket);
- if(m_pRecordSource->OnPacket(pPutPacket->pPacket, pPutPacket->nTimeOffset) == HXR_RETRY)
- return FALSE;
- HX_RELEASE(pPutPacket->pPacket);
- HX_DELETE(pPutPacket);
- m_PendingPutPackets.RemoveHead();
- }
- return TRUE;
- }
- void
- HXRecordControl::SetSource(IUnknown* pUnkSource)
- {
- if(m_pRecordSource)
- m_pRecordSource->SetSource(pUnkSource);
- }
- void
- HXRecordControl::OnEndOfPackets()
- {
- if(m_pRecordSource)
- m_pRecordSource->OnEndOfPackets();
- }