local bit = require 'bit' local floor, pi = math.floor, math.pi local bor, band, lshift, rshift = bit.bor, bit.band, bit.lshift, bit.rshift local n_sampling_max = 8192 local n_sampling_default = 256 local function check_sampling(n) if n then if n <= 1 then error('sampling points should be> 1') elseif n> n_sampling_max then print('warning: too many sampling points requested, ' .. 'limiting to ' .. n_sampling_max) n = n_sampling_max end else n = n_sampling_default end return n end function graph.ipath(f) local ln = graph.path(f()) for x, y in f do ln:line_to(x, y) end return ln end function graph.ipathp(f) local ln = graph.path() local move, line = ln.move_to, ln.line_to local function next(op) local x, y = f() if x and y then op(ln, x, y) return true end end local success repeat success, more = pcall(next, success and line or move) if not success then print('warning:', more) end until not more return ln end function graph.fxline(f, xi, xs, n) n = check_sampling(n) return graph.ipath(iter.sample(f, xi, xs, n)) end function graph.filine(f, a, b) return graph.ipath(iter.isample(f, a, b)) end function graph.xyline(x, y) local i0 = 1 local n = matrix.dim(x) local ln = graph.path(x[i0], y[i0]) for i=i0+1, i0+n-1 do ln:line_to(x[i], y[i]) end return ln end function graph.fxplot(f, xi, xs, color, n) n = check_sampling(n) local p = graph.plot() p:addline(graph.ipathp(iter.sample(f, xi, xs, n)), color) p:show() return p end function graph.fiplot(f, a, b, color) if not b then a, b, color = 1, a, b end local p = graph.plot() p:addline(graph.ipathp(iter.isample(f, a, b)), color) p:show() return p end local function add_square(p, lx, by, rx, ty) p:move_to(lx, by) p:line_to(rx, by) p:line_to(rx, ty) p:line_to(lx, ty) p:close() end function graph.fibars(f, a, b, color, fill) local wf = 0.5 * (fill or 1) local p = graph.plot() local sh = graph.path() for k = a, b do local y = f(k) add_square(sh, k-wf, 0, k+wf, y) end p:add(sh, color or 'black') p:show() return p end function graph.ibars(f) local b = graph.path() local lx, ly = f() local first = true for rx, ry in f do local dx = (rx-lx)/2 if first then add_square(b, lx-dx, 0, lx+dx, ly); first = false end add_square(b, lx+dx, 0, rx+dx, ry) lx, ly = rx, ry end return b end function graph.barplot(t) local nr, nc = #t, #t[1] - 1 local legend_text = t.legend local pad = 0.1 local dx = (1-2*pad)/nc local cat = {} local p = graph.plot() p.pad = true for k = 1, nr do local row = t[k] local label = row[1] cat[#cat+1] = k - 0.5 cat[#cat+1] = label for j = 1, nc do local x, y = (k-1) + pad + (j-1)*dx, row[j+1] local rect = graph.rect(x, 0, x+dx, y) p:add(rect, graph.webcolor(j)) end end if legend_text then for j = 1, nc do p:legend(legend_text[j], graph.webcolor(j), 'square') end end p:set_categories('x', cat) p.xlab_angle = pi/4 p:show() return p end function graph.segment(x1, y1, x2, y2) local p = graph.path(x1, y1) p:line_to(x2, y2) return p end function graph.rect(x1, y1, x2, y2) local p = graph.path() add_square(p, x1, y1, x2, y2) return p end local function rgba(r, g, b, a) r, g, b, a = tonumber(r), tonumber(g), tonumber(b), tonumber(a) local rb = band(lshift(r, 24), 0xff000000) local gb = band(lshift(g, 16), 0xff0000 ) local bb = band(lshift(b, 8 ), 0xff00 ) return bor(rb, gb, bb, a and band(a, 0xff) or 0xff) end local function rgba_decode(col) col = tonumber(col) local r = rshift(band(col, 0xff000000), 24) local g = rshift(band(col, 0xff0000), 16) local b = rshift(band(col, 0xff00), 8) local a = band(col, 0xff) return r, g, b, a end graph.rgba = rgba graph.rgb = rgba local function gamma(a) return (a / 255)^2.2 end local function gamma_inv(a) return 255 * a^(1/2.2) end local function color_blend(f1, r1, g1, b1, f2, r2, g2, b2, alpha) local r = gamma_inv(f1 * gamma(r1) + f2 * gamma(r2)) local g = gamma_inv(f1 * gamma(g1) + f2 * gamma(g2)) local b = gamma_inv(f1 * gamma(b1) + f2 * gamma(b2)) return rgba(r, g, b, alpha) end local lum_std = 190 graph.color = { red = rgba(lum_std, 0, 0), green = rgba(0, lum_std, 0), blue = rgba(0, 0, lum_std), magenta = rgba(lum_std, 0, lum_std), cyan = rgba(0, lum_std, lum_std), yellow = rgba(lum_std, lum_std, 0), black = rgba(0, 0, 0), white = rgba(0xff, 0xff, 0xff), decode = rgba_decode, combine = function(f1, c1, f2, c2) local r1, g1, b1 = rgba_decode(c1) if f2 and c2 then local r2, g2, b2 = rgba_decode(c2) return color_blend(f1, r1, g1, b1, f2, r2, g2, b2) else return rgba(f1*r1, f1*g1, f1*b1) end end } local bcolors = {'red', 'blue', 'green', 'magenta', 'cyan', 'yellow'} -- colors from a popular spreadsheet application + random colors from -- the web local wcolors = {0x4f81bd, 0xc0504d, 0x9bbb59, 0x695185, 0x3c8da3, 0xcc7b38, 0x58c3bb, 0xf29a86, 0xbd87bb, 0xc3a492, 0xf0d845, 0x2d7ac0, 0xfce4ca, 0x8b9068, 0xc0b99d, 0xD9CCB9, 0xDF7782, 0xE95D22, 0x017890, 0x613D2D} local hue_map = { {231, 0, 0 }, {231, 113, 0 }, {231, 211, 0 }, {156, 231, 0 }, {0, 231, 33 }, {0, 231, 156}, {0, 195, 231}, {0, 113, 231}, {0, 0, 231}, {132, 0, 231} } local function hue_choose(k) local e = hue_map[k] local r, g, b = e[1], e[2], e[3] return rgba(r, g, b, 255) end local function hue_color(p) local x = 10 - p * 9 local i = floor(x) if i < 1 or i+1> 10 then return i < 1 and hue_choose(1) or hue_choose(10) else local e1, e2 = hue_map[i], hue_map[i+1] local f2 = x-i return color_blend(1 - f2, e1[1], e1[2], e1[3], f2, e2[1], e2[2], e2[3]) end end function graph.rainbow(n) local p = #bcolors return graph.color[bcolors[(n-1) % p + 1]] end function graph.webcolor(n) local p = #wcolors return lshift(wcolors[(n-1) % p + 1], 8) + 0xff end local color_schema = { bluish = {232, 229, 217, 88, 37, 153}, redyellow = {229, 229, 0, 229, 0, 0}, darkgreen = {229, 229, 0, 0, 102, 0} } function graph.color_function(schema, alpha) local c = color_schema[schema] return function(a) return color_blend(1-a, c[1], c[2], c[3], a, c[4], c[5], c[6], alpha) end end graph.hue_color = hue_color function graph.plot_lines(ln, title) local p = graph.plot(title) for k=1, #ln do p:addline(ln[k], rainbow(k)) end p:show() return p end local function legend_symbol(sym, dx, dy) if sym == 'square' then return graph.rect(5+dx, 5+dy, 15+dx, 15+dy) elseif sym == 'line' then return graph.segment(2+dx, 10+dy, 18+dx, 10+dy), {{'stroke'}} else return graph.marker(10+dx, 10+dy, sym, 8) end end local function get_plot_legend(p) local lg = p:get_legend() if not lg then lg = graph.plot() lg.units = false lg.clip = false p:set_legend(lg) end return lg end local function plot_legend(self, text, color, symspec, trans) local lg = get_plot_legend(self) local env = debug.getfenv(self) local k = env.__lg_count or 0 local y = -k * 20 local sym, symtr = legend_symbol(symspec, 0, y) local tr = (trans and trans or symtr) lg:add(sym, color, tr) lg:add(graph.textshape(25, y + 6, text, 14), 'black') env.__lg_count = k+1 self:update() end local function plot_legend_title(self, text, color) color = color or graph.rgb(0, 0, 120) local lg = get_plot_legend(self) lg:add(graph.textshape(25, 30, text, 14), color) self:update() end local function redirect_plot() local reg = debug.getregistry() local mt = reg['GSL.plot'] local plot_index = mt.__index local function index_redirect(t, k) if k == 'legend' then return plot_legend elseif k == 'legend_title' then return plot_legend_title end return plot_index(t, k) end mt.__index = index_redirect end redirect_plot()

AltStyle によって変換されたページ (->オリジナル) /