File: vector.go

package info (click to toggle)
mumax3 3.11.1-1
  • links: PTS, VCS
  • area: contrib
  • in suites: forky, sid
  • size: 10,668 kB
  • sloc: makefile: 194; ansic: 155; sh: 86; javascript: 16
file content (61 lines) | stat: -rw-r--r-- 1,247 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
package data

import "math"

// 3-component vector
type Vector [3]float64

func (v Vector) X() float64 { return v[0] }
func (v Vector) Y() float64 { return v[1] }
func (v Vector) Z() float64 { return v[2] }

// Returns a*v.
func (v Vector) Mul(a float64) Vector {
	return Vector{a * v[0], a * v[1], a * v[2]}
}

// Returns (1/a)*v.
func (v Vector) Div(a float64) Vector {
	return v.Mul(1 / a)
}

// Returns a+b.
func (a Vector) Add(b Vector) Vector {
	return Vector{a[0] + b[0], a[1] + b[1], a[2] + b[2]}
}

// Returns a+s*b.
func (a Vector) MAdd(s float64, b Vector) Vector {
	return Vector{a[0] + s*b[0], a[1] + s*b[1], a[2] + s*b[2]}
}

// Returns a-b.
func (a Vector) Sub(b Vector) Vector {
	return Vector{a[0] - b[0], a[1] - b[1], a[2] - b[2]}
}

// Returns the norm of v.
func (v Vector) Len() float64 {
	len2 := v.Dot(v)
	return math.Sqrt(len2)
}

// Returns the dot (inner) product a.b.
func (a Vector) Dot(b Vector) float64 {
	return a[0]*b[0] + a[1]*b[1] + a[2]*b[2]
}

// Returns the cross (vector) product a x b
// in a right-handed coordinate system.
func (a Vector) Cross(b Vector) Vector {
	x := a[1]*b[2] - a[2]*b[1]
	y := a[2]*b[0] - a[0]*b[2]
	z := a[0]*b[1] - a[1]*b[0]
	return Vector{x, y, z}
}

const (
	X = 0
	Y = 1
	Z = 2
)