46 lines
1.2 KiB
Go
46 lines
1.2 KiB
Go
package gstreamer
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/go-gst/go-gst/gst"
|
|
"github.com/goccy/go-graphviz"
|
|
"github.com/goccy/go-graphviz/cgraph"
|
|
)
|
|
|
|
func gstreamerBinToPNG(bin *gst.Bin, filename string) error {
|
|
ctx := context.Background()
|
|
g, err := graphviz.New(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("error creating graphviz context: %w", err)
|
|
}
|
|
|
|
dotString := bin.DebugBinToDotData(gst.DebugGraphShowAll)
|
|
graph, err := cgraph.ParseBytes([]byte(dotString))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
//graph.SetDPI(150)
|
|
//setFontSize(graph, 20)
|
|
|
|
if err := g.RenderFilename(ctx, graph, graphviz.PNG, filename); err != nil {
|
|
return fmt.Errorf("error rendering graph: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func setFontSize(graph *cgraph.Graph, fontSize float64) {
|
|
graph.SetFontSize(20)
|
|
graph.Set("arrowSize", "3")
|
|
for node, err := graph.FirstNode(); err == nil && node != nil; node, err = graph.NextNode(node) {
|
|
node.SetFontSize(fontSize)
|
|
for edge, err := graph.FirstEdge(node); err == nil && edge != nil; edge, err = graph.NextEdge(edge, node) {
|
|
edge.SetFontSize(fontSize)
|
|
}
|
|
}
|
|
//for subGraph, err := graph.FirstSubGraph(); err == nil && subGraph != nil; subGraph, err = graph.NextSubGraph() {
|
|
// setFontSize(subGraph, fontSize)
|
|
//}
|
|
}
|