Skip to content

Commit 709a8cb

Browse files
committed
EnumSet complement
1 parent c22efa0 commit 709a8cb

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

buildSrc/src/main/kotlin/fastutil-ext-generator.gradle.kts

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -691,7 +691,7 @@ val enumSetTask = tasks.register<GenerateSrcTask>("enum-set") {
691691
appendLine("}")
692692
appendLine()
693693

694-
for (it in arrayOf("Array", "Sequence", "Iterable", "Stream")) {
694+
for (it in arrayOf("Array", "Sequence", "Iterable", "Iterator", "Stream")) {
695695
val generic1 = if (it == "Array") "<out E>" else "<E>"
696696
appendLine("inline fun $generic $it$generic1.toEnumSet(): EnumSet<E> {")
697697
withIndent {
@@ -700,6 +700,8 @@ val enumSetTask = tasks.register<GenerateSrcTask>("enum-set") {
700700
appendLine("}")
701701
appendLine()
702702
}
703+
704+
appendLine("inline fun <reified E : Enum<E>> EnumSet<E>.complement(): EnumSet<E> = EnumSet.complementOf(this)")
703705
}
704706
}
705707

0 commit comments

Comments
 (0)