From 1265c72531d0cf76b52508e33e9488b62688ae1b Mon Sep 17 00:00:00 2001
From: "mergify[bot]" <37929162+mergify[bot]@users.noreply.github.com>
Date: Tue, 9 Jan 2024 20:45:56 +0100
Subject: [PATCH] Add autodocs for nested and __map_ parameters (#216) (#221)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* Use new rst functions of GPL

* Add optional "context" for description of structs

* Make multiline text from yaml more robust

* Do not sort the parameters

(cherry picked from commit 11916f841bc091c7c0ef44a6c80541e245bc4855)

Co-authored-by: Christoph Fröhlich <christophfroehlich@users.noreply.github.com>
---
 _ext/generate_parameter_library.py | 72 ++++++++++++++++++++++++++++--
 1 file changed, 69 insertions(+), 3 deletions(-)

diff --git a/_ext/generate_parameter_library.py b/_ext/generate_parameter_library.py
index b48aa9ab40a..5bbd6ff1804 100644
--- a/_ext/generate_parameter_library.py
+++ b/_ext/generate_parameter_library.py
@@ -2,30 +2,96 @@
 from sphinx.util.docutils import SphinxDirective
 from sphinx.util.nodes import nested_parse_with_titles
 from docutils.statemachine import ViewList
+import yaml
+import re
 from generate_parameter_library_py.parse_yaml import (
     GenerateCode,
 )
 from generate_parameter_library_py.generate_markdown import (
     DefaultConfigMarkdown,
-    ParameterDetailMarkdown
+    ParameterDetailMarkdown,
+    RuntimeParameterDetailMarkdown
 )
 
+def insert_additional_parameters_after(items, keys, insert_map):
+  for key in insert_map.keys():
+      # Define the level you're looking for
+      level_to_find = key.split('.')[0]
+
+      # Reverse keys and find the index of the first occurrence of the level
+      reversed_keys = list(reversed(keys))
+      reversed_index = next((index for (index, item) in enumerate(reversed_keys) if item.split('.')[0] == level_to_find), None)
+
+      if reversed_index is not None:
+          # If the level is found, insert the runtime_param_item after it in the original list
+          index = len(keys) - 1 - reversed_index
+          items = items[:index+1] + [insert_map[key]] + items[index+1:]
+          keys = keys[:index+1] + [key] + keys[index+1:]
+      else:
+          # If the level is not found, append the runtime_param_item to the end of the list
+          items.append(insert_map[key])
+          keys.append(key)
+  return items, keys
+
+def insert_additional_parameters_before(items, keys, insert_map):
+  for key in insert_map.keys():
+      # Define the level you're looking for
+      level_to_find = key.split('.')[0]
+
+      # Find the index of the first occurrence of the level in items
+      index = next((index for (index, item) in enumerate(keys) if item.split('.')[0] == level_to_find), None)
+
+      if index is not None:
+          # If the level is found, insert the runtime_param_item after it
+          items = items[:index] + [insert_map[key]] + items[index:]
+          keys = keys[:index] + [key] + keys[index:]
+      else:
+          # If the level is not found, append the runtime_param_item to the end of the list
+          items.append(insert_map[key])
+          keys.append(key)
+  return items, keys
+
 class GeneraterParameterLibraryDetails(SphinxDirective):
     required_arguments = 1
-    optional_arguments = 0
+    optional_arguments = 1 # context yaml file, "key: string"
 
     def run(self):
         # get the absolute path from sphinx tree
         yaml_file = self.env.relfn2path(self.arguments[0], self.env.docname)[0]
+        if len(self.arguments) > 1:
+          context_yaml_file = self.env.relfn2path(self.arguments[1], self.env.docname)[0]
+          with open(context_yaml_file, 'r') as file:
+            context_yaml_data = yaml.safe_load(file)
+        else:
+          context_yaml_data = {}
 
         gen_param_struct = GenerateCode("rst")
         gen_param_struct.parse(yaml_file, "")
 
+        # general parameters
         param_details = [
             ParameterDetailMarkdown(param)
             for param in gen_param_struct.declare_parameters
         ]
-        docs = "\n".join(str(val) for val in param_details)
+        # runtime parameters, i.e., such with a __map_ key
+        runtime_param_details = [
+            RuntimeParameterDetailMarkdown(param)
+            for param in gen_param_struct.declare_dynamic_parameters
+        ]
+
+        param_strings_keys = [detail.declare_parameters.parameter_name  for detail in param_details]
+        param_items = [str(detail) for detail in param_details]
+        runtime_param_strings_map = {detail.declare_parameters.parameter_name: str(detail) for detail in runtime_param_details}
+        # add optional context data from yaml. we don't use a jinja template here -> add the indent manually
+        context_strings_map = {key: str(key) + "\n" +
+          '\n'.join('  ' + line for line in str(value).replace('\\t', '  ').splitlines()) + "\n"
+          for key, value in context_yaml_data.items()}
+
+        param_items, param_strings_keys = insert_additional_parameters_after(param_items, param_strings_keys, runtime_param_strings_map)
+        param_items, param_strings_keys = insert_additional_parameters_before(param_items, param_strings_keys, context_strings_map)
+
+        docs = "\n".join(param_items)
+
         # print(docs)
 
         # Add the content one line at a time.