/*
 * Copyright (c) 2022 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.
 */

// gtest
#include <gtest/gtest.h>
#include "window_test_utils.h"
#include "window_accessibility_controller.h"
#include "window_impl.h"
#include "wm_common.h"
using namespace testing;
using namespace testing::ext;

namespace OHOS {
namespace Rosen {
using Utils = WindowTestUtils;
class WindowDisplayZoomTest : public testing::Test {
public:
    static void SetUpTestCase();
    static void TearDownTestCase();
    virtual void SetUp() override;
    virtual void TearDown() override;
    Utils::TestWindowInfo windowInfo_;
};

void WindowDisplayZoomTest::SetUpTestCase()
{
}

void WindowDisplayZoomTest::TearDownTestCase()
{
}

void WindowDisplayZoomTest::SetUp()
{
    windowInfo_ = {
        .name = "zoomWindow",
        .rect = {0, 0, 300, 100},
        .type = WindowType::WINDOW_TYPE_APP_MAIN_WINDOW,
        .mode = WindowMode::WINDOW_MODE_FLOATING,
        .needAvoid = false,
        .parentLimit = false,
        .showWhenLocked = true,
        .parentId = INVALID_WINDOW_ID,
    };
}

void WindowDisplayZoomTest::TearDown()
{
}

namespace {
/**
 * @tc.name: DisplayZoom01
 * @tc.desc: test interface SetAnchorAndScale
 * @tc.type: FUNC
 * @tc.require: issueI5NGWL
 */
HWTEST_F(WindowDisplayZoomTest, DisplayZoom01, Function | MediumTest | Level3)
{
    WindowAccessibilityController::GetInstance().OffWindowZoom();
    sleep(1);

    windowInfo_.name = "DisplayZoom01";
    sptr<Window> window = Utils::CreateTestWindow(windowInfo_);
    if (window == nullptr) {
        return;
    }
    Window* ptr = window.GetRefPtr();
    WindowImpl* implPtr = (WindowImpl*)ptr;
    ASSERT_EQ(WMError::WM_OK, window->Show());
    sleep(1);
    Transform expect;
    ASSERT_TRUE(expect == implPtr->GetWindowProperty()->GetZoomTransform());

    WindowAccessibilityController::GetInstance().SetAnchorAndScale(0, 0, 2);
    sleep(1);
    Rect rect = window->GetRect();
    expect.pivotX_ = (0 - rect.posX_) * 1.0 / rect.width_;
    expect.pivotY_ = (0 - rect.posY_) * 1.0 / rect.height_;
    expect.scaleX_ = expect.scaleY_ = 2; // scale value
    ASSERT_TRUE(expect == implPtr->GetWindowProperty()->GetZoomTransform());

    WindowAccessibilityController::GetInstance().SetAnchorAndScale(0, 0, 2);
    sleep(1);
    expect.scaleX_ = expect.scaleY_ = 4; // scale value
    ASSERT_TRUE(expect == implPtr->GetWindowProperty()->GetZoomTransform());

    WindowAccessibilityController::GetInstance().SetAnchorAndScale(0, 0, 0.5);
    sleep(1);
    expect.scaleX_ = expect.scaleY_ = 2; // scale value
    ASSERT_TRUE(expect == implPtr->GetWindowProperty()->GetZoomTransform());

    WindowAccessibilityController::GetInstance().SetAnchorAndScale(0, 0, 0.1);
    sleep(1);
    ASSERT_TRUE(expect == implPtr->GetWindowProperty()->GetZoomTransform());

    WindowAccessibilityController::GetInstance().SetAnchorAndScale(0, 0, -0.1);
    sleep(1);
    ASSERT_TRUE(expect == implPtr->GetWindowProperty()->GetZoomTransform());

    WindowAccessibilityController::GetInstance().OffWindowZoom();
    window->Destroy();
}

/**
 * @tc.name: DisplayZoom02
 * @tc.desc: test interface SetAnchorOffset
 * @tc.type: FUNC
 * @tc.require: issueI5NGWL
 */
HWTEST_F(WindowDisplayZoomTest, DisplayZoom02, Function | MediumTest | Level3)
{
    windowInfo_.name = "DisplayZoom02";
    sptr<Window> window = Utils::CreateTestWindow(windowInfo_);
    if (window == nullptr) {
        return;
    }
    ASSERT_EQ(WMError::WM_OK, window->Show());
    sleep(1);

    WindowAccessibilityController::GetInstance().SetAnchorAndScale(0, 0, 2);
    sleep(1);
    WindowAccessibilityController::GetInstance().SetAnchorOffset(-100, -100);
    sleep(1);

    Transform expect;
    Rect rect = window->GetRect();
    expect.pivotX_ = (0 - rect.posX_) * 1.0 / rect.width_;
    expect.pivotY_ = (0 - rect.posY_) * 1.0 / rect.height_;
    expect.scaleX_ = expect.scaleY_ = 2; // scale value
    expect.translateX_ = expect.translateY_ = -100;  // translate value

    WindowAccessibilityController::GetInstance().SetAnchorOffset(200, 200);
    sleep(1);
    expect.translateX_ = expect.translateY_ = 0;
    WindowAccessibilityController::GetInstance().OffWindowZoom();
    window->Destroy();
}

/**
 * @tc.name: DisplayZoom03
 * @tc.desc: test interface OffWindowZoom
 * @tc.type: FUNC
 * @tc.require: issueI5NGWL
 */
HWTEST_F(WindowDisplayZoomTest, DisplayZoom03, Function | MediumTest | Level3)
{
    windowInfo_.name = "DisplayZoom03";
    sptr<Window> window = Utils::CreateTestWindow(windowInfo_);
    if (window == nullptr) {
        return;
    }
    Window* ptr = window.GetRefPtr();
    WindowImpl* implPtr = (WindowImpl*)ptr;
    ASSERT_EQ(WMError::WM_OK, window->Show());
    sleep(1);

    WindowAccessibilityController::GetInstance().SetAnchorAndScale(0, 0, 2);
    sleep(1);
    WindowAccessibilityController::GetInstance().OffWindowZoom();
    sleep(1);

    Transform expect;
    ASSERT_TRUE(expect == implPtr->GetWindowProperty()->GetZoomTransform());

    window->Destroy();
}

/**
 * @tc.name: DisplayZoom04
 * @tc.desc: test add and remove a window after zoom display
 * @tc.type: FUNC
 * @tc.require: issueI5NGWL
 */
HWTEST_F(WindowDisplayZoomTest, DisplayZoom04, Function | MediumTest | Level3)
{
    WindowAccessibilityController::GetInstance().SetAnchorAndScale(0, 0, 2);
    sleep(1);

    windowInfo_.name = "DisplayZoom04";
    sptr<Window> window = Utils::CreateTestWindow(windowInfo_);
    if (window == nullptr) {
        return;
    }
    Window* ptr = window.GetRefPtr();
    WindowImpl* implPtr = (WindowImpl*)ptr;
    ASSERT_EQ(WMError::WM_OK, window->Show());
    sleep(1);

    Transform expect;
    Rect rect = window->GetRect();
    expect.pivotX_ = (rect.width_ == 0) ? 0 : (0 - rect.posX_) * 1.0 / rect.width_;
    expect.pivotY_ = (rect.height_ == 0) ? 0 : (0 - rect.posY_) * 1.0 / rect.height_;
    expect.scaleX_ = expect.scaleY_ = 2; // scale value
    ASSERT_TRUE(expect == implPtr->GetWindowProperty()->GetZoomTransform());

    ASSERT_EQ(WMError::WM_OK, window->Hide());;
    WindowAccessibilityController::GetInstance().OffWindowZoom();
    sleep(1);

    ASSERT_EQ(WMError::WM_OK, window->Show());
    sleep(1);
    Transform identity;
    ASSERT_TRUE(identity == implPtr->GetWindowProperty()->GetZoomTransform());

    window->Destroy();
}

/**
 * @tc.name: DisplayZoom05
 * @tc.desc: test animate and zoom transform
 * @tc.type: FUNC
 * @tc.require: issueI5NGWL
 */
HWTEST_F(WindowDisplayZoomTest, DisplayZoom05, Function | MediumTest | Level3)
{
    WindowAccessibilityController::GetInstance().SetAnchorAndScale(0, 0, 2);
    sleep(1);

    windowInfo_.name = "DisplayZoom05";
    sptr<Window> window = Utils::CreateTestWindow(windowInfo_);
    if (window == nullptr) {
        return;
    }
    Window* ptr = window.GetRefPtr();
    WindowImpl* implPtr = (WindowImpl*)ptr;
    ASSERT_EQ(WMError::WM_OK, window->Show());
    sleep(1);

    Transform animate;
    animate.translateX_ = -100; // translate x value
    animate.translateZ_ = 100; // translate z value
    window->SetTransform(animate);
    sleep(1);

    Transform expect;
    Rect rect = window->GetRect();
    expect.pivotX_ = (rect.width_ == 0) ? 0 : (0 - rect.posX_) * 1.0 / rect.width_;
    expect.pivotY_ = (rect.height_ == 0) ? 0 : (0 - rect.posY_) * 1.0 / rect.height_;
    expect.scaleX_ = expect.scaleY_ = 1.7; // scale value
    Transform actual = implPtr->GetWindowProperty()->GetZoomTransform();

    auto isExpec = [](float a, float b) -> bool {
        return abs(a - b) < 0.1;
    };
    ASSERT_EQ(true, isExpec(actual.pivotX_, expect.pivotX_));
    ASSERT_EQ(true, isExpec(actual.pivotY_, expect.pivotY_));
    ASSERT_EQ(true, isExpec(actual.scaleX_, expect.scaleX_));
    ASSERT_EQ(true, isExpec(actual.scaleY_, expect.scaleY_));

    WindowAccessibilityController::GetInstance().OffWindowZoom();
    window->Destroy();
}

/**
 * @tc.name: DisplayZoom06
 * @tc.desc: test speical window type
 * @tc.type: FUNC
 * @tc.require: issueI5NGWL
 */
HWTEST_F(WindowDisplayZoomTest, DisplayZoom06, Function | MediumTest | Level3)
{
    WindowAccessibilityController::GetInstance().SetAnchorAndScale(0, 0, 2);
    sleep(1);
    windowInfo_.name = "DisplayZoom06";
    windowInfo_.type = WindowType::WINDOW_TYPE_INPUT_METHOD_FLOAT;
    sptr<Window> window = Utils::CreateTestWindow(windowInfo_);
    ASSERT_NE(nullptr, window);
    sleep(1);
    WindowAccessibilityController::GetInstance().OffWindowZoom();
    window->Destroy();
}
}
} // namespace Rosen
} // namespace OHOS