Merge pull request #2376 from DARMA-tasking/2375-extend-json-format-f…

…or-dervied-stats-and-lb-iterations #2375: scripts: add lb_iterations to JSON data file schema validator
DARMA-tasking · Dec 11, 2024 · e485ac7 · e485ac7
2 parents 6190033 + e28e310
commit e485ac7
Showing 1 changed file with 59 additions and 1 deletion.
diff --git a/scripts/LBDatafile_schema.py b/scripts/LBDatafile_schema.py
@@ -93,7 +93,65 @@ def validate_ids(field):
                         'bytes': float
                     }
                 ],
-                Optional('user_defined'): dict
+                Optional('user_defined'): dict,
+                Optional('lb_iterations'): [
+                    {
+                        'id': int,
+                        'tasks': [
+                            {
+                                'entity': And({
+                                    Optional('collection_id'): int,
+                                    'home': int,
+                                    Optional('id'): int,
+                                    Optional('seq_id'): int,
+                                    Optional('index'): [int],
+                                    'type': str,
+                                    'migratable': bool,
+                                    Optional('objgroup_id'): int
+                                }, validate_ids),
+                                'node': int,
+                                'resource': str,
+                                Optional('subphases'): [
+                                    {
+                                        'id': int,
+                                        'time': float,
+                                    }
+                                ],
+                                'time': float,
+                                Optional('user_defined'): dict,
+                                Optional('attributes'): dict
+                            },
+                        ],
+                        Optional('communications'): [
+                            {
+                                'type': str,
+                                'to': And({
+                                    'type': str,
+                                    Optional('id'): int,
+                                    Optional('seq_id'): int,
+                                    Optional('home'): int,
+                                    Optional('collection_id'): int,
+                                    Optional('migratable'): bool,
+                                    Optional('index'): [int],
+                                    Optional('objgroup_id'): int,
+                                }, validate_ids),
+                                'messages': int,
+                                'from': And({
+                                    'type': str,
+                                    Optional('id'): int,
+                                    Optional('seq_id'): int,
+                                    Optional('home'): int,
+                                    Optional('collection_id'): int,
+                                    Optional('migratable'): bool,
+                                    Optional('index'): [int],
+                                    Optional('objgroup_id'): int,
+                                }, validate_ids),
+                                'bytes': float
+                            }
+                        ],
+                        Optional('user_defined'): dict
+                    }
+                ]
             },
         ]
     }