1 /*
2  * Copyright 2019 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #pragma once
18 
19 #include <compositionengine/CompositionRefreshArgs.h>
20 #include <compositionengine/DisplayColorProfile.h>
21 #include <compositionengine/LayerFE.h>
22 #include <compositionengine/Output.h>
23 #include <compositionengine/OutputLayer.h>
24 #include <compositionengine/RenderSurface.h>
25 #include <compositionengine/impl/OutputCompositionState.h>
26 #include <gmock/gmock.h>
27 
28 namespace android::compositionengine::mock {
29 
30 class Output : public virtual compositionengine::Output {
31 public:
32     Output();
33     virtual ~Output();
34 
35     MOCK_CONST_METHOD0(isValid, bool());
36     MOCK_CONST_METHOD0(getDisplayId, std::optional<DisplayId>());
37 
38     MOCK_METHOD1(setCompositionEnabled, void(bool));
39     MOCK_METHOD1(setLayerCachingEnabled, void(bool));
40     MOCK_METHOD1(setLayerCachingTexturePoolEnabled, void(bool));
41     MOCK_METHOD3(setProjection, void(ui::Rotation, const Rect&, const Rect&));
42     MOCK_METHOD1(setDisplaySize, void(const ui::Size&));
43     MOCK_METHOD2(setLayerStackFilter, void(uint32_t, bool));
44     MOCK_CONST_METHOD0(getTransformHint, ui::Transform::RotationFlags());
45 
46     MOCK_METHOD1(setColorTransform, void(const compositionengine::CompositionRefreshArgs&));
47     MOCK_METHOD1(setColorProfile, void(const ColorProfile&));
48     MOCK_METHOD2(setDisplayBrightness, void(float, float));
49 
50     MOCK_CONST_METHOD1(dump, void(std::string&));
51     MOCK_CONST_METHOD2(dumpPlannerInfo, void(const Vector<String16>&, std::string&));
52     MOCK_CONST_METHOD0(getName, const std::string&());
53     MOCK_METHOD1(setName, void(const std::string&));
54 
55     MOCK_CONST_METHOD0(getDisplayColorProfile, compositionengine::DisplayColorProfile*());
56     MOCK_METHOD1(setDisplayColorProfile,
57                  void(std::unique_ptr<compositionengine::DisplayColorProfile>));
58 
59     MOCK_CONST_METHOD0(getRenderSurface, compositionengine::RenderSurface*());
60     MOCK_METHOD1(setRenderSurface, void(std::unique_ptr<compositionengine::RenderSurface>));
61 
62     MOCK_CONST_METHOD0(getState, const OutputCompositionState&());
63     MOCK_METHOD0(editState, OutputCompositionState&());
64 
65     MOCK_CONST_METHOD1(getDirtyRegion, Region(bool));
66     MOCK_CONST_METHOD2(belongsInOutput, bool(std::optional<uint32_t>, bool));
67     MOCK_CONST_METHOD1(belongsInOutput, bool(const sp<compositionengine::LayerFE>&));
68 
69     MOCK_CONST_METHOD1(getOutputLayerForLayer,
70                        compositionengine::OutputLayer*(const sp<compositionengine::LayerFE>&));
71     MOCK_METHOD0(clearOutputLayers, void());
72     MOCK_METHOD1(injectOutputLayerForTest,
73                  compositionengine::OutputLayer*(const sp<compositionengine::LayerFE>&));
74     MOCK_CONST_METHOD0(getOutputLayerCount, size_t());
75     MOCK_CONST_METHOD1(getOutputLayerOrderedByZByIndex, OutputLayer*(size_t));
76 
77     MOCK_METHOD1(setReleasedLayers, void(ReleasedLayers&&));
78 
79     MOCK_METHOD2(prepare, void(const compositionengine::CompositionRefreshArgs&, LayerFESet&));
80     MOCK_METHOD1(present, void(const compositionengine::CompositionRefreshArgs&));
81 
82     MOCK_METHOD2(rebuildLayerStacks,
83                  void(const compositionengine::CompositionRefreshArgs&, LayerFESet&));
84     MOCK_METHOD2(collectVisibleLayers,
85                  void(const compositionengine::CompositionRefreshArgs&,
86                       compositionengine::Output::CoverageState&));
87     MOCK_METHOD2(ensureOutputLayerIfVisible,
88                  void(sp<compositionengine::LayerFE>&, compositionengine::Output::CoverageState&));
89     MOCK_METHOD1(setReleasedLayers, void(const compositionengine::CompositionRefreshArgs&));
90 
91     MOCK_CONST_METHOD1(updateLayerStateFromFE, void(const CompositionRefreshArgs&));
92     MOCK_METHOD1(updateCompositionState, void(const CompositionRefreshArgs&));
93     MOCK_METHOD0(planComposition, void());
94     MOCK_METHOD1(writeCompositionState, void(const CompositionRefreshArgs&));
95     MOCK_METHOD1(updateColorProfile, void(const compositionengine::CompositionRefreshArgs&));
96 
97     MOCK_METHOD0(beginFrame, void());
98 
99     MOCK_METHOD0(prepareFrame, void());
100     MOCK_METHOD0(chooseCompositionStrategy, void());
101 
102     MOCK_METHOD1(devOptRepaintFlash, void(const compositionengine::CompositionRefreshArgs&));
103 
104     MOCK_METHOD1(finishFrame, void(const compositionengine::CompositionRefreshArgs&));
105 
106     MOCK_METHOD2(composeSurfaces,
107                  std::optional<base::unique_fd>(
108                          const Region&,
109                          const compositionengine::CompositionRefreshArgs& refreshArgs));
110     MOCK_CONST_METHOD0(getSkipColorTransform, bool());
111 
112     MOCK_METHOD0(postFramebuffer, void());
113     MOCK_METHOD1(renderCachedSets, void(const CompositionRefreshArgs&));
114     MOCK_METHOD0(presentAndGetFrameFences, compositionengine::Output::FrameFences());
115 
116     MOCK_METHOD3(generateClientCompositionRequests,
117                  std::vector<LayerFE::LayerSettings>(bool, Region&, ui::Dataspace));
118     MOCK_METHOD2(appendRegionFlashRequests,
119                  void(const Region&, std::vector<LayerFE::LayerSettings>&));
120     MOCK_METHOD1(setExpensiveRenderingExpected, void(bool));
121     MOCK_METHOD1(cacheClientCompositionRequests, void(uint32_t));
122 };
123 
124 } // namespace android::compositionengine::mock
125