third_party/WebKit/Source/core/testing/sim/SimCompositor.cpp

// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "core/testing/sim/SimCompositor.h"

#include "core/exported/WebViewImpl.h"
#include "core/frame/LocalFrame.h"
#include "core/frame/LocalFrameView.h"
#include "core/frame/WebLocalFrameImpl.h"
#include "core/layout/LayoutView.h"
#include "core/paint/PaintLayer.h"
#include "core/paint/compositing/CompositedLayerMapping.h"
#include "core/paint/compositing/PaintLayerCompositor.h"
#include "core/testing/sim/SimDisplayItemList.h"
#include "platform/graphics/GraphicsLayer.h"
#include "platform/wtf/Time.h"
#include "public/platform/WebRect.h"

namespace blink {

static void PaintLayers(GraphicsLayer& layer,
                        SimDisplayItemList& display_list) {
  if (layer.DrawsContent() && layer.HasTrackedRasterInvalidations()) {
    layer.WebContentLayerClientForTesting().PaintContents(&display_list);
    layer.ResetTrackedRasterInvalidations();
  }

  if (GraphicsLayer* mask_layer = layer.MaskLayer())
    PaintLayers(*mask_layer, display_list);
  if (GraphicsLayer* contents_clipping_mask_layer =
          layer.ContentsClippingMaskLayer())
    PaintLayers(*contents_clipping_mask_layer, display_list);

  for (auto child : layer.Children())
    PaintLayers(*child, display_list);
}

static void PaintFrames(LocalFrame& root, SimDisplayItemList& display_list) {
  GraphicsLayer* layer =
      root.View()->GetLayoutView()->Compositor()->RootGraphicsLayer();
  PaintLayers(*layer, display_list);
}

SimCompositor::SimCompositor()
    : needs_begin_frame_(false),
      defer_commits_(true),
      has_selection_(false),
      web_view_(nullptr),
      last_frame_time_monotonic_(0) {
  LocalFrameView::SetInitialTracksPaintInvalidationsForTesting(true);
}

SimCompositor::~SimCompositor() {
  LocalFrameView::SetInitialTracksPaintInvalidationsForTesting(false);
}

void SimCompositor::SetWebView(WebViewImpl& web_view) {
  web_view_ = &web_view;
}

void SimCompositor::SetNeedsBeginFrame() {
  needs_begin_frame_ = true;
}

void SimCompositor::SetDeferCommits(bool defer_commits) {
  defer_commits_ = defer_commits;
}

void SimCompositor::RegisterSelection(const WebSelection&) {
  has_selection_ = true;
}

void SimCompositor::ClearSelection() {
  has_selection_ = false;
}

SimDisplayItemList SimCompositor::BeginFrame(double time_delta_in_seconds) {
  DCHECK(web_view_);
  DCHECK(!defer_commits_);
  DCHECK(needs_begin_frame_);
  DCHECK_GT(time_delta_in_seconds, 0);
  needs_begin_frame_ = false;

  last_frame_time_monotonic_ += time_delta_in_seconds;

  web_view_->BeginFrame(last_frame_time_monotonic_);
  web_view_->UpdateAllLifecyclePhases();

  LocalFrame* root = web_view_->MainFrameImpl()->GetFrame();

  SimDisplayItemList display_list;
  PaintFrames(*root, display_list);

  return display_list;
}

}  // namespace blink