eson/structure
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add GetOrSet

Browse Source
This commit is contained in:
eson
2019-04-17 03:22:07 +08:00
parent 3bc9a99ca9
commit 8d0df4bfed
4 changed files with 104 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
sparse_array/array2/array2.go
View File

@@ -108,6 +108,29 @@ func (arr *Array2) Get(idx int) (interface{}, bool) {
return v, v != nil
}
func (arr *Array2) GetOrSet(idx int, DoSetValue func([]interface{}, int)) (result interface{}, isSet bool) {
yindex := idx / arr.xsize
xindex := idx % arr.xsize
xdata := arr.data[yindex]
if xdata == nil {
xdata = make([]interface{}, arr.xsize, arr.xsize)
arr.data[yindex] = xdata
}
result = xdata[xindex]
if result == nil {
DoSetValue(xdata, xindex)
result = xdata[xindex]
if result == nil {
panic("DoSetValue Not Set <nil> Value")
}
arr.sizes[yindex]++
return result, true
}
return result, false
}
func (arr *Array2) Del(idx int) (interface{}, bool) {
yindex := idx / arr.xsize
xindex := idx % arr.xsize

32
sparse_array/array3/array3.go
View File

@@ -163,6 +163,38 @@ func (arr *Array3) Get(idx int) (interface{}, bool) {
return v, v != nil
}
func (arr *Array3) GetOrSet(idx int, DoSetValue func([]interface{}, int)) (result interface{}, isSet bool) {
zindex := idx / arr.xyproduct
nidx := (idx % arr.xyproduct)
yindex := nidx / arr.xsize
xindex := nidx % arr.xsize
ydata := arr.data[zindex]
if ydata == nil {
ydata = make([][]interface{}, arr.ysize, arr.ysize)
arr.data[zindex] = ydata
}
xdata := ydata[yindex]
if xdata == nil {
xdata = make([]interface{}, arr.xsize, arr.xsize)
ydata[yindex] = xdata
arr.ysizes[zindex]++
}
result = xdata[xindex]
if result == nil {
DoSetValue(xdata, xindex)
result = xdata[xindex]
if result == nil {
panic("DoSetValue Not Set <nil> Value")
}
arr.xsizes[zindex][yindex]++
return result, false
}
return result, true
}
func (arr *Array3) Del(idx int) (interface{}, bool) {
zindex := idx / arr.xyproduct
nextsize := (idx % arr.xyproduct)