XYGraphStyledDoubleExampleView.java

/*******************************************************************************
 * Copyright (c) 2016 Bernhard Wedl and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Bernhard Wedl - initial API and implementation
 *******************************************************************************/

package org.eclipse.nebula.visualization.xygraph.exampleview;

import org.eclipse.draw2d.LightweightSystem;
import org.eclipse.nebula.visualization.xygraph.dataprovider.CircularBufferDataProvider;
import org.eclipse.nebula.visualization.xygraph.dataprovider.ISample;
import org.eclipse.nebula.visualization.xygraph.figures.IXYGraph;
import org.eclipse.nebula.visualization.xygraph.figures.ToolbarArmedXYGraph;
import org.eclipse.nebula.visualization.xygraph.figures.Trace;
import org.eclipse.nebula.visualization.xygraph.figures.Trace.PointStyle;
import org.eclipse.nebula.visualization.xygraph.figures.Trace.TraceType;
import org.eclipse.nebula.visualization.xygraph.figures.XYGraph;
import org.eclipse.nebula.visualization.xygraph.styleprovider.BasePointStyleProvider;
import org.eclipse.nebula.visualization.xygraph.util.XYGraphMediaFactory;
import org.eclipse.swt.SWT;
import org.eclipse.swt.graphics.Color;
import org.eclipse.swt.widgets.Canvas;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.ui.part.ViewPart;

public class XYGraphStyledDoubleExampleView extends ViewPart {

    public XYGraphStyledDoubleExampleView() {
    }

    class Rainbowstyle extends BasePointStyleProvider {

        @Override
        public Color getPointColor(ISample sample, Trace trace) {

            if (sample == null)
                return super.getPointColor(sample, trace);

            if (sample.getXValue() < 0.1)
                return XYGraphMediaFactory.getInstance().getColor(XYGraphMediaFactory.COLOR_PURPLE);

            if (sample.getXValue() < 0.2)
                return XYGraphMediaFactory.getInstance().getColor(XYGraphMediaFactory.COLOR_BLUE);

            if (sample.getXValue() < 0.3)
                return XYGraphMediaFactory.getInstance().getColor(XYGraphMediaFactory.COLOR_GREEN);

            if (sample.getXValue() < 0.4)
                return XYGraphMediaFactory.getInstance().getColor(XYGraphMediaFactory.COLOR_YELLOW);

            if (sample.getXValue() < 0.5)
                return XYGraphMediaFactory.getInstance().getColor(XYGraphMediaFactory.COLOR_ORANGE);

            if (sample.getXValue() < 0.6)
                return XYGraphMediaFactory.getInstance().getColor(XYGraphMediaFactory.COLOR_RED);

            return XYGraphMediaFactory.getInstance().getColor(XYGraphMediaFactory.COLOR_WHITE);

        }
    }

    @Override
    public void createPartControl(Composite parent) {

        // use LightweightSystem to create the bridge between SWT and draw2D
        final LightweightSystem lws = new LightweightSystem(new Canvas(parent, SWT.NONE));

		// create a new XY Graph.
		IXYGraph xyGraph = new XYGraph();

		ToolbarArmedXYGraph toolbarArmedXYGraph = new ToolbarArmedXYGraph(xyGraph);

        xyGraph.setTitle("Simple Styled Example");
        // set it as the content of LightwightSystem
        lws.setContents(toolbarArmedXYGraph);

        // create a trace data provider, which will provide the data to the trace.
        CircularBufferDataProvider traceDataProvider = new CircularBufferDataProvider(false);
        traceDataProvider.setBufferSize(100);
        traceDataProvider.setCurrentXDataArray(new double[] { 0, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6 });
        traceDataProvider.setCurrentYDataArray(new double[] { 1, 1, 1, 1, 1, 1, 1 });

        // create the trace
        Trace trace = new Trace("Trace-Styled XY Plot", xyGraph.getPrimaryXAxis(), xyGraph.getPrimaryYAxis(), traceDataProvider);

        // set trace property
        trace.setTraceColor(XYGraphMediaFactory.getInstance().getColor(XYGraphMediaFactory.COLOR_GRAY));
        trace.setTraceType(TraceType.POINT);

        // set point properties
        trace.setPointStyle(PointStyle.FILLED_SQUARE);
        trace.setPointSize(40);
        trace.setPointStyleProvider(new Rainbowstyle());

        // add the trace to xyGraph
        xyGraph.addTrace(trace);

        // perform AutoScale
        xyGraph.performAutoScale();

    }

    @Override
    public void setFocus() {

    }

}