From 72f25852875aadd98ab749c145feca6235e17623 Mon Sep 17 00:00:00 2001 From: Mordan Vitalii Date: Wed, 19 Jun 2024 14:25:49 +0500 Subject: [PATCH] Support comments with function names --- scripts/mea/et/internal_witness.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/scripts/mea/et/internal_witness.py b/scripts/mea/et/internal_witness.py index 5079a4e..e86d707 100644 --- a/scripts/mea/et/internal_witness.py +++ b/scripts/mea/et/internal_witness.py @@ -68,6 +68,7 @@ def __init__(self, logger): self._env_models_json = {} self._env_funcs_openers = {} self._env_threads = {} + self._env_funcs = {} self._notes = {} self._asserts = {} self._actions = [] @@ -275,6 +276,8 @@ def process_verifier_notes(self): self._logger.debug(f"Add note {note} for model function " f"'{self._resolve_function(func_id)}'") edge['note'] = self.process_comment(note) + if file_id in self._env_funcs and func_id in self._env_funcs[file_id]: + edge['env'] = self.process_comment(self._env_funcs[file_id][func_id]) if func_id in self._env_models: # TODO: not supported anymore env_note = self._env_models[func_id] @@ -413,6 +416,10 @@ def _parse_model_comments(self): if file_id not in self._env_threads: self._env_threads[file_id] = {} self._env_threads[file_id][self.add_function(data["function"])] = data["thread"] + if "comment" in data and "function" in data and data.get("type") == "CONTROL_FUNCTION_BEGIN": + if file_id not in self._env_funcs: + self._env_funcs[file_id] = {} + self._env_funcs[file_id][self.add_function(data["function"])] = data["comment"] if 'name' in data: func_data = { 'type': data['type'],