/*
 * Copyright 2012 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 <core/object.h>
#include <core/client.h>
#include <core/engine.h>

struct nvkm_object *
nvkm_object_search(struct nvkm_client *client, u64 handle,
                   const struct nvkm_object_func *func)
{
        struct nvkm_object *object;
        unsigned long flags;

        if (handle) {
                spin_lock_irqsave(&client->obj_lock, flags);
                struct rb_node *node = client->objroot.rb_node;
                while (node) {
                        object = rb_entry(node, typeof(*object), node);
                        if (handle < object->object)
                                node = node->rb_left;
                        else
                        if (handle > object->object)
                                node = node->rb_right;
                        else {
                                spin_unlock_irqrestore(&client->obj_lock, flags);
                                goto done;
                        }
                }
                spin_unlock_irqrestore(&client->obj_lock, flags);
                return ERR_PTR(-ENOENT);
        } else {
                object = &client->object;
        }

done:
        if (unlikely(func && object->func != func))
                return ERR_PTR(-EINVAL);
        return object;
}

void
nvkm_object_remove(struct nvkm_object *object)
{
        unsigned long flags;

        spin_lock_irqsave(&object->client->obj_lock, flags);
        if (!RB_EMPTY_NODE(&object->node))
                rb_erase(&object->node, &object->client->objroot);
        spin_unlock_irqrestore(&object->client->obj_lock, flags);
}

bool
nvkm_object_insert(struct nvkm_object *object)
{
        struct rb_node **ptr;
        struct rb_node *parent = NULL;
        unsigned long flags;

        spin_lock_irqsave(&object->client->obj_lock, flags);
        ptr = &object->client->objroot.rb_node;
        while (*ptr) {
                struct nvkm_object *this = rb_entry(*ptr, typeof(*this), node);
                parent = *ptr;
                if (object->object < this->object) {
                        ptr = &parent->rb_left;
                } else if (object->object > this->object) {
                        ptr = &parent->rb_right;
                } else {
                        spin_unlock_irqrestore(&object->client->obj_lock, flags);
                        return false;
                }
        }

        rb_link_node(&object->node, parent, ptr);
        rb_insert_color(&object->node, &object->client->objroot);
        spin_unlock_irqrestore(&object->client->obj_lock, flags);
        return true;
}

int
nvkm_object_mthd(struct nvkm_object *object, u32 mthd, void *data, u32 size)
{
        if (likely(object->func->mthd))
                return object->func->mthd(object, mthd, data, size);
        return -ENODEV;
}

int
nvkm_object_ntfy(struct nvkm_object *object, u32 mthd,
                 struct nvkm_event **pevent)
{
        if (likely(object->func->ntfy))
                return object->func->ntfy(object, mthd, pevent);
        return -ENODEV;
}

int
nvkm_object_map(struct nvkm_object *object, void *argv, u32 argc,
                enum nvkm_object_map *type, u64 *addr, u64 *size)
{
        if (likely(object->func->map))
                return object->func->map(object, argv, argc, type, addr, size);
        return -ENODEV;
}

int
nvkm_object_unmap(struct nvkm_object *object)
{
        if (likely(object->func->unmap))
                return object->func->unmap(object);
        return -ENODEV;
}

int
nvkm_object_bind(struct nvkm_object *object, struct nvkm_gpuobj *gpuobj,
                 int align, struct nvkm_gpuobj **pgpuobj)
{
        if (object->func->bind)
                return object->func->bind(object, gpuobj, align, pgpuobj);
        return -ENODEV;
}

int
nvkm_object_fini(struct nvkm_object *object, enum nvkm_suspend_state suspend)
{
        const char *action;
        struct nvkm_object *child;
        s64 time;
        int ret;

        switch (suspend) {
        case NVKM_POWEROFF:
        default:
                action = "fini";
                break;
        case NVKM_SUSPEND:
                action = "suspend";
                break;
        case NVKM_RUNTIME_SUSPEND:
                action = "runtime";
                break;
        }
        nvif_debug(object, "%s children...\n", action);
        time = ktime_to_us(ktime_get());
        list_for_each_entry_reverse(child, &object->tree, head) {
                ret = nvkm_object_fini(child, suspend);
                if (ret && suspend)
                        goto fail_child;
        }

        nvif_debug(object, "%s running...\n", action);
        if (object->func->fini) {
                ret = object->func->fini(object, suspend);
                if (ret) {
                        nvif_error(object, "%s failed with %d\n", action, ret);
                        if (suspend)
                                goto fail;
                }
        }

        time = ktime_to_us(ktime_get()) - time;
        nvif_debug(object, "%s completed in %lldus\n", action, time);
        return 0;

fail:
        if (object->func->init) {
                int rret = object->func->init(object);
                if (rret)
                        nvif_fatal(object, "failed to restart, %d\n", rret);
        }
fail_child:
        list_for_each_entry_continue_reverse(child, &object->tree, head) {
                nvkm_object_init(child);
        }
        return ret;
}

int
nvkm_object_init(struct nvkm_object *object)
{
        struct nvkm_object *child;
        s64 time;
        int ret;

        nvif_debug(object, "init running...\n");
        time = ktime_to_us(ktime_get());
        if (object->func->init) {
                ret = object->func->init(object);
                if (ret)
                        goto fail;
        }

        nvif_debug(object, "init children...\n");
        list_for_each_entry(child, &object->tree, head) {
                ret = nvkm_object_init(child);
                if (ret)
                        goto fail_child;
        }

        time = ktime_to_us(ktime_get()) - time;
        nvif_debug(object, "init completed in %lldus\n", time);
        return 0;

fail_child:
        list_for_each_entry_continue_reverse(child, &object->tree, head)
                nvkm_object_fini(child, NVKM_POWEROFF);
fail:
        nvif_error(object, "init failed with %d\n", ret);
        if (object->func->fini)
                object->func->fini(object, NVKM_POWEROFF);
        return ret;
}

void *
nvkm_object_dtor(struct nvkm_object *object)
{
        struct nvkm_object *child, *ctemp;
        void *data = object;
        s64 time;

        nvif_debug(object, "destroy children...\n");
        time = ktime_to_us(ktime_get());
        list_for_each_entry_safe(child, ctemp, &object->tree, head) {
                nvkm_object_del(&child);
        }

        nvif_debug(object, "destroy running...\n");
        nvkm_object_unmap(object);
        if (object->func->dtor)
                data = object->func->dtor(object);
        nvkm_engine_unref(&object->engine);
        time = ktime_to_us(ktime_get()) - time;
        nvif_debug(object, "destroy completed in %lldus...\n", time);
        return data;
}

void
nvkm_object_del(struct nvkm_object **pobject)
{
        struct nvkm_object *object = *pobject;
        if (object && !WARN_ON(!object->func)) {
                *pobject = nvkm_object_dtor(object);
                nvkm_object_remove(object);
                list_del(&object->head);
                kfree(*pobject);
                *pobject = NULL;
        }
}

void
nvkm_object_ctor(const struct nvkm_object_func *func,
                 const struct nvkm_oclass *oclass, struct nvkm_object *object)
{
        object->func = func;
        object->client = oclass->client;
        object->engine = nvkm_engine_ref(oclass->engine);
        object->oclass = oclass->base.oclass;
        object->handle = oclass->handle;
        object->object = oclass->object;
        INIT_LIST_HEAD(&object->head);
        INIT_LIST_HEAD(&object->tree);
        RB_CLEAR_NODE(&object->node);
        WARN_ON(IS_ERR(object->engine));
}

int
nvkm_object_new_(const struct nvkm_object_func *func,
                 const struct nvkm_oclass *oclass, void *data, u32 size,
                 struct nvkm_object **pobject)
{
        if (size == 0) {
                if (!(*pobject = kzalloc_obj(**pobject)))
                        return -ENOMEM;
                nvkm_object_ctor(func, oclass, *pobject);
                return 0;
        }
        return -ENOSYS;
}

static const struct nvkm_object_func
nvkm_object_func = {
};

int
nvkm_object_new(const struct nvkm_oclass *oclass, void *data, u32 size,
                struct nvkm_object **pobject)
{
        const struct nvkm_object_func *func =
                oclass->base.func ? oclass->base.func : &nvkm_object_func;
        return nvkm_object_new_(func, oclass, data, size, pobject);
}