Use ProbeTable to generate NP probes information

.github/workflows/copy_probe_features.yml

@@ -0,0 +1,31 @@
+name: Sync probe_features from ProbeTable
+
+on:
+  schedule:
+    - cron: '0 0 * * 1' # Every Monday at 00:00 UTC
+  workflow_dispatch:
+
+jobs:
+  copy-file:
+    runs-on: ubuntu-latest
+
+    steps:
+    - name: Checkout current repository
+      uses: actions/checkout@v4
+
+    - name: Copy file from external repo
+      run: |
+        # Download the file directly
+        curl -o src/probeinterface/resources/probe_features.json \
+          https://raw.githubusercontent.com/billkarsh/ProbeTable/refs/heads/main/Tables/probe_features.json
+
+    - name: Commit changes if any
+      run: |
+        git config --local user.email "[email protected]"
+        git config --local user.name "GitHub Action"
+
+        git add src/probeinterface/resources/probe_features.json
+
+        # Only commit if there are changes
+        git diff --staged --quiet || git commit -m "Update probe_features from ProbeTable"
+        git push

.gitignore

@@ -4,7 +4,6 @@ tests/*.prb
 tests/*.json
 tests/*/*.prb
 
-*.json
 *.prb
 tests/*/*.json
 tests/*/*.tsv

resources/generate_neuropixels_library.py

@@ -1,10 +1,12 @@
 import shutil
 from pathlib import Path
 
+import json
+
 import numpy as np
 import matplotlib.pyplot as plt
 
-from probeinterface.neuropixels_tools import npx_descriptions, probe_part_number_to_probe_type, _make_npx_probe_from_description
+from probeinterface.neuropixels_tools import make_npx_description, _make_npx_probe_from_description
 from probeinterface.plotting import plot_probe
 from probeinterface import write_probeinterface
 
@@ -18,24 +20,24 @@ def generate_all_npx():
     # if not base_folder.exists():
     base_folder.mkdir(exist_ok=True)
 
+    probe_features_filepath = Path(__file__).absolute().parent / Path("../src/probeinterface/resources/probe_features.json")
+    probe_features = json.load(open(probe_features_filepath, "r"))
+    probe_part_numbers = probe_features['neuropixels_probes'].keys()
+
 
-    for probe_number, probe_type in probe_part_number_to_probe_type.items():
+    for probe_number in probe_part_numbers:
 
         if probe_number is None:
             continue
 
-        if probe_number == "1110":
+        if probe_number == "NP1110":
             # the formula by the imrow table is wrong and more complicated
             continue
 
         probe_folder = base_folder / probe_number
         probe_folder.mkdir(exist_ok=True)
 
-        print(probe_number, probe_type)
-
-        probe_description = npx_descriptions[probe_type]
-
-
+        probe_description = make_npx_description(probe_number)
 
         num_shank = probe_description["shank_number"]
         contact_per_shank = probe_description["ncols_per_shank"] * probe_description["nrows_per_shank"]
@@ -99,8 +101,5 @@ def generate_all_npx():
 
 
 
-
-
-
 if __name__ == "__main__":
     generate_all_npx()