barchart.java

来自「测试工具」· Java 代码 · 共 110 行

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * 
 */
package org.apache.jmeter.testelement;

import java.awt.Dimension;
import java.awt.image.BufferedImage;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import javax.swing.JComponent;

import org.apache.jmeter.report.DataSet;
import org.apache.jmeter.visualizers.AxisGraph;
import org.apache.jmeter.visualizers.SamplingStatCalculator;

/**
 * The class is reponsible for returning 
 * @author pete
 *
 */
public class BarChart extends AbstractChart {

    public static final String REPORT_BAR_CHART_URL = "ReportChart.bar.chart.url";

    public BarChart() {
		super();
	}

    public String getURL() {
        return getPropertyAsString(REPORT_BAR_CHART_URL);
    }
    
    public void setURL(String url) {
        setProperty(REPORT_BAR_CHART_URL,url);
    }
    
    /**
     * Convert the data from SamplingStatCalculator to double array of array
     * @param data
     * @return
     */
    public double[][] convertToDouble(List data) {
        double[][] dataset = new double[1][data.size()];
        //Iterator itr = data.iterator();
        for (int idx=0; idx < data.size(); idx++) {
            SamplingStatCalculator stat = (SamplingStatCalculator)data.get(idx);
            dataset[0][idx] = getValue(stat);
        }
        return dataset;
    }
    
	public JComponent renderChart(List data) {
        ArrayList dset = new ArrayList();
        ArrayList xlabels = new ArrayList();
        Iterator itr = data.iterator();
        while (itr.hasNext()) {
            DataSet item = (DataSet)itr.next();
            SamplingStatCalculator ss = item.getStatistics(this.getURL());
            if (ss != null) {
                // we add the entry
                dset.add(ss);
                if ( getXLabel().equals(X_DATA_FILENAME_LABEL) ) {
                    xlabels.add(item.getDataSourceName());
                } else {
                    xlabels.add(item.getMonthDayYearDate());
                }
            }
        }
        double[][] dbset = convertToDouble(dset);
		return renderGraphics(dbset, (String[])xlabels.toArray(new String[xlabels.size()]));
	}
    
    public JComponent renderGraphics(double[][] data, String[] xAxisLabels) {
        AxisGraph panel = new AxisGraph();
        panel.setTitle(this.getTitle());
        panel.setData(data);
        panel.setXAxisLabels(xAxisLabels);
        panel.setYAxisLabels(this.getYLabel());
        panel.setXAxisTitle(this.getFormattedXAxis());
        panel.setYAxisTitle(this.getYAxis());
        // we should make this configurable eventually
        int width = getWidth();
        int height = getHeight();
        panel.setPreferredSize(new Dimension(width,height));
        panel.setSize(new Dimension(width,height));
        panel.setWidth(width);
        panel.setHeight(width);
        setBufferedImage(new BufferedImage(width,height,BufferedImage.TYPE_INT_RGB));
        panel.paintComponent(this.getBufferedImage().createGraphics());
        return panel;
    }
    
}