import java.awt.*;
import java.awt.event.*;
import javax.swing.*;
import ChartDirector.*;

public class waterfall implements DemoModule
{
    //Name of demo program
    public String toString() { return "Waterfall Chart"; }

    //Number of charts produced in this demo
    public int getNoOfCharts() { return 1; }

    //Main code for creating charts
    public void createChart(ChartViewer viewer, int index)
    {
        // 4 data points to represent the cash flow for the Q1 - Q4
        double[] data = {230, 140, 220, 330, 150};
		ArrayMath lineData = new ArrayMath(data);
		lineData.acc();
		double scaleFactor = lineData.max() / 100;
		if (scaleFactor == 0) {
			// Avoid division by zero error for zero data
			scaleFactor = 1;
		}
		lineData.div2(scaleFactor);

        // We want to plot a waterfall chart showing the 4 quarters as well as the
        // total
        String[] labels = {"Product 1", "Product 2", "Product 3", "Product 4",
            "Product 5", "Total"};
		String[] labels1 = null;
 
		XYChart c = new XYChart(600, 350, 0xFFFFFF, 0xFFFFFF, 0);
	   // Add a line layer for the pareto line
		 LineLayer lineLayer = c.addLineLayer2();
		 // Add the pareto line using deep blue (0000ff) as the color, with circle
		 // symbols
		 lineLayer.addDataSet(lineData.result(), 0x000000).setDataSymbol(Chart.CircleShape, 9, 0x000000, 0x000000);
		 // Set the line width to 2 pixel
		 lineLayer.setLineWidth(2);
		 // Bind the line layer to the secondary (right) y-axis.
		 lineLayer.setUseYAxis2();
		 c.yAxis2().setTitle("Cumulative % of Total Benefits");
		 c.yAxis2().setLabelFormat("{value}%");
			 // The top side of the bars in a waterfall chart is the accumulated data. We
			  // use the ChartDirector ArrayMath utility to accumulate the data. The
			  // "total" is handlsed by inserting a zero point at the end before
			  // accumulation (after accumulation it will become the total).
			  double[] boxTop = new ArrayMath(data).insert2(0, 1).acc().result();

			  // The botom side of the bars is just the top side of the previous bar. So we
			  // shifted the top side data to obtain the bottom side data.
			  double[] boxBottom = new ArrayMath(boxTop).shift(1, 0).result();

			  // The last point (total) is different. Its bottom side is always 0.
			  boxBottom[boxBottom.length - 1] = 0;

			  // Create a XYChart object of size 500 x 280 pixels. Set background color to
			  // light blue (ccccff), with 1 pixel 3D border effect.

			  // Add a title to the chart using 13 points Arial Bold Itatic font, with
			  // white (ffffff) text on a deep blue (0x80) background
		   //   c.addTitle("Top Business Benefits", "Arial Bold", 13, 0x000000);

			  // Set the plotarea at (55, 50) and of size 430 x 215 pixels. Use alternative
			  // white/grey background.
			  c.setPlotArea(85, 45, 450, 215, Chart.Transparent, Chart.Transparent);

			  // Set the labels on the x axis using Arial Bold font
			  c.xAxis().setLabels(labels1).setFontStyle("Arial",8);
			  c.xAxis().setIndent(true);

			  // Set the x-axis ticks and grid lines to be between the bars
			  c.xAxis().setTickOffset(0.5);

			  // Use Arial Bold as the y axis label font
			  c.yAxis().setLabelStyle("Arial",8);
	   c.addLegend(50, 300, false, "Arial", 8).setBackground(Chart.Transparent);
			  // Add a multi-color box-whisker layer to represent the waterfall bars
			  BoxWhiskerLayer layer = c.addBoxWhiskerLayer2(boxTop, boxBottom,  null, null, null, null, 0.5, labels);
			  //layer.setBoxColors(hexcodes);
			  layer.setBorderColor(Chart.Transparent);

        // Output the chart
        viewer.setChart(c);

        //include tool tip for the chart
        viewer.setImageMap(c.getHTMLImageMap("clickable", "",
            "title='{xLabel}: {={top}-{bottom}} millions'"));
    }

    //Allow this module to run as standalone program for easy testing
    public static void main(String[] args)
    {
        //Instantiate an instance of this demo module
        DemoModule demo = new waterfall();

        //Create and set up the main window
        JFrame frame = new JFrame(demo.toString());
        frame.addWindowListener(new WindowAdapter() {
            public void windowClosing(WindowEvent e) {System.exit(0);} });
        frame.getContentPane().setBackground(Color.white);

        // Create the chart and put them in the content pane
        ChartViewer viewer = new ChartViewer();
        demo.createChart(viewer, 0);
        frame.getContentPane().add(viewer);

        // Display the window
        frame.pack();
        frame.setVisible(true);
    }
}