/* 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/. */

#include "ComputePipeline.h"

#include "Device.h"
#include "mozilla/dom/WebGPUBinding.h"

namespace mozilla::webgpu {

GPU_IMPL_CYCLE_COLLECTION(ComputePipeline, mParent)
GPU_IMPL_JS_WRAP(ComputePipeline)

ComputePipeline::ComputePipeline(Device* const aParent, RawId aId)
    : ObjectBase(aParent->GetChild(), aId,
                 ffi::wgpu_client_drop_compute_pipeline),
      ChildOf(aParent) {}

ComputePipeline::~ComputePipeline() = default;

already_AddRefed<BindGroupLayout> ComputePipeline::GetBindGroupLayout(
    uint32_t aIndex) const {
  const RawId bglId = ffi::wgpu_client_compute_pipeline_get_bind_group_layout(
      GetClient(), mParent->GetId(), GetId(), aIndex);

  RefPtr<BindGroupLayout> object = new BindGroupLayout(mParent, bglId);
  return object.forget();
}

}  // namespace mozilla::webgpu
