package org.common.util.pdftohtml;

import java.awt.image.BufferedImage;

import java.awt.image.RenderedImage;

import java.io.File;

import java.io.FileOutputStream;

import java.util.ArrayList;

import java.util.List;

import javax.imageio.ImageIO;

import org.icepdf.core.pobjects.Document;

import org.icepdf.core.pobjects.Page;

import org.icepdf.core.util.GraphicsRenderingHints;

public class PdfToJpg

{

    public static void main(String[] arguments)

    {

        pdftoimage("D:\\pdf2htmlEX-v1.0\\PDF\\1.pdf","D:\\pdf2htmlEX-v1.0\\PDF\\a.jpg");

    }

    public static void pdftoimage(String pdfpath,String imagepath){

        Document document = new Document();

        try

        {

            document.setFile(pdfpath);

        }

        catch (Exception e1)

        {

            e1.printStackTrace();

        }

        int pages = document.getNumberOfPages();

        System.out.println("pages  " + pages);

        BufferedImage image = null;

        RenderedImage rendImage = null;

        List<BufferedImage> piclist = new ArrayList<BufferedImage>();

        for( int i = 0 ; i < pages ; i++ )

        {

            image = (BufferedImage)document.getPageImage(i,GraphicsRenderingHints.SCREEN,Page.BOUNDARY_TRIMBOX, 0f, 1.5f);;

            rendImage = image;

            piclist.add(image);

        }

        yPic(piclist, imagepath);

        document.dispose();

    }

    /**

     * 将宽度相同的图片，竖向追加在一起 ##注意：宽度必须相同

     *

     * @param piclist

     *            文件流数组

     * @param outPath

     *            输出路径

     */

    public static void yPic(List<BufferedImage> piclist, String outPath) {// 纵向处理图片

        if (piclist == null || piclist.size() <= 0) {

            System.out.println("图片数组为空!");

            return;

        }

        try {

            int height = 0, // 总高度

            width = 0, // 总宽度

            _height = 0, // 临时的高度 , 或保存偏移高度

            __height = 0, // 临时的高度，主要保存每个高度

            picNum = piclist.size();// 图片的数量

            File fileImg = null; // 保存读取出的图片

            int[] heightArray = new int[picNum]; // 保存每个文件的高度

            BufferedImage buffer = null; // 保存图片流

            List<int[]> imgRGB = new ArrayList<int[]>(); // 保存所有的图片的RGB

            int[] _imgRGB; // 保存一张图片中的RGB数据

            for (int i = 0; i < picNum; i++) {

                buffer = piclist.get(i);

                heightArray[i] = _height = buffer.getHeight();// 图片高度

                if (i == 0) {

                    width = buffer.getWidth();// 图片宽度

                }

                height += _height; // 获取总高度

                _imgRGB = new int[width * _height];// 从图片中读取RGB

                _imgRGB = buffer.getRGB(0, 0, width, _height, _imgRGB, 0, width);

                imgRGB.add(_imgRGB);

            }

            _height = 0; // 设置偏移高度为0

            // 生成新图片

            BufferedImage imageResult = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);

            for (int i = 0; i < picNum; i++) {

                __height = heightArray[i];

                if (i != 0) _height += __height; // 计算偏移高度

                imageResult.setRGB(0, _height, width, __height, imgRGB.get(i), 0, width); // 写入流中

            }

            File outFile = new File(outPath);

            ImageIO.write(imageResult, "jpg", outFile);// 写图片

            new FileOutputStream(outFile).close();

        } catch (Exception e) {

            e.printStackTrace();

        }

    }

}