1 /*
2  * Copyright 2022 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/DisplayColorProfile.h>
20 #include <compositionengine/RenderSurface.h>
21 #include <compositionengine/impl/Output.h>
22 #include <ui/Rect.h>
23 
24 #include "RenderArea.h"
25 
26 namespace android {
27 
28 struct ScreenCaptureOutputArgs {
29     const compositionengine::CompositionEngine& compositionEngine;
30     const compositionengine::Output::ColorProfile& colorProfile;
31     const RenderArea& renderArea;
32     ui::LayerStack layerStack;
33     std::shared_ptr<renderengine::ExternalTexture> buffer;
34     float sdrWhitePointNits;
35     float displayBrightnessNits;
36     // Counterintuitively, when targetBrightness > 1.0 then dim the scene.
37     float targetBrightness;
38     bool regionSampling;
39     bool treat170mAsSrgb;
40     bool dimInGammaSpaceForEnhancedScreenshots;
41     bool isProtected = false;
42     bool enableLocalTonemapping = false;
43 };
44 
45 // ScreenCaptureOutput is used to compose a set of layers into a preallocated buffer.
46 //
47 // SurfaceFlinger passes instances of ScreenCaptureOutput to CompositionEngine in calls to
48 // SurfaceFlinger::captureLayers and SurfaceFlinger::captureDisplay.
49 class ScreenCaptureOutput : public compositionengine::impl::Output {
50 public:
51     ScreenCaptureOutput(const RenderArea& renderArea,
52                         const compositionengine::Output::ColorProfile& colorProfile,
53                         bool regionSampling, bool dimInGammaSpaceForEnhancedScreenshots,
54                         bool enableLocalTonemapping);
55 
56     void updateColorProfile(const compositionengine::CompositionRefreshArgs&) override;
57 
58     std::vector<compositionengine::LayerFE::LayerSettings> generateClientCompositionRequests(
59             bool supportsProtectedContent, ui::Dataspace outputDataspace,
60             std::vector<compositionengine::LayerFE*>& outLayerFEs) override;
61 
62 protected:
getSkipColorTransform()63     bool getSkipColorTransform() const override { return false; }
64     renderengine::DisplaySettings generateClientCompositionDisplaySettings(
65             const std::shared_ptr<renderengine::ExternalTexture>& buffer) const override;
66 
67 private:
68     const RenderArea& mRenderArea;
69     const compositionengine::Output::ColorProfile& mColorProfile;
70     const bool mRegionSampling;
71     const bool mDimInGammaSpaceForEnhancedScreenshots;
72     const bool mEnableLocalTonemapping;
73 };
74 
75 std::shared_ptr<ScreenCaptureOutput> createScreenCaptureOutput(ScreenCaptureOutputArgs);
76 
77 } // namespace android
78