org.swtchart/src/org/swtchart/Chart.java

Go to the documentation of this file.

/*******************************************************************************
 * Copyright (c) 2008-2014 SWTChart project. All rights reserved. 
 * 
 * This code is distributed under the terms of the Eclipse Public License v1.0
 * which is available at http://www.eclipse.org/legal/epl-v10.html
 *******************************************************************************/
package org.swtchart;

import org.eclipse.swt.SWT;
import org.eclipse.swt.graphics.Color;
import org.eclipse.swt.graphics.GC;
import org.eclipse.swt.graphics.Image;
import org.eclipse.swt.graphics.ImageData;
import org.eclipse.swt.graphics.ImageLoader;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Control;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.Event;
import org.eclipse.swt.widgets.Listener;
import org.swtchart.internal.ChartLayout;
import org.swtchart.internal.ChartLayoutData;
import org.swtchart.internal.ChartTitle;
import org.swtchart.internal.Legend;
import org.swtchart.internal.PlotArea;
import org.swtchart.internal.Title;
import org.swtchart.internal.axis.AxisSet;
import org.swtchart.internal.series.SeriesSet;

public class Chart extends Composite implements Listener {

    private Title title;

    private Legend legend;

    private AxisSet axisSet;

    private PlotArea plotArea;

    private int orientation;

    private boolean compressEnabled;

    private boolean updateSuspended;

    public Chart(Composite parent, int style) {
        super(parent, style | SWT.DOUBLE_BUFFERED);

        orientation = SWT.HORIZONTAL;
        compressEnabled = true;
        updateSuspended = false;

        parent.layout();

        setLayout(new ChartLayout());

        title = new ChartTitle(this);
        title.setLayoutData(new ChartLayoutData(SWT.DEFAULT, 100));
        legend = new Legend(this, SWT.NONE);
        legend.setLayoutData(new ChartLayoutData(200, SWT.DEFAULT));
        plotArea = new PlotArea(this, SWT.NONE);
        axisSet = new AxisSet(this);

        updateLayout();

        addListener(SWT.Resize, this);
    }

    public ITitle getTitle() {
        return title;
    }

    public ILegend getLegend() {
        return legend;
    }

    public IAxisSet getAxisSet() {
        return axisSet;
    }

    public Composite getPlotArea() {
        return plotArea;
    }

    public ISeriesSet getSeriesSet() {
        return plotArea.getSeriesSet();
    }

    /*
     * @see Control#setBackground(Color)
     */
    @Override
    public void setBackground(Color color) {
        super.setBackground(color);

        for (Control child : getChildren()) {
            if (!(child instanceof PlotArea) && !(child instanceof Legend)) {
                child.setBackground(color);
            }
        }
    }

    public Color getBackgroundInPlotArea() {
        return plotArea.getBackground();
    }

    public void setBackgroundInPlotArea(Color color) {
        if (color != null && color.isDisposed()) {
            SWT.error(SWT.ERROR_INVALID_ARGUMENT);
        }
        plotArea.setBackground(color);
    }

    @Override
    public void setOrientation(int orientation) {
        if (orientation == SWT.HORIZONTAL || orientation == SWT.VERTICAL) {
            this.orientation = orientation;
        }
        updateLayout();
    }

    @Override
    public int getOrientation() {
        return orientation;
    }

    public void enableCompress(boolean enabled) {
        compressEnabled = enabled;
    }

    public boolean isCompressEnabled() {
        return compressEnabled;
    }

    public void suspendUpdate(boolean suspend) {
        if (updateSuspended == suspend) {
            return;
        }
        updateSuspended = suspend;

        // make sure that chart is updated
        if (!suspend) {
            updateLayout();
            ((SeriesSet) getSeriesSet()).updateStackAndRiserData();
        }
    }

    public boolean isUpdateSuspended() {
        return updateSuspended;
    }

    /*
     * @see Listener#handleEvent(Event)
     */
    public void handleEvent(Event event) {
        switch (event.type) {
        case SWT.Resize:
            updateLayout();
            redraw();
            break;
        default:
            break;
        }
    }

    public void updateLayout() {
        if (updateSuspended) {
            return;
        }

        if (legend != null) {
            legend.updateLayoutData();
        }

        if (title != null) {
            title.updateLayoutData();
        }

        if (axisSet != null) {
            axisSet.updateLayoutData();
        }

        layout();

        if (axisSet != null) {
            axisSet.refresh();
        }
    }

    /*
     * @see Control#update()
     */
    @Override
    public void update() {
        super.update();
        for (Control child : getChildren()) {
            child.update();
        }
    }

    /*
     * @see Widget#dispose()
     */
    @Override
    public void dispose() {
        title.dispose();
        legend.dispose();
        axisSet.dispose();
        plotArea.dispose();
        super.dispose();
    }

    /*
     * @see Control#redraw()
     */
    @Override
    public void redraw() {
        super.redraw();
        for (Control child : getChildren()) {
            child.redraw();
        }
    }

    public void save(String filename, int format) {
        Image image = new Image(Display.getDefault(), getBounds());

        GC gc = new GC(image);
        print(gc);
        gc.dispose();

        ImageData data = image.getImageData();
        ImageLoader loader = new ImageLoader();
        loader.data = new ImageData[] { data };
        loader.save(filename, format);
        image.dispose();
    }

    public void renderOffscreenImage(Image image) {
        GC gc = new GC(image);
        print(gc);
        gc.dispose();
    }
}

---