portapack-mayhem/firmware/common/simd.hpp

/*
 * Copyright (C) 2015 Jared Boone, ShareBrained Technology, Inc.
 *
 * This file is part of PortaPack.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2, or (at your option)
 * any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */

#ifndef __SIMD_H__
#define __SIMD_H__

#if defined(LPC43XX_M4)

#include <hal.h>

#include <cstdint>

template<typename T>
struct simd32_t {
	union {
		uint32_t raw;
		T vec;
	};

	operator uint32_t() const {
		return raw;
	}

	simd32_t& operator=(uint32_t v) {
		raw = v;
		return *this;
	}

	static_assert(sizeof(raw) == sizeof(vec), "simd32_t types are not the same size.");
};

template<typename T>
simd32_t<T>* simd32_ptr(T* const p) {
	return reinterpret_cast<simd32_t<T>*>(p);
}

struct vec4_s8 {
	union {
		int8_t v[4];
		uint32_t w;
	};
};

struct vec2_s16 {
	constexpr vec2_s16(
	) : v { 0, 0 }
	{
	}

	constexpr vec2_s16(
		const int16_t v
	) : v { v, v }
	{
	}
	
	constexpr vec2_s16(
		const int16_t v0,
		const int16_t v1
	) : v { v0, v1 }
	{
	}

	constexpr vec2_s16(
		const vec2_s16& other
	) : w { other.w }
	{
	}

	vec2_s16& operator=(const vec2_s16& other) {
		w = other.w;
		return *this;
	}

	union {
		int16_t v[2];
		uint32_t w;
	};
};

static inline vec4_s8 rev16(const vec4_s8 v) {
	vec4_s8 result;
	result.w = __REV16(v.w);
	return result;
}

static inline vec4_s8 pkhbt(const vec4_s8 v1, const vec4_s8 v2, const size_t sh = 0) {
	vec4_s8 result;
	result.w = __PKHBT(v1.w, v2.w, sh);
	return result;
}

static inline vec2_s16 pkhbt(const vec2_s16 v1, const vec2_s16 v2, const size_t sh = 0) {
	vec2_s16 result;
	result.w = __PKHBT(v1.w, v2.w, sh);
	return result;
}

static inline vec2_s16 pkhtb(const vec2_s16 v1, const vec2_s16 v2, const size_t sh = 0) {
	vec2_s16 result;
	result.w = __PKHTB(v1.w, v2.w, sh);
	return result;
}

static inline vec2_s16 sxtb16(const vec4_s8 v, const size_t sh = 0) {
	vec2_s16 result;
	result.w = __SXTB16(v.w, sh);
	return result;
}

static inline int32_t smlsd(const vec2_s16 v1, const vec2_s16 v2, const int32_t accum) {
	return __SMLSD(v1.w, v2.w, accum);
}

static inline int32_t smlad(const vec2_s16 v1, const vec2_s16 v2, const int32_t accum) {
	return __SMLAD(v1.w, v2.w, accum);
}

#endif /* defined(LPC43XX_M4) */

#endif/*__SIMD_H__*/