gecko/gfx/layers/ipc/SharedRGBImage.cpp
Edwin Flores d1720c9ea8 Bug 803394 - Add image type for RGB images in shared memory r=nical
--HG--
extra : rebase_source : 3be9bbbcf827c0796d7b0a39724f3372ac080632
2012-12-13 11:47:28 +13:00

126 lines
3.0 KiB
C++

/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "ImageContainerChild.h"
#include "ShadowLayers.h"
#include "SharedRGBImage.h"
#include "Shmem.h"
// Just big enough for a 1080p RGBA32 frame
#define MAX_FRAME_SIZE (16 * 1024 * 1024)
namespace mozilla {
namespace layers {
SharedRGBImage::SharedRGBImage(ImageContainerChild *aImageContainerChild) :
Image(nullptr, SHARED_RGB),
mSize(0, 0),
mImageContainerChild(aImageContainerChild),
mAllocated(false),
mShmem(new ipc::Shmem())
{
mImageContainerChild->AddRef();
}
SharedRGBImage::~SharedRGBImage()
{
mImageContainerChild->DeallocShmemAsync(*mShmem);
mImageContainerChild->Release();
delete mShmem;
}
already_AddRefed<SharedRGBImage>
SharedRGBImage::Create(ImageContainer *aImageContainer,
nsIntSize aSize,
gfxImageFormat aImageFormat)
{
NS_ASSERTION(aImageFormat == gfxASurface::ImageFormatARGB32 ||
aImageFormat == gfxASurface::ImageFormatRGB24 ||
aImageFormat == gfxASurface::ImageFormatRGB16_565,
"RGB formats supported only");
if (!aImageContainer) {
NS_WARNING("No ImageContainer to allocate SharedRGBImage");
return nullptr;
}
ImageFormat format = SHARED_RGB;
nsRefPtr<Image> image = aImageContainer->CreateImage(&format, 1);
if (!image) {
NS_WARNING("Failed to create SharedRGBImage");
return nullptr;
}
nsRefPtr<SharedRGBImage> rgbImage = static_cast<SharedRGBImage*>(image.get());
rgbImage->mSize = gfxIntSize(aSize.width, aSize.height);
rgbImage->mImageFormat = aImageFormat;
if (!rgbImage->AllocateBuffer(aSize, aImageFormat)) {
NS_WARNING("Failed to allocate shared memory for SharedRGBImage");
return nullptr;
}
return rgbImage.forget();
}
uint8_t *
SharedRGBImage::GetBuffer()
{
return mShmem->get<uint8_t>();
}
size_t
SharedRGBImage::GetBufferSize()
{
return mSize.width * mSize.height * gfxASurface::BytesPerPixel(mImageFormat);
}
gfxIntSize
SharedRGBImage::GetSize()
{
return mSize;
}
bool
SharedRGBImage::AllocateBuffer(nsIntSize aSize, gfxImageFormat aImageFormat)
{
if (mAllocated) {
NS_WARNING("Already allocated shmem");
return false;
}
size_t size = GetBufferSize();
if (size == 0 || size > MAX_FRAME_SIZE) {
NS_WARNING("Invalid frame size");
}
if (mImageContainerChild->AllocUnsafeShmemSync(size, OptimalShmemType(), mShmem)) {
mAllocated = true;
}
return mAllocated;
}
already_AddRefed<gfxASurface>
SharedRGBImage::GetAsSurface()
{
return nullptr;
}
SharedImage *
SharedRGBImage::ToSharedImage()
{
if (!mAllocated) {
return nullptr;
}
return new SharedImage(RGBImage(*mShmem,
nsIntRect(0, 0, mSize.width, mSize.height),
mImageFormat));
}
} // namespace layers
} // namespace mozilla