source: trunk/src/org/expeditee/io/PDF2Writer.java@ 309

package org.expeditee.io;

import java.awt.Dimension;
import java.awt.Image;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.Writer;
import java.util.Collection;
import java.util.List;

import org.expeditee.gui.Frame;
import org.expeditee.gui.FrameGraphics;
import org.expeditee.gui.UserSettings;
import org.expeditee.items.Item;
import org.expeditee.items.Picture;
import org.expeditee.items.Text;

import com.lowagie.text.Document;
import com.lowagie.text.DocumentException;
import com.lowagie.text.Font;
import com.lowagie.text.FontFactory;
import com.lowagie.text.Rectangle;
import com.lowagie.text.pdf.PdfContentByte;
import com.lowagie.text.pdf.PdfWriter;

/**
 * 
 * @author root
 *
 */
public class PDF2Writer extends DefaultFrameWriter {

        private Document _pdfDocument;

        private PdfWriter _pdfWriter;

        public PDF2Writer() {
                Dimension d = FrameGraphics.getMaxSize();
                _pdfDocument = new Document(new Rectangle(d.width, d.height));
        }

        @Override
        protected void initialise(Frame start, Writer writer) throws IOException {
                _format = ".pdf";
                super.initialise(start, writer);
                try {
                        _pdfWriter = PdfWriter.getInstance(_pdfDocument,
                                        new FileOutputStream(_filename));
                        _pdfDocument.open();
                        _pdfDocument.addCreationDate();
                        _pdfDocument.addAuthor(UserSettings.Username);
                        _pdfDocument.addCreator("Expeditee");
                } catch (DocumentException e) {
                        e.printStackTrace();
                        throw new IOException(e.getMessage());
                }
        }

        @Override
        protected void writeTitle(Text toWrite, List<Item> items)
                        throws IOException {
                _pdfDocument.addTitle(toWrite.getText());
                writeText(toWrite);
        }

        @Override
        protected String finalise() throws IOException {
                _pdfDocument.close();
                return super.finalise();
        }

        @Override
        protected void writeText(Text text) throws IOException {
                PdfContentByte cb = _pdfWriter.getDirectContent();
                // we tell the ContentByte we're ready to draw text
                cb.beginText();
                Font font = FontFactory.getFont(
                                Conversion.getPdfFont(text.getFamily()), text.getSize(), text
                                                .getPaintFont().getStyle(), text.getPaintColor());
                cb.setFontAndSize(font.getBaseFont(), text.getSize());
                // cb.setColorStroke(text.getPaintColor());
                cb.setColorFill(text.getPaintColor());

                // we draw some text on a certain position
                cb.setTextMatrix(text.getX(), _pdfWriter.getPageSize().getHeight() - text.getY());
                List<String> textLines = text.getTextList();
                //cb.showText(textLines.get(0));
                for(int i = 0; i < textLines.size(); i++){
                        cb.showText(textLines.get(i));
                        cb.moveText(0, -text.getLineHeight());
                }
                
                // we tell the contentByte, we've finished drawing text
                cb.endText();
        }

        @Override
        protected void writePicture(Picture pic) throws IOException {
                Image image = pic.getCroppedImage();
                try {
                        PdfContentByte cb = _pdfWriter.getDirectContent();
                        com.lowagie.text.Image iTextImage = com.lowagie.text.Image
                                        .getInstance(image, null);
                        iTextImage.setAbsolutePosition(pic.getX(), _pdfWriter.getPageSize().getHeight() - pic.getY() - pic.getHeight());
                        cb.addImage(iTextImage);
                } catch (DocumentException e) {
                        e.printStackTrace();
                }
        }
        
        @Override
        protected Collection<Item> getItemsToWrite(Frame toWrite) {
                return toWrite.getVisibleItems();
        }
}