Skip to content

Commit

Permalink
DOC: Automatically document settings
Browse files Browse the repository at this point in the history
  • Loading branch information
larsoner committed Feb 26, 2024
1 parent 61b3553 commit a82a34e
Show file tree
Hide file tree
Showing 13 changed files with 562 additions and 254 deletions.
3 changes: 3 additions & 0 deletions docs/build-docs.sh
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,9 @@ python $STEP_DIR/source/examples/gen_examples.py
echo "Generating pipeline table …"
python $STEP_DIR/source/features/gen_steps.py

echo "Generating config docs …"
python $STEP_DIR/source/settings/gen_settings.py

echo "Building the documentation …"
cd $STEP_DIR
mkdocs build
19 changes: 19 additions & 0 deletions docs/source/settings/execution.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@

::: mne_bids_pipeline._config
options:
members:
- n_jobs
- parallel_backend
- dask_open_dashboard
- dask_temp_dir
- dask_worker_memory_limit
- random_state
- shortest_event
- log_level
- mne_log_level
- on_error
- memory_location
- memory_subdir
- memory_file_method
- memory_verbose
- config_validation
140 changes: 140 additions & 0 deletions docs/source/settings/gen_settings.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,140 @@
"""Generate settings .md files."""

import re
from pathlib import Path

from tqdm import tqdm

import mne_bids_pipeline._config

root_file = Path(mne_bids_pipeline._config.__file__)
settings_dir = Path(__file__).parent

# Mapping between first two lower-case words in the section name and the desired
# file or folder name
section_to_file = { # .md will be added to the files
# root file
"general settings": "general",
# folder
"preprocessing": "preprocessing",
"break detection": "breaks",
"bad channel": "autobads",
"maxwell filter": "maxfilter",
"filtering": "filter",
"resampling": "resample",
"epoching": "epochs",
"filtering &": None, # just a header
"artifact removal": None,
"stimulation artifact": "stim_artifact",
"ssp, ica,": "ssp_ica",
"amplitude-based artifact": "artifacts",
# folder
"sensor-level analysis": "sensor",
"condition contrasts": "contrasts",
"decoding /": "mvpa",
"time-frequency analysis": "time_freq",
"group-level analysis": "group_level",
# folder
"source-level analysis": "source",
"bem surface": "bem",
"source space": "forward",
"inverse solution": "inverse",
# folder
"reports": "reports",
"report generation": "report_generation",
# root file
"execution": "execution",
}

option_header = """
::: mne_bids_pipeline._config
options:
members:"""
prefix = """\
- """

# We cannot use ast for this because it doesn't preserve comments. We could use
# something like redbaron, but our code is hopefully simple enough!
assign_re = re.compile(
# Line starts with annotation syntax (name captured by the first group).
r"^(\w+): "
# Then the annotation can be ...
"("
# ... a standard assignment ...
".+ = .+"
# ... or ...
"|"
# ... the start of a multiline type annotation like "a: Union["
r"(Union|Optional|Literal)\["
# To the end of the line.
")$",
re.MULTILINE,
)


def main():
print(f"Parsing {root_file} to generate settings .md files.")
# max file-level depth is 2 even though we have 3 subsection levels
levels = [None, None]
current_path, current_lines = None, list()
text = root_file.read_text("utf-8")
lines = text.splitlines()
lines += ["# #"] # add a dummy line to trigger the last write
in_header = False
for li, line in enumerate(tqdm(lines)):
line = line.rstrip()
if line.startswith("# #"): # a new (sub)section / file
this_def = line[2:]
this_level = this_def.split()[0]
assert this_level.count("#") == len(this_level), this_level
this_level = this_level.count("#") - 1
if this_level == 2:
# flatten preprocessing/filtering/filter to preprocessing/filter
# for example
this_level = 1
assert this_level in (0, 1), (this_level, this_def)
this_def = this_def[this_level + 2 :]
levels[this_level] = this_def
# Write current lines and reset
if len(current_lines) > 1: # more than just the header
assert current_path is not None, levels
if current_lines[0] == "": # this happens with tags
current_lines = current_lines[1:]
current_path.write_text("\n".join(current_lines + [""]), "utf-8")
if this_level == 0:
this_root = settings_dir
else:
this_root = settings_dir / f"{section_to_file[levels[0].lower()]}"
this_root.mkdir(exist_ok=True)
key = " ".join(this_def.split()[:2]).lower()
if key == "":
assert li == len(lines) - 1, (li, line)
continue # our dummy line
fname = section_to_file[key]
if fname is None:
current_path = None
else:
current_path = this_root / f"{fname}.md"
current_lines = []
in_header = True
continue

if in_header:
if line == "":
in_header = False
current_lines.append(option_header)
else:
assert line == "#" or line.startswith("# "), (li, line) # a comment
current_lines.append(line[2:])
continue

# Could be an option
match = assign_re.match(line)
if match is not None:
name, typ, desc = match.groups()
current_lines.append(f"{prefix}{name}")
continue


if __name__ == "__main__":
main()
18 changes: 2 additions & 16 deletions docs/source/settings/general.md
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@

::: mne_bids_pipeline._config
options:
members:
Expand Down Expand Up @@ -27,22 +28,7 @@
- eeg_reference
- eeg_template_montage
- drop_channels
- analyze_channels
- reader_extra_params
- read_raw_bids_verbose
- analyze_channels
- plot_psd_for_runs
- n_jobs
- parallel_backend
- dask_open_dashboard
- dask_temp_dir
- dask_worker_memory_limit
- random_state
- shortest_event
- memory_location
- memory_subdir
- memory_file_method
- memory_verbose
- config_validation
- log_level
- mne_log_level
- on_error
1 change: 1 addition & 0 deletions docs/source/settings/preprocessing/artifacts.md
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ tags:
You can do a quick average of blink data and check what the amplitude looks
like.


::: mne_bids_pipeline._config
options:
members:
Expand Down
8 changes: 4 additions & 4 deletions docs/source/settings/preprocessing/epochs.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,14 +13,14 @@ tags:
- rename_events
- on_rename_missing_events
- event_repeated
- conditions
- epochs_tmin
- epochs_tmax
- baseline
- epochs_metadata_tmin
- epochs_metadata_tmax
- epochs_metadata_keep_first
- epochs_metadata_keep_last
- epochs_metadata_query
- conditions
- epochs_tmin
- epochs_tmax
- rest_epochs_duration
- rest_epochs_overlap
- baseline
2 changes: 1 addition & 1 deletion docs/source/settings/preprocessing/filter.md
Original file line number Diff line number Diff line change
Expand Up @@ -30,8 +30,8 @@ of tips! 😇
members:
- l_freq
- h_freq
- notch_freq
- l_trans_bandwidth
- h_trans_bandwidth
- notch_freq
- notch_trans_bandwidth
- notch_widths
4 changes: 2 additions & 2 deletions docs/source/settings/preprocessing/ssp_ica.md
Original file line number Diff line number Diff line change
Expand Up @@ -17,11 +17,11 @@ tags:
- min_eog_epochs
- n_proj_eog
- n_proj_ecg
- ssp_meg
- ecg_proj_from_average
- eog_proj_from_average
- ssp_reject_eog
- ssp_meg
- ssp_reject_ecg
- ssp_reject_eog
- ssp_ecg_channel
- ica_reject
- ica_algorithm
Expand Down
2 changes: 1 addition & 1 deletion docs/source/settings/preprocessing/stim_artifact.md
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
---
----
tags:
- preprocessing
- artifact-removal
Expand Down
6 changes: 3 additions & 3 deletions docs/source/settings/sensor/mvpa.md
Original file line number Diff line number Diff line change
Expand Up @@ -18,10 +18,10 @@ tags:
- decoding_n_splits
- decoding_time_generalization
- decoding_time_generalization_decim
- decoding_csp
- decoding_csp_times
- decoding_csp_freqs
- n_boot
- cluster_forming_t_threshold
- cluster_n_permutations
- cluster_permutation_p_threshold
- decoding_csp
- decoding_csp_times
- decoding_csp_freqs
1 change: 0 additions & 1 deletion docs/source/settings/source/forward.md
Original file line number Diff line number Diff line change
Expand Up @@ -11,4 +11,3 @@ tags:
- mri_landmarks_kind
- spacing
- mindist
- source_info_path_update
Loading

0 comments on commit a82a34e

Please sign in to comment.