// Copyright 2017 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <ddk/debug.h>
#include <ddk/device.h>
#include <ddk/protocol/platform-defs.h>
#include <hw/reg.h>
#include <soc/aml-a113/a113-hw.h>
#include <soc/aml-a113/aml-tdm.h>
#include <zircon/assert.h>
#include <limits.h>

#include "gauss.h"

#define PDM_MMIO_BASE 0xff632000
#define EE_AUDIO_MMIO_BASE 0xff642000
#define PDM_IRQ (85 + 32)

static const pbus_mmio_t audio_in_mmios[] = {
    {
        .base = EE_AUDIO_MMIO_BASE, .length = PAGE_SIZE,
    },
    {
        .base = PDM_MMIO_BASE, .length = PAGE_SIZE,
    },
};

static const pbus_irq_t audio_in_irqs[] = {
    {
        .irq = PDM_IRQ,
        .mode = ZX_INTERRUPT_MODE_EDGE_HIGH,
    },
};

static const pbus_bti_t audio_in_btis[] = {
    {
        .iommu_index = 0,
        .bti_id = BTI_AUDIO_IN,
    },
};

static const pbus_dev_t gauss_audio_in_dev = {
    .name = "gauss-audio-in",
    .vid = PDEV_VID_GOOGLE,
    .pid = PDEV_PID_GAUSS,
    .did = PDEV_DID_GAUSS_AUDIO_IN,
    .mmios = audio_in_mmios,
    .mmio_count = countof(audio_in_mmios),
    .irqs = audio_in_irqs,
    .irq_count = countof(audio_in_irqs),
    .btis = audio_in_btis,
    .bti_count = countof(audio_in_btis),
};

static const pbus_mmio_t tdm_audio_mmios[] = {
    {
        .base = A113_TDM_PHYS_BASE,
        .length = 4096
    },
};

static const pbus_i2c_channel_t tdm_i2cs[] = {
    {
        .bus_id = AML_I2C_B,
        .address = 0x4C
    },
    {
        .bus_id = AML_I2C_B,
        .address = 0x4D
    },
    {
        .bus_id = AML_I2C_B,
        .address = 0x4E
    },
};

static const pbus_irq_t tdm_irqs[] = {
    {
        .irq = (90 + 32),
        .mode = ZX_INTERRUPT_MODE_EDGE_HIGH,
    },
};

static const pbus_bti_t tdm_btis[] = {
    {
        .iommu_index = 0,
        .bti_id = BTI_AUDIO_OUT,
    },
};

static const pbus_dev_t gauss_tdm_audio_dev = {
    .name = "gauss-tdm-audio",
    .vid = PDEV_VID_GOOGLE,
    .pid = PDEV_PID_GAUSS,
    .did = PDEV_DID_GAUSS_AUDIO_OUT,
    .irqs = tdm_irqs,
    .irq_count = countof(tdm_irqs),
    .mmios = tdm_audio_mmios,
    .mmio_count = countof(tdm_audio_mmios),
    .i2c_channels = tdm_i2cs,
    .i2c_channel_count = countof(tdm_i2cs),
    .btis = tdm_btis,
    .bti_count = countof(tdm_btis),
};

zx_status_t gauss_audio_init(gauss_bus_t* bus) {

    ZX_DEBUG_ASSERT(bus);
    zx_status_t status;

    // Add audio in and out devices.
    if ((status = pbus_device_add(&bus->pbus, &gauss_audio_in_dev, 0)) != ZX_OK) {
        zxlogf(ERROR, "a113_audio_init could not add gauss_audio_in_dev: %d\n", status);
        return status;
    }

    printf("Adding the tdm device\n");
    if ((status = pbus_device_add(&bus->pbus, &gauss_tdm_audio_dev, 0)) != ZX_OK) {
        zxlogf(ERROR, "a113_audio_init could not add gauss_tdm_audio_dev: %d\n", status);
    }

    return ZX_OK;
}