/*	$OpenBSD: ofw_power.c,v 1.2 2021/11/26 11:44:01 kettenis Exp $	*/
/*
 * Copyright (c) 2016 Mark Kettenis
 * Copyright (c) 2018 Patrick Wildt <patrick@blueri.se>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <sys/types.h>
#include <sys/systm.h>
#include <sys/malloc.h>

#include <dev/ofw/openfirm.h>
#include <dev/ofw/ofw_power.h>

LIST_HEAD(, power_domain_device) power_domain_devices =
	LIST_HEAD_INITIALIZER(power_domain_devices);

void
power_domain_register(struct power_domain_device *pd)
{
	pd->pd_cells = OF_getpropint(pd->pd_node, "#power-domain-cells", 0);
	pd->pd_phandle = OF_getpropint(pd->pd_node, "phandle", 0);
	if (pd->pd_phandle == 0)
		return;

	LIST_INSERT_HEAD(&power_domain_devices, pd, pd_list);
}

void
power_domain_enable_cells(uint32_t *cells, int on)
{
	struct power_domain_device *pd;
	uint32_t phandle = cells[0];

	LIST_FOREACH(pd, &power_domain_devices, pd_list) {
		if (pd->pd_phandle == phandle)
			break;
	}

	if (pd && pd->pd_enable)
		pd->pd_enable(pd->pd_cookie, &cells[1], on);
}

uint32_t *
power_domain_next_domain(uint32_t *cells)
{
	uint32_t phandle = cells[0];
	int node, ncells;

	node = OF_getnodebyphandle(phandle);
	if (node == 0)
		return NULL;

	ncells = OF_getpropint(node, "#power-domain-cells", 0);
	return cells + ncells + 1;
}

void
power_domain_do_enable_idx(int node, int idx, int on)
{
	uint32_t *domains;
	uint32_t *domain;
	int len;

	len = OF_getproplen(node, "power-domains");
	if (len <= 0)
		return;

	domains = malloc(len, M_TEMP, M_WAITOK);
	OF_getpropintarray(node, "power-domains", domains, len);

	domain = domains;
	while (domain && domain < domains + (len / sizeof(uint32_t))) {
		if (idx <= 0)
			power_domain_enable_cells(domain, on);
		if (idx == 0)
			break;
		domain = power_domain_next_domain(domain);
		idx--;
	}

	free(domains, M_TEMP, len);
}

void
power_domain_enable_idx(int node, int idx)
{
	power_domain_do_enable_idx(node, idx, 1);
}

void
power_domain_enable(int node)
{
	power_domain_do_enable_idx(node, 0, 1);
}

void
power_domain_disable_idx(int node, int idx)
{
	power_domain_do_enable_idx(node, idx, 0);
}

void
power_domain_disable(int node)
{
	power_domain_do_enable_idx(node, 0, 0);
}