DataConversionToNIX/mnetonix/edftonix.py

import os
import sys
import mne
import mnetonix as m2n


def main():
    if len(sys.argv) < 2:
        print("Please provide an EDF filename as argument")
        sys.exit(1)

    efname = sys.argv[1]
    locname = None
    if len(sys.argv) > 2:
        locname = sys.argv[2]
        locname = os.path.abspath(locname)
    root, ext = os.path.splitext(efname)
    nfname = root + os.path.extsep + "nix"
    print(f"Converting '{efname}' to NIX")
    # stim_channel=False marks the last channel as "EDF Annotations"
    ef = mne.io.read_raw_edf(efname, montage=locname,
                             preload=True, stim_channel=False)
    m2n.write_raw_mne(nfname, ef)
    ef.close()


if __name__ == "__main__":
    main()