// Copyright 2018 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/metadata.h>
#include <ddk/metadata/i2c.h>
#include <ddk/platform-defs.h>
#include <soc/aml-s905d2/s905d2-gpio.h>
#include <soc/aml-s905d2/s905d2-hw.h>

#include <limits.h>

#include "astro.h"

static const pbus_mmio_t i2c_mmios[] = {
    {
        .base = S905D2_I2C_AO_0_BASE,
        .length = 0x20,
    },
    {
        .base = S905D2_I2C2_BASE,
        .length = 0x20,
    },
    {
        .base = S905D2_I2C3_BASE,
        .length = 0x20,
    },
};

static const pbus_irq_t i2c_irqs[] = {
    {
        .irq = S905D2_I2C_AO_0_IRQ,
        .mode = ZX_INTERRUPT_MODE_EDGE_HIGH,
    },
    {
        .irq = S905D2_I2C2_IRQ,
        .mode = ZX_INTERRUPT_MODE_EDGE_HIGH,
    },
    {
        .irq = S905D2_I2C3_IRQ,
        .mode = ZX_INTERRUPT_MODE_EDGE_HIGH,
    },
};

static const i2c_channel_t i2c_channels[] = {
    // Backlight I2C
    {
        .bus_id = ASTRO_I2C_3,
        .address = I2C_BACKLIGHT_ADDR,
        .vid = PDEV_VID_TI,
        .pid = PDEV_PID_TI_LP8556,
        .did = PDEV_DID_TI_BACKLIGHT,
    },
    // Focaltech touch screen
    {
        .bus_id = ASTRO_I2C_2,
        .address = I2C_FOCALTECH_TOUCH_ADDR,
        // binds as composite device
    },
    // Goodix touch screen
    {
        .bus_id = ASTRO_I2C_2,
        .address = I2C_GOODIX_TOUCH_ADDR,
        // binds as composite device
    },
    // Light sensor
    {
        .bus_id = ASTRO_I2C_A0_0,
        .address = I2C_AMBIENTLIGHT_ADDR,
        // binds as composite device
    },
    // Audio output
    {
        .bus_id = ASTRO_I2C_3,
        .address = I2C_AUDIO_CODEC_ADDR,
        // binds as composite device
    },
};

static const pbus_metadata_t i2c_metadata[] = {
    {
        .type = DEVICE_METADATA_I2C_CHANNELS,
        .data_buffer = &i2c_channels,
        .data_size = sizeof(i2c_channels),
    }
};

static const pbus_dev_t i2c_dev = {
    .name = "i2c",
    .vid = PDEV_VID_AMLOGIC,
    .pid = PDEV_PID_GENERIC,
    .did = PDEV_DID_AMLOGIC_I2C,
    .mmio_list = i2c_mmios,
    .mmio_count = countof(i2c_mmios),
    .irq_list = i2c_irqs,
    .irq_count = countof(i2c_irqs),
    .metadata_list = i2c_metadata,
    .metadata_count = countof(i2c_metadata),
};

zx_status_t aml_i2c_init(aml_bus_t* bus) {
    // setup pinmux for our I2C busses

    //i2c_ao_0
    gpio_impl_set_alt_function(&bus->gpio, S905D2_GPIOAO(2), 1);
    gpio_impl_set_alt_function(&bus->gpio, S905D2_GPIOAO(3), 1);
    //i2c2
    gpio_impl_set_alt_function(&bus->gpio, S905D2_GPIOZ(14), 3);
    gpio_impl_set_alt_function(&bus->gpio, S905D2_GPIOZ(15), 3);
    //i2c3
    gpio_impl_set_alt_function(&bus->gpio, S905D2_GPIOA(14), 2);
    gpio_impl_set_alt_function(&bus->gpio, S905D2_GPIOA(15), 2);

    zx_status_t status = pbus_device_add(&bus->pbus, &i2c_dev);
    if (status != ZX_OK) {
        zxlogf(ERROR, "aml_i2c_init: pbus_device_add failed: %d\n", status);
        return status;
    }

    return ZX_OK;
}