package kevin.cn;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import javax.imageio.ImageIO;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.pdmodel.PDPage;
import org.apache.pdfbox.pdmodel.encryption.InvalidPasswordException;
import org.apache.pdfbox.rendering.ImageType;
import org.apache.pdfbox.rendering.PDFRenderer;
import org.apache.pdfbox.tools.PDFBox;
import kevin.cn.PdfUtil;
@SuppressWarnings("unused")
public class Test {
  //經過測試,dpi為96,100,105,120,150,200中,105顯示效果較為清晰,體積穩定,dpi越高圖片體積越大,一般電腦顯示分辨率為96
  public static final float DEFAULT_DPI = 105;
  //默認轉換的圖片格式為jpg
  public static final String DEFAULT_FORMAT = "jpg";
  public static void main(String[] args) throws Exception {
    pdfToImage("/media/kevin/FileData/JavaCode/pdfboxTest/企業信息化建設論文.pdf","/media/kevin/FileData/JavaCode/pdfboxTest/img/7.jpg",5);
  }

/**
   * pdf轉圖片
   *
   * @param pdfPath PDF路徑 
   * @imgPath img路徑
   * @page_end 要轉換的頁碼，也可以定義開始頁碼和結束頁碼，我這里只需要一頁，根據需求自行添加
   */
  public static void pdfToImage(String pdfPath, String imgPath,int page_end) {
    try {
      //圖像合并使用參數
      // 總寬度
      int width = 0;
      // 保存一張圖片中的RGB數據
      int[] singleImgRGB;
      int shiftHeight = 0;
      //保存每張圖片的像素值
      BufferedImage imageResult = null;
      //利用PdfBox生成圖像
      PDDocument pdDocument = PDDocument.load(new File(pdfPath));
      PDFRenderer renderer = new PDFRenderer(pdDocument);
      //循環每個頁碼
      for (int i = 0, len = pdDocument.getNumberOfPages(); i < len; i++) {
        if (i==page_end) {
          BufferedImage image = renderer.renderImageWithDPI(i, DEFAULT_DPI, ImageType.RGB);
          int imageHeight = image.getHeight();
          int imageWidth = image.getWidth();
          //計算高度和偏移量
          //使用第一張圖片寬度;
          width = imageWidth;
          //保存每頁圖片的像素值
          imageResult = new BufferedImage(width, imageHeight, BufferedImage.TYPE_INT_RGB);
          //這里有高度，可以將imageHeight*len，我這里值提取一頁所以不需要
           singleImgRGB = image.getRGB(0, 0, width, imageHeight, null, 0, width);
           // 寫入流中
           imageResult.setRGB(0, shiftHeight, width, imageHeight, singleImgRGB, 0, width);
        }else if(i>page_end) {
          continue;
        }
      }
      pdDocument.close();
      // 寫圖片
      ImageIO.write(imageResult, DEFAULT_FORMAT, new File(imgPath));
    } catch (Exception e) {
      e.printStackTrace();
    }
    //OVER
  }