1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
|
--!A cross-platform build utility based on Lua
--
-- Licensed under the Apache License, Version 2.0 (the "License");
-- you may not use this file except in compliance with the License.
-- You may obtain a copy of the License at
--
-- http://www.apache.org/licenses/LICENSE-2.0
--
-- Unless required by applicable law or agreed to in writing, software
-- distributed under the License is distributed on an "AS IS" BASIS,
-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-- See the License for the specific language governing permissions and
-- limitations under the License.
--
-- Copyright (C) 2015-present, TBOOX Open Source Group.
--
-- @author ruki
-- @file profiler.lua
--
-- define module
local profiler = {}
-- load modules
local os = require("base/os")
local path = require("base/path")
local table = require("base/table")
local utils = require("base/utils")
local string = require("base/string")
-- get the function title
function profiler:_func_title(funcinfo)
local name = funcinfo.name or 'anonymous'
local line = string.format("%d", funcinfo.linedefined or 0)
local source = funcinfo.short_src or 'C_FUNC'
if os.isfile(source) then
source = path.relative(source, xmake._PROGRAM_DIR)
end
return string.format("%-30s: %s: %s", name, source, line)
end
-- get the function report
function profiler:_func_report(funcinfo)
local title = self:_func_title(funcinfo)
local report = self._REPORTS_BY_TITLE[title]
if not report then
report =
{
title = self:_func_title(funcinfo)
, callcount = 0
, totaltime = 0
}
self._REPORTS_BY_TITLE[title] = report
table.insert(self._REPORTS, report)
end
return report
end
-- profiling call
function profiler:_profiling_call(funcinfo)
local report = self:_func_report(funcinfo)
report.calltime = os.clock()
report.callcount = report.callcount + 1
end
-- profiling return
function profiler:_profiling_return(funcinfo)
local stoptime = os.clock()
local report = self:_func_report(funcinfo)
if report.calltime and report.calltime > 0 then
report.totaltime = report.totaltime + (stoptime - report.calltime)
report.calltime = 0
end
end
-- the profiling handler
function profiler._profiling_handler(hooktype)
local funcinfo = debug.getinfo(2, 'nS')
if hooktype == "call" then
profiler:_profiling_call(funcinfo)
elseif hooktype == "return" then
profiler:_profiling_return(funcinfo)
end
end
-- the tracing handler
function profiler._tracing_handler(hooktype)
local funcinfo = debug.getinfo(2, 'nS')
if hooktype == "call" then
local name = funcinfo.name
local source = funcinfo.short_src
if name and source and source:endswith(".lua") then
local line = string.format("%d", funcinfo.linedefined or 0)
utils.print("%-30s: %s: %s", name, source, line)
end
end
end
-- start profiling
function profiler:start()
local mode = self:mode()
if mode and mode == "trace" then
debug.sethook(profiler._tracing_handler, 'cr', 0)
else
self._REPORTS = {}
self._REPORTS_BY_TITLE = {}
self._STARTIME = os.clock()
debug.sethook(profiler._profiling_handler, 'cr', 0)
end
end
-- stop profiling
function profiler:stop()
-- trace?
local mode = self:mode()
if mode and mode == "trace" then
-- stop to hook
debug.sethook()
else
-- save the stop time
self._STOPTIME = os.clock()
-- stop to hook
debug.sethook()
-- calculate the total time
local totaltime = self._STOPTIME - self._STARTIME
-- sort reports
table.sort(self._REPORTS, function(a, b)
return a.totaltime > b.totaltime
end)
-- show reports
for _, report in ipairs(self._REPORTS) do
-- calculate percent
local percent = (report.totaltime / totaltime) * 100
if percent < 1 then
break
end
-- trace
utils.print("%6.3f, %6.2f%%, %7d, %s", report.totaltime, percent, report.callcount, report.title)
end
end
end
-- get profiler mode
function profiler:mode()
local mode = self._MODE
if not mode then
mode = os.getenv("XMAKE_PROFILE")
self._MODE = mode
end
return mode
end
-- profiler is enabled?
function profiler:enabled()
return self:mode() ~= nil
end
-- return module
return profiler
|
