Prechádzať zdrojové kódy

gin commit from Ivory

Modified files: 2
Achilleas Koutsou 5 rokov pred
rodič
commit
c6c9029606
2 zmenil súbory, kde vykonal 14 pridanie a 7 odobranie
  1. 1 1
      EEG for MNE/LED_28_06_2012_104.nix
  2. 13 6
      readrawnix.py

+ 1 - 1
EEG for MNE/LED_28_06_2012_104.nix

@@ -1 +1 @@
-/annex/objects/MD5-s39611069--3b422c59c0eb14bd7c1af5364718ada7
+/annex/objects/MD5-s39611069--489fad083fdfb24f6cf68f21bbbc6948

+ 13 - 6
readrawnix.py

@@ -63,12 +63,7 @@ def merge_data_arrays(arrays):
     return np.array(rows)
 
 
-def main():
-    if len(sys.argv) < 2:
-        print("Please provide either a NIX filename as the first argument")
-        sys.exit(1)
-
-    nixfilename = sys.argv[1]
+def import_nix(nixfilename):
     nixfile = nix.File(nixfilename, mode=nix.FileMode.ReadOnly)
 
     # root, ext = os.path.splitext(nixfilename)
@@ -92,10 +87,22 @@ def main():
     else:
         nixrawdata = datagroup.data_arrays[0][:]
 
+    # Create MNE RawArray
     mnerawdata = mne.io.RawArray(nixrawdata, info)
 
     nixfile.close()
 
+    return mnerawdata
+
+
+def main():
+    if len(sys.argv) < 2:
+        print("Please provide either a NIX filename as the first argument")
+        sys.exit(1)
+
+    nixfilename = sys.argv[1]
+    import_nix(nixfilename)
+
 
 if __name__ == "__main__":
     main()