/*
 * SPDX-FileCopyrightText: 2006 Martin Decky
 *
 * SPDX-License-Identifier: BSD-3-Clause
 */

#include <abi/asmtool.h>
#include <arch/asm/regname.h>
#include <arch/fpu_context_struct.h>
#include <arch/msr.h>

.text

.macro FPU_CONTEXT_STORE r
	stfd fr0, FPU_CONTEXT_OFFSET_FR0(\r)
	stfd fr1, FPU_CONTEXT_OFFSET_FR1(\r)
	stfd fr2, FPU_CONTEXT_OFFSET_FR2(\r)
	stfd fr3, FPU_CONTEXT_OFFSET_FR3(\r)
	stfd fr4, FPU_CONTEXT_OFFSET_FR4(\r)
	stfd fr5, FPU_CONTEXT_OFFSET_FR5(\r)
	stfd fr6, FPU_CONTEXT_OFFSET_FR6(\r)
	stfd fr7, FPU_CONTEXT_OFFSET_FR7(\r)
	stfd fr8, FPU_CONTEXT_OFFSET_FR8(\r)
	stfd fr9, FPU_CONTEXT_OFFSET_FR9(\r)
	stfd fr10, FPU_CONTEXT_OFFSET_FR10(\r)
	stfd fr11, FPU_CONTEXT_OFFSET_FR11(\r)
	stfd fr12, FPU_CONTEXT_OFFSET_FR12(\r)
	stfd fr13, FPU_CONTEXT_OFFSET_FR13(\r)
	stfd fr14, FPU_CONTEXT_OFFSET_FR14(\r)
	stfd fr15, FPU_CONTEXT_OFFSET_FR15(\r)
	stfd fr16, FPU_CONTEXT_OFFSET_FR16(\r)
	stfd fr17, FPU_CONTEXT_OFFSET_FR17(\r)
	stfd fr18, FPU_CONTEXT_OFFSET_FR18(\r)
	stfd fr19, FPU_CONTEXT_OFFSET_FR19(\r)
	stfd fr20, FPU_CONTEXT_OFFSET_FR20(\r)
	stfd fr21, FPU_CONTEXT_OFFSET_FR21(\r)
	stfd fr22, FPU_CONTEXT_OFFSET_FR22(\r)
	stfd fr23, FPU_CONTEXT_OFFSET_FR23(\r)
	stfd fr24, FPU_CONTEXT_OFFSET_FR24(\r)
	stfd fr25, FPU_CONTEXT_OFFSET_FR25(\r)
	stfd fr26, FPU_CONTEXT_OFFSET_FR26(\r)
	stfd fr27, FPU_CONTEXT_OFFSET_FR27(\r)
	stfd fr28, FPU_CONTEXT_OFFSET_FR28(\r)
	stfd fr29, FPU_CONTEXT_OFFSET_FR29(\r)
	stfd fr30, FPU_CONTEXT_OFFSET_FR30(\r)
	stfd fr31, FPU_CONTEXT_OFFSET_FR31(\r)
.endm

.macro FPU_CONTEXT_LOAD r
	lfd fr0, FPU_CONTEXT_OFFSET_FR0(\r)
	lfd fr1, FPU_CONTEXT_OFFSET_FR1(\r)
	lfd fr2, FPU_CONTEXT_OFFSET_FR2(\r)
	lfd fr3, FPU_CONTEXT_OFFSET_FR3(\r)
	lfd fr4, FPU_CONTEXT_OFFSET_FR4(\r)
	lfd fr5, FPU_CONTEXT_OFFSET_FR5(\r)
	lfd fr6, FPU_CONTEXT_OFFSET_FR6(\r)
	lfd fr7, FPU_CONTEXT_OFFSET_FR7(\r)
	lfd fr8, FPU_CONTEXT_OFFSET_FR8(\r)
	lfd fr9, FPU_CONTEXT_OFFSET_FR9(\r)
	lfd fr10, FPU_CONTEXT_OFFSET_FR10(\r)
	lfd fr11, FPU_CONTEXT_OFFSET_FR11(\r)
	lfd fr12, FPU_CONTEXT_OFFSET_FR12(\r)
	lfd fr13, FPU_CONTEXT_OFFSET_FR13(\r)
	lfd fr14, FPU_CONTEXT_OFFSET_FR14(\r)
	lfd fr15, FPU_CONTEXT_OFFSET_FR15(\r)
	lfd fr16, FPU_CONTEXT_OFFSET_FR16(\r)
	lfd fr17, FPU_CONTEXT_OFFSET_FR17(\r)
	lfd fr18, FPU_CONTEXT_OFFSET_FR18(\r)
	lfd fr19, FPU_CONTEXT_OFFSET_FR19(\r)
	lfd fr20, FPU_CONTEXT_OFFSET_FR20(\r)
	lfd fr21, FPU_CONTEXT_OFFSET_FR21(\r)
	lfd fr22, FPU_CONTEXT_OFFSET_FR22(\r)
	lfd fr23, FPU_CONTEXT_OFFSET_FR23(\r)
	lfd fr24, FPU_CONTEXT_OFFSET_FR24(\r)
	lfd fr25, FPU_CONTEXT_OFFSET_FR25(\r)
	lfd fr26, FPU_CONTEXT_OFFSET_FR26(\r)
	lfd fr27, FPU_CONTEXT_OFFSET_FR27(\r)
	lfd fr28, FPU_CONTEXT_OFFSET_FR28(\r)
	lfd fr29, FPU_CONTEXT_OFFSET_FR29(\r)
	lfd fr30, FPU_CONTEXT_OFFSET_FR30(\r)
	lfd fr31, FPU_CONTEXT_OFFSET_FR31(\r)
.endm

FUNCTION_BEGIN(fpu_context_save)
	FPU_CONTEXT_STORE r3

	mffs fr0
	stfd fr0, FPU_CONTEXT_OFFSET_FPSCR(r3)

	blr
FUNCTION_END(fpu_context_save)

FUNCTION_BEGIN(fpu_context_restore)
	lfd fr0, FPU_CONTEXT_OFFSET_FPSCR(r3)
	mtfsf 0xff, fr0

	FPU_CONTEXT_LOAD r3

	blr
FUNCTION_END(fpu_context_restore)

FUNCTION_BEGIN(fpu_init)
	mfmsr r0
	ori r0, r0, MSR_FP

	# Disable FPU exceptions
	li r3, MSR_FE0 | MSR_FE1
	andc r0, r0, r3

	mtmsr r0
	isync

	blr
FUNCTION_END(fpu_init)

FUNCTION_BEGIN(fpu_enable)
	mfmsr r0
	ori r0, r0, MSR_FP
	mtmsr r0
	isync
	blr
FUNCTION_END(fpu_enable)

FUNCTION_BEGIN(fpu_disable)
	mfmsr r0
	li r3, MSR_FP
	andc r0, r0, r3
	mtmsr r0
	isync
	blr
FUNCTION_END(fpu_disable)