/*
 * Intel Smart Sound Technology (SST) DSP Core Driver
 *
 * Copyright (C) 2013, Intel Corporation. All rights reserved.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License version
 * 2 as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 */

#include <linux/slab.h>
#include <linux/export.h>
#include <linux/interrupt.h>
#include <linux/module.h>
#include <linux/platform_device.h>
#include <linux/io.h>
#include <linux/delay.h>

#include "sst-dsp.h"
#include "sst-dsp-priv.h"

#define CREATE_TRACE_POINTS
#include <trace/events/intel-sst.h>

/* Internal generic low-level SST IO functions - can be overidden */
void sst_shim32_write(void __iomem *addr, u32 offset, u32 value)
{
        writel(value, addr + offset);
}
EXPORT_SYMBOL_GPL(sst_shim32_write);

u32 sst_shim32_read(void __iomem *addr, u32 offset)
{
        return readl(addr + offset);
}
EXPORT_SYMBOL_GPL(sst_shim32_read);

void sst_shim32_write64(void __iomem *addr, u32 offset, u64 value)
{
        memcpy_toio(addr + offset, &value, sizeof(value));
}
EXPORT_SYMBOL_GPL(sst_shim32_write64);

u64 sst_shim32_read64(void __iomem *addr, u32 offset)
{
        u64 val;

        memcpy_fromio(&val, addr + offset, sizeof(val));
        return val;
}
EXPORT_SYMBOL_GPL(sst_shim32_read64);

static inline void _sst_memcpy_toio_32(volatile u32 __iomem *dest,
        u32 *src, size_t bytes)
{
        int i, words = bytes >> 2;

        for (i = 0; i < words; i++)
                writel(src[i], dest + i);
}

static inline void _sst_memcpy_fromio_32(u32 *dest,
        const volatile __iomem u32 *src, size_t bytes)
{
        int i, words = bytes >> 2;

        for (i = 0; i < words; i++)
                dest[i] = readl(src + i);
}

void sst_memcpy_toio_32(struct sst_dsp *sst,
        void __iomem *dest, void *src, size_t bytes)
{
        _sst_memcpy_toio_32(dest, src, bytes);
}
EXPORT_SYMBOL_GPL(sst_memcpy_toio_32);

void sst_memcpy_fromio_32(struct sst_dsp *sst, void *dest,
        void __iomem *src, size_t bytes)
{
        _sst_memcpy_fromio_32(dest, src, bytes);
}
EXPORT_SYMBOL_GPL(sst_memcpy_fromio_32);

/* Public API */
void sst_dsp_shim_write(struct sst_dsp *sst, u32 offset, u32 value)
{
        unsigned long flags;

        spin_lock_irqsave(&sst->spinlock, flags);
        sst->ops->write(sst->addr.shim, offset, value);
        spin_unlock_irqrestore(&sst->spinlock, flags);
}
EXPORT_SYMBOL_GPL(sst_dsp_shim_write);

u32 sst_dsp_shim_read(struct sst_dsp *sst, u32 offset)
{
        unsigned long flags;
        u32 val;

        spin_lock_irqsave(&sst->spinlock, flags);
        val = sst->ops->read(sst->addr.shim, offset);
        spin_unlock_irqrestore(&sst->spinlock, flags);

        return val;
}
EXPORT_SYMBOL_GPL(sst_dsp_shim_read);

void sst_dsp_shim_write64(struct sst_dsp *sst, u32 offset, u64 value)
{
        unsigned long flags;

        spin_lock_irqsave(&sst->spinlock, flags);
        sst->ops->write64(sst->addr.shim, offset, value);
        spin_unlock_irqrestore(&sst->spinlock, flags);
}
EXPORT_SYMBOL_GPL(sst_dsp_shim_write64);

u64 sst_dsp_shim_read64(struct sst_dsp *sst, u32 offset)
{
        unsigned long flags;
        u64 val;

        spin_lock_irqsave(&sst->spinlock, flags);
        val = sst->ops->read64(sst->addr.shim, offset);
        spin_unlock_irqrestore(&sst->spinlock, flags);

        return val;
}
EXPORT_SYMBOL_GPL(sst_dsp_shim_read64);

void sst_dsp_shim_write_unlocked(struct sst_dsp *sst, u32 offset, u32 value)
{
        sst->ops->write(sst->addr.shim, offset, value);
}
EXPORT_SYMBOL_GPL(sst_dsp_shim_write_unlocked);

u32 sst_dsp_shim_read_unlocked(struct sst_dsp *sst, u32 offset)
{
        return sst->ops->read(sst->addr.shim, offset);
}
EXPORT_SYMBOL_GPL(sst_dsp_shim_read_unlocked);

void sst_dsp_shim_write64_unlocked(struct sst_dsp *sst, u32 offset, u64 value)
{
        sst->ops->write64(sst->addr.shim, offset, value);
}
EXPORT_SYMBOL_GPL(sst_dsp_shim_write64_unlocked);

u64 sst_dsp_shim_read64_unlocked(struct sst_dsp *sst, u32 offset)
{
        return sst->ops->read64(sst->addr.shim, offset);
}
EXPORT_SYMBOL_GPL(sst_dsp_shim_read64_unlocked);

int sst_dsp_shim_update_bits_unlocked(struct sst_dsp *sst, u32 offset,
                                u32 mask, u32 value)
{
        bool change;
        unsigned int old, new;
        u32 ret;

        ret = sst_dsp_shim_read_unlocked(sst, offset);

        old = ret;
        new = (old & (~mask)) | (value & mask);

        change = (old != new);
        if (change)
                sst_dsp_shim_write_unlocked(sst, offset, new);

        return change;
}
EXPORT_SYMBOL_GPL(sst_dsp_shim_update_bits_unlocked);

int sst_dsp_shim_update_bits64_unlocked(struct sst_dsp *sst, u32 offset,
                                u64 mask, u64 value)
{
        bool change;
        u64 old, new;

        old = sst_dsp_shim_read64_unlocked(sst, offset);

        new = (old & (~mask)) | (value & mask);

        change = (old != new);
        if (change)
                sst_dsp_shim_write64_unlocked(sst, offset, new);

        return change;
}
EXPORT_SYMBOL_GPL(sst_dsp_shim_update_bits64_unlocked);

/* This is for registers bits with attribute RWC */
void sst_dsp_shim_update_bits_forced_unlocked(struct sst_dsp *sst, u32 offset,
                                u32 mask, u32 value)
{
        unsigned int old, new;
        u32 ret;

        ret = sst_dsp_shim_read_unlocked(sst, offset);

        old = ret;
        new = (old & (~mask)) | (value & mask);

        sst_dsp_shim_write_unlocked(sst, offset, new);
}
EXPORT_SYMBOL_GPL(sst_dsp_shim_update_bits_forced_unlocked);

int sst_dsp_shim_update_bits(struct sst_dsp *sst, u32 offset,
                                u32 mask, u32 value)
{
        unsigned long flags;
        bool change;

        spin_lock_irqsave(&sst->spinlock, flags);
        change = sst_dsp_shim_update_bits_unlocked(sst, offset, mask, value);
        spin_unlock_irqrestore(&sst->spinlock, flags);
        return change;
}
EXPORT_SYMBOL_GPL(sst_dsp_shim_update_bits);

int sst_dsp_shim_update_bits64(struct sst_dsp *sst, u32 offset,
                                u64 mask, u64 value)
{
        unsigned long flags;
        bool change;

        spin_lock_irqsave(&sst->spinlock, flags);
        change = sst_dsp_shim_update_bits64_unlocked(sst, offset, mask, value);
        spin_unlock_irqrestore(&sst->spinlock, flags);
        return change;
}
EXPORT_SYMBOL_GPL(sst_dsp_shim_update_bits64);

/* This is for registers bits with attribute RWC */
void sst_dsp_shim_update_bits_forced(struct sst_dsp *sst, u32 offset,
                                u32 mask, u32 value)
{
        unsigned long flags;

        spin_lock_irqsave(&sst->spinlock, flags);
        sst_dsp_shim_update_bits_forced_unlocked(sst, offset, mask, value);
        spin_unlock_irqrestore(&sst->spinlock, flags);
}
EXPORT_SYMBOL_GPL(sst_dsp_shim_update_bits_forced);

int sst_dsp_register_poll(struct sst_dsp *ctx, u32 offset, u32 mask,
                         u32 target, u32 time, char *operation)
{
        u32 reg;
        unsigned long timeout;
        int k = 0, s = 500;

        /*
         * split the loop into sleeps of varying resolution. more accurately,
         * the range of wakeups are:
         * Phase 1(first 5ms): min sleep 0.5ms; max sleep 1ms.
         * Phase 2:( 5ms to 10ms) : min sleep 0.5ms; max sleep 10ms
         * (usleep_range (500, 1000) and usleep_range(5000, 10000) are
         * both possible in this phase depending on whether k > 10 or not).
         * Phase 3: (beyond 10 ms) min sleep 5ms; max sleep 10ms.
         */

        timeout = jiffies + msecs_to_jiffies(time);
        while ((((reg = sst_dsp_shim_read_unlocked(ctx, offset)) & mask) != target)
                && time_before(jiffies, timeout)) {
                k++;
                if (k > 10)
                        s = 5000;

                usleep_range(s, 2*s);
        }

        if ((reg & mask) == target) {
                dev_dbg(ctx->dev, "FW Poll Status: reg=%#x %s successful\n",
                                        reg, operation);

                return 0;
        }

        dev_dbg(ctx->dev, "FW Poll Status: reg=%#x %s timedout\n",
                                        reg, operation);
        return -ETIME;
}
EXPORT_SYMBOL_GPL(sst_dsp_register_poll);

void sst_dsp_dump(struct sst_dsp *sst)
{
        if (sst->ops->dump)
                sst->ops->dump(sst);
}
EXPORT_SYMBOL_GPL(sst_dsp_dump);

void sst_dsp_reset(struct sst_dsp *sst)
{
        if (sst->ops->reset)
                sst->ops->reset(sst);
}
EXPORT_SYMBOL_GPL(sst_dsp_reset);

int sst_dsp_boot(struct sst_dsp *sst)
{
        if (sst->ops->boot)
                sst->ops->boot(sst);

        return 0;
}
EXPORT_SYMBOL_GPL(sst_dsp_boot);

int sst_dsp_wake(struct sst_dsp *sst)
{
        if (sst->ops->wake)
                return sst->ops->wake(sst);

        return 0;
}
EXPORT_SYMBOL_GPL(sst_dsp_wake);

void sst_dsp_sleep(struct sst_dsp *sst)
{
        if (sst->ops->sleep)
                sst->ops->sleep(sst);
}
EXPORT_SYMBOL_GPL(sst_dsp_sleep);

void sst_dsp_stall(struct sst_dsp *sst)
{
        if (sst->ops->stall)
                sst->ops->stall(sst);
}
EXPORT_SYMBOL_GPL(sst_dsp_stall);

void sst_dsp_ipc_msg_tx(struct sst_dsp *dsp, u32 msg)
{
        sst_dsp_shim_write_unlocked(dsp, SST_IPCX, msg | SST_IPCX_BUSY);
        trace_sst_ipc_msg_tx(msg);
}
EXPORT_SYMBOL_GPL(sst_dsp_ipc_msg_tx);

u32 sst_dsp_ipc_msg_rx(struct sst_dsp *dsp)
{
        u32 msg;

        msg = sst_dsp_shim_read_unlocked(dsp, SST_IPCX);
        trace_sst_ipc_msg_rx(msg);

        return msg;
}
EXPORT_SYMBOL_GPL(sst_dsp_ipc_msg_rx);

int sst_dsp_mailbox_init(struct sst_dsp *sst, u32 inbox_offset, size_t inbox_size,
        u32 outbox_offset, size_t outbox_size)
{
        sst->mailbox.in_base = sst->addr.lpe + inbox_offset;
        sst->mailbox.out_base = sst->addr.lpe + outbox_offset;
        sst->mailbox.in_size = inbox_size;
        sst->mailbox.out_size = outbox_size;
        return 0;
}
EXPORT_SYMBOL_GPL(sst_dsp_mailbox_init);

void sst_dsp_outbox_write(struct sst_dsp *sst, void *message, size_t bytes)
{
        u32 i;

        trace_sst_ipc_outbox_write(bytes);

        memcpy_toio(sst->mailbox.out_base, message, bytes);

        for (i = 0; i < bytes; i += 4)
                trace_sst_ipc_outbox_wdata(i, *(u32 *)(message + i));
}
EXPORT_SYMBOL_GPL(sst_dsp_outbox_write);

void sst_dsp_outbox_read(struct sst_dsp *sst, void *message, size_t bytes)
{
        u32 i;

        trace_sst_ipc_outbox_read(bytes);

        memcpy_fromio(message, sst->mailbox.out_base, bytes);

        for (i = 0; i < bytes; i += 4)
                trace_sst_ipc_outbox_rdata(i, *(u32 *)(message + i));
}
EXPORT_SYMBOL_GPL(sst_dsp_outbox_read);

void sst_dsp_inbox_write(struct sst_dsp *sst, void *message, size_t bytes)
{
        u32 i;

        trace_sst_ipc_inbox_write(bytes);

        memcpy_toio(sst->mailbox.in_base, message, bytes);

        for (i = 0; i < bytes; i += 4)
                trace_sst_ipc_inbox_wdata(i, *(u32 *)(message + i));
}
EXPORT_SYMBOL_GPL(sst_dsp_inbox_write);

void sst_dsp_inbox_read(struct sst_dsp *sst, void *message, size_t bytes)
{
        u32 i;

        trace_sst_ipc_inbox_read(bytes);

        memcpy_fromio(message, sst->mailbox.in_base, bytes);

        for (i = 0; i < bytes; i += 4)
                trace_sst_ipc_inbox_rdata(i, *(u32 *)(message + i));
}
EXPORT_SYMBOL_GPL(sst_dsp_inbox_read);

/* Module information */
MODULE_AUTHOR("Liam Girdwood");
MODULE_DESCRIPTION("Intel SST Core");
MODULE_LICENSE("GPL v2");