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

#ifndef COMPONENTS_EXO_FRAME_SINK_RESOURCE_MANAGER_H_
#define COMPONENTS_EXO_FRAME_SINK_RESOURCE_MANAGER_H_

#include "base/containers/flat_map.h"
#include "components/viz/common/resources/release_callback.h"
#include "components/viz/common/resources/resource_id.h"
#include "components/viz/common/resources/returned_resource.h"

namespace exo {

// This class manages the resource IDs and active resource callbacks suitable
// for implementing a frame sink.
class FrameSinkResourceManager {
 public:
  using ReleaseCallback = base::OnceCallback<void(viz::ReturnedResource)>;

  FrameSinkResourceManager();

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

  ~FrameSinkResourceManager();

  bool HasReleaseCallbackForResource(viz::ResourceId id);
  void SetResourceReleaseCallback(viz::ResourceId id, ReleaseCallback callback);
  viz::ResourceId AllocateResourceId();

  bool HasNoCallbacks() const;
  void ReclaimResource(viz::ReturnedResource resource);
  void ClearAllCallbacks();

 private:
  // A collection of callbacks used to release resources.
  using ResourceReleaseCallbackMap =
      base::flat_map<viz::ResourceId, ReleaseCallback>;
  ResourceReleaseCallbackMap release_callbacks_;

  // The id generator for the buffer.
  viz::ResourceIdGenerator id_generator_;
};

}  // namespace exo

#endif  // COMPONENTS_EXO_FRAME_SINK_RESOURCE_MANAGER_H_