// Copyright 2018 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_TEXTURE_VIRTUAL_DEVICE_MOJO_ADAPTER_H_
#define SERVICES_VIDEO_CAPTURE_TEXTURE_VIRTUAL_DEVICE_MOJO_ADAPTER_H_

#include "base/sequence_checker.h"
#include "media/capture/video/video_capture_buffer_pool.h"
#include "media/capture/video/video_frame_receiver_on_task_runner.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 TextureVirtualDeviceMojoAdapter : public mojom::TextureVirtualDevice,
                                        public Device {
 public:
  TextureVirtualDeviceMojoAdapter();

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

  ~TextureVirtualDeviceMojoAdapter() override;

  void SetReceiverDisconnectedCallback(base::OnceClosure callback);

  // mojom::TextureVirtualDevice implementation.
  void OnNewSharedImageBufferHandle(
      int32_t buffer_id,
      media::mojom::SharedImageBufferHandleSetPtr shared_image_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;

  // Device implementation.
  void Start(const media::VideoCaptureParams& requested_settings,
             mojo::PendingRemote<mojom::VideoFrameHandler> handler) override;
  void StartInProcess(
      const media::VideoCaptureParams& requested_settings,
      const base::WeakPtr<media::VideoFrameReceiver>& frame_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_;
  // Used when this device is started in process.
  base::WeakPtr<media::VideoFrameReceiver> video_frame_handler_in_process_;
  std::unordered_map<int32_t, media::mojom::SharedImageBufferHandleSetPtr>
      known_shared_image_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_TEXTURE_VIRTUAL_DEVICE_MOJO_ADAPTER_H_