ingo@318: package de.intevation.gnv.exports; ingo@318: ingo@320: import com.lowagie.text.Document; ingo@320: import com.lowagie.text.DocumentException; ingo@320: import com.lowagie.text.PageSize; ingo@320: import com.lowagie.text.Rectangle; sascha@623: ingo@320: import com.lowagie.text.pdf.PdfContentByte; ingo@320: import com.lowagie.text.pdf.PdfTemplate; ingo@320: import com.lowagie.text.pdf.PdfWriter; ingo@320: sascha@623: import de.intevation.artifactdatabase.XMLUtils; sascha@623: sascha@623: import de.intevation.gnv.chart.Chart; sascha@623: ingo@320: import java.awt.Graphics2D; sascha@623: import java.awt.Transparency; sascha@623: sascha@623: import java.awt.geom.Rectangle2D.Double; sascha@623: ingo@320: import java.awt.geom.Rectangle2D; sascha@623: ingo@617: import java.awt.image.BufferedImage; sascha@623: sascha@623: import java.io.IOException; ingo@318: import java.io.OutputStream; ingo@321: import java.io.OutputStreamWriter; ingo@321: import java.io.UnsupportedEncodingException; sascha@623: ingo@318: import javax.imageio.ImageIO; ingo@318: ingo@321: import org.apache.batik.svggen.SVGGraphics2D; ingo@321: import org.apache.batik.svggen.SVGGraphics2DIOException; sascha@623: ingo@318: import org.apache.log4j.Logger; ingo@318: ingo@318: import org.jfree.chart.JFreeChart; ingo@318: ingo@318: /** ingo@318: * @author Ingo Weinzierl ingo@318: */ ingo@318: public class ChartExportHelper { ingo@318: ingo@320: private static final String DEFAULT_PAGE_SIZE = "A4"; ingo@321: private static final String DEFAULT_ENCODING = "UTF-8"; ingo@318: ingo@318: private static Logger log = Logger.getLogger(ChartExportHelper.class); ingo@318: ingo@318: public static void exportImage( ingo@318: OutputStream out, ingo@318: JFreeChart chart, ingo@318: String format, ingo@318: int width, ingo@318: int height ingo@318: ) ingo@318: throws IOException ingo@318: { ingo@318: log.info("export chart as png"); ingo@318: ingo@318: ImageIO.write( ingo@318: chart.createBufferedImage( ingo@318: width, height, Transparency.BITMASK, null ingo@318: ), ingo@318: format, ingo@318: out ingo@318: ); ingo@318: } ingo@318: ingo@320: ingo@617: public static void exportHistograms( ingo@617: OutputStream out, ingo@617: Chart[] histograms, ingo@617: String format, ingo@617: int width, ingo@617: int height) ingo@617: throws IOException ingo@617: { ingo@617: log.info("export histograms"); ingo@617: ingo@617: int size = histograms.length; ingo@617: BufferedImage image = new BufferedImage( ingo@617: width, height*size, BufferedImage.TYPE_INT_RGB); ingo@617: Graphics2D g = image.createGraphics(); ingo@617: ingo@617: for (int i = 0; i < size; i++) { ingo@617: JFreeChart chart = histograms[i].generateChart(); ingo@617: chart.draw(g, new Rectangle2D.Double(0.0D, i*height, width, height)); ingo@617: } ingo@617: g.finalize(); ingo@617: ingo@617: ImageIO.write(image, format, out); ingo@617: } ingo@617: ingo@617: ingo@639: public static void exportHistogramsAsSVG( ingo@639: OutputStream out, ingo@639: Chart[] histograms, ingo@639: String encoding, ingo@639: int width, ingo@639: int height ingo@639: ) { ingo@639: log.info("export histograms as svg"); ingo@639: ingo@639: if (encoding == null) ingo@639: encoding = DEFAULT_ENCODING; ingo@639: ingo@639: org.w3c.dom.Document document = XMLUtils.newDocument(); ingo@639: SVGGraphics2D graphics = new SVGGraphics2D(document); ingo@639: ingo@639: int size = histograms.length; ingo@639: for (int i = 0; i < size; i++) { ingo@639: JFreeChart chart = histograms[i].generateChart(); ingo@639: chart.draw(graphics, new Rectangle2D.Double( ingo@639: 0.0D, i*height,width,height)); ingo@639: } ingo@639: graphics.finalize(); ingo@639: ingo@639: try { ingo@639: graphics.stream(new OutputStreamWriter(out, encoding)); ingo@639: } ingo@639: catch (SVGGraphics2DIOException svge) { ingo@639: log.error("Error while writing svg export to output stream.", svge); ingo@639: } ingo@639: catch (UnsupportedEncodingException uee) { ingo@639: log.error("Unsupported encoding: " + encoding, uee); ingo@639: } ingo@639: } ingo@639: ingo@639: ingo@321: public static void exportSVG( ingo@321: OutputStream out, ingo@321: JFreeChart chart, ingo@321: String encoding, ingo@321: int width, ingo@321: int height ingo@321: ) { ingo@321: log.info("export chart as svg"); ingo@321: ingo@321: if (encoding == null) ingo@321: encoding = DEFAULT_ENCODING; ingo@321: ingo@323: org.w3c.dom.Document document = XMLUtils.newDocument(); ingo@321: SVGGraphics2D graphics = new SVGGraphics2D(document); ingo@321: ingo@321: chart.draw(graphics, new Rectangle2D.Double(0.0D, 0.0D,width,height)); ingo@321: ingo@321: try { ingo@321: graphics.stream(new OutputStreamWriter(out, encoding)); ingo@321: } ingo@321: catch (SVGGraphics2DIOException svge) { ingo@321: log.error("Error while writing svg export to output stream.", svge); ingo@321: } ingo@321: catch (UnsupportedEncodingException uee) { ingo@321: log.error("Unsupported encoding: " + encoding, uee); ingo@321: } ingo@318: } ingo@318: ingo@320: ingo@320: public static void exportPDF( ingo@320: OutputStream out, ingo@320: JFreeChart chart, ingo@320: String pageFormat, ingo@320: boolean landscape, ingo@320: float marginLeft, ingo@320: float marginRight, ingo@320: float marginTop, ingo@320: float marginBottom ingo@320: ) { ingo@320: log.info("export chart as pdf."); ingo@320: ingo@320: if (pageFormat == null) ingo@320: pageFormat = DEFAULT_PAGE_SIZE; ingo@320: ingo@320: Rectangle page = PageSize.getRectangle(pageFormat); ingo@320: int pageWidth = (int) (page.getRight(marginRight) - page.getLeft(marginLeft)); ingo@320: int pageHeight = (int) (page.getTop(marginTop) - page.getBottom(marginBottom)); ingo@320: ingo@320: Document document = null; ingo@320: if (landscape) { ingo@320: document = new Document(page.rotate()); ingo@320: log.debug("Create landscape pdf."); ingo@320: } ingo@320: else ingo@320: document = new Document(page); ingo@320: ingo@320: try { ingo@320: PdfWriter writer = PdfWriter.getInstance(document, out); ingo@320: ingo@320: document.addSubject(chart.getTitle().getText()); ingo@320: document.addCreationDate(); ingo@320: document.open(); ingo@320: ingo@320: PdfContentByte content = writer.getDirectContent(); ingo@320: ingo@320: int width = 0; ingo@320: int height = 0; ingo@320: if (landscape) { ingo@320: width = pageHeight; ingo@320: height = pageWidth; ingo@320: } ingo@320: else { ingo@320: width = pageWidth; ingo@320: height = pageHeight; ingo@320: } ingo@320: ingo@320: PdfTemplate template = content.createTemplate(width, height); ingo@320: Graphics2D graphics = template.createGraphics(width, height); ingo@320: Rectangle2D area = new Rectangle2D.Double(0.0D, 0.0D,width,height); ingo@320: ingo@320: chart.draw(graphics, area); ingo@320: graphics.dispose(); ingo@320: content.addTemplate(template, marginLeft, marginBottom); ingo@320: } ingo@320: catch (DocumentException de) { ingo@320: log.error("Error while exporting chart to pdf.", de); ingo@320: } ingo@320: finally { ingo@320: document.close(); ingo@320: } ingo@318: } ingo@640: ingo@640: ingo@640: public static void exportHistogramsAsPDF( ingo@640: OutputStream out, ingo@640: Chart[] histograms, ingo@640: String pageFormat, ingo@640: boolean landscape, ingo@640: float marginLeft, ingo@640: float marginRight, ingo@640: float marginTop, ingo@640: float marginBottom ingo@640: ) { ingo@640: log.info("export histogram as pdf."); ingo@640: ingo@640: if (pageFormat == null) ingo@640: pageFormat = DEFAULT_PAGE_SIZE; ingo@640: ingo@640: Rectangle page = PageSize.getRectangle(pageFormat); ingo@640: int pageWidth = ingo@640: (int) (page.getRight(marginRight) - page.getLeft(marginLeft)); ingo@640: int pageHeight = ingo@640: (int) (page.getTop(marginTop) - page.getBottom(marginBottom)); ingo@640: ingo@640: Document document = null; ingo@640: if (landscape) { ingo@640: document = new Document(page.rotate()); ingo@640: log.debug("Create landscape pdf."); ingo@640: } ingo@640: else ingo@640: document = new Document(page); ingo@640: ingo@640: try { ingo@640: PdfWriter writer = PdfWriter.getInstance(document, out); ingo@640: ingo@640: document.addCreationDate(); ingo@640: document.open(); ingo@640: ingo@640: PdfContentByte content = writer.getDirectContent(); ingo@640: ingo@640: int width = 0; ingo@640: int height = 0; ingo@640: if (landscape) { ingo@640: width = pageHeight; ingo@640: height = pageWidth; ingo@640: } ingo@640: else { ingo@640: width = pageWidth; ingo@640: height = pageHeight; ingo@640: } ingo@640: ingo@640: int size = histograms.length; ingo@640: for (int i = 0; i < size; i++) { ingo@640: if (i > 0) { ingo@640: document.newPage(); ingo@640: } ingo@640: ingo@640: JFreeChart chart = histograms[i].generateChart(); ingo@640: PdfTemplate template = content.createTemplate(width, height); ingo@640: Graphics2D graphics = template.createGraphics(width, height); ingo@640: Rectangle2D area = new Rectangle2D.Double( ingo@640: 0.0D, 0.0D,width,height); ingo@640: ingo@640: chart.draw(graphics, area); ingo@640: graphics.dispose(); ingo@640: content.addTemplate(template, marginLeft, marginBottom); ingo@640: } ingo@640: } ingo@640: catch (DocumentException de) { ingo@640: log.error("Error while exporting chart to pdf.", de); ingo@640: } ingo@640: finally { ingo@640: document.close(); ingo@640: } ingo@640: } ingo@318: } ingo@320: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf-8 :