Add _MM_TRANSPOSE4_PS pseudo-macro. (#106)

This adds a strange macro, which I've replaced with a function, because it
seems there are not many better alternatives.

Also adds a test, and `#[allow(non_snake_case)]` to `#[simd_test]`.

Author: MaloJaffre

src/x86/sse.rs

@@ -875,6 +875,22 @@ pub unsafe fn _mm_undefined_ps() -> f32x4 {
     f32x4::splat(mem::uninitialized())
 }
 
+/// Transpose the 4x4 matrix formed by 4 rows of f32x4 in place.
+#[inline(always)]
+#[allow(non_snake_case)]
+#[target_feature = "+sse"]
+pub unsafe fn _MM_TRANSPOSE4_PS(row0: &mut f32x4, row1: &mut f32x4, row2: &mut f32x4, row3: &mut f32x4) {
+    let tmp0 = _mm_unpacklo_ps(*row0, *row1);
+    let tmp2 = _mm_unpacklo_ps(*row2, *row3);
+    let tmp1 = _mm_unpackhi_ps(*row0, *row1);
+    let tmp3 = _mm_unpackhi_ps(*row2, *row3);
+
+    mem::replace(row0, _mm_movelh_ps(tmp0, tmp2));
+    mem::replace(row1,_mm_movehl_ps(tmp2, tmp0));
+    mem::replace(row2, _mm_movelh_ps(tmp1, tmp3));
+    mem::replace(row3, _mm_movehl_ps(tmp3, tmp1));
+}
+
 #[allow(improper_ctypes)]
 extern {
     #[link_name = "llvm.x86.sse.add.ss"]
@@ -1291,4 +1307,19 @@ mod tests {
             sse::_MM_GET_EXCEPTION_STATE() & sse::_MM_EXCEPT_UNDERFLOW != 0;
         assert_eq!(underflow, true);
     }
+
+    #[simd_test = "sse"]
+    unsafe fn _MM_TRANSPOSE4_PS() {
+        let mut a = f32x4::new(1.0, 2.0, 3.0, 4.0);
+        let mut b = f32x4::new(5.0, 6.0, 7.0, 8.0);
+        let mut c = f32x4::new(9.0, 10.0, 11.0, 12.0);
+        let mut d = f32x4::new(13.0, 14.0, 15.0, 16.0);
+
+        sse::_MM_TRANSPOSE4_PS(&mut a, &mut b, &mut c, &mut d);
+
+        assert_eq!(a, f32x4::new(1.0, 5.0, 9.0, 13.0));
+        assert_eq!(b, f32x4::new(2.0, 6.0, 10.0, 14.0));
+        assert_eq!(c, f32x4::new(3.0, 7.0, 11.0, 15.0));
+        assert_eq!(d, f32x4::new(4.0, 8.0, 12.0, 16.0));
+    }
 }

stdsimd-test/simd-test-macro/src/lib.rs

@@ -46,6 +46,7 @@ pub fn simd_test(attr: proc_macro::TokenStream,
     let name: TokenStream = name.as_str().parse().unwrap();
 
     let ret: TokenStream = quote! {
+        #[allow(non_snake_case)]
         #[test]
         fn #name() {
             if cfg_feature_enabled!(#target_feature) {