htmlreporter.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.image.*;
import java.io.*;
import java.util.*;

import javax.imageio.*;

import org.apache.velocity.*;
import org.apache.velocity.app.*;
import org.jfree.chart.*;
import org.polepos.framework.*;


public class HTMLReporter extends GraphReporter {	
	private static final String TEMPLATEDIRNAME = "templates";
	public final static String ENCODING = "utf-8";
	public final static String OUTDIRNAME="doc/results/html";
	
	private File outdir=null;
	private List<Circuit> circuits=new ArrayList<Circuit>();
	private List<Lap> laps=new ArrayList<Lap>();
	private VelocityEngine engine=null;
	private Graph graph=null;
	
	protected void report(Graph graph) {
		try {
			Circuit oldcircuit=circuit();
			if(oldcircuit!=graph.circuit()) {
				if(oldcircuit==null) {
					setup();
				}
				renderCircuitPage();
				circuits.add(graph.circuit());
				laps.clear();
			}	
			this.graph=graph;
			laps.add(graph.lap());
			renderLapGraph(graph);			
			renderLapPage();
		} catch (Exception exc) {
			exc.printStackTrace();
		}	
	}

	private String lapFilePrefix() {
		return circuit().internalName()+"_"+lap().name();
	}

	protected void finish() {
		renderCircuitPage();
		renderIndexPage();
		copyStylesheet();
	}
	
	private void setup() throws Exception {
		outdir=new File(OUTDIRNAME);
		outdir.mkdirs();
		engine = new VelocityEngine();
		engine.setProperty( VelocityEngine.RUNTIME_LOG_LOGSYSTEM, this);
		engine.setProperty(VelocityEngine.ENCODING_DEFAULT,ENCODING);
		engine.setProperty(VelocityEngine.FILE_RESOURCE_LOADER_PATH,TEMPLATEDIRNAME);
		engine.init();
	}

	private void renderIndexPage() {
        List<TeamCar> distinct = new ArrayList<TeamCar>();
        for(TeamCar teamCar :graph.teamCars()){
            if(!distinct.contains(teamCar)){
                distinct.add(teamCar);
            }
        }
		VelocityContext context = new VelocityContext();
		context.put("includefile", "index.vhtml");
		context.put("teamcars", distinct);
		context.put("circuits", circuits);
		renderPage("index.html", context);
	}

	private void renderCircuitPage() {
		if(circuit()==null) {
			return;
		}
		VelocityContext context = new VelocityContext();
		context.put("includefile","circuitresults.vhtml");
		context.put("circuit",circuit());
		context.put("laps",laps);
		renderPage(circuit().internalName()+".html",context);
	}

	private void renderLapPage() {
		String lapfileprefix=lapFilePrefix();
		VelocityContext context = new VelocityContext();
		context.put("includefile","lapresult.vhtml");
		context.put("graph",graph);
		context.put("fileprefix",lapfileprefix);
		renderPage(lapfileprefix+".html",context);
	}

	private void renderLapGraph(Graph graph) throws IOException {
		JFreeChart chart=new ChartBuilder().createChart(graph);
		BufferedImage img=chart.createBufferedImage(750,500);
		ImageIO.write(img, "jpg", new File(outdir,lapFilePrefix()+".jpg"));
	}

	private void renderPage(String targetName,VelocityContext context) {
		BufferedWriter out=null;
		try {
			Template template=engine.getTemplate("baselayout.vhtml");
			out=new BufferedWriter(new OutputStreamWriter(new FileOutputStream(new File(outdir,targetName)),ENCODING));
			template.merge(context,out);
			out.close();
		} catch (Exception exc) {
			exc.printStackTrace();
		}
		finally {
			if(out!=null) {
				try {
					out.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
	}
	
	private void copyStylesheet() {
		File sourcefile=new File(new File(TEMPLATEDIRNAME),"style.css");
		File targetfile=new File(new File(OUTDIRNAME),"style.css");
		copyFile(sourcefile, targetfile);
	}

	private void copyFile(File sourcefile, File targetfile) {
		BufferedInputStream in=null;
		BufferedOutputStream out=null;
		try {
			in=new BufferedInputStream(new FileInputStream(sourcefile));
			out=new BufferedOutputStream(new FileOutputStream(targetfile));
			byte[] buffer=new byte[4096];
			int bytesread=0;
			while((bytesread=in.read(buffer))>=0) {
				out.write(buffer,0,bytesread);
			}
		} catch (IOException e) {
			e.printStackTrace();
		}
		finally {
			try {
				if(in!=null) {
					in.close();
				}
				if(out!=null) {
					out.close();
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}
	
	private Lap lap() {
		return getLast(laps);
	}

	private Circuit circuit() {
		return getLast(circuits);
	}
	
	private <Item> Item getLast(List<Item> list) {
		if(list.isEmpty()) {
			return null;
		}
		return list.get(list.size()-1);
	}
}