// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (C) 2014 Google, Inc.
 */

#include <linux/clk.h>
#include <linux/clk-provider.h>
#include <linux/kernel.h>
#include <linux/of.h>
#include <linux/of_address.h>
#include <linux/slab.h>

#include "clk.h"

struct pistachio_clk_provider *
pistachio_clk_alloc_provider(struct device_node *node, unsigned int num_clks)
{
        struct pistachio_clk_provider *p;

        p = kzalloc_obj(*p);
        if (!p)
                return p;

        p->clk_data.clks = kzalloc_objs(struct clk *, num_clks);
        if (!p->clk_data.clks)
                goto free_provider;
        p->clk_data.clk_num = num_clks;
        p->node = node;
        p->base = of_iomap(node, 0);
        if (!p->base) {
                pr_err("Failed to map clock provider registers\n");
                goto free_clks;
        }

        return p;

free_clks:
        kfree(p->clk_data.clks);
free_provider:
        kfree(p);
        return NULL;
}

void pistachio_clk_register_provider(struct pistachio_clk_provider *p)
{
        unsigned int i;

        for (i = 0; i < p->clk_data.clk_num; i++) {
                if (IS_ERR(p->clk_data.clks[i]))
                        pr_warn("Failed to register clock %d: %ld\n", i,
                                PTR_ERR(p->clk_data.clks[i]));
        }

        of_clk_add_provider(p->node, of_clk_src_onecell_get, &p->clk_data);
}

void pistachio_clk_register_gate(struct pistachio_clk_provider *p,
                                 struct pistachio_gate *gate,
                                 unsigned int num)
{
        struct clk *clk;
        unsigned int i;

        for (i = 0; i < num; i++) {
                clk = clk_register_gate(NULL, gate[i].name, gate[i].parent,
                                        CLK_SET_RATE_PARENT,
                                        p->base + gate[i].reg, gate[i].shift,
                                        0, NULL);
                p->clk_data.clks[gate[i].id] = clk;
        }
}

void pistachio_clk_register_mux(struct pistachio_clk_provider *p,
                                struct pistachio_mux *mux,
                                unsigned int num)
{
        struct clk *clk;
        unsigned int i;

        for (i = 0; i < num; i++) {
                clk = clk_register_mux(NULL, mux[i].name, mux[i].parents,
                                       mux[i].num_parents,
                                       CLK_SET_RATE_NO_REPARENT,
                                       p->base + mux[i].reg, mux[i].shift,
                                       get_count_order(mux[i].num_parents),
                                       0, NULL);
                p->clk_data.clks[mux[i].id] = clk;
        }
}

void pistachio_clk_register_div(struct pistachio_clk_provider *p,
                                struct pistachio_div *div,
                                unsigned int num)
{
        struct clk *clk;
        unsigned int i;

        for (i = 0; i < num; i++) {
                clk = clk_register_divider(NULL, div[i].name, div[i].parent,
                                           0, p->base + div[i].reg, 0,
                                           div[i].width, div[i].div_flags,
                                           NULL);
                p->clk_data.clks[div[i].id] = clk;
        }
}

void pistachio_clk_register_fixed_factor(struct pistachio_clk_provider *p,
                                         struct pistachio_fixed_factor *ff,
                                         unsigned int num)
{
        struct clk *clk;
        unsigned int i;

        for (i = 0; i < num; i++) {
                clk = clk_register_fixed_factor(NULL, ff[i].name, ff[i].parent,
                                                0, 1, ff[i].div);
                p->clk_data.clks[ff[i].id] = clk;
        }
}

void pistachio_clk_force_enable(struct pistachio_clk_provider *p,
                                unsigned int *clk_ids, unsigned int num)
{
        unsigned int i;
        int err;

        for (i = 0; i < num; i++) {
                struct clk *clk = p->clk_data.clks[clk_ids[i]];

                if (IS_ERR(clk))
                        continue;

                err = clk_prepare_enable(clk);
                if (err)
                        pr_err("Failed to enable clock %s: %d\n",
                               __clk_get_name(clk), err);
        }
}