uh-oh

core/src/processing/data/TableODS.java

@@ -0,0 +1,255 @@
+package processing.data;
+
+import java.io.*;
+import java.util.zip.*;
+
+import javax.xml.parsers.ParserConfigurationException;
+
+import org.xml.sax.SAXException;
+
+import processing.core.*;
+
+
+public class TableODS extends Table {
+
+  public TableODS(File odsFile) {
+    this(getContentXML(odsFile), null, false);
+  }
+
+
+  public TableODS(File odsFile, boolean actual) {
+    this(getContentXML(odsFile), null, actual);
+  }
+
+
+  public TableODS(PApplet parent, String filename) {
+    this(getContentXML(parent.createInput(filename)), null, false);
+  }
+
+
+  public TableODS(PApplet parent, String filename, boolean actual) {
+    this(getContentXML(parent.createInput(filename)), null, actual);
+  }
+
+
+  public TableODS(PApplet parent, String filename, String worksheet, boolean actual) {
+    this(getContentXML(parent.createInput(filename)), worksheet, actual);
+  }
+
+
+  /**
+   * Parse spreadsheet content.
+   * @param input InputStream of the content.xml file inside the .ods
+   */
+  protected TableODS(InputStream input, String worksheet, boolean actual) {
+    try {
+      InputStreamReader isr = new InputStreamReader(input, "UTF-8");
+      BufferedReader reader = new BufferedReader(isr);
+      read(reader, worksheet, actual);
+
+    } catch (UnsupportedEncodingException uee) {
+      uee.printStackTrace();
+    } catch (IOException e) {
+      e.printStackTrace();
+    } catch (ParserConfigurationException e) {
+      e.printStackTrace();
+    } catch (SAXException e) {
+      e.printStackTrace();
+    }
+  }
+
+
+  protected void read(BufferedReader reader, String worksheet, boolean actual) throws IOException, ParserConfigurationException, SAXException {
+    XML xml = new XML(reader);
+//    XML x = new XML(reader);
+//    PApplet.saveStrings(new File("/Users/fry/Desktop/namespacefix.xml"), new String[] { xml.toString() });
+//    PApplet.saveStrings(new File("/Users/fry/Desktop/newparser.xml"), new String[] { x.toString() });
+
+    // table files will have multiple sheets.. argh
+    // <table:table table:name="Sheet1" table:style-name="ta1" table:print="false">
+    // <table:table table:name="Sheet2" table:style-name="ta1" table:print="false">
+    // <table:table table:name="Sheet3" table:style-name="ta1" table:print="false">
+
+    XML[] sheets =
+      xml.getChildren("office:body/office:spreadsheet/table:table");
+      //xml.getChildren("office:body/office:spreadsheet/table:table/table");
+//    System.out.println("found " + sheets.length + " sheets.");
+
+    for (XML sheet : sheets) {
+//      System.out.println(sheet.getAttribute("table:name"));
+      if (worksheet == null || worksheet.equals(sheet.getString("table:name"))) {
+        parseSheet(sheet, actual);
+      }
+    }
+  }
+
+
+  protected void parseSheet(XML sheet, boolean actual) {
+    XML[] rows = sheet.getChildren("table:table-row");
+    //xml.getChildren("office:body/office:spreadsheet/table:table/table:table-row");
+
+    int rowIndex = 0;
+    for (XML row : rows) {
+      int rowRepeat = row.getInt("table:number-rows-repeated", 1);
+//      if (rowRepeat != 1) {
+//          System.out.println(rowRepeat + " " + rowCount + " " + (rowCount + rowRepeat));
+//      }
+      boolean rowNotNull = false;
+      XML[] cells = row.getChildren();
+      int columnIndex = 0;
+
+      for (XML cell : cells) {
+        int cellRepeat = cell.getInt("table:number-columns-repeated", 1);
+
+//        <table:table-cell table:formula="of:=SUM([.E7:.E8])" office:value-type="float" office:value="4150">
+//        <text:p>4150.00</text:p>
+//        </table:table-cell>
+
+        String cellData = actual ? cell.getString("office:value") : null;
+
+        // if there's an office:value in the cell, just roll with that
+        if (cellData == null) {
+          int cellKids = cell.getChildCount();
+          if (cellKids != 0) {
+            XML[] paragraphElements = cell.getChildren("text:p");
+            if (paragraphElements.length != 1) {
+              for (XML el : paragraphElements) {
+                System.err.println(el.toString());
+              }
+              throw new RuntimeException("found more than one text:p element");
+            }
+            XML textp = paragraphElements[0];
+            String textpContent = textp.getContent();
+            // if there are sub-elements, the content shows up as a child element
+            // (for which getName() returns null.. which seems wrong)
+            if (textpContent != null) {
+              cellData = textpContent;  // nothing fancy, the text is in the text:p element
+            } else {
+              XML[] textpKids = textp.getChildren();
+              StringBuffer cellBuffer = new StringBuffer();
+              for (XML kid : textpKids) {
+                String kidName = kid.getName();
+                if (kidName == null) {
+                  appendNotNull(kid, cellBuffer);
+
+                } else if (kidName.equals("text:s")) {
+                  int spaceCount = kid.getInt("text:c", 1);
+                  for (int space = 0; space < spaceCount; space++) {
+                    cellBuffer.append(' ');
+                  }
+                } else if (kidName.equals("text:span")) {
+                  appendNotNull(kid, cellBuffer);
+
+                } else if (kidName.equals("text:a")) {
+                  // <text:a xlink:href="http://blah.com/">blah.com</text:a>
+                  if (actual) {
+                    cellBuffer.append(kid.getString("xlink:href"));
+                  } else {
+                    appendNotNull(kid, cellBuffer);
+                  }
+
+                } else {
+                  appendNotNull(kid, cellBuffer);
+                  System.err.println(getClass().getName() + ": don't understand: " + kid);
+                  //throw new RuntimeException("I'm not used to this.");
+                }
+              }
+              cellData = cellBuffer.toString();
+            }
+            //setString(rowIndex, columnIndex, c); //text[0].getContent());
+            //columnIndex++;
+          }
+        }
+        for (int r = 0; r < cellRepeat; r++) {
+          if (cellData != null) {
+            //System.out.println("setting " + rowIndex + "," + columnIndex + " to " + cellData);
+            setString(rowIndex, columnIndex, cellData);
+          }
+          columnIndex++;
+          if (cellData != null) {
+//            if (columnIndex > columnMax) {
+//              columnMax = columnIndex;
+//            }
+            rowNotNull = true;
+          }
+        }
+      }
+      if (rowNotNull && rowRepeat > 1) {
+        String[] rowStrings = getStringRow(rowIndex);
+        for (int r = 1; r < rowRepeat; r++) {
+          addRow(rowStrings);
+        }
+      }
+      rowIndex += rowRepeat;
+//      if (rowNotNull) {
+//        rowMax = rowIndex;
+//      }
+    }
+//    if (rowMax != getRowCount()) {
+//      System.out.println("removing empty rows: " + rowMax + " instead of " + getRowCount());
+//      setRowCount(rowMax);
+//    }
+//    if (columnMax != getColumnCount()) {
+//      System.out.println("removing empty columns: " + columnMax + " instead of " + getColumnCount());
+//      setColumnCount(columnMax);
+//    }
+  }
+
+
+  protected void appendNotNull(XML kid, StringBuffer buffer) {
+    String content = kid.getContent();
+    if (content != null) {
+      buffer.append(content);
+    }
+  }
+
+
+//  static public PNode getContentXML(File file) {
+//    return new PNode(getContentReader(file));
+//  }
+
+
+//  static public BufferedReader getContentReader(File file) {
+//    return PApplet.createReader(getContentInput(file));
+//  }
+
+
+  /**
+   * Read zip file from a local file, and return the InputStream for content.xml.
+   */
+  static protected InputStream getContentXML(File file) {
+    try {
+      ZipFile zip = new ZipFile(file);
+      ZipEntry entry = zip.getEntry("content.xml");
+      return zip.getInputStream(entry);
+
+    } catch (IOException e) {
+      e.printStackTrace();
+    }
+    return null;
+  }
+
+
+  /**
+   * Read zip file from an InputStream, and return the InputStream for content.xml.
+   */
+  static protected InputStream getContentXML(InputStream input) {
+    ZipInputStream zis = new ZipInputStream(input);
+    ZipEntry entry = null;
+    try {
+      while ((entry = zis.getNextEntry()) != null) {
+        if (entry.getName().equals("content.xml")) {
+          return zis;
+//          InputStreamReader isr = new InputStreamReader(zis);
+//          BufferedReader reader = new BufferedReader(isr);
+//          read(reader, actual);
+//          break;
+          //return entry.getInputStream();
+        }
+      }
+    } catch (IOException e) {
+      e.printStackTrace();
+    }
+    return null;
+  }
+}