// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (c) 2014, The Linux Foundation. All rights reserved.
 */

#include <linux/kernel.h>
#include <linux/bitops.h>
#include <linux/regmap.h>
#include <linux/export.h>

#include "clk-regmap-mux.h"

static inline struct clk_regmap_mux *to_clk_regmap_mux(struct clk_hw *hw)
{
 return container_of(to_clk_regmap(hw), struct clk_regmap_mux, clkr);
}

static u8 mux_get_parent(struct clk_hw *hw)
{
 struct clk_regmap_mux *mux = to_clk_regmap_mux(hw);
 struct clk_regmap *clkr = to_clk_regmap(hw);
 unsigned int mask = GENMASK(mux->width - 1, 0);
 unsigned int val;

 regmap_read(clkr->regmap, mux->reg, &val);

 val >>= mux->shift;
 val &= mask;

 if (mux->parent_map)
  return qcom_find_cfg_index(hw, mux->parent_map, val);

 return val;
}

static int mux_set_parent(struct clk_hw *hw, u8 index)
{
 struct clk_regmap_mux *mux = to_clk_regmap_mux(hw);
 struct clk_regmap *clkr = to_clk_regmap(hw);
 unsigned int mask = GENMASK(mux->width + mux->shift - 1, mux->shift);
 unsigned int val;

 if (mux->parent_map)
  index = mux->parent_map[index].cfg;

 val = index;
 val <<= mux->shift;

 return regmap_update_bits(clkr->regmap, mux->reg, mask, val);
}

const struct clk_ops clk_regmap_mux_closest_ops = {
 .get_parent = mux_get_parent,
 .set_parent = mux_set_parent,
 .determine_rate = __clk_mux_determine_rate_closest,
};
EXPORT_SYMBOL_GPL(clk_regmap_mux_closest_ops);