mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
ffcedd7950
* ext/tk/lib/tk : bug fix and add Tcl/Tk extension support libraries git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@6559 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
666 lines
18 KiB
Ruby
666 lines
18 KiB
Ruby
#
|
|
# tkextlib/tcllib/plotchart.rb
|
|
# by Hidetoshi NAGAI (nagai@ai.kyutech.ac.jp)
|
|
#
|
|
# * Part of tcllib extension
|
|
# * Simple plotting and charting package
|
|
#
|
|
# (The following is the original description of the library.)
|
|
#
|
|
# Plotchart is a Tcl-only package that focuses on the easy creation of
|
|
# xy-plots, barcharts and other common types of graphical presentations.
|
|
# The emphasis is on ease of use, rather than flexibility. The procedures
|
|
# that create a plot use the entire canvas window, making the layout of the
|
|
# plot completely automatic.
|
|
#
|
|
# This results in the creation of an xy-plot in, say, ten lines of code:
|
|
# --------------------------------------------------------------------
|
|
# package require Plotchart
|
|
#
|
|
# canvas .c -background white -width 400 -height 200
|
|
# pack .c -fill both
|
|
#
|
|
# #
|
|
# # Create the plot with its x- and y-axes
|
|
# #
|
|
# set s [::Plotchart::createXYPlot .c {0.0 100.0 10.0} {0.0 100.0 20.0}]
|
|
#
|
|
# foreach {x y} {0.0 32.0 10.0 50.0 25.0 60.0 78.0 11.0 } {
|
|
# $s plot series1 $x $y
|
|
# }
|
|
#
|
|
# $s title "Data series"
|
|
# --------------------------------------------------------------------
|
|
#
|
|
# A drawback of the package might be that it does not do any data management.
|
|
# So if the canvas that holds the plot is to be resized, the whole plot must
|
|
# be redrawn. The advantage, though, is that it offers a number of plot and
|
|
# chart types:
|
|
#
|
|
# * XY-plots like the one shown above with any number of data series.
|
|
# * Stripcharts, a kind of XY-plots where the horizontal axis is adjusted
|
|
# automatically. The result is a kind of sliding window on the data
|
|
# series.
|
|
# * Polar plots, where the coordinates are polar instead of cartesian.
|
|
# * Isometric plots, where the scale of the coordinates in the two
|
|
# directions is always the same, i.e. a circle in world coordinates
|
|
# appears as a circle on the screen.
|
|
# You can zoom in and out, as well as pan with these plots (Note: this
|
|
# works best if no axes are drawn, the zooming and panning routines do
|
|
# not distinguish the axes), using the mouse buttons with the control
|
|
# key and the arrow keys with the control key.
|
|
# * Piecharts, with automatic scaling to indicate the proportions.
|
|
# * Barcharts, with either vertical or horizontal bars, stacked bars or
|
|
# bars side by side.
|
|
# * Timecharts, where bars indicate a time period and milestones or other
|
|
# important moments in time are represented by triangles.
|
|
# * 3D plots (both for displaying surfaces and 3D bars)
|
|
#
|
|
|
|
require 'tk'
|
|
|
|
# call setup script for general 'tkextlib' libraries
|
|
require 'tkextlib/setup.rb'
|
|
|
|
# call setup script
|
|
require File.join(File.dirname(File.expand_path(__FILE__)), 'setup.rb')
|
|
|
|
# TkPackage.require('Plotchart', '0.9')
|
|
TkPackage.require('Plotchart')
|
|
|
|
module Tk
|
|
module Tcllib
|
|
module Plotchart
|
|
end
|
|
end
|
|
end
|
|
|
|
module Tk::Tcllib::Plotchart
|
|
############################
|
|
def self.view_port(w, *args) # args := pxmin, pymin, pxmax, pymax
|
|
tk_call_without_enc('::Plotchart::viewPort', w.path, *(args.flatten))
|
|
end
|
|
|
|
def self.world_coordinates(w, *args) # args := xmin, ymin, xmax, ymax
|
|
tk_call_without_enc('::Plotchart::worldCoordinates',
|
|
w.path, *(args.flatten))
|
|
end
|
|
|
|
def self.world_3D_coordinates(w, *args)
|
|
# args := xmin, ymin, zmin, xmax, ymax, zmax
|
|
tk_call_without_enc('::Plotchart::world3DCoordinates',
|
|
w.path, *(args.flatten))
|
|
end
|
|
|
|
def self.coords_to_pixel(w, x, y)
|
|
list(tk_call_without_enc('::Plotchart::coordsToPixel', w.path, x, y))
|
|
end
|
|
|
|
def self.coords_3D_to_pixel(w, x, y, z)
|
|
list(tk_call_without_enc('::Plotchart::coords3DToPixel', w.path, x, y, z))
|
|
end
|
|
|
|
def self.polar_coordinates(w, radmax)
|
|
tk_call_without_enc('::Plotchart::polarCoordinates', w.path, radmax)
|
|
end
|
|
|
|
def self.polar_to_pixel(w, rad, phi)
|
|
list(tk_call_without_enc('::Plotchart::polarToPixel', w.path, rad, phi))
|
|
end
|
|
|
|
def self.pixel_to_coords(w, x, y)
|
|
list(tk_call_without_enc('::Plotchart::coordsToPixel', w.path, x, y))
|
|
end
|
|
|
|
def self.determine_scale(w, xmax, ymax)
|
|
tk_call_without_enc('::Plotchart::determineScale', w.path, xmax, ymax)
|
|
end
|
|
|
|
def self.set_zoom_pan(w)
|
|
tk_call_without_enc('::Plotchart::setZoomPan', w.path)
|
|
end
|
|
|
|
############################
|
|
module ChartMethod
|
|
include TkUtil
|
|
|
|
def title(str)
|
|
tk_call_without_enc(@chart, 'title', _get_eval_enc_str(str))
|
|
self
|
|
end
|
|
|
|
def save_plot(filename)
|
|
tk_call_without_enc(@chart, 'saveplot', filename)
|
|
self
|
|
end
|
|
|
|
def xtext(str)
|
|
tk_call_without_enc(@chart, 'xtext', _get_eval_enc_str(str))
|
|
self
|
|
end
|
|
|
|
def ytext(str)
|
|
tk_call_without_enc(@chart, 'ytext', _get_eval_enc_str(str))
|
|
self
|
|
end
|
|
|
|
def xconfig(key, value=None)
|
|
if key.kind_of?(Hash)
|
|
tk_call_without_enc(@chart, 'xconfig', *hash_kv(key, true))
|
|
else
|
|
tk_call_without_enc(@chart, 'xconfig',
|
|
"-#{key}", _get_eval_enc_str(value))
|
|
end
|
|
self
|
|
end
|
|
|
|
def yconfig(key, value=None)
|
|
if key.kind_of?(Hash)
|
|
tk_call_without_enc(@chart, 'yconfig', *hash_kv(key, true))
|
|
else
|
|
tk_call_without_enc(@chart, 'yconfig',
|
|
"-#{key}", _get_eval_enc_str(value))
|
|
end
|
|
self
|
|
end
|
|
|
|
############################
|
|
def view_port(*args) # args := pxmin, pymin, pxmax, pymax
|
|
tk_call_without_enc('::Plotchart::viewPort', @path, *(args.flatten))
|
|
self
|
|
end
|
|
|
|
def world_coordinates(*args) # args := xmin, ymin, xmax, ymax
|
|
tk_call_without_enc('::Plotchart::worldCoordinates',
|
|
@path, *(args.flatten))
|
|
self
|
|
end
|
|
|
|
def world_3D_coordinates(*args)
|
|
# args := xmin, ymin, zmin, xmax, ymax, zmax
|
|
tk_call_without_enc('::Plotchart::world3DCoordinates',
|
|
@path, *(args.flatten))
|
|
self
|
|
end
|
|
|
|
def coords_to_pixel(x, y)
|
|
list(tk_call_without_enc('::Plotchart::coordsToPixel', @path, x, y))
|
|
end
|
|
|
|
def coords_3D_to_pixel(x, y, z)
|
|
list(tk_call_without_enc('::Plotchart::coords3DToPixel', @path, x, y, z))
|
|
end
|
|
|
|
def polar_coordinates(radmax)
|
|
tk_call_without_enc('::Plotchart::polarCoordinates', @path, radmax)
|
|
self
|
|
end
|
|
|
|
def polar_to_pixel(rad, phi)
|
|
list(tk_call_without_enc('::Plotchart::polarToPixel', @path, rad, phi))
|
|
end
|
|
|
|
def pixel_to_coords(x, y)
|
|
list(tk_call_without_enc('::Plotchart::coordsToPixel', @path, x, y))
|
|
end
|
|
|
|
def determine_scale(xmax, ymax)
|
|
tk_call_without_enc('::Plotchart::determineScale', @path, xmax, ymax)
|
|
self
|
|
end
|
|
|
|
def set_zoom_pan()
|
|
tk_call_without_enc('::Plotchart::setZoomPan', @path)
|
|
self
|
|
end
|
|
end
|
|
|
|
############################
|
|
class XYPlot < TkCanvas
|
|
include ChartMethod
|
|
|
|
TkCommandNames = ['::Plotchart::createXYPlot'.freeze].freeze
|
|
|
|
def initialize(*args) # args := ([parent,] xaxis, yaxis [, keys])
|
|
# xaxis := Array of [minimum, maximum, stepsize]
|
|
# yaxis := Array of [minimum, maximum, stepsize]
|
|
if args[0].kind_of?(Array)
|
|
@xaxis = args.shift
|
|
@yaxis = args.shift
|
|
|
|
super(*args) # create canvas widget
|
|
else
|
|
parent = args.shift
|
|
|
|
@xaxis = args.shift
|
|
@yaxis = args.shift
|
|
|
|
if parent.kind_of?(TkCanvas)
|
|
@path = parent.path
|
|
else
|
|
super(parent, *args) # create canvas widget
|
|
end
|
|
end
|
|
|
|
@chart = _create_chart
|
|
end
|
|
|
|
def _create_chart
|
|
p self.class::TkCommandNames[0] if $DEBUG
|
|
tk_call_without_enc(self.class::TkCommandNames[0], @path,
|
|
array2tk_list(@xaxis), array2tk_list(@yaxis))
|
|
end
|
|
private :_create_chart
|
|
|
|
def plot(series, x, y)
|
|
tk_call_without_enc(@chart, 'plot', _get_eval_enc_str(series), x, y)
|
|
self
|
|
end
|
|
|
|
def dataconfig(series, key, value=None)
|
|
if key.kind_of?(Hash)
|
|
tk_call_without_enc(@chart, 'dataconfig', series, *hash_kv(key, true))
|
|
else
|
|
tk_call_without_enc(@chart, 'dataconfig', series,
|
|
"-#{key}", _get_eval_enc_str(value))
|
|
end
|
|
end
|
|
end
|
|
|
|
############################
|
|
class Stripchart < XYPlot
|
|
TkCommandNames = ['::Plotchart::createStripchart'.freeze].freeze
|
|
end
|
|
|
|
############################
|
|
class PolarPlot < TkCanvas
|
|
include ChartMethod
|
|
|
|
TkCommandNames = ['::Plotchart::createPolarplot'.freeze].freeze
|
|
|
|
def initialize(*args) # args := ([parent,] radius_data [, keys])
|
|
# radius_data := Array of [maximum_radius, stepsize]
|
|
if args[0].kind_of?(Array)
|
|
@radius_data = args.shift
|
|
|
|
super(*args) # create canvas widget
|
|
else
|
|
parent = args.shift
|
|
|
|
@radius_data = args.shift
|
|
|
|
if parent.kind_of?(TkCanvas)
|
|
@path = parent.path
|
|
else
|
|
super(parent, *args) # create canvas widget
|
|
end
|
|
end
|
|
|
|
@chart = _create_chart
|
|
end
|
|
|
|
def _create_chart
|
|
p self.class::TkCommandNames[0] if $DEBUG
|
|
tk_call_without_enc(self.class::TkCommandNames[0], @path,
|
|
array2tk_list(@radius_data))
|
|
end
|
|
private :_create_chart
|
|
|
|
def plot(series, radius, angle)
|
|
tk_call_without_enc(@chart, 'plot', _get_eval_enc_str(series),
|
|
radius, angle)
|
|
self
|
|
end
|
|
|
|
def dataconfig(series, key, value=None)
|
|
if key.kind_of?(Hash)
|
|
tk_call_without_enc(@chart, 'dataconfig', series, *hash_kv(key, true))
|
|
else
|
|
tk_call_without_enc(@chart, 'dataconfig', series,
|
|
"-#{key}", _get_eval_enc_str(value))
|
|
end
|
|
end
|
|
end
|
|
Polarplot = PolarPlot
|
|
|
|
############################
|
|
class IsometricPlot < TkCanvas
|
|
include ChartMethod
|
|
|
|
TkCommandNames = ['::Plotchart::createIsometricPlot'.freeze].freeze
|
|
|
|
def initialize(*args) # args := ([parent,] xaxis, yaxis, [, step] [, keys])
|
|
# xaxis := Array of [minimum, maximum]
|
|
# yaxis := Array of [minimum, maximum]
|
|
# step := Float of stepsize | "noaxes" | :noaxes
|
|
if args[0].kind_of?(Array)
|
|
@xaxis = args.shift
|
|
@yaxis = args.shift
|
|
|
|
if args[0].kind_of?(Hash)
|
|
@stepsize = :noaxes
|
|
else
|
|
@stepsize = args.shift
|
|
end
|
|
|
|
super(*args) # create canvas widget
|
|
else
|
|
parent = args.shift
|
|
|
|
@xaxis = args.shift
|
|
@yaxis = args.shift
|
|
|
|
if args[0].kind_of?(Hash)
|
|
@stepsize = :noaxes
|
|
else
|
|
@stepsize = args.shift
|
|
end
|
|
|
|
if parent.kind_of?(TkCanvas)
|
|
@path = parent.path
|
|
else
|
|
super(parent, *args) # create canvas widget
|
|
end
|
|
end
|
|
|
|
@chart = _create_chart
|
|
end
|
|
|
|
def _create_chart
|
|
p self.class::TkCommandNames[0] if $DEBUG
|
|
tk_call_without_enc(self.class::TkCommandNames[0], @path,
|
|
array2tk_list(@xaxis), array2tk_list(@yaxis),
|
|
@stepsize)
|
|
end
|
|
private :_create_chart
|
|
|
|
def plot(type, *args)
|
|
self.__send__("plot_#{type.to_s.tr('-', '_')}", *args)
|
|
end
|
|
|
|
def plot_rectangle(*args) # args := x1, y1, x2, y2, color
|
|
tk_call_without_enc(@chart, 'plot', 'rectangle', *(args.flatten))
|
|
self
|
|
end
|
|
|
|
def plot_filled_rectangle(*args) # args := x1, y1, x2, y2, color
|
|
tk_call_without_enc(@chart, 'plot', 'filled-rectangle', *(args.flatten))
|
|
self
|
|
end
|
|
|
|
def plot_circle(*args) # args := xc, yc, radius, color
|
|
tk_call_without_enc(@chart, 'plot', 'circle', *(args.flatten))
|
|
self
|
|
end
|
|
|
|
def plot_filled_circle(*args) # args := xc, yc, radius, color
|
|
tk_call_without_enc(@chart, 'plot', 'filled-circle', *(args.flatten))
|
|
self
|
|
end
|
|
end
|
|
Isometricplot = IsometricPlot
|
|
|
|
############################
|
|
class Plot3D < TkCanvas
|
|
include ChartMethod
|
|
|
|
TkCommandNames = ['::Plotchart::create3DPlot'.freeze].freeze
|
|
|
|
def initialize(*args) # args := ([parent,] xaxis, yaxis, zaxis [, keys])
|
|
# xaxis := Array of [minimum, maximum, stepsize]
|
|
# yaxis := Array of [minimum, maximum, stepsize]
|
|
# zaxis := Array of [minimum, maximum, stepsize]
|
|
if args[0].kind_of?(Array)
|
|
@xaxis = args.shift
|
|
@yaxis = args.shift
|
|
@zaxis = args.shift
|
|
|
|
super(*args) # create canvas widget
|
|
else
|
|
parent = args.shift
|
|
|
|
@xaxis = args.shift
|
|
@yaxis = args.shift
|
|
@zaxis = args.shift
|
|
|
|
if parent.kind_of?(TkCanvas)
|
|
@path = parent.path
|
|
else
|
|
super(parent, *args) # create canvas widget
|
|
end
|
|
end
|
|
|
|
@chart = _create_chart
|
|
end
|
|
|
|
def _create_chart
|
|
p self.class::TkCommandNames[0] if $DEBUG
|
|
tk_call_without_enc(self.class::TkCommandNames[0], @path,
|
|
array2tk_list(@xaxis),
|
|
array2tk_list(@yaxis),
|
|
array2tk_list(@zaxis))
|
|
end
|
|
private :_create_chart
|
|
|
|
def plot_function(cmd=Proc.new)
|
|
Tk.ip_eval("proc #{@path}_#{@chart} {x y} {#{install_cmd(cmd)} $x $y}")
|
|
tk_call_without_enc(@chart, 'plotfunc', "#{@path}_#{@chart}")
|
|
self
|
|
end
|
|
|
|
def grid_size(nxcells, nycells)
|
|
tk_call_without_enc(@chart, 'gridsize', nxcells, nycells)
|
|
self
|
|
end
|
|
|
|
def plot_data(dat)
|
|
# dat has to be provided as a 2 level array.
|
|
# 1st level contains rows, drawn in y-direction,
|
|
# and each row is an array whose elements are drawn in x-direction,
|
|
# for the columns.
|
|
tk_call_without_enc(@chart, 'plotdata', dat)
|
|
self
|
|
end
|
|
|
|
def colour(fill, border)
|
|
# configure the colours to use for polygon borders and inner area
|
|
tk_call_without_enc(@chart, 'colour', fill, border)
|
|
self
|
|
end
|
|
alias colours colour
|
|
alias colors colour
|
|
alias color colour
|
|
end
|
|
|
|
############################
|
|
class Piechart < TkCanvas
|
|
include ChartMethod
|
|
|
|
TkCommandNames = ['::Plotchart::createPiechart'.freeze].freeze
|
|
|
|
def initialize(*args) # args := ([parent] [, keys])
|
|
if args[0].kind_of?(TkCanvas)
|
|
parent = args.shift
|
|
@path = parent.path
|
|
else
|
|
super(*args) # create canvas widget
|
|
end
|
|
@chart = _create_chart
|
|
end
|
|
|
|
def _create_chart
|
|
p self.class::TkCommandNames[0] if $DEBUG
|
|
tk_call_without_enc(self.class::TkCommandNames[0], @path)
|
|
end
|
|
private :_create_chart
|
|
|
|
def plot(*dat) # argument is a list of [label, value]
|
|
tk_call_without_enc(@chart, 'plot', dat.flatten)
|
|
self
|
|
end
|
|
end
|
|
|
|
############################
|
|
class Barchart < TkCanvas
|
|
include ChartMethod
|
|
|
|
TkCommandNames = ['::Plotchart::createBarchart'.freeze].freeze
|
|
|
|
def initialize(*args)
|
|
# args := ([parent,] xlabels, ylabels [, series] [, keys])
|
|
# xlabels, ylabels := labels | axis ( depend on normal or horizontal )
|
|
# labels := Array of [label, label, ...]
|
|
# (It determines the number of bars that will be plotted per series.)
|
|
# axis := Array of [minimum, maximum, stepsize]
|
|
# series := Integer number of data series | 'stacked' | :stacked
|
|
if args[0].kind_of?(Array)
|
|
@xlabels = args.shift
|
|
@ylabels = args.shift
|
|
|
|
if args[0].kind_of?(Hash)
|
|
@series_size = :stacked
|
|
else
|
|
@series_size = args.shift
|
|
end
|
|
|
|
super(*args) # create canvas widget
|
|
else
|
|
parent = args.shift
|
|
|
|
@xlabels = args.shift
|
|
@ylabels = args.shift
|
|
|
|
if args[0].kind_of?(Hash)
|
|
@series_size = :stacked
|
|
else
|
|
@series_size = args.shift
|
|
end
|
|
|
|
if parent.kind_of?(TkCanvas)
|
|
@path = parent.path
|
|
else
|
|
super(parent, *args) # create canvas widget
|
|
end
|
|
end
|
|
|
|
@chart = _create_chart
|
|
end
|
|
|
|
def _create_chart
|
|
p self.class::TkCommandNames[0] if $DEBUG
|
|
tk_call_without_enc(self.class::TkCommandNames[0], @path,
|
|
array2tk_list(@xlabels), array2tk_list(@ylabels),
|
|
@series_size)
|
|
end
|
|
private :_create_chart
|
|
|
|
def plot(series, dat, col=None)
|
|
tk_call_without_enc(@chart, 'plot', series, dat, col)
|
|
self
|
|
end
|
|
|
|
def colours(*cols)
|
|
# set the colours to be used
|
|
tk_call_without_enc(@chart, 'colours', *cols)
|
|
self
|
|
end
|
|
alias colour colours
|
|
alias colors colours
|
|
alias color colours
|
|
end
|
|
|
|
############################
|
|
class HorizontalBarchart < Barchart
|
|
TkCommandNames = ['::Plotchart::createHorizontalBarchart'.freeze].freeze
|
|
end
|
|
|
|
############################
|
|
class Timechart < TkCanvas
|
|
include ChartMethod
|
|
|
|
TkCommandNames = ['::Plotchart::createTimechart'.freeze].freeze
|
|
|
|
def initialize(*args)
|
|
# args := ([parent,] time_begin, time_end, items [, keys])
|
|
# time_begin := String of time format (e.g. "1 january 2004")
|
|
# time_end := String of time format (e.g. "1 january 2004")
|
|
# items := Expected/maximum number of items
|
|
# ( This determines the vertical spacing. )
|
|
if args[0].kind_of?(Array)
|
|
@time_begin = args.shift
|
|
@time_end = args.shift
|
|
@items = args.shift
|
|
|
|
super(*args) # create canvas widget
|
|
else
|
|
parent = args.shift
|
|
|
|
@time_begin = args.shift
|
|
@time_end = args.shift
|
|
@items = args.shift
|
|
|
|
if parent.kind_of?(TkCanvas)
|
|
@path = parent.path
|
|
else
|
|
super(parent, *args) # create canvas widget
|
|
end
|
|
end
|
|
|
|
@chart = _create_chart
|
|
end
|
|
|
|
def _create_chart
|
|
p self.class::TkCommandNames[0] if $DEBUG
|
|
tk_call_without_enc(self.class::TkCommandNames[0], @path,
|
|
@time_begin, @time_end, @items)
|
|
end
|
|
private :_create_chart
|
|
|
|
def period(txt, time_begin, time_end, col=None)
|
|
tk_call_without_enc(@chart, 'period', txt, time_begin, time_end, col)
|
|
self
|
|
end
|
|
|
|
def milestone(txt, time, col=None)
|
|
tk_call_without_enc(@chart, 'milestone', txt, time, col)
|
|
self
|
|
end
|
|
|
|
def vertline(txt, time)
|
|
tk_call_without_enc(@chart, 'vertline', txt, time)
|
|
self
|
|
end
|
|
end
|
|
|
|
############################
|
|
class PlotSeries < TkObject
|
|
SeriesID_TBL = TkCore::INTERP.create_table
|
|
Series_ID = ['series'.freeze, '00000'.taint].freeze
|
|
TkCore::INTERP.init_ip_env{ SeriesID_TBL.clear }
|
|
|
|
def self.id2obj(chart, id)
|
|
path = chart.path
|
|
return id unless SeriesID_TBL[path]
|
|
SeriesID_TBL[path][id]? SeriesID_TBL[path][id]: id
|
|
end
|
|
|
|
def initialize(chart, keys=nil)
|
|
@parent = @chart_obj = chart
|
|
@ppath = @chart_obj.path
|
|
@path = @series = @id = Series_ID.join(TkCore::INTERP._ip_id_)
|
|
SeriesID_TBL[@id] = self
|
|
SeriesID_TBL[@ppath] = {} unless SeriesID_TBL[@ppath]
|
|
SeriesID_TBL[@ppath][@id] = self
|
|
Series_ID[1].succ!
|
|
dataconfig(keys) if keys.kind_of?(Hash)
|
|
end
|
|
|
|
def plot(*args)
|
|
@chart_obj.plot(@series, *args)
|
|
end
|
|
|
|
def dataconfig(key, value=None)
|
|
@chart_obj.dataconfig(@series, key, value)
|
|
end
|
|
end
|
|
end
|