/*
 * Copyright (c) 2021-2023 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef FOUNDATION_ACE_FRAMEWORKS_CORE_COMMON_WINDOW_H
#define FOUNDATION_ACE_FRAMEWORKS_CORE_COMMON_WINDOW_H

#include <memory>

#include "base/geometry/ng/rect_t.h"
#include "base/mousestyle/mouse_style.h"
#include "base/thread/task_executor.h"
#include "base/utils/macros.h"
#include "base/utils/noncopyable.h"
#include "core/common/ace_page.h"
#include "core/common/platform_window.h"

namespace OHOS::Rosen {
class RSUIDirector;
}

namespace OHOS::Ace {

namespace NG {
class FrameNode;
} // namespace NG

enum class WidthBreakpoint {WIDTH_XS, WIDTH_SM, WIDTH_MD, WIDTH_LG, WIDTH_XL, WIDTH_XXL, UNDEFINED};
enum class HeightBreakpoint {HEIGHT_SM, HEIGHT_MD, HEIGHT_LG};

class ACE_EXPORT Window : public std::enable_shared_from_this<Window> {
public:
    Window() = default;
    explicit Window(std::unique_ptr<PlatformWindow> platformWindow);
    virtual ~Window() = default;

    uint32_t GetWindowId() const
    {
        return windowId_;
    }

    virtual void RequestFrame();
    virtual void ForceFlushVsync(uint64_t nanoTimestamp, uint64_t frameCount) {}

    virtual void FlushFrameRate(int32_t rate, int32_t animatorExpectedFrameRate, int32_t rateTyte) {}

    virtual void SetTaskExecutor(const RefPtr<TaskExecutor>& taskExecutor) {}

    virtual void SetInstanceId(int32_t instanceId) {}

    virtual void Init() {}

    virtual void InitArkUI_X() {}

    virtual void Destroy()
    {
        platformWindow_->Destroy();
        platformWindow_.reset();
    }

    virtual void SetRootRenderNode(const RefPtr<RenderNode>& root);

    virtual void SetRootFrameNode(const RefPtr<NG::FrameNode>& root) {}

    virtual void RecordFrameTime(uint64_t timeStamp, const std::string& name) {}

    virtual void FlushTasks(std::function<void()> callback = nullptr) {}

    virtual std::shared_ptr<Rosen::RSUIDirector> GetRSUIDirector() const
    {
        return nullptr;
    }

    virtual void FlushLayoutSize(int32_t width, int32_t height) {}

    virtual bool FlushAnimation(uint64_t timeStamp)
    {
        return false;
    }

    virtual bool HasFirstFrameAnimation()
    {
        return false;
    }

    virtual void FlushAnimationStartTime(uint64_t timeStamp) {}

    virtual void FlushModifier() {}

    virtual bool HasUIRunningAnimation()
    {
        return false;
    }

    virtual void OnVsync(uint64_t nanoTimestamp, uint64_t frameCount);

    virtual void SetVsyncCallback(AceVsyncCallback&& callback);

    virtual void OnShow()
    {
        onShow_ = true;
    }

    virtual void OnHide()
    {
        onShow_ = false;
    }

    bool IsHide() const
    {
        return !onShow_;
    }

    bool GetUiDvsyncSwitch() const
    {
        return dvsyncOn_;
    }

    int64_t GetDeadlineByFrameCount(int64_t deadline, int64_t ts, int64_t frameBufferCount);

    void SetDensity(double density)
    {
        density_ = density;
    }

    void SetGetWindowRectImpl(std::function<Rect()>&& callback)
    {
        windowRectImpl_ = std::move(callback);
    }

    void InitGetGlobalWindowRectCallback(std::function<Rect()>&& callback)
    {
        globalDisplayWindowRectImpl_ = std::move(callback);
    }

    virtual Rect GetCurrentWindowRect() const
    {
        Rect rect;
        if (windowRectImpl_) {
            rect = windowRectImpl_();
        }
        return rect;
    }

    virtual Rect GetGlobalDisplayWindowRect() const
    {
        Rect rect;
        if (globalDisplayWindowRectImpl_) {
            rect = globalDisplayWindowRectImpl_();
        }
        return rect;
    }

    virtual void SetDrawTextAsBitmap(bool useBitmap) {}

    virtual float GetRefreshRate() const
    {
        return 0.0f;
    }

    uint64_t GetLastRequestVsyncTime() const
    {
        return lastRequestVsyncTime_;
    }

    int64_t GetLastVsyncEndTimestamp() const
    {
        return lastVsyncEndTimestamp_;
    }

    void SetLastVsyncEndTimestamp(int64_t lastVsyncEndTimestamp)
    {
        lastVsyncEndTimestamp_ = lastVsyncEndTimestamp;
    }

    virtual void SetKeepScreenOn(bool keepScreenOn) {};
    virtual void SetViewKeepScreenOn(bool keepScreenOn) {};

    virtual int64_t GetVSyncPeriod() const
    {
        return 0;
    };

    virtual std::string GetWindowName() const
    {
        static std::string name;
        return name;
    };

    void SetCursor(MouseFormat cursor)
    {
        cursor_ = cursor;
    }

    MouseFormat GetCursor() const
    {
        return cursor_;
    }

    void SetUserSetCursor(bool isUserSetCursor)
    {
        isUserSetCursor_ = isUserSetCursor;
    }

    bool IsUserSetCursor() const
    {
        return isUserSetCursor_;
    }

    virtual int32_t GetCurrentRefreshRateMode() const
    {
        return -1;
    }

    virtual int32_t GetAnimateExpectedRate() const
    {
        return 0;
    }

    virtual void Lock() {}

    virtual void Unlock() {}

    virtual void SetUiDvsyncSwitch(bool dvsyncSwitch);

    virtual uint32_t GetStatusBarHeight() const
    {
        return 0;
    }

    virtual bool GetIsRequestVsync()
    {
        return false;
    }

    virtual void NotifyExtensionTimeout(int32_t errorCode) {}

    virtual void NotifySnapshotUpdate() {}

    virtual bool GetIsRequestFrame()
    {
        return false;
    }

    void SetForceVsyncRequests(bool forceVsyncRequests);

    WidthBreakpoint GetWidthBreakpoint(const WidthLayoutBreakPoint& layoutBreakpoints) const;
    HeightBreakpoint GetHeightBreakpoint(const HeightLayoutBreakPoint& layoutBreakpoints) const;

    virtual void SetDVSyncUpdate(uint64_t dvsyncTime) {}
protected:
    bool isRequestVsync_ = false;
    bool onShow_ = true;
    double density_ = 1.0;
    MouseFormat cursor_ = MouseFormat::DEFAULT;
    bool isUserSetCursor_ = false;

    struct VsyncCallback {
        AceVsyncCallback callback_ = nullptr;
        int32_t containerId_ = -1;
    };
    std::list<struct VsyncCallback> callbacks_;

    uint64_t lastRequestVsyncTime_ = 0;
    int64_t lastVsyncEndTimestamp_ = 0;
    uint32_t windowId_ = 0;
    bool dvsyncOn_ = false;
    int64_t lastDVsyncInbihitPredictTs_ = 0;
    bool forceVsync_ = false;

private:
    std::function<Rect()> windowRectImpl_;
    std::function<Rect()> globalDisplayWindowRectImpl_;
    std::unique_ptr<PlatformWindow> platformWindow_;

    ACE_DISALLOW_COPY_AND_MOVE(Window);
};

} // namespace OHOS::Ace

#endif // FOUNDATION_ACE_FRAMEWORKS_CORE_COMMON_WINDOW_H