// 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 <unistd.h>
#include <zircon/hw/gpt.h>

#include <ddk/debug.h>
#include <ddk/device.h>
#include <ddk/io-buffer.h>
#include <ddk/metadata.h>
#include <ddk/metadata/nand.h>
#include <ddk/platform-defs.h>
#include <ddk/protocol/gpio.h>
#include <ddk/protocol/platform/bus.h>
#include <hw/reg.h>
#include <soc/aml-a113/a113-hw.h>

#include "gauss-hw.h"
#include "gauss.h"

static const pbus_mmio_t raw_nand_mmios[] = {
    {
        /* nandreg : Registers for NAND controller */
        .base = GAUSS_RAW_NAND_REG,
        .length = 0x2000,
    },
    {
        /* clockreg : Clock Register for NAND controller */
        /*
         * From the Linux devicetree. This is the base SD_EMMC_CLOCK
         * register (for port C)
         */
        .base = GAUSS_RAW_NAND_CLKREG,
        .length = 0x4, /* Just 4 bytes */
    },
};

static const pbus_irq_t raw_nand_irqs[] = {
    {
        .irq = GAUSS_RAW_NAND_IRQ,
    },
};

static const pbus_bti_t raw_nand_btis[] = {
    {
        .iommu_index = 0,
        .bti_id = BTI_AML_RAW_NAND,
    },
};

static const nand_config_t config = {
    .bad_block_config =
        {
            .type = kAmlogicUboot,
            .aml_uboot =
                {
                    .table_start_block = 20,
                    .table_end_block = 23,
                },
        },
    .extra_partition_config_count = 1,
    .extra_partition_config =
        {
            {
                .type_guid = GUID_BOOTLOADER_VALUE,
                .copy_count = 4,
            },
        },
};

static const pbus_metadata_t raw_nand_metadata[] = {
    {
        .type = DEVICE_METADATA_PRIVATE,
        .data_buffer = &config,
        .data_size = sizeof(config),
    },
};

static const pbus_boot_metadata_t raw_nand_boot_metadata[] = {
    {
        .zbi_type = DEVICE_METADATA_PARTITION_MAP,
        .zbi_extra = 0,
    },
};

static const pbus_dev_t raw_nand_dev = {
    .name = "aml_raw_nand",
    .vid = PDEV_VID_AMLOGIC,
    .pid = PDEV_PID_GENERIC,
    .did = PDEV_DID_AMLOGIC_RAW_NAND,
    .mmio_list = raw_nand_mmios,
    .mmio_count = countof(raw_nand_mmios),
    .irq_list = raw_nand_irqs,
    .irq_count = countof(raw_nand_irqs),
    .bti_list = raw_nand_btis,
    .bti_count = countof(raw_nand_btis),
    .metadata_list = raw_nand_metadata,
    .metadata_count = countof(raw_nand_metadata),
    .boot_metadata_list = raw_nand_boot_metadata,
    .boot_metadata_count = countof(raw_nand_boot_metadata),
};

zx_status_t gauss_raw_nand_init(gauss_bus_t* bus) {
  zx_status_t status;

  // set alternate functions to enable raw_nand
  status = gpio_impl_set_alt_function(&bus->gpio, A113_GPIOBOOT(8), 2);
  if (status != ZX_OK)
    return status;
  status = gpio_impl_set_alt_function(&bus->gpio, A113_GPIOBOOT(9), 2);
  if (status != ZX_OK)
    return status;
  status = gpio_impl_set_alt_function(&bus->gpio, A113_GPIOBOOT(10), 2);
  if (status != ZX_OK)
    return status;
  status = gpio_impl_set_alt_function(&bus->gpio, A113_GPIOBOOT(11), 2);
  if (status != ZX_OK)
    return status;
  status = gpio_impl_set_alt_function(&bus->gpio, A113_GPIOBOOT(12), 2);
  if (status != ZX_OK)
    return status;
  status = gpio_impl_set_alt_function(&bus->gpio, A113_GPIOBOOT(13), 2);
  if (status != ZX_OK)
    return status;

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

  return ZX_OK;
}