зеркало из https://github.com/mozilla/pjs.git
306 строки
8.7 KiB
C++
306 строки
8.7 KiB
C++
/* -*- Mode: C++; tab-width: 20; indent-tabs-mode: nil; c-basic-offset: 4 -*-
|
|
* ***** BEGIN LICENSE BLOCK *****
|
|
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
|
|
*
|
|
* The contents of this file are subject to the Mozilla Public License Version
|
|
* 1.1 (the "License"); you may not use this file except in compliance with
|
|
* the License. You may obtain a copy of the License at
|
|
* http://www.mozilla.org/MPL/
|
|
*
|
|
* Software distributed under the License is distributed on an "AS IS" basis,
|
|
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
|
|
* for the specific language governing rights and limitations under the
|
|
* License.
|
|
*
|
|
* The Original Code is Mozilla Corporation code.
|
|
*
|
|
* The Initial Developer of the Original Code is Mozilla Foundation.
|
|
* Portions created by the Initial Developer are Copyright (C) 2009
|
|
* the Initial Developer. All Rights Reserved.
|
|
*
|
|
* Contributor(s):
|
|
* Robert O'Callahan <robert@ocallahan.org>
|
|
*
|
|
* Alternatively, the contents of this file may be used under the terms of
|
|
* either the GNU General Public License Version 2 or later (the "GPL"), or
|
|
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
|
|
* in which case the provisions of the GPL or the LGPL are applicable instead
|
|
* of those above. If you wish to allow use of your version of this file only
|
|
* under the terms of either the GPL or the LGPL, and not to allow others to
|
|
* use your version of this file under the terms of the MPL, indicate your
|
|
* decision by deleting the provisions above and replace them with the notice
|
|
* and other provisions required by the GPL or the LGPL. If you do not delete
|
|
* the provisions above, a recipient may use your version of this file under
|
|
* the terms of any one of the MPL, the GPL or the LGPL.
|
|
*
|
|
* ***** END LICENSE BLOCK ***** */
|
|
|
|
#include "mozilla/Monitor.h"
|
|
|
|
#include "ImageLayers.h"
|
|
#include "BasicLayers.h"
|
|
#include "gfxImageSurface.h"
|
|
|
|
#ifdef XP_MACOSX
|
|
#include "gfxQuartzImageSurface.h"
|
|
#endif
|
|
|
|
#include "cairo.h"
|
|
|
|
#include "yuv_convert.h"
|
|
|
|
using mozilla::Monitor;
|
|
|
|
namespace mozilla {
|
|
namespace layers {
|
|
|
|
/**
|
|
* All our images can yield up a cairo surface and their size.
|
|
*/
|
|
class BasicImageImplData {
|
|
public:
|
|
/**
|
|
* This must be called on the main thread.
|
|
*/
|
|
virtual already_AddRefed<gfxASurface> GetAsSurface() = 0;
|
|
|
|
gfxIntSize GetSize() { return mSize; }
|
|
|
|
protected:
|
|
gfxIntSize mSize;
|
|
};
|
|
|
|
/**
|
|
* Since BasicLayers only paint on the main thread, handling a CairoImage
|
|
* is extremely simple. We just hang on to a reference to the surface and
|
|
* return that surface when BasicImageLayer::Paint asks for it via
|
|
* BasicImageContainer::GetAsSurface.
|
|
*/
|
|
class BasicCairoImage : public CairoImage, public BasicImageImplData {
|
|
public:
|
|
BasicCairoImage() : CairoImage(static_cast<BasicImageImplData*>(this)) {}
|
|
|
|
virtual void SetData(const Data& aData)
|
|
{
|
|
mSurface = aData.mSurface;
|
|
mSize = aData.mSize;
|
|
}
|
|
|
|
virtual already_AddRefed<gfxASurface> GetAsSurface()
|
|
{
|
|
NS_ASSERTION(NS_IsMainThread(), "Must be main thread");
|
|
nsRefPtr<gfxASurface> surface = mSurface.get();
|
|
return surface.forget();
|
|
}
|
|
|
|
protected:
|
|
nsCountedRef<nsMainThreadSurfaceRef> mSurface;
|
|
};
|
|
|
|
/**
|
|
* We handle YCbCr by converting to RGB when the image is initialized
|
|
* (which should be done off the main thread). The RGB results are stored
|
|
* in a memory buffer and converted to a cairo surface lazily.
|
|
*/
|
|
class BasicPlanarYCbCrImage : public PlanarYCbCrImage, public BasicImageImplData {
|
|
public:
|
|
BasicPlanarYCbCrImage() :
|
|
PlanarYCbCrImage(static_cast<BasicImageImplData*>(this))
|
|
{}
|
|
|
|
virtual void SetData(const Data& aData);
|
|
|
|
virtual already_AddRefed<gfxASurface> GetAsSurface();
|
|
|
|
protected:
|
|
nsAutoArrayPtr<PRUint8> mBuffer;
|
|
nsCountedRef<nsMainThreadSurfaceRef> mSurface;
|
|
};
|
|
|
|
void
|
|
BasicPlanarYCbCrImage::SetData(const Data& aData)
|
|
{
|
|
// Do some sanity checks to prevent integer overflow
|
|
if (aData.mYSize.width > 16384 || aData.mYSize.height > 16384) {
|
|
NS_ERROR("Illegal width or height");
|
|
return;
|
|
}
|
|
size_t size = aData.mPicSize.width*aData.mPicSize.height*4;
|
|
mBuffer = new PRUint8[size];
|
|
if (!mBuffer) {
|
|
// out of memory
|
|
return;
|
|
}
|
|
|
|
gfx::YUVType type = gfx::YV12;
|
|
if (aData.mYSize.width == aData.mCbCrSize.width &&
|
|
aData.mYSize.height == aData.mCbCrSize.height) {
|
|
type = gfx::YV24;
|
|
}
|
|
else if (aData.mYSize.width / 2 == aData.mCbCrSize.width &&
|
|
aData.mYSize.height == aData.mCbCrSize.height) {
|
|
type = gfx::YV16;
|
|
}
|
|
else if (aData.mYSize.width / 2 == aData.mCbCrSize.width &&
|
|
aData.mYSize.height / 2 == aData.mCbCrSize.height ) {
|
|
type = gfx::YV12;
|
|
}
|
|
else {
|
|
NS_ERROR("YCbCr format not supported");
|
|
}
|
|
|
|
// Convert from YCbCr to RGB now
|
|
gfx::ConvertYCbCrToRGB32(aData.mYChannel,
|
|
aData.mCbChannel,
|
|
aData.mCrChannel,
|
|
mBuffer,
|
|
aData.mPicX,
|
|
aData.mPicY,
|
|
aData.mPicSize.width,
|
|
aData.mPicSize.height,
|
|
aData.mYStride,
|
|
aData.mCbCrStride,
|
|
aData.mPicSize.width*4,
|
|
type);
|
|
mSize = aData.mPicSize;
|
|
}
|
|
|
|
static cairo_user_data_key_t imageSurfaceDataKey;
|
|
|
|
static void
|
|
DestroyBuffer(void* aBuffer)
|
|
{
|
|
delete[] static_cast<PRUint8*>(aBuffer);
|
|
}
|
|
|
|
already_AddRefed<gfxASurface>
|
|
BasicPlanarYCbCrImage::GetAsSurface()
|
|
{
|
|
NS_ASSERTION(NS_IsMainThread(), "Must be main thread");
|
|
|
|
if (mSurface) {
|
|
nsRefPtr<gfxASurface> result = mSurface.get();
|
|
return result.forget();
|
|
}
|
|
|
|
if (!mBuffer) {
|
|
return nsnull;
|
|
}
|
|
nsRefPtr<gfxImageSurface> imgSurface =
|
|
new gfxImageSurface(mBuffer, mSize,
|
|
mSize.width * 4,
|
|
gfxASurface::ImageFormatRGB24);
|
|
if (!imgSurface) {
|
|
return nsnull;
|
|
}
|
|
|
|
// Pass ownership of the buffer to the surface
|
|
imgSurface->SetData(&imageSurfaceDataKey, mBuffer.forget(), DestroyBuffer);
|
|
|
|
nsRefPtr<gfxASurface> result = imgSurface.get();
|
|
#if defined(XP_MACOSX)
|
|
nsRefPtr<gfxQuartzImageSurface> quartzSurface =
|
|
new gfxQuartzImageSurface(imgSurface);
|
|
if (quartzSurface) {
|
|
result = quartzSurface.forget();
|
|
}
|
|
#endif
|
|
mSurface = result.get();
|
|
|
|
return result.forget();
|
|
}
|
|
|
|
/**
|
|
* Our image container is very simple. It's really just a factory
|
|
* for the image objects. We use a Monitor to synchronize access to
|
|
* mImage.
|
|
*/
|
|
class BasicImageContainer : public ImageContainer {
|
|
public:
|
|
BasicImageContainer(BasicLayerManager* aManager) :
|
|
ImageContainer(aManager), mMonitor("BasicImageContainer")
|
|
{}
|
|
virtual already_AddRefed<Image> CreateImage(const Image::Format* aFormats,
|
|
PRUint32 aNumFormats);
|
|
virtual void SetCurrentImage(Image* aImage);
|
|
virtual already_AddRefed<Image> GetCurrentImage();
|
|
virtual already_AddRefed<gfxASurface> GetCurrentAsSurface(gfxIntSize* aSize);
|
|
|
|
protected:
|
|
Monitor mMonitor;
|
|
nsRefPtr<Image> mImage;
|
|
};
|
|
|
|
/**
|
|
* Returns true if aFormat is in the given format array.
|
|
*/
|
|
static PRBool
|
|
FormatInList(const Image::Format* aFormats, PRUint32 aNumFormats,
|
|
Image::Format aFormat)
|
|
{
|
|
for (PRUint32 i = 0; i < aNumFormats; ++i) {
|
|
if (aFormats[i] == aFormat) {
|
|
return PR_TRUE;
|
|
}
|
|
}
|
|
return PR_FALSE;
|
|
}
|
|
|
|
already_AddRefed<Image>
|
|
BasicImageContainer::CreateImage(const Image::Format* aFormats,
|
|
PRUint32 aNumFormats)
|
|
{
|
|
nsRefPtr<Image> image;
|
|
// Prefer cairo surfaces because they're native for us
|
|
if (FormatInList(aFormats, aNumFormats, Image::CAIRO_SURFACE)) {
|
|
image = new BasicCairoImage();
|
|
} else if (FormatInList(aFormats, aNumFormats, Image::PLANAR_YCBCR)) {
|
|
image = new BasicPlanarYCbCrImage();
|
|
}
|
|
return image.forget();
|
|
}
|
|
|
|
void
|
|
BasicImageContainer::SetCurrentImage(Image* aImage)
|
|
{
|
|
MonitorAutoEnter mon(mMonitor);
|
|
mImage = aImage;
|
|
}
|
|
|
|
already_AddRefed<Image>
|
|
BasicImageContainer::GetCurrentImage()
|
|
{
|
|
MonitorAutoEnter mon(mMonitor);
|
|
nsRefPtr<Image> image = mImage;
|
|
return image.forget();
|
|
}
|
|
|
|
static BasicImageImplData*
|
|
ToImageData(Image* aImage)
|
|
{
|
|
return static_cast<BasicImageImplData*>(aImage->GetImplData());
|
|
}
|
|
|
|
already_AddRefed<gfxASurface>
|
|
BasicImageContainer::GetCurrentAsSurface(gfxIntSize* aSizeResult)
|
|
{
|
|
NS_PRECONDITION(NS_IsMainThread(), "Must be called on main thread");
|
|
|
|
MonitorAutoEnter mon(mMonitor);
|
|
if (!mImage) {
|
|
return nsnull;
|
|
}
|
|
*aSizeResult = ToImageData(mImage)->GetSize();
|
|
return ToImageData(mImage)->GetAsSurface();
|
|
}
|
|
|
|
already_AddRefed<ImageContainer>
|
|
BasicLayerManager::CreateImageContainer()
|
|
{
|
|
nsRefPtr<ImageContainer> container = new BasicImageContainer(this);
|
|
return container.forget();
|
|
}
|
|
|
|
}
|
|
}
|