pdfreporter.java

个人感觉非常不错的poleperst-acid

/* 
This file is part of the PolePosition database benchmark
http://www.polepos.org

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public
License along with this program; if not, write to the Free
Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
MA  02111-1307, USA. */

package org.polepos.reporters;

import java.awt.*;
import java.awt.Font;
import java.io.*;
import java.util.*;
import java.util.List;

import org.jfree.chart.*;
import org.polepos.framework.*;

import com.lowagie.text.*;
import com.lowagie.text.pdf.*;


public class PDFReporter extends GraphReporter {
	private Circuit mCircuit;
	private Document mDocument;
	private PdfWriter mWriter;
    
    private com.lowagie.text.Font h1Font = FontFactory.getFont(FontFactory.HELVETICA,15,Font.BOLD);
    private com.lowagie.text.Font h2Font = FontFactory.getFont(FontFactory.HELVETICA,12,Font.BOLD);
    
    private com.lowagie.text.Font bigFont = FontFactory.getFont(FontFactory.HELVETICA,10,Font.BOLD);
    private com.lowagie.text.Font smallFont = FontFactory.getFont(FontFactory.HELVETICA,9,Font.PLAIN);
    
	
    protected void report(Graph graph) {
        
		if(mDocument==null) {
	        setupDocument(PATH);
            try {
                renderFirstPage(graph);
            } catch (DocumentException e) {
                e.printStackTrace();
            }
		}
		if(mDocument==null) {
			return;
		}
        
        Circuit circuit = graph.circuit();
        if(! circuit.equals(mCircuit)){
            mCircuit = circuit;
        }
        
		JFreeChart chart = new ChartBuilder().createChart(graph);
		
        chart.setBackgroundPaint(null);
        try {
			renderTable(graph);
			renderChart(chart);
			mDocument.newPage();
		} catch (Exception e) {
			e.printStackTrace();
		}
    }
    
    private void renderFirstPage(Graph graph) throws DocumentException{
        if(mDocument == null){
            return;
        }
        
        Paragraph para=new Paragraph();
        para.add(new Chunk("PolePosition\n",h1Font));
        para.add(new Chunk("the open source database benchmark\n",smallFont));
        para.add(linked(new Chunk(WEBSITE + "\n\n\n", smallFont), WEBSITE));
        para.add(new Chunk("Participating teams\n\n",h2Font));
        
        mDocument.add(para);
        
        List printed = new ArrayList();
        
        for(TeamCar teamCar :graph.teamCars()){
            Team team = teamCar.getTeam();
            String webSite = team.website();
            if(webSite != null){
                if(! printed.contains(team)){
                    printed.add(team);
                    renderTeam(team.name(), team.description(), webSite);
                }
            }else{
                Car car = teamCar.getCar();
                webSite = car.getWebsite();
                if(webSite != null){
                    if(! printed.contains(car)){
                        printed.add(car);
                        renderTeam(car.name(), car.description(), webSite);
                    }
                }
            }
        }
        mDocument.newPage();
    }
    
    private void renderTeam(String name, String description, String website) throws DocumentException{
        Paragraph para=new Paragraph();
        para.add(linked(new Chunk(name + "\n",bigFont), website));
        if(description != null){
            para.add(linked(new Chunk(description + "\n",smallFont), website));
        }
        if(website != null){
            para.add(linked(new Chunk(website + "\n", smallFont), website));
        }
        para.add(new Chunk("\n",smallFont));
        mDocument.add(para);
    }
    
    private Element linked(Chunk chunk, String link){
        if(link == null){
            return chunk;
        }
        Anchor anchor = new Anchor(chunk);
        anchor.setReference(link);
        return anchor;
    }
    
    

	private void setupDocument(String path) {
		String fileName = path + "/" + "PolePosition.pdf";
		File file = new File(fileName);
		file.delete();
		mDocument = new Document();
		try {
			mWriter = PdfWriter.getInstance(mDocument, new FileOutputStream(file));
			mDocument.open();
		} catch (Exception exc) {
			exc.printStackTrace();
			mDocument=null;
		}
	}

    @Override
	public void endSeason() {
        super.endSeason();
        if(mDocument != null){
            mDocument.close();
        }
        if(mWriter != null){
            mWriter.close();
        }
	}

	private void renderTable(Graph graph) throws DocumentException {
		Paragraph para=new Paragraph();
		// para.setAlignment(Element.ALIGN_CENTER);
        
        Circuit circuit = graph.circuit();
        Lap lap = graph.lap();
        
        para.add(new Chunk("Circuit: " + circuit.name()+ "\n",bigFont));
        para.add(new Chunk(circuit.description() + "\n",smallFont));
        para.add(new Chunk("Lap: " + lap.name()+ "\n\n",bigFont));
        
		List<TeamCar> teamCars=graph.teamCars();
		List<TurnSetup> setups=graph.setups();
		Table table = setupTable(graph);
		int idx=1;
        addTableCell(table, 0, 0,"t [time in ms]" , null,false,true);
		for(TurnSetup setup : setups) {
            StringBuffer header = new StringBuffer();
            boolean first = true;
            for(SetupProperty sp : setup.properties()){
                if(! first){
                    header.append("\n");
                }
                String name = sp.name();
                if(! name.equals("commitinterval")){
                    header.append(name);
                    header.append(":");
                    header.append(sp.value());
                    first = false;
                }
            }
			addTableCell(table, idx, 0, header.toString(),null, true,true);
			idx++;
		}
		table.endHeaders();
		int vidx=1;
		for(TeamCar teamCar : teamCars) {
			addTableCell(table,0,vidx,teamCar.toString(),teamCar.website(),true,false);
			int hidx=1;
			for(TurnSetup setup : setups) {
				String text=String.valueOf(graph.timeFor(teamCar,setup));
				addTableCell(table,hidx,vidx,text, null,false,false);
				hidx++;
			}
			vidx++;
		}
		para.add(table);
        para.add(new Chunk("\n",bigFont));
        mDocument.add(para);
	}

	private void addTableCell(Table table, int hidx, int vidx, String text, String link, boolean bold,boolean header) throws BadElementException {
        Chunk chunk = new Chunk(text,FontFactory.getFont(FontFactory.HELVETICA,9,(bold ? Font.BOLD : Font.PLAIN)));
        chunk.setTextRise(3);
		Cell cell=new Cell(linked(chunk, link));
		cell.setHeader(header);
        if(! header){
            cell.setNoWrap(true);
        }
        cell.setVerticalAlignment(Element.ALIGN_CENTER);
		table.addCell(cell,new Point(vidx,hidx));
	}
	
	private void renderChart(JFreeChart chart) throws DocumentException, BadElementException {
        int width = 500;
        int height = 300;
		PdfContentByte cb = mWriter.getDirectContent();
		PdfTemplate tp = cb.createTemplate(width, height);
		Graphics2D graphics = tp.createGraphics(width, height, new DefaultFontMapper());
		java.awt.Rectangle area = new java.awt.Rectangle(0, 0, width, height);
		chart.draw(graphics, area);
		graphics.dispose();
		ImgTemplate imgtmpl=new ImgTemplate(tp);
		//imgtmpl.setAlignment(Element.ALIGN_CENTER);
		mDocument.add(imgtmpl);
	}

	private Table setupTable(Graph graph) throws BadElementException {
		Table table=new Table(graph.setups().size()+1);
		table.setAutoFillEmptyCells(true);
		table.setSpaceInsideCell(2);
        
		table.setBorderWidth(0);
		table.setDefaultCellBorder(1);
		table.setTableFitsPage(true);
		return table;
	}

	protected void finish() {
	}
}