Dashboard: update python parser

src/python/impactx/dashboard/Input/utils.py

@@ -6,6 +6,7 @@
 License: BSD-3-Clause-LBNL
 """
 
+from pathlib import Path
 from typing import Union
 
 from .. import state
@@ -139,3 +140,18 @@
         current_input = getattr(state, state_name)
         numeric_input = GeneralFunctions.convert_to_numeric(current_input)
         setattr(state, state_name, numeric_input)
+
+    def get_impactx_root_dir() -> Path | None:
+        """
+        Locates the ImpactX source directory.
+
+        Looks for the outermost parent directory named 'impactx' that contains a '.git' folder.
+        """
+
+        current_directory = Path(__file__).resolve()
+        root_dir = None
+
+        for parent_dir in current_directory.parents:
+            if parent_dir.name == "impactx" and (parent_dir / ".git").is_dir():
+                root_dir = parent_dir  # keep going until we reach the highest match
+        return root_dir

src/python/impactx/dashboard/Input/visualization/lattice/statistics.py

@@ -69,9 +69,10 @@ def update_length_statistics() -> None:
     @staticmethod
     def update_element_counts() -> dict[str, int]:
         """
-        Computes the element counts in the lattice list.
+        Computes the element counts in the lattice list
+        and stores them in descending order by count.
 
-        :return: Dictionary of element names and their counts, sorted by count descending.
+        :return: Dictionary of element counts indexed by element name.
         """
         counts = {}
         for element in state.selected_lattice_list:

src/python/impactx/dashboard/Toolbar/examples_loader/loader.py

@@ -16,15 +16,15 @@
 
 DASHBOARD_EXAMPLES = {
     "fodo/run_fodo.py",
-    # "chicane/run_chicane_csr.py",
-    # "fodo_space_charge/run_fodo_envelope_sc.py",
-    # "apochromatic/run_apochromatic.py",
-    # "kurth/run_kurth_10nC_periodic.py",
-    # "expanding_beam/run_expanding_fft.py",
-    # "expanding_beam/run_expanding_envelope.py",
-    # "iota_lattice/run_iotalattice.py",
-    # "cyclotron/run_cyclotron.py",
-    # "dogleg/run_dogleg.py",
+    "chicane/run_chicane_csr.py",
+    "fodo_space_charge/run_fodo_envelope_sc.py",
+    "apochromatic/run_apochromatic.py",
+    # "kurth/run_kurth_10nC_periodic.py", - running into recursion issues
+    "expanding_beam/run_expanding_fft.py",
+    "expanding_beam/run_expanding_envelope.py",
+    "iota_lattice/run_iotalattice.py",
+    "cyclotron/run_cyclotron.py",
+    "dogleg/run_dogleg.py",
 }
 
 
@@ -64,7 +64,7 @@ def _get_example_content(file_name: str) -> dict:
         Retrieve the selected ImpactX example file and populate the UI with its values.
         """
 
-        impactx_directory = DashboardExamplesLoader.get_impactx_path()
+        impactx_directory = GeneralFunctions.get_impactx_root_dir()
         impactx_example_file_path = impactx_directory / "examples" / file_name
 
         file_content_as_str = impactx_example_file_path.read_text()
@@ -81,8 +81,9 @@ def load_impactx_examples() -> None:
 
         state.impactx_example_list.clear()
 
-        impactx_directory = DashboardExamplesLoader.get_impactx_path()
+        impactx_directory = GeneralFunctions.get_impactx_root_dir()
         impactx_examples_directory = impactx_directory / "examples"
+        print(" the examples directory is ", impactx_examples_directory)
 
         for path in impactx_examples_directory.glob("**/run*"):
             relative_path = path.relative_to(impactx_examples_directory)

src/python/impactx/dashboard/Toolbar/file_imports/python/helper.py

@@ -53,6 +53,10 @@ def parse_single_inputs(content: str) -> dict:
                 pattern_match = re.search(pattern.format(parameter_name), content)
                 if pattern_match:
                     value = ast.literal_eval(pattern_match.group(1))
+
+                    if parameter_name == "space_charge" and isinstance(value, bool):
+                        value = str(value).lower()
+
                     reference_dictionary[parameter_name] = value
                     break
 
@@ -61,8 +65,14 @@ def parse_single_inputs(content: str) -> dict:
             r"\bkin_energy_MeV\s*=\s*([^#\n]+)", content
         )
         if kin_energy_pattern_match:
-            kin_energy_value = kin_energy_pattern_match.group(1)
-            reference_dictionary["kin_energy_on_ui"] = kin_energy_value
+            kin_energy_value = kin_energy_pattern_match.group(1).strip()
+
+            try:
+                parsed_value = ast.literal_eval(kin_energy_value)
+            except (ValueError, SyntaxError):
+                parsed_value = reference_dictionary.get("kin_energy_MeV")
+
+            reference_dictionary["kin_energy_on_ui"] = parsed_value
 
         return reference_dictionary
 
@@ -71,11 +81,24 @@ def parse_list_inputs(content: str) -> dict:
         """
         Parses list-based simulation parameters from the simulation file content.
 
+        Some of the dashboard inputs are lists, such as n_cell = [16, 16, 16].
+        We have to look inside of the brackets to retrieve the individual values.
+
         :param content: The content of the ImpactX simulation file.
         """
         dictionary = {}
-        list_inputs = ["n_cell", "prob_relative"]
-        list_parsing = "{} = (\\[.*?\\])"
+        list_inputs = [
+            "n_cell",
+            "prob_relative",
+            "blocking_factor_x",
+            "blocking_factor_y",
+            "blocking_factor_z",
+        ]
+
+        # The below regex will capture everything inside of the brackets
+        # it assumes that the list will be placed in a single line (not multiple).
+        # Example: n_cell = [16, 16, 16] -->  [16, 16, 16]
+        list_parsing = r"{}\s*=\s*(\[[^\]]*\])"
 
         for input_name in list_inputs:
             match = re.search(list_parsing.format(input_name), content)
@@ -86,9 +109,13 @@ def parse_list_inputs(content: str) -> dict:
                     for i, dim in enumerate(["x", "y", "z"]):
                         dictionary[f"n_cell_{dim}"] = values[i]
 
-                if input_name == "prob_relative":
+                elif input_name == "prob_relative":
                     dictionary["prob_relative"] = values
 
+                elif input_name.startswith("blocking_factor_"):
+                    # Convert [16] -> 16
+                    dictionary[input_name] = values[0]
+
         return dictionary
 
     @staticmethod
@@ -134,33 +161,6 @@ def extract_parameters(distribution_type, parsing_pattern):
 
         return dictionary
 
-    @staticmethod
-    def parse_lattice_elements(content: str) -> dict:
-        """
-        Parses lattice elements from the simulation file content.
-
-        :param content: The content of the ImpactX simulation file.
-        """
-
-        dictionary = {"lattice_elements": []}
-        used_variables = set()
-
-        lattice_elements = re.findall(r"elements\.(\w+)\((.*?)\)", content)
-
-        for element_name, element_parameter in lattice_elements:
-            element = {"element": element_name, "parameters": {}}
-
-            parameter_pairs = re.findall(r"(\w+)=([^,\)]+)", element_parameter)
-            for parameter_name, parameter_value in parameter_pairs:
-                parameter_value_cleaned = parameter_value.strip("'\"")
-                element["parameters"][parameter_name] = parameter_value_cleaned
-                used_variables.add(parameter_value_cleaned)
-
-            dictionary["lattice_elements"].append(element)
-
-        dictionary["used_lattice_variables"] = used_variables
-        return dictionary
-
     @staticmethod
     def parse_variables(content: str, used_vars: set) -> dict:
         """