CastPack.h

// CastPack.h:
// ------------
//

#ifndef __IPSDKUTIL_INSTRUCTIONSET_DETAIL_AVX_CASTPACK_H__
#define __IPSDKUTIL_INSTRUCTIONSET_DETAIL_AVX_CASTPACK_H__

#include <IPSDKUtil/IPSDKUtilExports.h>
#include <IPSDKUtil/InstructionSet/BaseMaskPack.h>
#include <IPSDKUtil/InstructionSet/BasePack.h>
#include <IPSDKUtil/InstructionSet/detail/CastPack.h>
#include <IPSDKUtil/InstructionSet/Comparison/isLess.h>
#include <IPSDKUtil/Tools/arrayUtils.h>
#include <IPSDKUtil/Tools/ForceInline.h>

#include <boost/type_traits/is_signed.hpp>

namespace ipsdk {
namespace simd {
namespace detail {


template <typename TIn, typename TOut>
struct CastPack<eInstructionSet::eIS_Avx, TIn, TOut,
                typename boost::enable_if_c<boost::is_same<TIn, TOut>::value>::type>
{ 
    static IPSDK_FORCEINLINE BasePack<ePackType::ePT_Avx, TOut> 
    act(const BasePack<ePackType::ePT_Avx, TIn>& in)
    {
        return in;
    }

    static IPSDK_FORCEINLINE
    void
    act(const BasePack<ePackType::ePT_Avx, TIn>& in,
        BasePack<ePackType::ePT_Avx, TOut> & out)
    {
        typedef typename AvxType<TIn>::Type RegType;
        arrayop::copyArray<sizeof(TIn), RegType>(in._val, out._val);
    }

    static IPSDK_FORCEINLINE
    void
    act(const BaseMaskPack<ePackType::ePT_Avx, TIn>& inMask,
        BasePack<ePackType::ePT_Avx, TOut> & out)
    {
        typedef typename AvxType<TIn>::Type RegType;
        arrayop::copyArray<sizeof(TIn), RegType>(inMask._val, out._val);
    }
};

template <>
struct CastPack<eInstructionSet::eIS_Avx, ipReal32, ipReal64>
{ 
    static IPSDK_FORCEINLINE BasePack<ePackType::ePT_Avx, ipReal64> 
    act(const BasePack<ePackType::ePT_Avx, ipReal32>& in)
    {
        BasePack<ePackType::ePT_Avx, ipReal64> out;

        out._val[0] = _mm256_cvtps_pd(_mm256_extractf128_ps(in._val[0], 0));
        out._val[1] = _mm256_cvtps_pd(_mm256_extractf128_ps(in._val[0], 1));

        out._val[2] = _mm256_cvtps_pd(_mm256_extractf128_ps(in._val[1], 0));
        out._val[3] = _mm256_cvtps_pd(_mm256_extractf128_ps(in._val[1], 1));

        out._val[4] = _mm256_cvtps_pd(_mm256_extractf128_ps(in._val[2], 0));
        out._val[5] = _mm256_cvtps_pd(_mm256_extractf128_ps(in._val[2], 1));

        out._val[6] = _mm256_cvtps_pd(_mm256_extractf128_ps(in._val[3], 0));
        out._val[7] = _mm256_cvtps_pd(_mm256_extractf128_ps(in._val[3], 1));
        
        return out;
    }

    static IPSDK_FORCEINLINE
    void
    act(const BasePack<ePackType::ePT_Avx, ipReal32>& in,
         BasePack<ePackType::ePT_Avx, ipReal64>& out)
    {
        out._val[0] = _mm256_cvtps_pd(_mm256_extractf128_ps(in._val[0], 0));
        out._val[1] = _mm256_cvtps_pd(_mm256_extractf128_ps(in._val[0], 1));

        out._val[2] = _mm256_cvtps_pd(_mm256_extractf128_ps(in._val[1], 0));
        out._val[3] = _mm256_cvtps_pd(_mm256_extractf128_ps(in._val[1], 1));

        out._val[4] = _mm256_cvtps_pd(_mm256_extractf128_ps(in._val[2], 0));
        out._val[5] = _mm256_cvtps_pd(_mm256_extractf128_ps(in._val[2], 1));

        out._val[6] = _mm256_cvtps_pd(_mm256_extractf128_ps(in._val[3], 0));
        out._val[7] = _mm256_cvtps_pd(_mm256_extractf128_ps(in._val[3], 1));
    }
};

template <>
struct CastPack<eInstructionSet::eIS_Avx, ipReal64, ipReal32>
{ 
    static IPSDK_FORCEINLINE BasePack<ePackType::ePT_Avx, ipReal32> 
    act(const BasePack<ePackType::ePT_Avx, ipReal64>& in)
    {
        BasePack<ePackType::ePT_Avx, ipReal32> out;

        out._val[0] = _mm256_castps128_ps256(_mm256_cvtpd_ps(in._val[0]));
        out._val[0] = _mm256_insertf128_ps(out._val[0],
            _mm256_cvtpd_ps(in._val[1]), 1);

        out._val[1] = _mm256_castps128_ps256(_mm256_cvtpd_ps(in._val[2]));
        out._val[1] = _mm256_insertf128_ps(out._val[0],
            _mm256_cvtpd_ps(in._val[3]), 1);

        out._val[2] = _mm256_castps128_ps256(_mm256_cvtpd_ps(in._val[4]));
        out._val[2] = _mm256_insertf128_ps(out._val[0],
            _mm256_cvtpd_ps(in._val[5]), 1);

        out._val[3] = _mm256_castps128_ps256(_mm256_cvtpd_ps(in._val[6]));
        out._val[3] = _mm256_insertf128_ps(out._val[0],
            _mm256_cvtpd_ps(in._val[7]), 1);

        return out;
    }

    static IPSDK_FORCEINLINE
    void
    act(const BasePack<ePackType::ePT_Avx, ipReal64>& in,
         BasePack<ePackType::ePT_Avx, ipReal32>& out)
    {
        out._val[0] = _mm256_castps128_ps256(_mm256_cvtpd_ps(in._val[0]));
        out._val[0] = _mm256_insertf128_ps(out._val[0],
            _mm256_cvtpd_ps(in._val[1]), 1);

        out._val[1] = _mm256_castps128_ps256(_mm256_cvtpd_ps(in._val[2]));
        out._val[1] = _mm256_insertf128_ps(out._val[1],
            _mm256_cvtpd_ps(in._val[3]), 1);

        out._val[2] = _mm256_castps128_ps256(_mm256_cvtpd_ps(in._val[4]));
        out._val[2] = _mm256_insertf128_ps(out._val[2],
            _mm256_cvtpd_ps(in._val[5]), 1);

        out._val[3] = _mm256_castps128_ps256(_mm256_cvtpd_ps(in._val[6]));
        out._val[3] = _mm256_insertf128_ps(out._val[3],
            _mm256_cvtpd_ps(in._val[7]), 1);
    }
};


} // end of namespace detail
} // end of namespace simd
} // end of namespace ipsdk

#endif // __IPSDKUTIL_INSTRUCTIONSET_DETAIL_AVX_CASTPACK_H__