gecko-dev/dom/webgpu/ComputePipeline.cpp

53 строки
1.7 KiB
C++

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* 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 "ipc/WebGPUChild.h"
#include "mozilla/dom/WebGPUBinding.h"
namespace mozilla {
namespace webgpu {
GPU_IMPL_CYCLE_COLLECTION(ComputePipeline, mParent)
GPU_IMPL_JS_WRAP(ComputePipeline)
ComputePipeline::ComputePipeline(Device* const aParent, RawId aId,
RawId aImplicitPipelineLayoutId,
nsTArray<RawId>&& aImplicitBindGroupLayoutIds)
: ChildOf(aParent),
mImplicitPipelineLayoutId(aImplicitPipelineLayoutId),
mImplicitBindGroupLayoutIds(std::move(aImplicitBindGroupLayoutIds)),
mId(aId) {}
ComputePipeline::~ComputePipeline() { Cleanup(); }
void ComputePipeline::Cleanup() {
if (mValid && mParent) {
mValid = false;
auto bridge = mParent->GetBridge();
if (bridge && bridge->IsOpen()) {
bridge->SendComputePipelineDestroy(mId);
if (mImplicitPipelineLayoutId) {
bridge->SendImplicitLayoutDestroy(mImplicitPipelineLayoutId,
mImplicitBindGroupLayoutIds);
}
}
}
}
already_AddRefed<BindGroupLayout> ComputePipeline::GetBindGroupLayout(
uint32_t index) const {
const RawId id = index < mImplicitBindGroupLayoutIds.Length()
? mImplicitBindGroupLayoutIds[index]
: 0;
RefPtr<BindGroupLayout> object = new BindGroupLayout(mParent, id, false);
return object.forget();
}
} // namespace webgpu
} // namespace mozilla