fix: Implemented __is_lvalue_reference and __is_rvalue_reference type…

… traits
robertoraggi · Oct 29, 2023 · 845f336 · 845f336
1 parent 7f94af4
commit 845f336
Show file tree

Hide file tree

Showing 2 changed files with 77 additions and 0 deletions.
diff --git a/src/parser/cxx/parser.cc b/src/parser/cxx/parser.cc
@@ -203,6 +203,34 @@ struct Parser::TypeTraits {
     constexpr auto operator()(auto) const -> bool { return false; }
   } is_volatile;
 
+  struct {
+    constexpr auto operator()(const LvalueReferenceType*) const -> bool {
+      return true;
+    }
+
+    constexpr auto operator()(auto) const -> bool { return false; }
+  } is_lvalue_reference;
+
+  struct {
+    constexpr auto operator()(const RvalueReferenceType*) const -> bool {
+      return true;
+    }
+
+    constexpr auto operator()(auto) const -> bool { return false; }
+  } is_rvalue_reference;
+
+  struct {
+    constexpr auto operator()(const LvalueReferenceType*) const -> bool {
+      return true;
+    }
+
+    constexpr auto operator()(const RvalueReferenceType*) const -> bool {
+      return true;
+    }
+
+    constexpr auto operator()(auto) const -> bool { return false; }
+  } is_reference;
+
   struct {
     constexpr auto operator()(const LvalueReferenceType* type) const
         -> const Type* {
@@ -2489,6 +2517,23 @@ auto Parser::parse_builtin_call_expression(ExpressionAST*& yyast) -> bool {
         break;
       }
 
+      case TokenKind::T___IS_LVALUE_REFERENCE: {
+        ast->constValue =
+            visit(TypeTraits{*this}.is_lvalue_reference, firstType);
+        break;
+      }
+
+      case TokenKind::T___IS_RVALUE_REFERENCE: {
+        ast->constValue =
+            visit(TypeTraits{*this}.is_rvalue_reference, firstType);
+        break;
+      }
+
+      case TokenKind::T___IS_REFERENCE: {
+        ast->constValue = visit(TypeTraits{*this}.is_reference, firstType);
+        break;
+      }
+
       default:;
     }  // switch
   }

diff --git a/tests/unit_tests/sema/type_traits_01.cc b/tests/unit_tests/sema/type_traits_01.cc
@@ -42,3 +42,35 @@ static_assert(__is_volatile(volatile void*));
 
 // expected-error@1 {{static assert failed}}
 static_assert(__is_volatile(int));
+
+//
+// is_lvalue_reference trait
+//
+
+static_assert(__is_lvalue_reference(int&));
+static_assert(__is_lvalue_reference(const int&));
+
+// expected-error@1 {{static assert failed}}
+static_assert(__is_lvalue_reference(int&&));
+
+//
+// is_rvalue_reference trait
+//
+
+static_assert(__is_rvalue_reference(int&&));
+static_assert(__is_rvalue_reference(const int&&));
+
+// expected-error@1 {{static assert failed}}
+static_assert(__is_rvalue_reference(int&));
+
+//
+// is_reference trait
+//
+
+static_assert(__is_reference(int&));
+static_assert(__is_reference(const int&));
+static_assert(__is_reference(int&&));
+static_assert(__is_reference(const int&&));
+
+// expected-error@1 {{static assert failed}}
+static_assert(__is_reference(int));