mirror of
https://github.com/Polynominal/Piefiller.git
synced 2024-11-12 11:14:21 +00:00
376 lines
9.3 KiB
Lua
376 lines
9.3 KiB
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 |