P4_F64vec2.h

Go to the documentation of this file.

#ifndef L1Algo_F64vec2P4_H
#define L1Algo_F64vec2P4_H
 
#include "vec_arithmetic.h"
 
#include <cmath>
#include <emmintrin.h>
#include <iostream>
 
/**********************************
 *
 *   Vector of four single doubles
 *
 **********************************/
 
// #pragma pack(push,16)/* Must ensure class & union 16-B aligned */
 
// typedef __m128d Vectordouble __attribute__ ((aligned(16)));
 
const union {
    double d;
    long long i;
} __d_one = {(double)1.};
 
const union {
    long long i[2];
    __m128d m;
} __f64vec2_abs_mask_cheat = {{0x7fffffffffffffffll, 0x7fffffffffffffffll}},
  __f64vec2_sgn_mask_cheat = {{0x8000000000000000ull, 0x8000000000000000ull}}, __f64vec2_zero_cheat = {{0, 0}},
  __f64vec2_one_cheat = {{__d_one.i, __d_one.i}}, __f64vec2_true_cheat = {{0xFFFFFFFFFFFFFFFF, 0xFFFFFFFFFFFFFFFF}},
  __f64vec2_false_cheat = {{0x0000000000000000, 0x0000000000000000}};
 
#define _f64vec2_abs_mask (static_cast<F64vec2>(__f64vec2_abs_mask_cheat.m))
#define _f64vec2_sgn_mask (static_cast<F64vec2>(__f64vec2_sgn_mask_cheat.m))
#define _f64vec2_zero (static_cast<F64vec2>(__f64vec2_zero_cheat.m))
#define _f64vec2_one (static_cast<F64vec2>(__f64vec2_one_cheat.m))
#define _f64vec2_true (static_cast<F64vec2>(__f64vec2_true_cheat.m))
#define _f64vec2_false (static_cast<F64vec2>(__f64vec2_false_cheat.m))
 
class F64vec2
{
  public:
    __m128d v;
 
    double& operator[](int i) { return (reinterpret_cast<double*>(&v))[i]; }
    double operator[](int i) const { return (reinterpret_cast<const double*>(&v))[i]; }
 
    F64vec2()
        : v(_mm_set_pd1(0))
    {}
    F64vec2(const __m128d& a)
        : v(a)
    {}
    F64vec2(const double& a)
        : v(_mm_set_pd1(a))
    {}
 
    F64vec2(const double& f0, const double& f1)
        : v(_mm_set_pd(f1, f0))
    {}
 
    /* Conversion function */
    operator __m128d() const { return v; } /* Convert to __m128d */
 
    /* Arithmetic Operators */
    friend F64vec2 operator+(const F64vec2& a, const F64vec2& b) { return _mm_add_pd(a, b); }
    friend F64vec2 operator-(const F64vec2& a, const F64vec2& b) { return _mm_sub_pd(a, b); }
    friend F64vec2 operator*(const F64vec2& a, const F64vec2& b) { return _mm_mul_pd(a, b); }
    friend F64vec2 operator/(const F64vec2& a, const F64vec2& b) { return _mm_div_pd(a, b); }
 
    /* Functions */
    friend F64vec2 min(const F64vec2& a, const F64vec2& b) { return _mm_min_pd(a, b); }
    friend F64vec2 max(const F64vec2& a, const F64vec2& b) { return _mm_max_pd(a, b); }
 
    /* Square Root */
    friend F64vec2 sqrt(const F64vec2& a) { return _mm_sqrt_pd(a); }
 
    /* Reciprocal( inverse) Square Root */
    friend F64vec2 rsqrt(const F64vec2& a) { return 1. / sqrt(a); }
 
    /* Reciprocal (inversion) */
    // friend F64vec2 rcp  ( const F64vec2 &a ){ return _mm_rcp_pd  (a); }
    /* Reciprocal (inversion) */
    // friend F64vec2 rcp  ( const F64vec2 &a ){ return 1. / a; }
    /* NewtonRaphson Reciprocal
      [2 * rcppd(x) - (x * rcppd(x) * rcppd(x))] */
    friend F64vec2 rcp(const F64vec2& a) { return 1. / a; }
 
    /* Absolute value */
    friend F64vec2 fabs(const F64vec2& a) { return _mm_and_pd(a, _f64vec2_abs_mask); }
 
    /* Sign */
    friend F64vec2 sgn(const F64vec2& a) { return _mm_or_pd(_mm_and_pd(a, _f64vec2_sgn_mask), _f64vec2_one); }
    friend F64vec2 asgnb(const F64vec2& a, const F64vec2& b) { return _mm_or_pd(_mm_and_pd(b, _f64vec2_sgn_mask), a); }
 
    /* Logical */
 
    friend F64vec2 operator&(const F64vec2& a, const F64vec2& b)
    {   // mask returned
        return _mm_and_pd(a, b);
    }
    friend F64vec2 operator|(const F64vec2& a, const F64vec2& b)
    {   // mask returned
        return _mm_or_pd(a, b);
    }
    friend F64vec2 operator^(const F64vec2& a, const F64vec2& b)
    {   // mask returned
        return _mm_xor_pd(a, b);
    }
    friend F64vec2 operator!(const F64vec2& a)
    {   // mask returned
        return _mm_xor_pd(a, _f64vec2_true);
    }
    // friend F64vec2 operator||( const F64vec2 &a, const F64vec2 &b ){ // mask returned
    //   return _mm_or_pd(a, b);
    // }
 
    /* Comparison */
 
    friend F64vec2 operator<(const F64vec2& a, const F64vec2& b)
    {   // mask returned
        return _mm_cmplt_pd(a, b);
    }
    friend F64vec2 operator<=(const F64vec2& a, const F64vec2& b)
    {   // mask returned
        return _mm_cmple_pd(a, b);
    }
    friend F64vec2 operator>(const F64vec2& a, const F64vec2& b)
    {   // mask returned
        return _mm_cmpgt_pd(a, b);
    }
    friend F64vec2 operator>=(const F64vec2& a, const F64vec2& b)
    {   // mask returned
        return _mm_cmpge_pd(a, b);
    }
    friend F64vec2 operator==(const F64vec2& a, const F64vec2& b)
    {   // mask returned
        return _mm_cmpeq_pd(a, b);
    }
 
#define if3(a, b, c) ((a) & (b)) | ((!(a)) & (c))   // analog (a) ? b : c
 
#define NotEmpty(a) bool((a)[0]) | bool((a)[1]) | bool((a)[2]) | bool((a)[3])
#define Empty(a) !(bool((a)[0]) | bool((a)[1]) | bool((a)[2]) | bool((a)[3]))
    // bool NotEmpty(const F64vec2 &a) { return a[0]||a[1]||a[2]||a[3]; }
    // bool    Empty(const F64vec2 &a) { return !(a[0]||a[1]||a[2]||a[3]); } // optimize
    friend F64vec2 bool2int(const F64vec2& a)
    {   // mask returned
        return if3(a, 1, 0);
    }
 
    /* Define all operators for consistensy */
 
    vec_arithmetic(F64vec2, double);
 
    /* Non intrinsic functions */
 
#define _f1(A, F) F64vec2(F(A[0]), F(A[1]))
 
    friend F64vec2 exp(const F64vec2& a) { return _f1(a, exp); }
    friend F64vec2 log(const F64vec2& a) { return _f1(a, log); }
    friend F64vec2 sin(const F64vec2& a) { return _f1(a, sin); }
    friend F64vec2 cos(const F64vec2& a) { return _f1(a, cos); }
    friend F64vec2 acos(const F64vec2& a) { return _f1(a, acos); }
 
#undef _f1
 
    friend F64vec2 atan2(const F64vec2& y, const F64vec2& x)
    {
        const F64vec2 pi(3.1415926535897932);
        const F64vec2 pi_2 = pi / 2;
        const F64vec2 zero(0);
 
        const F64vec2& xZero = F64vec2(x == zero);
        const F64vec2& yZero = F64vec2(y == zero);
        const F64vec2& xNeg = F64vec2(x < zero);
        const F64vec2& yNeg = F64vec2(y < zero);
 
        const F64vec2& absX = fabs(x);
        const F64vec2& absY = fabs(y);
 
        F64vec2 a = absY / absX;
        const F64vec2 pi_4 = pi / 4;
        const F64vec2& gt_tan_3pi_8 = F64vec2(a > F64vec2(2.414213562373095));
        const F64vec2& gt_tan_pi_8 = F64vec2(a > F64vec2(0.4142135623730950)) & F64vec2(!gt_tan_3pi_8);
        const F64vec2 minusOne(-1);
        F64vec2 b(zero);
        b = (pi_2 & gt_tan_3pi_8) + (F64vec2(!gt_tan_3pi_8) & b);
        b = (pi_4 & gt_tan_pi_8) + (F64vec2(!gt_tan_pi_8) & b);
        a = (gt_tan_3pi_8 & (minusOne / a)) + (F64vec2(!gt_tan_3pi_8) & a);
        a = (gt_tan_pi_8 & ((absY - absX) / (absY + absX))) + (F64vec2(!gt_tan_pi_8) & a);
        const F64vec2& a2 = a * a;
        b +=
            (((8.05374449538e-2 * a2 - 1.38776856032E-1) * a2 + 1.99777106478E-1) * a2 - 3.33329491539E-1) * a2 * a + a;
        F64vec2 xyNeg = F64vec2(xNeg ^ yNeg);
        b = (xyNeg & (-b)) + (F64vec2(!xyNeg) & b);
        xyNeg = F64vec2(xNeg & !yNeg);
        b = (xyNeg & (b + pi)) + (F64vec2(!xyNeg) & b);
        xyNeg = F64vec2(xNeg & yNeg);
        b = (xyNeg & (b - pi)) + (F64vec2(!xyNeg) & b);
        xyNeg = F64vec2(xZero & yZero);
        b = (xyNeg & zero) + (F64vec2(!xyNeg) & b);
        xyNeg = F64vec2(xZero & yNeg);
        b = (xyNeg & (-pi_2)) + (F64vec2(!xyNeg) & b);
        return b;
    }
 
    friend std::ostream& operator<<(std::ostream& strm, const F64vec2& a)
    {
        strm << "[" << a[0] << " " << a[1] << " " << a[2] << " " << a[3] << "]";
        return strm;
    }
 
    friend std::istream& operator>>(std::istream& strm, F64vec2& a)
    {
        double tmp;
        strm >> tmp;
        a = tmp;
        return strm;
    }
 
} __attribute__((aligned(16)));
 
typedef F64vec2 fvec;
typedef double fscal;
const int fvecLen = 2;
// #define fvec_true  _f64vec2_true
// #define fvec_false _f64vec2_false
#define _fvecalignment __attribute__((aligned(16)))
 
#include "std_alloc.h"
 
#endif