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
|
// star.go defines stars and actions that can be used on them
// Copyright (C) 2019 Emile Hansmaennel
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
package viewer
import (
"encoding/json"
"fmt"
"io/ioutil"
"log"
"net/http"
"strconv"
"github.com/ajstarks/svgo"
"github.com/gorilla/mux"
"git.darknebu.la/GalaxySimulator/structs"
)
const (
width = 1920 * 8
height = 1920 * 8
)
var (
treeArray []*structs.Node
)
func drawTree(w http.ResponseWriter, r *http.Request) {
log.Println("The drawtree handler was accessed")
w.Header().Set("Content-Type", "image/svg+xml")
// get the tree index
vars := mux.Vars(r)
treeindex, _ := strconv.ParseInt(vars["treeindex"], 10, 0)
// define the svg
s := svg.New(w)
s.Start(width, height)
s.Rect(0, 0, width, height, s.RGB(0, 0, 0))
s.Gtransform(fmt.Sprintf("translate(%d,%d)", width/2, height/2))
getGalaxy(treeindex)
listOfStars := treeArray[treeindex].GetAllStars()
// draw the galaxy
drawStars(s, listOfStars)
drawBoxes(s, treeindex)
s.Gend()
s.End()
}
func drawStars(s *svg.SVG, listOfStars []structs.Star2D) {
log.Println("[ ] Drawing the stars")
for _, star := range listOfStars {
x := int(star.C.X / 2000)
y := int(star.C.Y / 2000)
s.Circle(x, y, 1, s.RGB(255, 255, 255))
}
log.Println("[ ] Done drawing the stars")
}
func drawBoxes(s *svg.SVG, treeindex int64) {
log.Println("[ ] Drawing the Boxes")
drawBox(s, treeArray[treeindex])
log.Println("[ ] Done drawing the Boxes")
}
func drawBox(s *svg.SVG, node *structs.Node) {
if node.Boundry != (structs.BoundingBox{}) {
x := int(node.Boundry.Center.X / 2000)
y := int(node.Boundry.Center.Y / 2000)
w := int(node.Boundry.Width / 2000)
s.CenterRect(x, y, w, w, "fill:none;stroke:white")
}
for i := 0; i < len(node.Subtrees); i++ {
if node.Subtrees[i] != nil {
drawBox(s, node.Subtrees[i])
}
}
}
func getGalaxy(index int64) {
log.Println("[ ] Getting the Galaxy")
// make a http-post request to the databse requesting the tree
requesturl := fmt.Sprintf("http://db.nbg1.emile.space/dumptree/%d", index)
resp, err := http.Get(requesturl)
if err != nil {
panic(err)
}
defer resp.Body.Close()
body, readerr := ioutil.ReadAll(resp.Body)
if readerr != nil {
panic(readerr)
}
tree := &structs.Node{}
jsonUnmarshalErr := json.Unmarshal(body, tree)
if jsonUnmarshalErr != nil {
panic(jsonUnmarshalErr)
}
// if the treeArray is not long enough, fill it
for int(index) > len(treeArray) {
emptyNode := structs.NewNode(structs.NewBoundingBox(structs.NewVec2(0, 0), 10))
treeArray = append(treeArray, emptyNode)
}
treeArray = append(treeArray, tree)
log.Println("[ ] Done Getting the galaxy")
}
func main() {
router := mux.NewRouter()
router.HandleFunc("/drawtree/{treeindex}", drawTree).Methods("GET")
log.Fatal(http.ListenAndServe(":80", router))
}
|
