• Home
• org.jetbrains.kotlin
• kotlin-compiler-embeddable
• 2.0.20-RC2
• source code
• setUtils.kt

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

org.jetbrains.kotlin.utils.setUtils.kt Maven / Gradle / Ivy

/*
 * Copyright 2010-2023 JetBrains s.r.o. and Kotlin Programming Language contributors.
 * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
 */

package org.jetbrains.kotlin.utils

/**
 * Works almost as regular flatMap, but returns a set and returns null if any lambda call returned null
 */
inline fun  Iterable.flatMapToNullableSet(transform: (T) -> Iterable?): Set? =
    flatMapTo(mutableSetOf()) { transform(it) ?: return null }.ifEmpty { emptySet() }

/**
 * Maps all elements of this non-empty collection with the given [transform] function to a new mutable set, or returns [emptySet] if this
 * collection is empty.
 *
 * [mapToSetOrEmpty] should be preferred over `collection.mapTo(mutableSetOf()) { ... }` when `collection` may be empty and the resulting
 * set may be cached, because [mapToSetOrEmpty] saves memory by avoiding the creation of an empty mutable set.
 */
inline fun  Collection.mapToSetOrEmpty(transform: (T) -> R): Set =
    if (isNotEmpty()) mapTo(mutableSetOf(), transform) else emptySet()

inline fun  Collection.filterToSetOrEmpty(predicate: (T) -> Boolean): Set =
    filterTo(mutableSetOf(), predicate).ifEmpty { emptySet() }