example_push_get_inspect_test.go

package proio_test

import (
	"bytes"
	"fmt"

	"github.com/proio-org/go-proio"
	model "github.com/proio-org/go-proio-pb/model/example"
)

func Example_pushGetInspect() {
	buffer := &bytes.Buffer{}
	writer := proio.NewWriter(buffer)

	eventOut := proio.NewEvent()

	// Create entries and hold onto their IDs for referencing

	parent := &model.Particle{Pdg: 443}
	parentID := eventOut.AddEntry("Particle", parent)
	eventOut.TagEntry(parentID, "Truth", "Primary")

	child1 := &model.Particle{Pdg: 11}
	child2 := &model.Particle{Pdg: -11}
	childIDs := eventOut.AddEntries("Particle", child1, child2)
	for _, id := range childIDs {
		eventOut.TagEntry(id, "Truth", "GenStable")
	}

	parent.Child = append(parent.Child, childIDs...)
	child1.Parent = append(child1.Parent, parentID)
	child2.Parent = append(child2.Parent, parentID)

	writer.Push(eventOut)

	writer.Flush()

	// Event created and serialized, now to deserialize and inspect

	reader := proio.NewReader(buffer)
	eventIn := reader.Next()

	mcParts := eventIn.TaggedEntries("Primary")
	fmt.Print(len(mcParts), " Primary particle(s)...\n")
	for i, parentID := range mcParts {
		part := eventIn.GetEntry(parentID).(*model.Particle)
		fmt.Print(i, ". PDG: ", part.GetPdg(), "\n")
		fmt.Print("  ", len(part.Child), " children...\n")
		for j, childID := range part.Child {
			fmt.Print("  ", j, ". PDG: ", eventIn.GetEntry(childID).(*model.Particle).GetPdg(), "\n")
		}
	}

	// Output:
	// 1 Primary particle(s)...
	// 0. PDG: 443
	//   2 children...
	//   0. PDG: 11
	//   1. PDG: -11
}