Merge pull request #235 from MIT-LCP/signame

Adds original signame function

Author: Lucas-Mc

wfdb/__init__.py

@@ -1,5 +1,5 @@
 from .io.record import (Record, MultiRecord, rdheader, rdrecord, rdsamp,
-                        wrsamp, dl_database, edf2mit, sampfreq)
+                        wrsamp, dl_database, edf2mit, sampfreq, signame)
 from .io.annotation import (Annotation, rdann, wrann, show_ann_labels,
                             show_ann_classes)
 from .io.download import get_dbs, get_record_list, dl_files, set_db_index_url

wfdb/io/__init__.py

@@ -1,5 +1,5 @@
 from .record import (Record, MultiRecord, rdheader, rdrecord, rdsamp, wrsamp,
-                     dl_database, edf2mit, sampfreq, SIGNAL_CLASSES)
+                     dl_database, edf2mit, sampfreq, signame, SIGNAL_CLASSES)
 from ._signal import est_res, wr_dat_file
 from .annotation import (Annotation, rdann, wrann, show_ann_labels,
                          show_ann_classes)

wfdb/io/record.py

@@ -2061,6 +2061,61 @@ def sampfreq(record_name, pn_dir=None):
         print('{}\t{}'.format(sig,samp))
 
 
+def signame(record_name, pn_dir=None, sig_nums=[]):
+    """
+    Read a WFDB record file and return the signal names.
+
+    Parameters
+    ----------
+    record_name : str
+        The name of the WFDB record to be read, without any file
+        extensions. If the argument contains any path delimiter
+        characters, the argument will be interpreted as PATH/BASE_RECORD.
+        Both relative and absolute paths are accepted. If the `pn_dir`
+        parameter is set, this parameter should contain just the base
+        record name, and the files fill be searched for remotely.
+        Otherwise, the data files will be searched for in the local path.
+    pn_dir : str, optional
+        Option used to stream data from Physionet. The Physionet
+        database directory from which to find the required record files.
+        eg. For record '100' in 'http://physionet.org/content/mitdb'
+        pn_dir='mitdb'.
+    sig_nums : list, optional
+        A list of the signal numbers to be outputted.
+
+    Returns
+    -------
+    N/A
+
+    Examples
+    --------
+    >>> wfdb.signame('sample-data/test01_00s')
+    >>> ECG 1
+    >>> ECG 2
+    >>> ECG 3
+    >>> ECG 4
+
+    >>> wfdb.signame('sample-data/test01_00s', sig_nums=[1,3])
+    >>> ECG 2
+    >>> ECG 4
+
+    """
+    if (pn_dir is not None) and ('.' not in pn_dir):
+        dir_list = pn_dir.split(os.sep)
+        pn_dir = posixpath.join(dir_list[0], get_version(dir_list[0]),
+                                 *dir_list[1:])
+
+    record = rdheader(record_name, pn_dir=pn_dir)
+    if len(sig_nums) > 0:
+        for n in sig_nums:
+            try:
+                print(record.sig_name[n])
+            except IndexError:
+                raise Exception('sig_nums value {} out of range'.format(n))
+    else:
+        print(*record.sig_name, sep='\n')
+
+
 def _get_wanted_channels(wanted_sig_names, record_sig_names, pad=False):
     """
     Given some wanted signal names, and the signal names contained in a