kscience/kmath
9
0
Fork 1
Code Issues Pull Requests 6 Packages Projects Releases 16 Wiki Activity

Feature/advanced optimization #404

Merged
altavir merged 17 commits from feature/advanced-optimization into dev 2021-08-29 11:44:10 +03:00
Conversation 0 Commits 17 Files Changed 154 +6 -4
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 9b8da4cdcc - Show all commits

10
kmath-core/src/commonMain/kotlin/space/kscience/kmath/misc/Featured.kt
View File

@ -16,13 +16,14 @@ public interface Featured<F : Any> {
public typealias FeatureKey<T> = KClass<out T> public typealias FeatureKey<T> = KClass<out T>
public interface Feature<F: Feature<F>> { public interface Feature<F : Feature<F>> {
/** /**
* A key used for extraction * A key used for extraction
*/ */
@Suppress("UNCHECKED_CAST") @Suppress("UNCHECKED_CAST")
public val key: FeatureKey<F> get() = this::class as FeatureKey<F> public val key: FeatureKey<F>
get() = this::class as FeatureKey<F>
} }
/** /**
@ -30,7 +31,7 @@ public interface Feature<F: Feature<F>> {
*/ */
public class FeatureSet<F : Feature<F>> private constructor(public val features: Map<FeatureKey<F>, F>) : Featured<F> { public class FeatureSet<F : Feature<F>> private constructor(public val features: Map<FeatureKey<F>, F>) : Featured<F> {
@Suppress("UNCHECKED_CAST") @Suppress("UNCHECKED_CAST")
override fun <T : F> getFeature(type: FeatureKey<T>): T? = features[type] as? T override fun <T : F> getFeature(type: FeatureKey<T>): T? = features[type]?.let { it as T }
public inline fun <reified T : F> getFeature(): T? = getFeature(T::class) public inline fun <reified T : F> getFeature(): T? = getFeature(T::class)
@ -49,6 +50,7 @@ public class FeatureSet<F : Feature<F>> private constructor(public val features:
public companion object { public companion object {
public fun <F : Feature<F>> of(vararg features: F): FeatureSet<F> = FeatureSet(features.associateBy { it.key }) public fun <F : Feature<F>> of(vararg features: F): FeatureSet<F> = FeatureSet(features.associateBy { it.key })
public fun <F : Feature<F>> of(features: Iterable<F>): FeatureSet<F> = FeatureSet(features.associateBy { it.key }) public fun <F : Feature<F>> of(features: Iterable<F>): FeatureSet<F> =
FeatureSet(features.associateBy { it.key })
} }
} }