Extracting formatted annotations

Author: ishiikurisu

java/br/unb/biologiaanimal/edf/EDF.java

@@ -218,7 +218,14 @@ public void toCsv(String filePath)
     }
 
     // TODO Read annotations
+    /**
+     * Gets a list of the annotations on the EDF file, lest there are
+     * annotations.
+     * @return An array containing one annotation for each entry
+     * @throws NoSuchFieldException
+     */
     public String[] getAnnotations()
+    throws NoSuchFieldException
     {
         return reader.getAnnotations();
     }

java/br/unb/biologiaanimal/edf/EDFReader.java

@@ -284,8 +284,50 @@ private long[] getLimits(String param)
     }
 
     // TODO Get annotations
+    /**
+     * Gets a list of the annotations on the EDF file, lest there are
+     * annotations.
+     * @return An array containing one annotation for each entry
+     * @throws NoSuchFieldException
+     */
     public String[] getAnnotations()
+    throws NoSuchFieldException
+    {
+        int annotationsChannel = getAnnotationsChannelIndex();
+
+        if (annotationsChannel < 0) {
+            throw new NoSuchFieldException();
+        }
+
+        return getRawNotes().split("\n");
+    }
+
+    private String getRawNotes()
     {
-        throw new UnsupportedOperationException();
+        byte[] raw = this.getRecord("EDF Annotations");
+        String outlet = "";
+        boolean inside = false;
+
+        for (int i = 0; i < raw.length; ++i)
+        {
+            if (inside) {
+                if (raw[i] == 0) {
+                    inside = false;
+                    outlet += "\n";
+                }
+                else {
+                    byte it = raw[i];
+                    boolean fact = (it == (byte) 20) || (it == (byte) 21);
+                    raw[i] = (fact)? (byte) ' ' : it;
+                    outlet += (char) raw[i];
+                }
+            }
+            else if (raw[i] == '+' || raw[i] == '-') {
+                inside = true;
+                outlet += (char) raw[i];
+            }
+        }
+
+        return outlet;
     }
 }

target/edf.jar

@@ -28,18 +28,6 @@ public static void main(String[] args) {
             allLabels += (i+1) + ". " + labels[i] + "\n";
         }
         System.out.println(allLabels);
-        System.out.println("--- # Let's write something");
-        System.out.println("-- " + edf.sayHi());
-        System.out.println("-- Writing to ASCII format");
-        System.out.println("-- + Single channel");
-        try { edf.toSingleChannelAscii("data\\edf\\HCT\\ECG.ascii", "ECG"); }
-        catch (Exception any) { System.out.println(any); }
-        System.out.println("-- + All channels");
-        try { edf.toAscii("data\\edf\\HCT-4-23.ascii"); }
-        catch (Exception any) { System.out.println(any); }
-        System.out.println("-- + To CSV");
-        try { edf.toCsv("data\\edf\\HCT-4-23.java.ascii"); }
-        catch (Exception any) { System.out.println(any); }
         System.out.println("-- + Getting annotations");
         try
         {
@@ -54,6 +42,19 @@ public static void main(String[] args) {
             System.out.println(any);
         }
 
+        System.out.println("--- # Let's write something");
+        System.out.println("-- " + edf.sayHi());
+        System.out.println("-- Writing to ASCII format");
+        System.out.println("-- + Single channel");
+        try { edf.toSingleChannelAscii("data\\edf\\HCT\\ECG.ascii", "ECG"); }
+        catch (Exception any) { System.out.println(any); }
+        System.out.println("-- + All channels");
+        try { edf.toAscii("data\\edf\\HCT-4-23.ascii"); }
+        catch (Exception any) { System.out.println(any); }
+        System.out.println("-- + To CSV");
+        try { edf.toCsv("data\\edf\\HCT-4-23.java.ascii"); }
+        catch (Exception any) { System.out.println(any); }
+
         System.out.println("...");
     }
 }