#!/usr/bin/env python3
# Copyright (C) 2024 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.

from python.generators.diff_tests.testing import Path
from python.generators.diff_tests.testing import Csv, TextProto
from python.generators.diff_tests.testing import DiffTestBlueprint
from python.generators.diff_tests.testing import TestSuite


class Frames(TestSuite):

  def test_android_frames_get_frame_table_with_id(self):
    return DiffTestBlueprint(
        trace=Path('../../metrics/graphics/android_doframe_depth.py'),
        query="""
        INCLUDE PERFETTO MODULE android.frames.timeline;
        SELECT frame_id, name, depth
        FROM android_frames_choreographer_do_frame doframe
        JOIN slice USING(id);
        """,
        out=Csv("""
        "frame_id","name","depth"
        10,"Choreographer#doFrame 10",0
        11,"Choreographer#doFrame 11",1
        12,"Choreographer#doFrame 12",4
        13,"Choreographer#doFrame 13",3
        """))

  def test_android_frames_choreographer_do_frame(self):
    return DiffTestBlueprint(
        trace=Path('../../metrics/graphics/android_jank_cuj.py'),
        query="""
        INCLUDE PERFETTO MODULE android.frames.timeline;

        SELECT * FROM android_frames_choreographer_do_frame;
        """,
        out=Csv("""
        "id","frame_id","ui_thread_utid","upid","ts"
        2,10,2,2,0
        20,20,2,2,20000000
        27,30,2,2,30000000
        40,40,2,2,40000000
        51,60,2,2,70000000
        61,90,2,2,100000000
        69,100,2,2,200000000
        79,110,2,2,300000000
        85,120,2,2,400000000
        93,130,2,2,550000000
        99,140,2,2,608500000
        105,145,2,2,655000000
        108,150,2,2,700000000
        114,160,2,2,800000000
        147,1000,2,2,1100000000
        """))

  def test_android_frames_draw_frame(self):
    return DiffTestBlueprint(
        trace=Path('../../metrics/graphics/android_jank_cuj.py'),
        query="""
        INCLUDE PERFETTO MODULE android.frames.timeline;

        SELECT * FROM android_frames_draw_frame;
        """,
        out=Csv("""
        "id","frame_id","render_thread_utid","upid"
        13,10,4,2
        21,20,4,2
        28,30,4,2
        46,40,4,2
        55,60,4,2
        63,90,4,2
        66,90,4,2
        72,100,4,2
        75,100,4,2
        80,110,4,2
        86,120,4,2
        95,130,4,2
        101,140,4,2
        106,145,4,2
        111,150,4,2
        115,160,4,2
        153,1000,4,2
        """))

  def test_android_frames(self):
    return DiffTestBlueprint(
        trace=Path('../../metrics/graphics/android_jank_cuj.py'),
        query="""
        INCLUDE PERFETTO MODULE android.frames.timeline;

        SELECT * FROM android_frames;
        """,
        out=Csv("""
        "frame_id","ts","dur","do_frame_id","draw_frame_id","actual_frame_timeline_id","expected_frame_timeline_id","render_thread_utid","ui_thread_utid","actual_frame_timeline_count","expected_frame_timeline_count","draw_frame_count","upid","process_name"
        10,0,16000000,2,13,1,0,4,2,1,1,1,2,"com.android.systemui"
        20,8000000,28000000,20,21,17,16,4,2,1,1,1,2,"com.android.systemui"
        30,30000000,25000000,27,28,26,25,4,2,1,1,1,2,"com.android.systemui"
        40,40000000,40000000,40,46,42,41,4,2,1,1,1,2,"com.android.systemui"
        60,70000000,20000000,51,55,53,52,4,2,2,1,1,2,"com.android.systemui"
        90,100000000,23000000,61,63,60,59,4,2,1,1,2,2,"com.android.systemui"
        100,200000000,22000000,69,72,71,70,4,2,1,1,2,2,"com.android.systemui"
        110,300000000,80000000,79,80,77,76,4,2,2,1,1,2,"com.android.systemui"
        120,400000000,61000000,85,86,84,83,4,2,1,1,1,2,"com.android.systemui"
        130,500000000,6000000,93,95,91,90,4,2,2,1,1,2,"com.android.systemui"
        140,608600000,17000000,99,101,100,97,4,2,1,1,1,2,"com.android.systemui"
        145,650000000,20000000,105,106,104,103,4,2,1,1,1,2,"com.android.systemui"
        150,700500000,14500000,108,111,110,109,4,2,1,1,1,2,"com.android.systemui"
        160,800000000,2000000,114,115,"[NULL]",113,4,2,0,1,1,2,"com.android.systemui"
        1000,1100000000,500000000,147,153,145,144,4,2,1,1,1,2,"com.android.systemui"
        """))

  def test_android_frames_layers(self):
    return DiffTestBlueprint(
        trace=Path('../../metrics/graphics/android_jank_cuj.py'),
        query="""
          INCLUDE PERFETTO MODULE android.frames.timeline;

          SELECT * FROM android_frames_layers;
          """,
        out=Csv("""
          "frame_id","ts","dur","ts_end","do_frame_id","draw_frame_id","actual_frame_timeline_id","expected_frame_timeline_id","render_thread_utid","ui_thread_utid","layer_id","layer_name","upid","process_name"
          10,0,16000000,16000000,2,13,1,0,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          20,8000000,28000000,36000000,20,21,17,16,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          30,30000000,25000000,55000000,27,28,26,25,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          40,40000000,40000000,80000000,40,46,42,41,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          60,70000000,10000000,80000000,51,55,53,52,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          60,70000000,20000000,90000000,51,55,54,52,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          90,100000000,23000000,123000000,61,63,60,59,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          90,100000000,23000000,123000000,61,66,60,59,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          100,200000000,22000000,222000000,69,72,71,70,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          100,200000000,22000000,222000000,69,75,71,70,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          110,300000000,61000000,361000000,79,80,77,76,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          110,300000000,80000000,380000000,79,80,78,76,4,2,1,"TX - JankyLayer#1",2,"com.android.systemui"
          120,400000000,61000000,461000000,85,86,84,83,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          130,500000000,2000000,502000000,93,95,91,90,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          130,550000000,6000000,556000000,93,95,94,90,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          140,608600000,17000000,625600000,99,101,100,97,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          145,650000000,20000000,670000000,105,106,104,103,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          150,700500000,14500000,715000000,108,111,110,109,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          160,800000000,2000000,802000000,114,115,"[NULL]",113,4,2,"[NULL]","[NULL]",2,"com.android.systemui"
          1000,1100000000,500000000,1600000000,147,153,145,144,4,2,0,"TX - NotificationShade#0",2,"com.android.systemui"
          """))

  def test_android_first_frame_after(self):
    return DiffTestBlueprint(
        trace=Path('../../metrics/graphics/android_jank_cuj.py'),
        query="""
        INCLUDE PERFETTO MODULE android.frames.timeline;

        SELECT frame_id FROM android_first_frame_after(100000000);
        """,
        out=Csv("""
        "frame_id"
        100
        """))

  def test_android_frames_overrun(self):
    return DiffTestBlueprint(
        trace=Path('../../metrics/graphics/android_jank_cuj.py'),
        query="""
        INCLUDE PERFETTO MODULE android.frames.per_frame_metrics;

        SELECT * FROM android_frames_overrun;
        """,
        out=Csv("""
        "frame_id","overrun"
        10,0
        20,8000000
        30,5000000
        40,20000000
        60,-10000000
        90,3000000
        100,2000000
        110,41000000
        120,41000000
        130,-18000000
        140,5600000
        145,0
        150,-5000000
        160,266000000
        190,0
        200,16000000
        1000,480000000
        """))

  def test_android_app_vsync_delay_per_frame(self):
    return DiffTestBlueprint(
        trace=Path('../../metrics/graphics/android_jank_cuj.py'),
        query="""
        INCLUDE PERFETTO MODULE android.frames.per_frame_metrics;

        SELECT * FROM android_app_vsync_delay_per_frame;
        """,
        out=Csv("""
        "frame_id","app_vsync_delay"
        10,0
        20,0
        30,0
        40,0
        60,0
        90,0
        100,0
        110,0
        120,0
        130,0
        140,8600000
        145,0
        150,500000
        1000,0
        """))

  def test_android_cpu_time_per_frame(self):
    return DiffTestBlueprint(
        trace=Path('../../metrics/graphics/android_jank_cuj.py'),
        query="""
        INCLUDE PERFETTO MODULE android.frames.per_frame_metrics;

        SELECT * FROM android_cpu_time_per_frame;
        """,
        out=Csv("""
        "frame_id","app_vsync_delay","do_frame_dur","draw_frame_dur","cpu_time"
        10,0,5000000,1000000,6000000
        20,0,3000000,4000000,7000000
        30,0,3000000,19000000,22000000
        40,0,13000000,7000000,20000000
        60,0,10000000,9000000,19000000
        90,0,15000000,8000000,23000000
        100,0,15000000,8000000,23000000
        110,0,15000000,2000000,17000000
        120,0,15000000,2000000,17000000
        130,0,5000000,4000000,9000000
        140,8600000,1500000,17000000,27100000
        145,0,20000000,3000000,23000000
        150,500000,2000000,13800000,16300000
        1000,0,100000000,150000000,250000000
        """))

  def test_android_frame_stats(self):
    return DiffTestBlueprint(
        trace=Path('../../metrics/graphics/android_jank_cuj.py'),
        query="""
        INCLUDE PERFETTO MODULE android.frames.per_frame_metrics;

        SELECT * FROM android_frame_stats;
        """,
        out=Csv("""
        "frame_id","overrun","cpu_time","ui_time","was_jank","was_slow_frame","was_big_jank","was_huge_jank"
        10,0,6000000,5000000,"[NULL]","[NULL]","[NULL]","[NULL]"
        20,8000000,7000000,3000000,1,"[NULL]","[NULL]","[NULL]"
        30,5000000,22000000,3000000,1,1,"[NULL]","[NULL]"
        40,20000000,20000000,13000000,1,"[NULL]","[NULL]","[NULL]"
        60,-10000000,19000000,10000000,"[NULL]","[NULL]","[NULL]","[NULL]"
        90,3000000,23000000,15000000,1,1,"[NULL]","[NULL]"
        100,2000000,23000000,15000000,1,1,"[NULL]","[NULL]"
        110,41000000,17000000,15000000,1,"[NULL]","[NULL]","[NULL]"
        120,41000000,17000000,15000000,1,"[NULL]","[NULL]","[NULL]"
        130,-18000000,9000000,5000000,"[NULL]","[NULL]","[NULL]","[NULL]"
        140,5600000,27100000,1500000,1,1,"[NULL]","[NULL]"
        145,0,23000000,20000000,"[NULL]",1,"[NULL]","[NULL]"
        150,-5000000,16300000,2000000,"[NULL]","[NULL]","[NULL]","[NULL]"
        160,266000000,7000000,2000000,1,"[NULL]","[NULL]","[NULL]"
        1000,480000000,250000000,100000000,1,1,1,1
        """))