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

cleanup parallel field example

Browse Source
This commit is contained in:
Alexander Nozik 2021-01-24 10:26:56 +03:00
parent 9829a16a32
commit 3bbb39f30d
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
examples/src/main/kotlin/kscience/kmath/structures/ParallelRealNDField.kt
View File

@ -12,12 +12,14 @@ import java.util.stream.IntStream
* A demonstration implementation of NDField over Real using Java [DoubleStream] for parallel execution * A demonstration implementation of NDField over Real using Java [DoubleStream] for parallel execution
*/ */
@OptIn(UnstableKMathAPI::class) @OptIn(UnstableKMathAPI::class)
public class StreamRealNDField( class StreamRealNDField(
shape: IntArray, override val shape: IntArray,
) : BufferedNDField<Double, RealField>(shape, RealField, Buffer.Companion::real), ) : NDField<Double, RealField>,
RingWithNumbers<NDStructure<Double>>, RingWithNumbers<NDStructure<Double>>,
ExtendedField<NDStructure<Double>> { ExtendedField<NDStructure<Double>> {
private val strides = DefaultStrides(shape)
override val elementContext: RealField get() = RealField
override val zero: NDBuffer<Double> by lazy { produce { zero } } override val zero: NDBuffer<Double> by lazy { produce { zero } }
override val one: NDBuffer<Double> by lazy { produce { one } } override val one: NDBuffer<Double> by lazy { produce { one } }
@ -26,7 +28,7 @@ public class StreamRealNDField(
return produce { d } return produce { d }
} }
override val NDStructure<Double>.buffer: RealBuffer private val NDStructure<Double>.buffer: RealBuffer
get() = when { get() = when {
!shape.contentEquals(this@StreamRealNDField.shape) -> throw ShapeMismatchException( !shape.contentEquals(this@StreamRealNDField.shape) -> throw ShapeMismatchException(
this@StreamRealNDField.shape, this@StreamRealNDField.shape,