func testDiagSolver(n int) {
fmt.Println("Initializing Sparse matrix structure\n")
tmp := cmplxSparse.New()
matrixSize := int(2 * n)
fmt.Println("Making sparse ", matrixSize, "x", matrixSize, " Hamiltonian tridiagonal matrix\n")
cmplxSparse.MakeHamTriDiag(n, tmp)
for idx0 := 0; idx0 < matrixSize; idx0++ {
tmp.Data[idx0][2] -= complex(0.0, 5.0)
}
fmt.Println("Accessing matrix elements of Hamiltonian (m,n):")
for idx0 := 0; idx0 < matrixSize; idx0++ {
for idx1 := 0; idx1 < matrixSize; idx1++ {
test := cmplxSparse.AccessMatrix(idx0, idx1, tmp)
fmt.Printf("%f ", test)
}
fmt.Printf("\n")
}
fmt.Printf("\n")
tmp0 := cmplxSparse.SparseCopy(tmp)
cmplxSparse.SparseDiagLU(tmp0)
fmt.Println("Accessing matrix elements of LU (m,n):")
for idx0 := 0; idx0 < matrixSize; idx0++ {
for idx1 := 0; idx1 < matrixSize; idx1++ {
test := cmplxSparse.AccessMatrix(idx0, idx1, tmp0)
fmt.Printf("%f ", test)
}
fmt.Printf("\n")
}
fmt.Printf("\n")
InvBuffer := make([][]complex128, matrixSize)
for idx0 := range InvBuffer {
InvBuffer[idx0] = make([]complex128, matrixSize)
for idx1 := range InvBuffer[idx0] {
InvBuffer[idx0][idx1] = 0.0 + 0.0i
}
}
for idx0 := range InvBuffer {
InvBuffer[idx0][idx0] = 1.0 + 0.0i
BufferMatrix := cmplxSparse.SparseDiagLinearSolver(tmp0, InvBuffer[idx0])
InvBuffer[idx0] = BufferMatrix
}
fmt.Println("Accessing matrix elements of inverse (m,n):")
for idx0 := range InvBuffer {
for idx1 := range InvBuffer[idx0] {
fmt.Printf("%.15g ", InvBuffer[idx0][idx1])
}
fmt.Printf("\n")
}
}
func testLU(n int) {
fmt.Println("Initializing Sparse matrix structure\n")
tmp := cmplxSparse.New()
matrixSize := int(2 * n)
fmt.Println("Making sparse ", matrixSize, "x", matrixSize, " Hamiltonian tridiagonal matrix\n")
cmplxSparse.MakeHamTriDiag(n, tmp)
for idx0 := 0; idx0 < matrixSize; idx0++ {
tmp.Data[idx0][2] -= complex(0.0, 5.0)
}
LU_mat := cmplxSparse.SparseCopy(tmp)
cmplxSparse.SparseDiagLU(LU_mat)
fmt.Println("Accessing matrix elements (m,n):")
for idx0 := 0; idx0 < matrixSize; idx0++ {
for idx1 := 0; idx1 < matrixSize; idx1++ {
test := cmplxSparse.AccessMatrix(idx0, idx1, LU_mat)
fmt.Printf("%f ", test)
}
fmt.Printf("\n")
}
fmt.Println("\n")
}
