services/video_capture/gpu_memory_buffer_virtual_device_mojo_adapter.h

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

#ifndef SERVICES_VIDEO_CAPTURE_GPU_MEMORY_BUFFER_VIRTUAL_DEVICE_MOJO_ADAPTER_H_
#define SERVICES_VIDEO_CAPTURE_GPU_MEMORY_BUFFER_VIRTUAL_DEVICE_MOJO_ADAPTER_H_

#include <map>

#include "base/functional/callback.h"
#include "base/sequence_checker.h"
#include "media/capture/video/video_capture_buffer_pool.h"
#include "media/capture/video_capture_types.h"
#include "mojo/public/cpp/bindings/pending_remote.h"
#include "mojo/public/cpp/bindings/remote.h"
#include "services/video_capture/device.h"
#include "services/video_capture/public/cpp/video_frame_access_handler.h"
#include "services/video_capture/public/mojom/producer.mojom.h"
#include "services/video_capture/public/mojom/video_frame_handler.mojom.h"
#include "services/video_capture/public/mojom/virtual_device.mojom.h"

namespace video_capture {

class GpuMemoryBufferVirtualDeviceMojoAdapter
    : public mojom::GpuMemoryBufferVirtualDevice,
      public Device {
 public:
  GpuMemoryBufferVirtualDeviceMojoAdapter();
  GpuMemoryBufferVirtualDeviceMojoAdapter(
      GpuMemoryBufferVirtualDeviceMojoAdapter&) = delete;
  GpuMemoryBufferVirtualDeviceMojoAdapter& operator=(
      GpuMemoryBufferVirtualDeviceMojoAdapter&) = delete;

  GpuMemoryBufferVirtualDeviceMojoAdapter(
      const GpuMemoryBufferVirtualDeviceMojoAdapter&) = delete;
  GpuMemoryBufferVirtualDeviceMojoAdapter& operator=(
      const GpuMemoryBufferVirtualDeviceMojoAdapter&) = delete;

  ~GpuMemoryBufferVirtualDeviceMojoAdapter() override;

  void SetReceiverDisconnectedCallback(base::OnceClosure callback);

  // mojom::GpuMemoryBufferVirtualDevice implementation.
  void OnNewGpuMemoryBufferHandle(
      int32_t buffer_id,
      gfx::GpuMemoryBufferHandle gmb_handle) override;
  void OnFrameAccessHandlerReady(
      mojo::PendingRemote<video_capture::mojom::VideoFrameAccessHandler>
          pending_frame_access_handler) override;
  void OnFrameReadyInBuffer(
      int32_t buffer_id,
      media::mojom::VideoFrameInfoPtr frame_info) override;
  void OnBufferRetired(int buffer_id) override;

  // mojom::Device implementation.
  void Start(const media::VideoCaptureParams& requested_settings,
             mojo::PendingRemote<mojom::VideoFrameHandler> handler) override;
  void MaybeSuspend() override;
  void Resume() override;
  void GetPhotoState(GetPhotoStateCallback callback) override;
  void SetPhotoOptions(media::mojom::PhotoSettingsPtr settings,
                       SetPhotoOptionsCallback callback) override;
  void TakePhoto(TakePhotoCallback callback) override;
  void ProcessFeedback(const media::VideoCaptureFeedback& feedback) override;
  void RequestRefreshFrame() override;

  void Stop();

 private:
  void OnReceiverConnectionErrorOrClose();

  base::OnceClosure optional_receiver_disconnected_callback_;
  // The |video_frame_handler_| can be bound and unbound during the lifetime of
  // this adapter (e.g. due to Start(), Stop() and Start() again).
  mojo::Remote<mojom::VideoFrameHandler> video_frame_handler_;
  std::map<int32_t, gfx::GpuMemoryBufferHandle> known_buffer_handles_;
  // Because the adapter's lifetime can be greater than |video_frame_handler_|,
  // each handler that is bound gets forwarded its own instance of
  // VideoFrameAccessHandlerForwarder.
  scoped_refptr<VideoFrameAccessHandlerRemote> frame_access_handler_remote_;
  bool video_frame_handler_has_forwarder_ = false;
  SEQUENCE_CHECKER(sequence_checker_);
};

}  // namespace video_capture

#endif  // SERVICES_VIDEO_CAPTURE_GPU_MEMORY_BUFFER_VIRTUAL_DEVICE_MOJO_ADAPTER_H_