indexer.PDFTextExtractor.java Source code

Introduction

Here is the source code for indexer.PDFTextExtractor.java

Source

/*
 * Copyright (c) 2016 Chris Bellis
 * This software is subject to the MIT License, see LICENSE.txt in the root of the repository.
 */

package indexer;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.text.PDFTextStripper;

import java.io.File;
import java.io.IOException;
import java.io.Writer;

/**
 * Created by Chris on 8/19/2015.
 * Simple utility class for extracting text from a PDF
 */
public class PDFTextExtractor {
    private static final Log log = LogFactory.getLog(PDFTextExtractor.class);
    // Instantiate this so that we can just have it accessable
    private PDFTextStripper stripper;

    public PDFTextExtractor() {
        try {
            stripper = new PDFTextStripper();
        } catch (IOException e) {
            log.error("Could not create PDF Text Stripper", e);
        }

    }

    /**
     * Given a PDF file, gets all the text from it
     *
     * @param filename The filename to get all the text from
     * @return The fulltext of the file
     * @throws IOException
     */
    public String extractText(String filename) throws IOException {
        PDDocument document = getPDDocument(filename);
        String res = stripper.getText(document);
        document.close();
        return res;
    }

    public void extractTextToWriter(String filename, Writer out) throws IOException {
        PDDocument document = getPDDocument(filename);
        stripper.writeText(document, out);
        document.close();
    }

    private PDDocument getPDDocument(String filename) throws IOException {
        if (stripper == null)
            throw new IOException("ERROR: PDFStripper was not created");
        File file = new File(filename);
        if (!file.exists())
            throw new IOException("ERROR: " + filename + " doesn't exist");
        return PDDocument.load(file);
    }
}