chromium/src/third_party/blink/renderer/platform/widget/compositing/widget_compositor.h

// Copyright 2020 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.

#ifndef THIRD_PARTY_BLINK_RENDERER_PLATFORM_WIDGET_COMPOSITING_WIDGET_COMPOSITOR_H_
#define THIRD_PARTY_BLINK_RENDERER_PLATFORM_WIDGET_COMPOSITING_WIDGET_COMPOSITOR_H_

#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/receiver.h"
#include "third_party/blink/public/mojom/page/widget.mojom-blink.h"
#include "third_party/blink/renderer/platform/platform_export.h"
#include "third_party/blink/renderer/platform/widget/compositing/widget_swap_queue.h"

namespace cc {
class LayerTreeHost;
}

namespace blink {

class WidgetBase;

class PLATFORM_EXPORT WidgetCompositor
    : public base::RefCountedThreadSafe<WidgetCompositor>,
      public mojom::blink::WidgetCompositor {
 public:
  WidgetCompositor(
      base::WeakPtr<WidgetBase> widget_base,
      scoped_refptr<base::SingleThreadTaskRunner> main_task_runner,
      scoped_refptr<base::SingleThreadTaskRunner> compositor_task_runner,
      mojo::PendingReceiver<mojom::blink::WidgetCompositor> receiver);
  ~WidgetCompositor() override = default;
  WidgetCompositor(const WidgetCompositor&) = delete;
  WidgetCompositor& operator=(const WidgetCompositor&) = delete;

  void Shutdown();

  // mojom::WidgetCompositor:
  void VisualStateRequest(VisualStateRequestCallback callback) override;

  virtual cc::LayerTreeHost* LayerTreeHost() const;

 private:
  void BindOnThread(
      mojo::PendingReceiver<mojom::blink::WidgetCompositor> receiver);
  void ResetOnThread();
  void CreateQueueSwapPromise(base::OnceCallback<void(int)> drain_callback,
                              base::OnceClosure swap_callback,
                              VisualStateRequestCallback callback);
  void VisualStateResponse();
  void DrainQueue(int source_frame_number);
  bool CalledOnValidCompositorThread();

  // Note that |widget_base_| is safe to be accessed on the main thread.
  base::WeakPtr<WidgetBase> widget_base_;
  scoped_refptr<base::SingleThreadTaskRunner> main_task_runner_;
  scoped_refptr<base::SingleThreadTaskRunner> compositor_task_runner_;
  mojo::Receiver<mojom::blink::WidgetCompositor> receiver_{this};
  std::unique_ptr<WidgetSwapQueue> swap_queue_;
};

}  // namespace blink

#endif  // THIRD_PARTY_BLINK_RENDERER_PLATFORM_WIDGET_COMPOSITING_WIDGET_COMPOSITOR_H_