1 /*
2 * Copyright (C) 2007 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 * 1. Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR
17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26 #include "config.h"
27 #include "ProgressTracker.h"
28
29 #include "DocumentLoader.h"
30 #include "Frame.h"
31 #include "FrameLoader.h"
32 #include "FrameLoaderClient.h"
33 #include "ResourceResponse.h"
34 #include <wtf/CurrentTime.h>
35
36 using std::min;
37
38 namespace WebCore {
39
40 // Always start progress at initialProgressValue. This helps provide feedback as
41 // soon as a load starts.
42 static const double initialProgressValue = 0.1;
43
44 // Similarly, always leave space at the end. This helps show the user that we're not done
45 // until we're done.
46 static const double finalProgressValue = 0.9; // 1.0 - initialProgressValue
47
48 static const int progressItemDefaultEstimatedLength = 1024 * 16;
49
50 struct ProgressItem {
ProgressItemWebCore::ProgressItem51 ProgressItem(long long length)
52 : bytesReceived(0)
53 , estimatedLength(length) { }
54
55 long long bytesReceived;
56 long long estimatedLength;
57 };
58
ProgressTracker()59 ProgressTracker::ProgressTracker()
60 : m_uniqueIdentifier(0)
61 , m_totalPageAndResourceBytesToLoad(0)
62 , m_totalBytesReceived(0)
63 , m_lastNotifiedProgressValue(0)
64 , m_lastNotifiedProgressTime(0)
65 , m_progressNotificationInterval(0.02)
66 , m_progressNotificationTimeInterval(0.1)
67 , m_finalProgressChangedSent(false)
68 , m_progressValue(0)
69 , m_numProgressTrackedFrames(0)
70 {
71 }
72
~ProgressTracker()73 ProgressTracker::~ProgressTracker()
74 {
75 deleteAllValues(m_progressItems);
76 }
77
estimatedProgress() const78 double ProgressTracker::estimatedProgress() const
79 {
80 return m_progressValue;
81 }
82
reset()83 void ProgressTracker::reset()
84 {
85 deleteAllValues(m_progressItems);
86 m_progressItems.clear();
87
88 m_totalPageAndResourceBytesToLoad = 0;
89 m_totalBytesReceived = 0;
90 m_progressValue = 0;
91 m_lastNotifiedProgressValue = 0;
92 m_lastNotifiedProgressTime = 0;
93 m_finalProgressChangedSent = false;
94 m_numProgressTrackedFrames = 0;
95 m_originatingProgressFrame = 0;
96 }
97
progressStarted(Frame * frame)98 void ProgressTracker::progressStarted(Frame* frame)
99 {
100 // LOG (Progress, "frame %p(%@), _private->numProgressTrackedFrames %d, _private->originatingProgressFrame %p", frame, [frame name], _private->numProgressTrackedFrames, _private->originatingProgressFrame);
101
102 frame->loader()->client()->willChangeEstimatedProgress();
103
104 if (m_numProgressTrackedFrames == 0 || m_originatingProgressFrame == frame) {
105 reset();
106 m_progressValue = initialProgressValue;
107 m_originatingProgressFrame = frame;
108
109 m_originatingProgressFrame->loader()->client()->postProgressStartedNotification();
110 }
111 m_numProgressTrackedFrames++;
112
113 frame->loader()->client()->didChangeEstimatedProgress();
114 }
115
progressCompleted(Frame * frame)116 void ProgressTracker::progressCompleted(Frame* frame)
117 {
118 // LOG (Progress, "frame %p(%@), _private->numProgressTrackedFrames %d, _private->originatingProgressFrame %p", frame, [frame name], _private->numProgressTrackedFrames, _private->originatingProgressFrame);
119
120 if (m_numProgressTrackedFrames <= 0)
121 return;
122
123 frame->loader()->client()->willChangeEstimatedProgress();
124
125 m_numProgressTrackedFrames--;
126 if (m_numProgressTrackedFrames == 0 ||
127 (frame == m_originatingProgressFrame && m_numProgressTrackedFrames != 0))
128 finalProgressComplete();
129
130 frame->loader()->client()->didChangeEstimatedProgress();
131 }
132
finalProgressComplete()133 void ProgressTracker::finalProgressComplete()
134 {
135 // LOG (Progress, "");
136
137 RefPtr<Frame> frame = m_originatingProgressFrame.release();
138
139 // Before resetting progress value be sure to send client a least one notification
140 // with final progress value.
141 if (!m_finalProgressChangedSent) {
142 m_progressValue = 1;
143 frame->loader()->client()->postProgressEstimateChangedNotification();
144 }
145
146 reset();
147
148 frame->loader()->client()->setMainFrameDocumentReady(true);
149 frame->loader()->client()->postProgressFinishedNotification();
150 }
151
incrementProgress(unsigned long identifier,const ResourceResponse & response)152 void ProgressTracker::incrementProgress(unsigned long identifier, const ResourceResponse& response)
153 {
154 // LOG (Progress, "_private->numProgressTrackedFrames %d, _private->originatingProgressFrame %p", _private->numProgressTrackedFrames, _private->originatingProgressFrame);
155
156 if (m_numProgressTrackedFrames <= 0)
157 return;
158
159 long long estimatedLength = response.expectedContentLength();
160 if (estimatedLength < 0)
161 estimatedLength = progressItemDefaultEstimatedLength;
162
163 m_totalPageAndResourceBytesToLoad += estimatedLength;
164
165 if (ProgressItem* item = m_progressItems.get(identifier)) {
166 item->bytesReceived = 0;
167 item->estimatedLength = estimatedLength;
168 } else
169 m_progressItems.set(identifier, new ProgressItem(estimatedLength));
170 }
171
incrementProgress(unsigned long identifier,const char *,int length)172 void ProgressTracker::incrementProgress(unsigned long identifier, const char*, int length)
173 {
174 ProgressItem* item = m_progressItems.get(identifier);
175
176 // FIXME: Can this ever happen?
177 if (!item)
178 return;
179
180 m_originatingProgressFrame->loader()->client()->willChangeEstimatedProgress();
181
182 unsigned bytesReceived = length;
183 double increment, percentOfRemainingBytes;
184 long long remainingBytes, estimatedBytesForPendingRequests;
185
186 item->bytesReceived += bytesReceived;
187 if (item->bytesReceived > item->estimatedLength) {
188 m_totalPageAndResourceBytesToLoad += ((item->bytesReceived * 2) - item->estimatedLength);
189 item->estimatedLength = item->bytesReceived * 2;
190 }
191
192 int numPendingOrLoadingRequests = m_originatingProgressFrame->loader()->numPendingOrLoadingRequests(true);
193 estimatedBytesForPendingRequests = progressItemDefaultEstimatedLength * numPendingOrLoadingRequests;
194 remainingBytes = ((m_totalPageAndResourceBytesToLoad + estimatedBytesForPendingRequests) - m_totalBytesReceived);
195 if (remainingBytes > 0) // Prevent divide by 0.
196 percentOfRemainingBytes = (double)bytesReceived / (double)remainingBytes;
197 else
198 percentOfRemainingBytes = 1.0;
199
200 // For documents that use WebCore's layout system, treat first layout as the half-way point.
201 // FIXME: The hasHTMLView function is a sort of roundabout way of asking "do you use WebCore's layout system".
202 bool useClampedMaxProgress = m_originatingProgressFrame->loader()->client()->hasHTMLView()
203 && !m_originatingProgressFrame->loader()->firstLayoutDone();
204 double maxProgressValue = useClampedMaxProgress ? 0.5 : finalProgressValue;
205 increment = (maxProgressValue - m_progressValue) * percentOfRemainingBytes;
206 m_progressValue += increment;
207 m_progressValue = min(m_progressValue, maxProgressValue);
208 ASSERT(m_progressValue >= initialProgressValue);
209
210 m_totalBytesReceived += bytesReceived;
211
212 double now = currentTime();
213 double notifiedProgressTimeDelta = now - m_lastNotifiedProgressTime;
214
215 // LOG (Progress, "_private->progressValue %g, _private->numProgressTrackedFrames %d", _private->progressValue, _private->numProgressTrackedFrames);
216 double notificationProgressDelta = m_progressValue - m_lastNotifiedProgressValue;
217 if ((notificationProgressDelta >= m_progressNotificationInterval ||
218 notifiedProgressTimeDelta >= m_progressNotificationTimeInterval) &&
219 m_numProgressTrackedFrames > 0) {
220 if (!m_finalProgressChangedSent) {
221 if (m_progressValue == 1)
222 m_finalProgressChangedSent = true;
223
224 m_originatingProgressFrame->loader()->client()->postProgressEstimateChangedNotification();
225
226 m_lastNotifiedProgressValue = m_progressValue;
227 m_lastNotifiedProgressTime = now;
228 }
229 }
230
231 m_originatingProgressFrame->loader()->client()->didChangeEstimatedProgress();
232 }
233
completeProgress(unsigned long identifier)234 void ProgressTracker::completeProgress(unsigned long identifier)
235 {
236 ProgressItem* item = m_progressItems.get(identifier);
237
238 // FIXME: Can this happen?
239 if (!item)
240 return;
241
242 // Adjust the total expected bytes to account for any overage/underage.
243 long long delta = item->bytesReceived - item->estimatedLength;
244 m_totalPageAndResourceBytesToLoad += delta;
245 item->estimatedLength = item->bytesReceived;
246
247 m_progressItems.remove(identifier);
248 delete item;
249 }
250
createUniqueIdentifier()251 unsigned long ProgressTracker::createUniqueIdentifier()
252 {
253 return ++m_uniqueIdentifier;
254 }
255
256
257 }
258