org.sigmah.server.report.renderer.itext.ItextChartRenderer.java Source code

Introduction

Here is the source code for org.sigmah.server.report.renderer.itext.ItextChartRenderer.java

Source

/*
 * All Sigmah code is released under the GNU General Public License v3
 * See COPYRIGHT.txt and LICENSE.txt.
 */

package org.sigmah.server.report.renderer.itext;

import com.google.inject.Inject;
import com.lowagie.text.DocWriter;
import com.lowagie.text.Document;
import com.lowagie.text.Image;
import com.lowagie.text.pdf.PdfContentByte;
import com.lowagie.text.pdf.PdfWriter;
import org.sigmah.server.report.renderer.ChartRendererJC;
import org.sigmah.shared.report.model.PivotChartElement;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;

/**
 * Renders {@link org.sigmah.shared.report.model.PivotChartElement PivotChartElement}s into
 * an iText document.
 *
 * @author Alex Bertram
 */
public class ItextChartRenderer implements ItextRenderer<PivotChartElement> {
    private static final int RESOLUTION = 150;

    private final ChartRendererJC chartRenderer;

    @Inject
    public ItextChartRenderer(ChartRendererJC chartRenderer) {
        this.chartRenderer = chartRenderer;
    }

    public void render(DocWriter writer, Document doc, PivotChartElement element) {

        try {
            doc.add(ThemeHelper.elementTitle(element.getTitle()));
            ItextRendererHelper.addFilterDescription(doc, element.getContent().getFilterDescriptions());

            float width = doc.getPageSize().getWidth() - doc.rightMargin() - doc.leftMargin();
            float height = (doc.getPageSize().getHeight() - doc.topMargin() - doc.bottomMargin()) / 3f;

            if (writer instanceof PdfWriter) {

                // We can render the chart directly as vector graphics
                // in the PDF file

                PdfWriter pdfWriter = (PdfWriter) writer;
                PdfContentByte cb = pdfWriter.getDirectContent();
                Graphics2D g2d = cb.createGraphics(width, height);
                chartRenderer.render(element, false, g2d, (int) width, (int) height, 72);
                g2d.dispose();

            } else {

                // For RTF/Html we embed as a GIF

                width = width / 72f * RESOLUTION;
                height = height / 72f * RESOLUTION;

                BufferedImage chartImage = chartRenderer.renderImage(element, false, (int) width, (int) height,
                        RESOLUTION);

                ByteArrayOutputStream baos = new ByteArrayOutputStream();
                ImageIO.write(chartImage, "GIF", baos);

                Image image = Image.getInstance(baos.toByteArray());
                image.scalePercent(72f / RESOLUTION * 100f);

                doc.add(image);

            }

        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}