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

#include <android/hardware/graphics/allocator/4.0/IAllocator.h>
#include <android/hardware/graphics/mapper/4.0/IMapper.h>

#include <memory>

#include "cros_gralloc/cros_gralloc_driver.h"
#include "cros_gralloc/cros_gralloc_helpers.h"

class CrosGralloc4Allocator : public android::hardware::graphics::allocator::V4_0::IAllocator {
  public:
    CrosGralloc4Allocator() = default;

    android::hardware::Return<void> allocate(const android::hardware::hidl_vec<uint8_t>& descriptor,
                                             uint32_t count, allocate_cb hidl_cb) override;

    android::hardware::graphics::mapper::V4_0::Error init();

  private:
    android::hardware::graphics::mapper::V4_0::Error allocate(
            const android::hardware::graphics::mapper::V4_0::IMapper::BufferDescriptorInfo&
                    description,
            uint32_t* outStride, android::hardware::hidl_handle* outHandle);

    std::shared_ptr<cros_gralloc_driver> mDriver;
};