fix: use attr_map to scope varid iteration in APIContainRelation online_check

Add Checker_data.attr_map (var_type → attr_name → set[VarInstId]), populated
in _set_var_map when an attribute is first observed for a variable. Replace the
broad type_map iteration in APIContainRelation.online_check and
query_var_changes_within_time_and_process with attr_map lookups that only visit
varids known to carry the attribute. This eliminates the KeyError when frozen
parameters (or any variable lacking a tracked attribute) appear in type_map.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: Essoz

traincheck/invariant/contain_relation.py

@@ -1315,35 +1315,35 @@ def online_check(
 
         if isinstance(child_param, (VarTypeParam, VarNameParam)):
             events = []
+            attr_name = child_param.attr_name
             with checker_data.lock:
-                if child_param.var_type in checker_data.type_map:
-                    for varid in checker_data.type_map[child_param.var_type]:
-                        if isinstance(child_param, VarNameParam):
-                            if varid.var_name != child_param.var_name:
-                                continue
-                            attr_name = child_param.attr_name
-                        elif isinstance(child_param, VarTypeParam):
-                            attr_name = child_param.attr_name
-                        if attr_name not in checker_data.varid_map[varid]:
+                candidate_varids = checker_data.attr_map.get(
+                    child_param.var_type, {}
+                ).get(attr_name, set())
+                if isinstance(child_param, VarNameParam):
+                    candidate_varids = {
+                        v
+                        for v in candidate_varids
+                        if v.var_name == child_param.var_name
+                    }
+                for varid in candidate_varids:
+                    for i in reversed(
+                        range(1, len(checker_data.varid_map[varid][attr_name]))
+                    ):
+                        change_time = checker_data.varid_map[varid][attr_name][
+                            i
+                        ].liveness.start_time
+                        if change_time <= pre_time:
+                            break
+                        if change_time > post_time:
                             continue
-                        for i in reversed(
-                            range(1, len(checker_data.varid_map[varid][attr_name]))
-                        ):
-
-                            change_time = checker_data.varid_map[varid][attr_name][
-                                i
-                            ].liveness.start_time
-                            if change_time <= pre_time:
-                                break
-                            if change_time > post_time:
-                                continue
-                            new_state = checker_data.varid_map[varid][attr_name][i]
-                            old_state = checker_data.varid_map[varid][attr_name][i - 1]
-                            if new_state.value == old_state.value:
-                                continue
-                            if new_state.liveness.end_time is None:
-                                new_state = copy.deepcopy(new_state)
-                            events.append((old_state, new_state))
+                        new_state = checker_data.varid_map[varid][attr_name][i]
+                        old_state = checker_data.varid_map[varid][attr_name][i - 1]
+                        if new_state.value == old_state.value:
+                            continue
+                        if new_state.liveness.end_time is None:
+                            new_state = copy.deepcopy(new_state)
+                        events.append((old_state, new_state))
 
             if check_relation_first:
                 for event in events:

traincheck/onlinechecker/streamhandler_filesystem.py

@@ -34,6 +34,7 @@ def __init__(self, file_path, checker_data: Checker_data):
 
         self.varid_map = checker_data.varid_map
         self.type_map = checker_data.type_map
+        self.attr_map = checker_data.attr_map
         self.pt_map = checker_data.pt_map
         self.process_to_vars = checker_data.process_to_vars
         self.args_map = checker_data.args_map
@@ -155,6 +156,11 @@ def _set_var_map(self, trace_record):
 
                     if attr_name not in self.varid_map[varid]:
                         self.varid_map[varid][attr_name] = []
+                        if varid.var_type not in self.attr_map:
+                            self.attr_map[varid.var_type] = {}
+                        if attr_name not in self.attr_map[varid.var_type]:
+                            self.attr_map[varid.var_type][attr_name] = set()
+                        self.attr_map[varid.var_type][attr_name].add(varid)
                     else:
                         self.varid_map[varid][attr_name][-1].liveness.end_time = (
                             trace_record["time"]

traincheck/onlinechecker/utils.py

@@ -17,6 +17,7 @@ def __init__(self, needed_data):
         self.check_queue = queue.Queue()
         self.varid_map = {}
         self.type_map = {}
+        self.attr_map: dict[str, dict[str, set]] = {}
         self.pt_map = {}
         self.process_to_vars = {}
         self.args_map = {}
@@ -189,7 +190,7 @@ def query_var_changes_within_time_and_process(
     """Extract all variable change events from the trace, within a specific time range and process."""
     events = []
     with checker_data.lock:
-        for varid in checker_data.type_map[var_type]:
+        for varid in checker_data.attr_map.get(var_type, {}).get(attr_name, set()):
             for i in reversed(range(1, len(checker_data.varid_map[varid][attr_name]))):
 
                 change_time = checker_data.varid_map[varid][attr_name][