123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181 |
- /* ====================================================================
- 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.hssf.eventusermodel;
-
- import java.io.InputStream;
- import java.io.IOException;
- import java.util.Set;
-
- import org.apache.poi.hssf.record.*;
- import org.apache.poi.poifs.filesystem.DirectoryNode;
- import org.apache.poi.poifs.filesystem.POIFSFileSystem;
- import static org.apache.poi.hssf.model.InternalWorkbook.WORKBOOK_DIR_ENTRY_NAMES;
-
- /**
- * Low level event based HSSF reader. Pass either a DocumentInputStream to
- * process events along with a request object or pass a POIFS POIFSFileSystem to
- * processWorkbookEvents along with a request.
- *
- * This will cause your file to be processed a record at a time. Each record with
- * a static id matching one that you have registered in your HSSFRequest will be passed
- * to your associated HSSFListener.
- */
- public class HSSFEventFactory {
- /** Creates a new instance of HSSFEventFactory */
- public HSSFEventFactory() {
- // no instance fields
- }
-
- /**
- * Processes a file into essentially record events.
- *
- * @param req an Instance of HSSFRequest which has your registered listeners
- * @param fs a POIFS filesystem containing your workbook
- *
- * @throws IOException if the workbook contained errors
- */
- public void processWorkbookEvents(HSSFRequest req, POIFSFileSystem fs) throws IOException {
- processWorkbookEvents(req, fs.getRoot());
- }
-
- /**
- * Processes a file into essentially record events.
- *
- * @param req an Instance of HSSFRequest which has your registered listeners
- * @param dir a DirectoryNode containing your workbook
- *
- * @throws IOException if the workbook contained errors
- */
- public void processWorkbookEvents(HSSFRequest req, DirectoryNode dir) throws IOException {
- // some old documents have "WORKBOOK" or "BOOK"
- String name = null;
- Set<String> entryNames = dir.getEntryNames();
- for (String potentialName : WORKBOOK_DIR_ENTRY_NAMES) {
- if (entryNames.contains(potentialName)) {
- name = potentialName;
- break;
- }
- }
- // If in doubt, go for the default
- if (name == null) {
- name = WORKBOOK_DIR_ENTRY_NAMES[0];
- }
-
- try (InputStream in = dir.createDocumentInputStream(name)) {
- processEvents(req, in);
- }
- }
-
- /**
- * Processes a file into essentially record events.
- *
- * @param req an Instance of HSSFRequest which has your registered listeners
- * @param fs a POIFS filesystem containing your workbook
- * @return numeric user-specified result code.
- *
- * @throws HSSFUserException if the processing should be aborted
- * @throws IOException if the workbook contained errors
- */
- public short abortableProcessWorkbookEvents(HSSFRequest req, POIFSFileSystem fs)
- throws IOException, HSSFUserException {
- return abortableProcessWorkbookEvents(req, fs.getRoot());
- }
-
- /**
- * Processes a file into essentially record events.
- *
- * @param req an Instance of HSSFRequest which has your registered listeners
- * @param dir a DirectoryNode containing your workbook
- * @return numeric user-specified result code.
- *
- * @throws HSSFUserException if the processing should be aborted
- * @throws IOException if the workbook contained errors
- */
- public short abortableProcessWorkbookEvents(HSSFRequest req, DirectoryNode dir)
- throws IOException, HSSFUserException {
- try (InputStream in = dir.createDocumentInputStream("Workbook")) {
- return abortableProcessEvents(req, in);
- }
- }
-
- /**
- * Processes a DocumentInputStream into essentially Record events.
- *
- * If an <code>AbortableHSSFListener</code> causes a halt to processing during this call
- * the method will return just as with <code>abortableProcessEvents</code>, but no
- * user code or <code>HSSFUserException</code> will be passed back.
- *
- * @see org.apache.poi.poifs.filesystem.POIFSFileSystem#createDocumentInputStream(String)
- * @param req an Instance of HSSFRequest which has your registered listeners
- * @param in a DocumentInputStream obtained from POIFS's POIFSFileSystem object
- */
- public void processEvents(HSSFRequest req, InputStream in) {
- try {
- genericProcessEvents(req, in);
- } catch (HSSFUserException hue) {
- /*If an HSSFUserException user exception is thrown, ignore it.*/
- }
- }
-
-
- /**
- * Processes a DocumentInputStream into essentially Record events.
- *
- * @see org.apache.poi.poifs.filesystem.POIFSFileSystem#createDocumentInputStream(String)
- * @param req an Instance of HSSFRequest which has your registered listeners
- * @param in a DocumentInputStream obtained from POIFS's POIFSFileSystem object
- * @return numeric user-specified result code.
- *
- * @throws HSSFUserException if the processing should be aborted
- */
- public short abortableProcessEvents(HSSFRequest req, InputStream in)
- throws HSSFUserException {
- return genericProcessEvents(req, in);
- }
-
- /**
- * Processes a DocumentInputStream into essentially Record events.
- *
- * @see org.apache.poi.poifs.filesystem.POIFSFileSystem#createDocumentInputStream(String)
- * @param req an Instance of HSSFRequest which has your registered listeners
- * @param in a DocumentInputStream obtained from POIFS's POIFSFileSystem object
- * @return numeric user-specified result code.
- */
- private short genericProcessEvents(HSSFRequest req, InputStream in)
- throws HSSFUserException {
- short userCode = 0;
-
- // Create a new RecordStream and use that
- RecordFactoryInputStream recordStream = new RecordFactoryInputStream(in, false);
-
- // Process each record as they come in
- while(true) {
- org.apache.poi.hssf.record.Record r = recordStream.nextRecord();
- if(r == null) {
- break;
- }
- userCode = req.processRecord(r);
- if (userCode != 0) {
- break;
- }
- }
-
- // All done, return our last code
- return userCode;
- }
- }
|