#!/usr/bin/env python3
#
# 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.

import os
from pathlib import Path
import re
import tempfile

from binary_cache_builder import BinaryCacheBuilder
from . test_utils import TestBase, TestHelper


class TestAnnotate(TestBase):
    def test_annotate(self):
        testdata_file = TestHelper.testdata_path('runtest_two_functions_arm64_perf.data')

        # Build binary_cache.
        binary_cache_builder = BinaryCacheBuilder(TestHelper.ndk_path, False)
        binary_cache_builder.build_binary_cache(testdata_file, [TestHelper.testdata_dir])

        # Generate annotated files.
        source_dir = TestHelper.testdata_dir
        self.run_cmd(['annotate.py', '-i', testdata_file, '-s',
                      str(source_dir), '--summary-width', '1000'])

        # Check annotated files.
        annotate_dir = Path('annotated_files')
        summary_file = annotate_dir / 'summary'
        check_items = [
            re.compile(r'100.00% \| 100.00% \| .+two_functions.cpp'),
            '100.00% | 0.00%  | main (line 20)',
            '50.06%  | 50.06% | line 16',
        ]
        self.check_strings_in_file(summary_file, check_items)

        source_files = list(annotate_dir.glob('**/*.cpp'))
        self.assertEqual(len(source_files), 1)
        source_file = source_files[0]
        self.assertEqual(source_file.name, 'two_functions.cpp')
        check_items = ['/* Total 50.06%, Self 50.06%          */    *p = i;']
        self.check_strings_in_file(source_file, check_items)

    def test_sample_filters(self):
        def get_report(filter: str):
            self.run_cmd(['annotate.py', '-i', TestHelper.testdata_path(
                'perf_display_bitmaps.data')] + filter.split())

        get_report('--exclude-pid 31850')
        get_report('--include-pid 31850')
        get_report('--pid 31850')
        get_report('--exclude-tid 31881')
        get_report('--include-tid 31881')
        get_report('--tid 31881')
        get_report('--exclude-process-name com.example.android.displayingbitmaps')
        get_report('--include-process-name com.example.android.displayingbitmaps')
        get_report('--exclude-thread-name com.example.android.displayingbitmaps')
        get_report('--include-thread-name com.example.android.displayingbitmaps')

        with tempfile.NamedTemporaryFile('w', delete=False) as filter_file:
            filter_file.write('GLOBAL_BEGIN 684943449406175\nGLOBAL_END 684943449406176')
            filter_file.flush()
            get_report('--filter-file ' + filter_file.name)
        os.unlink(filter_file.name)

    def test_show_art_frames(self):
        self.run_cmd(
            ['annotate.py', '-i', TestHelper.testdata_path('perf_with_interpreter_frames.data'),
             '--show-art-frames'])
        summary = Path('annotated_files') / 'summary'
        self.check_strings_in_file(summary, 'total period: 9800649')