Add Annotations for Monster schema and example buffer

1 files changed, 19 insertions, 0 deletions

diff --git a/scripts/generate_code.py b/scripts/generate_code.py
index 939b7c0b..248c2f40 100755
--- a/scripts/generate_code.py
+++ b/scripts/generate_code.py
@@ -97,6 +97,12 @@ def flatc_reflection(options, location, target):
         shutil.move(str(new_reflection_path), str(original_reflection_path))
     shutil.rmtree(str(Path(reflection_path, temp_dir)))
 
+def flatc_annotate(schema, file, include=None, cwd=tests_path):
+    cmd = [str(flatc_path)]
+    if include:
+        cmd += ["-I"] + [include]
+    cmd += ["--annotate", schema, file]
+    result = subprocess.run(cmd, cwd=str(cwd), check=True)
 
 # Glob a pattern relative to file path
 def glob(path, pattern):
@@ -272,6 +278,19 @@ flatc(
     schema="monster_test.fbs",
 )
 
+# Generate the annotated binary of the monster_test binary schema.
+flatc_annotate(
+    schema="../reflection/reflection.fbs", 
+    file="monster_test.bfbs", 
+    include="include_test"
+)
+
+flatc_annotate(
+    schema="monster_test.fbs", 
+    file="monsterdata_test.mon", 
+    include="include_test"
+)
+
 flatc(
     CPP_OPTS
     + NO_INCL_OPTS