RISC-V Immediate Decoder Template Metaprogram

riscv-imm.cc

/*
 * RISC-V Immediate Decoder Template Metaprogram
 *
 * Author: Michael Clark <[email protected]>
 */

#include <cstdio>
#include <cstdint>

/* sign extension template */

template <typename T, unsigned B>
inline T sign_extend(const T x)
{
	struct { T x:B; } s;
	return s.x = x;
}

/*
 * bit range template
 *
 * N    =   decoded MSB offset
 * M    =   decoded LSB offset
 */

template<int N, int M = N>
struct B
{
	enum { n = N };
	enum { m = M };
	enum { width = N - M + 1 };
	enum { mask = ((1 << (N + 1)) - 1) ^ ((1 << M) - 1) };
};

/*
 * immediate bit range segment
 *
 * K    =   instruction MSB offset
 * L    =   instruction LSB offset
 * Args =   decoded bit offsets i.e. B<N:M>, ...
 */

template<int K, int L, typename... Args>
struct S;

template<int K, int L>
struct S<K,L>
{
	enum { offset = 0 };

	static inline constexpr int64_t decode(uint64_t inst) { return 0; }
};

template<int K, int L, typename H, typename... T>
struct S<K,L,H,T...> : S<K,L,T...>
{
	typedef S<K,L,T...> I;

	enum { offset = I::offset + H::width };
	enum { shift = offset + L - H::width - H::m };

	static inline constexpr int64_t decode(uint64_t inst) {
		return ((shift < 0 ? inst << -shift : inst >> shift) & H::mask) |
			I::decode(inst);
	}
};

/*
 * immediate bit range notation template
 *
 * W    =   number of bits for sign extension
 * Args =   bit range segments i.e. S<K,L, B<N:M>, ...>, ...
 */

template<int W, typename... Args>
struct imm_impl_t;

template<int W>
struct imm_impl_t<W>
{
	static inline constexpr int64_t decode(uint64_t inst) { return 0; }
};

template<int W, typename H, typename... T>
struct imm_impl_t<W,H,T...> : imm_impl_t<W,T...>
{
	typedef imm_impl_t<W,T...> I;

	static inline constexpr int64_t decode(uint64_t inst) {
		return I::decode(inst) | H::decode(inst);
	}
};

template<int W, typename... Args>
struct imm_t : imm_impl_t<W,Args...>
{
	typedef imm_impl_t<W,Args...> I;

	static constexpr int64_t decode(uint64_t inst) {
		return sign_extend<int64_t,W>(I::decode(inst));
	}
};

/*
 * immediate bit range notation template examples
 * from riscv-compressed-spec-v1.9, page 9
 */

/*
 *      12         10  6               2
 * CB   offset[8|4:3]  offset[7:6|2:1|5]
 */
typedef imm_t<9, S<12,10, B<8>,B<4,3>>, S<6,2, B<7,6>,B<2,1>,B<5>>> CB;

/*
 *      12                          2
 * CJ   offset[11|4|9:8|10|6|7|3:1|5]
 */
typedef imm_t<12, S<12,2, B<11>,B<4>,B<9,8>,B<10>,B<6>,B<7>,B<3,1>,B<5>>> CJ;


/* immediate decoder test functions */

extern "C" int64_t __attribute__ ((noinline)) test_cb(uint64_t inst)
{
	return CB::decode(inst);
}

extern "C" int64_t __attribute__ ((noinline)) test_cj(uint64_t inst)
{
	return CJ::decode(inst);
}

/* main */

int main()
{
	/* bne  a0,zero,52 */
	printf("%s\n", test_cb(0xe915) == 52 ? "pass" : "fail");

	/* jal  zero,-4 */
	printf("%s\n", test_cj(0xbff5) == -4 ? "pass" : "fail");
}