add rt cores

Author: zschafft-intel

conf/config.yaml

@@ -57,18 +57,21 @@ demo:
   io_system:
     ip: 10.34.106.119
     nic: enp3s0
-    ssh_user: user
-    ssh_password: "password"   # set SSH_PASSWORD env var, or use key-based auth
+    ssh_user: intel
+    ssh_password: "Qwepoi!23"   # set SSH_PASSWORD env var, or use key-based auth
     ssh_port: 22
     # Host-side IP on the macvlan subnet (must not clash with any container IP).
     # Allows the IO host to reach containers via:  curl http://192.168.10.12:8080
     shim_ip: 192.168.10.253
+    t_cpus: "1,3"
   control_system:
     ip: localhost
     nic: eno12399np0
     # Host-side IP on the macvlan subnet (must not clash with any container IP).
     # Allows the control host to reach containers via:  curl http://192.168.10.15:8080
     shim_ip: 192.168.10.254
+    t_cpus: "3,5,7,9"
+    t_mems: "1"
 
   # Map host_ip:<host_port> → container_ip:<container_port> via iptables.
   # This lets external clients access container services

src/hde2e.py

@@ -66,7 +66,24 @@ def _sudo_cmd(self, cmd: str) -> str:
         pw = self._resolve_sudo_password()
         # Use printf to avoid issues with special chars in the password.
         return f"printf '%s\\n' '{pw}' | sudo -S {cmd}"
-
+    def _realtime_core_control(self, instance_num: str) -> Optional[str]:
+        """Return a CPU core list string for the real-time Control PLC instance."""
+        # Example mapping: Control_PLC_01 → cores 0-3, Control_PLC_02 → cores 4-7
+        t_cpus = self.config.demo.control_system.get("t_cpus")
+        t_cpus = t_cpus.split(",")
+        if len(t_cpus) < 4:
+            print("[WARN] Not enough t_cpus specified for control_system. Expected at least 4.")
+            return None
+        if instance_num == "01":
+            return f"{t_cpus[int(instance_num)-1]},{t_cpus[int(instance_num)]}" # cores 3,5
+        elif instance_num == "02":
+            return f"{t_cpus[int(instance_num)]},{t_cpus[len(t_cpus)-1]}" # cores 7,9
+    def _realtime_core_io(self, instance_num: str) -> Optional[str]:
+        """Return a CPU core list string for the real-time IO PLC instance."""
+        # Example mapping: IO_PLC_01 → cores 8-11, IO_PLC_02 → cores 12-15
+        t_cpus = self.config.demo.io_system.get("t_cpus")
+        t_cpus = t_cpus.split(",")
+        return str(t_cpus[instance_num - 1])
     #  Low-level helpers
     @staticmethod
     def _run_command(cmd: List[str], capture: bool = False) -> subprocess.CompletedProcess:
@@ -540,7 +557,7 @@ def launch_instance(
         # 1. Resolve the container IP from the JSON config
         container_ip = self._extract_ip_from_config(app_type, instance_num)
 
-        cpuset_cpus = cpuset_cpus or self.config.demo.get("cpuset_cpus")
+        cpuset_cpus = self._realtime_core_io(int(instance_num)) if app_type == "io" else self._realtime_core_control(instance_num)
         cpuset_mems = cpuset_mems or self.config.demo.get("cpuset_mems")
         if not container_ip:
             print(f"[ERROR] No IP found for {container_name}")
@@ -654,6 +671,8 @@ def start_control(self) -> int:
                 ssh=None,
                 network=DEFAULT_NETWORK,
                 nic=nic,
+                cpuset_mems=self.config.demo.control_system.get("cpuset_mems"),
+                cpuset_cpus=self._realtime_core_control(instance_num),
             )
             if rc != 0:
                 return rc
@@ -700,6 +719,7 @@ def start_io(self) -> int:
                     ssh=ssh,
                     network=DEFAULT_NETWORK,
                     nic=nic,
+                    cpuset_cpus=self._realtime_core_io(int(instance_num)),
                 )
                 if rc != 0:
                     return rc