/*
 * This file is part of LiquidBounce (https://github.com/CCBlueX/LiquidBounce)
 *
 * Copyright (c) 2015 - 2026 CCBlueX
 *
 * LiquidBounce is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * LiquidBounce is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with LiquidBounce. If not, see <https://www.gnu.org/licenses/>.
 */
package net.ccbluex.liquidbounce.utils.kotlin

import it.unimi.dsi.fastutil.objects.Object2IntMap
import it.unimi.dsi.fastutil.objects.Object2IntOpenHashMap
import net.ccbluex.fastutil.fastIterator
import java.util.function.Function

fun <K, V> memorizingFunction(map: MutableMap<K, V>, mappingFunction: Function<K, V>): Function<K, V> =
    Function { key -> map.computeIfAbsent(key, mappingFunction) }

inline fun <K : Any, V : Any> Map<K, V>.immutableCopy(): Map<K, V> =
    java.util.Map.copyOf(this)

/**
 * Sums the values for matching keys from another map to this map.
 * Modifies the current collection.
 */
fun <K> Object2IntOpenHashMap<K>.sumValues(anotherMap: Object2IntMap<K>): Object2IntMap<K> {
    anotherMap.fastIterator().forEach {
        this.addTo(it.key, it.intValue)
    }
    return this
}
