Exemplo n.º 1
0
// Solve Ux = y, replace b with solution
// column oriented backward substitution
func (U mesh) SolveUx(y vec.Vectorer) {
	n := U.c
	for j := n - 1; j >= 1; j-- {
		y.SetAt(j, y.GetAt(j)/U.GetAtNode(j, j))
		for k := 0; k < j; k++ {
			y.SetAt(k, y.GetAt(k)-y.GetAt(j)*U.GetAtNode(k, j))
		}
	}
	y.SetAt(0, y.GetAt(0)/U.GetAtNode(0, 0))
}
Exemplo n.º 2
0
// Solve Ly = b for y, replace b with solution
// column oriented forward substitution
func (L mesh) SolveLy(b vec.Vectorer) {
	n := L.c
	for j := 0; j < n-1; j++ {
		b.SetAt(j, b.GetAt(j)/L.GetAtNode(j, j))
		for k := j + 1; k < n; k++ {
			b.SetAt(k, b.GetAt(k)-b.GetAt(j)*L.GetAtNode(k, j))
		}
	}
	b.SetAt(n-1, b.GetAt(n-1)/L.GetAtNode(n-1, n-1))
}
Exemplo n.º 3
0
// GetCol returns the c'th column of m into vector, v
// of length equal to no. of rows of mesh
func (m mesh) GetCol(v vec.Vectorer, c int) {
	off := m.off
	for i := range v.Slice() {
		v.SetAt(i, m.elems.GetAt(i*off+c))
	}
}
Exemplo n.º 4
0
// GetRow returns the r'th column of m into vector, v
// of length equal to no. of cols of mesh
func (m mesh) GetRow(v vec.Vectorer, r int) {
	off := r * m.off
	for j := range v.Slice() {
		v.SetAt(j, m.elems.GetAt(off+j))
	}
}
Exemplo n.º 5
0
// GetDiag puts the diagonal of the mesh into
// the vector, m must be a square matrix
func (m mesh) GetDiag(d vec.Vectorer) {
	for e := range d.Slice() {
		d.SetAt(e, m.GetAtNode(e, e))
	}
}