source: trunk/src/org/expeditee/importer/pdfImporter.java@ 919

/**
 *    pdfImporter.java
 *    Copyright (C) 2010 New Zealand Digital Library, http://expeditee.org
 *
 *    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 3 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, see <http://www.gnu.org/licenses/>.
 */

package org.expeditee.importer;

import java.awt.Color;
import java.awt.Graphics;
import java.awt.Image;
import java.awt.Point;
import java.awt.Rectangle;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.io.RandomAccessFile;
import java.nio.ByteBuffer;
import java.nio.channels.FileChannel;

import javax.imageio.ImageIO;

import org.expeditee.gui.Browser;
import org.expeditee.gui.Frame;
import org.expeditee.gui.FrameCreator;
import org.expeditee.gui.FrameGraphics;
import org.expeditee.gui.FrameIO;
import org.expeditee.gui.MessageBay;
import org.expeditee.items.Item;
import org.expeditee.items.Text;

import com.sun.pdfview.PDFFile;
import com.sun.pdfview.PDFPage;

public class pdfImporter implements FileImporter {
        
        public Item importFile(final File f, Point location) throws IOException {
                if (location == null || f == null) {
                        return null;
                }
                
                final int x = 0;
                final int y = 60;
                final int width = Browser._theBrowser.getSize().width;
                System.out.println(width);
                
                final String name = FrameIO.ConvertToValidFramesetName(f.getName().substring(0,f.getName().lastIndexOf('.'))).toLowerCase();
                
                //check if the file is a pdf
                if(!f.getAbsolutePath().substring(f.getAbsolutePath().lastIndexOf('.')+1).toLowerCase().equals("pdf"))
                {
                        return null;
                }
                
                final Text link = FrameDNDTransferHandler.importString(name, location);
                link.setLink(name+"1");
                
                final Frame frameset;
                try
                {
                        frameset = FrameIO.CreateNewFrameset(name);
                }
                catch(Exception e)
                {
                        MessageBay.displayMessage("Frameset \"" + name + "\" already exists, creating a link to it");
                        return link;//if the frameset exists just make a link to the existing frameset
                }
                
                final String framesetPath = frameset.getPath()+name+File.separator;
                
                System.out.println("PATH = " + framesetPath);
                
                new Thread() {
                        public void run() {
                                try {
                                        MessageBay.displayMessage("Importing " + f.getName() + "...");
                                        //load a pdf from a byte buffer
                      RandomAccessFile raf = new RandomAccessFile(f, "r");
                      FileChannel channel = raf.getChannel();
                      ByteBuffer buf = channel.map(FileChannel.MapMode.READ_ONLY, 0, channel.size());
                      PDFFile pdffile = new PDFFile(buf);
                      int pages = pdffile.getNumPages();
                      
                      Frame currentFrame=frameset;
                                        Frame _currentFrame=null;
                                        Text nextButton=null, prevButton=null;
                                        final float spacing=((Text)FrameCreator.createButton("Next", null, null, 10F, 10F)).getBoundsWidth() + 30F;
                      
                      //make images from the pdf pages, write frames with those images
                                        for(int i=1; i<=pages; i++)
                                        {
                                                //remove the title from the frame
                                                /* if(FrameDNDTransferHandler.modifier!=1 &&
                                                                currentFrame!=null &&
                                                                currentFrame.getTitleItem()!=null)
                                                        ((Text)currentFrame.getTitleItem()).delete(); */
                                          //get the pdf page
                              PDFPage page = pdffile.getPage(i);
                              //get the width and height for the page
                              int w=(int)page.getBBox().getWidth(), h=(int)page.getBBox().getHeight();
                              final double res=1080;
                              if(w<res || h<res)
                              {
                                      double ws=res/w;
                                      double hs=res/h;
                                      if(ws>hs)
                                      {
                                              w*=ws;
                                              h*=ws;
                                      }
                                      else
                                      {
                                              w*=hs;
                                              h*=hs;
                                      }
                              }
                              //generate the image
                              Image img = page.getImage(w, h, //width & height
                                      new Rectangle(0, 0, (int)page.getBBox().getWidth(), (int)page.getBBox().getHeight()), //clip rect
                                      null, //null for the ImageObserver
                                      true, //fill background with white
                                      true //block until drawing is done
                                      );
                              //Create a buffered image to store the image in
                              BufferedImage bimg = new BufferedImage(img.getWidth(null), img.getHeight(null), BufferedImage.TYPE_INT_RGB);
                              Graphics g = bimg.createGraphics();
                              //Paint the image onto the buffered image
                              g.drawImage(img, 0, 0, null);
                              g.dispose();
                              //save it as a file
                              File out = new File(framesetPath+i+".png");
                              ImageIO.write(bimg, "png", out);
                              //generate a frame with that image
                              System.out.println(width);
                              System.out.println("@i: "+framesetPath+i+".png "+width);
                                                currentFrame.addText(x, y, "@i: "+framesetPath+i+".png "+width, null);
                                                if(i>1)
                                                {
                                                  //put a next button on the previous frame (points to current frame)
                                                  nextButton=(Text)FrameCreator.createButton("Next", null, null, 10F, 10F);
                                                        nextButton.setID(_currentFrame.getNextItemID());
                                                        nextButton.addAction("next");
                                                        _currentFrame.addItem(nextButton);
                                                        //put a previous button on the current frame (points to previous frame)
                                                        prevButton=(Text)FrameCreator.createButton("Prev", null, null, (i<pages)?spacing:10F, 10F);
                                                        prevButton.setID(currentFrame.getNextItemID());
                                                        prevButton.addAction("previous");
                                                        currentFrame.addItem(prevButton);
                                                }
                                                else
                                                {
                                                        prevButton=(Text)FrameCreator.createButton("Home", null, null, (i<pages)?spacing:10F, 10F);
                                                        prevButton.setID(currentFrame.getNextItemID());
                                                        prevButton.addAction("GotoHome");
                                                        currentFrame.addItem(prevButton);
                                                }
                                                FrameIO.SaveFrame(_currentFrame,true);
                                                _currentFrame=currentFrame;
                                                if(i<pages) currentFrame = FrameIO.CreateFrame(frameset.getFramesetName(), name, null);
                                        }
                                        prevButton=(Text)FrameCreator.createButton("Home", null, null, spacing, 10F);
                                        prevButton.setID(currentFrame.getNextItemID());
                                        prevButton.addAction("gotohome");
                                        currentFrame.addItem(prevButton);
                                        FrameIO.SaveFrame(currentFrame,true);
                                        MessageBay.displayMessage(f.getName() + " import complete", Color.GREEN);
                                        FrameGraphics.requestRefresh(true);
                                } catch (Exception e) {
                                        e.printStackTrace();
                                        MessageBay.errorMessage(e.getMessage());
                                }
                        }
                }.start();
                FrameGraphics.refresh(true);
                //return source;
                return link;
        }
}