status-im/whispervis
2
0
Fork 0
mirror of https://github.com/status-im/whispervis.git synced 2025-02-02 16:33:56 +00:00
Code Issues Projects Releases Wiki Activity

Initial refactoring to use graphx

Browse Source
This commit is contained in:
Ivan Daniluk 2018-07-12 12:31:15 +02:00
parent 9ee56d0720
commit a502d29b8f
No known key found for this signature in database
GPG Key ID: 97ED33CE024E1DBF
4 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
main.go
View File

@ -4,8 +4,8 @@ import (
"flag"
"log"
"github.com/divan/graph-experiments/graph"
"github.com/divan/graph-experiments/layout"
"github.com/divan/graphx/formats"
"github.com/divan/graphx/layout"
)
func main() {
@ -13,7 +13,7 @@ func main() {
iterations := flag.Int("i", 600, "Graph layout iterations to run (0 = auto, buggy)")
flag.Parse()
data, err := graph.NewGraphFromJSON("network.json")
data, err := formats.FromD3JSON("network.json")
if err != nil {
log.Fatal(err)
}
@ -26,7 +26,7 @@ func main() {
log.Printf("Loaded propagation data: %d timestamps\n", len(plog.Timestamps))
log.Printf("Initializing layout...")
repelling := layout.NewGravityForce(-100.0, layout.BarneHutMethod)
repelling := layout.NewGravityForce(-10.0, layout.BarneHutMethod)
springs := layout.NewSpringForce(0.01, 5.0, layout.ForEachLink)
drag := layout.NewDragForce(0.4, layout.ForEachNode)
layout3D := layout.New(data, repelling, springs, drag)

4
ws.go
View File

@ -5,8 +5,8 @@ import (
"log"
"net/http"
"github.com/divan/graph-experiments/graph"
"github.com/divan/graph-experiments/layout"
"github.com/divan/graphx/graph"
"github.com/divan/graphx/layout"
"github.com/gorilla/websocket"
)

7
ws_graph.go
View File

@ -5,14 +5,15 @@ import (
"encoding/json"
"log"
"github.com/divan/graph-experiments/export"
"github.com/divan/graph-experiments/graph"
"github.com/divan/graphx/formats"
"github.com/divan/graphx/graph"
"github.com/gorilla/websocket"
)
func (ws *WSServer) sendGraphData(c *websocket.Conn) {
var buf bytes.Buffer
err := export.NewJSON(&buf, false).ExportGraph(ws.graph)
d3json := formats.NewD3JSON(&buf, false)
err := d3json.ExportGraph(ws.graph)
if err != nil {
log.Fatal("Can't marshal graph to JSON")
}

8
ws_positions.go
View File

@ -21,11 +21,11 @@ func (ws *WSServer) updatePositions() {
// positions
nodes := ws.layout.Nodes()
positions := []*position{}
for i := 0; i < len(nodes); i++ {
for _, node := range nodes {
pos := &position{
X: nodes[i].X,
Y: nodes[i].Y,
Z: nodes[i].Z,
X: node.X,
Y: node.Y,
Z: node.Z,
}
positions = append(positions, pos)
}