/*
 * Copyright (C) 2021 The Android Open Source Project
 *
 * 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.
 */

#include <gtest/gtest.h>
#include <gmock/gmock.h>

#include <JankTracker.h>
#include <utils/TimeUtils.h>

using namespace android;
using namespace android::uirenderer;

class TestFrameMetricsObserver : public FrameMetricsObserver {
public:
    void notify(const int64_t*) {}
};

TEST(JankTracker, noJank) {
    std::mutex mutex;
    ProfileDataContainer container(mutex);
    JankTracker jankTracker(&container);
    std::unique_ptr<FrameMetricsReporter> reporter = std::make_unique<FrameMetricsReporter>();

    uint64_t frameNumber = 0;
    uint32_t surfaceId = 0;

    FrameInfo* info = jankTracker.startFrame();
    info->set(FrameInfoIndex::IntendedVsync) = 100_ms;
    info->set(FrameInfoIndex::Vsync) = 101_ms;
    info->set(FrameInfoIndex::SwapBuffersCompleted) = 115_ms;
    info->set(FrameInfoIndex::GpuCompleted) = 115_ms;
    info->set(FrameInfoIndex::FrameCompleted) = 115_ms;
    info->set(FrameInfoIndex::FrameInterval) = 16_ms;
    info->set(FrameInfoIndex::FrameDeadline) = 120_ms;
    jankTracker.finishFrame(*info, reporter, frameNumber, surfaceId);

    info = jankTracker.startFrame();
    info->set(FrameInfoIndex::IntendedVsync) = 116_ms;
    info->set(FrameInfoIndex::Vsync) = 117_ms;
    info->set(FrameInfoIndex::SwapBuffersCompleted) = 129_ms;
    info->set(FrameInfoIndex::GpuCompleted) = 131_ms;
    info->set(FrameInfoIndex::FrameCompleted) = 131_ms;
    info->set(FrameInfoIndex::FrameInterval) = 16_ms;
    info->set(FrameInfoIndex::FrameDeadline) = 136_ms;
    jankTracker.finishFrame(*info, reporter, frameNumber, surfaceId);

    ASSERT_EQ(2, container.get()->totalFrameCount());
    ASSERT_EQ(0, container.get()->jankFrameCount());
}


TEST(JankTracker, jank) {
    std::mutex mutex;
    ProfileDataContainer container(mutex);
    JankTracker jankTracker(&container);
    std::unique_ptr<FrameMetricsReporter> reporter = std::make_unique<FrameMetricsReporter>();

    uint64_t frameNumber = 0;
    uint32_t surfaceId = 0;

    FrameInfo* info = jankTracker.startFrame();
    info->set(FrameInfoIndex::IntendedVsync) = 100_ms;
    info->set(FrameInfoIndex::Vsync) = 101_ms;
    info->set(FrameInfoIndex::SwapBuffersCompleted) = 115_ms;
    info->set(FrameInfoIndex::GpuCompleted) = 121_ms;
    info->set(FrameInfoIndex::FrameCompleted) = 121_ms;
    info->set(FrameInfoIndex::FrameInterval) = 16_ms;
    info->set(FrameInfoIndex::FrameDeadline) = 120_ms;
    jankTracker.finishFrame(*info, reporter, frameNumber, surfaceId);

    ASSERT_EQ(1, container.get()->totalFrameCount());
    ASSERT_EQ(1, container.get()->jankFrameCount());
}

TEST(JankTracker, legacyJankButNoRealJank) {
    std::mutex mutex;
    ProfileDataContainer container(mutex);
    JankTracker jankTracker(&container);
    std::unique_ptr<FrameMetricsReporter> reporter = std::make_unique<FrameMetricsReporter>();

    uint64_t frameNumber = 0;
    uint32_t surfaceId = 0;

    FrameInfo* info = jankTracker.startFrame();
    info->set(FrameInfoIndex::IntendedVsync) = 100_ms;
    info->set(FrameInfoIndex::Vsync) = 101_ms;
    info->set(FrameInfoIndex::SwapBuffersCompleted) = 117_ms;
    info->set(FrameInfoIndex::GpuCompleted) = 118_ms;
    info->set(FrameInfoIndex::FrameCompleted) = 118_ms;
    info->set(FrameInfoIndex::FrameInterval) = 16_ms;
    info->set(FrameInfoIndex::FrameDeadline) = 120_ms;
    jankTracker.finishFrame(*info, reporter, frameNumber, surfaceId);

    ASSERT_EQ(1, container.get()->totalFrameCount());
    ASSERT_EQ(0, container.get()->jankFrameCount());
    ASSERT_EQ(1, container.get()->jankLegacyFrameCount());
}

TEST(JankTracker, doubleStuffed) {
    std::mutex mutex;
    ProfileDataContainer container(mutex);
    JankTracker jankTracker(&container);
    std::unique_ptr<FrameMetricsReporter> reporter = std::make_unique<FrameMetricsReporter>();

    uint64_t frameNumber = 0;
    uint32_t surfaceId = 0;

    // First frame janks
    FrameInfo* info = jankTracker.startFrame();
    info->set(FrameInfoIndex::IntendedVsync) = 100_ms;
    info->set(FrameInfoIndex::Vsync) = 101_ms;
    info->set(FrameInfoIndex::SwapBuffersCompleted) = 115_ms;
    info->set(FrameInfoIndex::GpuCompleted) = 121_ms;
    info->set(FrameInfoIndex::FrameCompleted) = 121_ms;
    info->set(FrameInfoIndex::FrameInterval) = 16_ms;
    info->set(FrameInfoIndex::FrameDeadline) = 120_ms;
    jankTracker.finishFrame(*info, reporter, frameNumber, surfaceId);

    ASSERT_EQ(1, container.get()->jankFrameCount());

    // Second frame is long, but doesn't jank because double-stuffed.
    info = jankTracker.startFrame();
    info->set(FrameInfoIndex::IntendedVsync) = 116_ms;
    info->set(FrameInfoIndex::Vsync) = 122_ms;
    info->set(FrameInfoIndex::SwapBuffersCompleted) = 129_ms;
    info->set(FrameInfoIndex::GpuCompleted) = 137_ms;
    info->set(FrameInfoIndex::FrameCompleted) = 137_ms;
    info->set(FrameInfoIndex::FrameInterval) = 16_ms;
    info->set(FrameInfoIndex::FrameDeadline) = 136_ms;
    jankTracker.finishFrame(*info, reporter, frameNumber, surfaceId);

    ASSERT_EQ(2, container.get()->totalFrameCount());
    ASSERT_EQ(1, container.get()->jankFrameCount());
}

TEST(JankTracker, doubleStuffedThenPauseThenJank) {
    std::mutex mutex;
    ProfileDataContainer container(mutex);
    JankTracker jankTracker(&container);
    std::unique_ptr<FrameMetricsReporter> reporter = std::make_unique<FrameMetricsReporter>();

    uint64_t frameNumber = 0;
    uint32_t surfaceId = 0;

    // First frame janks
    FrameInfo* info = jankTracker.startFrame();
    info->set(FrameInfoIndex::IntendedVsync) = 100_ms;
    info->set(FrameInfoIndex::Vsync) = 101_ms;
    info->set(FrameInfoIndex::SwapBuffersCompleted) = 115_ms;
    info->set(FrameInfoIndex::GpuCompleted) = 121_ms;
    info->set(FrameInfoIndex::FrameCompleted) = 121_ms;
    info->set(FrameInfoIndex::FrameInterval) = 16_ms;
    info->set(FrameInfoIndex::FrameDeadline) = 120_ms;
    jankTracker.finishFrame(*info, reporter, frameNumber, surfaceId);

    ASSERT_EQ(1, container.get()->jankFrameCount());

    // Second frame is long, but doesn't jank because double-stuffed.
    info = jankTracker.startFrame();
    info->set(FrameInfoIndex::IntendedVsync) = 116_ms;
    info->set(FrameInfoIndex::Vsync) = 122_ms;
    info->set(FrameInfoIndex::SwapBuffersCompleted) = 129_ms;
    info->set(FrameInfoIndex::GpuCompleted) = 137_ms;
    info->set(FrameInfoIndex::FrameCompleted) = 137_ms;
    info->set(FrameInfoIndex::FrameInterval) = 16_ms;
    info->set(FrameInfoIndex::FrameDeadline) = 136_ms;
    jankTracker.finishFrame(*info, reporter, frameNumber, surfaceId);

    ASSERT_EQ(1, container.get()->jankFrameCount());

    // Thirdframe is long and skips one frame some double stuffed logic gets reset
    info = jankTracker.startFrame();
    info->set(FrameInfoIndex::IntendedVsync) = 148_ms;
    info->set(FrameInfoIndex::Vsync) = 148_ms;
    info->set(FrameInfoIndex::SwapBuffersCompleted) = 160_ms;
    info->set(FrameInfoIndex::GpuCompleted) = 169_ms;
    info->set(FrameInfoIndex::FrameCompleted) = 169_ms;
    info->set(FrameInfoIndex::FrameInterval) = 16_ms;
    info->set(FrameInfoIndex::FrameDeadline) = 168_ms;
    jankTracker.finishFrame(*info, reporter, frameNumber, surfaceId);

    ASSERT_EQ(3, container.get()->totalFrameCount());
    ASSERT_EQ(2, container.get()->jankFrameCount());
}

TEST(JankTracker, doubleStuffedTwoIntervalBehind) {
    std::mutex mutex;
    ProfileDataContainer container(mutex);
    JankTracker jankTracker(&container);
    std::unique_ptr<FrameMetricsReporter> reporter = std::make_unique<FrameMetricsReporter>();

    uint64_t frameNumber = 0;
    uint32_t surfaceId = 0;

    // First frame janks
    FrameInfo* info = jankTracker.startFrame();
    info->set(FrameInfoIndex::IntendedVsync) = 100_ms;
    info->set(FrameInfoIndex::Vsync) = 101_ms;
    info->set(FrameInfoIndex::SwapBuffersCompleted) = 107_ms;
    info->set(FrameInfoIndex::GpuCompleted) = 117_ms;
    info->set(FrameInfoIndex::FrameCompleted) = 117_ms;
    info->set(FrameInfoIndex::FrameInterval) = 16_ms;
    info->set(FrameInfoIndex::FrameDeadline) = 116_ms;
    jankTracker.finishFrame(*info, reporter, frameNumber, surfaceId);

    ASSERT_EQ(1, container.get()->jankFrameCount());

    // Second frame is long, but doesn't jank because double-stuffed.
    // Second frame duration is between 1*interval ~ 2*interval
    info = jankTracker.startFrame();
    info->set(FrameInfoIndex::IntendedVsync) = 116_ms;
    info->set(FrameInfoIndex::Vsync) = 116_ms;
    info->set(FrameInfoIndex::SwapBuffersCompleted) = 129_ms;
    info->set(FrameInfoIndex::GpuCompleted) = 133_ms;
    info->set(FrameInfoIndex::FrameCompleted) = 133_ms;
    info->set(FrameInfoIndex::FrameInterval) = 16_ms;
    info->set(FrameInfoIndex::FrameDeadline) = 132_ms;
    jankTracker.finishFrame(*info, reporter, frameNumber, surfaceId);

    ASSERT_EQ(1, container.get()->jankFrameCount());

    // Third frame is even longer, cause a jank
    // Third frame duration is between 2*interval ~ 3*interval
    info = jankTracker.startFrame();
    info->set(FrameInfoIndex::IntendedVsync) = 132_ms;
    info->set(FrameInfoIndex::Vsync) = 132_ms;
    info->set(FrameInfoIndex::SwapBuffersCompleted) = 160_ms;
    info->set(FrameInfoIndex::GpuCompleted) = 165_ms;
    info->set(FrameInfoIndex::FrameCompleted) = 165_ms;
    info->set(FrameInfoIndex::FrameInterval) = 16_ms;
    info->set(FrameInfoIndex::FrameDeadline) = 148_ms;
    jankTracker.finishFrame(*info, reporter, frameNumber, surfaceId);

    ASSERT_EQ(2, container.get()->jankFrameCount());

    // 4th frame is double-stuffed with a 2 * interval latency
    // 4th frame duration is between 2*interval ~ 3*interval
    info = jankTracker.startFrame();
    info->set(FrameInfoIndex::IntendedVsync) = 148_ms;
    info->set(FrameInfoIndex::Vsync) = 148_ms;
    info->set(FrameInfoIndex::SwapBuffersCompleted) = 170_ms;
    info->set(FrameInfoIndex::GpuCompleted) = 181_ms;
    info->set(FrameInfoIndex::FrameCompleted) = 181_ms;
    info->set(FrameInfoIndex::FrameInterval) = 16_ms;
    info->set(FrameInfoIndex::FrameDeadline) = 164_ms;
    jankTracker.finishFrame(*info, reporter, frameNumber, surfaceId);

    ASSERT_EQ(2, container.get()->jankFrameCount());
}