Add @VisibleForTesting to fix ChromePublic release build.
[chromium-blink-merge.git] / gpu / tools / compositor_model_bench / render_model_utils.h
blob8d6ed3c45c89cef2bd941090fcaa79a6ba6ea946
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
3 // found in the LICENSE file.
5 // Whole-tree processing that's likely to be helpful in multiple render models.
7 #ifndef GPU_TOOLS_COMPOSITOR_MODEL_BENCH_RENDER_MODEL_UTILS_H_
8 #define GPU_TOOLS_COMPOSITOR_MODEL_BENCH_RENDER_MODEL_UTILS_H_
10 #include <map>
11 #include <set>
12 #include <vector>
14 #include "base/compiler_specific.h"
15 #include "base/memory/scoped_ptr.h"
16 #include "gpu/tools/compositor_model_bench/render_tree.h"
18 // This is a visitor that runs over the tree structure that was built from the
19 // configuration file. It creates OpenGL textures (random checkerboards) that
20 // match the specifications of the original textures and overwrites the old
21 // texture ID's in the tree, replacing them with the matching new textures.
22 class TextureGenerator : public RenderNodeVisitor {
23 public:
24 typedef scoped_ptr<uint8[]> ImagePtr;
25 typedef std::vector<Tile>::iterator tile_iter;
27 explicit TextureGenerator(RenderNode* root);
28 ~TextureGenerator() override;
30 // RenderNodeVisitor functions look for textures and pass them
31 // off to HandleTexture (which behaves appropriately depending
32 // on which pass we are in.)
33 void BeginVisitRenderNode(RenderNode* node) override;
34 void BeginVisitCCNode(CCNode* node) override;
36 private:
37 enum TextureGenStage {
38 DiscoveryStage,
39 RemappingStage,
40 ImageGenerationStage
43 void DiscoverInputIDs(RenderNode* root);
44 void GenerateGLTexIDs();
45 void AssignIDMapping();
46 void WriteOutNewIDs(RenderNode* root);
47 void AllocateImageArray();
48 void BuildTextureImages(RenderNode* root);
49 void HandleTexture(int* texID, int width, int height, GLenum format);
50 void GenerateImageForTexture(int texID, int width, int height, GLenum format);
52 TextureGenStage stage_;
53 std::set<int> discovered_ids_;
54 scoped_ptr<GLuint[]> tex_ids_;
55 std::map<int, int> remapped_ids_;
56 scoped_ptr<ImagePtr[]> image_data_;
57 int images_generated_;
58 std::set<int> ids_for_completed_textures_;
61 #endif // GPU_TOOLS_COMPOSITOR_MODEL_BENCH_RENDER_MODEL_UTILS_H_