update docker images

Signed-off-by: Yutaka Kondo <[email protected]>
autowarefoundation · Sep 20, 2024 · f444680 · f444680
1 parent 45a41a5
commit f444680
Showing 1 changed file with 64 additions and 24 deletions.
diff --git a/scripts/measure_workflows.py b/scripts/measure_workflows.py
@@ -164,16 +164,28 @@ def auth(dxf, response):
         dxf.authenticate(github_actor, github_token, response=response)
 
     docker_images = {
-        "base-amd64": [],
-        "autoware-core-amd64": [],
-        "autoware-universe-amd64": [],
-        "devel-amd64": [],
-        "runtime-amd64": [],
-        "base-cuda-amd64": [],
-        "autoware-core-cuda-amd64": [],
-        "autoware-universe-cuda-amd64": [],
-        "devel-cuda-amd64": [],
-        "runtime-cuda-amd64": [],
+        "base": [],
+        "core": [],
+        "core-devel": [],
+        "universe-sensing-perception": [],
+        "universe-sensing-perception-devel": [],
+        "universe-localization-mapping": [],
+        "universe-localization-mapping-devel": [],
+        "universe-planning-control": [],
+        "universe-planning-control-devel": [],
+        "universe": [],
+        "universe-devel": [],
+        "base-cuda": [],
+        "core-cuda": [],
+        "core-devel-cuda": [],
+        "universe-sensing-perception-cuda": [],
+        "universe-sensing-perception-devel-cuda": [],
+        "universe-localization-mapping-cuda": [],
+        "universe-localization-mapping-devel-cuda": [],
+        "universe-planning-control-cuda": [],
+        "universe-planning-control-devel-cuda": [],
+        "universe-cuda": [],
+        "universe-devel-cuda": [],
     }
 
     dxf = DXF("ghcr.io", f"{DOCKER_ORGS}/{DOCKER_IMAGE}", auth)
@@ -185,20 +197,48 @@ def auth(dxf, response):
         if not tag.endswith("amd64"):
             continue
         docker_image = ""
-        for key in (
-            "base",
-            "autoware-core",
-            "autoware-universe",
-            "devel",
-            "runtime",
-        ):
-            if key in tag:
-                docker_image = (
-                    key
-                    + ("-cuda-" if "cuda" in tag else "-")
-                    + "amd64"
-                )
-                break
+        if "autoware-" in tag:
+            for key in (
+                "autoware-core",
+                "autoware-universe",
+            ):
+                if key in tag:
+                    docker_image = (
+                        ("core-devel" if "core" in tag else "universe-devel")
+                        + ("-cuda" if "cuda" in tag else "")
+                    )
+                    break
+        else:
+            matched = False
+            for key in ("universe-sensing-perception",
+                        "universe-localization-mapping",
+                        "universe-planning-control"):
+                if key in tag:
+                    docker_image = (
+                        key
+                        + ("-devel" if "devel" in tag else "")
+                        + ("-cuda" if "cuda" in tag else "")
+                    )
+                    matched = True
+                    break
+            if not matched:
+                if "universe" in tag:
+                    docker_image = (
+                        key
+                        + ("-devel" if "devel" in tag else "")
+                        + ("-cuda" if "cuda" in tag else "")
+                    )
+                elif "devel" in tag:
+                    docker_image = (
+                        "universe-devel"
+                        + ("-cuda" if "cuda" in tag else "")
+                    )
+                elif "runtime" in tag:
+                    docker_image = (
+                        "universe"
+                        + ("-cuda" if "cuda" in tag else "")
+                    )
+        print(docker_image)
         if docker_image == "":
             continue