/******************************************************************************
 * Copyright (c) Advanced Micro Devices, Inc. All rights reserved.
 *
 * SPDX-License-Identifier: MIT
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to
 * deal in the Software without restriction, including without limitation the
 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
 * sell copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
 * IN THE SOFTWARE.
 *****************************************************************************/

#ifndef _TESTER_ARGUMENTS_HPP_
#define _TESTER_ARGUMENTS_HPP_

#include <climits>
#include <cstdint>
#include <rocshmem/rocshmem.hpp>
#include <string>
#include <iostream>


enum TeamSplitType {
  ROCSHMEM_TEST_TEAM_DUP = 0,    // Dup parent team
  ROCSHMEM_TEST_TEAM_SINGLE,     // each PE will be its own team
  ROCSHMEM_TEST_TEAM_BLOCK,      // split parent into two halves
  ROCSHMEM_TEST_TEAM_ODDEVEN,    // odd-even splitting
  ROCSHMEM_TEST_TEAM_SHARED,     // predefined ROCSHMEM_TEAM_SHARED
  ROCSHMEM_TEST_TEAM_SUBSET_PARENT, // split a subset parent team (not TEAM_WORLD)
};

enum UserBufType {
  USER_BUF_TYPE_HOST,
  USER_BUF_TYPE_DEVICE,
  USER_BUF_TYPE_FINE,
  USER_BUF_TYPE_UNCACHED,
  USER_BUF_TYPE_MANAGED,
  USER_BUF_TYPE_HEAP,
};

/*-----------------------------------------
 * Atomics Addressing modes (contention model)
 *-----------------------------------------*/
enum class AddrMode : int {
  PerGrid,     // all WGs -> same address
  PerBlock,    // each WG -> its own address (default)
};

class TesterArguments {
 public:
  TesterArguments(int argc, char *argv[]);

  /**
   * Initialize rocshmem members
   * Valid after rocshmem_init function called.
   */
  void get_arguments();

 private:
  /**
   * Output method which displays available command line options
   */
  static void show_usage(std::string executable_name);

public:
  /**
   * Program name (argv[0]) for path resolution, e.g. finding HSACO next to binary
   */
  std::string executable_name;

  /**
   * Arguments obtained from command line
   */
  unsigned num_wgs = 1;
  unsigned num_threads = 1;
  unsigned algorithm = 0;
  size_t min_msg_size = 1;
  size_t max_msg_size = 1 << 20;
  size_t max_volume_size = 0;
  unsigned wg_size = 64;
  unsigned thread_access = 64;
  unsigned coal_coef = 64;
  unsigned op_type = 0;
  unsigned shmem_context = rocshmem::ROCSHMEM_CTX_WG_PRIVATE;
  AddrMode addr_mode = AddrMode::PerBlock;
  enum UserBufType local_buf_type = USER_BUF_TYPE_HEAP;

  /**
   * Arguments obtained from rocshmem
   */
  int numprocs = INT_MAX;
  int myid = INT_MAX;

  /**
   * Defaults tester values
   */
  int loop = 10;
  int skip = 10;
  int loop_large = 10;
  int batch = 0;
  bool verif = true;
  size_t large_message_size = 32768;

  TeamSplitType team_type = ROCSHMEM_TEST_TEAM_DUP;
};

#endif