PdfSnapShotExporter.java

/*
 * Copyright 2013 Serdar.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package de.fub.maps.project.snapshot.impl;

import com.itextpdf.awt.PdfGraphics2D;
import com.itextpdf.text.Document;
import com.itextpdf.text.DocumentException;
import com.itextpdf.text.Rectangle;
import com.itextpdf.text.pdf.PdfContentByte;
import com.itextpdf.text.pdf.PdfWriter;
import de.fub.maps.project.snapshot.api.AbstractComponentSnapShotExporter;
import de.fub.maps.project.snapshot.api.ComponentSnapShotExporter;
import java.awt.Component;
import java.awt.Dimension;
import java.awt.Image;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import javax.swing.SwingUtilities;
import org.netbeans.api.annotations.common.StaticResource;
import org.openide.util.Exceptions;
import org.openide.util.ImageUtilities;
import org.openide.util.NbBundle;
import org.openide.util.RequestProcessor;
import org.openide.util.lookup.ServiceProvider;

/**
 * Implementation of the ComponentSnapShotExporter, which handler the
 * exportation/printing of a Component as PDF file.
 *
 * @author Serdar
 */
@NbBundle.Messages({
    "CLT_PdfSnapShotExporter_Description=Exports a component as Pdf File.",
    "CLT_PdfSnapShotExporter_Name=Pdf"
})
@ServiceProvider(service = ComponentSnapShotExporter.class)
public class PdfSnapShotExporter extends AbstractComponentSnapShotExporter {

    @StaticResource
    private static final String ICON_PATH = "de/fub/maps/project/snapshot/impl/pdfIcon.png";

    @Override
    public Image getIconImage() {
        return ImageUtilities.loadImage(ICON_PATH, false);
    }

    @Override
    public String getName() {
        return Bundle.CLT_PdfSnapShotExporter_Name();
    }

    @Override
    public String getShortDescription() {
        return Bundle.CLT_PdfSnapShotExporter_Description();
    }

    @Override
    public void export(Component component) {
        File showFileChoose = showFileChoose("pdf");
        if (showFileChoose != null) {
            handleExport(showFileChoose, component);
        }
    }

    private void handleExport(final File selectedFile, final Component component) {
        RequestProcessor.getDefault().post(new Runnable() {
            @Override
            public void run() {
                SwingUtilities.invokeLater(new Runnable() {
                    @Override
                    public void run() {

                        if (component != null) {
                            PdfGraphics2D pdfGraphics2D = null;
                            try {
                                Dimension preferredSize = component.getSize();
                                Dimension dimension = preferredSize; //DimensionUtil.computeToA4Pdf(preferredSize);
                                // step 1
                                Document document = new Document(new Rectangle(dimension.width, dimension.height));
                                // step 2
                                PdfWriter writer = PdfWriter.getInstance(document, new FileOutputStream(selectedFile));
                                // step 3
                                document.open();
//                                document.newPage();
                                // step 4
                                PdfContentByte cb = writer.getDirectContent();
//                                PdfTemplate map = cb.createTemplate(dimension.width * 10, dimension.height * 10);
                                pdfGraphics2D = new PdfGraphics2D(cb, dimension.width, dimension.height);
//                                component.setPreferredSize(dimension);
//                                component.setSize(dimension);
//                                component.revalidate();
//                                component.repaint();
                                // paintAll must be called, a simple paint does
                                //not change the size of the component
                                component.printAll(pdfGraphics2D);
//                                component.setPreferredSize(preferredSize);
//                                component.setSize(preferredSize);
//                                component.revalidate();
//                                component.repaint();

                                pdfGraphics2D.dispose();
//                                cb.addTemplate(map, 0, 0);
                                // step 5
                                document.close();
                            } catch (FileNotFoundException ex) {
                                Exceptions.printStackTrace(ex);
                            } catch (DocumentException ex) {
                                Exceptions.printStackTrace(ex);
                            } finally {
                                if (pdfGraphics2D != null) {
                                    pdfGraphics2D.dispose();
                                }
                            }
                        }
                    }
                });
            }
        });
    }
}