ImageGStreamerCairo.cpp [plain text]
#include "config.h"
#include "ImageGStreamer.h"
#if ENABLE(VIDEO) && USE(GSTREAMER)
#include "GStreamerUtilities.h"
#include <cairo.h>
#include <gst/gst.h>
#include <gst/video/gstvideometa.h>
using namespace std;
using namespace WebCore;
ImageGStreamer::ImageGStreamer(GstSample* sample)
{
GstCaps* caps = gst_sample_get_caps(sample);
GstVideoInfo videoInfo;
gst_video_info_init(&videoInfo);
if (!gst_video_info_from_caps(&videoInfo, caps))
return;
ASSERT(GST_VIDEO_INFO_N_PLANES(&videoInfo) == 1);
GstBuffer* buffer = gst_sample_get_buffer(sample);
if (!gst_video_frame_map(&m_videoFrame, &videoInfo, buffer, GST_MAP_READ))
return;
unsigned char* bufferData = reinterpret_cast<unsigned char*>(GST_VIDEO_FRAME_PLANE_DATA(&m_videoFrame, 0));
int stride = GST_VIDEO_FRAME_PLANE_STRIDE(&m_videoFrame, 0);
int width = GST_VIDEO_FRAME_WIDTH(&m_videoFrame);
int height = GST_VIDEO_FRAME_HEIGHT(&m_videoFrame);
RefPtr<cairo_surface_t> surface;
cairo_format_t cairoFormat;
#if G_BYTE_ORDER == G_LITTLE_ENDIAN
cairoFormat = (GST_VIDEO_FRAME_FORMAT(&m_videoFrame) == GST_VIDEO_FORMAT_BGRA) ? CAIRO_FORMAT_ARGB32 : CAIRO_FORMAT_RGB24;
#else
cairoFormat = (GST_VIDEO_FRAME_FORMAT(&m_videoFrame) == GST_VIDEO_FORMAT_ARGB) ? CAIRO_FORMAT_ARGB32 : CAIRO_FORMAT_RGB24;
#endif
if (cairoFormat == CAIRO_FORMAT_ARGB32) {
unsigned char* surfaceData = static_cast<unsigned char*>(fastMalloc(height * stride));
unsigned char* surfacePixel = surfaceData;
for (int x = 0; x < width; x++) {
for (int y = 0; y < height; y++) {
#if G_BYTE_ORDER == G_LITTLE_ENDIAN
unsigned short alpha = bufferData[3];
surfacePixel[0] = (bufferData[0] * alpha + 128) / 255;
surfacePixel[1] = (bufferData[1] * alpha + 128) / 255;
surfacePixel[2] = (bufferData[2] * alpha + 128) / 255;
surfacePixel[3] = alpha;
#else
unsigned short alpha = bufferData[0];
surfacePixel[0] = alpha;
surfacePixel[1] = (bufferData[1] * alpha + 128) / 255;
surfacePixel[2] = (bufferData[2] * alpha + 128) / 255;
surfacePixel[3] = (bufferData[3] * alpha + 128) / 255;
#endif
bufferData += 4;
surfacePixel += 4;
}
}
surface = adoptRef(cairo_image_surface_create_for_data(surfaceData, cairoFormat, width, height, stride));
static cairo_user_data_key_t s_surfaceDataKey;
cairo_surface_set_user_data(surface.get(), &s_surfaceDataKey, surfaceData, [](void* data) { fastFree(data); });
} else
surface = adoptRef(cairo_image_surface_create_for_data(bufferData, cairoFormat, width, height, stride));
ASSERT(cairo_surface_status(surface.get()) == CAIRO_STATUS_SUCCESS);
m_image = BitmapImage::create(WTFMove(surface));
if (GstVideoCropMeta* cropMeta = gst_buffer_get_video_crop_meta(buffer))
setCropRect(FloatRect(cropMeta->x, cropMeta->y, cropMeta->width, cropMeta->height));
}
ImageGStreamer::~ImageGStreamer()
{
if (m_image)
m_image = nullptr;
gst_video_frame_unmap(&m_videoFrame);
}
#endif // USE(GSTREAMER)