forked from biogo/biogo
/
trapezoid.go
81 lines (69 loc) · 2.09 KB
/
trapezoid.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
72
73
74
75
76
77
78
79
80
81
// Copyright ©2011-2012 The bíogo Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package filter
// Type to store a successfully filtered w × e Parallelogram
type Trapezoid struct {
Top, Bottom int // B (query) coords of top and bottom of trapzoidal zone
Left, Right int // Left and right diagonals of trapzoidal zone
}
type trapezoid struct {
next *trapezoid // Organized in a list linked on this field
Trapezoid
}
// prependFrontTo prepends a copy of front of list onto the front of the list
// taking the allocation from the receiver which acts as a pool. It returns the
// next element of the pool.
func (tr *trapezoid) prependFrontTo(list *trapezoid) *trapezoid {
next := tr.next
*tr = *list
list.join(tr)
return next
}
// join joins list to the receiver, returning the receiver.
func (tr *trapezoid) join(list *trapezoid) *trapezoid {
tr.next = list
return tr
}
// Return the receiver and the subsequent trapezoid in the list.
func (tr *trapezoid) decapitate() (*trapezoid, *trapezoid) {
return tr, tr.next
}
// clip is the trapezoid trimming method used during merge.
func (tr *trapezoid) clip(lagPosition, lagClip int) {
if bottom := lagClip + tr.Left; tr.Bottom < bottom {
tr.Bottom = bottom
}
if top := lagPosition + tr.Right; tr.Top > top {
tr.Top = top
}
midPosition := (tr.Bottom + tr.Top) / 2
if left := midPosition - lagPosition; tr.Left < left {
tr.Left = left
}
if right := midPosition - lagClip; tr.Right > right {
tr.Right = right
}
}
// Trapezoids implements the sort.Sort interface.
type Trapezoids []Trapezoid
// Return the sum of all Trapezoids in the slice.
func (tr Trapezoids) Sum() (a, b int) {
for _, t := range tr {
la, lb := t.Top-t.Bottom, t.Right-t.Left
a, b = a+la, b+lb
}
return
}
// Required for sort.Interface
func (tr Trapezoids) Len() int {
return len(tr)
}
// Required for sort.Interface
func (tr Trapezoids) Less(i, j int) bool {
return tr[i].Bottom < tr[j].Bottom
}
// Required for sort.Interface
func (tr Trapezoids) Swap(i, j int) {
tr[i], tr[j] = tr[j], tr[i]
}