2008-05-13 Kevin McCullough <kmccullough@apple.com>
[WebKit-https.git] / JavaScriptCore / profiler / FunctionCallProfile.cpp
1 /*
2  * Copyright (C) 2008 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  *
8  * 1.  Redistributions of source code must retain the above copyright
9  *     notice, this list of conditions and the following disclaimer.
10  * 2.  Redistributions in binary form must reproduce the above copyright
11  *     notice, this list of conditions and the following disclaimer in the
12  *     documentation and/or other materials provided with the distribution.
13  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
14  *     its contributors may be used to endorse or promote products derived
15  *     from this software without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
18  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
20  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
22  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
23  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
24  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27  */
28
29 #include "config.h"
30 #include "FunctionCallProfile.h"
31
32 #include "Profiler.h"
33 #include "DateMath.h"
34
35 #include <stdio.h>
36
37 namespace KJS {
38
39 FunctionCallProfile::FunctionCallProfile(const UString& name)
40     : m_functionName(name)
41     , m_timeSum(0)
42     , m_numberOfCalls(0)
43 {
44     m_startTime = getCurrentUTCTime();
45 }
46
47 void FunctionCallProfile::willExecute()
48 {
49     m_startTime = getCurrentUTCTime();
50 }
51
52 void FunctionCallProfile::didExecute(Vector<UString> stackNames, unsigned int stackIndex)
53 {
54     if (stackIndex && stackIndex == stackNames.size()) {
55         ASSERT(stackNames[stackIndex - 1] == m_functionName);
56         endAndRecordCall();
57         return;
58     }
59
60     for (StackIterator currentChild = m_children.begin(); currentChild != m_children.end() && stackIndex < stackNames.size(); ++currentChild) {
61         if ((*currentChild)->functionName() == stackNames[stackIndex]) {
62             (*currentChild)->didExecute(stackNames, ++stackIndex);
63             return;
64         }
65     }
66 }
67
68 void FunctionCallProfile::addChild(RefPtr<FunctionCallProfile>& child)
69 {
70     if (!child)
71         return;
72
73     for (StackIterator currentChild = m_children.begin(); currentChild != m_children.end(); ++currentChild) {
74         if ((*currentChild)->functionName() == child->functionName())
75             return;
76     }
77
78     m_children.append(child);
79 }
80
81 FunctionCallProfile* FunctionCallProfile::findChild(const UString& name)
82 {
83     for (StackIterator currentChild = m_children.begin(); currentChild != m_children.end(); ++currentChild) {
84         if ((*currentChild)->functionName() == name)
85             return (*currentChild).get();
86     }
87
88     return 0;
89 }
90
91 void FunctionCallProfile::stopProfiling()
92 {
93     if (m_startTime)
94         endAndRecordCall();
95
96     StackIterator endOfChildren = m_children.end();
97     for (StackIterator it = m_children.begin(); it != endOfChildren; ++it)
98         (*it)->stopProfiling();
99 }
100
101 void FunctionCallProfile::printDataInspectorStyle(int indentLevel) const
102 {
103     // Print function names
104     if (indentLevel) {
105         for (int i = 0; i < indentLevel; ++i)
106             printf("  ");
107
108         printf("%.3fms %s\n", m_timeSum, m_functionName.UTF8String().c_str());
109     } else
110         printf("%s\n", m_functionName.UTF8String().c_str());
111
112     ++indentLevel;
113
114     // Print children's names and information
115     for (StackIterator currentChild = m_children.begin(); currentChild != m_children.end(); ++currentChild)
116         (*currentChild)->printDataInspectorStyle(indentLevel);
117 }
118
119 // print the profiled data in a format that matches the tool sample's output.
120 double FunctionCallProfile::printDataSampleStyle(int indentLevel, FunctionCallHashCount& countedFunctions) const
121 {
122     printf("    ");
123
124     // Print function names
125     const char* name = m_functionName.UTF8String().c_str();
126     double sampleCount = m_timeSum * 1000;
127     if (indentLevel) {
128         for (int i = 0; i < indentLevel; ++i)
129             printf("  ");
130
131          countedFunctions.add(m_functionName.rep());
132
133         printf("%.0f %s\n", sampleCount ? sampleCount : 1, name);
134     } else
135         printf("%s\n", name);
136
137     ++indentLevel;
138
139     // Print children's names and information
140     double sumOfChildrensCount = 0.0;
141     for (StackIterator currentChild = m_children.begin(); currentChild != m_children.end(); ++currentChild)
142         sumOfChildrensCount += (*currentChild)->printDataSampleStyle(indentLevel, countedFunctions);
143
144     sumOfChildrensCount *= 1000;    //
145     // Print remainder of samples to match sample's output
146     if (sumOfChildrensCount < sampleCount) {
147         printf("    ");
148         while (indentLevel--)
149             printf("  ");
150
151         printf("%.0f %s\n", sampleCount - sumOfChildrensCount, m_functionName.UTF8String().c_str());
152     }
153
154     return m_timeSum;
155 }
156
157 void FunctionCallProfile::endAndRecordCall()
158 {
159     m_timeSum += getCurrentUTCTime() - m_startTime;
160     m_startTime = 0.0;
161
162     ++m_numberOfCalls;
163 }
164
165 }   // namespace KJS