/*
 * Copyright 2016 Red Hat Inc.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 *
 * Authors: Ben Skeggs
 */
#include "priv.h"
#include "cgrp.h"
#include "chan.h"
#include "chid.h"
#include "runl.h"

#include <core/memory.h>
#include <subdev/timer.h>

#include <nvif/class.h>

void
gk110_chan_preempt(struct nvkm_chan *chan)
{
 struct nvkm_cgrp *cgrp = chan->cgrp;

 if (cgrp->hw) {
  cgrp->func->preempt(cgrp);
  return;
 }

 gf100_chan_preempt(chan);
}

const struct nvkm_chan_func
gk110_chan = {
 .inst = &gf100_chan_inst,
 .userd = &gk104_chan_userd,
 .ramfc = &gk104_chan_ramfc,
 .bind = gk104_chan_bind,
 .unbind = gk104_chan_unbind,
 .start = gk104_chan_start,
 .stop = gk104_chan_stop,
 .preempt = gk110_chan_preempt,
};

static void
gk110_cgrp_preempt(struct nvkm_cgrp *cgrp)
{
 nvkm_wr32(cgrp->runl->fifo->engine.subdev.device, 0x002634, 0x01000000 | cgrp->id);
}

const struct nvkm_cgrp_func
gk110_cgrp = {
 .preempt = gk110_cgrp_preempt,
};

void
gk110_runl_insert_cgrp(struct nvkm_cgrp *cgrp, struct nvkm_memory *memory, u64 offset)
{
 nvkm_wo32(memory, offset + 0, (cgrp->chan_nr << 26) | (128 << 18) |
          (3 << 14) | 0x00002000 | cgrp->id);
 nvkm_wo32(memory, offset + 4, 0x00000000);
}

const struct nvkm_runl_func
gk110_runl = {
 .size = 8,
 .update = nv50_runl_update,
 .insert_cgrp = gk110_runl_insert_cgrp,
 .insert_chan = gk104_runl_insert_chan,
 .commit = gk104_runl_commit,
 .wait = nv50_runl_wait,
 .pending = gk104_runl_pending,
 .block = gk104_runl_block,
 .allow = gk104_runl_allow,
 .fault_clear = gk104_runl_fault_clear,
 .preempt_pending = gf100_runl_preempt_pending,
};

int
gk110_fifo_chid_ctor(struct nvkm_fifo *fifo, int nr)
{
 int ret;

 ret = nvkm_chid_new(&nvkm_chan_event, &fifo->engine.subdev, nr, 0, nr, &fifo->cgid);
 if (ret)
  return ret;

 return gf100_fifo_chid_ctor(fifo, nr);
}

static const struct nvkm_fifo_func
gk110_fifo = {
 .chid_nr = gk104_fifo_chid_nr,
 .chid_ctor = gk110_fifo_chid_ctor,
 .runq_nr = gf100_fifo_runq_nr,
 .runl_ctor = gk104_fifo_runl_ctor,
 .init = gk104_fifo_init,
 .init_pbdmas = gk104_fifo_init_pbdmas,
 .intr = gk104_fifo_intr,
 .intr_mmu_fault_unit = gf100_fifo_intr_mmu_fault_unit,
 .intr_ctxsw_timeout = gf100_fifo_intr_ctxsw_timeout,
 .mmu_fault = &gk104_fifo_mmu_fault,
 .nonstall = &gf100_fifo_nonstall,
 .runl = &gk110_runl,
 .runq = &gk104_runq,
 .engn = &gk104_engn,
 .engn_ce = &gk104_engn_ce,
 .cgrp = {{ 0, 0, KEPLER_CHANNEL_GROUP_A  }, &gk110_cgrp },
 .chan = {{ 0, 0, KEPLER_CHANNEL_GPFIFO_B }, &gk110_chan },
};

int
gk110_fifo_new(struct nvkm_device *device, enum nvkm_subdev_type type, int inst,
        struct nvkm_fifo **pfifo)
{
 return nvkm_fifo_new_(&gk110_fifo, device, type, inst, pfifo);
}