]> source.dussan.org Git - poi.git/commitdiff
New HPSF based TextExtractor for document metadata, org.apache.poi.hpsf.extractor...
authorNick Burch <nick@apache.org>
Tue, 12 Aug 2008 18:44:50 +0000 (18:44 +0000)
committerNick Burch <nick@apache.org>
Tue, 12 Aug 2008 18:44:50 +0000 (18:44 +0000)
git-svn-id: https://svn.apache.org/repos/asf/poi/trunk@685260 13f79535-47bb-0310-9956-ffa450edef68

src/documentation/content/xdocs/changes.xml
src/documentation/content/xdocs/status.xml
src/java/org/apache/poi/POITextExtractor.java
src/java/org/apache/poi/hpsf/CustomProperties.java
src/java/org/apache/poi/hpsf/DocumentSummaryInformation.java
src/java/org/apache/poi/hpsf/SpecialPropertySet.java
src/java/org/apache/poi/hpsf/SummaryInformation.java
src/java/org/apache/poi/hpsf/extractor/HPFSPropertiesExtractor.java [new file with mode: 0644]
src/testcases/org/apache/poi/hpsf/extractor/TestHPFSPropertiesExtractor.java [new file with mode: 0644]

index e290d73df8d2c2739835d6f27bfd121418521ccf..6a0cae2672ab48e11f0fd72189a942342ec1608e 100644 (file)
@@ -37,6 +37,7 @@
 
                <!-- Don't forget to update status.xml too! -->
         <release version="3.1.1-alpha1" date="2008-??-??">
+           <action dev="POI-DEVELOPERS" type="add">New HPSF based TextExtractor for document metadata, org.apache.poi.hpsf.extractor.HPFSPropertiesExtractor</action>
            <action dev="POI-DEVELOPERS" type="fix">Properly update the array of Slide's text runs in HSLF when new text shapes are added</action>
            <action dev="POI-DEVELOPERS" type="fix">45590 - Fix for Header/footer extraction for .ppt files saved in Office 2007</action>
            <action dev="POI-DEVELOPERS" type="fix">Big improvement in how HWPF handles unicode text, and more sanity checking of text ranges within HWPF</action>
index 9a6810660c6b4036d899e9286a8387ef2ea2898d..0146f55e7928d07093e1c97c750b78ef4736da97 100644 (file)
@@ -34,6 +34,7 @@
        <!-- Don't forget to update changes.xml too! -->
     <changes>
         <release version="3.1.1-alpha1" date="2008-??-??">
+           <action dev="POI-DEVELOPERS" type="add">New HPSF based TextExtractor for document metadata, org.apache.poi.hpsf.extractor.HPFSPropertiesExtractor</action>
            <action dev="POI-DEVELOPERS" type="fix">Properly update the array of Slide's text runs in HSLF when new text shapes are added</action>
            <action dev="POI-DEVELOPERS" type="fix">45590 - Fix for Header/footer extraction for .ppt files saved in Office 2007</action>
            <action dev="POI-DEVELOPERS" type="fix">Big improvement in how HWPF handles unicode text, and more sanity checking of text ranges within HWPF</action>
index 3ba71880ebbdeb7316cc3f0b74a0a609a1ac4a8f..a7ffd44197936dde97006093787d9588f4b6bea7 100644 (file)
@@ -37,6 +37,14 @@ public abstract class POITextExtractor {
        public POITextExtractor(POIDocument document) {
                this.document = document;
        }
+       /**
+        * Creates a new text extractor, using the same
+        *  document as another text extractor. Normally
+        *  only used by properties extractors.
+        */
+       protected POITextExtractor(POITextExtractor otherExtractor) {
+               this.document = otherExtractor.document;
+       }
        
        /**
         * Retrieves all the text from the document.
index 24b19e5d046e8cc434fb247d2b32ae034d736a9f..420fc2f9bb991ed47657864b2e130609aac0861f 100644 (file)
@@ -21,6 +21,7 @@ import java.util.Date;
 import java.util.HashMap;
 import java.util.Iterator;
 import java.util.Map;
+import java.util.Set;
 
 import org.apache.poi.hpsf.wellknown.PropertyIDMap;
 
@@ -293,8 +294,18 @@ public class CustomProperties extends HashMap
         final CustomProperty cp = new CustomProperty(p, name);
         return put(cp);
     }
-
+    
     /**
+     * Returns a set of all the names of our
+     *  custom properties
+     */
+    public Set keySet() {
+       return dictionaryNameToID.keySet();
+       }
+
+
+
+       /**
      * <p>Sets the codepage.</p>
      *
      * @param codepage the codepage
index b7a7c9ae6d3e46061383d141c3a46b6c8a088a26..62c6127ee4f543a152a485414d43aa5133f783e6 100644 (file)
@@ -45,6 +45,9 @@ public class DocumentSummaryInformation extends SpecialPropertySet
     public static final String DEFAULT_STREAM_NAME =
         "\005DocumentSummaryInformation";
 
+    public PropertyIDMap getPropertySetIDMap() {
+       return PropertyIDMap.getDocumentSummaryInformationProperties();
+    }
 
 
     /**
index 6a02bbc188218bf9e139f9c9bb83dcbe0a4b903c..f415bd5d128862d5757fb1cb3a49c7cc31ac0c84 100644 (file)
@@ -22,6 +22,7 @@ import java.io.InputStream;
 import java.io.OutputStream;
 import java.util.List;
 
+import org.apache.poi.hpsf.wellknown.PropertyIDMap;
 import org.apache.poi.poifs.filesystem.DirectoryEntry;
 
 /**
@@ -57,6 +58,11 @@ import org.apache.poi.poifs.filesystem.DirectoryEntry;
  */
 public abstract class SpecialPropertySet extends MutablePropertySet
 {
+       /**
+        * The id to name mapping of the properties
+        *  in this set.
+        */
+       public abstract PropertyIDMap getPropertySetIDMap();
 
     /**
      * <p>The "real" property set <code>SpecialPropertySet</code>
index 66d9ce0937e9bc06acfb4bf3ba5da222c0c75cfe..a143e2bad0f12c712c77a82bed2d987d75e368d8 100644 (file)
@@ -40,6 +40,9 @@ public class SummaryInformation extends SpecialPropertySet
      */
     public static final String DEFAULT_STREAM_NAME = "\005SummaryInformation";
 
+    public PropertyIDMap getPropertySetIDMap() {
+       return PropertyIDMap.getSummaryInformationProperties();
+    }
 
 
     /**
diff --git a/src/java/org/apache/poi/hpsf/extractor/HPFSPropertiesExtractor.java b/src/java/org/apache/poi/hpsf/extractor/HPFSPropertiesExtractor.java
new file mode 100644 (file)
index 0000000..c85f1bb
--- /dev/null
@@ -0,0 +1,144 @@
+/* ====================================================================
+   Licensed to the Apache Software Foundation (ASF) under one or more
+   contributor license agreements.  See the NOTICE file distributed with
+   this work for additional information regarding copyright ownership.
+   The ASF licenses this file to You under the Apache License, Version 2.0
+   (the "License"); you may not use this file except in compliance with
+   the License.  You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+==================================================================== */
+package org.apache.poi.hpsf.extractor;
+
+import java.io.IOException;
+import java.io.OutputStream;
+import java.util.Iterator;
+
+import org.apache.poi.POIDocument;
+import org.apache.poi.POITextExtractor;
+import org.apache.poi.hpsf.CustomProperties;
+import org.apache.poi.hpsf.DocumentSummaryInformation;
+import org.apache.poi.hpsf.Property;
+import org.apache.poi.hpsf.SpecialPropertySet;
+import org.apache.poi.hpsf.SummaryInformation;
+import org.apache.poi.hpsf.wellknown.PropertyIDMap;
+import org.apache.poi.poifs.filesystem.POIFSFileSystem;
+import org.apache.poi.util.LittleEndian;
+
+/**
+ * Extracts all of the HPSF properties, both
+ *  build in and custom, returning them in 
+ *  textual form.
+ */
+public class HPFSPropertiesExtractor extends POITextExtractor {
+       public HPFSPropertiesExtractor(POITextExtractor mainExtractor) {
+               super(mainExtractor);
+       }
+       public HPFSPropertiesExtractor(POIDocument doc) {
+               super(doc);
+       }
+       public HPFSPropertiesExtractor(POIFSFileSystem fs) {
+               super(new PropertiesOnlyDocument(fs));
+       }
+       
+       public String getDocumentSummaryInformationText() {
+               DocumentSummaryInformation dsi = document.getDocumentSummaryInformation();
+               StringBuffer text = new StringBuffer();
+
+               // Normal properties
+               text.append( getPropertiesText(dsi) );
+               
+               // Now custom ones
+               CustomProperties cps = dsi.getCustomProperties();
+               Iterator keys = cps.keySet().iterator();
+               while(keys.hasNext()) {
+                       String key = (String)keys.next();
+                       String val = getPropertyValueText( cps.get(key) );
+                       text.append(key + " = " + val + "\n");
+               }
+               
+               // All done
+               return text.toString();
+       }
+       public String getSummaryInformationText() {
+               SummaryInformation si = document.getSummaryInformation();
+               
+               // Just normal properties
+               return getPropertiesText(si);
+       }
+       
+       private static String getPropertiesText(SpecialPropertySet ps) {
+               if(ps == null) {
+                       // Not defined, oh well
+                       return "";
+               }
+               
+               StringBuffer text = new StringBuffer();
+               
+               PropertyIDMap idMap = ps.getPropertySetIDMap();
+               Property[] props = ps.getProperties();
+               for(int i=0; i<props.length; i++) {
+                       String type = Long.toString( props[i].getID() ); 
+                       Object typeObj = idMap.get(props[i].getID());
+                       if(typeObj != null) {
+                               type = typeObj.toString();
+                       }
+                       
+                       String val = getPropertyValueText( props[i].getValue() );
+                       text.append(type + " = " + val + "\n");
+               }
+               
+               return text.toString();
+       }
+       private static String getPropertyValueText(Object val) {
+               if(val == null) {
+                       return "(not set)";
+               }
+               if(val instanceof byte[]) {
+                       byte[] b = (byte[])val;
+                       if(b.length == 0) {
+                               return "";
+                       }
+                       if(b.length == 1) {
+                               return Byte.toString(b[0]);
+                       }
+                       if(b.length == 2) {
+                               return Integer.toString( LittleEndian.getUShort(b) );
+                       }
+                       if(b.length == 4) {
+                               return Long.toString( LittleEndian.getUInt(b) );
+                       }
+                       // Maybe it's a string? who knows!
+                       return new String(b);
+               }
+               return val.toString();
+       }
+
+       /**
+        * Return the text of all the properties defined in
+        *  the document.
+        */
+       public String getText() {
+               return getSummaryInformationText() + getDocumentSummaryInformationText();
+       }
+
+       /**
+        * So we can get at the properties of any 
+        *  random OLE2 document.
+        */
+       private static class PropertiesOnlyDocument extends POIDocument {
+               private PropertiesOnlyDocument(POIFSFileSystem fs) {
+                       super(fs);
+               }
+
+               public void write(OutputStream out) throws IOException {
+                       throw new IllegalStateException("Unable to write, only for properties!");
+               }
+       }
+}
diff --git a/src/testcases/org/apache/poi/hpsf/extractor/TestHPFSPropertiesExtractor.java b/src/testcases/org/apache/poi/hpsf/extractor/TestHPFSPropertiesExtractor.java
new file mode 100644 (file)
index 0000000..7d96787
--- /dev/null
@@ -0,0 +1,115 @@
+/* ====================================================================
+   Licensed to the Apache Software Foundation (ASF) under one or more
+   contributor license agreements.  See the NOTICE file distributed with
+   this work for additional information regarding copyright ownership.
+   The ASF licenses this file to You under the Apache License, Version 2.0
+   (the "License"); you may not use this file except in compliance with
+   the License.  You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+==================================================================== */
+package org.apache.poi.hpsf.extractor;
+
+import java.io.File;
+import java.io.FileInputStream;
+
+import org.apache.poi.hssf.extractor.ExcelExtractor;
+import org.apache.poi.hssf.usermodel.HSSFWorkbook;
+import org.apache.poi.poifs.filesystem.POIFSFileSystem;
+
+import junit.framework.TestCase;
+
+public class TestHPFSPropertiesExtractor extends TestCase {
+       private String dir;
+       
+    protected void setUp() throws Exception {
+       dir = System.getProperty("HPSF.testdata.path");
+       assertNotNull("HPSF.testdata.path not set", dir);
+       }
+    
+       public void testNormalProperties() throws Exception {
+               POIFSFileSystem fs = new POIFSFileSystem(
+                               new FileInputStream(new File(dir, "TestMickey.doc"))
+               );
+               HPFSPropertiesExtractor ext = new HPFSPropertiesExtractor(fs);
+               ext.getText();
+               
+               // Check each bit in turn
+               String sinfText = ext.getSummaryInformationText();
+               String dinfText = ext.getDocumentSummaryInformationText();
+               
+               assertTrue(sinfText.indexOf("TEMPLATE = Normal") > -1);
+               assertTrue(sinfText.indexOf("SUBJECT = sample subject") > -1);
+               assertTrue(dinfText.indexOf("MANAGER = sample manager") > -1);
+               assertTrue(dinfText.indexOf("COMPANY = sample company") > -1);
+               
+               // Now overall
+               String text = ext.getText();
+               assertTrue(text.indexOf("TEMPLATE = Normal") > -1);
+               assertTrue(text.indexOf("SUBJECT = sample subject") > -1);
+               assertTrue(text.indexOf("MANAGER = sample manager") > -1);
+               assertTrue(text.indexOf("COMPANY = sample company") > -1);
+       }
+       public void testNormalUnicodeProperties() throws Exception {
+               POIFSFileSystem fs = new POIFSFileSystem(
+                               new FileInputStream(new File(dir, "TestUnicode.xls"))
+               );
+               HPFSPropertiesExtractor ext = new HPFSPropertiesExtractor(fs);
+               ext.getText();
+               
+               // Check each bit in turn
+               String sinfText = ext.getSummaryInformationText();
+               String dinfText = ext.getDocumentSummaryInformationText();
+               
+               assertTrue(sinfText.indexOf("AUTHOR = marshall") > -1);
+               assertTrue(sinfText.indexOf("TITLE = Titel: \u00c4h") > -1);
+               assertTrue(dinfText.indexOf("COMPANY = Schreiner") > -1);
+               assertTrue(dinfText.indexOf("SCALE = false") > -1);
+               
+               // Now overall
+               String text = ext.getText();
+               assertTrue(text.indexOf("AUTHOR = marshall") > -1);
+               assertTrue(text.indexOf("TITLE = Titel: \u00c4h") > -1);
+               assertTrue(text.indexOf("COMPANY = Schreiner") > -1);
+               assertTrue(text.indexOf("SCALE = false") > -1);
+       }
+       public void testCustomProperties() throws Exception {
+               POIFSFileSystem fs = new POIFSFileSystem(
+                               new FileInputStream(new File(dir, "TestMickey.doc"))
+               );
+               HPFSPropertiesExtractor ext = new HPFSPropertiesExtractor(fs);
+
+               // Custom properties are part of the document info stream
+               String dinfText = ext.getDocumentSummaryInformationText();
+               assertTrue(dinfText.indexOf("Client = sample client") > -1);
+               assertTrue(dinfText.indexOf("Division = sample division") > -1);
+               
+               String text = ext.getText();
+               assertTrue(text.indexOf("Client = sample client") > -1);
+               assertTrue(text.indexOf("Division = sample division") > -1);
+       }
+    
+    public void testConstructors() throws Exception {
+               POIFSFileSystem fs = new POIFSFileSystem(
+                               new FileInputStream(new File(dir, "TestUnicode.xls"))
+               );
+               HSSFWorkbook wb = new HSSFWorkbook(fs);
+               ExcelExtractor excelExt = new ExcelExtractor(wb);
+               
+               String fsText = (new HPFSPropertiesExtractor(fs)).getText();
+               String hwText = (new HPFSPropertiesExtractor(wb)).getText();
+               String eeText = (new HPFSPropertiesExtractor(excelExt)).getText();
+               
+               assertEquals(fsText, hwText);
+               assertEquals(fsText, eeText);
+               
+               assertTrue(fsText.indexOf("AUTHOR = marshall") > -1);
+               assertTrue(fsText.indexOf("TITLE = Titel: \u00c4h") > -1);
+    }
+}