ENH: Add ARTEMIS123 helmet (#5272)

doc/whats_new.rst

@@ -25,6 +25,8 @@ Changelog
 
 - Add :func:`mne.channels.make_1020_channel_selections` to group 10/20-named EEG channels by location, by `Jona Sassenhagen`_
 
+- Add helmet for Artemis123 for :func:`mne.viz.plot_alignment` by `Eric Larson`_
+
 Bug
 ~~~
 

examples/visualization/plot_meg_sensors.py

@@ -3,8 +3,7 @@
 Plotting sensor layouts of MEG systems
 ======================================
 
-In this example, sensor layouts of different MEG systems
-are shown.
+In this example, sensor layouts of different MEG systems are shown.
 """
 # Author: Eric Larson <larson.eric.d@gmail.com>
 #
@@ -24,27 +23,28 @@ print(__doc__)
 
 bti_path = op.abspath(op.dirname(mne.__file__)) + '/io/bti/tests/data/'
 kit_path = op.abspath(op.dirname(mne.__file__)) + '/io/kit/tests/data/'
-raws = dict(
-    Neuromag=read_raw_fif(sample.data_path() +
-                          '/MEG/sample/sample_audvis_raw.fif'),
-    CTF_275=read_raw_ctf(spm_face.data_path() +
-                         '/MEG/spm/SPM_CTF_MEG_example_faces1_3D.ds'),
-    Magnes_3600wh=read_raw_bti(op.join(bti_path, 'test_pdf_linux'),
-                               op.join(bti_path, 'test_config_linux'),
-                               op.join(bti_path, 'test_hs_linux')),
-    KIT=read_raw_kit(op.join(kit_path, 'test.sqd')),
-    Artemis123=read_raw_artemis123(op.join(
+raws = {
+    'Neuromag': read_raw_fif(sample.data_path() +
+                             '/MEG/sample/sample_audvis_raw.fif'),
+    'CTF 275': read_raw_ctf(spm_face.data_path() +
+                            '/MEG/spm/SPM_CTF_MEG_example_faces1_3D.ds'),
+    'Magnes 3600wh': read_raw_bti(op.join(bti_path, 'test_pdf_linux'),
+                                  op.join(bti_path, 'test_config_linux'),
+                                  op.join(bti_path, 'test_hs_linux')),
+    'KIT': read_raw_kit(op.join(kit_path, 'test.sqd')),
+    'Artemis123': read_raw_artemis123(op.join(
         testing.data_path(), 'ARTEMIS123',
-        'Artemis_Data_2017-04-14-10h-38m-59s_Phantom_1k_HPI_1s.bin'))
-)
+        'Artemis_Data_2017-04-14-10h-38m-59s_Phantom_1k_HPI_1s.bin')),
+}
 
-for system, raw in raws.items():
+for system, raw in sorted(raws.items()):
     meg = ['helmet', 'sensors']
     # We don't have coil definitions for KIT refs, so exclude them
     if system != 'KIT':
         meg.append('ref')
     fig = plot_alignment(raw.info, trans=None, dig=False, eeg=False,
-                         surfaces=[], meg=meg, coord_frame='meg')
+                         surfaces=[], meg=meg, coord_frame='meg',
+                         verbose=True)
     text = mlab.title(system)
     text.x_position = 0.5
     text.y_position = 0.95
@@ -52,4 +52,5 @@ for system, raw in raws.items():
     text.property.justification = 'center'
     text.actor.text_scale_mode = 'none'
     text.property.bold = True
+    text.property.font_size = 40
     mlab.draw(fig)

mne/channels/channels.py

@@ -51,6 +51,9 @@ def _get_meg_system(info):
             elif coil_type == FIFF.FIFFV_COIL_BABY_GRAD:
                 system = 'BabySQUID'
                 break
+            elif coil_type == FIFF.FIFFV_COIL_ARTEMIS123_GRAD:
+                system = 'ARTEMIS123'
+                break
     return system