Fixup embed script

Signed-off-by: Michael Carroll <[email protected]>

sdf/embedSdf.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 import inspect
+import sys
 from pathlib import Path, PurePosixPath
 
 """"Script for generating a C++ file that contains the content from all SDF files"""
@@ -130,15 +131,26 @@ def get_file_header_epilog() -> str:
 if __name__ == "__main__":
     copyright = get_copyright_notice()
     prolog = get_file_header_prolog()
-
     map_str = ""
-    for sdf_version in SUPPORTED_SDF_VERSIONS:
-        pathlist = Path(sdf_version).glob('*.sdf')
-        map_str += get_map_content(pathlist)
 
-    for sdf_conversion in SUPPORTED_SDF_CONVERSIONS:
-        pathlist = Path(sdf_conversion).glob('*.convert')
-        map_str += get_map_content(pathlist)
+    if len(sys.argv) > 1:
+        # Sources are passed in as arguments (Bazel)
+        for filename in sorted(sys.argv[1:]):
+            _, relative_path = filename.split('/sdf/')
+
+            with open(filename, 'r', encoding='utf-8') as data:
+                content = data.read()
+                map_str += embed_sdf_content(relative_path, content)
+
+    else:
+        # Assume we are in current working directory
+        for sdf_version in SUPPORTED_SDF_VERSIONS:
+            pathlist = Path(sdf_version).glob('*.sdf')
+            map_str += get_map_content(pathlist)
+
+        for sdf_conversion in SUPPORTED_SDF_CONVERSIONS:
+            pathlist = Path(sdf_conversion).glob('*.convert')
+            map_str += get_map_content(pathlist)
 
     # remove the last comma
     map_str = map_str[:-2]

src/SDF_TEST.cc

@@ -21,7 +21,6 @@
 #include <gz/utils/Environment.hh>
 #include <gz/utils/SuppressWarning.hh>
 
-#include "test_config.hh"
 #include "test_utils.hh"
 
 #include "sdf/sdf.hh"