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

#ifndef MEDIA_GPU_WINDOWS_D3D12_FENCE_H_
#define MEDIA_GPU_WINDOWS_D3D12_FENCE_H_

#include <d3d12.h>
#include <wrl.h>

#include "base/memory/ref_counted.h"
#include "base/memory/scoped_refptr.h"
#include "media/gpu/media_gpu_export.h"
#include "media/gpu/windows/d3d11_status.h"
#include "media/gpu/windows/d3d_com_defs.h"

namespace media {

// D3D12Fence wraps a ID3D12Fence pointer and its last signaled fence value.
class MEDIA_GPU_EXPORT D3D12Fence
    : public base::RefCountedThreadSafe<D3D12Fence> {
 public:
  REQUIRE_ADOPTION_FOR_REFCOUNTED_TYPE();

  explicit D3D12Fence(ComD3D12Fence fence);

  static scoped_refptr<D3D12Fence> Create(
      ID3D12Device* device,
      D3D12_FENCE_FLAGS flags = D3D12_FENCE_FLAG_NONE);

  // Get the underlying ID3D12Fence pointer.
  ID3D12Fence* Get() const;

  // Get the last signaled fence value.
  uint64_t Value() const;

  // Get the fence value completed by GPU.
  uint64_t GetCompletedValue() const;

  // Let |command_queue| signal the fence and return the corresponding fence
  // value to be waited for elsewhere.
  D3D11Status::Or<uint64_t> Signal(ID3D12CommandQueue& command_queue);

  // Wait on CPU until the |fence_value| is signaled.
  D3D11Status WaitCPU(uint64_t fence_value) const;

  // Let D3D11 |device_context| wait on GPU until the |fence_value| is signaled.
  // The D3D11Fence should have been checked to be supported before calling
  // this.
  D3D11Status WaitGPU(ID3D11DeviceContext& device_context,
                      uint64_t fence_value);

  // Signal the fence and wait on CPU until the fence is signaled.
  D3D11Status SignalAndWaitCPU(ID3D12CommandQueue& command_queue);

 private:
  friend class RefCountedThreadSafe;
  ~D3D12Fence();

  ComD3D12Fence fence_;
  uint64_t fence_value_ = 0;

  ComD3D11Fence d3d11_fence_;
};

}  // namespace media

#endif  // MEDIA_GPU_WINDOWS_D3D12_FENCE_H_