/* SPDX-License-Identifier: MIT
 *
 * Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved.
 */
#include "gr.h"

#include <engine/fifo.h>
#include <engine/gr/priv.h>

static int
nvkm_rm_gr_obj_ctor(const struct nvkm_oclass *oclass, void *argv, u32 argc,
		    struct nvkm_object **pobject)
{
	struct r535_gr_chan *chan = container_of(oclass->parent, typeof(*chan), object);

	return nvkm_rm_engine_obj_new(&chan->chan->rm.object, chan->chan->id, oclass, pobject);
}

static int
nvkm_rm_gr_fini(struct nvkm_gr *base, bool suspend)
{
	struct nvkm_rm *rm = base->engine.subdev.device->gsp->rm;
	struct r535_gr *gr = container_of(base, typeof(*gr), base);

	if (rm->api->gr->scrubber.fini)
		rm->api->gr->scrubber.fini(gr);

	return 0;
}

static int
nvkm_rm_gr_init(struct nvkm_gr *base)
{
	struct nvkm_rm *rm = base->engine.subdev.device->gsp->rm;
	struct r535_gr *gr = container_of(base, typeof(*gr), base);
	int ret;

	if (rm->api->gr->scrubber.init) {
		ret = rm->api->gr->scrubber.init(gr);
		if (ret)
			return ret;
	}

	return 0;
}

int
nvkm_rm_gr_new(struct nvkm_rm *rm)
{
	const u32 classes[] = {
		rm->gpu->gr.class.i2m,
		rm->gpu->gr.class.twod,
		rm->gpu->gr.class.threed,
		rm->gpu->gr.class.compute,
	};
	struct nvkm_gr_func *func;
	struct r535_gr *gr;

	func = kzalloc(struct_size(func, sclass, ARRAY_SIZE(classes) + 1), GFP_KERNEL);
	if (!func)
		return -ENOMEM;

	func->dtor = r535_gr_dtor;
	func->oneinit = r535_gr_oneinit;
	func->init = nvkm_rm_gr_init;
	func->fini = nvkm_rm_gr_fini;
	func->units = r535_gr_units;
	func->chan_new = r535_gr_chan_new;

	for (int i = 0; i < ARRAY_SIZE(classes); i++) {
		func->sclass[i].oclass = classes[i];
		func->sclass[i].minver = -1;
		func->sclass[i].maxver = 0;
		func->sclass[i].ctor = nvkm_rm_gr_obj_ctor;
	}

	gr = kzalloc(sizeof(*gr), GFP_KERNEL);
	if (!gr) {
		kfree(func);
		return -ENOMEM;
	}

	nvkm_gr_ctor(func, rm->device, NVKM_ENGINE_GR, 0, true, &gr->base);
	gr->scrubber.chid = -1;
	rm->device->gr = &gr->base;
	return 0;
}