VideoSampleBufferCompressor.h [plain text]
#pragma once
#if ENABLE(MEDIA_STREAM) && USE(AVFOUNDATION)
#include <CoreMedia/CoreMedia.h>
#include <VideoToolbox/VTErrors.h>
typedef struct opaqueCMSampleBuffer *CMSampleBufferRef;
typedef struct OpaqueVTCompressionSession *VTCompressionSessionRef;
namespace WebCore {
class VideoSampleBufferCompressor {
WTF_MAKE_FAST_ALLOCATED;
public:
static std::unique_ptr<VideoSampleBufferCompressor> create(CMVideoCodecType, CMBufferQueueTriggerCallback, void* callbackObject);
~VideoSampleBufferCompressor();
void setBitsPerSecond(unsigned);
void finish();
void addSampleBuffer(CMSampleBufferRef);
CMSampleBufferRef getOutputSampleBuffer();
RetainPtr<CMSampleBufferRef> takeOutputSampleBuffer();
unsigned bitRate() const;
private:
explicit VideoSampleBufferCompressor(CMVideoCodecType);
bool initialize(CMBufferQueueTriggerCallback, void* callbackObject);
void processSampleBuffer(CMSampleBufferRef);
bool initCompressionSession(CMVideoFormatDescriptionRef);
static void videoCompressionCallback(void *refCon, void*, OSStatus, VTEncodeInfoFlags, CMSampleBufferRef);
dispatch_queue_t m_serialDispatchQueue;
CMVideoCodecType m_outputCodecType { kCMVideoCodecType_H264 };
RetainPtr<CMBufferQueueRef> m_outputBufferQueue;
RetainPtr<VTCompressionSessionRef> m_vtSession;
bool m_isEncoding { false };
float m_maxKeyFrameIntervalDuration { 2.0 };
unsigned m_expectedFrameRate { 30 };
Optional<unsigned> m_outputBitRate;
};
}
#endif