uk.co.petertribble.jkstat.gui.KstatSetChart.java Source code

Introduction

Here is the source code for uk.co.petertribble.jkstat.gui.KstatSetChart.java

Source

/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (the "License").  You may not use this file except in compliance
 * with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

package uk.co.petertribble.jkstat.gui;

import java.util.Map;
import java.util.HashMap;
import java.util.List;
import java.util.ArrayList;
import java.util.Date;
import uk.co.petertribble.jkstat.api.*;

import org.jfree.chart.ChartFactory;
import org.jfree.data.time.*;

/**
 * A graphical chart of a KstatSet, showing one or multiple statistics.
 *
 * @author Peter Tribble
 */
public class KstatSetChart extends KstatBaseChart {

    private KstatSet kss;
    private TimeSeriesCollection dataset;
    private Map<String, TimeSeries> tsmap;
    private Map<Kstat, ChartableKstat> kMap;

    /**
     * Create a Chart of the given statistic.
     *
     * @param jkstat a {@code JKstat}
     * @param kss the {@code KstatSet} supplying the data
     * @param statistic the statistic to be charted
     * @param showdelta if true, show rates, else show absolute values
     */
    public KstatSetChart(JKstat jkstat, KstatSet kss, String statistic, boolean showdelta) {
        this.jkstat = jkstat;
        this.kss = kss;
        this.showdelta = showdelta;
        init(statistic);
    }

    /**
     * Create a Chart of the given statistics.
     *
     * @param jkstat a {@code JKstat}
     * @param kss the {@code KstatSet} supplying the data
     * @param statistics the statistics to be charted
     * @param showdelta if true, show rates, else show absolute values
     */
    public KstatSetChart(JKstat jkstat, KstatSet kss, List<String> statistics, boolean showdelta) {
        this.jkstat = jkstat;
        this.kss = kss;
        this.showdelta = showdelta;
        init(statistics);
    }

    private void init(String statistic) {
        List<String> statistics = new ArrayList<String>();
        statistics.add(statistic);
        init(statistics);
    }

    private void init(List<String> statistics) {
        dataset = new TimeSeriesCollection();
        tsmap = new HashMap<String, TimeSeries>();
        kMap = new HashMap<Kstat, ChartableKstat>();

        for (Kstat ks : kss.getKstats()) {
            kMap.put(ks, new ChartableKstat(jkstat, ks));

            for (String statistic : KstatUtil.numericStatistics(jkstat, ks)) {
                String s = ks.getTriplet() + ":" + statistic;
                tsmap.put(s, new TimeSeries(s));
            }
            for (String statistic : statistics) {
                dataset.addSeries(tsmap.get(ks.getTriplet() + ":" + statistic));
            }
        }

        if (jkstat instanceof SequencedJKstat) {
            readAll(((SequencedJKstat) jkstat).newInstance());
        } else {
            setMaxAge(maxage);
            updateAccessory();
        }

        String ylabel = showdelta ? KstatResources.getString("CHART.RATE")
                : KstatResources.getString("CHART.VALUE");

        chart = ChartFactory.createTimeSeriesChart(kss.toString() == null ? "statistic" : kss.toString(),
                KstatResources.getString("CHART.TIME"), ylabel, dataset, true, true, false);

        setAxes();

        if (!(jkstat instanceof SequencedJKstat)) {
            startLoop();
        }
    }

    @Override
    public void addStatistic(String statistic) {
        for (Kstat ks : kss.getKstats()) {
            dataset.addSeries(tsmap.get(ks.getTriplet() + ":" + statistic));
        }
    }

    @Override
    public void removeStatistic(String statistic) {
        for (Kstat ks : kss.getKstats()) {
            dataset.removeSeries(tsmap.get(ks.getTriplet() + ":" + statistic));
        }
    }

    @Override
    public void setMaxAge(int maxage) {
        this.maxage = maxage;
        for (String s : tsmap.keySet()) {
            tsmap.get(s).setMaximumItemAge(maxage);
        }
    }

    /*
     * read all the data from the kstat sequence
     */
    private void readAll(SequencedJKstat sjkstat) {
        for (ChartableKstat ck : kMap.values()) {
            ck.setJKstat(sjkstat);
        }
        do {
            // FIXME update the KstatSet if it changes
            for (Kstat ks : kss.getKstats()) {
                if (sjkstat.getKstat(ks) != null) {
                    readOne(ks, new Millisecond(new Date(sjkstat.getTime())));
                }
            }
        } while (sjkstat.next());
    }

    @Override
    public void updateAccessory() {
        for (Kstat ks : kss.getKstats()) {
            readOne(ks, new Millisecond());
        }
    }

    /*
     * Get and update the appropriate data.
     */
    private void readOne(Kstat ks, Millisecond ms) {
        ChartableKstat cks = kMap.get(ks);
        cks.update();
        for (String statistic : KstatUtil.numericStatistics(jkstat, ks)) {
            tsmap.get(ks.getTriplet() + ":" + statistic).add(ms,
                    showdelta ? cks.getRate(statistic) : (double) cks.getValue(statistic));
        }
    }
}