whispervis/graph.go

package main

import (
	"runtime"

	"github.com/gopherjs/vecty"
)

// UpdateGraph starts graph layout simulation.
func (p *Page) UpdateGraph() {
	p.loader.Reset()
	p.loaded = false
	vecty.Rerender(p)

	config := p.forceEditor.Config()
	p.loader.SetSteps(config.Steps)
	for i := 0; i < config.Steps; i++ {
		p.layout.UpdatePositions()
		p.loader.Inc()
		vecty.Rerender(p.loader)
		runtime.Gosched()
	}
	p.loaded = true
	// TODO(divan): remove previous objects
	p.webgl.RemoveObjects()
	p.webgl.CreateObjects(p.layout.Positions(), p.layout.Links())

	vecty.Rerender(p)
}

// ApplyForces applies current forces to the objects, and runs
// a single simulation run to update positions.
func (p *Page) ApplyForces() {
	fc := p.forceEditor.Config()
	p.layout.SetConfig(fc.Config)
	p.layout.UpdatePositions()
	p.webgl.updatePositions()
	p.webgl.rt.Disable()
}