r/sustaining_gazes
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity
sustaining_gazes/lib/3rdParty/dlib/include/dlib/simd/simd4i.h
unknown 57e58a6949 Master commit of OpenFace.
2016-04-28 15:40:36 -04:00

411 lines
12 KiB
C++
Raw Blame History

// Copyright (C) 2013 Davis E. King (davis@dlib.net)
// License: Boost Software License See LICENSE.txt for the full license.
#ifndef DLIB_sIMD4I_Hh_
#define DLIB_sIMD4I_Hh_
#include "simd_check.h"
#include "../uintn.h"
namespace dlib
{
#ifdef DLIB_HAVE_SSE2
class simd4i
{
public:
typedef int32 type;
inline simd4i() {}
inline simd4i(int32 f) { x = _mm_set1_epi32(f); }
inline simd4i(int32 r0, int32 r1, int32 r2, int32 r3) { x = _mm_setr_epi32(r0,r1,r2,r3); }
inline simd4i(const __m128i& val):x(val) {}
inline simd4i& operator=(const __m128i& val)
{
x = val;
return *this;
}
inline operator __m128i() const { return x; }
inline void load_aligned(const type* ptr) { x = _mm_load_si128((const __m128i*)ptr); }
inline void store_aligned(type* ptr) const { _mm_store_si128((__m128i*)ptr, x); }
inline void load(const type* ptr) { x = _mm_loadu_si128((const __m128i*)ptr); }
inline void store(type* ptr) const { _mm_storeu_si128((__m128i*)ptr, x); }
inline unsigned int size() const { return 4; }
inline int32 operator[](unsigned int idx) const
{
int32 temp[4];
store(temp);
return temp[idx];
}
private:
__m128i x;
};
#else
class simd4i
{
public:
typedef int32 type;
inline simd4i() {}
inline simd4i(int32 f) { x[0]=f; x[1]=f; x[2]=f; x[3]=f; }
inline simd4i(int32 r0, int32 r1, int32 r2, int32 r3) { x[0]=r0; x[1]=r1; x[2]=r2; x[3]=r3;}
struct rawarray
{
int32 a[4];
};
inline simd4i(const rawarray& a) { x[0]=a.a[0]; x[1]=a.a[1]; x[2]=a.a[2]; x[3]=a.a[3]; }
inline void load_aligned(const type* ptr)
{
x[0] = ptr[0];
x[1] = ptr[1];
x[2] = ptr[2];
x[3] = ptr[3];
}
inline void store_aligned(type* ptr) const
{
ptr[0] = x[0];
ptr[1] = x[1];
ptr[2] = x[2];
ptr[3] = x[3];
}
inline void load(const type* ptr)
{
x[0] = ptr[0];
x[1] = ptr[1];
x[2] = ptr[2];
x[3] = ptr[3];
}
inline void store(type* ptr) const
{
ptr[0] = x[0];
ptr[1] = x[1];
ptr[2] = x[2];
ptr[3] = x[3];
}
inline unsigned int size() const { return 4; }
inline int32 operator[](unsigned int idx) const { return x[idx]; }
private:
int32 x[4];
};
#endif
// ----------------------------------------------------------------------------------------
inline std::ostream& operator<<(std::ostream& out, const simd4i& item)
{
int32 temp[4];
item.store(temp);
out << "(" << temp[0] << ", " << temp[1] << ", " << temp[2] << ", " << temp[3] << ")";
return out;
}
// ----------------------------------------------------------------------------------------
inline simd4i operator+ (const simd4i& lhs, const simd4i& rhs)
{
#ifdef DLIB_HAVE_SSE2
return _mm_add_epi32(lhs, rhs);
#else
return simd4i(lhs[0]+rhs[0],
lhs[1]+rhs[1],
lhs[2]+rhs[2],
lhs[3]+rhs[3]);
#endif
}
inline simd4i& operator+= (simd4i& lhs, const simd4i& rhs)
{ return lhs = lhs + rhs; return lhs;}
// ----------------------------------------------------------------------------------------
inline simd4i operator- (const simd4i& lhs, const simd4i& rhs)
{
#ifdef DLIB_HAVE_SSE2
return _mm_sub_epi32(lhs, rhs);
#else
return simd4i(lhs[0]-rhs[0],
lhs[1]-rhs[1],
lhs[2]-rhs[2],
lhs[3]-rhs[3]);
#endif
}
inline simd4i& operator-= (simd4i& lhs, const simd4i& rhs)
{ return lhs = lhs - rhs; return lhs;}
// ----------------------------------------------------------------------------------------
inline simd4i operator* (const simd4i& lhs, const simd4i& rhs)
{
#ifdef DLIB_HAVE_SSE41
return _mm_mullo_epi32(lhs, rhs);
#elif defined(DLIB_HAVE_SSE2)
int32 _lhs[4]; lhs.store(_lhs);
int32 _rhs[4]; rhs.store(_rhs);
return simd4i(_lhs[0]*_rhs[0],
_lhs[1]*_rhs[1],
_lhs[2]*_rhs[2],
_lhs[3]*_rhs[3]);
#else
return simd4i(lhs[0]*rhs[0],
lhs[1]*rhs[1],
lhs[2]*rhs[2],
lhs[3]*rhs[3]);
#endif
}
inline simd4i& operator*= (simd4i& lhs, const simd4i& rhs)
{ return lhs = lhs * rhs; return lhs;}
// ----------------------------------------------------------------------------------------
inline simd4i operator& (const simd4i& lhs, const simd4i& rhs)
{
#ifdef DLIB_HAVE_SSE2
return _mm_and_si128(lhs, rhs);
#else
return simd4i(lhs[0]&rhs[0],
lhs[1]&rhs[1],
lhs[2]&rhs[2],
lhs[3]&rhs[3]);
#endif
}
inline simd4i& operator&= (simd4i& lhs, const simd4i& rhs)
{ return lhs = lhs & rhs; return lhs;}
// ----------------------------------------------------------------------------------------
inline simd4i operator| (const simd4i& lhs, const simd4i& rhs)
{
#ifdef DLIB_HAVE_SSE2
return _mm_or_si128(lhs, rhs);
#else
return simd4i(lhs[0]|rhs[0],
lhs[1]|rhs[1],
lhs[2]|rhs[2],
lhs[3]|rhs[3]);
#endif
}
inline simd4i& operator|= (simd4i& lhs, const simd4i& rhs)
{ return lhs = lhs | rhs; return lhs;}
// ----------------------------------------------------------------------------------------
inline simd4i operator^ (const simd4i& lhs, const simd4i& rhs)
{
#ifdef DLIB_HAVE_SSE2
return _mm_xor_si128(lhs, rhs);
#else
return simd4i(lhs[0]^rhs[0],
lhs[1]^rhs[1],
lhs[2]^rhs[2],
lhs[3]^rhs[3]);
#endif
}
inline simd4i& operator^= (simd4i& lhs, const simd4i& rhs)
{ return lhs = lhs ^ rhs; return lhs;}
// ----------------------------------------------------------------------------------------
inline simd4i operator~ (const simd4i& lhs)
{
#ifdef DLIB_HAVE_SSE2
return _mm_xor_si128(lhs, _mm_set1_epi32(0xFFFFFFFF));
#else
return simd4i(~lhs[0],
~lhs[1],
~lhs[2],
~lhs[3]);
#endif
}
// ----------------------------------------------------------------------------------------
inline simd4i operator<< (const simd4i& lhs, const int& rhs)
{
#ifdef DLIB_HAVE_SSE2
return _mm_sll_epi32(lhs,_mm_cvtsi32_si128(rhs));
#else
return simd4i(lhs[0]<<rhs,
lhs[1]<<rhs,
lhs[2]<<rhs,
lhs[3]<<rhs);
#endif
}
inline simd4i& operator<<= (simd4i& lhs, const int& rhs)
{ return lhs = lhs << rhs; return lhs;}
// ----------------------------------------------------------------------------------------
inline simd4i operator>> (const simd4i& lhs, const int& rhs)
{
#ifdef DLIB_HAVE_SSE2
return _mm_sra_epi32(lhs,_mm_cvtsi32_si128(rhs));
#else
return simd4i(lhs[0]>>rhs,
lhs[1]>>rhs,
lhs[2]>>rhs,
lhs[3]>>rhs);
#endif
}
inline simd4i& operator>>= (simd4i& lhs, const int& rhs)
{ return lhs = lhs >> rhs; return lhs;}
// ----------------------------------------------------------------------------------------
inline simd4i operator== (const simd4i& lhs, const simd4i& rhs)
{
#ifdef DLIB_HAVE_SSE2
return _mm_cmpeq_epi32(lhs, rhs);
#else
return simd4i(lhs[0]==rhs[0] ? 0xFFFFFFFF : 0,
lhs[1]==rhs[1] ? 0xFFFFFFFF : 0,
lhs[2]==rhs[2] ? 0xFFFFFFFF : 0,
lhs[3]==rhs[3] ? 0xFFFFFFFF : 0);
#endif
}
// ----------------------------------------------------------------------------------------
inline simd4i operator!= (const simd4i& lhs, const simd4i& rhs)
{
#ifdef DLIB_HAVE_SSE2
return ~(lhs==rhs);
#else
return simd4i(lhs[0]!=rhs[0] ? 0xFFFFFFFF : 0,
lhs[1]!=rhs[1] ? 0xFFFFFFFF : 0,
lhs[2]!=rhs[2] ? 0xFFFFFFFF : 0,
lhs[3]!=rhs[3] ? 0xFFFFFFFF : 0);
#endif
}
// ----------------------------------------------------------------------------------------
inline simd4i operator< (const simd4i& lhs, const simd4i& rhs)
{
#ifdef DLIB_HAVE_SSE2
return _mm_cmplt_epi32(lhs, rhs);
#else
return simd4i(lhs[0]<rhs[0] ? 0xFFFFFFFF : 0,
lhs[1]<rhs[1] ? 0xFFFFFFFF : 0,
lhs[2]<rhs[2] ? 0xFFFFFFFF : 0,
lhs[3]<rhs[3] ? 0xFFFFFFFF : 0);
#endif
}
// ----------------------------------------------------------------------------------------
inline simd4i operator> (const simd4i& lhs, const simd4i& rhs)
{
return rhs < lhs;
}
// ----------------------------------------------------------------------------------------
inline simd4i operator<= (const simd4i& lhs, const simd4i& rhs)
{
#ifdef DLIB_HAVE_SSE2
return ~(lhs > rhs);
#else
return simd4i(lhs[0]<=rhs[0] ? 0xFFFFFFFF : 0,
lhs[1]<=rhs[1] ? 0xFFFFFFFF : 0,
lhs[2]<=rhs[2] ? 0xFFFFFFFF : 0,
lhs[3]<=rhs[3] ? 0xFFFFFFFF : 0);
#endif
}
// ----------------------------------------------------------------------------------------
inline simd4i operator>= (const simd4i& lhs, const simd4i& rhs)
{
return rhs <= lhs;
}
// ----------------------------------------------------------------------------------------
inline simd4i min (const simd4i& lhs, const simd4i& rhs)
{
#ifdef DLIB_HAVE_SSE41
return _mm_min_epi32(lhs, rhs);
#elif defined(DLIB_HAVE_SSE2)
int32 _lhs[4]; lhs.store(_lhs);
int32 _rhs[4]; rhs.store(_rhs);
return simd4i(std::min(_lhs[0],_rhs[0]),
std::min(_lhs[1],_rhs[1]),
std::min(_lhs[2],_rhs[2]),
std::min(_lhs[3],_rhs[3]));
#else
return simd4i(std::min(lhs[0],rhs[0]),
std::min(lhs[1],rhs[1]),
std::min(lhs[2],rhs[2]),
std::min(lhs[3],rhs[3]));
#endif
}
// ----------------------------------------------------------------------------------------
inline simd4i max (const simd4i& lhs, const simd4i& rhs)
{
#ifdef DLIB_HAVE_SSE41
return _mm_max_epi32(lhs, rhs);
#elif defined(DLIB_HAVE_SSE2)
int32 _lhs[4]; lhs.store(_lhs);
int32 _rhs[4]; rhs.store(_rhs);
return simd4i(std::max(_lhs[0],_rhs[0]),
std::max(_lhs[1],_rhs[1]),
std::max(_lhs[2],_rhs[2]),
std::max(_lhs[3],_rhs[3]));
#else
return simd4i(std::max(lhs[0],rhs[0]),
std::max(lhs[1],rhs[1]),
std::max(lhs[2],rhs[2]),
std::max(lhs[3],rhs[3]));
#endif
}
// ----------------------------------------------------------------------------------------
inline int32 sum(const simd4i& item)
{
#ifdef DLIB_HAVE_SSE3
simd4i temp = _mm_hadd_epi32(item,item);
temp = _mm_hadd_epi32(temp,temp);
return _mm_cvtsi128_si32(temp);
#elif defined(DLIB_HAVE_SSE2)
int32 temp[4];
item.store(temp);
return temp[0]+temp[1]+temp[2]+temp[3];
#else
return item[0]+item[1]+item[2]+item[3];
#endif
}
// ----------------------------------------------------------------------------------------
// perform cmp ? a : b
inline simd4i select(const simd4i& cmp, const simd4i& a, const simd4i& b)
{
#ifdef DLIB_HAVE_SSE41
return _mm_blendv_epi8(b,a,cmp);
#elif defined(DLIB_HAVE_SSE2)
return ((cmp&a) | _mm_andnot_si128(cmp,b));
#else
return ((cmp&a) | (~cmp&b));
#endif
}
// ----------------------------------------------------------------------------------------
}
#endif // DLIB_sIMD4I_Hh_
Reference in a new issue View git blame Copy permalink