Move LocalFrame::DragImageForSelection DragController

third_party/WebKit/Source/core/page/DragController.cpp

 /*
  * Copyright (C) 2007, 2009, 2010 Apple Inc. All rights reserved.
  * Copyright (C) 2008 Google Inc.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
@@ skipping 58 matching lines @@
 #include "core/page/DragState.h"
 #include "core/page/Page.h"
 #include "core/svg/graphics/SVGImageForContainer.h"
 #include "platform/DragImage.h"
 #include "platform/SharedBuffer.h"
 #include "platform/geometry/IntRect.h"
 #include "platform/geometry/IntSize.h"
 #include "platform/graphics/BitmapImage.h"
 #include "platform/graphics/Image.h"
 #include "platform/graphics/ImageOrientation.h"
+#include "platform/graphics/paint/PaintRecordBuilder.h"
 #include "platform/loader/fetch/ResourceFetcher.h"
 #include "platform/loader/fetch/ResourceRequest.h"
 #include "platform/weborigin/SecurityOrigin.h"
 #include "platform/wtf/Assertions.h"
 #include "platform/wtf/CurrentTime.h"
 #include "platform/wtf/PassRefPtr.h"
 #include "platform/wtf/RefPtr.h"
 #include "public/platform/WebCommon.h"
 #include "public/platform/WebDragData.h"
 #include "public/platform/WebDragOperation.h"
@@ skipping 984 matching lines @@
       link_url, link_text, font_description, device_scale_factor);
 
   IntSize size = drag_image ? drag_image->Size() : IntSize();
   IntPoint drag_image_offset(-size.Width() / 2, -kLinkDragBorderInset);
   drag_loc = IntPoint(mouse_dragged_point.X() + drag_image_offset.X(),
                       mouse_dragged_point.Y() + drag_image_offset.Y());
 
   return drag_image;
 }
 
+std::unique_ptr<DragImage> DragController::DragImageForSelection(
+    LocalFrame* src,

[Xiaocheng, 2017/06/02 18:26:47]

Can we pass a |const LocalFrame&| parameter?

+    float opacity) {
+  if (!src->Selection().ComputeVisibleSelectionInDOMTreeDeprecated().IsRange())
+    return nullptr;
+
+  src->View()->UpdateAllLifecyclePhasesExceptPaint();
+  DCHECK(src->GetDocument()->IsActive());
+
+  FloatRect painting_rect = FloatRect(src->Selection().Bounds());
+  GlobalPaintFlags paint_flags =
+      kGlobalPaintSelectionOnly | kGlobalPaintFlattenCompositingLayers;
+
+  PaintRecordBuilder builder(
+      DataTransfer::DeviceSpaceBounds(painting_rect, *src));
+  src->View()->PaintContents(builder.Context(), paint_flags,
+                             EnclosingIntRect(painting_rect));
+  return DataTransfer::CreateDragImage(
+      *src, opacity, kDoNotRespectImageOrientation, painting_rect, builder,
+      PropertyTreeState::Root());
+}
+
 bool DragController::StartDrag(LocalFrame* src,
                                const DragState& state,
                                const WebMouseEvent& drag_event,
                                const IntPoint& drag_origin) {
 #if DCHECK_IS_ON()
   DCHECK(DragTypeIsValid(state.drag_type_));
 #endif
   DCHECK(src);
   if (!src->View() || src->ContentLayoutItem().IsNull())
     return false;
@@ skipping 24 matching lines @@
   std::unique_ptr<DragImage> drag_image =
       data_transfer->CreateDragImage(drag_offset, src);
   if (drag_image) {
     drag_location = DragLocationForDHTMLDrag(mouse_dragged_point, drag_origin,
                                              drag_offset, !link_url.IsEmpty());
   }
 
   Node* node = state.drag_src_.Get();
   if (state.drag_type_ == kDragSourceActionSelection) {
     if (!drag_image) {
-      drag_image = src->DragImageForSelection(kDragImageAlpha);
+      drag_image = DragController::DragImageForSelection(src, kDragImageAlpha);

[Xiaocheng, 2017/06/02 18:26:47]

nit: Remove |DragController|

       drag_location = DragLocationForSelectionDrag(src);
     }
     DoSystemDrag(drag_image.get(), drag_location, drag_origin, data_transfer,
                  src, false);
   } else if (state.drag_type_ == kDragSourceActionImage) {
     if (image_url.IsEmpty() || !node || !node->IsElementNode())
       return false;
     Element* element = ToElement(node);
     Image* image = GetImage(element);
     if (!image || image->IsNull() || !image->Data() || !image->Data()->size())
@@ skipping 131 matching lines @@
 
 DEFINE_TRACE(DragController) {
   visitor->Trace(page_);
   visitor->Trace(document_under_mouse_);
   visitor->Trace(drag_initiator_);
   visitor->Trace(drag_state_);
   visitor->Trace(file_input_element_under_mouse_);
 }
 
 }  // namespace blink