/* This file is part of the KDE project. | |
Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies). | |
This library is free software: you can redistribute it and/or modify | |
it under the terms of the GNU Lesser General Public License as published by | |
the Free Software Foundation, either version 2.1 or 3 of the License. | |
This library is distributed in the hope that it will be useful, | |
but WITHOUT ANY WARRANTY; without even the implied warranty of | |
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
GNU Lesser General Public License for more details. | |
You should have received a copy of the GNU Lesser General Public License | |
along with this library. If not, see <http://www.gnu.org/licenses/>. | |
*/ | |
#include "iodevicereader.h" | |
#include "qasyncreader.h" | |
#include "mediagraph.h" | |
#include <phonon/streaminterface.h> | |
#include <dshow.h> | |
#include <initguid.h> | |
QT_BEGIN_NAMESPACE | |
#ifndef QT_NO_PHONON_ABSTRACTMEDIASTREAM | |
namespace Phonon | |
{ | |
namespace DS9 | |
{ | |
//these mediatypes define a stream, its type will be autodetected by DirectShow | |
static QVector<AM_MEDIA_TYPE> getMediaTypes() | |
{ | |
//the order here is important because otherwise, | |
//directshow might not be able to detect the stream type correctly | |
AM_MEDIA_TYPE mt = { MEDIATYPE_Stream, MEDIASUBTYPE_Avi, TRUE, FALSE, 1, GUID_NULL, 0, 0, 0}; | |
QVector<AM_MEDIA_TYPE> ret; | |
//AVI stream | |
ret << mt; | |
//WAVE stream | |
mt.subtype = MEDIASUBTYPE_WAVE; | |
ret << mt; | |
//normal auto-detect stream (must be at the end!) | |
mt.subtype = MEDIASUBTYPE_NULL; | |
ret << mt; | |
return ret; | |
} | |
class StreamReader : public QAsyncReader, public Phonon::StreamInterface | |
{ | |
public: | |
StreamReader(QBaseFilter *parent, const Phonon::MediaSource &source, const MediaGraph *mg) : | |
QAsyncReader(parent, getMediaTypes()), | |
m_seekable(false), m_pos(0), m_size(-1), m_mediaGraph(mg) | |
{ | |
connectToSource(source); | |
} | |
//for Phonon::StreamInterface | |
void writeData(const QByteArray &data) | |
{ | |
m_pos += data.size(); | |
m_buffer += data; | |
} | |
void endOfData() | |
{ | |
} | |
void setStreamSize(qint64 newSize) | |
{ | |
QMutexLocker locker(&m_mutex); | |
m_size = newSize; | |
} | |
void setStreamSeekable(bool s) | |
{ | |
QMutexLocker locker(&m_mutex); | |
m_seekable = s; | |
} | |
//virtual pure members | |
//implementation from IAsyncReader | |
STDMETHODIMP Length(LONGLONG *total, LONGLONG *available) | |
{ | |
QMutexLocker locker(&m_mutex); | |
if (total) { | |
*total = m_size; | |
} | |
if (available) { | |
*available = m_size; | |
} | |
return S_OK; | |
} | |
HRESULT read(LONGLONG pos, LONG length, BYTE *buffer, LONG *actual) | |
{ | |
Q_ASSERT(!m_mutex.tryLock()); | |
if (m_mediaGraph->isStopping()) { | |
return VFW_E_WRONG_STATE; | |
} | |
if(m_size != 1 && pos + length > m_size) { | |
//it tries to read outside of the boundaries | |
return E_FAIL; | |
} | |
if (m_pos - m_buffer.size() != pos) { | |
if (!m_seekable) { | |
return S_FALSE; | |
} | |
m_pos = pos; | |
seekStream(pos); | |
m_buffer.clear(); | |
} | |
int oldSize = m_buffer.size(); | |
while (m_buffer.size() < int(length)) { | |
needData(); | |
if (oldSize == m_buffer.size()) { | |
break; //we didn't get any data | |
} | |
oldSize = m_buffer.size(); | |
} | |
int bytesRead = qMin(m_buffer.size(), int(length)); | |
qMemCopy(buffer, m_buffer.data(), bytesRead); | |
//truncate the buffer | |
m_buffer = m_buffer.mid(bytesRead); | |
if (actual) { | |
*actual = bytesRead; //initialization | |
} | |
return bytesRead == length ? S_OK : S_FALSE; | |
} | |
public: | |
//for Phonon::StreamInterface | |
QByteArray m_buffer; | |
bool m_seekable; | |
qint64 m_pos; | |
qint64 m_size; | |
const MediaGraph *m_mediaGraph; | |
}; | |
IODeviceReader::IODeviceReader(const Phonon::MediaSource &source, const MediaGraph *mg) : | |
QBaseFilter(CLSID_NULL) | |
{ | |
//create the output pin | |
m_streamReader = new StreamReader(this, source, mg); | |
} | |
IODeviceReader::~IODeviceReader() | |
{ | |
} | |
} | |
} | |
#endif //QT_NO_PHONON_ABSTRACTMEDIASTREAM | |
QT_END_NAMESPACE |