/
scatter.go
71 lines (62 loc) · 1.43 KB
/
scatter.go
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
62
63
64
65
66
67
68
69
70
71
package petscgo
/*
#cgo pkg-config: PETSc ompi
#include "petsc.h"
*/
import "C"
import (
"errors"
)
// Scatter wraps the VecScatter context
type Scatter struct {
s C.VecScatter
}
// NewScatter creates a new scatter context
func NewScatter(src, dest *Vec, sis, dis *IS) (*Scatter, error) {
s := new(Scatter)
perr := C.VecScatterCreate(src.v, sis.is, dest.v, dis.is, &s.s)
if perr != 0 {
return nil, errors.New("Error creating scatter context")
}
return s, nil
}
// Destroy destroys the scatter context
func (s *Scatter) Destroy() error {
perr := C.VecScatterDestroy(&s.s)
if perr != 0 {
return errors.New("Error destroying scatter context")
}
return nil
}
// Begin starts a scatter
func (s *Scatter) Begin(src, dest *Vec, add, forward bool) error {
var iora C.InsertMode = C.INSERT_VALUES
if add {
iora = C.ADD_VALUES
}
var mode C.ScatterMode = C.SCATTER_FORWARD
if !forward {
mode = C.SCATTER_REVERSE
}
perr := C.VecScatterBegin(s.s, src.v, dest.v, iora, mode)
if perr != 0 {
return errors.New("Error starting scatter")
}
return nil
}
// End ends the scatter
func (s *Scatter) End(src, dest *Vec, add, forward bool) error {
var iora C.InsertMode = C.INSERT_VALUES
if add {
iora = C.ADD_VALUES
}
var mode C.ScatterMode = C.SCATTER_FORWARD
if !forward {
mode = C.SCATTER_REVERSE
}
perr := C.VecScatterEnd(s.s, src.v, dest.v, iora, mode)
if perr != 0 {
return errors.New("Error ending scatter")
}
return nil
}