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
|
package backend
import (
"database/sql"
"fmt"
"log"
)
// GenForestTree generates a forest representation of the tree with the given index
func GenForestTree(database *sql.DB, index int64) string {
db = database
rootNodeID := getRootNodeID(index)
return genForestTreeNode(rootNodeID)
}
// genForestTreeNodes returns a sub-representation of a given node in forest format
func genForestTreeNode(nodeID int64) string {
var returnString string
// get the subnode ids
var subnode [4]int64
query := fmt.Sprintf("SELECT subnode[1], subnode[2], subnode[3], subnode[4] FROM nodes WHERE node_id=%d", nodeID)
err := db.QueryRow(query).Scan(&subnode[0], &subnode[1], &subnode[2], &subnode[3])
if err != nil {
log.Fatalf("[ E ] updateTotalMassNode query: %v\n\t\t\t query: %s\n", err, query)
}
returnString += "["
// iterate over all subnodes updating their total masses
for _, subnodeID := range subnode {
if subnodeID != 0 {
centerOfMass := getCenterOfMass(nodeID)
mass := getNodeTotalMass(nodeID)
returnString += fmt.Sprintf("%.0f %.0f %.0f", centerOfMass.X, centerOfMass.Y, mass)
returnString += genForestTreeNode(subnodeID)
} else {
if getStarID(nodeID) != 0 {
coords := getStarCoordinates(nodeID)
starID := getStarID(nodeID)
mass := getStarMass(starID)
returnString += fmt.Sprintf("[%.0f %.0f %.0f]", coords.X, coords.Y, mass)
} else {
returnString += fmt.Sprintf("[0 0]")
}
// break, this stops a star from being counted multiple (4) times
break
}
}
returnString += "]"
return returnString
}
|
