Merge pull request #1889 from Smit-create/i-1867

Support real <-> unsigned integer cast

Author: certik

integration_tests/expr_01u.py

@@ -1,4 +1,4 @@
-from lpython import inline, u32
+from lpython import inline, u32, u64, i64
 
 @inline
 def uadd(x: u32, y: u32) -> u32:
@@ -20,6 +20,14 @@ def main1():
     z = uand_op(x, y)
     assert z == u32(16)
 
+    # test issue 1867
+    cycles_count: u64
+    end_cycle: i64 = i64(100)
+    start_cycle: i64 = i64(4)
+    num_iters: i64 = i64(4)
+    cycles_count = u64((end_cycle - start_cycle) / num_iters)
+    assert cycles_count == u64(24)
+
 
 main1()
 

src/libasr/ASR.asdl

@@ -425,7 +425,9 @@ cast_kind
     | IntegerToCharacter
     | LogicalToCharacter
     | UnsignedIntegerToInteger
+    | UnsignedIntegerToReal
     | IntegerToUnsignedInteger
+    | RealToUnsignedInteger
     | CPtrToUnsignedInteger
     | UnsignedIntegerToCPtr
 

src/libasr/codegen/asr_to_c_cpp.h

@@ -1491,12 +1491,28 @@ R"(#include <stdio.h>
                 last_expr_precedence = 2;
                 break;
             }
+            case (ASR::cast_kindType::RealToUnsignedInteger) : {
+                int dest_kind = ASRUtils::extract_kind_from_ttype_t(x.m_type);
+                src = "(uint" + std::to_string(dest_kind * 8) + "_t)(" + src + ")";
+                last_expr_precedence = 2;
+                break;
+            }
             case (ASR::cast_kindType::UnsignedIntegerToInteger) : {
                 int dest_kind = ASRUtils::extract_kind_from_ttype_t(x.m_type);
                 src = "(int" + std::to_string(dest_kind * 8) + "_t)(" + src + ")";
                 last_expr_precedence = 2;
                 break;
             }
+            case (ASR::cast_kindType::UnsignedIntegerToReal) : {
+                int dest_kind = ASRUtils::extract_kind_from_ttype_t(x.m_type);
+                switch (dest_kind) {
+                    case 4: src = "(float)(" + src + ")"; break;
+                    case 8: src = "(double)(" + src + ")"; break;
+                    default: throw CodeGenError("Cast IntegerToReal: Unsupported Kind " + std::to_string(dest_kind));
+                }
+                last_expr_precedence = 2;
+                break;
+            }
             case (ASR::cast_kindType::ComplexToComplex) : {
                 break;
             }

src/libasr/codegen/asr_to_llvm.cpp

@@ -6585,6 +6585,11 @@ class ASRToLLVMVisitor : public ASR::BaseVisitor<ASRToLLVMVisitor>
                 tmp = builder->CreateSIToFP(tmp, getFPType(a_kind, false));
 		        break;
             }
+            case (ASR::cast_kindType::UnsignedIntegerToReal) : {
+                int a_kind = ASRUtils::extract_kind_from_ttype_t(x.m_type);
+                tmp = builder->CreateSIToFP(tmp, getFPType(a_kind, false));
+		        break;
+            }
             case (ASR::cast_kindType::LogicalToReal) : {
                 int a_kind = ASRUtils::extract_kind_from_ttype_t(x.m_type);
                 tmp = builder->CreateUIToFP(tmp, getFPType(a_kind, false));
@@ -6597,6 +6602,13 @@ class ASRToLLVMVisitor : public ASR::BaseVisitor<ASRToLLVMVisitor>
                 tmp = builder->CreateFPToSI(tmp, target_type);
                 break;
             }
+            case (ASR::cast_kindType::RealToUnsignedInteger) : {
+                llvm::Type *target_type;
+                int a_kind = ASRUtils::extract_kind_from_ttype_t(x.m_type);
+                target_type = getIntType(a_kind);
+                tmp = builder->CreateFPToSI(tmp, target_type);
+                break;
+            }
             case (ASR::cast_kindType::RealToComplex) : {
                 llvm::Type *target_type;
                 llvm::Value *zero;

src/lpython/semantics/python_ast_to_asr.cpp

@@ -55,13 +55,15 @@ namespace CastingUtil {
         {std::make_pair(ASR::ttypeType::Real, ASR::ttypeType::Complex), ASR::cast_kindType::RealToComplex},
         {std::make_pair(ASR::ttypeType::Real, ASR::ttypeType::Integer), ASR::cast_kindType::RealToInteger},
         {std::make_pair(ASR::ttypeType::Real, ASR::ttypeType::Logical), ASR::cast_kindType::RealToLogical},
+        {std::make_pair(ASR::ttypeType::Real, ASR::ttypeType::UnsignedInteger), ASR::cast_kindType::RealToUnsignedInteger},
         {std::make_pair(ASR::ttypeType::Integer, ASR::ttypeType::Complex), ASR::cast_kindType::IntegerToComplex},
         {std::make_pair(ASR::ttypeType::Integer, ASR::ttypeType::Real), ASR::cast_kindType::IntegerToReal},
         {std::make_pair(ASR::ttypeType::Integer, ASR::ttypeType::Logical), ASR::cast_kindType::IntegerToLogical},
         {std::make_pair(ASR::ttypeType::Integer, ASR::ttypeType::UnsignedInteger), ASR::cast_kindType::IntegerToUnsignedInteger},
         {std::make_pair(ASR::ttypeType::Logical, ASR::ttypeType::Real), ASR::cast_kindType::LogicalToReal},
         {std::make_pair(ASR::ttypeType::Logical, ASR::ttypeType::Integer), ASR::cast_kindType::LogicalToInteger},
         {std::make_pair(ASR::ttypeType::UnsignedInteger, ASR::ttypeType::Integer), ASR::cast_kindType::UnsignedIntegerToInteger},
+        {std::make_pair(ASR::ttypeType::UnsignedInteger, ASR::ttypeType::Real), ASR::cast_kindType::UnsignedIntegerToReal}
     };
 
     // Data structure which contains casting rules for equal intrinsic