gecko-dev/gfx/2d/SourceSurfaceD2D1.cpp

249 строки
6.7 KiB
C++

/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* 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 "SourceSurfaceD2D1.h"
#include "DrawTargetD2D1.h"
namespace mozilla {
namespace gfx {
SourceSurfaceD2D1::SourceSurfaceD2D1(ID2D1Image* aImage,
ID2D1DeviceContext* aDC,
SurfaceFormat aFormat,
const IntSize& aSize, DrawTargetD2D1* aDT)
: mImage(aImage),
mDC(aDC),
mDevice(Factory::GetD2D1Device()),
mFormat(aFormat),
mSize(aSize),
mDrawTarget(aDT),
mOwnsCopy(false) {
aImage->QueryInterface((ID2D1Bitmap1**)getter_AddRefs(mRealizedBitmap));
if (aDT) {
mSnapshotLock = aDT->mSnapshotLock;
}
}
SourceSurfaceD2D1::~SourceSurfaceD2D1() {
if (mOwnsCopy) {
DrawTargetD2D1::mVRAMUsageSS -=
mSize.width * mSize.height * BytesPerPixel(mFormat);
}
}
bool SourceSurfaceD2D1::IsValid() const {
return mDevice == Factory::GetD2D1Device();
}
already_AddRefed<DataSourceSurface> SourceSurfaceD2D1::GetDataSurface() {
Maybe<MutexAutoLock> lock;
if (mSnapshotLock) {
lock.emplace(*mSnapshotLock);
}
if (!EnsureRealizedBitmap()) {
gfxCriticalError() << "Failed to realize a bitmap, device "
<< hexa(mDevice);
return nullptr;
}
HRESULT hr;
RefPtr<ID2D1Bitmap1> softwareBitmap;
D2D1_BITMAP_PROPERTIES1 props;
props.dpiX = 96;
props.dpiY = 96;
props.pixelFormat = D2DPixelFormat(mFormat);
props.colorContext = nullptr;
props.bitmapOptions =
D2D1_BITMAP_OPTIONS_CANNOT_DRAW | D2D1_BITMAP_OPTIONS_CPU_READ;
hr = mDC->CreateBitmap(D2DIntSize(mSize), nullptr, 0, props,
(ID2D1Bitmap1**)getter_AddRefs(softwareBitmap));
if (FAILED(hr)) {
gfxCriticalError() << "Failed to create software bitmap: " << mSize
<< " Code: " << hexa(hr);
return nullptr;
}
D2D1_POINT_2U point = D2D1::Point2U(0, 0);
D2D1_RECT_U rect = D2D1::RectU(0, 0, mSize.width, mSize.height);
hr = softwareBitmap->CopyFromBitmap(&point, mRealizedBitmap, &rect);
if (FAILED(hr)) {
gfxWarning() << "Failed to readback into software bitmap. Code: "
<< hexa(hr);
return nullptr;
}
return MakeAndAddRef<DataSourceSurfaceD2D1>(softwareBitmap, mFormat);
}
bool SourceSurfaceD2D1::EnsureRealizedBitmap() {
if (mRealizedBitmap) {
return true;
}
// Why aren't we using mDevice here or anywhere else?
RefPtr<ID2D1Device> device = Factory::GetD2D1Device();
if (!device) {
return false;
}
RefPtr<ID2D1DeviceContext> dc;
device->CreateDeviceContext(D2D1_DEVICE_CONTEXT_OPTIONS_NONE,
getter_AddRefs(dc));
D2D1_BITMAP_PROPERTIES1 props;
props.dpiX = 96;
props.dpiY = 96;
props.pixelFormat = D2DPixelFormat(mFormat);
props.colorContext = nullptr;
props.bitmapOptions = D2D1_BITMAP_OPTIONS_TARGET;
dc->CreateBitmap(D2DIntSize(mSize), nullptr, 0, props,
(ID2D1Bitmap1**)getter_AddRefs(mRealizedBitmap));
dc->SetTarget(mRealizedBitmap);
dc->BeginDraw();
dc->DrawImage(mImage);
dc->EndDraw();
return true;
}
void SourceSurfaceD2D1::DrawTargetWillChange() {
MOZ_ASSERT(mSnapshotLock);
mSnapshotLock->AssertCurrentThreadOwns();
// At this point in time this should always be true here.
MOZ_ASSERT(mRealizedBitmap);
RefPtr<ID2D1Bitmap1> oldBitmap = mRealizedBitmap;
D2D1_BITMAP_PROPERTIES1 props;
props.dpiX = 96;
props.dpiY = 96;
props.pixelFormat = D2DPixelFormat(mFormat);
props.colorContext = nullptr;
props.bitmapOptions = D2D1_BITMAP_OPTIONS_TARGET;
HRESULT hr =
mDC->CreateBitmap(D2DIntSize(mSize), nullptr, 0, props,
(ID2D1Bitmap1**)getter_AddRefs(mRealizedBitmap));
if (FAILED(hr)) {
gfxCriticalError()
<< "Failed to create bitmap to make DrawTarget copy. Size: " << mSize
<< " Code: " << hexa(hr);
MarkIndependent();
return;
}
D2D1_POINT_2U point = D2D1::Point2U(0, 0);
D2D1_RECT_U rect = D2D1::RectU(0, 0, mSize.width, mSize.height);
mRealizedBitmap->CopyFromBitmap(&point, oldBitmap, &rect);
mImage = mRealizedBitmap;
DrawTargetD2D1::mVRAMUsageSS +=
mSize.width * mSize.height * BytesPerPixel(mFormat);
mOwnsCopy = true;
// Ensure the object stays alive for the duration of MarkIndependent.
RefPtr<SourceSurfaceD2D1> deathGrip = this;
// We now no longer depend on the source surface content remaining the same.
MarkIndependent();
}
void SourceSurfaceD2D1::MarkIndependent() {
if (mDrawTarget) {
MOZ_ASSERT(mDrawTarget->mSnapshot == this);
mDrawTarget->mSnapshot = nullptr;
mDrawTarget = nullptr;
}
}
DataSourceSurfaceD2D1::DataSourceSurfaceD2D1(ID2D1Bitmap1* aMappableBitmap,
SurfaceFormat aFormat)
: mBitmap(aMappableBitmap),
mFormat(aFormat),
mIsMapped(false),
mImplicitMapped(false) {}
DataSourceSurfaceD2D1::~DataSourceSurfaceD2D1() {
if (mImplicitMapped) {
mBitmap->Unmap();
}
}
IntSize DataSourceSurfaceD2D1::GetSize() const {
D2D1_SIZE_F size = mBitmap->GetSize();
return IntSize(int32_t(size.width), int32_t(size.height));
}
uint8_t* DataSourceSurfaceD2D1::GetData() {
EnsureMapped();
return mMap.bits;
}
bool DataSourceSurfaceD2D1::Map(MapType aMapType,
MappedSurface* aMappedSurface) {
// DataSourceSurfaces used with the new Map API should not be used with
// GetData!!
MOZ_ASSERT(!mImplicitMapped);
MOZ_ASSERT(!mIsMapped);
D2D1_MAP_OPTIONS options;
if (aMapType == MapType::READ) {
options = D2D1_MAP_OPTIONS_READ;
} else {
gfxWarning() << "Attempt to map D2D1 DrawTarget for writing.";
return false;
}
D2D1_MAPPED_RECT map;
if (FAILED(mBitmap->Map(D2D1_MAP_OPTIONS_READ, &map))) {
gfxCriticalError() << "Failed to map bitmap (M).";
return false;
}
aMappedSurface->mData = map.bits;
aMappedSurface->mStride = map.pitch;
mIsMapped = !!aMappedSurface->mData;
return mIsMapped;
}
void DataSourceSurfaceD2D1::Unmap() {
MOZ_ASSERT(mIsMapped);
mIsMapped = false;
mBitmap->Unmap();
}
int32_t DataSourceSurfaceD2D1::Stride() {
EnsureMapped();
return mMap.pitch;
}
void DataSourceSurfaceD2D1::EnsureMapped() {
// Do not use GetData() after having used Map!
MOZ_ASSERT(!mIsMapped);
if (mImplicitMapped) {
return;
}
if (FAILED(mBitmap->Map(D2D1_MAP_OPTIONS_READ, &mMap))) {
gfxCriticalError() << "Failed to map bitmap (EM).";
return;
}
mImplicitMapped = true;
}
} // namespace gfx
} // namespace mozilla