Merge pull request #2081 from Shaikh-Ubaid/support_unsigned_int_unary_operators

Support unsigned int unary operators

Author: certik

integration_tests/CMakeLists.txt

@@ -552,12 +552,14 @@ RUN(NAME test_ConstantEllipsis  LABLES cpython llvm c)
 RUN(NAME test_max_min        LABELS cpython llvm c)
 RUN(NAME test_global         LABELS cpython llvm c)
 RUN(NAME test_global_decl    LABELS cpython llvm c)
-RUN(NAME test_integer_bitnot LABELS cpython llvm c wasm)
-RUN(NAME test_unsign_int_bitnot LABELS cpython llvm c)
 RUN(NAME test_ifexp_01          LABELS cpython llvm c)
 RUN(NAME test_ifexp_02          LABELS cpython llvm c)
-RUN(NAME test_unary_minus    LABELS cpython llvm c)
-RUN(NAME test_unary_plus     LABELS cpython llvm c)
+RUN(NAME test_unary_op_01    LABELS cpython llvm c) # unary minus
+RUN(NAME test_unary_op_02    LABELS cpython llvm c) # unary plus
+RUN(NAME test_unary_op_03    LABELS cpython llvm c wasm) # unary bitinvert
+RUN(NAME test_unary_op_04    LABELS cpython llvm c) # unary bitinvert
+RUN(NAME test_unary_op_05    LABELS cpython llvm c) # unsigned unary minus, plus
+RUN(NAME test_unary_op_06    LABELS cpython llvm c) # unsigned unary bitnot
 RUN(NAME test_bool_binop     LABELS cpython llvm c)
 RUN(NAME test_issue_518      LABELS cpython llvm c NOFAST)
 RUN(NAME structs_01          LABELS cpython llvm c)

integration_tests/test_unary_minus.py integration_tests/test_unary_op_01.pyintegration_tests/test_unary_minus.py renamed to integration_tests/test_unary_op_01.py

@@ -0,0 +1,27 @@
+from lpython import u16, u32, u64
+
+def f():
+
+    i: u16
+    i = u16(67)
+    print(-i, +i, -(-i))
+    assert -i == u16(65469)
+    assert +i == u16(67)
+    assert -(-i) == u16(67)
+
+    j: u32
+    j = u32(25)
+    print(-j, +j, -(-j))
+    assert -j == u32(4294967271)
+    assert +j == u32(25)
+    assert -(-j) == u32(25)
+
+    k: u64
+    k = u64(100000000000123)
+    print(-k, +k, -(-k))
+    # TODO: We are unable to store the following u64 in AST/R
+    # assert -k == u64(18446644073709551493)
+    assert +k == u64(100000000000123)
+    assert -(-k) == u64(100000000000123)
+
+f()

integration_tests/test_unary_plus.py integration_tests/test_unary_op_02.pyintegration_tests/test_unary_plus.py renamed to integration_tests/test_unary_op_02.py

@@ -0,0 +1,35 @@
+from lpython import i32, i64, u16, u64, f32, f64
+
+def f():
+
+    i: i32
+    j: i64
+    i = -67
+    j = i64(0)
+
+    print(i, j)
+    print(not i, not j)
+    assert (not i) == False
+    assert (not j) == True
+
+    k: u16
+    l: u64
+    k = u16(0)
+    l = u64(55)
+
+    print(k, l)
+    print(not k, not l)
+    assert (not k) == True
+    assert (not l) == False
+
+    m: f32
+    n: f64
+    m = f32(0.0)
+    n = -3.14
+
+    print(m, n)
+    print(not m, not n)
+    assert (not m) == True
+    assert (not n) == False
+
+f()

integration_tests/test_integer_bitnot.py integration_tests/test_unary_op_03.pyintegration_tests/test_integer_bitnot.py renamed to integration_tests/test_unary_op_03.py

@@ -430,6 +430,7 @@ cast_kind
     | LogicalToCharacter
     | UnsignedIntegerToInteger
     | UnsignedIntegerToReal
+    | UnsignedIntegerToLogical
     | IntegerToUnsignedInteger
     | RealToUnsignedInteger
     | CPtrToUnsignedInteger

…egration_tests/test_unsign_int_bitnot.py integration_tests/test_unary_op_04.pyintegration_tests/test_unsign_int_bitnot.py renamed to integration_tests/test_unary_op_04.py integration_tests/test_unsign_int_bitnot.py renamed to integration_tests/test_unary_op_04.py

@@ -1830,7 +1830,8 @@ PyMODINIT_FUNC PyInit_lpython_module_)" + fn_name + R"((void) {
                 last_expr_precedence = 2;
                 break;
             }
-            case (ASR::cast_kindType::IntegerToLogical) : {
+            case (ASR::cast_kindType::IntegerToLogical) :
+            case (ASR::cast_kindType::UnsignedIntegerToLogical) : {
                 src = "(bool)(" + src + ")";
                 last_expr_precedence = 2;
                 break;
@@ -2061,6 +2062,12 @@ PyMODINIT_FUNC PyInit_lpython_module_)" + fn_name + R"((void) {
         handle_UnaryMinus(x);
     }
 
+    void visit_UnsignedIntegerUnaryMinus(const ASR::UnsignedIntegerUnaryMinus_t &x) {
+        handle_UnaryMinus(x);
+        int kind = ASRUtils::extract_kind_from_ttype_t(ASRUtils::expr_type(x.m_arg));
+        src = "(uint" + std::to_string(kind * 8) + "_t)" + src;
+    }
+
     void visit_RealUnaryMinus(const ASR::RealUnaryMinus_t &x) {
         handle_UnaryMinus(x);
     }

integration_tests/test_unary_op_05.py

@@ -6208,6 +6208,17 @@ class ASRToLLVMVisitor : public ASR::BaseVisitor<ASRToLLVMVisitor>
         tmp = builder->CreateSub(zero, tmp);
     }
 
+    void visit_UnsignedIntegerUnaryMinus(const ASR::UnsignedIntegerUnaryMinus_t &x) {
+        if (x.m_value) {
+            this->visit_expr_wrapper(x.m_value, true);
+            return;
+        }
+        this->visit_expr_wrapper(x.m_arg, true);
+        int kind = ASRUtils::extract_kind_from_ttype_t(ASRUtils::expr_type(x.m_arg));
+        llvm::Value *one = llvm::ConstantInt::get(context, llvm::APInt(kind * 8, 1, true));
+        tmp = builder->CreateAdd(builder->CreateNot(tmp), one); // compute 2's complement
+    }
+
     void visit_RealUnaryMinus(const ASR::RealUnaryMinus_t &x) {
         if (x.m_value) {
             this->visit_expr_wrapper(x.m_value, true);
@@ -6777,7 +6788,8 @@ class ASRToLLVMVisitor : public ASR::BaseVisitor<ASRToLLVMVisitor>
                 tmp = complex_from_floats(tmp, zero, complex_type);
                 break;
             }
-            case (ASR::cast_kindType::IntegerToLogical) : {
+            case (ASR::cast_kindType::IntegerToLogical) :
+            case (ASR::cast_kindType::UnsignedIntegerToLogical) : {
                 ASR::ttype_t* curr_type = extract_ttype_t_from_expr(x.m_arg);
                 LCOMPILERS_ASSERT(curr_type != nullptr)
                 int a_kind = ASRUtils::extract_kind_from_ttype_t(curr_type);