Piefiller/piefiller.lua

local path = ...
local piefiller = {}
local function hsvToRgb(h, s, v)
	local h,s,v = h,1,1
    local h,s,v = math.fmod(h,360),s,v
    if s==0 then return { v,v,v } end
    h=h/60
    i=math.floor(h)
    f=h-i
    p=v*(1-s)
    q=v*(1-s*f)
    t=v*(1-s*(1-f))
    
    if i==0 then r=v g=t b=p
    elseif i==1 then r=q g=v b=p
    elseif i==2 then r=p g=v b=t
    elseif i==3 then r=p g=q b=v
    elseif i==4 then r=t g=p b=v
    else r=v g=p b=q
    end
    
    r=r*255
    g=g*255
    b=b*255
    return { r,g,b }
end
local function copy(t)
	local ret = {} 
	for i,v in pairs(t) do 
		ret[i] = v 
		if type(v) == "table" then 
			ret[i] = copy(v)
		end 
	end 
	return ret 
end 
function setColor(...)
	local args = {...} 
	love.graphics.setColor(args[1] or 255,args[2] or 255,args[3] or 255,args[4] or 255)
end 
local color_data = {}
local colors = {}  
function piefiller:new()
    local self = {} 
    setmetatable(self,{__index = piefiller})
    self.data = {}
	self.parsed = {} 
    self.last = 0 
	self.timer = 0 
    self:reset()
	self.depth = 2 
	self.small = false 
    self.x = 0 
	self.y = 0
	self.scale = 1
	self.step = 1
	self.keys = {
		reset = "r",
		increase_depth = "down", 
		decrease_depth = "up",
		increase_step_size = ",",
		decrease_step_size = ".",
		shorten_names = "s",
		show_hidden = "c",
		save_to_file = "p",
	}
	return self
end
function piefiller:reset()
	self.data = {} 
	self.x = 0 
	self.y = 0
	self.scale = 1
	for i=0,300 do 
		table.insert(colors,hsvToRgb(i,100,100))
	end 
end 
function piefiller:setKey(table_or_mode,key)
	if type(table_or_mode) == "table" then 
		self.keys = table_or_mode
		for i,v in pairs(table_or_mode) do 
			if self.keys[i] then self.keys[i] = v end			
		end 
	elseif type(table_or_mode) == "string" then 
		if not self.keys[table_or_mode] then error("Invalid mode: "..tostring(table_or_mode)) end
		self.keys[table_or_mode] = key
	else 
		error("Expected table or string got:"..type(table_or_mode))
	end 
end 

function piefiller:parse(caller,parent)
	return {
		parent = parent,
		func = caller.func,
		count = 0,
		time = 0,
		child_time = 0, 
		named_child_time = 0, 
		children = {},
		children_time = {},
		info = caller,
		kids = {},
	}
end 
function piefiller:attach()
   self.last = os.clock()
   local function hook()
        local depth = self.depth 
        local caller = debug.getinfo(depth)
        local taken = os.clock() - self.last
		if caller then 
			local last_caller
			local own = string.find(caller.source,path)
			if caller.func ~= hook and not own then  
				while caller do 
					if last_caller and not self.view_children then
						local name = caller.func 
						local lc = self.data[last_caller.func]
						if not lc.kids[name] then 
							lc.kids[name] = self:parse(caller,last_caller)
						end 
						local kid = lc.kids[name]
						kid.count = kid.count + 1 
						kid.time = kid.time + taken 
					else 
						local name = caller.func 
						local raw = self.data[name] 
						if not raw then
							self.data[name] = self:parse(caller,last_caller)
						end
						raw = self.data[name]
						raw.count = raw.count + 1 
						raw.time = raw.time + taken
						last_caller = caller
					end 
					depth = depth + 1 
					caller = debug.getinfo(depth)
				end 
			end
		end 
    end
	local step = 10^self.step
	if self.step < 0 then 
		step = 1/-self.step 
	end 
	debug.sethook(hook,"",step)
end 
function piefiller:detach(stop)
    local totaltime = 0
    local parsed = {}
    local no = 0 
    for i,v in pairs(self.data) do 
        no = no + 1
        totaltime = totaltime + v.time 
		local i = no 
        parsed[i] = {}
        parsed[i].name = v.info.name
        parsed[i].time = v.time 
		parsed[i].src = v.info.source
        parsed[i].def = v.info.linedefined
        parsed[i].cur = v.info.currentline
		parsed[i].item = v 
		parsed[i].caller = v.info
		if not color_data[v.func] then 
			local i = math.random(#colors)
			color_data[v.func] = colors[i]
			table.remove(colors,i)
		end 
		
		parsed[i].color = color_data[v.func]
	end 
    local prc = totaltime/100
    for i,v in ipairs(parsed) do 
        parsed[i].prc = v.time/prc
    end 
    self.parsed = parsed 
    self.totaltime = totaltime 
	if not stop then debug.sethook() end 
end 
local largeFont = love.graphics.newFont(25)
function piefiller:draw(args)
    local loading 
	local oldFont = love.graphics.getFont()
	local args = args or {}
	local rad = args["rad"] or 200 
    local mode = args["mode"] or "simple"
    local ret = args["return"]
    local pi = math.pi  
	local arc = love.graphics.arc 
	local w,h = love.graphics.getDimensions()
	
	love.graphics.push()
	
	love.graphics.translate(self.x,self.y)
	love.graphics.scale(self.scale)
	
	if self.parsed and self.totaltime > 0  then
        local lastangle = 0
        local font = love.graphics.getFont()
		for i,v in ipairs(self.parsed) do 
            local color = v.color
			local cx,cy = w/2,h/2
			local angle = math.rad(3.6*v.prc)
				setColor(color)
				arc("fill",cx,cy,rad,lastangle,lastangle + angle)
				setColor(colors.black)
				if v.prc > 1 then 
					arc("line",cx,cy,rad,lastangle,lastangle + angle)
				end 
				lastangle = lastangle + angle 
				setColor()
        end 
		lastangle = 0 
		for i,v in ipairs(self.parsed) do 
			
				local color = v.color
				local cx,cy = w/2,h/2
				local angle = math.rad(3.6*v.prc)
				local x = cx + rad * math.cos(lastangle + angle/2)
				local y = cy + rad * math.sin(lastangle + angle/2)
				if self.small then 
					txt = tostring(v.src).." @: "..tostring(v.name)
				else 
					txt = tostring(math.ceil(v.prc)).." % "..tostring(v.name).." : "..tostring(v.src).." @: "..tostring(v.def)
				end 
				local fw = font:getWidth(txt)
				local sx = 1 
				if cx < x then 
					sx = -1 
					fw = 0 
				end 
				if cy + rad/2 < y then 
					y = y + font:getHeight()  
				elseif cy + rad/2 > y then 
					y = y - font:getHeight()
				end 
				local ofx 
				love.graphics.print(txt,((x) + (-(fw+20))*sx),y)
				lastangle = lastangle + angle 		
		end
	else 
		loading  = true 
    end 
	self.timer = self.timer + love.timer.getDelta() 
	if self.timer > 20 then 
		self.timer = 0 
	end 
	love.graphics.setFont(largeFont)
	local t = "Depth: "..self.depth.." with step: "..self.step
	local fw = largeFont:getWidth(t)
	local fh = largeFont:getHeight()
	love.graphics.print(t,w/2 - fw/2,(fh+5))
	if loading then 
		t = "Loading..."
		fw = largeFont:getWidth(t)
		love.graphics.print("Loading...",w/2 - fw/2,h/2)
	end 
	
	love.graphics.pop() 
	
	love.graphics.setFont(oldFont)
end 
function piefiller:mousepressed(x,y,b)
	if b == "wu" then
		local scale = self.scale - math.floor((0.05*self.scale)*1000)/1000
		if scale > 0 and scale > 0.1 then
			local lastzoom = self.scale
			local mouse_x = x - self.x
			local mouse_y = y - self.y
			self.scale = scale
			local newx = mouse_x * (self.scale/lastzoom)
			local newy = mouse_y * (self.scale/lastzoom)
			self.x = self.x + (mouse_x-newx)
			self.y = self.y + (mouse_y-newy)
		else
			self.scale = 0.1
		end
	elseif b == "wd" then
		local scale = self.scale + math.floor((0.05*self.scale)*1000)/1000
		local scalex = self.scale
		if scale > 0 and scale < 20 then
			local lastzoom = self.scale
			local mouse_x = x - self.x
			local mouse_y = y - self.y
			self.scale = scale
			local newx = mouse_x * (self.scale/lastzoom)
			local newy = mouse_y * (self.scale/lastzoom)
			self.x = self.x + (mouse_x-newx)
			self.y = self.y + (mouse_y-newy)
		else
			self.scale = 20
		end
	end		
end
function piefiller:keypressed(key)
	local mode 
	for i,v in pairs(self.keys) do 
		if key == v then 
			mode = i
			break 
		end 
	end 
	if mode then 
		if mode == "reset" then 
			self:reset()
		elseif mode == "increase_depth" then 
			self:reset()	
			self.depth = self.depth + 1 
		elseif mode == "decrease_depth" then 
			self:reset()	
			self.depth = self.depth - 1 
		elseif mode == "increase_step_size" then 
			self.step = self.step - 1
		elseif mode == "decrease_step_size" then 
			self.step = self.step +1 
		elseif mode == "shorten_names" then
			self.small = not self.small 
		elseif mode == "show_hidden" then 
			self:reset()
			self.view_children = not self.view_children
		elseif mode == "save_to_file" then 	
			local parsed = copy(self.parsed)
			table.sort(parsed,function(a,b)
				return a.prc > b.prc
			end)
			local d = {"Depth: "..self.depth.." with step: "..self.step.."\r\n".."Total time: "..self.totaltime.."\r\n"} 
			
			for i,v in ipairs(parsed) do 
				local instance = {
					"-----"..(v.name or "def@"..v.def).."-----",
					"source:"..v.src..":"..v.def,
					"current line: "..v.cur, 
					"time: "..v.time,
					"percentage: "..math.ceil(v.prc).." %",
					"----------------",
				}
				for i,v in ipairs(instance) do 
					instance[i] = v.."\r\n"
				end 
				table.insert(d,table.concat(instance))
			end 
			local data = table.concat(d)
			love.filesystem.write("Profile",data)
			love.system.openURL(love.filesystem.getRealDirectory("Profile"))
		end
	end 
end 
function piefiller:unpack(fn)
	local data = {
		items = {}, 
		about = {
			depth = self.depth,
			step = self.step, 
			totalTime = self.totaltime,
		}, 
	} 
	for i,v in ipairs(self.parsed) do 
		local a = {
			name = v.name,
			line_defined = v.def, 
			current_line = v.cur,
			source = v.src,
			time_taken = v.time, 
			percentage = v.prc, 
			caller = v.caller,
		}
		if fn then 
			assert(type(fn) == "function","Expected function got:"..type(fn))
			fn(a)
		end 
		table.insert(data.items,a)
	end 
	return data 
end 
return piefiller