import org.csstudio.swt.xygraph.dataprovider.CircularBufferDataProvider;
import org.csstudio.swt.xygraph.dataprovider.Sample;
import org.csstudio.swt.xygraph.figures.ToolbarArmedXYGraph;
import org.csstudio.swt.xygraph.figures.Trace;
import org.csstudio.swt.xygraph.figures.Trace.TraceType;
import org.csstudio.swt.xygraph.figures.XYGraph;
import org.csstudio.swt.xygraph.figures.XYGraphFlags;
import org.eclipse.draw2d.LightweightSystem;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.Shell;
import org.junit.Test;

/** JUnit demo of the plot with 'staircase' data, including gaps,
 *  as it can be used for showing historic data.
 *  @author Kay Kasemir
 */
@SuppressWarnings("nls")
public class StaircaseTest
{
    private static int next_x = 1;

    @Test
    public void testStaircasePlot()
    {
        // Main window (shell)
        final Shell shell = new Shell();
        shell.setSize(800, 500);
        shell.open();
        
        // XYGraph
        final LightweightSystem lws = new LightweightSystem(shell);
        final ToolbarArmedXYGraph plot = new ToolbarArmedXYGraph(new XYGraph(),
                XYGraphFlags.SEPARATE_ZOOM | XYGraphFlags.STAGGER);
        final XYGraph xygraph = plot.getXYGraph();
        xygraph.setTransparent(false);
        xygraph.setTitle("You should see a line. Zoom out to see more data");
        lws.setContents(plot);

        // Add data & trace
        final CircularBufferDataProvider data = new CircularBufferDataProvider(true);
        data.addSample(new Sample(next_x++, 1, 1, 1, 0, 0));
        data.addSample(new Sample(next_x++, 2, 1, 1, 0, 0));
        data.addSample(new Sample(next_x++, 3, 1, 1, 0, 0));
        data.addSample(new Sample(next_x++, 1, 1, 1, 0, 0));
        data.addSample(new Sample(next_x++, 1, 1, 1, 0, 0));
        data.addSample(new Sample(next_x++, 1, 1, 1, 0, 0));
        // Add Double.NaN gap, single point
        data.addSample(new Sample(next_x++, Double.NaN, 0, 0, 0, 0, "Disconnected"));
        data.addSample(new Sample(next_x++, 1, 0, 0, 0, 0));
        // Another gap, single point
        data.addSample(new Sample(next_x++, Double.NaN, 0, 0, 0, 0, "Disconnected"));
        data.addSample(new Sample(next_x++, 2, 0, 0, 0, 0));
        // Last value is valid 'forever'
        data.addSample(new Sample(Double.MAX_VALUE, 2, 0, 0, 0, 0));

        // Always looked OK with this range
        xygraph.primaryXAxis.setRange(data.getXDataMinMax());
        xygraph.primaryYAxis.setRange(data.getYDataMinMax());
        
        // With STEP_HORIZONTALLY this should have shown just a horizontal
        // line, but a bug resulted in nothing when both the 'start' and 'end'
        // point of the horizontal line were outside the plot range.
        // Similarly, using STEP_VERTICALLY failed to draw anything when
        // both end-points of the horizontal or vertical section of the step
        // were outside the plot. (fixed)
        // 
        // There's still a question about handling 'YErrorInArea':
        // For now, the axis intersection removes x/y errors,
        // so when moving a sample with y error left or right outside
        // of the plot range, the error area suddenly shrinks when
        // the axis intersection is assumed to have +-0 y error.
        xygraph.primaryXAxis.setRange(4.1, 4.9);
        
        // Gap, start of X range, sample @ x==8, gap @ 9, end of range.
        // Bug failed to show line from that sample up to gap @ 9.
        xygraph.primaryXAxis.setRange(7.5, 9.5);
        
        final Trace trace = new Trace("Demo", xygraph.primaryXAxis,
                xygraph.primaryYAxis, data);
        trace.setTraceType(TraceType.STEP_HORIZONTALLY);
//        trace.setTraceType(TraceType.STEP_VERTICALLY);
        
//        // SOLID_LINE does not show individual points
//        trace.setTraceType(TraceType.SOLID_LINE);
//        trace.setPointStyle(PointStyle.CIRCLE);
        
        trace.setErrorBarEnabled(true);
        trace.setDrawYErrorInArea(true);
        xygraph.addTrace(trace);        
        
        // SWT main loop
        final Display display = Display.getDefault();
        while (!shell.isDisposed()) {
          if (!display.readAndDispatch())
            display.sleep();
        }
    }
}