ChartDirector 6.1 (Java Edition)

General Curve Fitting




This example demonstrates the LOWESS general purpose curve fitting that is built into ChartDirector.

In regression curve fitting, a function type have to be chosen first, and the data are used to compute the best parameters of that function. For example, in polynomial regression, the function type is a polynomial of a given degree, and the data are used to determine the optimal polynomial coefficients.

However, in many applications, the function type is not known or not expressible in algebraic form. To handle these cases, ChartDirector supports a general curve fitting algorithm known as LOWESS. For details, please refer to ArrayMath.lowess and ArrayMath.lowess2.

The LOWESS algorithm only adjusts the points to smoothen them. They still need to be joined together. In this example, the adjusted points are joined with a spline layer using XYChart.addSplineLayer.

Source Code Listing

[JSP Version] jspdemo/curvefitting.jsp
<%@page import="ChartDirector.*" %>
<%
// Use random table to generate a random series. The random table is set to 1 col x 51 rows, with 9
// as the seed
RanTable rantable = new RanTable(9, 1, 51);

// Set the 1st column to start from 100, with changes between rows from -5 to +5
rantable.setCol(0, 100, -5, 5);

// Get the 1st column of the random table as the data set
double[] data = rantable.getCol(0);

// Create a XYChart object of size 600 x 300 pixels
XYChart c = new XYChart(600, 300);

// Set the plotarea at (50, 35) and of size 500 x 240 pixels. Enable both the horizontal and
// vertical grids by setting their colors to grey (0xc0c0c0)
c.setPlotArea(50, 35, 500, 240).setGridColor(0xc0c0c0, 0xc0c0c0);

// Add a title to the chart using 18 point Times Bold Itatic font.
c.addTitle("LOWESS Generic Curve Fitting Algorithm", "Times New Roman Bold Italic", 18);

// Set the y axis line width to 3 pixels
c.yAxis().setWidth(3);

// Add a title to the x axis using 12pt Arial Bold Italic font
c.xAxis().setTitle("Server Load (TPS)", "Arial Bold Italic", 12);

// Set the x axis line width to 3 pixels
c.xAxis().setWidth(3);

// Set the x axis scale from 0 - 50, with major tick every 5 units and minor tick every 1 unit
c.xAxis().setLinearScale(0, 50, 5, 1);

// Add a blue layer to the chart
LineLayer layer = c.addLineLayer2();

// Add a red (0x80ff0000) data set to the chart with square symbols
layer.addDataSet(data, 0x80ff0000).setDataSymbol(Chart.SquareSymbol);

// Set the line width to 2 pixels
layer.setLineWidth(2);

// Use lowess for curve fitting, and plot the fitted data using a spline layer with line width set
// to 3 pixels
c.addSplineLayer(new ArrayMath(data).lowess().result(), 0x0000ff).setLineWidth(3);

// Set zero affinity to 0 to make sure the line is displayed in the most detail scale
c.yAxis().setAutoScale(0, 0, 0);

// Output the chart
String chart1URL = c.makeSession(request, "chart1");

// Include tool tip for the chart
String imageMap1 = c.getHTMLImageMap("", "", "title='({x}, {value|2})'");
%>
<html>
<body style="margin:5px 0px 0px 5px">
<div style="font-size:18pt; font-family:verdana; font-weight:bold">
    General Curve Fitting
</div>
<hr color="#000080">
<div style="font-size:10pt; font-family:verdana; margin-bottom:1.5em">
    <a href="viewsource.jsp?file=<%=request.getServletPath()%>">View Source Code</a>
</div>
<img src='<%=response.encodeURL("getchart.jsp?"+chart1URL)%>'
    usemap="#map1" border="0">
<map name="map1"><%=imageMap1%></map>
</body>
</html>

[Java Version] javademo/curvefitting.java
import java.awt.*;
import java.awt.event.*;
import javax.swing.*;
import ChartDirector.*;

public class curvefitting implements DemoModule
{
    //Name of demo program
    public String toString() { return "General Curve Fitting"; }

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

    //Main code for creating charts
    public void createChart(ChartViewer viewer, int chartIndex)
    {
        // Use random table to generate a random series. The random table is set to 1 col x 51 rows,
        // with 9 as the seed
        RanTable rantable = new RanTable(9, 1, 51);

        // Set the 1st column to start from 100, with changes between rows from -5 to +5
        rantable.setCol(0, 100, -5, 5);

        // Get the 1st column of the random table as the data set
        double[] data = rantable.getCol(0);

        // Create a XYChart object of size 600 x 300 pixels
        XYChart c = new XYChart(600, 300);

        // Set the plotarea at (50, 35) and of size 500 x 240 pixels. Enable both the horizontal and
        // vertical grids by setting their colors to grey (0xc0c0c0)
        c.setPlotArea(50, 35, 500, 240).setGridColor(0xc0c0c0, 0xc0c0c0);

        // Add a title to the chart using 18 point Times Bold Itatic font.
        c.addTitle("LOWESS Generic Curve Fitting Algorithm", "Times New Roman Bold Italic", 18);

        // Set the y axis line width to 3 pixels
        c.yAxis().setWidth(3);

        // Add a title to the x axis using 12pt Arial Bold Italic font
        c.xAxis().setTitle("Server Load (TPS)", "Arial Bold Italic", 12);

        // Set the x axis line width to 3 pixels
        c.xAxis().setWidth(3);

        // Set the x axis scale from 0 - 50, with major tick every 5 units and minor tick every 1
        // unit
        c.xAxis().setLinearScale(0, 50, 5, 1);

        // Add a blue layer to the chart
        LineLayer layer = c.addLineLayer2();

        // Add a red (0x80ff0000) data set to the chart with square symbols
        layer.addDataSet(data, 0x80ff0000).setDataSymbol(Chart.SquareSymbol);

        // Set the line width to 2 pixels
        layer.setLineWidth(2);

        // Use lowess for curve fitting, and plot the fitted data using a spline layer with line
        // width set to 3 pixels
        c.addSplineLayer(new ArrayMath(data).lowess().result(), 0x0000ff).setLineWidth(3);

        // Set zero affinity to 0 to make sure the line is displayed in the most detail scale
        c.yAxis().setAutoScale(0, 0, 0);

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

        //include tool tip for the chart
        viewer.setImageMap(c.getHTMLImageMap("clickable", "", "title='({x}, {value|2})'"));
    }

    //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 curvefitting();

        //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);
    }
}