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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
|
package structs
import (
"encoding/json"
"fmt"
"io/ioutil"
"os/exec"
)
type Node struct {
Boundry BoundingBox // Spatial outreach of the quadtree
CenterOfMass Vec2 // Center of mass of the cell
TotalMass float64 // Total mass of all the stars in the cell
Depth int // Depth of the cell in the tree
Star Star2D // The actual star
// NW, NE, SW, SE
Subtrees [4]*Node // The child subtrees
}
// NewRoot returns a pointer to a node defined as a root node. It taks the with of the BoundingBox as an argument
// resulting in a node that should (in theory) fit the whole galaxy if defined correctly.
func NewRoot(BoundingBoxWidth float64) *Node {
return &Node{
Boundry: BoundingBox{
Center: Vec2{0, 0},
Width: BoundingBoxWidth,
},
CenterOfMass: Vec2{},
TotalMass: 0,
Depth: 0,
Star: Star2D{},
Subtrees: [4]*Node{},
}
}
// Create a new new node using the given bounding box
func NewNode(bounadry BoundingBox) *Node {
return &Node{Boundry: bounadry}
}
// Subdivide the tree
func (n *Node) subdivide() {
// define new values defining the new BoundaryBoxes
newBoundaryWidth := n.Boundry.Width / 2
newBoundaryPosX := n.Boundry.Center.X + (newBoundaryWidth / 2)
newBoundaryPosY := n.Boundry.Center.Y + (newBoundaryWidth / 2)
newBoundaryNegX := n.Boundry.Center.X - (newBoundaryWidth / 2)
newBoundaryNegY := n.Boundry.Center.Y - (newBoundaryWidth / 2)
// define the new Subtrees
n.Subtrees[0] = NewNode(BoundingBox{Vec2{newBoundaryNegX, newBoundaryPosY}, newBoundaryWidth})
n.Subtrees[1] = NewNode(BoundingBox{Vec2{newBoundaryPosX, newBoundaryPosY}, newBoundaryWidth})
n.Subtrees[2] = NewNode(BoundingBox{Vec2{newBoundaryNegX, newBoundaryNegY}, newBoundaryWidth})
n.Subtrees[3] = NewNode(BoundingBox{Vec2{newBoundaryPosX, newBoundaryNegY}, newBoundaryWidth})
}
// Insert inserts the given star into the Node or the tree it is called on
func (n *Node) Insert(star Star2D) {
fmt.Printf("Hello, this is the insert function, I'm inserting the star %v", star)
// prevent the function to recurse to deep into the tree
if n.Boundry.Width < 0.1 {
return
}
// if the subtree does not contain a node, insert the star
if n.Star == (Star2D{}) {
// if a subtree is present, insert the star into that subtree
if n.Subtrees != [4]*Node{} {
QuadrantBlocking := star.getRelativePositionInt(n.Boundry)
n.Subtrees[QuadrantBlocking].Insert(star)
// directly insert the star into the node
} else {
n.Star = star
}
// Move the star blocking the slot into it's subtree using a recursive call on this function
// and add the star to the slot
} else {
// if the node does not all ready have child nodes, subdivide it
if n.Subtrees == ([4]*Node{}) {
n.subdivide()
}
// Insert the blocking star into it's subtree
QuadrantBlocking := n.Star.getRelativePositionInt(n.Boundry)
n.Subtrees[QuadrantBlocking].Insert(n.Star)
n.Star = Star2D{}
// Insert the blocking star into it's subtree
QuadrantBlockingNew := star.getRelativePositionInt(n.Boundry)
n.Subtrees[QuadrantBlockingNew].Insert(star)
star = Star2D{}
}
fmt.Println("Done inserting %v, the tree noe looks like this: %v", star, n)
}
// GenForestTree draws the subtree it is called on. If there is a star inside of the root node, the node is drawn
// The method returns a string depicting the tree in latex forest structure
func (n Node) GenForestTree(node *Node) string {
returnstring := "["
// if there is a star in the node, add the stars coordinates to the return string
if n.Star != (Star2D{}) {
returnstring += fmt.Sprintf("%.0f %.0f", n.Star.C.X, n.Star.C.Y)
}
// iterate over all the subtrees and call the GenForestTree method on the subtrees containing children
for i := 0; i < len(n.Subtrees); i++ {
if n.Subtrees[i] != nil {
returnstring += n.Subtrees[i].GenForestTree(n.Subtrees[i])
} else {
returnstring += "[]"
}
}
// Post-tree brace
returnstring += "]"
return returnstring
}
// DrawTreeLaTeX writes the tree it is called on to a texfile defined by the outpath parameter and
// calls lualatex to build the tex-file
func (n Node) DrawTreeLaTeX(outpath string) {
// define all the stuff in front of the tree
preamble := `\documentclass{article}
\usepackage{tikz}
\usepackage{forest}
\usepackage{adjustbox}
\begin{document}
\begin{adjustbox}{max size={\textwidth}{\textheight}}
\begin{forest}
for tree={,draw, s sep+=0.25em}
`
// define all the stuff after the tree
poststring := `
\end{forest}
\end{adjustbox}
\end{document}
`
// combine all the strings
data := []byte(fmt.Sprintf("%s%s%s", preamble, n.GenForestTree(&n), poststring))
// write them to a file
writeerr := ioutil.WriteFile(outpath, data, 0644)
if writeerr != nil {
panic(writeerr)
}
// build the pdf
cmd := exec.Command("lualatex", outpath)
runerr := cmd.Run()
if runerr != nil {
panic(runerr)
}
}
// GetAllStars returns all the stars in the tree it is called on in an array
func (n Node) GetAllStars() []string {
// define a list to store the stars
listOfNodes := []string{}
// if there is a star in the node, append the star to the list
if n.Star != (Star2D{}) {
starJson, err := json.Marshal(n.Star)
if err != nil {
panic(err)
}
listOfNodes = append(listOfNodes, string(starJson))
}
// iterate over all the subtrees
for i := 0; i < len(n.Subtrees); i++ {
if n.Subtrees[i] != nil {
// insert all the stars from the subtrees into the list of nodes
for _, star := range n.Subtrees[i].GetAllStars() {
listOfNodes = append(listOfNodes, star)
}
}
}
return listOfNodes
}
|