/* SPDX-License-Identifier: GPL-2.0-or-later */
/*
 * Copyright (C) 2005 Mips Technologies
 * Author: Chris Dearman, chris@mips.com derived from fpu.h
 */
#ifndef _ASM_DSP_H
#define _ASM_DSP_H

#include <asm/cpu.h>
#include <asm/cpu-features.h>
#include <asm/hazards.h>
#include <asm/mipsregs.h>

#define DSP_DEFAULT 0x00000000
#define DSP_MASK 0x3f

#define __enable_dsp_hazard()      \
do {         \
 asm("_ehb");       \
} while (0)

static inline void __init_dsp(void)
{
 mthi1(0);
 mtlo1(0);
 mthi2(0);
 mtlo2(0);
 mthi3(0);
 mtlo3(0);
 wrdsp(DSP_DEFAULT, DSP_MASK);
}

static inline void init_dsp(void)
{
 if (cpu_has_dsp)
  __init_dsp();
}

#define __save_dsp(tsk)       \
do {         \
 tsk->thread.dsp.dspr[0] = mfhi1();    \
 tsk->thread.dsp.dspr[1] = mflo1();    \
 tsk->thread.dsp.dspr[2] = mfhi2();    \
 tsk->thread.dsp.dspr[3] = mflo2();    \
 tsk->thread.dsp.dspr[4] = mfhi3();    \
 tsk->thread.dsp.dspr[5] = mflo3();    \
 tsk->thread.dsp.dspcontrol = rddsp(DSP_MASK);   \
} while (0)

#define save_dsp(tsk)       \
do {         \
 if (cpu_has_dsp)      \
  __save_dsp(tsk);     \
} while (0)

#define __restore_dsp(tsk)      \
do {         \
 mthi1(tsk->thread.dsp.dspr[0]);     \
 mtlo1(tsk->thread.dsp.dspr[1]);     \
 mthi2(tsk->thread.dsp.dspr[2]);     \
 mtlo2(tsk->thread.dsp.dspr[3]);     \
 mthi3(tsk->thread.dsp.dspr[4]);     \
 mtlo3(tsk->thread.dsp.dspr[5]);     \
 wrdsp(tsk->thread.dsp.dspcontrol, DSP_MASK);   \
} while (0)

#define restore_dsp(tsk)      \
do {         \
 if (cpu_has_dsp)      \
  __restore_dsp(tsk);     \
} while (0)

#define __get_dsp_regs(tsk)      \
({         \
 if (tsk == current)      \
  __save_dsp(current);     \
         \
 tsk->thread.dsp.dspr;      \
})

#endif /* _ASM_DSP_H */