/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 2018 Emmanuel Vadot <manu@freebsd.org>
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/param.h>
#include <sys/systm.h>
#include <sys/bus.h>
#include <sys/rman.h>
#include <sys/kernel.h>
#include <sys/module.h>
#include <machine/bus.h>

#include <dev/fdt/simplebus.h>

#include <dev/ofw/ofw_bus.h>
#include <dev/ofw/ofw_bus_subr.h>

#ifdef __aarch64__
#include "opt_soc.h"
#endif

#include <dev/clk/clk_div.h>
#include <dev/clk/clk_fixed.h>
#include <dev/clk/clk_mux.h>

#include <dev/hwreset/hwreset.h>

#include <dev/clk/allwinner/aw_ccung.h>

#include <dt-bindings/clock/sun8i-de2.h>
#include <dt-bindings/reset/sun8i-de2.h>

enum CCU_DE2 {
	H3_CCU = 1,
	A64_CCU,
};

/* Non exported clocks */
#define	CLK_MIXER0_DIV	3
#define	CLK_MIXER1_DIV	4
#define	CLK_WB_DIV	5

static struct aw_ccung_reset h3_de2_ccu_resets[] = {
	CCU_RESET(RST_MIXER0, 0x08, 0)
	CCU_RESET(RST_WB, 0x08, 2)
};

static struct aw_ccung_reset a64_de2_ccu_resets[] = {
	CCU_RESET(RST_MIXER0, 0x08, 0)
	CCU_RESET(RST_MIXER1, 0x08, 1)
	CCU_RESET(RST_WB, 0x08, 2)
};

static struct aw_ccung_gate h3_de2_ccu_gates[] = {
	CCU_GATE(CLK_BUS_MIXER0, "mixer0", "mixer0-div", 0x00, 0)
	CCU_GATE(CLK_BUS_WB, "wb", "wb-div", 0x00, 2)

	CCU_GATE(CLK_MIXER0, "bus-mixer0", "bus-de", 0x04, 0)
	CCU_GATE(CLK_WB, "bus-wb", "bus-de", 0x04, 2)
};

static struct aw_ccung_gate a64_de2_ccu_gates[] = {
	CCU_GATE(CLK_BUS_MIXER0, "mixer0", "mixer0-div", 0x00, 0)
	CCU_GATE(CLK_BUS_MIXER1, "mixer1", "mixer1-div", 0x00, 1)
	CCU_GATE(CLK_BUS_WB, "wb", "wb-div", 0x00, 2)

	CCU_GATE(CLK_MIXER0, "bus-mixer0", "bus-de", 0x04, 0)
	CCU_GATE(CLK_MIXER1, "bus-mixer1", "bus-de", 0x04, 1)
	CCU_GATE(CLK_WB, "bus-wb", "bus-de", 0x04, 2)
};

static const char *div_parents[] = {"de"};

NM_CLK(mixer0_div_clk,
    CLK_MIXER0_DIV,			/* id */
    "mixer0-div", div_parents,		/* names, parents */
    0x0C,				/* offset */
    0, 0, 1, AW_CLK_FACTOR_FIXED,	/* N factor (fake)*/
    0, 4, 0, 0,				/* M flags */
    0, 0,				/* mux */
    0,					/* gate */
    AW_CLK_SCALE_CHANGE);	/* flags */

NM_CLK(mixer1_div_clk,
    CLK_MIXER1_DIV,			/* id */
    "mixer1-div", div_parents,		/* names, parents */
    0x0C,				/* offset */
    0, 0, 1, AW_CLK_FACTOR_FIXED,	/* N factor (fake)*/
    4, 4, 0, 0,				/* M flags */
    0, 0,				/* mux */
    0,					/* gate */
    AW_CLK_SCALE_CHANGE);		/* flags */

NM_CLK(wb_div_clk,
    CLK_WB_DIV,				/* id */
    "wb-div", div_parents,		/* names, parents */
    0x0C,				/* offset */
    0, 0, 1, AW_CLK_FACTOR_FIXED,	/* N factor (fake)*/
    8, 4, 0, 0,				/* M flags */
    0, 0,				/* mux */
    0,					/* gate */
    AW_CLK_SCALE_CHANGE);		/* flags */

static struct aw_ccung_clk h3_de2_ccu_clks[] = {
	{ .type = AW_CLK_NM, .clk.nm = &mixer0_div_clk},
	{ .type = AW_CLK_NM, .clk.nm = &wb_div_clk},
};

static struct aw_ccung_clk a64_de2_ccu_clks[] = {
	{ .type = AW_CLK_NM, .clk.nm = &mixer0_div_clk},
	{ .type = AW_CLK_NM, .clk.nm = &mixer1_div_clk},
	{ .type = AW_CLK_NM, .clk.nm = &wb_div_clk},
};

static struct ofw_compat_data compat_data[] = {
	{"allwinner,sun8i-h3-de2-clk", H3_CCU},
	{"allwinner,sun50i-a64-de2-clk", A64_CCU},
	{NULL,             0}
};

static int
ccu_de2_probe(device_t dev)
{

	if (!ofw_bus_status_okay(dev))
		return (ENXIO);

	if (ofw_bus_search_compatible(dev, compat_data)->ocd_data == 0)
		return (ENXIO);

	device_set_desc(dev, "Allwinner DE2 Clock Control Unit");
	return (BUS_PROBE_DEFAULT);
}

static int
ccu_de2_attach(device_t dev)
{
	struct aw_ccung_softc *sc;
	phandle_t node;
	clk_t mod, bus;
	hwreset_t rst_de;
	enum CCU_DE2 type;

	sc = device_get_softc(dev);
	node = ofw_bus_get_node(dev);

	type = (enum CCU_DE2)ofw_bus_search_compatible(dev, compat_data)->ocd_data;

	switch (type) {
	case H3_CCU:
		sc->resets = h3_de2_ccu_resets;
		sc->nresets = nitems(h3_de2_ccu_resets);
		sc->gates = h3_de2_ccu_gates;
		sc->ngates = nitems(h3_de2_ccu_gates);
		sc->clks = h3_de2_ccu_clks;
		sc->nclks = nitems(h3_de2_ccu_clks);
		break;
	case A64_CCU:
		sc->resets = a64_de2_ccu_resets;
		sc->nresets = nitems(a64_de2_ccu_resets);
		sc->gates = a64_de2_ccu_gates;
		sc->ngates = nitems(a64_de2_ccu_gates);
		sc->clks = a64_de2_ccu_clks;
		sc->nclks = nitems(a64_de2_ccu_clks);
		break;
	}

	if (hwreset_get_by_ofw_idx(dev, node, 0, &rst_de) != 0) {
		device_printf(dev, "Cannot get de reset\n");
		return (ENXIO);
	}
	if (hwreset_deassert(rst_de) != 0) {
		device_printf(dev, "Cannot de-assert de reset\n");
		return (ENXIO);
	}

	if (clk_get_by_ofw_name(dev, node, "mod", &mod) != 0) {
		device_printf(dev, "Cannot get mod clock\n");
		return (ENXIO);
	}
	if (clk_enable(mod) != 0) {
		device_printf(dev, "Cannot enable mod clock\n");
		return (ENXIO);
	}

	if (clk_get_by_ofw_name(dev, node, "bus", &bus) != 0) {
		device_printf(dev, "Cannot get bus clock\n");
		return (ENXIO);
	}
	if (clk_enable(bus) != 0) {
		device_printf(dev, "Cannot enable bus clock\n");
		return (ENXIO);
	}

	return (aw_ccung_attach(dev));
}

static device_method_t ccu_de2_methods[] = {
	/* Device interface */
	DEVMETHOD(device_probe,		ccu_de2_probe),
	DEVMETHOD(device_attach,	ccu_de2_attach),

	DEVMETHOD_END
};

DEFINE_CLASS_1(ccu_de2, ccu_de2_driver, ccu_de2_methods,
  sizeof(struct aw_ccung_softc), aw_ccung_driver);

EARLY_DRIVER_MODULE(ccu_de2, simplebus, ccu_de2_driver, 0, 0,
    BUS_PASS_RESOURCE + BUS_PASS_ORDER_LAST);