Implement LUP decomposition in GSL module

This commit is contained in:
Iaroslav Postovalov 2021-01-24 01:42:20 +07:00
parent 54d61c016e
commit d9ebadd22a
No known key found for this signature in database
GPG Key ID: 46E15E4A31B3BCD7
13 changed files with 568 additions and 220 deletions

View File

@ -14,38 +14,53 @@ private fun KtPsiFactory.createMatrixClass(
kotlinTypeName: String, kotlinTypeName: String,
kotlinTypeAlias: String = kotlinTypeName kotlinTypeAlias: String = kotlinTypeName
) { ) {
fun fn(pattern: String) = fn(pattern, cTypeName)
val className = "Gsl${kotlinTypeAlias}Matrix" val className = "Gsl${kotlinTypeAlias}Matrix"
val structName = sn("gsl_matrixR", cTypeName) val structName = sn("gsl_matrixR", cTypeName)
@Language("kotlin") val text = """internal class $className( @Language("kotlin") val text = """internal class $className(
override val nativeHandle: CPointer<$structName>, override val rawNativeHandle: CPointer<$structName>,
scope: DeferScope scope: AutofreeScope,
) : GslMatrix<$kotlinTypeName, $structName>(scope) { ) : GslMatrix<$kotlinTypeName, $structName>(scope) {
override val rowNum: Int override val rowNum: Int
get() = nativeHandleChecked().pointed.size1.toInt() get() = nativeHandle.pointed.size1.toInt()
override val colNum: Int override val colNum: Int
get() = nativeHandleChecked().pointed.size2.toInt() get() = nativeHandle.pointed.size2.toInt()
override val rows: Buffer<Buffer<$kotlinTypeName>>
get() = VirtualBuffer(rowNum) { r ->
Gsl${kotlinTypeAlias}Vector(
${fn("gsl_matrixRrow")}(nativeHandle, r.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override val columns: Buffer<Buffer<$kotlinTypeName>>
get() = VirtualBuffer(rowNum) { c ->
Gsl${kotlinTypeAlias}Vector(
${fn("gsl_matrixRcolumn")}(nativeHandle, c.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override operator fun get(i: Int, j: Int): $kotlinTypeName = override operator fun get(i: Int, j: Int): $kotlinTypeName =
${fn("gsl_matrixRget", cTypeName)}(nativeHandleChecked(), i.toULong(), j.toULong()) ${fn("gsl_matrixRget")}(nativeHandle, i.toULong(), j.toULong())
override operator fun set(i: Int, j: Int, value: ${kotlinTypeName}): Unit = override operator fun set(i: Int, j: Int, value: ${kotlinTypeName}): Unit =
${fn("gsl_matrixRset", cTypeName)}(nativeHandleChecked(), i.toULong(), j.toULong(), value) ${fn("gsl_matrixRset")}(nativeHandle, i.toULong(), j.toULong(), value)
override fun copy(): $className { override fun copy(): $className {
val new = requireNotNull(${fn("gsl_matrixRalloc", cTypeName)}(rowNum.toULong(), colNum.toULong())) val new = requireNotNull(${fn("gsl_matrixRalloc")}(rowNum.toULong(), colNum.toULong()))
${fn("gsl_matrixRmemcpy", cTypeName)}(new, nativeHandleChecked()) ${fn("gsl_matrixRmemcpy")}(new, nativeHandle)
return $className(new, scope) return $className(new, scope)
} }
override fun close(): Unit = ${fn("gsl_matrixRfree", cTypeName)}(nativeHandleChecked()) override fun close(): Unit = ${fn("gsl_matrixRfree")}(nativeHandle)
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is $className) if (other is $className)
return ${ return ${fn("gsl_matrixRequal")}(nativeHandle, other.nativeHandle) == 1
fn("gsl_matrixRequal", cTypeName)
}(nativeHandleChecked(), other.nativeHandleChecked()) == 1
return super.equals(other) return super.equals(other)
} }
@ -64,7 +79,7 @@ fun matricesCodegen(outputFile: String, project: Project = createProject()) {
f += createNewLine(2) f += createNewLine(2)
f += createImportDirective(ImportPath.fromString("kotlinx.cinterop.*")) f += createImportDirective(ImportPath.fromString("kotlinx.cinterop.*"))
f += createNewLine(1) f += createNewLine(1)
f += createImportDirective(ImportPath.fromString("kscience.kmath.linear.*")) f += createImportDirective(ImportPath.fromString("kscience.kmath.structures.*"))
f += createNewLine(1) f += createNewLine(1)
f += createImportDirective(ImportPath.fromString("org.gnu.gsl.*")) f += createImportDirective(ImportPath.fromString("org.gnu.gsl.*"))
f += createNewLine(2) f += createNewLine(2)

View File

@ -14,38 +14,36 @@ private fun KtPsiFactory.createVectorClass(
kotlinTypeName: String, kotlinTypeName: String,
kotlinTypeAlias: String = kotlinTypeName kotlinTypeAlias: String = kotlinTypeName
) { ) {
fun fn(pattern: String) = fn(pattern, cTypeName)
val className = "Gsl${kotlinTypeAlias}Vector" val className = "Gsl${kotlinTypeAlias}Vector"
val structName = sn("gsl_vectorR", cTypeName) val structName = sn("gsl_vectorR", cTypeName)
@Language("kotlin") val text = @Language("kotlin") val text =
"""internal class $className(override val nativeHandle: CPointer<$structName>, scope: DeferScope) : """internal class $className(override val rawNativeHandle: CPointer<$structName>, scope: AutofreeScope) :
GslVector<$kotlinTypeName, $structName>(scope) { GslVector<$kotlinTypeName, $structName>(scope) {
override val size: Int override val size: Int
get() = nativeHandleChecked().pointed.size.toInt() get() = nativeHandle.pointed.size.toInt()
override operator fun get(index: Int): $kotlinTypeName = override operator fun get(index: Int): $kotlinTypeName =
${fn("gsl_vectorRget", cTypeName)}(nativeHandleChecked(), index.toULong()) ${fn("gsl_vectorRget")}(nativeHandle, index.toULong())
override operator fun set(index: Int, value: $kotlinTypeName): Unit = override operator fun set(index: Int, value: $kotlinTypeName): Unit =
${fn("gsl_vectorRset", cTypeName)}(nativeHandleChecked(), index.toULong(), value) ${fn("gsl_vectorRset")}(nativeHandle, index.toULong(), value)
override fun copy(): $className { override fun copy(): $className {
val new = requireNotNull(${fn("gsl_vectorRalloc", cTypeName)}(size.toULong())) val new = requireNotNull(${fn("gsl_vectorRalloc")}(size.toULong()))
${fn("gsl_vectorRmemcpy", cTypeName)}(new, nativeHandleChecked()) ${fn("gsl_vectorRmemcpy")}(new, nativeHandle)
return ${className}(new, scope) return ${className}(new, scope)
} }
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is $className) if (other is $className)
return ${ return ${fn("gsl_vectorRequal")}(nativeHandle, other.nativeHandle) == 1
fn("gsl_vectorRequal",
cTypeName)
}(nativeHandleChecked(), other.nativeHandleChecked()) == 1
return super.equals(other) return super.equals(other)
} }
override fun close(): Unit = ${fn("gsl_vectorRfree", cTypeName)}(nativeHandleChecked()) override fun close(): Unit = ${fn("gsl_vectorRfree")}(nativeHandle)
}""" }"""
f += createClass(text) f += createClass(text)

View File

@ -1,5 +1,6 @@
package kscience.kmath.linear package kscience.kmath.linear
import kscience.kmath.misc.UnstableKMathAPI
import kscience.kmath.operations.Ring import kscience.kmath.operations.Ring
import kscience.kmath.operations.SpaceOperations import kscience.kmath.operations.SpaceOperations
import kscience.kmath.operations.invoke import kscience.kmath.operations.invoke
@ -8,18 +9,22 @@ import kscience.kmath.structures.Buffer
import kscience.kmath.structures.BufferFactory import kscience.kmath.structures.BufferFactory
import kscience.kmath.structures.Matrix import kscience.kmath.structures.Matrix
import kscience.kmath.structures.asSequence import kscience.kmath.structures.asSequence
import kotlin.reflect.KClass
/** /**
* Basic operations on matrices. Operates on [Matrix] * Basic operations on matrices. Operates on [Matrix].
*
* @param T the type of items in the matrices.
* @param M the type of operated matrices.
*/ */
public interface MatrixContext<T : Any, out M : Matrix<T>> : SpaceOperations<Matrix<T>> { public interface MatrixContext<T : Any, out M : Matrix<T>> : SpaceOperations<Matrix<T>> {
/** /**
* Produce a matrix with this context and given dimensions * Produces a matrix with this context and given dimensions.
*/ */
public fun produce(rows: Int, columns: Int, initializer: (i: Int, j: Int) -> T): M public fun produce(rows: Int, columns: Int, initializer: (i: Int, j: Int) -> T): M
/** /**
* Produce a point compatible with matrix space (and possibly optimized for it) * Produces a point compatible with matrix space (and possibly optimized for it).
*/ */
public fun point(size: Int, initializer: (Int) -> T): Point<T> = Buffer.boxing(size, initializer) public fun point(size: Int, initializer: (Int) -> T): Point<T> = Buffer.boxing(size, initializer)
@ -66,8 +71,19 @@ public interface MatrixContext<T : Any, out M : Matrix<T>> : SpaceOperations<Mat
*/ */
public operator fun T.times(m: Matrix<T>): M = m * this public operator fun T.times(m: Matrix<T>): M = m * this
public companion object { /**
* Gets a feature from the matrix. This function may return some additional features than
* [kscience.kmath.structures.NDStructure.getFeature].
*
* @param F the type of feature.
* @param m the matrix.
* @param type the [KClass] instance of [F].
* @return a feature object or `null` if it isn't present.
*/
@UnstableKMathAPI
public fun <F : Any> getFeature(m: Matrix<T>, type: KClass<F>): F? = m.getFeature(type)
public companion object {
/** /**
* A structured matrix with custom buffer * A structured matrix with custom buffer
*/ */
@ -84,9 +100,31 @@ public interface MatrixContext<T : Any, out M : Matrix<T>> : SpaceOperations<Mat
} }
} }
/**
* Gets a feature from the matrix. This function may return some additional features than
* [kscience.kmath.structures.NDStructure.getFeature].
*
* @param T the type of items in the matrices.
* @param M the type of operated matrices.
* @param F the type of feature.
* @receiver the [MatrixContext] of [T].
* @param m the matrix.
* @return a feature object or `null` if it isn't present.
*/
@UnstableKMathAPI
public inline fun <T : Any, reified F : Any> MatrixContext<T, *>.getFeature(m: Matrix<T>): F? =
getFeature(m, F::class)
/**
* Partial implementation of [MatrixContext] for matrices of [Ring].
*
* @param T the type of items in the matrices.
* @param R the type of ring of matrix elements.
* @param M the type of operated matrices.
*/
public interface GenericMatrixContext<T : Any, R : Ring<T>, out M : Matrix<T>> : MatrixContext<T, M> { public interface GenericMatrixContext<T : Any, R : Ring<T>, out M : Matrix<T>> : MatrixContext<T, M> {
/** /**
* The ring context for matrix elements * The ring instance of [T].
*/ */
public val elementContext: R public val elementContext: R

View File

@ -241,11 +241,22 @@ public class ArrayBuffer<T>(private val array: Array<T>) : MutableBuffer<T> {
override fun copy(): MutableBuffer<T> = ArrayBuffer(array.copyOf()) override fun copy(): MutableBuffer<T> = ArrayBuffer(array.copyOf())
} }
/** /**
* Returns an [ArrayBuffer] that wraps the original array. * Returns an [ArrayBuffer] that wraps the original array.
*/ */
public fun <T> Array<T>.asBuffer(): ArrayBuffer<T> = ArrayBuffer(this) public fun <T> Array<T>.asBuffer(): ArrayBuffer<T> = ArrayBuffer(this)
/**
* Creates a new [ArrayBuffer] with the specified [size], where each element is calculated by calling the specified
* [init] function.
*
* The function [init] is called for each array element sequentially starting from the first one.
* It should return the value for an array element given its index.
*/
public inline fun <reified T> ArrayBuffer(size: Int, init: (Int) -> T): ArrayBuffer<T> =
Array(size) { i -> init(i) }.asBuffer()
/** /**
* Immutable wrapper for [MutableBuffer]. * Immutable wrapper for [MutableBuffer].
* *

View File

@ -1,5 +1,5 @@
package=org.gnu.gsl package=org.gnu.gsl
headers=gsl/gsl_blas.h gsl/gsl_linalg.h headers=gsl/gsl_blas.h gsl/gsl_linalg.h gsl/gsl_permute_matrix.h
compilerOpts.linux=-I/usr/include compilerOpts.linux=-I/usr/include
compilerOpts.osx=-I/usr/local -I/usr/local/include compilerOpts.osx=-I/usr/local -I/usr/local/include
linkerOpts.linux=-L/usr/lib64 -L/usr/lib/x86_64-linux-gnu -lgsl linkerOpts.linux=-L/usr/lib64 -L/usr/lib/x86_64-linux-gnu -lgsl

View File

@ -2,6 +2,8 @@ package kscience.kmath.gsl
import kotlinx.cinterop.* import kotlinx.cinterop.*
import kscience.kmath.operations.Complex import kscience.kmath.operations.Complex
import kscience.kmath.structures.Buffer
import kscience.kmath.structures.VirtualBuffer
import org.gnu.gsl.* import org.gnu.gsl.*
internal fun CValue<gsl_complex>.toKMath(): Complex = useContents { Complex(dat[0], dat[1]) } internal fun CValue<gsl_complex>.toKMath(): Complex = useContents { Complex(dat[0], dat[1]) }
@ -11,58 +13,74 @@ internal fun Complex.toGsl(): CValue<gsl_complex> = cValue {
dat[1] = im dat[1] = im
} }
internal class GslComplexMatrix(override val nativeHandle: CPointer<gsl_matrix_complex>, scope: DeferScope) : internal class GslComplexMatrix(override val rawNativeHandle: CPointer<gsl_matrix_complex>, scope: AutofreeScope) :
GslMatrix<Complex, gsl_matrix_complex>(scope) { GslMatrix<Complex, gsl_matrix_complex>(scope) {
override val rowNum: Int override val rowNum: Int
get() = nativeHandleChecked().pointed.size1.toInt() get() = nativeHandle.pointed.size1.toInt()
override val colNum: Int override val colNum: Int
get() = nativeHandleChecked().pointed.size2.toInt() get() = nativeHandle.pointed.size2.toInt()
override val rows: Buffer<Buffer<Complex>>
get() = VirtualBuffer(rowNum) { r ->
GslComplexVector(
gsl_matrix_complex_row(nativeHandle, r.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override val columns: Buffer<Buffer<Complex>>
get() = VirtualBuffer(rowNum) { c ->
GslComplexVector(
gsl_matrix_complex_column(nativeHandle, c.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override operator fun get(i: Int, j: Int): Complex = override operator fun get(i: Int, j: Int): Complex =
gsl_matrix_complex_get(nativeHandleChecked(), i.toULong(), j.toULong()).toKMath() gsl_matrix_complex_get(nativeHandle, i.toULong(), j.toULong()).toKMath()
override operator fun set(i: Int, j: Int, value: Complex): Unit = override operator fun set(i: Int, j: Int, value: Complex): Unit =
gsl_matrix_complex_set(nativeHandleChecked(), i.toULong(), j.toULong(), value.toGsl()) gsl_matrix_complex_set(nativeHandle, i.toULong(), j.toULong(), value.toGsl())
override fun copy(): GslComplexMatrix { override fun copy(): GslComplexMatrix {
val new = requireNotNull(gsl_matrix_complex_alloc(rowNum.toULong(), colNum.toULong())) val new = requireNotNull(gsl_matrix_complex_alloc(rowNum.toULong(), colNum.toULong()))
gsl_matrix_complex_memcpy(new, nativeHandleChecked()) gsl_matrix_complex_memcpy(new, nativeHandle)
return GslComplexMatrix(new, scope) return GslComplexMatrix(new, scope)
} }
override fun close(): Unit = gsl_matrix_complex_free(nativeHandleChecked()) override fun close(): Unit = gsl_matrix_complex_free(nativeHandle)
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslComplexMatrix) if (other is GslComplexMatrix)
return gsl_matrix_complex_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_matrix_complex_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
} }
internal class GslComplexVector(override val nativeHandle: CPointer<gsl_vector_complex>, scope: DeferScope) : internal class GslComplexVector(override val rawNativeHandle: CPointer<gsl_vector_complex>, scope: AutofreeScope) :
GslVector<Complex, gsl_vector_complex>(scope) { GslVector<Complex, gsl_vector_complex>(scope) {
override val size: Int override val size: Int
get() = nativeHandleChecked().pointed.size.toInt() get() = nativeHandle.pointed.size.toInt()
override fun get(index: Int): Complex = gsl_vector_complex_get(nativeHandleChecked(), index.toULong()).toKMath() override fun get(index: Int): Complex = gsl_vector_complex_get(nativeHandle, index.toULong()).toKMath()
override fun set(index: Int, value: Complex): Unit = override fun set(index: Int, value: Complex): Unit =
gsl_vector_complex_set(nativeHandleChecked(), index.toULong(), value.toGsl()) gsl_vector_complex_set(nativeHandle, index.toULong(), value.toGsl())
override fun copy(): GslComplexVector { override fun copy(): GslComplexVector {
val new = requireNotNull(gsl_vector_complex_alloc(size.toULong())) val new = requireNotNull(gsl_vector_complex_alloc(size.toULong()))
gsl_vector_complex_memcpy(new, nativeHandleChecked()) gsl_vector_complex_memcpy(new, nativeHandle)
return GslComplexVector(new, scope) return GslComplexVector(new, scope)
} }
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslComplexVector) if (other is GslComplexVector)
return gsl_vector_complex_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_vector_complex_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
override fun close(): Unit = gsl_vector_complex_free(nativeHandleChecked()) override fun close(): Unit = gsl_vector_complex_free(nativeHandle)
} }

View File

@ -1,33 +1,40 @@
package kscience.kmath.gsl package kscience.kmath.gsl
import kotlinx.cinterop.AutofreeScope
import kotlinx.cinterop.CStructVar import kotlinx.cinterop.CStructVar
import kotlinx.cinterop.DeferScope
import kscience.kmath.structures.Matrix import kscience.kmath.structures.Matrix
import kscience.kmath.structures.NDStructure import kscience.kmath.structures.NDStructure
import kscience.kmath.structures.asSequence
/** /**
* Wraps gsl_matrix_* objects from GSL. * Wraps gsl_matrix_* objects from GSL.
*/ */
public abstract class GslMatrix<T : Any, H : CStructVar> internal constructor(scope: DeferScope) : public abstract class GslMatrix<T : Any, H : CStructVar> internal constructor(scope: AutofreeScope) :
GslMemoryHolder<H>(scope), Matrix<T> { GslMemoryHolder<H>(scope), Matrix<T> {
internal abstract operator fun set(i: Int, j: Int, value: T) internal abstract operator fun set(i: Int, j: Int, value: T)
internal abstract fun copy(): GslMatrix<T, H> internal abstract fun copy(): GslMatrix<T, H>
public override fun equals(other: Any?): Boolean { public override fun equals(other: Any?): Boolean {
return NDStructure.equals(this, other as? NDStructure<*> ?: return false) return NDStructure.contentEquals(this, other as? NDStructure<*> ?: return false)
} }
public override fun hashCode(): Int { public override fun hashCode(): Int {
var ret = 7 var ret = 7
val nRows = rowNum ret = ret * 31 + rowNum
val nCols = colNum ret = ret * 31 + colNum
ret = ret * 31 + nRows
ret = ret * 31 + nCols
for (row in 0 until nRows) for (row in 0 until rowNum)
for (col in 0 until nCols) for (col in 0 until colNum)
ret = ret * 31 + (11 * (row + 1) + 17 * (col + 1)) * this[row, col].hashCode() ret = ret * 31 + (11 * (row + 1) + 17 * (col + 1)) * this[row, col].hashCode()
return ret return ret
} }
public override fun toString(): String = if (rowNum <= 5 && colNum <= 5)
"Matrix(rowsNum = $rowNum, colNum = $colNum)\n" +
rows.asSequence().joinToString(prefix = "(", postfix = ")", separator = "\n ") { buffer ->
buffer.asSequence().joinToString(separator = "\t") { it.toString() }
}
else
"Matrix(rowsNum = $rowNum, colNum = $colNum)"
} }

View File

@ -1,31 +1,33 @@
package kscience.kmath.gsl package kscience.kmath.gsl
import kotlinx.cinterop.CStructVar import kotlinx.cinterop.*
import kotlinx.cinterop.DeferScope import kscience.kmath.linear.*
import kotlinx.cinterop.memScoped import kscience.kmath.misc.UnstableKMathAPI
import kotlinx.cinterop.pointed
import kscience.kmath.linear.MatrixContext
import kscience.kmath.linear.Point
import kscience.kmath.operations.Complex import kscience.kmath.operations.Complex
import kscience.kmath.operations.ComplexField import kscience.kmath.operations.ComplexField
import kscience.kmath.operations.toComplex import kscience.kmath.operations.toComplex
import kscience.kmath.structures.Matrix import kscience.kmath.structures.Matrix
import kscience.kmath.structures.toList
import org.gnu.gsl.* import org.gnu.gsl.*
import kotlin.reflect.KClass
import kotlin.reflect.cast
internal inline fun <T : Any, H : CStructVar> GslMatrix<T, H>.fill(initializer: (Int, Int) -> T): GslMatrix<T, H> = internal inline fun <T : Any, H : CStructVar> GslMatrix<T, H>.fill(initializer: (Int, Int) -> T): GslMatrix<T, H> =
apply { apply {
(0 until rowNum).forEach { row -> (0 until colNum).forEach { col -> this[row, col] = initializer(row, col) } } for (row in 0 until rowNum) {
for (col in 0 until colNum) this[row, col] = initializer(row, col)
}
} }
internal inline fun <T : Any, H : CStructVar> GslVector<T, H>.fill(initializer: (Int) -> T): GslVector<T, H> = internal inline fun <T : Any, H : CStructVar> GslVector<T, H>.fill(initializer: (Int) -> T): GslVector<T, H> =
apply { (0 until size).forEach { index -> this[index] = initializer(index) } } apply {
for (index in 0 until size) this[index] = initializer(index)
}
/** /**
* Represents matrix context implementing where all the operations are delegated to GSL. * Represents matrix context implementing where all the operations are delegated to GSL.
*/ */
public abstract class GslMatrixContext<T : Any, H1 : CStructVar, H2 : CStructVar> internal constructor( public abstract class GslMatrixContext<T : Any, H1 : CStructVar, H2 : CStructVar> : MatrixContext<T, GslMatrix<T, H1>> {
internal val scope: DeferScope,
) : MatrixContext<T, GslMatrix<T, H1>> {
init { init {
ensureHasGslErrorHandler() ensureHasGslErrorHandler()
} }
@ -52,19 +54,22 @@ public abstract class GslMatrixContext<T : Any, H1 : CStructVar, H2 : CStructVar
public override fun produce(rows: Int, columns: Int, initializer: (i: Int, j: Int) -> T): GslMatrix<T, H1> = public override fun produce(rows: Int, columns: Int, initializer: (i: Int, j: Int) -> T): GslMatrix<T, H1> =
produceDirtyMatrix(rows, columns).fill(initializer) produceDirtyMatrix(rows, columns).fill(initializer)
public override fun point(size: Int, initializer: (Int) -> T): Point<T> = public override fun point(size: Int, initializer: (Int) -> T): GslVector<T, H2> =
produceDirtyVector(size).fill(initializer) produceDirtyVector(size).fill(initializer)
} }
/** /**
* Represents [Double] matrix context implementing where all the operations are delegated to GSL. * Represents [Double] matrix context implementing where all the operations are delegated to GSL.
*/ */
public class GslRealMatrixContext(scope: DeferScope) : GslMatrixContext<Double, gsl_matrix, gsl_vector>(scope) { public class GslRealMatrixContext(internal val scope: AutofreeScope) :
override fun produceDirtyMatrix(rows: Int, columns: Int): GslMatrix<Double, gsl_matrix> = GslMatrixContext<Double, gsl_matrix, gsl_vector>() {
GslRealMatrix(nativeHandle = requireNotNull(gsl_matrix_alloc(rows.toULong(), columns.toULong())), scope = scope) override fun produceDirtyMatrix(rows: Int, columns: Int): GslMatrix<Double, gsl_matrix> = GslRealMatrix(
rawNativeHandle = requireNotNull(gsl_matrix_alloc(rows.toULong(), columns.toULong())),
scope = scope,
)
override fun produceDirtyVector(size: Int): GslVector<Double, gsl_vector> = override fun produceDirtyVector(size: Int): GslVector<Double, gsl_vector> =
GslRealVector(nativeHandle = requireNotNull(gsl_vector_alloc(size.toULong())), scope = scope) GslRealVector(rawNativeHandle = requireNotNull(gsl_vector_alloc(size.toULong())), scope = scope)
public override fun Matrix<Double>.dot(other: Matrix<Double>): GslMatrix<Double, gsl_matrix> { public override fun Matrix<Double>.dot(other: Matrix<Double>): GslMatrix<Double, gsl_matrix> {
val x = toGsl().nativeHandle val x = toGsl().nativeHandle
@ -105,6 +110,58 @@ public class GslRealMatrixContext(scope: DeferScope) : GslMatrixContext<Double,
gsl_matrix_sub(g1.nativeHandle, b.toGsl().nativeHandle) gsl_matrix_sub(g1.nativeHandle, b.toGsl().nativeHandle)
return g1 return g1
} }
@Suppress("IMPLICIT_CAST_TO_ANY")
@UnstableKMathAPI
public override fun <F : Any> getFeature(m: Matrix<Double>, type: KClass<F>): F? = when (type) {
LupDecompositionFeature::class -> object : LupDecompositionFeature<Double> {
private val lup by lazy {
val lu = m.toGsl().copy()
val n = lu.rowNum
val perm = gsl_permutation_alloc(n.toULong())
memScoped {
val i = alloc<IntVar>()
gsl_linalg_LU_decomp(lu.nativeHandle, perm, i.ptr)
}
val one = produce(n, n) { i, j -> if (i == j) 1.0 else 0.0 }
val p = produce(n, n) { _, _ -> 0.0 }
for (j in 0 until perm!!.pointed.size.toInt()) {
println("7 $j")
val k = gsl_permutation_get(perm, j.toULong()).toInt()
val col = one.columns[k]
println(p.toGsl())
println(col.toList())
gsl_matrix_set_col(p.nativeHandle, j.toULong(), col.toGsl().nativeHandle)
}
gsl_permutation_free(perm)
lu to p
}
override val l by lazy {
VirtualMatrix(lup.first.shape[0], lup.first.shape[1]) { i, j ->
when {
j < i -> lup.first[i, j]
j == i -> 1.0
else -> 0.0
}
} + LFeature
}
override val u by lazy {
VirtualMatrix(
lup.first.shape[0],
lup.first.shape[1],
) { i, j -> if (j >= i) lup.first[i, j] else 0.0 } + UFeature
}
override val p by lazy(lup::second)
}
else -> super.getFeature(m, type)
}?.let(type::cast)
} }
/** /**
@ -116,22 +173,22 @@ public fun <R> GslRealMatrixContext(block: GslRealMatrixContext.() -> R): R =
/** /**
* Represents [Float] matrix context implementing where all the operations are delegated to GSL. * Represents [Float] matrix context implementing where all the operations are delegated to GSL.
*/ */
public class GslFloatMatrixContext(scope: DeferScope) : public class GslFloatMatrixContext(internal val scope: AutofreeScope) :
GslMatrixContext<Float, gsl_matrix_float, gsl_vector_float>(scope) { GslMatrixContext<Float, gsl_matrix_float, gsl_vector_float>() {
override fun produceDirtyMatrix(rows: Int, columns: Int): GslMatrix<Float, gsl_matrix_float> = GslFloatMatrix( override fun produceDirtyMatrix(rows: Int, columns: Int): GslMatrix<Float, gsl_matrix_float> = GslFloatMatrix(
nativeHandle = requireNotNull(gsl_matrix_float_alloc(rows.toULong(), columns.toULong())), rawNativeHandle = requireNotNull(gsl_matrix_float_alloc(rows.toULong(), columns.toULong())),
scope = scope, scope = scope,
) )
override fun produceDirtyVector(size: Int): GslVector<Float, gsl_vector_float> = override fun produceDirtyVector(size: Int): GslVector<Float, gsl_vector_float> =
GslFloatVector(nativeHandle = requireNotNull(value = gsl_vector_float_alloc(size.toULong())), scope = scope) GslFloatVector(rawNativeHandle = requireNotNull(value = gsl_vector_float_alloc(size.toULong())), scope = scope)
public override fun Matrix<Float>.dot(other: Matrix<Float>): GslMatrix<Float, gsl_matrix_float> { public override fun Matrix<Float>.dot(other: Matrix<Float>): GslMatrix<Float, gsl_matrix_float> {
val x = toGsl().nativeHandle val x = toGsl().nativeHandle
val a = other.toGsl().nativeHandle val a = other.toGsl().nativeHandle
val result = requireNotNull(gsl_matrix_float_calloc(a.pointed.size1, a.pointed.size2)) val result = requireNotNull(gsl_matrix_float_calloc(a.pointed.size1, a.pointed.size2))
gsl_blas_sgemm(CblasNoTrans, CblasNoTrans, 1f, x, a, 1f, result) gsl_blas_sgemm(CblasNoTrans, CblasNoTrans, 1f, x, a, 1f, result)
return GslFloatMatrix(nativeHandle = result, scope = scope) return GslFloatMatrix(rawNativeHandle = result, scope = scope)
} }
public override fun Matrix<Float>.dot(vector: Point<Float>): GslVector<Float, gsl_vector_float> { public override fun Matrix<Float>.dot(vector: Point<Float>): GslVector<Float, gsl_vector_float> {
@ -139,7 +196,7 @@ public class GslFloatMatrixContext(scope: DeferScope) :
val a = vector.toGsl().nativeHandle val a = vector.toGsl().nativeHandle
val result = requireNotNull(gsl_vector_float_calloc(a.pointed.size)) val result = requireNotNull(gsl_vector_float_calloc(a.pointed.size))
gsl_blas_sgemv(CblasNoTrans, 1f, x, a, 1f, result) gsl_blas_sgemv(CblasNoTrans, 1f, x, a, 1f, result)
return GslFloatVector(nativeHandle = result, scope = scope) return GslFloatVector(rawNativeHandle = result, scope = scope)
} }
public override fun Matrix<Float>.times(value: Float): GslMatrix<Float, gsl_matrix_float> { public override fun Matrix<Float>.times(value: Float): GslMatrix<Float, gsl_matrix_float> {
@ -176,22 +233,22 @@ public fun <R> GslFloatMatrixContext(block: GslFloatMatrixContext.() -> R): R =
/** /**
* Represents [Complex] matrix context implementing where all the operations are delegated to GSL. * Represents [Complex] matrix context implementing where all the operations are delegated to GSL.
*/ */
public class GslComplexMatrixContext(scope: DeferScope) : public class GslComplexMatrixContext(internal val scope: AutofreeScope) :
GslMatrixContext<Complex, gsl_matrix_complex, gsl_vector_complex>(scope) { GslMatrixContext<Complex, gsl_matrix_complex, gsl_vector_complex>() {
override fun produceDirtyMatrix(rows: Int, columns: Int): GslMatrix<Complex, gsl_matrix_complex> = GslComplexMatrix( override fun produceDirtyMatrix(rows: Int, columns: Int): GslMatrix<Complex, gsl_matrix_complex> = GslComplexMatrix(
nativeHandle = requireNotNull(gsl_matrix_complex_alloc(rows.toULong(), columns.toULong())), rawNativeHandle = requireNotNull(gsl_matrix_complex_alloc(rows.toULong(), columns.toULong())),
scope = scope, scope = scope,
) )
override fun produceDirtyVector(size: Int): GslVector<Complex, gsl_vector_complex> = override fun produceDirtyVector(size: Int): GslVector<Complex, gsl_vector_complex> =
GslComplexVector(nativeHandle = requireNotNull(gsl_vector_complex_alloc(size.toULong())), scope = scope) GslComplexVector(rawNativeHandle = requireNotNull(gsl_vector_complex_alloc(size.toULong())), scope = scope)
public override fun Matrix<Complex>.dot(other: Matrix<Complex>): GslMatrix<Complex, gsl_matrix_complex> { public override fun Matrix<Complex>.dot(other: Matrix<Complex>): GslMatrix<Complex, gsl_matrix_complex> {
val x = toGsl().nativeHandle val x = toGsl().nativeHandle
val a = other.toGsl().nativeHandle val a = other.toGsl().nativeHandle
val result = requireNotNull(gsl_matrix_complex_calloc(a.pointed.size1, a.pointed.size2)) val result = requireNotNull(gsl_matrix_complex_calloc(a.pointed.size1, a.pointed.size2))
gsl_blas_zgemm(CblasNoTrans, CblasNoTrans, ComplexField.one.toGsl(), x, a, ComplexField.one.toGsl(), result) gsl_blas_zgemm(CblasNoTrans, CblasNoTrans, ComplexField.one.toGsl(), x, a, ComplexField.one.toGsl(), result)
return GslComplexMatrix(nativeHandle = result, scope = scope) return GslComplexMatrix(rawNativeHandle = result, scope = scope)
} }
public override fun Matrix<Complex>.dot(vector: Point<Complex>): GslVector<Complex, gsl_vector_complex> { public override fun Matrix<Complex>.dot(vector: Point<Complex>): GslVector<Complex, gsl_vector_complex> {
@ -225,6 +282,58 @@ public class GslComplexMatrixContext(scope: DeferScope) :
gsl_matrix_complex_sub(g1.nativeHandle, b.toGsl().nativeHandle) gsl_matrix_complex_sub(g1.nativeHandle, b.toGsl().nativeHandle)
return g1 return g1
} }
@Suppress("IMPLICIT_CAST_TO_ANY")
@UnstableKMathAPI
public override fun <F : Any> getFeature(m: Matrix<Complex>, type: KClass<F>): F? = when (type) {
LupDecompositionFeature::class -> object : LupDecompositionFeature<Complex> {
private val lup by lazy {
val lu = m.toGsl().copy()
val n = lu.rowNum
val perm = gsl_permutation_alloc(n.toULong())
memScoped {
val i = alloc<IntVar>()
gsl_linalg_complex_LU_decomp(lu.nativeHandle, perm, i.ptr)
}
val one = produce(n, n) { i, j -> if (i == j) 1.0.toComplex() else 0.0.toComplex() }
val p = produce(n, n) { _, _ -> 0.0.toComplex() }
for (j in 0 until perm!!.pointed.size.toInt()) {
println("7 $j")
val k = gsl_permutation_get(perm, j.toULong()).toInt()
val col = one.columns[k]
println(p.toGsl())
println(col.toList())
gsl_matrix_complex_set_col(p.nativeHandle, j.toULong(), col.toGsl().nativeHandle)
}
gsl_permutation_free(perm)
lu to p
}
override val l by lazy {
VirtualMatrix(lup.first.shape[0], lup.first.shape[1]) { i, j ->
when {
j < i -> lup.first[i, j]
j == i -> 1.0.toComplex()
else -> 0.0.toComplex()
}
} + LFeature
}
override val u by lazy {
VirtualMatrix(
lup.first.shape[0],
lup.first.shape[1],
) { i, j -> if (j >= i) lup.first[i, j] else 0.0.toComplex() } + UFeature
}
override val p by lazy(lup::second)
}
else -> super.getFeature(m, type)
}?.let(type::cast)
} }
/** /**

View File

@ -1,5 +1,6 @@
package kscience.kmath.gsl package kscience.kmath.gsl
import kotlinx.cinterop.AutofreeScope
import kotlinx.cinterop.CPointer import kotlinx.cinterop.CPointer
import kotlinx.cinterop.CStructVar import kotlinx.cinterop.CStructVar
import kotlinx.cinterop.DeferScope import kotlinx.cinterop.DeferScope
@ -12,23 +13,26 @@ import kotlinx.cinterop.DeferScope
* *
* @param scope the scope where this object is declared. * @param scope the scope where this object is declared.
*/ */
public abstract class GslMemoryHolder<H : CStructVar> internal constructor(internal val scope: DeferScope) { public abstract class GslMemoryHolder<H : CStructVar> internal constructor(internal val scope: AutofreeScope) {
internal abstract val nativeHandle: CPointer<H> internal abstract val rawNativeHandle: CPointer<H>
private var isClosed: Boolean = false private var isClosed: Boolean = false
internal val nativeHandle: CPointer<H>
get() {
check(!isClosed) { "The use of GSL object that is closed." }
return rawNativeHandle
}
internal var shouldBeFreed = true
init { init {
ensureHasGslErrorHandler() ensureHasGslErrorHandler()
scope.defer { scope.defer {
close() if (shouldBeFreed) close()
isClosed = true isClosed = true
} }
} }
internal fun nativeHandleChecked(): CPointer<H> {
check(!isClosed) { "The use of GSL object that is closed." }
return nativeHandle
}
internal abstract fun close() internal abstract fun close()
} }

View File

@ -1,13 +1,13 @@
package kscience.kmath.gsl package kscience.kmath.gsl
import kotlinx.cinterop.AutofreeScope
import kotlinx.cinterop.CStructVar import kotlinx.cinterop.CStructVar
import kotlinx.cinterop.DeferScope
import kscience.kmath.linear.Point import kscience.kmath.linear.Point
/** /**
* Wraps gsl_vector_* objects from GSL. * Wraps gsl_vector_* objects from GSL.
*/ */
public abstract class GslVector<T, H : CStructVar> internal constructor(scope: DeferScope) : public abstract class GslVector<T, H : CStructVar> internal constructor(scope: AutofreeScope) :
GslMemoryHolder<H>(scope), Point<T> { GslMemoryHolder<H>(scope), Point<T> {
internal abstract operator fun set(index: Int, value: T) internal abstract operator fun set(index: Int, value: T)
internal abstract fun copy(): GslVector<T, H> internal abstract fun copy(): GslVector<T, H>

View File

@ -1,260 +1,388 @@
package kscience.kmath.gsl package kscience.kmath.gsl
import kotlinx.cinterop.* import kotlinx.cinterop.*
import kscience.kmath.linear.* import kscience.kmath.structures.*
import org.gnu.gsl.* import org.gnu.gsl.*
internal class GslRealMatrix( internal class GslRealMatrix(
override val nativeHandle: CPointer<gsl_matrix>, override val rawNativeHandle: CPointer<gsl_matrix>,
scope: DeferScope scope: AutofreeScope,
) : GslMatrix<Double, gsl_matrix>(scope) { ) : GslMatrix<Double, gsl_matrix>(scope) {
override val rowNum: Int override val rowNum: Int
get() = nativeHandleChecked().pointed.size1.toInt() get() = nativeHandle.pointed.size1.toInt()
override val colNum: Int override val colNum: Int
get() = nativeHandleChecked().pointed.size2.toInt() get() = nativeHandle.pointed.size2.toInt()
override val rows: Buffer<Buffer<Double>>
get() = VirtualBuffer(rowNum) { r ->
GslRealVector(
gsl_matrix_row(nativeHandle, r.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override val columns: Buffer<Buffer<Double>>
get() = VirtualBuffer(rowNum) { c ->
GslRealVector(
gsl_matrix_column(nativeHandle, c.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override operator fun get(i: Int, j: Int): Double = override operator fun get(i: Int, j: Int): Double =
gsl_matrix_get(nativeHandleChecked(), i.toULong(), j.toULong()) gsl_matrix_get(nativeHandle, i.toULong(), j.toULong())
override operator fun set(i: Int, j: Int, value: Double): Unit = override operator fun set(i: Int, j: Int, value: Double): Unit =
gsl_matrix_set(nativeHandleChecked(), i.toULong(), j.toULong(), value) gsl_matrix_set(nativeHandle, i.toULong(), j.toULong(), value)
override fun copy(): GslRealMatrix { override fun copy(): GslRealMatrix {
val new = requireNotNull(gsl_matrix_alloc(rowNum.toULong(), colNum.toULong())) val new = requireNotNull(gsl_matrix_alloc(rowNum.toULong(), colNum.toULong()))
gsl_matrix_memcpy(new, nativeHandleChecked()) gsl_matrix_memcpy(new, nativeHandle)
return GslRealMatrix(new, scope) return GslRealMatrix(new, scope)
} }
override fun close(): Unit = gsl_matrix_free(nativeHandleChecked()) override fun close(): Unit = gsl_matrix_free(nativeHandle)
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslRealMatrix) if (other is GslRealMatrix)
return gsl_matrix_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_matrix_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
} }
internal class GslFloatMatrix( internal class GslFloatMatrix(
override val nativeHandle: CPointer<gsl_matrix_float>, override val rawNativeHandle: CPointer<gsl_matrix_float>,
scope: DeferScope scope: AutofreeScope,
) : GslMatrix<Float, gsl_matrix_float>(scope) { ) : GslMatrix<Float, gsl_matrix_float>(scope) {
override val rowNum: Int override val rowNum: Int
get() = nativeHandleChecked().pointed.size1.toInt() get() = nativeHandle.pointed.size1.toInt()
override val colNum: Int override val colNum: Int
get() = nativeHandleChecked().pointed.size2.toInt() get() = nativeHandle.pointed.size2.toInt()
override val rows: Buffer<Buffer<Float>>
get() = VirtualBuffer(rowNum) { r ->
GslFloatVector(
gsl_matrix_float_row(nativeHandle, r.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override val columns: Buffer<Buffer<Float>>
get() = VirtualBuffer(rowNum) { c ->
GslFloatVector(
gsl_matrix_float_column(nativeHandle, c.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override operator fun get(i: Int, j: Int): Float = override operator fun get(i: Int, j: Int): Float =
gsl_matrix_float_get(nativeHandleChecked(), i.toULong(), j.toULong()) gsl_matrix_float_get(nativeHandle, i.toULong(), j.toULong())
override operator fun set(i: Int, j: Int, value: Float): Unit = override operator fun set(i: Int, j: Int, value: Float): Unit =
gsl_matrix_float_set(nativeHandleChecked(), i.toULong(), j.toULong(), value) gsl_matrix_float_set(nativeHandle, i.toULong(), j.toULong(), value)
override fun copy(): GslFloatMatrix { override fun copy(): GslFloatMatrix {
val new = requireNotNull(gsl_matrix_float_alloc(rowNum.toULong(), colNum.toULong())) val new = requireNotNull(gsl_matrix_float_alloc(rowNum.toULong(), colNum.toULong()))
gsl_matrix_float_memcpy(new, nativeHandleChecked()) gsl_matrix_float_memcpy(new, nativeHandle)
return GslFloatMatrix(new, scope) return GslFloatMatrix(new, scope)
} }
override fun close(): Unit = gsl_matrix_float_free(nativeHandleChecked()) override fun close(): Unit = gsl_matrix_float_free(nativeHandle)
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslFloatMatrix) if (other is GslFloatMatrix)
return gsl_matrix_float_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_matrix_float_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
} }
internal class GslShortMatrix( internal class GslShortMatrix(
override val nativeHandle: CPointer<gsl_matrix_short>, override val rawNativeHandle: CPointer<gsl_matrix_short>,
scope: DeferScope scope: AutofreeScope,
) : GslMatrix<Short, gsl_matrix_short>(scope) { ) : GslMatrix<Short, gsl_matrix_short>(scope) {
override val rowNum: Int override val rowNum: Int
get() = nativeHandleChecked().pointed.size1.toInt() get() = nativeHandle.pointed.size1.toInt()
override val colNum: Int override val colNum: Int
get() = nativeHandleChecked().pointed.size2.toInt() get() = nativeHandle.pointed.size2.toInt()
override val rows: Buffer<Buffer<Short>>
get() = VirtualBuffer(rowNum) { r ->
GslShortVector(
gsl_matrix_short_row(nativeHandle, r.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override val columns: Buffer<Buffer<Short>>
get() = VirtualBuffer(rowNum) { c ->
GslShortVector(
gsl_matrix_short_column(nativeHandle, c.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override operator fun get(i: Int, j: Int): Short = override operator fun get(i: Int, j: Int): Short =
gsl_matrix_short_get(nativeHandleChecked(), i.toULong(), j.toULong()) gsl_matrix_short_get(nativeHandle, i.toULong(), j.toULong())
override operator fun set(i: Int, j: Int, value: Short): Unit = override operator fun set(i: Int, j: Int, value: Short): Unit =
gsl_matrix_short_set(nativeHandleChecked(), i.toULong(), j.toULong(), value) gsl_matrix_short_set(nativeHandle, i.toULong(), j.toULong(), value)
override fun copy(): GslShortMatrix { override fun copy(): GslShortMatrix {
val new = requireNotNull(gsl_matrix_short_alloc(rowNum.toULong(), colNum.toULong())) val new = requireNotNull(gsl_matrix_short_alloc(rowNum.toULong(), colNum.toULong()))
gsl_matrix_short_memcpy(new, nativeHandleChecked()) gsl_matrix_short_memcpy(new, nativeHandle)
return GslShortMatrix(new, scope) return GslShortMatrix(new, scope)
} }
override fun close(): Unit = gsl_matrix_short_free(nativeHandleChecked()) override fun close(): Unit = gsl_matrix_short_free(nativeHandle)
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslShortMatrix) if (other is GslShortMatrix)
return gsl_matrix_short_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_matrix_short_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
} }
internal class GslUShortMatrix( internal class GslUShortMatrix(
override val nativeHandle: CPointer<gsl_matrix_ushort>, override val rawNativeHandle: CPointer<gsl_matrix_ushort>,
scope: DeferScope scope: AutofreeScope,
) : GslMatrix<UShort, gsl_matrix_ushort>(scope) { ) : GslMatrix<UShort, gsl_matrix_ushort>(scope) {
override val rowNum: Int override val rowNum: Int
get() = nativeHandleChecked().pointed.size1.toInt() get() = nativeHandle.pointed.size1.toInt()
override val colNum: Int override val colNum: Int
get() = nativeHandleChecked().pointed.size2.toInt() get() = nativeHandle.pointed.size2.toInt()
override val rows: Buffer<Buffer<UShort>>
get() = VirtualBuffer(rowNum) { r ->
GslUShortVector(
gsl_matrix_ushort_row(nativeHandle, r.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override val columns: Buffer<Buffer<UShort>>
get() = VirtualBuffer(rowNum) { c ->
GslUShortVector(
gsl_matrix_ushort_column(nativeHandle, c.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override operator fun get(i: Int, j: Int): UShort = override operator fun get(i: Int, j: Int): UShort =
gsl_matrix_ushort_get(nativeHandleChecked(), i.toULong(), j.toULong()) gsl_matrix_ushort_get(nativeHandle, i.toULong(), j.toULong())
override operator fun set(i: Int, j: Int, value: UShort): Unit = override operator fun set(i: Int, j: Int, value: UShort): Unit =
gsl_matrix_ushort_set(nativeHandleChecked(), i.toULong(), j.toULong(), value) gsl_matrix_ushort_set(nativeHandle, i.toULong(), j.toULong(), value)
override fun copy(): GslUShortMatrix { override fun copy(): GslUShortMatrix {
val new = requireNotNull(gsl_matrix_ushort_alloc(rowNum.toULong(), colNum.toULong())) val new = requireNotNull(gsl_matrix_ushort_alloc(rowNum.toULong(), colNum.toULong()))
gsl_matrix_ushort_memcpy(new, nativeHandleChecked()) gsl_matrix_ushort_memcpy(new, nativeHandle)
return GslUShortMatrix(new, scope) return GslUShortMatrix(new, scope)
} }
override fun close(): Unit = gsl_matrix_ushort_free(nativeHandleChecked()) override fun close(): Unit = gsl_matrix_ushort_free(nativeHandle)
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslUShortMatrix) if (other is GslUShortMatrix)
return gsl_matrix_ushort_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_matrix_ushort_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
} }
internal class GslLongMatrix( internal class GslLongMatrix(
override val nativeHandle: CPointer<gsl_matrix_long>, override val rawNativeHandle: CPointer<gsl_matrix_long>,
scope: DeferScope scope: AutofreeScope,
) : GslMatrix<Long, gsl_matrix_long>(scope) { ) : GslMatrix<Long, gsl_matrix_long>(scope) {
override val rowNum: Int override val rowNum: Int
get() = nativeHandleChecked().pointed.size1.toInt() get() = nativeHandle.pointed.size1.toInt()
override val colNum: Int override val colNum: Int
get() = nativeHandleChecked().pointed.size2.toInt() get() = nativeHandle.pointed.size2.toInt()
override val rows: Buffer<Buffer<Long>>
get() = VirtualBuffer(rowNum) { r ->
GslLongVector(
gsl_matrix_long_row(nativeHandle, r.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override val columns: Buffer<Buffer<Long>>
get() = VirtualBuffer(rowNum) { c ->
GslLongVector(
gsl_matrix_long_column(nativeHandle, c.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override operator fun get(i: Int, j: Int): Long = override operator fun get(i: Int, j: Int): Long =
gsl_matrix_long_get(nativeHandleChecked(), i.toULong(), j.toULong()) gsl_matrix_long_get(nativeHandle, i.toULong(), j.toULong())
override operator fun set(i: Int, j: Int, value: Long): Unit = override operator fun set(i: Int, j: Int, value: Long): Unit =
gsl_matrix_long_set(nativeHandleChecked(), i.toULong(), j.toULong(), value) gsl_matrix_long_set(nativeHandle, i.toULong(), j.toULong(), value)
override fun copy(): GslLongMatrix { override fun copy(): GslLongMatrix {
val new = requireNotNull(gsl_matrix_long_alloc(rowNum.toULong(), colNum.toULong())) val new = requireNotNull(gsl_matrix_long_alloc(rowNum.toULong(), colNum.toULong()))
gsl_matrix_long_memcpy(new, nativeHandleChecked()) gsl_matrix_long_memcpy(new, nativeHandle)
return GslLongMatrix(new, scope) return GslLongMatrix(new, scope)
} }
override fun close(): Unit = gsl_matrix_long_free(nativeHandleChecked()) override fun close(): Unit = gsl_matrix_long_free(nativeHandle)
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslLongMatrix) if (other is GslLongMatrix)
return gsl_matrix_long_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_matrix_long_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
} }
internal class GslULongMatrix( internal class GslULongMatrix(
override val nativeHandle: CPointer<gsl_matrix_ulong>, override val rawNativeHandle: CPointer<gsl_matrix_ulong>,
scope: DeferScope scope: AutofreeScope,
) : GslMatrix<ULong, gsl_matrix_ulong>(scope) { ) : GslMatrix<ULong, gsl_matrix_ulong>(scope) {
override val rowNum: Int override val rowNum: Int
get() = nativeHandleChecked().pointed.size1.toInt() get() = nativeHandle.pointed.size1.toInt()
override val colNum: Int override val colNum: Int
get() = nativeHandleChecked().pointed.size2.toInt() get() = nativeHandle.pointed.size2.toInt()
override val rows: Buffer<Buffer<ULong>>
get() = VirtualBuffer(rowNum) { r ->
GslULongVector(
gsl_matrix_ulong_row(nativeHandle, r.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override val columns: Buffer<Buffer<ULong>>
get() = VirtualBuffer(rowNum) { c ->
GslULongVector(
gsl_matrix_ulong_column(nativeHandle, c.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override operator fun get(i: Int, j: Int): ULong = override operator fun get(i: Int, j: Int): ULong =
gsl_matrix_ulong_get(nativeHandleChecked(), i.toULong(), j.toULong()) gsl_matrix_ulong_get(nativeHandle, i.toULong(), j.toULong())
override operator fun set(i: Int, j: Int, value: ULong): Unit = override operator fun set(i: Int, j: Int, value: ULong): Unit =
gsl_matrix_ulong_set(nativeHandleChecked(), i.toULong(), j.toULong(), value) gsl_matrix_ulong_set(nativeHandle, i.toULong(), j.toULong(), value)
override fun copy(): GslULongMatrix { override fun copy(): GslULongMatrix {
val new = requireNotNull(gsl_matrix_ulong_alloc(rowNum.toULong(), colNum.toULong())) val new = requireNotNull(gsl_matrix_ulong_alloc(rowNum.toULong(), colNum.toULong()))
gsl_matrix_ulong_memcpy(new, nativeHandleChecked()) gsl_matrix_ulong_memcpy(new, nativeHandle)
return GslULongMatrix(new, scope) return GslULongMatrix(new, scope)
} }
override fun close(): Unit = gsl_matrix_ulong_free(nativeHandleChecked()) override fun close(): Unit = gsl_matrix_ulong_free(nativeHandle)
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslULongMatrix) if (other is GslULongMatrix)
return gsl_matrix_ulong_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_matrix_ulong_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
} }
internal class GslIntMatrix( internal class GslIntMatrix(
override val nativeHandle: CPointer<gsl_matrix_int>, override val rawNativeHandle: CPointer<gsl_matrix_int>,
scope: DeferScope scope: AutofreeScope,
) : GslMatrix<Int, gsl_matrix_int>(scope) { ) : GslMatrix<Int, gsl_matrix_int>(scope) {
override val rowNum: Int override val rowNum: Int
get() = nativeHandleChecked().pointed.size1.toInt() get() = nativeHandle.pointed.size1.toInt()
override val colNum: Int override val colNum: Int
get() = nativeHandleChecked().pointed.size2.toInt() get() = nativeHandle.pointed.size2.toInt()
override val rows: Buffer<Buffer<Int>>
get() = VirtualBuffer(rowNum) { r ->
GslIntVector(
gsl_matrix_int_row(nativeHandle, r.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override val columns: Buffer<Buffer<Int>>
get() = VirtualBuffer(rowNum) { c ->
GslIntVector(
gsl_matrix_int_column(nativeHandle, c.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override operator fun get(i: Int, j: Int): Int = override operator fun get(i: Int, j: Int): Int =
gsl_matrix_int_get(nativeHandleChecked(), i.toULong(), j.toULong()) gsl_matrix_int_get(nativeHandle, i.toULong(), j.toULong())
override operator fun set(i: Int, j: Int, value: Int): Unit = override operator fun set(i: Int, j: Int, value: Int): Unit =
gsl_matrix_int_set(nativeHandleChecked(), i.toULong(), j.toULong(), value) gsl_matrix_int_set(nativeHandle, i.toULong(), j.toULong(), value)
override fun copy(): GslIntMatrix { override fun copy(): GslIntMatrix {
val new = requireNotNull(gsl_matrix_int_alloc(rowNum.toULong(), colNum.toULong())) val new = requireNotNull(gsl_matrix_int_alloc(rowNum.toULong(), colNum.toULong()))
gsl_matrix_int_memcpy(new, nativeHandleChecked()) gsl_matrix_int_memcpy(new, nativeHandle)
return GslIntMatrix(new, scope) return GslIntMatrix(new, scope)
} }
override fun close(): Unit = gsl_matrix_int_free(nativeHandleChecked()) override fun close(): Unit = gsl_matrix_int_free(nativeHandle)
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslIntMatrix) if (other is GslIntMatrix)
return gsl_matrix_int_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_matrix_int_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
} }
internal class GslUIntMatrix( internal class GslUIntMatrix(
override val nativeHandle: CPointer<gsl_matrix_uint>, override val rawNativeHandle: CPointer<gsl_matrix_uint>,
scope: DeferScope scope: AutofreeScope,
) : GslMatrix<UInt, gsl_matrix_uint>(scope) { ) : GslMatrix<UInt, gsl_matrix_uint>(scope) {
override val rowNum: Int override val rowNum: Int
get() = nativeHandleChecked().pointed.size1.toInt() get() = nativeHandle.pointed.size1.toInt()
override val colNum: Int override val colNum: Int
get() = nativeHandleChecked().pointed.size2.toInt() get() = nativeHandle.pointed.size2.toInt()
override val rows: Buffer<Buffer<UInt>>
get() = VirtualBuffer(rowNum) { r ->
GslUIntVector(
gsl_matrix_uint_row(nativeHandle, r.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override val columns: Buffer<Buffer<UInt>>
get() = VirtualBuffer(rowNum) { c ->
GslUIntVector(
gsl_matrix_uint_column(nativeHandle, c.toULong()).placeTo(scope).pointed.vector.ptr,
scope,
).apply { shouldBeFreed = false }
}
override operator fun get(i: Int, j: Int): UInt = override operator fun get(i: Int, j: Int): UInt =
gsl_matrix_uint_get(nativeHandleChecked(), i.toULong(), j.toULong()) gsl_matrix_uint_get(nativeHandle, i.toULong(), j.toULong())
override operator fun set(i: Int, j: Int, value: UInt): Unit = override operator fun set(i: Int, j: Int, value: UInt): Unit =
gsl_matrix_uint_set(nativeHandleChecked(), i.toULong(), j.toULong(), value) gsl_matrix_uint_set(nativeHandle, i.toULong(), j.toULong(), value)
override fun copy(): GslUIntMatrix { override fun copy(): GslUIntMatrix {
val new = requireNotNull(gsl_matrix_uint_alloc(rowNum.toULong(), colNum.toULong())) val new = requireNotNull(gsl_matrix_uint_alloc(rowNum.toULong(), colNum.toULong()))
gsl_matrix_uint_memcpy(new, nativeHandleChecked()) gsl_matrix_uint_memcpy(new, nativeHandle)
return GslUIntMatrix(new, scope) return GslUIntMatrix(new, scope)
} }
override fun close(): Unit = gsl_matrix_uint_free(nativeHandleChecked()) override fun close(): Unit = gsl_matrix_uint_free(nativeHandle)
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslUIntMatrix) if (other is GslUIntMatrix)
return gsl_matrix_uint_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_matrix_uint_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }

View File

@ -3,219 +3,219 @@ package kscience.kmath.gsl
import kotlinx.cinterop.* import kotlinx.cinterop.*
import org.gnu.gsl.* import org.gnu.gsl.*
internal class GslRealVector(override val nativeHandle: CPointer<gsl_vector>, scope: DeferScope) : internal class GslRealVector(override val rawNativeHandle: CPointer<gsl_vector>, scope: AutofreeScope) :
GslVector<Double, gsl_vector>(scope) { GslVector<Double, gsl_vector>(scope) {
override val size: Int override val size: Int
get() = nativeHandleChecked().pointed.size.toInt() get() = nativeHandle.pointed.size.toInt()
override operator fun get(index: Int): Double = override operator fun get(index: Int): Double =
gsl_vector_get(nativeHandleChecked(), index.toULong()) gsl_vector_get(nativeHandle, index.toULong())
override operator fun set(index: Int, value: Double): Unit = override operator fun set(index: Int, value: Double): Unit =
gsl_vector_set(nativeHandleChecked(), index.toULong(), value) gsl_vector_set(nativeHandle, index.toULong(), value)
override fun copy(): GslRealVector { override fun copy(): GslRealVector {
val new = requireNotNull(gsl_vector_alloc(size.toULong())) val new = requireNotNull(gsl_vector_alloc(size.toULong()))
gsl_vector_memcpy(new, nativeHandleChecked()) gsl_vector_memcpy(new, nativeHandle)
return GslRealVector(new, scope) return GslRealVector(new, scope)
} }
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslRealVector) if (other is GslRealVector)
return gsl_vector_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_vector_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
override fun close(): Unit = gsl_vector_free(nativeHandleChecked()) override fun close(): Unit = gsl_vector_free(nativeHandle)
} }
internal class GslFloatVector(override val nativeHandle: CPointer<gsl_vector_float>, scope: DeferScope) : internal class GslFloatVector(override val rawNativeHandle: CPointer<gsl_vector_float>, scope: AutofreeScope) :
GslVector<Float, gsl_vector_float>(scope) { GslVector<Float, gsl_vector_float>(scope) {
override val size: Int override val size: Int
get() = nativeHandleChecked().pointed.size.toInt() get() = nativeHandle.pointed.size.toInt()
override operator fun get(index: Int): Float = override operator fun get(index: Int): Float =
gsl_vector_float_get(nativeHandleChecked(), index.toULong()) gsl_vector_float_get(nativeHandle, index.toULong())
override operator fun set(index: Int, value: Float): Unit = override operator fun set(index: Int, value: Float): Unit =
gsl_vector_float_set(nativeHandleChecked(), index.toULong(), value) gsl_vector_float_set(nativeHandle, index.toULong(), value)
override fun copy(): GslFloatVector { override fun copy(): GslFloatVector {
val new = requireNotNull(gsl_vector_float_alloc(size.toULong())) val new = requireNotNull(gsl_vector_float_alloc(size.toULong()))
gsl_vector_float_memcpy(new, nativeHandleChecked()) gsl_vector_float_memcpy(new, nativeHandle)
return GslFloatVector(new, scope) return GslFloatVector(new, scope)
} }
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslFloatVector) if (other is GslFloatVector)
return gsl_vector_float_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_vector_float_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
override fun close(): Unit = gsl_vector_float_free(nativeHandleChecked()) override fun close(): Unit = gsl_vector_float_free(nativeHandle)
} }
internal class GslShortVector(override val nativeHandle: CPointer<gsl_vector_short>, scope: DeferScope) : internal class GslShortVector(override val rawNativeHandle: CPointer<gsl_vector_short>, scope: AutofreeScope) :
GslVector<Short, gsl_vector_short>(scope) { GslVector<Short, gsl_vector_short>(scope) {
override val size: Int override val size: Int
get() = nativeHandleChecked().pointed.size.toInt() get() = nativeHandle.pointed.size.toInt()
override operator fun get(index: Int): Short = override operator fun get(index: Int): Short =
gsl_vector_short_get(nativeHandleChecked(), index.toULong()) gsl_vector_short_get(nativeHandle, index.toULong())
override operator fun set(index: Int, value: Short): Unit = override operator fun set(index: Int, value: Short): Unit =
gsl_vector_short_set(nativeHandleChecked(), index.toULong(), value) gsl_vector_short_set(nativeHandle, index.toULong(), value)
override fun copy(): GslShortVector { override fun copy(): GslShortVector {
val new = requireNotNull(gsl_vector_short_alloc(size.toULong())) val new = requireNotNull(gsl_vector_short_alloc(size.toULong()))
gsl_vector_short_memcpy(new, nativeHandleChecked()) gsl_vector_short_memcpy(new, nativeHandle)
return GslShortVector(new, scope) return GslShortVector(new, scope)
} }
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslShortVector) if (other is GslShortVector)
return gsl_vector_short_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_vector_short_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
override fun close(): Unit = gsl_vector_short_free(nativeHandleChecked()) override fun close(): Unit = gsl_vector_short_free(nativeHandle)
} }
internal class GslUShortVector(override val nativeHandle: CPointer<gsl_vector_ushort>, scope: DeferScope) : internal class GslUShortVector(override val rawNativeHandle: CPointer<gsl_vector_ushort>, scope: AutofreeScope) :
GslVector<UShort, gsl_vector_ushort>(scope) { GslVector<UShort, gsl_vector_ushort>(scope) {
override val size: Int override val size: Int
get() = nativeHandleChecked().pointed.size.toInt() get() = nativeHandle.pointed.size.toInt()
override operator fun get(index: Int): UShort = override operator fun get(index: Int): UShort =
gsl_vector_ushort_get(nativeHandleChecked(), index.toULong()) gsl_vector_ushort_get(nativeHandle, index.toULong())
override operator fun set(index: Int, value: UShort): Unit = override operator fun set(index: Int, value: UShort): Unit =
gsl_vector_ushort_set(nativeHandleChecked(), index.toULong(), value) gsl_vector_ushort_set(nativeHandle, index.toULong(), value)
override fun copy(): GslUShortVector { override fun copy(): GslUShortVector {
val new = requireNotNull(gsl_vector_ushort_alloc(size.toULong())) val new = requireNotNull(gsl_vector_ushort_alloc(size.toULong()))
gsl_vector_ushort_memcpy(new, nativeHandleChecked()) gsl_vector_ushort_memcpy(new, nativeHandle)
return GslUShortVector(new, scope) return GslUShortVector(new, scope)
} }
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslUShortVector) if (other is GslUShortVector)
return gsl_vector_ushort_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_vector_ushort_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
override fun close(): Unit = gsl_vector_ushort_free(nativeHandleChecked()) override fun close(): Unit = gsl_vector_ushort_free(nativeHandle)
} }
internal class GslLongVector(override val nativeHandle: CPointer<gsl_vector_long>, scope: DeferScope) : internal class GslLongVector(override val rawNativeHandle: CPointer<gsl_vector_long>, scope: AutofreeScope) :
GslVector<Long, gsl_vector_long>(scope) { GslVector<Long, gsl_vector_long>(scope) {
override val size: Int override val size: Int
get() = nativeHandleChecked().pointed.size.toInt() get() = nativeHandle.pointed.size.toInt()
override operator fun get(index: Int): Long = override operator fun get(index: Int): Long =
gsl_vector_long_get(nativeHandleChecked(), index.toULong()) gsl_vector_long_get(nativeHandle, index.toULong())
override operator fun set(index: Int, value: Long): Unit = override operator fun set(index: Int, value: Long): Unit =
gsl_vector_long_set(nativeHandleChecked(), index.toULong(), value) gsl_vector_long_set(nativeHandle, index.toULong(), value)
override fun copy(): GslLongVector { override fun copy(): GslLongVector {
val new = requireNotNull(gsl_vector_long_alloc(size.toULong())) val new = requireNotNull(gsl_vector_long_alloc(size.toULong()))
gsl_vector_long_memcpy(new, nativeHandleChecked()) gsl_vector_long_memcpy(new, nativeHandle)
return GslLongVector(new, scope) return GslLongVector(new, scope)
} }
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslLongVector) if (other is GslLongVector)
return gsl_vector_long_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_vector_long_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
override fun close(): Unit = gsl_vector_long_free(nativeHandleChecked()) override fun close(): Unit = gsl_vector_long_free(nativeHandle)
} }
internal class GslULongVector(override val nativeHandle: CPointer<gsl_vector_ulong>, scope: DeferScope) : internal class GslULongVector(override val rawNativeHandle: CPointer<gsl_vector_ulong>, scope: AutofreeScope) :
GslVector<ULong, gsl_vector_ulong>(scope) { GslVector<ULong, gsl_vector_ulong>(scope) {
override val size: Int override val size: Int
get() = nativeHandleChecked().pointed.size.toInt() get() = nativeHandle.pointed.size.toInt()
override operator fun get(index: Int): ULong = override operator fun get(index: Int): ULong =
gsl_vector_ulong_get(nativeHandleChecked(), index.toULong()) gsl_vector_ulong_get(nativeHandle, index.toULong())
override operator fun set(index: Int, value: ULong): Unit = override operator fun set(index: Int, value: ULong): Unit =
gsl_vector_ulong_set(nativeHandleChecked(), index.toULong(), value) gsl_vector_ulong_set(nativeHandle, index.toULong(), value)
override fun copy(): GslULongVector { override fun copy(): GslULongVector {
val new = requireNotNull(gsl_vector_ulong_alloc(size.toULong())) val new = requireNotNull(gsl_vector_ulong_alloc(size.toULong()))
gsl_vector_ulong_memcpy(new, nativeHandleChecked()) gsl_vector_ulong_memcpy(new, nativeHandle)
return GslULongVector(new, scope) return GslULongVector(new, scope)
} }
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslULongVector) if (other is GslULongVector)
return gsl_vector_ulong_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_vector_ulong_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
override fun close(): Unit = gsl_vector_ulong_free(nativeHandleChecked()) override fun close(): Unit = gsl_vector_ulong_free(nativeHandle)
} }
internal class GslIntVector(override val nativeHandle: CPointer<gsl_vector_int>, scope: DeferScope) : internal class GslIntVector(override val rawNativeHandle: CPointer<gsl_vector_int>, scope: AutofreeScope) :
GslVector<Int, gsl_vector_int>(scope) { GslVector<Int, gsl_vector_int>(scope) {
override val size: Int override val size: Int
get() = nativeHandleChecked().pointed.size.toInt() get() = nativeHandle.pointed.size.toInt()
override operator fun get(index: Int): Int = override operator fun get(index: Int): Int =
gsl_vector_int_get(nativeHandleChecked(), index.toULong()) gsl_vector_int_get(nativeHandle, index.toULong())
override operator fun set(index: Int, value: Int): Unit = override operator fun set(index: Int, value: Int): Unit =
gsl_vector_int_set(nativeHandleChecked(), index.toULong(), value) gsl_vector_int_set(nativeHandle, index.toULong(), value)
override fun copy(): GslIntVector { override fun copy(): GslIntVector {
val new = requireNotNull(gsl_vector_int_alloc(size.toULong())) val new = requireNotNull(gsl_vector_int_alloc(size.toULong()))
gsl_vector_int_memcpy(new, nativeHandleChecked()) gsl_vector_int_memcpy(new, nativeHandle)
return GslIntVector(new, scope) return GslIntVector(new, scope)
} }
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslIntVector) if (other is GslIntVector)
return gsl_vector_int_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_vector_int_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
override fun close(): Unit = gsl_vector_int_free(nativeHandleChecked()) override fun close(): Unit = gsl_vector_int_free(nativeHandle)
} }
internal class GslUIntVector(override val nativeHandle: CPointer<gsl_vector_uint>, scope: DeferScope) : internal class GslUIntVector(override val rawNativeHandle: CPointer<gsl_vector_uint>, scope: AutofreeScope) :
GslVector<UInt, gsl_vector_uint>(scope) { GslVector<UInt, gsl_vector_uint>(scope) {
override val size: Int override val size: Int
get() = nativeHandleChecked().pointed.size.toInt() get() = nativeHandle.pointed.size.toInt()
override operator fun get(index: Int): UInt = override operator fun get(index: Int): UInt =
gsl_vector_uint_get(nativeHandleChecked(), index.toULong()) gsl_vector_uint_get(nativeHandle, index.toULong())
override operator fun set(index: Int, value: UInt): Unit = override operator fun set(index: Int, value: UInt): Unit =
gsl_vector_uint_set(nativeHandleChecked(), index.toULong(), value) gsl_vector_uint_set(nativeHandle, index.toULong(), value)
override fun copy(): GslUIntVector { override fun copy(): GslUIntVector {
val new = requireNotNull(gsl_vector_uint_alloc(size.toULong())) val new = requireNotNull(gsl_vector_uint_alloc(size.toULong()))
gsl_vector_uint_memcpy(new, nativeHandleChecked()) gsl_vector_uint_memcpy(new, nativeHandle)
return GslUIntVector(new, scope) return GslUIntVector(new, scope)
} }
override fun equals(other: Any?): Boolean { override fun equals(other: Any?): Boolean {
if (other is GslUIntVector) if (other is GslUIntVector)
return gsl_vector_uint_equal(nativeHandleChecked(), other.nativeHandleChecked()) == 1 return gsl_vector_uint_equal(nativeHandle, other.nativeHandle) == 1
return super.equals(other) return super.equals(other)
} }
override fun close(): Unit = gsl_vector_uint_free(nativeHandleChecked()) override fun close(): Unit = gsl_vector_uint_free(nativeHandle)
} }

View File

@ -3,6 +3,8 @@ package kscience.kmath.gsl
import kscience.kmath.linear.RealMatrixContext import kscience.kmath.linear.RealMatrixContext
import kscience.kmath.operations.invoke import kscience.kmath.operations.invoke
import kscience.kmath.structures.Matrix import kscience.kmath.structures.Matrix
import kscience.kmath.structures.asIterable
import kscience.kmath.structures.toList
import kotlin.random.Random import kotlin.random.Random
import kotlin.test.Test import kotlin.test.Test
import kotlin.test.assertEquals import kotlin.test.assertEquals
@ -39,4 +41,22 @@ internal class GslMatrixRealTest {
assertEquals(mat, mat3) assertEquals(mat, mat3)
assertEquals(mat2, mat3) assertEquals(mat2, mat3)
} }
@Test
fun rows() = GslRealMatrixContext {
val mat = produce(2, 2) { i, j -> i.toDouble() + j }
mat.rows.asIterable().zip(listOf(listOf(0.0, 1.0), listOf(1.0, 2.0))).forEach { (a, b) ->
assertEquals(a.toList(), b)
}
}
@Test
fun columns() = GslRealMatrixContext {
val mat = produce(2, 2) { i, j -> i.toDouble() + j }
mat.columns.asIterable().zip(listOf(listOf(0.0, 1.0), listOf(1.0, 2.0))).forEach { (a, b) ->
assertEquals(a.toList(), b)
}
}
} }