InjectedBundleRangeHandle.cpp [plain text]
#include "config.h"
#include "InjectedBundleRangeHandle.h"
#include "ShareableBitmap.h"
#include "WebImage.h"
#include <JavaScriptCore/APICast.h>
#include <WebCore/Document.h>
#include <WebCore/FloatRect.h>
#include <WebCore/Frame.h>
#include <WebCore/FrameSelection.h>
#include <WebCore/FrameView.h>
#include <WebCore/GraphicsContext.h>
#include <WebCore/IntRect.h>
#include <WebCore/JSRange.h>
#include <WebCore/Page.h>
#include <WebCore/Range.h>
#include <WebCore/VisibleSelection.h>
#include <wtf/HashMap.h>
#include <wtf/NeverDestroyed.h>
using namespace WebCore;
namespace WebKit {
typedef HashMap<Range*, InjectedBundleRangeHandle*> DOMHandleCache;
static DOMHandleCache& domHandleCache()
{
static NeverDestroyed<DOMHandleCache> cache;
return cache;
}
PassRefPtr<InjectedBundleRangeHandle> InjectedBundleRangeHandle::getOrCreate(JSContextRef, JSObjectRef object)
{
Range* range = JSRange::toWrapped(toJS(object));
return getOrCreate(range);
}
PassRefPtr<InjectedBundleRangeHandle> InjectedBundleRangeHandle::getOrCreate(Range* range)
{
if (!range)
return 0;
DOMHandleCache::AddResult result = domHandleCache().add(range, nullptr);
if (!result.isNewEntry)
return PassRefPtr<InjectedBundleRangeHandle>(result.iterator->value);
auto rangeHandle = InjectedBundleRangeHandle::create(range);
result.iterator->value = rangeHandle.ptr();
return WTFMove(rangeHandle);
}
Ref<InjectedBundleRangeHandle> InjectedBundleRangeHandle::create(Range* range)
{
return adoptRef(*new InjectedBundleRangeHandle(range));
}
InjectedBundleRangeHandle::InjectedBundleRangeHandle(Range* range)
: m_range(range)
{
}
InjectedBundleRangeHandle::~InjectedBundleRangeHandle()
{
domHandleCache().remove(m_range.get());
}
Range* InjectedBundleRangeHandle::coreRange() const
{
return m_range.get();
}
WebCore::IntRect InjectedBundleRangeHandle::boundingRectInWindowCoordinates() const
{
FloatRect boundingRect = m_range->absoluteBoundingRect();
Frame* frame = m_range->ownerDocument().frame();
return frame->view()->contentsToWindow(enclosingIntRect(boundingRect));
}
PassRefPtr<WebImage> InjectedBundleRangeHandle::renderedImage(SnapshotOptions options)
{
Document& ownerDocument = m_range->ownerDocument();
Frame* frame = ownerDocument.frame();
if (!frame)
return nullptr;
FrameView* frameView = frame->view();
if (!frameView)
return nullptr;
VisibleSelection oldSelection = frame->selection().selection();
frame->selection().setSelection(VisibleSelection(*m_range));
float scaleFactor = (options & SnapshotOptionsExcludeDeviceScaleFactor) ? 1 : frame->page()->deviceScaleFactor();
IntRect paintRect = enclosingIntRect(m_range->absoluteBoundingRect());
IntSize backingStoreSize = paintRect.size();
backingStoreSize.scale(scaleFactor);
RefPtr<ShareableBitmap> backingStore = ShareableBitmap::createShareable(backingStoreSize, ShareableBitmap::SupportsAlpha);
if (!backingStore)
return nullptr;
auto graphicsContext = backingStore->createGraphicsContext();
graphicsContext->scale(FloatSize(scaleFactor, scaleFactor));
paintRect.move(frameView->frameRect().x(), frameView->frameRect().y());
paintRect.moveBy(-frameView->scrollPosition());
graphicsContext->translate(-paintRect.x(), -paintRect.y());
PaintBehavior oldPaintBehavior = frameView->paintBehavior();
PaintBehavior paintBehavior = oldPaintBehavior | PaintBehaviorSelectionOnly | PaintBehaviorFlattenCompositingLayers;
if (options & SnapshotOptionsForceBlackText)
paintBehavior |= PaintBehaviorForceBlackText;
if (options & SnapshotOptionsForceWhiteText)
paintBehavior |= PaintBehaviorForceWhiteText;
frameView->setPaintBehavior(paintBehavior);
ownerDocument.updateLayout();
frameView->paint(*graphicsContext, paintRect);
frameView->setPaintBehavior(oldPaintBehavior);
frame->selection().setSelection(oldSelection);
return WebImage::create(backingStore);
}
}