Building of merged JAR of modelled and unmodelled classes in rt.jar.

Author: marek-trtik

benchmarks/LIBRARIES/models/model/mkrtm.py

@@ -0,0 +1,66 @@
+import os
+import shutil
+import argparse
+import tempfile
+import contextlib
+
+
+@contextlib.contextmanager
+def _working_dir(new_dir):
+    previous_dir = os.getcwd()
+    if not os.path.isdir(new_dir):
+        os.makedirs(new_dir)
+    os.chdir(new_dir)
+    yield
+    os.chdir(previous_dir)
+
+
+def _parse_cmd_line():
+    parser = argparse.ArgumentParser(
+        description="Creates a JAR file consisting of modelled class files and "
+                    "other class files in the passed 'rt.jar'.")
+    parser.add_argument("rt", type=str,
+                        help="A path-name of the 'rt.jar' file.")
+    parser.add_argument("--output-name", type=str, default="rtm.jar",
+                        help="A name of the final JAR file (without path).")
+    cmdline = parser.parse_args()
+    if not os.path.isfile(cmdline.rt):
+        print("ERROR: The passed file '" + cmdline.rt + "' does not exist.")
+        exit(1)
+    if os.path.splitext(cmdline.rt)[1].lower() != ".jar":
+        print("ERROR: The passed file '" + cmdline.rt + "' does not have '.jar' extension.")
+        exit(1)
+    cmdline.rt = os.path.abspath(cmdline.rt)
+    return cmdline
+
+
+def _mkrtm(cmdline):
+    model_classes_root_dir = os.path.abspath(os.path.join("target", "classes"))
+    temp_root_dir = tempfile.mkdtemp()
+    classes_root_dir = os.path.join(temp_root_dir, "classes")
+    rt_unpack_root_dir = os.path.join(temp_root_dir, "RT_UNPACK")
+    dst_jar_pathname = os.path.abspath(os.path.join("target", cmdline.output_name))
+
+    shutil.copytree(model_classes_root_dir, classes_root_dir)
+    with _working_dir(rt_unpack_root_dir):
+        os.system("jar -xvf  '" + cmdline.rt + "'")
+    for root, _, file_names in os.walk(rt_unpack_root_dir):
+        for filename in file_names:
+            if os.path.splitext(filename)[1].lower() == ".class":
+                src_pathname = os.path.abspath(os.path.join(root, filename))
+                rel_pathname = os.path.relpath(src_pathname, rt_unpack_root_dir)
+                dst_pathname = os.path.join(classes_root_dir, rel_pathname)
+                if not os.path.isfile(dst_pathname):
+                    if not os.path.isdir(os.path.dirname(dst_pathname)):
+                        os.makedirs(os.path.dirname(dst_pathname))
+                    shutil.copy(src_pathname, dst_pathname)
+    with _working_dir(classes_root_dir):
+        os.system("jar cvfm " + os.path.basename(dst_jar_pathname) + " " +
+                  os.path.join(rt_unpack_root_dir, "META-INF", "MANIFEST.MF") +
+                  " .")
+    shutil.copy(os.path.join(classes_root_dir, os.path.basename(dst_jar_pathname)),
+                dst_jar_pathname)
+
+
+if __name__ == '__main__':
+    _mkrtm(_parse_cmd_line())