Skip to content

Fix #3675: don't unnecessarily wrap arrays when eliminating java varargs #3869

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 5 commits into from
Jan 23, 2018
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 10 additions & 0 deletions compiler/src/dotty/tools/dotc/transform/ElimRepeated.scala
Original file line number Diff line number Diff line change
Expand Up @@ -94,6 +94,8 @@ class ElimRepeated extends MiniPhase with InfoTransformer { thisPhase =>
private def seqToArray(tree: Tree, pt: Type)(implicit ctx: Context): Tree = tree match {
case SeqLiteral(elems, elemtpt) =>
JavaSeqLiteral(elems, elemtpt)
case app@Apply(fun, args) if isWrappedArray(app) => // rewrite a call to `wrapXArray(arr)` to `arr`
args.head
case _ =>
val elemType = tree.tpe.elemType
var elemClass = elemType.classSymbol
Expand All @@ -106,6 +108,14 @@ class ElimRepeated extends MiniPhase with InfoTransformer { thisPhase =>
// Because of phantomclasses, the Java array's type might not conform to the return type
}

/** Determines whether `tree` is a method call to Predef.wrapXArray */
private def isWrappedArray(tree: Apply)(implicit ctx: Context): Boolean = {
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@allanrenucci That's much nicer! Thanks! PTAL

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think the name of the method and doc are confusing. This tests that the tree is a method call to Predef.wrappedXArray, not that the tree represents a wrapped array. I would prefer:

/** Is the tree a method call to Predef.wrapXArray */
def isWrapArrayMethodCall

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done.

val elemTpe = tree.tpe.elemType
val wrapMethodName = TreeGen.wrapArrayMethodName(elemTpe)
val wrapMethodSym = defn.ScalaPredefModuleRef.denot.requiredMethod(wrapMethodName)
tree.fun.symbol == wrapMethodSym
}

override def transformTypeApply(tree: TypeApply)(implicit ctx: Context): Tree =
transformTypeOfTree(tree)

Expand Down
28 changes: 28 additions & 0 deletions compiler/test/dotty/tools/backend/jvm/DottyBytecodeTests.scala
Original file line number Diff line number Diff line change
Expand Up @@ -187,4 +187,32 @@ class TestBCode extends DottyBytecodeTest {
diffInstructions(instructions1, instructions2))
}
}

/** Verifies that arrays are not unnecessarily wrapped when passed to Java varargs methods */
@Test def dontWrapArraysInJavaVarargs = {
val source =
"""
|import java.nio.file._
|class Test {
| def test(xs: Array[String]) = {
| val p4 = Paths.get("Hello", xs: _*)
| }
|}
""".stripMargin

checkBCode(source) { dir =>
val moduleIn = dir.lookupName("Test.class", directory = false)
val moduleNode = loadClassNode(moduleIn.input)
val method = getMethod(moduleNode, "test")

val arrayWrapped = instructionsFromMethod(method).exists { instr: Instruction =>
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

You can do:

val arrayWrapped = instructionsFromMethod(method).exists {
  case inv: Invoke => inv.name.contains("wrapRefArray")
  case _ => false
}

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done.

instr match {
case inv: Invoke => inv.name.contains("wrapRefArray")
case _ => false
}
}

assert(!arrayWrapped, "Arrays should not be wrapped when passed to a Java varargs method\n")
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I verified that this fails if I comment out the changes to ElimRepeated.

}
}
}