ChartDirector 6.0 (Ruby 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

[Ruby On Rails Version - Controller] app/controllers/curvefitting_controller.rb
require("chartdirector")

class CurvefittingController < ApplicationController

    def index()
        @title = "General Curve Fitting"
        @ctrl_file = File.expand_path(__FILE__)
        @noOfCharts = 1
        render :template => "templates/chartview"
    end

    #
    # Render and deliver the chart
    #
    def getchart()
        # 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 = ChartDirector::RanTable.new(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
        data = rantable.getCol(0)

        # Create a XYChart object of size 600 x 300 pixels
        c = ChartDirector::XYChart.new(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", "timesbi.ttf", 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)", "arialbi.ttf", 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
        layer = c.addLineLayer2()

        # Add a red (0x80ff0000) data set to the chart with square symbols
        layer.addDataSet(data, 0x80ff0000).setDataSymbol(ChartDirector::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(ChartDirector::ArrayMath.new(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
        send_data(c.makeChart2(ChartDirector::PNG), :type => "image/png", :disposition => "inline")

    end

end

[Ruby On Rails Version - View] app/views/templates/chartview.html.erb
<html>
<body style="margin:5px 0px 0px 5px">

<!-- Title -->
<div style="font-size:18pt; font-family:verdana; font-weight:bold">
    <%= @title %>
</div>
<hr style="border:solid 1px #000080" />

<!-- Source Code Listing Link -->
<div style="font-size:9pt; font-family:verdana; margin-bottom:1.5em">
    <%= link_to "Source Code Listing", 
        :controller => "cddemo", :action => "viewsource",
        :ctrl_file => @ctrl_file, :view_file => File.expand_path(__FILE__) %>
</div>

<!-- Create one or more IMG tags to display the demo chart(s) -->
<% 0.upto(@noOfCharts - 1) do |i| %>
    <img src="<%= url_for(:action => "getchart", :img => i) %>">
<% end %>

</body>
</html>

[Command Line Version] rubydemo/curvefitting.rb
#!/usr/bin/env ruby
require("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 = ChartDirector::RanTable.new(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
data = rantable.getCol(0)

# Create a XYChart object of size 600 x 300 pixels
c = ChartDirector::XYChart.new(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", "timesbi.ttf", 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)", "arialbi.ttf", 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
layer = c.addLineLayer2()

# Add a red (0x80ff0000) data set to the chart with square symbols
layer.addDataSet(data, 0x80ff0000).setDataSymbol(ChartDirector::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(ChartDirector::ArrayMath.new(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
c.makeChart("curvefitting.png")