OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef CHROME_RENDERER_PRERENDER_PRERENDER_HELPER_H_ | 5 #ifndef CHROME_RENDERER_PRERENDER_PRERENDER_HELPER_H_ |
6 #define CHROME_RENDERER_PRERENDER_PRERENDER_HELPER_H_ | 6 #define CHROME_RENDERER_PRERENDER_PRERENDER_HELPER_H_ |
7 | 7 |
8 #include "base/compiler_specific.h" | 8 #include "base/compiler_specific.h" |
9 #include "base/macros.h" | 9 #include "base/macros.h" |
10 #include "base/time/time.h" | 10 #include "base/time/time.h" |
11 #include "chrome/common/prerender_types.h" | 11 #include "chrome/common/prerender_types.h" |
12 #include "content/public/renderer/render_frame_observer.h" | 12 #include "content/public/renderer/render_frame_observer.h" |
13 #include "content/public/renderer/render_frame_observer_tracker.h" | 13 #include "content/public/renderer/render_frame_observer_tracker.h" |
14 | 14 |
15 namespace prerender { | 15 namespace prerender { |
16 | 16 |
17 // Helper class to track whether its RenderFrame is currently being prerendered. | 17 // Helper class to track whether its RenderFrame is currently being prerendered. |
18 // Created when prerendering starts and deleted as soon as it stops. | 18 // Created when prerendering starts and deleted as soon as it stops. |
19 class PrerenderHelper | 19 class PrerenderHelper |
20 : public content::RenderFrameObserver, | 20 : public content::RenderFrameObserver, |
21 public content::RenderFrameObserverTracker<PrerenderHelper> { | 21 public content::RenderFrameObserverTracker<PrerenderHelper> { |
22 public: | 22 public: |
23 explicit PrerenderHelper(content::RenderFrame* render_frame); | 23 PrerenderHelper(content::RenderFrame* render_frame, |
| 24 PrerenderMode prerender_mode); |
| 25 |
| 26 // Must be called on a sub-frame. Inherits the prerender mode from the main |
| 27 // frame. |
| 28 explicit PrerenderHelper(content::RenderFrame* sub_frame); |
| 29 |
24 ~PrerenderHelper() override; | 30 ~PrerenderHelper() override; |
25 | 31 |
26 // Returns true if |render_view| is currently prerendering. | 32 // Returns true if |render_view| is currently prerendering. |
27 static bool IsPrerendering(const content::RenderFrame* render_frame); | 33 static bool IsPrerendering(const content::RenderFrame* render_frame); |
28 | 34 |
| 35 static PrerenderMode GetPrerenderMode( |
| 36 const content::RenderFrame* render_frame); |
| 37 |
29 private: | 38 private: |
30 // RenderFrameObserver implementation. | 39 // RenderFrameObserver implementation. |
31 bool OnMessageReceived(const IPC::Message& message) override; | 40 bool OnMessageReceived(const IPC::Message& message) override; |
32 void OnDestruct() override; | 41 void OnDestruct() override; |
33 | 42 |
34 void OnSetIsPrerendering(PrerenderMode mode); | 43 void OnSetIsPrerendering(PrerenderMode mode); |
35 | 44 |
| 45 PrerenderMode prerender_mode_; |
| 46 |
36 DISALLOW_COPY_AND_ASSIGN(PrerenderHelper); | 47 DISALLOW_COPY_AND_ASSIGN(PrerenderHelper); |
37 }; | 48 }; |
38 | 49 |
39 } // namespace prerender | 50 } // namespace prerender |
40 | 51 |
41 #endif // CHROME_RENDERER_PRERENDER_PRERENDER_HELPER_H_ | 52 #endif // CHROME_RENDERER_PRERENDER_PRERENDER_HELPER_H_ |
OLD | NEW |