diff options
| author | Mel <einebeere@gmail.com> | 2022-05-27 16:44:22 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-05-27 16:49:02 +0000 |
| commit | 47c4cd3705bee9d7154c42ce95aef6f8a19e0661 (patch) | |
| tree | f6b2d502cbc256914a6f7181e2cf623460f9d912 /pkg/libs/rangemap/rangemap.go | |
| parent | c2d4bf51de9a2d721168c62b14b89f5281ed366e (diff) | |
| download | jinx-47c4cd3705bee9d7154c42ce95aef6f8a19e0661.tar.zst jinx-47c4cd3705bee9d7154c42ce95aef6f8a19e0661.zip | |
Add debug info to compiled VM code
Diffstat (limited to 'pkg/libs/rangemap/rangemap.go')
| -rw-r--r-- | pkg/libs/rangemap/rangemap.go | 58 |
1 files changed, 58 insertions, 0 deletions
diff --git a/pkg/libs/rangemap/rangemap.go b/pkg/libs/rangemap/rangemap.go new file mode 100644 index 0000000..df891a6 --- /dev/null +++ b/pkg/libs/rangemap/rangemap.go @@ -0,0 +1,58 @@ +package rangemap + +type RangeMap[D any] struct { + ranges []rangeEntry + data []D +} + +func New[D any]() RangeMap[D] { + return RangeMap[D]{ + ranges: []rangeEntry{}, + } +} + +func (rm *RangeMap[D]) AppendToLast(to int, data D) bool { + if to < 0 { + return false + } + + from := 0 + if len(rm.ranges) != 0 { + last := rm.ranges[len(rm.ranges)-1] + if last.to >= to { + return false + } + from = last.to + 1 + } + + rm.ranges = append(rm.ranges, rangeEntry{ + from: from, + to: to, + }) + + rm.data = append(rm.data, data) + return true +} + +func (rm *RangeMap[D]) Get(i int) (*D, bool) { + left := 0 + right := len(rm.ranges) - 1 + for left <= right { + mid := (left + right) / 2 + entry := rm.ranges[mid] + if i < entry.from { + right = mid - 1 + } else if i > entry.to { + left = mid + 1 + } else { + return &rm.data[mid], true + } + } + + return nil, false +} + +type rangeEntry struct { + from int + to int +} |
