akvirtualcamera/dshow/VirtualCamera/src/mediasample2.cpp
Gonzalo Exequiel Pedone 84d630e422
Windows plugin working, 'update' command still needs elevated privileges
but all other commands can be run as normal user.
2020-10-08 20:07:51 -03:00

116 lines
3.8 KiB
C++

/* akvirtualcamera, virtual camera for Mac and Windows.
* Copyright (C) 2020 Gonzalo Exequiel Pedone
*
* akvirtualcamera is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* akvirtualcamera 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with akvirtualcamera. If not, see <http://www.gnu.org/licenses/>.
*
* Web-Site: http://webcamoid.github.io/
*/
#include "mediasample2.h"
#include "PlatformUtils/src/utils.h"
#include "VCamUtils/src/utils.h"
namespace AkVCam
{
class MediaSample2Private
{
public:
DWORD m_data {0};
DWORD m_typeSpecificFlags {0};
DWORD m_sampleFlags {0};
DWORD m_streamId {0};
AM_MEDIA_TYPE *m_mediaType {nullptr};
};
}
AkVCam::MediaSample2::MediaSample2(IMemAllocator *memAllocator,
LONG bufferSize,
LONG align,
LONG prefix):
MediaSample(memAllocator, bufferSize, align, prefix)
{
this->d = new MediaSample2Private;
}
AkVCam::MediaSample2::~MediaSample2()
{
deleteMediaType(&this->d->m_mediaType);
delete this->d;
}
HRESULT AkVCam::MediaSample2::SetProperties(DWORD cbProperties,
const BYTE *pbProperties)
{
AkLogFunction();
if (cbProperties < sizeof(AM_SAMPLE2_PROPERTIES))
return E_INVALIDARG;
if (!pbProperties)
return E_POINTER;
auto properties =
reinterpret_cast<const AM_SAMPLE2_PROPERTIES *>(pbProperties);
if (properties->pbBuffer || properties->cbBuffer)
return E_INVALIDARG;
this->d->m_data = properties->cbData;
this->d->m_typeSpecificFlags = properties->dwTypeSpecificFlags;
this->d->m_sampleFlags = properties->dwSampleFlags;
this->SetDiscontinuity(this->d->m_sampleFlags & AM_SAMPLE_DATADISCONTINUITY);
this->SetSyncPoint(this->d->m_sampleFlags & AM_SAMPLE_SPLICEPOINT);
this->SetPreroll(this->d->m_sampleFlags & AM_SAMPLE_PREROLL);
this->SetActualDataLength(properties->lActual);
auto start = properties->tStart;
auto stop = properties->tStop;
this->SetTime(&start, &stop);
this->SetMediaTime(&start, &stop);
this->d->m_streamId = properties->dwStreamId;
deleteMediaType(&this->d->m_mediaType);
this->d->m_mediaType = createMediaType(properties->pMediaType);
this->SetMediaType(properties->pMediaType);
return S_OK;
}
HRESULT AkVCam::MediaSample2::GetProperties(DWORD cbProperties,
BYTE *pbProperties)
{
AkLogFunction();
if (cbProperties < sizeof(AM_SAMPLE2_PROPERTIES))
return E_INVALIDARG;
if (!pbProperties)
return E_POINTER;
auto properties =
reinterpret_cast<AM_SAMPLE2_PROPERTIES *>(pbProperties);
properties->cbData = this->d->m_data;
properties->dwTypeSpecificFlags = this->d->m_typeSpecificFlags;
properties->dwSampleFlags = this->d->m_sampleFlags;
properties->lActual = this->GetActualDataLength();
this->GetTime(&properties->tStart, &properties->tStop);
properties->dwStreamId = this->d->m_streamId;
deleteMediaType(&this->d->m_mediaType);
this->GetMediaType(&this->d->m_mediaType);
properties->pMediaType = this->d->m_mediaType;
this->GetPointer(&properties->pbBuffer);
properties->cbBuffer = this->GetSize();
return S_OK;
}