#ifndef __NVIF_MMU_H__
#define __NVIF_MMU_H__
#include <nvif/object.h>

struct nvif_mmu {
        struct nvif_object object;
        u8  dmabits;
        u8  heap_nr;
        u8  type_nr;
        u8  kind_inv;
        u16 kind_nr;
        s32 mem;

        struct {
                u64 size;
        } *heap;

        struct {
#define NVIF_MEM_VRAM                                                      0x01
#define NVIF_MEM_HOST                                                      0x02
#define NVIF_MEM_COMP                                                      0x04
#define NVIF_MEM_DISP                                                      0x08
#define NVIF_MEM_KIND                                                      0x10
#define NVIF_MEM_MAPPABLE                                                  0x20
#define NVIF_MEM_COHERENT                                                  0x40
#define NVIF_MEM_UNCACHED                                                  0x80
                u8 type;
                u8 heap;
        } *type;

        u8 *kind;
};

int nvif_mmu_ctor(struct nvif_object *, const char *name, s32 oclass,
                  struct nvif_mmu *);
void nvif_mmu_dtor(struct nvif_mmu *);

static inline bool
nvif_mmu_kind_valid(struct nvif_mmu *mmu, u8 kind)
{
        if (kind) {
                if (kind >= mmu->kind_nr || mmu->kind[kind] == mmu->kind_inv)
                        return false;
        }
        return true;
}

static inline int
nvif_mmu_type(struct nvif_mmu *mmu, u8 mask)
{
        int i;
        for (i = 0; i < mmu->type_nr; i++) {
                if ((mmu->type[i].type & mask) == mask)
                        return i;
        }
        return -EINVAL;
}
#endif