package charts;

import java.awt.Color;
import java.awt.Font;
import java.io.IOException;
import java.io.PrintWriter;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import javax.servlet.http.HttpSession;
import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartRenderingInfo;
import org.jfree.chart.ChartUtilities;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.DateAxis;
import org.jfree.chart.axis.NumberAxis;
import org.jfree.chart.block.BlockContainer;
import org.jfree.chart.block.BorderArrangement;
import org.jfree.chart.block.EmptyBlock;
import org.jfree.chart.entity.StandardEntityCollection;
import org.jfree.chart.labels.StandardXYToolTipGenerator;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.renderer.xy.XYItemRenderer;
import org.jfree.chart.renderer.xy.XYLineAndShapeRenderer;
import org.jfree.chart.servlet.ServletUtilities;
import org.jfree.chart.title.CompositeTitle;
import org.jfree.chart.title.LegendTitle;
import org.jfree.chart.title.TextTitle;
import org.jfree.data.time.Month;
import org.jfree.data.time.TimeSeries;
import org.jfree.data.time.TimeSeriesCollection;
import org.jfree.data.xy.XYDataset;
import org.jfree.ui.RectangleEdge;
public class MultiChart
{
	 /**
     * 构造报表的结果集
     * @return
     */
    private static XYDataset createDataset1() {
        TimeSeries s1 = new TimeSeries("居民中长期/中长期贷款", Month.class);
        s1.add(new Month(1, 2007), 0.51);
        s1.add(new Month(2, 2007), 0.26);
        s1.add(new Month(3, 2007), 0.22);
        s1.add(new Month(4, 2007), 0.27);
        s1.add(new Month(5, 2007), 0.26);
        s1.add(new Month(6, 2007), 0.24);
        s1.add(new Month(7, 2007), 0.42);
      
        
        TimeSeriesCollection dataset = new TimeSeriesCollection();
        dataset.addSeries(s1);
        
        return dataset;
    }
    /**
     * 构造报表的结果集
     * @return
     */
    private static XYDataset createDataset2() {
        TimeSeries s2 = new TimeSeries("居民贷款/非金融贷款", Month.class);
        s2.add(new Month(1, 2007), 0.18);
        s2.add(new Month(2, 2007), 0.20);
        s2.add(new Month(3, 2007), 0.24);
        s2.add(new Month(4, 2007), 0.30);
        s2.add(new Month(5, 2007), 0.35);
        s2.add(new Month(6, 2007), 0.38);
        s2.add(new Month(7, 2007), 0.48);
        
        TimeSeriesCollection dataset = new TimeSeriesCollection();
        dataset.addSeries(s2);
        
        return dataset;

    }
    /**
     * 根据结果集构造JFreechart报表对象
     * @param 
     * @return
     */
    private static JFreeChart createChart(XYDataset dataSet1,XYDataset dataSet2) {
       
        JFreeChart jfreechart = ChartFactory.createTimeSeriesChart(
            "居民类贷款占比比较", 
            "日期", 
            "比例",
            dataSet1, 
            false,     
            true,    
            false      
        );
        jfreechart.addSubtitle(new TextTitle("------《2008年银行业行业报告》", new Font(
                "Dialog", Font.ITALIC, 10)));
        
        XYPlot plot = (XYPlot) jfreechart.getPlot();
        NumberAxis axis1 = (NumberAxis) plot.getRangeAxis();
        axis1.setNumberFormatOverride(new DecimalFormat("0.0%"));
        
        NumberAxis axis2 = new NumberAxis("比例");
        axis2.setAutoRangeIncludesZero(false);
        axis2.setNumberFormatOverride(new DecimalFormat("0.0%"));
        
        plot.setRangeAxis(1, axis2);
        plot.setDataset(1, dataSet2);
        plot.mapDatasetToRangeAxis(1, 1);
        XYItemRenderer renderer = plot.getRenderer();
        renderer.setBaseToolTipGenerator(
                StandardXYToolTipGenerator.getTimeSeriesInstance());
        if (renderer instanceof XYLineAndShapeRenderer) {
            XYLineAndShapeRenderer rr = (XYLineAndShapeRenderer) renderer;
            rr.setBaseShapesVisible(true);
            rr.setBaseShapesFilled(true);
        }
        
        XYLineAndShapeRenderer renderer2 = new XYLineAndShapeRenderer();
        renderer2.setSeriesPaint(0, Color.black);
        renderer2.setBaseShapesVisible(true);
        renderer2.setBaseToolTipGenerator(
                StandardXYToolTipGenerator.getTimeSeriesInstance());
        plot.setRenderer(1, renderer2);
              
        DateAxis axis = (DateAxis) plot.getDomainAxis();
        axis.setDateFormatOverride(new SimpleDateFormat("yyyy-MMM"));

        //创建两个图例，并分别放在两条坐标轴下
        LegendTitle legend1 = new LegendTitle(renderer);
        LegendTitle legend2 = new LegendTitle(renderer2);
        BlockContainer container = new BlockContainer(new BorderArrangement());
        container.add(legend1, RectangleEdge.LEFT);
        container.add(legend2, RectangleEdge.RIGHT);
        container.add(new EmptyBlock(2000, 0));
        CompositeTitle legends = new CompositeTitle(container);
        legends.setPosition(RectangleEdge.BOTTOM);
        jfreechart.addSubtitle(legends);
        
        return jfreechart;
    }
    /**
     * 生成折线图的图片，返回图片文件的名称
     * @param session
     * @param pw
     * @return
     */
    public static String generateMultiChart(HttpSession session, PrintWriter pw)
    {
        String filename = null;
        XYDataset dataset1 = createDataset1();
        XYDataset dataset2 = createDataset2();
        JFreeChart chart = createChart(dataset1,dataset2);
        ChartRenderingInfo info = new ChartRenderingInfo(
                new StandardEntityCollection());
        try
        {
            filename = ServletUtilities.saveChartAsPNG(chart, 600, 350, info,
                    session);
        }
        catch (IOException e)
        {
            e.printStackTrace();
        }
        try
        {
            ChartUtilities.writeImageMap(pw, filename, info, false);
        }
        catch (IOException e)
        {
            e.printStackTrace();
        }
        pw.flush();
        return filename;
    }
}