1/*
2 * Copyright (c) 2021-2022 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 *     http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16#ifndef FOUNDATION_DMSERVER_DISPLAY_MANAGER_SERVICE_H
17#define FOUNDATION_DMSERVER_DISPLAY_MANAGER_SERVICE_H
18
19#include <map>
20#include <mutex>
21
22#include <system_ability.h>
23#include <surface.h>
24#include <ui/rs_display_node.h>
25
26#include "atomic_map.h"
27#include "dm_common.h"
28#include "display_dumper.h"
29#include "screen.h"
30#include "abstract_display.h"
31#include "abstract_display_controller.h"
32#include "abstract_screen_controller.h"
33#include "display_change_listener.h"
34#include "display_cutout_controller.h"
35#include "display_manager_stub.h"
36#include "display_power_controller.h"
37#include "singleton_delegator.h"
38#include "window_info_queried_listener.h"
39
40namespace OHOS::Rosen {
41class DisplayManagerService : public SystemAbility, public DisplayManagerStub {
42DECLARE_SYSTEM_ABILITY(DisplayManagerService);
43WM_DECLARE_SINGLE_INSTANCE_BASE(DisplayManagerService);
44
45public:
46    int Dump(int fd, const std::vector<std::u16string>& args) override;
47    void OnStart() override;
48    void OnStop() override;
49    ScreenId CreateVirtualScreen(VirtualScreenOption option,
50        const sptr<IRemoteObject>& displayManagerAgent) override;
51    DMError DestroyVirtualScreen(ScreenId screenId) override;
52    DMError SetVirtualScreenSurface(ScreenId screenId, sptr<IBufferProducer> surface) override;
53    DMError IsScreenRotationLocked(bool& isLocked) override;
54    DMError SetScreenRotationLocked(bool isLocked) override;
55    DMError SetScreenRotationLockedFromJs(bool isLocked) override;
56
57    sptr<DisplayInfo> GetDefaultDisplayInfo() override;
58    sptr<DisplayInfo> GetDisplayInfoById(DisplayId displayId) override;
59    sptr<DisplayInfo> GetDisplayInfoByScreen(ScreenId screenId) override;
60    sptr<CutoutInfo> GetCutoutInfo(DisplayId displayId) override;
61    DMError SetOrientation(ScreenId screenId, Orientation orientation) override;
62    DMError SetOrientationFromWindow(ScreenId screenId, Orientation orientation, bool withAnimation);
63    bool SetRotationFromWindow(ScreenId screenId, Rotation targetRotation, bool withAnimation);
64    void SetGravitySensorSubscriptionEnabled();
65    std::shared_ptr<Media::PixelMap> GetDisplaySnapshot(DisplayId displayId, DmErrorCode* errorCode) override;
66    DMError HasPrivateWindow(DisplayId id, bool& hasPrivateWindow) override;
67    // colorspace, gamut
68    DMError GetScreenSupportedColorGamuts(ScreenId screenId, std::vector<ScreenColorGamut>& colorGamuts) override;
69    DMError GetScreenColorGamut(ScreenId screenId, ScreenColorGamut& colorGamut) override;
70    DMError SetScreenColorGamut(ScreenId screenId, int32_t colorGamutIdx) override;
71    DMError GetScreenGamutMap(ScreenId screenId, ScreenGamutMap& gamutMap) override;
72    DMError SetScreenGamutMap(ScreenId screenId, ScreenGamutMap gamutMap) override;
73    DMError SetScreenColorTransform(ScreenId screenId) override;
74
75    DMError RegisterDisplayManagerAgent(const sptr<IDisplayManagerAgent>& displayManagerAgent,
76        DisplayManagerAgentType type) override;
77    DMError UnregisterDisplayManagerAgent(const sptr<IDisplayManagerAgent>& displayManagerAgent,
78        DisplayManagerAgentType type) override;
79    bool WakeUpBegin(PowerStateChangeReason reason) override;
80    bool WakeUpEnd() override;
81    bool SuspendBegin(PowerStateChangeReason reason) override;
82    bool SuspendEnd() override;
83    bool SetSpecifiedScreenPower(ScreenId, ScreenPowerState, PowerStateChangeReason) override;
84    bool SetScreenPowerForAll(ScreenPowerState state, PowerStateChangeReason reason) override;
85    ScreenPowerState GetScreenPower(ScreenId dmsScreenId) override;
86    bool SetDisplayState(DisplayState state) override;
87    void UpdateRSTree(DisplayId displayId, DisplayId parentDisplayId, std::shared_ptr<RSSurfaceNode>& surfaceNode,
88        bool isAdd, bool isMultiDisplay);
89    DMError AddSurfaceNodeToDisplay(DisplayId displayId,
90        std::shared_ptr<RSSurfaceNode>& surfaceNode, bool onTop = true) override;
91    DMError RemoveSurfaceNodeFromDisplay(DisplayId displayId, std::shared_ptr<RSSurfaceNode>& surfaceNode) override;
92    DisplayState GetDisplayState(DisplayId displayId) override;
93    bool TryToCancelScreenOff() override;
94    bool SetScreenBrightness(uint64_t screenId, uint32_t level) override;
95    uint32_t GetScreenBrightness(uint64_t screenId) override;
96    void NotifyDisplayEvent(DisplayEvent event) override;
97    bool SetFreeze(std::vector<DisplayId> displayIds, bool isFreeze) override;
98
99    DMError MakeMirror(ScreenId mainScreenId, std::vector<ScreenId> mirrorScreenIds, ScreenId& screenGroupId) override;
100    DMError MakeExpand(std::vector<ScreenId> screenId, std::vector<Point> startPoints,
101        ScreenId& screenGroupId) override;
102    DMError StopMirror(const std::vector<ScreenId>& mirrorScreenIds) override;
103    DMError StopExpand(const std::vector<ScreenId>& expandScreenIds) override;
104    void RemoveVirtualScreenFromGroup(std::vector<ScreenId> screens) override;
105    sptr<ScreenInfo> GetScreenInfoById(ScreenId screenId) override;
106    sptr<ScreenGroupInfo> GetScreenGroupInfoById(ScreenId screenId) override;
107    ScreenId GetScreenGroupIdByScreenId(ScreenId screenId);
108    DMError GetAllScreenInfos(std::vector<sptr<ScreenInfo>>& screenInfos) override;
109
110    std::vector<DisplayId> GetAllDisplayIds() override;
111    DMError SetScreenActiveMode(ScreenId screenId, uint32_t modeId) override;
112    DMError SetVirtualPixelRatio(ScreenId screenId, float virtualPixelRatio) override;
113    DMError SetResolution(ScreenId screenId, uint32_t width, uint32_t height,
114        float virtualPixelRatio) override { return DMError::DM_OK; }
115    DMError GetDensityInCurResolution(ScreenId screenId,
116        float& virtualPixelRatio) override { return DMError::DM_OK; }
117    void RegisterDisplayChangeListener(sptr<IDisplayChangeListener> listener);
118    void RegisterWindowInfoQueriedListener(const sptr<IWindowInfoQueriedListener>& listener);
119    void NotifyPrivateWindowStateChanged(bool hasPrivate);
120    std::vector<DisplayPhysicalResolution> GetAllDisplayPhysicalResolution() override;
121
122private:
123    DisplayManagerService();
124    ~DisplayManagerService() = default;
125    bool Init();
126    void NotifyDisplayStateChange(DisplayId defaultDisplayId, sptr<DisplayInfo> displayInfo,
127        const std::map<DisplayId, sptr<DisplayInfo>>& displayInfoMap, DisplayStateChangeType type);
128    void NotifyScreenshot(DisplayId displayId);
129    ScreenId GetScreenIdByDisplayId(DisplayId displayId) const;
130    void ConfigureDisplayManagerService();
131    void ConfigureWaterfallDisplayCompressionParams();
132
133    std::recursive_mutex mutex_;
134    static inline SingletonDelegator<DisplayManagerService> delegator_;
135    sptr<AbstractDisplayController> abstractDisplayController_;
136    sptr<AbstractScreenController> abstractScreenController_;
137    sptr<DisplayPowerController> displayPowerController_;
138    sptr<DisplayCutoutController> displayCutoutController_;
139    sptr<IDisplayChangeListener> displayChangeListener_;
140    sptr<IWindowInfoQueriedListener> windowInfoQueriedListener_;
141    sptr<DisplayDumper> displayDumper_;
142    AtomicMap<ScreenId, uint32_t> accessTokenIdMaps_;
143    bool isAutoRotationOpen_;
144    std::vector<DisplayPhysicalResolution> allDisplayPhysicalResolution_ {};
145};
146} // namespace OHOS::Rosen
147
148#endif // FOUNDATION_DMSERVER_DISPLAY_MANAGER_SERVICE_H