Java tutorial
/* * PdfGenerator.java * * This file is part of the STS-Tool project. * Copyright (c) 2011-2012 "University of Trento - DISI" All rights reserved. * * Is strictly forbidden to remove this copyright notice from this source code. * * Disclaimer of Warranty: * STS-Tool (this software) is provided "as-is" and without warranty of any kind, * express, implied or otherwise, including without limitation, any warranty of * merchantability or fitness for a particular purpose. * In no event shall the copyright holder or contributors be liable for any direct, * indirect, incidental, special, exemplary, or consequential damages * including, but not limited to, procurement of substitute goods or services; * loss of use, data, or profits; or business interruption) however caused and on * any theory of liability, whether in contract, strict liability, or tort (including * negligence or otherwise) arising in any way out of the use of this software, even * if advised of the possibility of such damage. * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License version 3 * as published by the Free Software Foundation with the addition of the * following permission added to Section 15 as permitted in Section 7(a): * FOR ANY PART OF THE COVERED WORK IN WHICH THE COPYRIGHT IS OWNED BY * "University of Trento - DISI","University of Trento - DISI" DISCLAIMS THE * WARRANTY OF NON INFRINGEMENT OF THIRD PARTY RIGHTS. * * See the GNU Affero General Public License for more details. * You should have received a copy of the GNU Affero General Public License * along with this program; if not, see http://www.gnu.org/licenses or write to * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, * Boston, MA, 02110-1301 USA, or download the license from the following URL: * http://www.sts-tool.eu/License.php * * For more information, please contact STS-Tool group at this * address: ststool@disi.unitn.it * */ package eu.aniketos.wp1.ststool.report.pdfgenerator; import java.io.IOException; import java.io.OutputStream; import java.util.ArrayList; import java.util.List; import org.eclipse.core.runtime.IProgressMonitor; import org.eclipse.core.runtime.SubProgressMonitor; import com.itextpdf.text.Chunk; import com.itextpdf.text.Document; import com.itextpdf.text.DocumentException; import com.itextpdf.text.Element; import com.itextpdf.text.Image; import com.itextpdf.text.PageSize; import com.itextpdf.text.Paragraph; import com.itextpdf.text.Phrase; import com.itextpdf.text.Rectangle; import com.itextpdf.text.Utilities; import com.itextpdf.text.pdf.ColumnText; import com.itextpdf.text.pdf.PdfContentByte; import com.itextpdf.text.pdf.PdfPageEventHelper; import com.itextpdf.text.pdf.PdfWriter; import com.itextpdf.text.pdf.draw.DottedLineSeparator; public class PdfGenerator { private final static float HEADER_HEIGHT = Utilities.millimetersToPoints(20f); private final static float FOOTER_HEIGHT = Utilities.millimetersToPoints(30f); private final static float MARGIN_RIGHT = Utilities.millimetersToPoints(23f); private final static float MARGIN_LEFT = Utilities.millimetersToPoints(23f); private final static float MARGIN_UP = HEADER_HEIGHT + Utilities.millimetersToPoints(4f); private final static float MARGIN_DOWN = FOOTER_HEIGHT + Utilities.millimetersToPoints(2f); private ReportContentFactory rcf; private List<Paragraph> indexElements = new ArrayList<Paragraph>(); Image footerImage; public PdfGenerator(ReportContentFactory rcf) { this.rcf = rcf; footerImage = rcf.getFooterImage(); } /** * Creates a PDF document. * * @param pdfFilename * the path to the new PDF document * @throws DocumentException * @throws IOException */ public void createPdf(OutputStream os, IProgressMonitor monitor) throws DocumentException, IOException { //createPdf(filename, createPdf(filename,0)); createPdf(os, -1, monitor); } private int createPdf(OutputStream os, int indexSize, IProgressMonitor monitor) throws DocumentException, IOException { Document document = new Document(PageSize.A4, MARGIN_LEFT, MARGIN_RIGHT, MARGIN_UP, MARGIN_DOWN); PdfWriter writer = PdfWriter.getInstance(document, os); writer.setStrictImageSequence(true); writer.setLinearPageMode();// used to reorder page; // writer.setPdfVersion(PdfWriter.PDF_VERSION_1_6); ChapterSectionTOC event = new ChapterSectionTOC(); writer.setPageEvent(event); event.pagenumber = rcf.startingPageNumber(); writer.setBoxSize("art", new Rectangle(MARGIN_LEFT, 0, PageSize.A4.getWidth() - MARGIN_RIGHT, FOOTER_HEIGHT)); monitor.beginTask("Generating Report", 100); document.open(); rcf.buildContent(writer, document, new SubProgressMonitor(monitor, 85)); if (rcf.buildIndex()) { document.newPage(); event.setIndex(true); int toc = writer.getPageNumber(); buildIndex(writer, document); document.newPage(); monitor.worked(10); int total = writer.reorderPages(null); int tocLenght = total - toc + 1; int[] order = new int[total]; order[0] = 1; for (int i = 0; i < tocLenght; i++) { order[i + 1] = toc + i; } for (int i = 2; i < toc; i++) { order[i + tocLenght - 1] = i; } writer.reorderPages(order); document.close(); monitor.worked(5); return tocLenght; } return -1; } private void buildIndex(PdfWriter writer, Document document) throws DocumentException { document.add(rcf.getIndexTitle()); for (Paragraph p : indexElements) { document.add(p); } } /** * Inner class to keep track of the TOC and to draw lines under ever chapter and section. */ class ChapterSectionTOC extends PdfPageEventHelper { final static int INDEX_INDENTATION = 10; /** List with the titles. */ Phrase header = new Phrase(" "); int pagenumber = -1; boolean index = false; public boolean isIndex() { return index; } public void setIndex(boolean index) { this.index = index; } @Override public void onChapter(PdfWriter writer, Document document, float position, Paragraph title) { header = new Phrase("header : " + title.getContent()); Paragraph p = new Paragraph(); p.add(new Chunk(title.getContent(), rcf.getIndexChapterFont())); Chunk c = new Chunk(Integer.toString(pagenumber), rcf.getIndexNumberFont()); p.add(new Chunk(new DottedLineSeparator(), PageSize.A4.getWidth() - MARGIN_LEFT - MARGIN_RIGHT - c.getWidthPoint() - INDEX_INDENTATION - 1, false)); p.setIndentationLeft(INDEX_INDENTATION); p.add(c); indexElements.add(p); } @Override public void onChapterEnd(PdfWriter writer, Document document, float position) { } @Override public void onSection(PdfWriter writer, Document document, float position, int depth, Paragraph title) { Paragraph p = new Paragraph(); p.add(new Chunk(title.getContent(), rcf.getIndexSectionFont())); Chunk c = new Chunk(Integer.toString(pagenumber), rcf.getIndexNumberFont()); p.add(new Chunk(new DottedLineSeparator(), PageSize.A4.getWidth() - MARGIN_LEFT - MARGIN_RIGHT - c.getWidthPoint() - (INDEX_INDENTATION * depth) - 1, false)); p.add(c); p.setIndentationLeft(INDEX_INDENTATION * depth); indexElements.add(p); } @Override public void onSectionEnd(PdfWriter writer, Document document, float position) { } @Override public void onOpenDocument(PdfWriter writer, Document document) { } /** * Increase the page number. * * @see com.itextpdf.text.pdf.PdfPageEventHelper#onStartPage(com.itextpdf.text.pdf.PdfWriter, com.itextpdf.text.Document) */ @Override public void onStartPage(PdfWriter writer, Document document) { pagenumber++; } /** * Adds the header and the footer. * * @see com.itextpdf.text.pdf.PdfPageEventHelper#onEndPage(com.itextpdf.text.pdf.PdfWriter, com.itextpdf.text.Document) */ @Override public void onEndPage(PdfWriter writer, Document document) { if (writer.getPageNumber() == 1) return; if (isIndex()) { } else { PdfContentByte cb = writer.getDirectContent(); cb.saveState(); cb.setLineWidth(1); /** ----------------------HEADER-------------------- */ cb.moveTo(MARGIN_LEFT, PageSize.A4.getHeight() - HEADER_HEIGHT - 2); cb.lineTo(PageSize.A4.getWidth() - MARGIN_RIGHT, PageSize.A4.getHeight() - HEADER_HEIGHT - 2); cb.stroke(); //ColumnText.showTextAligned(writer.getDirectContent(),Element.ALIGN_CENTER, header,PageSize.A4.getWidth()/2,PageSize.A4.getHeight()-HEADER_HEIGHT, 0); /** ----------------------FOOTER-------------------- */ cb.moveTo(MARGIN_LEFT, FOOTER_HEIGHT); cb.lineTo(PageSize.A4.getWidth() - MARGIN_RIGHT, FOOTER_HEIGHT); cb.stroke(); footerImage.setAbsolutePosition((PageSize.A4.getWidth() - footerImage.getPlainWidth()) / 2, FOOTER_HEIGHT - footerImage.getPlainHeight() - 5); try { cb.addImage(footerImage); } catch (DocumentException e) { e.printStackTrace(); } Phrase p = rcf.getPageNumberPhrase(pagenumber); ColumnText.showTextAligned(writer.getDirectContent(), Element.ALIGN_RIGHT, p, PageSize.A4.getWidth() - MARGIN_RIGHT, FOOTER_HEIGHT - p.getFont().getSize() - 2, 0); cb.restoreState(); } if (SHOW_AREA) { PdfContentByte under = writer.getDirectContentUnder(); under.saveState(); under.setRGBColorFill(0xFF, 0xD7, 0x00); under.rectangle(MARGIN_LEFT, MARGIN_DOWN, PageSize.A4.getWidth() - MARGIN_RIGHT - MARGIN_LEFT, PageSize.A4.getHeight() - MARGIN_UP - MARGIN_DOWN); under.fill(); under.restoreState(); under.saveState(); under.setRGBColorFill(0x99, 0xcc, 0xff); under.rectangle(MARGIN_LEFT, PageSize.A4.getHeight() - HEADER_HEIGHT, PageSize.A4.getWidth() - MARGIN_RIGHT - MARGIN_LEFT, HEADER_HEIGHT); under.fill(); under.restoreState(); under.saveState(); under.setRGBColorFill(0x99, 0xcc, 0xff); under.rectangle(MARGIN_LEFT, 0, PageSize.A4.getWidth() - MARGIN_RIGHT - MARGIN_LEFT, FOOTER_HEIGHT); under.fill(); under.restoreState(); } } } private final static boolean SHOW_AREA = false; }