Merge pull request #586 from JohanMabille/logical

Implemented operator! for batch of complex

Author: JohanMabille

include/xsimd/types/xsimd_batch.hpp

@@ -247,6 +247,7 @@ struct batch<std::complex<T>, A> {
   batch(real_batch const& real) : m_real(real), m_imag(0) {}
   batch(T val) : m_real(val), m_imag(0) {}
   batch(std::initializer_list<value_type> data) { *this = load_unaligned(data.begin()); }
+  explicit batch(batch_bool_type b);
 
   static XSIMD_NO_DISCARD batch load_aligned(const T* real_src, const T* imag_src=nullptr);
   static XSIMD_NO_DISCARD batch load_unaligned(const T* real_src, const T* imag_src=nullptr);
@@ -293,6 +294,7 @@ struct batch<std::complex<T>, A> {
   }
 
   // unary operators
+  batch_bool_type operator!() const { return operator==(batch(0)); }
   batch operator~() const { return {~m_real, ~m_imag}; }
   batch operator-() const { return {-m_real, -m_imag};}
   batch operator+() const { return {+m_real, +m_imag}; }
@@ -564,6 +566,13 @@ batch_bool<T, A> batch_bool<T, A>::load_unaligned(bool const* mem) {
 
 // batch complex implementation
 //
+
+template <class T, class A>
+batch<std::complex<T>, A>::batch(batch_bool_type b)
+    : m_real(b), m_imag(0)
+{
+}
+
 template <class T, class A>
 batch<std::complex<T>, A> batch<std::complex<T>, A>::load_aligned(const T* real_src, const T* imag_src)
 {

test/test_batch_complex.cpp

@@ -533,6 +533,18 @@ class batch_complex_test : public testing::Test
         }
     }
 
+    void test_boolean_conversion() const
+    {
+        // !batch
+        {
+            array_type expected;
+            std::transform(lhs.cbegin(), lhs.cend(), expected.begin(),
+                            [](const value_type& l) { return l == value_type(0); });
+            batch_type res = (batch_type)!batch_lhs();
+            EXPECT_BATCH_EQ(res, expected) << print_function_name("!batch");
+        }
+    }
+
 private:
 
     batch_type batch_lhs() const
@@ -596,3 +608,8 @@ TYPED_TEST(batch_complex_test, fused_operations)
 {
     this->test_fused_operations();
 }
+
+TYPED_TEST(batch_complex_test, boolean_conversion)
+{
+    this->test_boolean_conversion();
+}