Print infix operations in infix form

Author: mbovel

compiler/src/dotty/tools/dotc/printing/RefinedPrinter.scala

@@ -32,6 +32,7 @@ import dotty.tools.dotc.util.SourcePosition
 import dotty.tools.dotc.ast.untpd.{MemberDef, Modifiers, PackageDef, RefTree, Template, TypeDef, ValOrDefDef}
 import cc.*
 import dotty.tools.dotc.parsing.JavaParsers
+import dotty.tools.dotc.transform.TreeExtractors.BinaryOp
 
 class RefinedPrinter(_ctx: Context) extends PlainPrinter(_ctx) {
 
@@ -387,6 +388,10 @@ class RefinedPrinter(_ctx: Context) extends PlainPrinter(_ctx) {
 
     def optDotPrefix(tree: This) = optText(tree.qual)(_ ~ ".") provided !isLocalThis(tree)
 
+    /** Should a binary operation with this operator be printed infix? */
+    def isInfix(op: Symbol) =
+      op.exists && (op.isDeclaredInfix || op.name.isOperatorName)
+
     def caseBlockText(tree: Tree): Text = tree match {
       case Block(stats, expr) => toText(stats :+ expr, "\n")
       case expr => toText(expr)
@@ -478,6 +483,13 @@ class RefinedPrinter(_ctx: Context) extends PlainPrinter(_ctx) {
         optDotPrefix(tree) ~ keywordStr("this") ~ idText(tree)
       case Super(qual: This, mix) =>
         optDotPrefix(qual) ~ keywordStr("super") ~ optText(mix)("[" ~ _ ~ "]")
+      case BinaryOp(l, op, r) if isInfix(op) =>
+        val isRightAssoc = op.name.endsWith(":")
+        val opPrec = parsing.precedence(op.name)
+        val leftPrec = if isRightAssoc then opPrec + 1 else opPrec
+        val rightPrec = if !isRightAssoc then opPrec + 1 else opPrec
+        changePrec(opPrec):
+          atPrec(leftPrec)(toText(l))  ~ " " ~ toText(op.name) ~ " " ~ atPrec(rightPrec)(toText(r))
       case app @ Apply(fun, args) =>
         if (fun.hasType && fun.symbol == defn.throwMethod)
           changePrec (GlobalPrec) {

tests/printing/export-param-flags.check

@@ -2,7 +2,7 @@
 package <empty> {
   final lazy module val A: A = new A()
   final module class A() extends Object() { this: A.type =>
-    inline def inlinedParam(inline x: Int): Int = x.+(x):Int
+    inline def inlinedParam(inline x: Int): Int = x + x:Int
   }
   final lazy module val Exported: Exported = new Exported()
   final module class Exported() extends Object() { this: Exported.type =>

tests/printing/infix-operations.check

@@ -0,0 +1,35 @@
+[[syntax trees at end of                     typer]] // tests/printing/infix-operations.scala
+package <empty> {
+  class C(a: Int) extends Object() {
+    private[this] val a: Int
+    def foo(b: Int): C = this
+    def +(b: Int): C = this
+  }
+  final lazy module val infix-operations$package: infix-operations$package =
+    new infix-operations$package()
+  final module class infix-operations$package() extends Object() {
+    this: infix-operations$package.type =>
+    @main def main: Unit =
+      {
+        val v1: Int = 1 + 2 + 3
+        val v2: Int = 1 + (2 + 3)
+        val v3: Int = 1 + 2 * 3
+        val v4: Int = (1 + 2) * 3
+        val v5: Int = 1 + 2:Int
+        val v6: Int = 1 + 2:Int
+        val c: C = new C(2)
+        val v7: C = c.foo(3)
+        val v8: C = c + 3
+        ()
+      }
+  }
+  final class main() extends Object() {
+    <static> def main(args: Array[String]): Unit =
+      try main catch
+        {
+          case error @ _:scala.util.CommandLineParser.ParseError =>
+            scala.util.CommandLineParser.showError(error)
+        }
+  }
+}
+

tests/printing/infix-operations.scala

@@ -0,0 +1,15 @@
+class C(a: Int):
+  def foo(b: Int): C = this
+  def +(b: Int): C = this
+
+@main def main =
+  val v1 = 1 + 2 + 3
+  val v2 = 1 + (2 + 3)
+  val v3 = 1 + 2 * 3
+  val v4 = (1 + 2) * 3
+  val v5 = (1 + 2):Int
+  val v6 = 1 + 2:Int // same as above
+
+  val c = new C(2) // must not be printed in infix form
+  val v7 = c.foo(3) // must not be printed in infix form
+  val v8 = c + 3

tests/printing/lambdas.check

@@ -2,26 +2,26 @@
 package <empty> {
   final lazy module val Main: Main = new Main()
   final module class Main() extends Object() { this: Main.type =>
-    val f1: Int => Int = (x: Int) => x.+(1)
-    val f2: (Int, Int) => Int = (x: Int, y: Int) => x.+(y)
-    val f3: Int => Int => Int = (x: Int) => (y: Int) => x.+(y)
-    val f4: [T] => (x: Int) => Int = [T >: Nothing <: Any] => (x: Int) => x.+(1)
+    val f1: Int => Int = (x: Int) => x + 1
+    val f2: (Int, Int) => Int = (x: Int, y: Int) => x + y
+    val f3: Int => Int => Int = (x: Int) => (y: Int) => x + y
+    val f4: [T] => (x: Int) => Int = [T >: Nothing <: Any] => (x: Int) => x + 1
     val f5: [T] => (x: Int) => Int => Int = [T >: Nothing <: Any] => (x: Int)
-       => (y: Int) => x.+(y)
+       => (y: Int) => x + y
     val f6: Int => Int = (x: Int) =>
       {
-        val x2: Int = x.+(1)
-        x2.+(1)
+        val x2: Int = x + 1
+        x2 + 1
       }
-    def f7(x: Int): Int = x.+(1)
+    def f7(x: Int): Int = x + 1
     val f8: Int => Int = (x: Int) => Main.f7(x)
     val l: List[Int] = List.apply[Int]([1,2,3 : Int]*)
-    Main.l.map[Int]((_$1: Int) => _$1.+(1))
-    Main.l.map[Int]((x: Int) => x.+(1))
+    Main.l.map[Int]((_$1: Int) => _$1 + 1)
+    Main.l.map[Int]((x: Int) => x + 1)
     Main.l.map[Int]((x: Int) =>
       {
-        val x2: Int = x.+(1)
-        x2.+(1)
+        val x2: Int = x + 1
+        x2 + 1
       }
     )
     Main.l.map[Int]((x: Int) => Main.f7(x))

tests/printing/posttyper/i22533.check

@@ -10,7 +10,7 @@ package <empty> {
     override def equals(x$0: Any): Boolean =
       x$0 match
         {
-          case x$0 @ _:Foo @unchecked => this.u.==(x$0.u)
+          case x$0 @ _:Foo @unchecked => this.u == x$0.u
           case _ => false
         }
     private[this] val u: Int

tests/printing/transformed/lazy-vals-legacy.check

@@ -26,8 +26,8 @@ package <empty> {
         {
           val flag: Long = scala.runtime.LazyVals.get(this, A.OFFSET$_m_0)
           val state: Long = scala.runtime.LazyVals.STATE(flag, 0)
-          if state.==(3) then return A.x$lzy1 else
-            if state.==(0) then
+          if state == 3 then return A.x$lzy1 else
+            if state == 0 then
               if scala.runtime.LazyVals.CAS(this, A.OFFSET$_m_0, flag, 1, 0)
                  then
                 try

tests/printing/transformed/lazy-vals-new.check

@@ -24,14 +24,14 @@ package <empty> {
       {
         val result: Object = A.x$lzy1
         if result.isInstanceOf[Int] then Int.unbox(result) else
-          if result.eq(scala.runtime.LazyVals.NullValue) then Int.unbox(null)
+          if result eq scala.runtime.LazyVals.NullValue then Int.unbox(null)
              else Int.unbox(A.x$lzyINIT1())
       }
     private def x$lzyINIT1(): Object =
       while <empty> do
         {
           val current: Object = A.x$lzy1
-          if current.eq(null) then
+          if current eq null then
             if
               scala.runtime.LazyVals.objCAS(this, A.OFFSET$_m_0, null,
                 scala.runtime.LazyVals.Evaluating)
@@ -42,7 +42,7 @@ package <empty> {
                 try
                   {
                     resultNullable = Int.box(2)
-                    if resultNullable.eq(null) then
+                    if resultNullable eq null then
                       result = scala.runtime.LazyVals.NullValue else
                       result = resultNullable
                     ()
@@ -69,7 +69,7 @@ package <empty> {
               current.isInstanceOf[
                 scala.runtime.LazyVals.LazyVals$LazyValControlState]
              then
-              if current.eq(scala.runtime.LazyVals.Evaluating) then
+              if current eq scala.runtime.LazyVals.Evaluating then
                 {
                   scala.runtime.LazyVals.objCAS(this, A.OFFSET$_m_0, current,
                     new scala.runtime.LazyVals.LazyVals$Waiting())

tests/run/i13181.scala

@@ -1 +1 @@
-@main def Test = assert(scala.compiletime.codeOf(1+2) == "1.+(2)")
+@main def Test = assert(scala.compiletime.codeOf(1+2) == "1 + 2")

tests/run/typeCheckErrors.check

@@ -1 +1 @@
-List(Error(value check is not a member of Unit,compileError("1" * 2).check(""),22,Typer), Error(argument to compileError must be a statically known String but was: augmentString("1").*(2),compileError("1" * 2).check(""),13,Typer))
+List(Error(value check is not a member of Unit,compileError("1" * 2).check(""),22,Typer), Error(argument to compileError must be a statically known String but was: augmentString("1") * 2,compileError("1" * 2).check(""),13,Typer))