path: root/Master/texmf-dist/tex/lualatex/bezierplot/bezierplot.lua

#!/usr/bin/env lua
-- Linus Romer, published 2018 under LPPL Version 1.3c
-- version 1.4 2019-06-18
abs = math.abs
acos = math.acos
asin = math.asin
atan = math.atan
cos = math.cos
exp = math.exp
e = math.exp(1)
log = math.log
pi = math.pi
sin = math.sin
sqrt = math.sqrt
tan = math.tan
huge = math.huge

-- just a helper for debugging:
local function printdifftable(t)
	for i = 1,#t do
		for j = 1, 6 do
			if j < 5 then
				io.write(t[i][j].." ")
			else
				if t[i][j] then
					io.write("true ")
				else
					io.write("false ")
				end
			end
		end
		io.write("\n")
	end
end

-- cube root defined for all real numbers x
function cbrt(x)
	if x < 0 then
		return -(-x)^(1/3)
	else
		return x^(1/3)
	end
end

function sgn(x)
	if x<0 then
		return -1
	elseif x>0 then
		return 1
	else
		return 0
	end
end

local function round(num, decimals)
	local result = tonumber(string.format("%." .. (decimals or 0) .. "f", num))
	if abs(result) == 0 then
		return 0
	else
		return result
	end
end

-- check if a point (x,y) satisfies xmin <= x <= xmax and ymin <= < <= ymax
local function is_in_window(x,y,xmin,xmax,ymin,ymax)
	if x >= xmin and x <= xmax and y >= ymin and y <= ymax then
		return true
	else
		return false
	end
end

local function evaluate(s)
	local tempfunc = assert(load("return " .. s))
	return tempfunc()
end


-- 5-stencil method
-- return from a graph from f in the form {{x,y},...}
-- the derivatives in form {{x,y,dy/dx,ddy/ddx,extrema,inflection},...}
local function diffgraph(func,graph,h)
	local dgraph = {}	
	local l = #graph
	if l < 4 then -- this is not worth the pain...
		for i = 1, l do 
			table.insert(dgraph,{graph[i][1],graph[i][2],0,0,false,false})
		end
	else
		local yh = func(graph[1][1]-h)
		local yhh = func(graph[1][1]-2*h)
		if yhh > -math.huge and yhh < math.huge  -- if defined at all
		and yh > -math.huge and yh < math.huge then
			dgraph[1] = {graph[1][1],graph[1][2],
			(yhh-8*yh+8*graph[2][2]-graph[3][2])/(12*h),
			(-yhh+16*yh-30*graph[1][2]+16*graph[2][2]-graph[3][2])
			/(12*h^2)}
			dgraph[2] = {graph[2][1],graph[2][2],
			(yh-8*graph[1][2]+8*graph[3][2]-graph[4][2])/(12*h),
			(-yh+16*graph[1][2]-30*graph[2][2]+16*graph[3][2]-graph[4][2])
			/(12*h^2)}
		else -- take neighbour values
			dgraph[1] = {graph[1][1],graph[1][2],
			(graph[1][2]-8*graph[2][2]+8*graph[4][2]-graph[5][2])/(12*h),
			(-graph[1][2]+16*graph[2][2]-30*graph[3][2]
			+16*graph[4][2]-graph[5][2])/(12*h^2)}
			dgraph[2] = {graph[2][1],graph[2][2],
			(graph[1][2]-8*graph[2][2]+8*graph[4][2]-graph[5][2])/(12*h),
			(-graph[1][2]+16*graph[2][2]-30*graph[3][2]
			+16*graph[4][2]-graph[5][2])/(12*h^2)}
		end
		for i = 3, l-2 do 
			table.insert(dgraph,{graph[i][1],graph[i][2],
			(graph[i-2][2]-8*graph[i-1][2]+8*graph[i+1][2]-graph[i+2][2])
			/(12*h),
			(-graph[i-2][2]+16*graph[i-1][2]-30*graph[i][2]
			+16*graph[i+1][2]-graph[i+2][2])
			/(12*h^2)})
		end
		yh = func(graph[l][1]+h)
		yhh = func(graph[l][1]+2*h)
		if yhh > -math.huge and yhh < math.huge  -- if defined at all
		and yh > -math.huge and yh < math.huge then
			dgraph[l-1] = {graph[l-1][1],graph[l-1][2],
			(graph[l-3][2]-8*graph[l-2][2]+8*graph[l][2]-yh)/(12*h),
			(-graph[l-3][2]+16*graph[l-2][2]-30*graph[l-1][2]
			+16*graph[l][2]-yh)/(12*h^2)}
			dgraph[l] = {graph[l][1],graph[l][2],
			(graph[l-2][2]-8*graph[l-1][2]+8*yh-yhh)/(12*h),
			(-graph[l-2][2]+16*graph[l-1][2]-30*graph[l][2]
			+16*yh-yhh)/(12*h^2)}
		else	
			-- take neighbour values
			dgraph[l] = {graph[l][1],graph[l][2],
			(graph[l-4][2]-8*graph[l-3][2]+8*graph[l-1][2]-graph[l][2])
			/(12*h),
			(-graph[l-4][2]+16*graph[l-3][2]-30*graph[l-2][2]
			+16*graph[l-1][2]-graph[l][2])/(12*h^2)}
			dgraph[l-1] = {graph[l-1][1],graph[l-2][2],
			(graph[l-4][2]-8*graph[l-3][2]+8*graph[l-1][2]-graph[l][2])
			/(12*h),
			(-graph[l-4][2]+16*graph[l-3][2]-30*graph[l-2][2]
			+16*graph[l-1][2]-graph[l][2])/(12*h^2)}
		end
		-- add information about being extremum / inflection point (true/false)
		for i = 1, l do 
			dgraph[i][5] = false -- dy/dx == 0 ? default, may change later
			dgraph[i][6] = false -- ddy/ddx == 0 ? default, may change later
		end
		for i = 1, l-1 do 
			-- if no gap is inbetween
			if not (dgraph[i+1][1] - dgraph[i][1] > 1.5*h) then
				-- check for dy/dx == 0 
				-- if not already determined as near dy/dx=0
				if dgraph[i][3] == 0 then
					if dgraph[i+1][3] == 0 then --take the later
						dgraph[i+1][5] = true
						dgraph[i][5] = false
					else
						dgraph[i][5] = true
					end
				elseif abs(dgraph[i][3]*dgraph[i+1][3]) 
				~= dgraph[i][3]*dgraph[i+1][3] then -- this must be near
					if abs(dgraph[i][4]) <= abs(dgraph[i+1][4]) then
						dgraph[i][5] = true
					else
						dgraph[i+1][5] = true
					end
				end
				-- check for ddy/ddx == 0 
				-- if not already determined as near ddy/ddx=0
				if (not dgraph[i][6]) and (abs(dgraph[i][4]*dgraph[i+1][4]) 
					~= dgraph[i][4]*dgraph[i+1][4]) then -- this must be near
					if abs(dgraph[i][4]) <= abs(dgraph[i+1][4]) then
						dgraph[i][6] = true
					else
						dgraph[i+1][6] = true
					end
				end
			end
		end
	end
	return dgraph
end

-- simplified diffgraph function, the function is derived only once
-- return from a graph from f in the form {{x,y},...}
-- the derivatives in form {{x,y,dy/dx},...}
-- we start with index 1 and then always jump indexjump to the next
-- index
local function diffgraphsimple(func,graph,h,indexjump)
	local dgraph = {}	
	local l = math.floor((#graph-1)/indexjump)*indexjump+1
	if l < 2 then -- this is not worth the pain...
		dgraph = {graph[1][1],graph[1][2],0}
	else
		local yh = func(graph[1][1]-h)
		local yhh = func(graph[1][1]-2*h)
		if yhh > -math.huge and yhh < math.huge  -- if defined at all
		and yh > -math.huge and yh < math.huge then
			dgraph[1] = {graph[1][1],graph[1][2],
			(yhh-8*yh+8*graph[2][2]-graph[3][2])/(12*h)}
		else -- take neighbour values
			dgraph[1] = {graph[1][1],graph[1][2],
			(graph[1][2]-8*graph[2][2]+8*graph[4][2]-graph[5][2])
			/(12*h)}
		end
		for i = 1+indexjump, l-1, indexjump do 
			table.insert(dgraph,{graph[i][1],graph[i][2],
			(graph[i-2][2]-8*graph[i-1][2]+8*graph[i+1][2]-graph[i+2][2])
			/(12*h)})
		end
		yh = func(graph[l][1]+h)
		yhh = func(graph[l][1]+2*h)
		if yhh > -math.huge and yhh < math.huge  -- if defined at all
		and yh > -math.huge and yh < math.huge then

			table.insert(dgraph,{graph[l][1],graph[l][2],
			(graph[l-2][2]-8*graph[l-1][2]+8*yh-yhh)/(12*h)})
		else	
			-- take neighbour values
			table.insert(dgraph,{graph[l][1],graph[l][2],
			(graph[l-4][2]-8*graph[l-3][2]+8*graph[l-1][2]-graph[l][2])
			/(12*h)})
		end
	end
	return dgraph
end

-- diffgraph for cubic function ax^3+bx^2+cx+d
-- returns the derivatives in form {{x,y,dy/dx,ddy/ddx},...}
-- if isinverse = true then the coordinates will be inversed
local function diffgraphcubic(graph,a,b,c,d,isinverse)
	local dgraph = {}	
	local l = #graph
	for i = 1, l do 
		if isinverse then
			dgraph[#dgraph+1] = {graph[i][2],graph[i][1],c
			+graph[i][2]*(2*b+3*a*graph[i][2]),6*a*graph[i][2]+2*b}
		else
			dgraph[#dgraph+1] = {graph[i][1],graph[i][2],c
			+graph[i][1]*(2*b+3*a*graph[i][1]),6*a*graph[i][1]+2*b}
		end
	end
	return dgraph
end

-- checks for 100 x, if the function given by funcstring
-- fits the graph g (up to maxerror) after filling in
-- the parameters a, b, c, d
-- if the graph is inverted, then isinverse has to be set true
local function do_parameters_fit(a,b,c,d,funcstring,funcgraph,maxerror,isinverse)
	if not (a > -math.huge and a < math.huge and b > -math.huge and b < math.huge  and
	c > -math.huge and c < math.huge and d > -math.huge and d < math.huge) then
		return false
	end
	local funcx = string.gsub(funcstring, "a", a)
	local funcx = string.gsub(funcx, "b", b)
	local funcx = string.gsub(funcx, "c", c)
	local funcx = string.gsub(funcx, "d", d)
	local func = assert(load("local x = ...; return "..funcx))
	for i = 1, #funcgraph, math.max(1,math.floor(0.01*#funcgraph)) do 
		if isinverse then
			if abs(func(funcgraph[i][2])-funcgraph[i][1]) 
			> maxerror then
				return false
			end
		else
			if abs(func(funcgraph[i][1])-funcgraph[i][2]) 
			> maxerror then
				return false
			end
		end
	end
	return true
end

-- f(x)=a*x^3+b*x+c
local function parameters_cubic(xp,yp,xq,yq,xr,yr,xs,ys)
	local a = (((xp^2 * xq) * yr) - ((xp^2 * xq) * ys) 
	- ((xp^2 * xr) * yq) + ((xp^2 * xr) * ys) + ((xp^2 * xs) * yq)
	- ((xp^2 * xs) * yr) - ((xp * xq^2) * yr) + ((xp * xq^2) * ys) 
	+ ((xp * xr^2) * yq) - ((xp * xr^2) * ys) - ((xp * xs^2) * yq) 
	+ ((xp * xs^2) * yr) + ((xq^2 * xr) * yp) - ((xq^2 * xr) * ys) 
	- ((xq^2 * xs) * yp) + ((xq^2 * xs) * yr) - ((xq * xr^2) * yp) 
	+ ((xq * xr^2) * ys) + ((xq * xs^2) * yp) - ((xq * xs^2) * yr) 
	+ ((xr^2 * xs) * yp) - ((xr^2 * xs) * yq) - ((xr * xs^2) * yp) 
	+ ((xr * xs^2) * yq)) / 
	(((xp^3 * xq^2) * xr) - ((xp^3 * xq^2) * xs) 
	- ((xp^3 * xq) * xr^2) + ((xp^3 * xq) * xs^2) 
	+ ((xp^3 * xr^2) * xs) - ((xp^3 * xr) * xs^2) 
	- ((xp^2 * xq^3) * xr) + ((xp^2 * xq^3) * xs) 
	+ ((xp^2 * xq) * xr^3) - ((xp^2 * xq) * xs^3) 
	- ((xp^2 * xr^3) * xs) + ((xp^2 * xr) * xs^3) 
	+ ((xp * xq^3) * xr^2) - ((xp * xq^3) * xs^2) 
	- ((xp * xq^2) * xr^3) + ((xp * xq^2) * xs^3) 
	+ ((xp * xr^3) * xs^2) - ((xp * xr^2) * xs^3) 
	- ((xq^3 * xr^2) * xs) + ((xq^3 * xr) * xs^2) 
	+ ((xq^2 * xr^3) * xs) - ((xq^2 * xr) * xs^3) 
	- ((xq * xr^3) * xs^2) + ((xq * xr^2) * xs^3))
	local b = ((((-xp^3) * xq) * yr) + ((xp^3 * xq) * ys) 
	+ ((xp^3 * xr) * yq) - ((xp^3 * xr) * ys) - ((xp^3 * xs) * yq) 
	+ ((xp^3 * xs) * yr) + ((xp * xq^3) * yr) - ((xp * xq^3) * ys) 
	- ((xp * xr^3) * yq) + ((xp * xr^3) * ys) + ((xp * xs^3) * yq) 
	- ((xp * xs^3) * yr) - ((xq^3 * xr) * yp) + ((xq^3 * xr) * ys) 
	+ ((xq^3 * xs) * yp) - ((xq^3 * xs) * yr) + ((xq * xr^3) * yp) 
	- ((xq * xr^3) * ys) - ((xq * xs^3) * yp) + ((xq * xs^3) * yr) 
	- ((xr^3 * xs) * yp) + ((xr^3 * xs) * yq) + ((xr * xs^3) * yp) 
	- ((xr * xs^3) * yq)) / 
	(((xp^3 * xq^2) * xr) - ((xp^3 * xq^2) * xs) 
	- ((xp^3 * xq) * xr^2) + ((xp^3 * xq) * xs^2) 
	+ ((xp^3 * xr^2) * xs) - ((xp^3 * xr) * xs^2) 
	- ((xp^2 * xq^3) * xr) + ((xp^2 * xq^3) * xs) 
	+ ((xp^2 * xq) * xr^3) - ((xp^2 * xq) * xs^3) 
	- ((xp^2 * xr^3) * xs) + ((xp^2 * xr) * xs^3) 
	+ ((xp * xq^3) * xr^2) - ((xp * xq^3) * xs^2) 
	- ((xp * xq^2) * xr^3) + ((xp * xq^2) * xs^3) 
	+ ((xp * xr^3) * xs^2) - ((xp * xr^2) * xs^3) 
	- ((xq^3 * xr^2) * xs) + ((xq^3 * xr) * xs^2) 
	+ ((xq^2 * xr^3) * xs) - ((xq^2 * xr) * xs^3) 
	- ((xq * xr^3) * xs^2) + ((xq * xr^2) * xs^3))
	local c = (((xp^3 * xq^2) * yr) - ((xp^3 * xq^2) * ys) 
	- ((xp^3 * xr^2) * yq) + ((xp^3 * xr^2) * ys) 
	+ ((xp^3 * xs^2) * yq) - ((xp^3 * xs^2) * yr) 
	- ((xp^2 * xq^3) * yr) + ((xp^2 * xq^3) * ys) 
	+ ((xp^2 * xr^3) * yq) - ((xp^2 * xr^3) * ys) 
	- ((xp^2 * xs^3) * yq) + ((xp^2 * xs^3) * yr) 
	+ ((xq^3 * xr^2) * yp) - ((xq^3 * xr^2) * ys) 
	- ((xq^3 * xs^2) * yp) + ((xq^3 * xs^2) * yr) 
	- ((xq^2 * xr^3) * yp) + ((xq^2 * xr^3) * ys) 
	+ ((xq^2 * xs^3) * yp) - ((xq^2 * xs^3) * yr) 
	+ ((xr^3 * xs^2) * yp) - ((xr^3 * xs^2) * yq) 
	- ((xr^2 * xs^3) * yp) + ((xr^2 * xs^3) * yq)) / 
	(((xp^3 * xq^2) * xr) - ((xp^3 * xq^2) * xs) 
	- ((xp^3 * xq) * xr^2) + ((xp^3 * xq) * xs^2) 
	+ ((xp^3 * xr^2) * xs) - ((xp^3 * xr) * xs^2) 
	- ((xp^2 * xq^3) * xr) + ((xp^2 * xq^3) * xs) 
	+ ((xp^2 * xq) * xr^3) - ((xp^2 * xq) * xs^3) 
	- ((xp^2 * xr^3) * xs) + ((xp^2 * xr) * xs^3) 
	+ ((xp * xq^3) * xr^2) - ((xp * xq^3) * xs^2) 
	- ((xp * xq^2) * xr^3) + ((xp * xq^2) * xs^3) 
	+ ((xp * xr^3) * xs^2) - ((xp * xr^2) * xs^3) 
	- ((xq^3 * xr^2) * xs) + ((xq^3 * xr) * xs^2) 
	+ ((xq^2 * xr^3) * xs) - ((xq^2 * xr) * xs^3) 
	- ((xq * xr^3) * xs^2) + ((xq * xr^2) * xs^3))
	local d = ((((xp^(3) * xq^(2)) * xr) * ys) 
	- (((xp^(3) * xq^(2)) * xs) * yr) - (((xp^(3) * xq) * xr^(2)) * ys) 
	+ (((xp^(3) * xq) * xs^(2)) * yr) + (((xp^(3) * xr^(2)) * xs) * yq) 
	- (((xp^(3) * xr) * xs^(2)) * yq) - (((xp^(2) * xq^(3)) * xr) * ys) 
	+ (((xp^(2) * xq^(3)) * xs) * yr) + (((xp^(2) * xq) * xr^(3)) * ys) 
	- (((xp^(2) * xq) * xs^(3)) * yr) - (((xp^(2) * xr^(3)) * xs) * yq) 
	+ (((xp^(2) * xr) * xs^(3)) * yq) + (((xp * xq^(3)) * xr^(2)) * ys) 
	- (((xp * xq^(3)) * xs^(2)) * yr) - (((xp * xq^(2)) * xr^(3)) * ys) 
	+ (((xp * xq^(2)) * xs^(3)) * yr) + (((xp * xr^(3)) * xs^(2)) * yq) 
	- (((xp * xr^(2)) * xs^(3)) * yq) - (((xq^(3) * xr^(2)) * xs) * yp) 
	+ (((xq^(3) * xr) * xs^(2)) * yp) + (((xq^(2) * xr^(3)) * xs) * yp) 
	- (((xq^(2) * xr) * xs^(3)) * yp) - (((xq * xr^(3)) * xs^(2)) * yp) 
	+ (((xq * xr^(2)) * xs^(3)) * yp)) / 
	(((xp^(3) * xq^(2)) * xr) - 
	((xp^(3) * xq^(2)) * xs) - ((xp^(3) * xq) * xr^(2)) 
	+ ((xp^(3) * xq) * xs^(2)) + ((xp^(3) * xr^(2)) * xs) 
	- ((xp^(3) * xr) * xs^(2)) - ((xp^(2) * xq^(3)) * xr) 
	+ ((xp^(2) * xq^(3)) * xs) + ((xp^(2) * xq) * xr^(3)) 
	- ((xp^(2) * xq) * xs^(3)) - ((xp^(2) * xr^(3)) * xs) 
	+ ((xp^(2) * xr) * xs^(3)) + ((xp * xq^(3)) * xr^(2)) 
	- ((xp * xq^(3)) * xs^(2)) - ((xp * xq^(2)) * xr^(3)) 
	+ ((xp * xq^(2)) * xs^(3)) + ((xp * xr^(3)) * xs^(2)) 
	- ((xp * xr^(2)) * xs^(3)) - ((xq^(3) * xr^(2)) * xs) 
	+ ((xq^(3) * xr) * xs^(2)) + ((xq^(2) * xr^(3)) * xs) 
	- ((xq^(2) * xr) * xs^(3)) - ((xq * xr^(3)) * xs^(2)) 
	+ ((xq * xr^(2)) * xs^(3)))
	return a, b, c, d
end

-- f(x)=a*x+b
local function parameters_affine(xp,yp,xq,yq)
	local a = (yp - yq) / (xp - xq)
	local b = ((xp * yq) - (xq * yp)) / (xp - xq)
	return a, b
end

-- what is the sum of the squared error
-- when comparing the bezier path
-- p.. control q and r .. s
-- with the graph g from index starti to endi
-- (looking at the points at roughly t=.33 and t=.67)
local function squareerror(f,g,starti,endi,qx,qy,rx,ry)
	local result = 0
	for t = .1, .9, .1 do
		x = (1-t)^3*g[starti][1]+3*t*(1-t)^2*qx+3*t^2*(1-t)*rx+t^3*g[endi][1]
		y = (1-t)^3*g[starti][2]+3*t*(1-t)^2*qy+3*t^2*(1-t)*ry+t^3*g[endi][2]
		result = result + (y-f(x))^2
	end
	return result
end

-- converts a table with bezier point information
-- to a string with rounded values 
-- the path is reversed, if rev is true
-- e.g. if bezierpoints = {{0,1},{2,3,4,5,6,7},{8,9,10,11,12,13}}
-- then 
-- (0,1) .. controls (2,3) and (4,5) .. (6,7) .. controls 
-- (8,9) and (10,11) .. (12,13)
-- will be returned
-- the notation "pgfplots" will change the notation to
-- YES: \addplot coordinates {(0,1) (6,7) (2,3) (4,5) (6,7) (12,13) (8,9) (10,11)}
-- NO: 0  1 \\ 6  7 \\ 2  3 \\ 4  5 \\ \\ 6  7 \\ 12 13 \\ 8  9 \\ 10 11 \\
-- As pgfplots does not connect the bezier segments
-- reverse paths are not implemented 
local function beziertabletostring(bezierpoints,rndx,rndy,rev,notation)
	local bezierstring = ""
	local b = {{round(bezierpoints[1][1],rndx),round(bezierpoints[1][2],rndy)}} -- rounded and then 
	-- reduced points (if identical after rounding)
	-- rounding
	for i = 2, #bezierpoints do
		-- check if x--coordinates are identical
		if round(bezierpoints[i][#bezierpoints[i]-1],rndx) ~=  b[#b][#b[#b]-1] then
			b[#b+1] = {}
			for j = 1, #bezierpoints[i] do
				if j % 2 == 0 then -- x coordinate
					b[#b][j] = round(bezierpoints[i][j],rndx)
				else
					b[#b][j] = round(bezierpoints[i][j],rndy)
				end
			end
		end
	end
	if #b > 1 then -- if not empty or single point
		-- check if bezierstring contains only straight lines
		local onlystraightlines = true
		for i = 1, #b do
			if #b[i] > 2 then
				onlystraightlines = false
				break
			end
		end
		if onlystraightlines then
			if rev then
				bezierstring = "(" .. b[#b][1] .. "," .. b[#b][2] ..")"
					for i = #b-1, 1, -1 do
						bezierstring = bezierstring .. 
							" -- (" .. b[i][1] .. "," .. b[i][2] ..")"
					end
			else
				if notation == "pgfplots" then
					bezierstring = "\\addplot coordinates {(" 
						.. b[1][1] .. "," .. b[1][2] .. ") (" 
						.. b[2][1] .. "," .. b[2][2] .. ") (" 
						.. b[1][1] .. "," .. b[1][2] .. ") (" 
						.. b[2][1] .. "," .. b[2][2] .. ") }" 
				else -- notation = tikz
					bezierstring = "(" .. b[1][1] .. "," .. b[1][2] ..")"
					for i = 2, #b do
						bezierstring = bezierstring .. 
							" -- (" .. b[i][1] .. "," .. b[i][2] ..")"
					end
				end	
			end
		else
			if rev then
				bezierstring = "(" .. b[#b][#b[#b]-1] .. "," 
				.. b[#b][#b[#b]] ..")" -- initial point
				for i = #b, 2, -1 do
					if #b[i] >= 6 then -- cubic bezier spline
						bezierstring = bezierstring .. " .. controls (" 
						.. b[i][3] .. "," .. b[i][4] ..") and ("
						.. b[i][1] .. "," .. b[i][2] .. ") .. (" 
						.. b[i-1][#b[i-1]-1] .. "," .. b[i-1][#b[i-1]]..")"
					else
						bezierstring = bezierstring .. " (" 
						.. b[i-1][#b[i-1]-1] .. "," .. b[i-1][#b[i-1]] ..")"
					end
				end
			else
				if notation == "pgfplots" then
					bezierstring = "\\addplot coordinates {"
					for i = 1, #b-1 do
						if #b[i+1] >= 6 then -- cubic bezier spline
							bezierstring = bezierstring .. "("
							.. b[i][#b[i]-1] .. "," .. b[i][#b[i]] .. ") (" 
							.. b[i+1][5] .. "," .. b[i+1][6] .. ") ("  
							.. b[i+1][1] .. "," .. b[i+1][2] .. ") (" 
							.. b[i+1][3] .. "," .. b[i+1][4] .. ") " 
						end
					end
					bezierstring = bezierstring .. "}"
				else -- notation = tikz
					bezierstring = "(" .. b[1][1] .. "," 
					.. b[1][2] ..")" -- initial point
					for i = 2, #b do
						if #b[i] >= 6 then -- cubic bezier spline
							bezierstring = bezierstring .. " .. controls (" 
							.. b[i][1] .. "," .. b[i][2] ..") and ("
							.. b[i][3] .. "," .. b[i][4] .. ") .. (" 
							.. b[i][5] .. "," .. b[i][6]..")"
						else
							bezierstring = bezierstring .. " (" 
							.. b[i][1] .. "," .. b[i][2] ..")"
						end
					end
				end
			end
		end
	end
	return bezierstring
end

-- take end points of a graph g of the function f
-- (from indices starti to endi)
-- without extrema or inflection points inbetween 
-- and try to approximate it with a cubic bezier curve
-- (round to rndx and rndy when printing)
-- if maxerror <= 0, the function will not be recursive anymore
local function graphtobezierapprox(f,g,starti,endi,maxerror)
	local px = g[starti][1]
	local py = g[starti][2]
	local dp = g[starti][3]
	local sx = g[endi][1]
	local sy = g[endi][2]
	local ds = g[endi][3]
	-- we compute the corner point c, where the controls would meet
	local cx = ((dp * px) - (ds * sx) - py + sy) / (dp - ds)
	local cy = (dp * ((ds * px) - (ds * sx) - py + sy) / (dp - ds)) + py
	-- now we slide q between p and c & r between s and c
	-- and search for the best qx and best rx
	local qx = px+.01*(cx-px)
	local qy = py+.01*(cy-py)
	local rx = sx+.01*(cx-sx)
	local ry = sy+.01*(cy-sy)
	local err = squareerror(f,g,starti,endi,qx,qy,rx,ry)
	for i = 2, 99 do
		for j = 2, 99 do
			xa = px+i*.01*(cx-px)
			ya = py+i*.01*(cy-py)
			xb = sx+j*.01*(cx-sx)
			yb = sy+j*.01*(cy-sy)
			-- now check, if xa and xb fit better
			-- than the last qx and rx did
			-- (sum of squares must be smaller)
			local newerror = squareerror(f,g,starti,endi,xa,ya,xb,yb)
			if newerror < err then
				qx = xa
				qy = ya
				rx = xb
				ry = yb
				err = newerror
			end
		end
	end
	if maxerror > 0 then
		-- check if it is close enough: (recycling err, xa, ya)
		err = 0
		for t = .1, .9, .1 do
			xa = (1-t)^3*g[starti][1]+3*t*(1-t)^2*qx+3*t^2*(1-t)*rx+t^3*g[endi][1]
			ya = (1-t)^3*g[starti][2]+3*t*(1-t)^2*qy+3*t^2*(1-t)*ry+t^3*g[endi][2]
			if abs(ya-f(xa)) > err then
				err = abs(ya-f(xa))
				err = abs(ya-f(xa))
			end
		end
		if (err <= maxerror)
		and qx > -math.huge and qx < math.huge
		and qy > -math.huge and qy < math.huge
		and rx > -math.huge and ry < math.huge
		and sx > -math.huge and sy < math.huge
		then
			return {qx,qy,rx,ry,sx,sy}
		else
			-- search for an intermediate point where the graph has the same
			-- slope as the line from the start point to the end point:
			local interindex = math.floor(.5*starti+.5*endi) -- will change
			for i = starti + 1, endi - 1 do
				if abs(g[i][3]-(g[endi][2]-g[starti][2])
				/(g[endi][1]-g[starti][1])) 
				< abs(g[interindex][3]-(g[endi][2]-g[starti][2])
				/(g[endi][1]-g[starti][1])) then
					interindex = i
				end
			end
			local left = graphtobezierapprox(f,g,starti,interindex,maxerror)
			local right = graphtobezierapprox(f,g,interindex,endi,maxerror)
			for i=1, #right do --now append the right to the left:
				left[#left+1] = right[i]
			end
			return left
		end
	elseif qx > -math.huge and qx < math.huge
	and qy > -math.huge and qy < math.huge
	and rx > -math.huge and ry < math.huge
	and sx > -math.huge and sy < math.huge
	then
		return {qx,qy,rx,ry,sx,sy}
	end
end

-- like above but exact for quadratic and cubic (if not inverse)
-- resp. exact for squareroot and cuberoot (if inverse)
local function graphtobezier(g,starti,endi,isinverse)
	local px = g[starti][1]
	local py = g[starti][2]
	local dp = g[starti][3]
	local sx = g[endi][1]
	local sy = g[endi][2]
	local ds = g[endi][3]
	local qx = px+(sx-px)/3
	local rx = px+2*(sx-px)/3
	local qy = py+(qx-px)*dp
	local ry = sy+(rx-sx)*ds
	if isinverse then
		return {qy,qx,ry,rx,sy,sx}
	else
		return {qx,qy,rx,ry,sx,sy}
	end
end

-- main function
function bezierplot(functionstring,xminstring,xmaxstring,yminstring,ymaxstring,samplesstring,notation)
	local fstringreplaced = string.gsub(functionstring, "%*%*", "^")
	local f = assert(load("local x = ...; return " .. fstringreplaced)) 
	local xmin = evaluate(xminstring)
	local xmax = evaluate(xmaxstring)
	local ymin = evaluate(yminstring)
	local ymax = evaluate(ymaxstring)
	local samples = evaluate(samplesstring)
	local isreverse = false
	if xmin > xmax then
		isreverse = true
	elseif xmin == xmax then
		xmax = xmin + 10
	end
	xmin, xmax = math.min(xmin,xmax), math.max(xmin,xmax)
	if ymin == ymax then
		ymax = ymin + 10
	end
	ymin, ymax = math.min(ymin,ymax), math.max(ymin,ymax)
	local xsteps = 50000 
	-- if samples < 2 the samples will be chosen as wisely as possible
	local arbitrary_samples = true
	if samples >= 2 then
		arbitrary_samples = false
		xsteps = (samples-1)*math.max(2,math.floor(xsteps/(samples-1)))
	end
	local xstep = (xmax-xmin)/xsteps
	-- the output of the x coordinates will be rounded to rndx digits
	local rndx = math.max(0,math.floor(5.5-log(xmax-xmin)/log(10)))
	local xerror = abs(xmax-xmin)/(10^rndx)
	-- the output of the y coordinates will be rounded to rndy digits
	local rndy = math.max(0,math.floor(5.5-log(ymax-ymin)/log(10)))
	local yerror = (ymax-ymin)/(10^rndy)
	-- determine parts of the graph that are inside window
	local graphs = {} -- graph split to the connected parts
	local graph = {} -- graphs concatenated (needed for function type)
	local outside = true -- value is outside window
	local i = 0
	local j = 0
	local yminreal -- determine the real minimimum of the y coord.
	local ymaxreal -- just decring
	local yminrealfound = false
	local ymaxrealfound = false
	for n = 0, xsteps do
		local x = xmin + n/xsteps*(xmax-xmin)
		if n == xsteps then
			x = xmax
		end
		local y = f(x)
		if (y >= ymin-.1*yerror and ymin ~= -huge or y > ymin and ymin == -huge)
		and (y <= ymax+.1*yerror and ymax ~= huge or y < ymax and ymax == huge)
		then -- inside
			if outside then -- if it was outside before
				outside = false
				j = 0
				i = i + 1
				graphs[i] = {}
			end
			j = j + 1
			graphs[i][j] = {x,y}
			graph[#graph+1] = {x,y}
			if not yminrealfound or yminrealfound and y < yminreal then
				yminreal = y
				yminrealfound = true
			end
			if not ymaxrealfound or ymaxrealfound and y > ymaxreal then
				ymaxreal = y
				ymaxrealfound = true
			end
		else
			outside = true
		end
	end
	
	-- some redefinitions
	if #graph ~= 0 and yminreal ~= ymaxreal then
		ymin = yminreal
		ymax = ymaxreal
		rndy = math.max(0,math.floor(5.5-log(ymax-ymin)/log(10)))
		yerror = (ymax-ymin)/(10^rndy)
	end
	
	-- check for the function type (for this, we need the concatenated
	-- parts of the graph)
	-- go through the connected parts
	local functiontype = "unknown"
	local a, b, c, d -- possible function parameter
	-- check for affine functions:
	local l = #graph
	a, b = parameters_affine(graph[1][1],graph[1][2],
	graph[l][1],graph[l][2])
	if do_parameters_fit(a,b,0,0,"a*x+b",graph,yerror,false) then
		functiontype = "affine"
	else -- check for cubic functions (includes quadratic functions)
		a, b, c, d = parameters_cubic(graph[1][1],graph[1][2],
		graph[math.floor(l/3)][1],graph[math.floor(l/3)][2],
		graph[math.floor(2*l/3)][1],graph[math.floor(2*l/3)][2],
		graph[l][1],graph[l][2])
		if do_parameters_fit(a,b,c,d,"a*x^3+b*x^2+c*x+d",graph,
		yerror,false) then
			functiontype = "cubic"
		else -- check for cuberoot functions (includes squareroots)
			a, b, c, d = parameters_cubic(graph[1][2],graph[1][1],
			graph[math.floor(l/3)][2],graph[math.floor(l/3)][1],
			graph[math.floor(2*l/3)][2],graph[math.floor(2*l/3)][1],
			graph[l][2],graph[l][1])
			if do_parameters_fit(a,b,c,d,"a*x^3+b*x^2+c*x+d",graph,
			xerror,true) then
				functiontype = "cuberoot"
			end
		end
	end
			
	local bezierpoints = {}
	-- the bezier path (0,1) .. controls 
	-- (2,3) and (4,5) .. (6,7) .. controls 
	-- (8,9) and (10,11) .. (12,13)
	-- will be stored as
	-- bezierpoints={{0,1},{2,3,4,5,6,7},{8,9,10,11,12,13}}
	
	if functiontype == "affine" then
		if arbitrary_samples then
			bezierpoints = {{graph[1][1],graph[1][2]},{graph[#graph][1],
			graph[#graph][2]}}
		else -- we can here savely assume that graphs has only one part,
		-- therefore graphs[1]=graph
			for i = 1, #graph, math.floor(xsteps/(samples-1)) do
				bezierpoints[#bezierpoints+1] = {graph[i][1],graph[i][2]}
			end
		end
	elseif functiontype == "cubic" then 
		local extrema_inflections = {} -- store the extrema and
		-- inflection points for arbitrary samples
		if arbitrary_samples then
			if math.abs(a) < yerror*1e-10 then -- quadratic case (one extremum)
				if is_in_window(-c/(2*b),(-c^2+4*b*d)/(4*b),xmin,xmax,
				ymin,ymax) then
					extrema_inflections = {{-c/(2*b),(-c^2+4*b*d)/(4*b)}}
				end
			else -- cubic case (two extrema and one inflection point)
				-- we order the points with the help of sgn
				-- check for first extrema
				if is_in_window((-sgn(a)*sqrt(-3*a*c+b^2)-b)/(3*a),
				(2*b^3+27*a^2*d-9*a*b*c+sqrt(b^2-3*a*c)*sgn(a)*
				(2*b^2-6*a*c))/(27*a^2),xmin,xmax,ymin,ymax) then
					extrema_inflections[#extrema_inflections+1] = 
					{(-sgn(a)*sqrt(-3*a*c+b^2)-b)/(3*a),(2*b^3+27*a^2*d-
					9*a*b*c+sqrt(b^2-3*a*c)*sgn(a)*(2*b^2-6*a*c))/(27*a^2)}
				end
				-- check for inflection point (has to be inbetween)
				if is_in_window(-b/(3*a),(2*b^3+27*a^2*d-9*a*b*c)
				/(27*a^2),xmin,xmax,ymin,ymax) then
					extrema_inflections[#extrema_inflections+1]={-b/(3*a),
					(2*b^3+27*a^2*d-9*a*b*c)/(27*a^2)}
				end
				-- check for second extrema
				if is_in_window((sgn(a)*sqrt(-3*a*c+b^2)-b)/(3*a),
				(2*b^3+27*a^2*d-9*a*b*c+sqrt(b^2-3*a*c)*sgn(a)*
				(-2*b^2+6*a*c))/(27*a^2),xmin,xmax,ymin,ymax) then
					extrema_inflections[#extrema_inflections+1] = 
					{(sgn(a)*sqrt(-3*a*c+b^2)-b)/(3*a),(2*b^3+27*a^2*d-
					9*a*b*c+sqrt(b^2-3*a*c)*sgn(a)*(-2*b^2+6*a*c))/(27*a^2)}
				end
			end
		end
		for part = 1, #graphs do 
			bezierpoints[#bezierpoints+1] = {graphs[part][1][1],
			graphs[part][1][2]} -- initial points
			local graphsamples = {}-- will be the graph reduced to the 
			-- samples (or the most important points)
			local dg -- will be the differentiated graph
			if arbitrary_samples then -- add extrema and inflection 
			-- points to the border points
				graphsamples = {{graphs[part][1][1],
					graphs[part][1][2]}}
				for j = 1, #extrema_inflections do
					if extrema_inflections[j][1] > math.min(
					graphs[part][1][1] ,graphs[part][#graphs[part]][1])
					and extrema_inflections[j][1] < math.max(
					graphs[part][1][1] ,graphs[part][#graphs[part]][1])
					then
						graphsamples[#graphsamples+1] = 
							{extrema_inflections[j][1],
							extrema_inflections[j][2]}
					end
				end
				graphsamples[#graphsamples+1] = 
					{graphs[part][#graphs[part]][1],
					graphs[part][#graphs[part]][2]}
			else
				for i = 1, #graphs[part], xsteps/(samples-1) do
					graphsamples[#graphsamples+1] = 
						{graphs[part][i][1],graphs[part][i][2]}
				end
			end
			dg = diffgraphcubic(graphsamples,a,b,c,d,false)
			for i = 2, #dg do
				bezierpoints[#bezierpoints+1] = graphtobezier(dg,i-1,i,false)
			end
		end
	elseif functiontype == "cuberoot" then 
		local inflection = {} -- store the inflection point
		if arbitrary_samples and math.abs(a) ~= 0
		and is_in_window((2*b^3+27*a^2*d-9*a*b*c)/(27*a^2),-b/(3*a),
		xmin,xmax,ymin,ymax) then
			inflection = {(2*b^3+27*a^2*d-9*a*b*c)/(27*a^2),-b/(3*a)}
		end
		-- (there cannot be more than one part)
		bezierpoints[#bezierpoints+1] = {graphs[1][1][1],
		graphs[1][1][2]} -- initial points
		local graphsamples = {}-- will be the graph reduced to the 
		-- samples (or the most important points)
		local dg -- will be the differentiated graph
		if arbitrary_samples then -- add inflection point (if exis.)
			graphsamples = {{graphs[1][1][1],
				graphs[1][1][2]}}
			if #inflection > 0 and inflection[1] > math.min(
				graphs[1][1][1],graphs[1][#graphs[1]][1])
				and inflection[1] < math.max(
				graphs[1][1][1],graphs[1][#graphs[1]][1])
				then
					graphsamples[#graphsamples+1] = 
						{inflection[1],inflection[2]}
			end
			graphsamples[#graphsamples+1] = 
				{graphs[1][#graphs[1]][1],
				graphs[1][#graphs[1]][2]}
		else
			for i = 1, #graphs[1], xsteps/(samples-1) do
				graphsamples[#graphsamples+1] = 
					{graphs[1][i][1],graphs[1][i][2]}
			end
		end
		dg = diffgraphcubic(graphsamples,a,b,c,d,true)
		for i = 2, #dg do
			bezierpoints[#bezierpoints+1] = graphtobezier(dg,i-1,i,true)
		end
	else	
	---------- generic case (no special function) ----------------		
		if arbitrary_samples then
			-- go through the connected parts
			for part = 1, #graphs do 
				local dg = diffgraph(f,graphs[part],xstep)
				--printdifftable(dg) -- for debugging
				bezierpoints[#bezierpoints+1] = {dg[1][1],dg[1][2]}
				local startindex = 1
				for k = 2, #dg do
					if dg[k][5] or dg[k][6] then -- extrema and inflection points
						local tobeadded = graphtobezierapprox(
						f,dg,startindex,k,10*yerror)
						-- tobeadded may contain a multiple of 6 entries
					-- e.g. {1,2,3,4,5,6,7,8,9,10,11,12}
						for i = 1, math.floor(#tobeadded/6) do
							bezierpoints[#bezierpoints+1] = {}
							for j = 1, 6 do
								bezierpoints[#bezierpoints][j] = tobeadded[(i-1)*6+j]
							end
						end
						startindex = k
					end
				end
				if startindex ~= #dg then -- if no special points inbetween
					local tobeadded = graphtobezierapprox(f,dg,
					startindex,#dg,10*yerror)
					-- tobeadded may contain a multiple of 6 entries
					-- e.g. {1,2,3,4,5,6,7,8,9,10,11,12}
					for i = 1, math.floor(#tobeadded/6) do
						bezierpoints[#bezierpoints+1] = {}
						for j = 1, 6 do
							bezierpoints[#bezierpoints][j] = tobeadded[(i-1)*6+j]
						end
					end
				end
			end
		else -- fixed samples in the generic case
			-- go through the connected parts
			for part = 1, #graphs do 
				local dg = diffgraphsimple(f,graphs[part],xstep,
					math.floor(0.5+xsteps/(samples-1)))
				bezierpoints[#bezierpoints+1] = {dg[1][1],dg[1][2]} -- initial points
				for i = 2, #dg do
					bezierpoints[#bezierpoints+1] = graphtobezier(dg,i-1,i,false)
				end
			end
		end
	end
	return beziertabletostring(bezierpoints,rndx,rndy,isreverse,notation)		
end

-- main program --

if not pcall(debug.getlocal, 4, 1) then
--if debug.getinfo(3) == nil then
	if #arg >= 1 then
		local xmin = -5
		local xmax = 5
		if #arg >= 2 then 
			xmin = arg[2]
		end
		if #arg >= 3 then
			xmax = arg[3]
		end
		local ymin = -5
		local ymax = 5
		if #arg >= 4 then 
			ymin = arg[4]
		end
		if #arg >= 5 then 
			ymax = arg[5]
		end
		local samples = 0
		if #arg >= 6 then 
			samples = arg[6]
		end
		local notation = "tikz"
		if #arg >= 7 then 
			notation = arg[7] 
		end
		print(bezierplot(arg[1],xmin,xmax,ymin,ymax,samples,notation))
	end
end