/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef GMPSharedMemManager_h_
#define GMPSharedMemManager_h_

#include "mozilla/ipc/Shmem.h"
#include "nsTArray.h"

namespace mozilla::gmp {

class GMPVideoi420FrameImpl;

enum class GMPSharedMemClass { Decoded, Encoded };

class GMPSharedMemManager {
 public:
  GMPSharedMemManager() = default;

  virtual ~GMPSharedMemManager();

  bool MgrTakeShmem(GMPSharedMemClass aClass, ipc::Shmem* aMem);
  bool MgrTakeShmem(GMPSharedMemClass aClass, size_t aSize, ipc::Shmem* aMem);
  void MgrGiveShmem(GMPSharedMemClass aClass, ipc::Shmem&& aMem);
  void MgrPurgeShmems();

  virtual bool MgrAllocShmem(size_t aSize, ipc::Shmem* aMem) { return false; }
  virtual void MgrDeallocShmem(ipc::Shmem& aMem) = 0;

  virtual void MgrDecodedFrameDestroyed(GMPVideoi420FrameImpl* aFrame) {}

 protected:
  virtual bool MgrIsOnOwningThread() const = 0;
  virtual bool MgrCanSend() const = 0;

  static constexpr size_t kMaxPools = 2;

 private:
  void PurgeSmallerShmem(nsTArray<ipc::Shmem>& aPool, size_t aSize);

  static constexpr size_t kMaxPoolLength = 16;
  AutoTArray<ipc::Shmem, kMaxPoolLength> mPool[kMaxPools];
};

}  // namespace mozilla::gmp

#endif  // GMPSharedMemManager_h_