#!/usr/bin/env python
#
# Copyright (C) 2020 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 argparse
import os
import sys

import ltp_test_cases
from common import filter_utils

def run(arch: str, n_bit: int, is_low_mem: bool, is_hwasan: bool, run_staging: bool, output_file: str):

    ltp_tests = ltp_test_cases.LtpTestCases(None)

    test_filter = filter_utils.Filter()
    ltp_tests.GenConfig(
        arch,
        n_bit,
        test_filter,
        output_file=output_file,
        run_staging=run_staging,
        is_low_mem=is_low_mem,
        is_hwasan=is_hwasan)

if __name__ == '__main__':
    arg_parser = argparse.ArgumentParser(
        description='Generate LTP configuration files for VTS')
    arg_parser.add_argument('--arch',
                            dest='arch',
                            type=str,
                            choices=['arm', 'riscv', 'x86'],
                            required=True,
                            help="Target device architecture")
    arg_parser.add_argument('--bitness',
                            dest='bitness',
                            type=int,
                            choices=[32, 64],
                            required=True,
                            help="Target device architecture bitness")
    arg_parser.add_argument('--low-mem',
                            dest='is_low_mem',
                            type=str,
                            choices=['True', 'False'],
                            default='False',
                            help="Target device is low memory device")
    arg_parser.add_argument('--hwasan',
                            dest='is_hwasan',
                            type=str,
                            choices=['True', 'False'],
                            default='False',
                            help="Target device is hwasan")
    arg_parser.add_argument('--staging',
                            dest='run_staging',
                            type=str,
                            choices=['True', 'False'],
                            default='False',
                            help="Run all the tests, except from the disabled ones")
    arg_parser.add_argument('output_file_path',
                            help="Path for the output file")
    args = arg_parser.parse_args()

    run(arch=args.arch,
        n_bit=str(args.bitness),
        is_low_mem=args.is_low_mem == 'True',
        is_hwasan=args.is_hwasan == 'True',
        run_staging=args.run_staging == 'True',
        output_file=args.output_file_path)