org.jahia.modules.docviewer.PDFBoxPDF2ImageConverterService.java Source code

Introduction

Here is the source code for org.jahia.modules.docviewer.PDFBoxPDF2ImageConverterService.java

Source

/**
 * This file is part of Jahia, next-generation open source CMS:
 * Jahia's next-generation, open source CMS stems from a widely acknowledged vision
 * of enterprise application convergence - web, search, document, social and portal -
 * unified by the simplicity of web content management.
 *
 * For more information, please visit http://www.jahia.com.
 *
 * Copyright (C) 2002-2011 Jahia Solutions Group SA. All rights reserved.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 * As a special exception to the terms and conditions of version 2.0 of
 * the GPL (or any later version), you may redistribute this Program in connection
 * with Free/Libre and Open Source Software ("FLOSS") applications as described
 * in Jahia's FLOSS exception. You should have received a copy of the text
 * describing the FLOSS exception, and it is also available here:
 * http://www.jahia.com/license
 *
 * Commercial and Supported Versions of the program (dual licensing):
 * alternatively, commercial and supported versions of the program may be used
 * in accordance with the terms and conditions contained in a separate
 * written agreement between you and Jahia Solutions Group SA.
 *
 * If you are unsure which license is appropriate for your use,
 * please contact the sales department at sales@jahia.com.
 */

package org.jahia.modules.docviewer;

import java.awt.image.BufferedImage;
import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;

import org.apache.commons.io.IOUtils;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.pdmodel.PDPage;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Creates images for PDF document pages using PDFBox library.
 * 
 * @author Sergiy Shyrkov
 */
public class PDFBoxPDF2ImageConverterService implements PDF2ImageConverter {

    private static final Logger logger = LoggerFactory.getLogger(PDFBoxPDF2ImageConverterService.class);

    private int imageType = BufferedImage.TYPE_INT_RGB;

    private int resolution = 96;

    public BufferedImage getImageOfPage(InputStream pdfInputStream, int pageNumber) throws Exception {
        BufferedImage image = null;

        PDDocument pdfDoc = null;
        try {
            pdfDoc = PDDocument.load(pdfInputStream);
            PDPage page = (PDPage) pdfDoc.getDocumentCatalog().getAllPages().get(pageNumber);
            image = page.convertToImage(imageType, resolution);
        } catch (IndexOutOfBoundsException e) {
            logger.warn("No page with the number {} found in the PDF document", pageNumber);
        } finally {
            try {
                pdfDoc.close();
            } catch (Exception e) {
                // ignore
            }
        }

        return image;
    }

    public BufferedImage getImageOfPage(File pdfFile, int pageNumber) throws Exception {
        BufferedImage image = null;

        InputStream is = null;
        try {
            is = new BufferedInputStream(new FileInputStream(pdfFile));
            image = getImageOfPage(is, pageNumber);
        } finally {
            IOUtils.closeQuietly(is);
        }

        return image;
    }

    public void setImageType(int imageType) {
        this.imageType = imageType;
    }

    public void setResolution(int resolution) {
        this.resolution = resolution;
    }

}