Skip to content

Commit 2b7ceed

Browse files
dwijnanddwijnand
committed
Re-implement with PolySig support w/ a Typed
1 parent 14a80fd commit 2b7ceed

File tree

11 files changed

+107
-119
lines changed

11 files changed

+107
-119
lines changed

‎compiler/src/dotty/tools/backend/sjs/JSCodeGen.scala‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4726,7 +4726,7 @@ class JSCodeGen()(using genCtx: Context) {
47264726
Set[Symbol](
47274727
defn.BoxedUnitClass, defn.BoxedBooleanClass, defn.BoxedCharClass, defn.BoxedByteClass,
47284728
defn.BoxedShortClass, defn.BoxedIntClass, defn.BoxedLongClass, defn.BoxedFloatClass,
4729-
defn.BoxedDoubleClass, defn.StringClass, defn.VoidClass
4729+
defn.BoxedDoubleClass, defn.StringClass, jsdefn.JavaLangVoidClass
47304730
)
47314731
}
47324732

‎compiler/src/dotty/tools/backend/sjs/JSDefinitions.scala‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,9 @@ final class JSDefinitions()(using Context) {
2626
@threadUnsafe lazy val NoinlineAnnotType: TypeRef = requiredClassRef("scala.noinline")
2727
def NoinlineAnnot(using Context) = NoinlineAnnotType.symbol.asClass
2828

29+
@threadUnsafe lazy val JavaLangVoidType: TypeRef = requiredClassRef("java.lang.Void")
30+
def JavaLangVoidClass(using Context) = JavaLangVoidType.symbol.asClass
31+
2932
@threadUnsafe lazy val ScalaJSJSPackageVal = requiredPackage("scala.scalajs.js")
3033
@threadUnsafe lazy val ScalaJSJSPackageClass = ScalaJSJSPackageVal.moduleClass.asClass
3134
@threadUnsafe lazy val JSPackage_typeOfR = ScalaJSJSPackageClass.requiredMethodRef("typeOf")

‎compiler/src/dotty/tools/dotc/core/Definitions.scala‎

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -485,9 +485,6 @@ class Definitions {
485485
}
486486
def NullType: TypeRef = NullClass.typeRef
487487

488-
@tu lazy val VoidType: TypeRef = requiredClassRef("java.lang.Void")
489-
def VoidClass(using Context) = VoidType.symbol.asClass
490-
491488
@tu lazy val InvokerModule = requiredModule("scala.runtime.coverage.Invoker")
492489
@tu lazy val InvokedMethodRef = InvokerModule.requiredMethodRef("invoked")
493490

@@ -735,7 +732,6 @@ class Definitions {
735732
}
736733
def JavaEnumType = JavaEnumClass.typeRef
737734

738-
739735
@tu lazy val MethodHandleClass: ClassSymbol = requiredClass("java.lang.invoke.MethodHandle")
740736
@tu lazy val MethodHandlesLookupClass: ClassSymbol = requiredClass("java.lang.invoke.MethodHandles.Lookup")
741737
@tu lazy val VarHandleClass: ClassSymbol = requiredClass("java.lang.invoke.VarHandle")

‎compiler/src/dotty/tools/dotc/core/tasty/TastyPrinter.scala‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,7 @@ class TastyPrinter(bytes: Array[Byte]) {
129129
printName(); printName()
130130
case VALDEF | DEFDEF | TYPEDEF | TYPEPARAM | PARAM | NAMEDARG | BIND =>
131131
printName(); printTrees()
132-
case REFINEDtype | TERMREFin | TYPEREFin | SELECTin | SELECTinPoly =>
132+
case REFINEDtype | TERMREFin | TYPEREFin | SELECTin =>
133133
printName(); printTree(); printTrees()
134134
case RETURN | HOLE =>
135135
printNat(); printTrees()

‎compiler/src/dotty/tools/dotc/core/tasty/TreePickler.scala‎

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -413,14 +413,6 @@ class TreePickler(pickler: TastyPickler) {
413413
writeByte(if name.isTypeName then SELECTtpt else SELECT)
414414
pickleNameAndSig(name, sig, ename)
415415
pickleTree(qual)
416-
else if defn.wasPolymorphicSignature(tree.symbol) then
417-
writeByte(SELECTinPoly)
418-
withLength {
419-
pickleNameAndSig(name, tree.symbol.signature, ename)
420-
pickleTree(qual)
421-
pickleType(tree.symbol.owner.typeRef)
422-
pickleType(tree.tpe.widenSingleton, richTypes = true) // this widens to a MethodType, so need richTypes
423-
}
424416
else // select from owner
425417
writeByte(SELECTin)
426418
withLength {

‎compiler/src/dotty/tools/dotc/core/tasty/TreeUnpickler.scala‎

Lines changed: 46 additions & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -1131,6 +1131,7 @@ class TreeUnpickler(reader: TastyReader,
11311131

11321132
def makeSelect(qual: Tree, name: Name, denot: Denotation): Select =
11331133
var qualType = qual.tpe.widenIfUnstable
1134+
val owner = denot.symbol.maybeOwner
11341135
val tpe0 = name match
11351136
case name: TypeName => TypeRef(qualType, name, denot)
11361137
case name: TermName => TermRef(qualType, name, denot)
@@ -1142,43 +1143,6 @@ class TreeUnpickler(reader: TastyReader,
11421143
val denot = accessibleDenot(qual.tpe.widenIfUnstable, name, sig, target)
11431144
makeSelect(qual, name, denot)
11441145

1145-
def readSelectIn(): Select =
1146-
var sname = readName()
1147-
val qual = readTerm()
1148-
val ownerTpe = readType()
1149-
val owner = ownerTpe.typeSymbol
1150-
val SignedName(name, sig, target) = sname: @unchecked // only methods with params use SELECTin
1151-
val qualType = qual.tpe.widenIfUnstable
1152-
val prefix = ctx.typeAssigner.maybeSkolemizePrefix(qualType, name)
1153-
1154-
/** Tasty should still be able to resolve a method from another root class,
1155-
* even if it has been moved to a super type,
1156-
* or an override has been removed.
1157-
*
1158-
* This is tested in
1159-
* - sbt-test/tasty-compat/remove-override
1160-
* - sbt-test/tasty-compat/move-method
1161-
*/
1162-
def lookupInSuper =
1163-
val cls = ownerTpe.classSymbol
1164-
if cls.exists then
1165-
cls.asClass.classDenot
1166-
.findMember(name, cls.thisType, EmptyFlags, excluded=Private)
1167-
.atSignature(sig, target)
1168-
else
1169-
NoDenotation
1170-
1171-
val denot =
1172-
val d = ownerTpe.decl(name).atSignature(sig, target)
1173-
(if !d.exists then lookupInSuper else d).asSeenFrom(prefix)
1174-
1175-
makeSelect(qual, name, denot)
1176-
1177-
def readSelectInPoly(): Select =
1178-
val tree = readSelectIn()
1179-
val info = readType()
1180-
tree.withType(tree.symbol.copy(info = info).termRef)
1181-
11821146
def readQualId(): (untpd.Ident, TypeRef) =
11831147
val qual = readTerm().asInstanceOf[untpd.Ident]
11841148
(untpd.Ident(qual.name).withSpan(qual.span), qual.tpe.asInstanceOf[TypeRef])
@@ -1269,13 +1233,25 @@ class TreeUnpickler(reader: TastyReader,
12691233
val fn = readTerm()
12701234
val args = until(end)(readTerm())
12711235
if fn.symbol.isConstructor then constructorApply(fn, args)
1272-
else tpd.Apply(fn, args)
1236+
else if defn.isPolymorphicSignature(fn.symbol) then
1237+
val info = MethodType(args.map(_.tpe.widen), defn.ObjectType)
1238+
val fun2 = fn.withType(fn.symbol.copy(info = info).termRef)
1239+
tpd.Apply(fun2, args)
1240+
else
1241+
tpd.Apply(fn, args)
12731242
case TYPEAPPLY =>
12741243
tpd.TypeApply(readTerm(), until(end)(readTpt()))
12751244
case TYPED =>
12761245
val expr = readTerm()
12771246
val tpt = readTpt()
1278-
Typed(expr, tpt)
1247+
expr match
1248+
case Apply(fun, args) if defn.wasPolymorphicSignature(fun.symbol) =>
1249+
val info = MethodType(args.map(_.tpe.widen), tpt.tpe)
1250+
val fun2 = fun.withType(fun.symbol.copy(info = info).termRef)
1251+
val expr2 = tpd.cpy.Apply(expr)(fun2, args)
1252+
Typed(expr2, tpt)
1253+
case _ =>
1254+
Typed(expr, tpt)
12791255
case ASSIGN =>
12801256
Assign(readTerm(), readTerm())
12811257
case BLOCK =>
@@ -1328,8 +1304,37 @@ class TreeUnpickler(reader: TastyReader,
13281304
case SELECTouter =>
13291305
val levels = readNat()
13301306
readTerm().outerSelect(levels, SkolemType(readType()))
1331-
case SELECTin => readSelectIn()
1332-
case SELECTinPoly => readSelectInPoly()
1307+
case SELECTin =>
1308+
var sname = readName()
1309+
val qual = readTerm()
1310+
val ownerTpe = readType()
1311+
val owner = ownerTpe.typeSymbol
1312+
val SignedName(name, sig, target) = sname: @unchecked // only methods with params use SELECTin
1313+
val qualType = qual.tpe.widenIfUnstable
1314+
val prefix = ctx.typeAssigner.maybeSkolemizePrefix(qualType, name)
1315+
1316+
/** Tasty should still be able to resolve a method from another root class,
1317+
* even if it has been moved to a super type,
1318+
* or an override has been removed.
1319+
*
1320+
* This is tested in
1321+
* - sbt-test/tasty-compat/remove-override
1322+
* - sbt-test/tasty-compat/move-method
1323+
*/
1324+
def lookupInSuper =
1325+
val cls = ownerTpe.classSymbol
1326+
if cls.exists then
1327+
cls.asClass.classDenot
1328+
.findMember(name, cls.thisType, EmptyFlags, excluded=Private)
1329+
.atSignature(sig, target)
1330+
else
1331+
NoDenotation
1332+
1333+
val denot =
1334+
val d = ownerTpe.decl(name).atSignature(sig, target)
1335+
(if !d.exists then lookupInSuper else d).asSeenFrom(prefix)
1336+
1337+
makeSelect(qual, name, denot)
13331338
case REPEATED =>
13341339
val elemtpt = readTpt()
13351340
SeqLiteral(until(end)(readTerm()), elemtpt)

‎compiler/src/dotty/tools/dotc/typer/Applications.scala‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -952,7 +952,8 @@ trait Applications extends Compatibility {
952952
// being brought into existence as needed
953953
val info = MethodType(proto.typedArgs().map(_.tpe.widen), resultType)
954954
val fun2 = fun1.withType(funRef.symbol.copy(info = info).termRef)
955-
simpleApply(fun2, proto)
955+
val app = simpleApply(fun2, proto)
956+
Typed(app, TypeTree(resultType))
956957
else
957958
val app = ApplyTo(tree, fun1, funRef, proto, pt)
958959
convertNewGenericArray(

‎project/Build.scala‎

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1816,10 +1816,9 @@ object Build {
18161816
settings(disableDocSetting).
18171817
settings(
18181818
versionScheme := Some("semver-spec"),
1819-
if (mode == Bootstrapped) Def.settings(
1820-
commonMiMaSettings,
1821-
mimaBinaryIssueFilters ++= MiMaFilters.TastyCore,
1822-
) else {
1819+
if (mode == Bootstrapped) {
1820+
commonMiMaSettings
1821+
} else {
18231822
Nil
18241823
}
18251824
)

‎project/MiMaFilters.scala‎

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,4 @@ object MiMaFilters {
2222
ProblemFilters.exclude[MissingClassProblem]("scala.caps$Pure"),
2323
ProblemFilters.exclude[MissingClassProblem]("scala.caps$unsafe$"),
2424
)
25-
val TastyCore: Seq[ProblemFilter] = Seq(
26-
ProblemFilters.exclude[MissingMethodProblem]("dotty.tools.tasty.TastyFormat.SELECTinPoly"),
27-
)
2825
}

‎tasty/src/dotty/tools/tasty/TastyFormat.scala‎

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,6 @@ Standard-Section: "ASTs" TopLevelStat*
8686
IDENT NameRef Type -- Used when term ident’s type is not a TermRef
8787
SELECT possiblySigned_NameRef qual_Term -- qual.name
8888
SELECTin Length possiblySigned_NameRef qual_Term owner_Type -- qual.name, referring to a symbol declared in owner that has the given signature (see note below)
89-
SELECTinPoly Length possiblySigned_NameRef qual_Term owner_Type method_Type -- like SELECTin, but with the method_Type too (because signature polymorphic)
9089
QUALTHIS typeIdent_Tree -- id.this, different from THIS in that it contains a qualifier ident with position.
9190
NEW clsType_Term -- new cls
9291
THROW throwableExpr_Term -- throw throwableExpr
@@ -579,7 +578,6 @@ object TastyFormat {
579578
// final val ??? = 178
580579
// final val ??? = 179
581580
final val METHODtype = 180
582-
final val SELECTinPoly = 181
583581

584582
final val MATCHtype = 190
585583
final val MATCHtpt = 191
@@ -779,7 +777,6 @@ object TastyFormat {
779777
case TERMREFin => "TERMREFin"
780778
case TYPEREFin => "TYPEREFin"
781779
case SELECTin => "SELECTin"
782-
case SELECTinPoly => "SELECTinPoly"
783780

784781
case REFINEDtype => "REFINEDtype"
785782
case REFINEDtpt => "REFINEDtpt"
@@ -810,7 +807,7 @@ object TastyFormat {
810807
*/
811808
def numRefs(tag: Int): Int = tag match {
812809
case VALDEF | DEFDEF | TYPEDEF | TYPEPARAM | PARAM | NAMEDARG | RETURN | BIND |
813-
SELFDEF | REFINEDtype | TERMREFin | TYPEREFin | SELECTin | SELECTinPoly | HOLE => 1
810+
SELFDEF | REFINEDtype | TERMREFin | TYPEREFin | SELECTin | HOLE => 1
814811
case RENAMED | PARAMtype => 2
815812
case POLYtype | TYPELAMBDAtype | METHODtype => -1
816813
case _ => 0

0 commit comments

Comments
 (0)