groot/rdict: generate RVersioner if not implemented

go-hep · Feb 11, 2022 · 2a34461 · 2a34461
1 parent d23d2f4
commit 2a34461
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/groot/rdict/gen_streamer.go b/groot/rdict/gen_streamer.go
@@ -130,7 +130,7 @@ func (g *genStreamer) Generate(typeName string) error {
 	}
 
 	if !types.Implements(tn.Type(), g.rvers) && !types.Implements(types.NewPointer(tn.Type()), g.rvers) {
-		return fmt.Errorf("type %q does not implement %q", tn.Pkg().Path()+"."+tn.Name(), "go-hep.org/x/hep/groot/rbytes.RVersioner")
+		g.genRVersioner(typ, typeName)
 	}
 
 	g.genStreamer(typ, typeName)
@@ -177,6 +177,10 @@ func (o *%[1]s) MarshalROOT(w *rbytes.WBuffer) (int, error) {
 // 	)
 // }
 
+func (g *genStreamer) genRVersioner(t types.Type, typeName string) {
+	g.printf("func (*%s) RVersion() int16 { return 1 }\n\n", typeName)
+}
+
 func (g *genStreamer) genStreamer(t types.Type, typeName string) {
 	g.printf(`func init() {
 	// Streamer for %[1]s.