[CodeGen] Add sar(signed arithmetic shift right) support

Author: eee4017

include/soll/AST/Expr.h

@@ -241,7 +241,7 @@ class BinaryOperator : public Expr {
     return Opc == BO_Add || Opc == BO_Sub;
   }
   static inline constexpr bool isShiftOp(Opcode Opc) {
-    return Opc == BO_Shl || Opc == BO_Shr;
+    return Opc == BO_Shl || Opc == BO_Shr || Opc == BO_AShr;
   }
   static inline constexpr bool isBitwiseOp(Opcode Opc) {
     return Opc >= BO_And && Opc <= BO_Or;

include/soll/AST/OperationKinds.h

@@ -34,6 +34,7 @@ enum BinaryOperatorKind {
   // Bitwise shift operators.
   BO_Shl,
   BO_Shr,
+  BO_AShr,
   // Bitwise AND operator.
   BO_And,
   // Bitwise XOR operator.

lib/CodeGen/ExprEmitter.cpp

@@ -407,6 +407,18 @@ ExprValuePtr ExprEmitter::visit(const BinaryOperator *BO) {
       }
       break;
     }
+    case BinaryOperatorKind::BO_AShr: {
+      auto LHSType = LHS->getType();
+      auto RHSType = RHS->getType();
+      auto returnType =
+          LHSType->getIntegerBitWidth() > RHSType->getIntegerBitWidth()
+              ? LHSType
+              : RHSType;
+      V = Builder.CreateAShr(Builder.CreateZExtOrTrunc(LHS, returnType),
+                             Builder.CreateZExtOrTrunc(RHS, returnType),
+                             "BO_AShr");
+      break;
+    }
     case BinaryOperatorKind::BO_And:
     case BinaryOperatorKind::BO_AsmAnd:
       V = Builder.CreateAnd(LHS, RHS, "BO_And");

lib/Sema/SemaExprAsm.cpp

@@ -669,8 +669,11 @@ Sema::CreateAsmBuiltinCallExpr(SourceRange L, const AsmIdentifier &Callee,
   case AsmIdentifier::SpecialIdentifier::shr:
     return std::make_unique<AsmBinaryOperator>(
         L, std::move(Args[1]), std::move(Args[0]), std::move(ReturnTy), BO_Shr);
-  // TODO: case AsmIdentifier::SpecialIdentifier::sars256:
-  // TODO: case AsmIdentifier::SpecialIdentifier::sar:
+  case AsmIdentifier::SpecialIdentifier::sars256:
+  case AsmIdentifier::SpecialIdentifier::sar:
+    return std::make_unique<AsmBinaryOperator>(L, std::move(Args[1]),
+                                               std::move(Args[0]),
+                                               std::move(ReturnTy), BO_AShr);
   default: ///< treated as normal CallExpr
     break;
   }