mirror of
https://github.com/gonum/gonum.git
synced 2025-10-23 15:13:31 +08:00
3fa9374bd4
This merges the three packages, matrix, mat64, and cmat128. It then renames this big package to mat. It fixes the import statements and corresponding code
144 lines
3.0 KiB
Go
144 lines
3.0 KiB
Go
// Copyright ©2014 The gonum Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// This file must be kept in sync with index_no_bound_checks.go.
|
|
|
|
//+build bounds
|
|
|
|
package mat
|
|
|
|
// At returns the element at row i, column j.
|
|
func (m *Dense) At(i, j int) float64 {
|
|
return m.at(i, j)
|
|
}
|
|
|
|
func (m *Dense) at(i, j int) float64 {
|
|
if uint(i) >= uint(m.mat.Rows) {
|
|
panic(ErrRowAccess)
|
|
}
|
|
if uint(j) >= uint(m.mat.Cols) {
|
|
panic(ErrColAccess)
|
|
}
|
|
return m.mat.Data[i*m.mat.Stride+j]
|
|
}
|
|
|
|
// Set sets the element at row i, column j to the value v.
|
|
func (m *Dense) Set(i, j int, v float64) {
|
|
m.set(i, j, v)
|
|
}
|
|
|
|
func (m *Dense) set(i, j int, v float64) {
|
|
if uint(i) >= uint(m.mat.Rows) {
|
|
panic(ErrRowAccess)
|
|
}
|
|
if uint(j) >= uint(m.mat.Cols) {
|
|
panic(ErrColAccess)
|
|
}
|
|
m.mat.Data[i*m.mat.Stride+j] = v
|
|
}
|
|
|
|
// At returns the element at row i.
|
|
// It panics if i is out of bounds or if j is not zero.
|
|
func (v *Vector) At(i, j int) float64 {
|
|
if j != 0 {
|
|
panic(ErrColAccess)
|
|
}
|
|
return v.at(i)
|
|
}
|
|
|
|
func (v *Vector) at(i int) float64 {
|
|
if uint(i) >= uint(v.n) {
|
|
panic(ErrRowAccess)
|
|
}
|
|
return v.mat.Data[i*v.mat.Inc]
|
|
}
|
|
|
|
// SetVec sets the element at row i to the value val.
|
|
// It panics if i is out of bounds.
|
|
func (v *Vector) SetVec(i int, val float64) {
|
|
v.setVec(i, val)
|
|
}
|
|
|
|
func (v *Vector) setVec(i int, val float64) {
|
|
if uint(i) >= uint(v.n) {
|
|
panic(ErrVectorAccess)
|
|
}
|
|
v.mat.Data[i*v.mat.Inc] = val
|
|
}
|
|
|
|
// At returns the element at row i and column j.
|
|
func (t *SymDense) At(i, j int) float64 {
|
|
return t.at(i, j)
|
|
}
|
|
|
|
func (t *SymDense) at(i, j int) float64 {
|
|
if uint(i) >= uint(t.mat.N) {
|
|
panic(ErrRowAccess)
|
|
}
|
|
if uint(j) >= uint(t.mat.N) {
|
|
panic(ErrColAccess)
|
|
}
|
|
if i > j {
|
|
i, j = j, i
|
|
}
|
|
return t.mat.Data[i*t.mat.Stride+j]
|
|
}
|
|
|
|
// SetSym sets the elements at (i,j) and (j,i) to the value v.
|
|
func (t *SymDense) SetSym(i, j int, v float64) {
|
|
t.set(i, j, v)
|
|
}
|
|
|
|
func (t *SymDense) set(i, j int, v float64) {
|
|
if uint(i) >= uint(t.mat.N) {
|
|
panic(ErrRowAccess)
|
|
}
|
|
if uint(j) >= uint(t.mat.N) {
|
|
panic(ErrColAccess)
|
|
}
|
|
if i > j {
|
|
i, j = j, i
|
|
}
|
|
t.mat.Data[i*t.mat.Stride+j] = v
|
|
}
|
|
|
|
// At returns the element at row i, column j.
|
|
func (t *TriDense) At(i, j int) float64 {
|
|
return t.at(i, j)
|
|
}
|
|
|
|
func (t *TriDense) at(i, j int) float64 {
|
|
if uint(i) >= uint(t.mat.N) {
|
|
panic(ErrRowAccess)
|
|
}
|
|
if uint(j) >= uint(t.mat.N) {
|
|
panic(ErrColAccess)
|
|
}
|
|
isUpper := t.isUpper()
|
|
if (isUpper && i > j) || (!isUpper && i < j) {
|
|
return 0
|
|
}
|
|
return t.mat.Data[i*t.mat.Stride+j]
|
|
}
|
|
|
|
// SetTri sets the element of the triangular matrix at row i, column j to the value v.
|
|
// It panics if the location is outside the appropriate half of the matrix.
|
|
func (t *TriDense) SetTri(i, j int, v float64) {
|
|
t.set(i, j, v)
|
|
}
|
|
|
|
func (t *TriDense) set(i, j int, v float64) {
|
|
if uint(i) >= uint(t.mat.N) {
|
|
panic(ErrRowAccess)
|
|
}
|
|
if uint(j) >= uint(t.mat.N) {
|
|
panic(ErrColAccess)
|
|
}
|
|
isUpper := t.isUpper()
|
|
if (isUpper && i > j) || (!isUpper && i < j) {
|
|
panic(ErrTriangleSet)
|
|
}
|
|
t.mat.Data[i*t.mat.Stride+j] = v
|
|
}
|