#
# Copyright (C) 2018-2022 The ESPResSo project
#
# This file is part of ESPResSo.
#
# ESPResSo is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# ESPResSo is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#

include(ProcessorCount)
ProcessorCount(NP)

if(EXISTS ${MPIEXEC})
  # OpenMPI 3.0 and higher checks the number of processes against the number of
  # CPUs
  execute_process(
    COMMAND ${MPIEXEC} --version RESULT_VARIABLE mpi_version_result
    OUTPUT_VARIABLE mpi_version_output ERROR_VARIABLE mpi_version_output)
  if(mpi_version_result EQUAL 0 AND mpi_version_output MATCHES
                                    "\\(Open(RTE| MPI)\\) ([3-9]\\.|1[0-9])")
    set(ESPRESSO_MPIEXEC_PREFLAGS "--oversubscribe")
  else()
    set(ESPRESSO_MPIEXEC_PREFLAGS "")
  endif()
endif()

function(SET_BENCHMARK_PROPERTIES)
  set_tests_properties(
    ${ARGV0} PROPERTIES RUN_SERIAL TRUE SKIP_REGULAR_EXPRESSION
                        "espressomd.FeaturesError: Missing features")
endfunction()

function(PYTHON_BENCHMARK)
  cmake_parse_arguments(
    BENCHMARK "" "FILE;RUN_WITH_MPI;MIN_NUM_PROC;MAX_NUM_PROC"
    "ARGUMENTS;DEPENDENCIES" ${ARGN})
  get_filename_component(BENCHMARK_NAME ${BENCHMARK_FILE} NAME_WE)
  foreach(argument IN LISTS BENCHMARK_ARGUMENTS)
    string(REGEX REPLACE "[^-a-zA-Z0-9_\\.]+" "_" argument ${argument})
    string(REGEX REPLACE "^[-_]+" "" argument ${argument})
    set(BENCHMARK_NAME "${BENCHMARK_NAME}__${argument}")
  endforeach(argument)
  list(APPEND BENCHMARK_DEPENDENCIES ${BENCHMARK_FILE} benchmarks.py)
  foreach(dependency IN LISTS BENCHMARK_DEPENDENCIES)
    configure_file(${dependency} ${CMAKE_CURRENT_BINARY_DIR}/${dependency})
  endforeach(dependency)
  set(BENCHMARK_FILE "${CMAKE_CURRENT_BINARY_DIR}/${BENCHMARK_FILE}")
  list(APPEND BENCHMARK_ARGUMENTS
       "--output=${CMAKE_BINARY_DIR}/benchmarks.csv.part")

  # default values
  if(NOT DEFINED BENCHMARK_RUN_WITH_MPI)
    set(BENCHMARK_RUN_WITH_MPI TRUE)
  endif()
  if(NOT DEFINED BENCHMARK_MIN_NUM_PROC)
    set(BENCHMARK_MIN_NUM_PROC 1)
  endif()
  if(NOT DEFINED BENCHMARK_MAX_NUM_PROC)
    set(BENCHMARK_MAX_NUM_PROC ${NP})
  endif()
  # parallel schemes
  if(EXISTS ${MPIEXEC} AND ${BENCHMARK_RUN_WITH_MPI})
    set(BENCHMARK_CONFIGURATIONS "sentinel")
    foreach(BENCHMARK_NUM_PROC 1 2 4 8 16)
      if(${BENCHMARK_MAX_NUM_PROC} GREATER_EQUAL ${BENCHMARK_NUM_PROC}
         AND ${BENCHMARK_MIN_NUM_PROC} LESS_EQUAL ${BENCHMARK_NUM_PROC}
         AND ${NP} GREATER_EQUAL ${BENCHMARK_NUM_PROC})
        list(APPEND BENCHMARK_CONFIGURATIONS ${BENCHMARK_NUM_PROC})
      endif()
    endforeach(BENCHMARK_NUM_PROC)
    list(REMOVE_AT BENCHMARK_CONFIGURATIONS 0)
    foreach(BENCHMARK_NUM_PROC IN LISTS BENCHMARK_CONFIGURATIONS)
      set(BENCHMARK_TEST_NAME
          benchmark__${BENCHMARK_NAME}__parallel_${BENCHMARK_NUM_PROC})
      add_test(
        NAME ${BENCHMARK_TEST_NAME}
        COMMAND
          ${MPIEXEC} ${ESPRESSO_MPIEXEC_PREFLAGS} ${MPIEXEC_NUMPROC_FLAG}
          ${BENCHMARK_NUM_PROC} ${MPIEXEC_PREFLAGS}
          ${CMAKE_BINARY_DIR}/pypresso ${BENCHMARK_FILE} ${BENCHMARK_ARGUMENTS}
          ${MPIEXEC_POSTFLAGS})
      set_benchmark_properties(${BENCHMARK_TEST_NAME})
    endforeach(BENCHMARK_NUM_PROC)
  else()
    set(BENCHMARK_TEST_NAME benchmark__${BENCHMARK_NAME}__serial)
    add_test(NAME ${BENCHMARK_TEST_NAME}
             COMMAND ${CMAKE_BINARY_DIR}/pypresso ${BENCHMARK_FILE}
                     ${BENCHMARK_ARGUMENTS})
    set_benchmark_properties(${BENCHMARK_TEST_NAME})
  endif()
endfunction(PYTHON_BENCHMARK)

python_benchmark(FILE lj.py ARGUMENTS
                 "--particles_per_core=1000;--volume_fraction=0.50")
python_benchmark(FILE lj.py ARGUMENTS
                 "--particles_per_core=1000;--volume_fraction=0.02")
python_benchmark(FILE lj.py ARGUMENTS
                 "--particles_per_core=10000;--volume_fraction=0.50")
python_benchmark(FILE lj.py ARGUMENTS
                 "--particles_per_core=10000;--volume_fraction=0.02")
python_benchmark(FILE mc_acid_base_reservoir.py ARGUMENTS
                 "--particles_per_core=500;--mode=benchmark")
python_benchmark(
  FILE lj.py ARGUMENTS
  "--particles_per_core=1000;--volume_fraction=0.10;--bonds" RUN_WITH_MPI FALSE)
python_benchmark(
  FILE lj.py ARGUMENTS
  "--particles_per_core=10000;--volume_fraction=0.10;--bonds" RUN_WITH_MPI
  FALSE)
python_benchmark(
  FILE p3m.py ARGUMENTS
  "--particles_per_core=1000;--volume_fraction=0.25;--prefactor=4")
python_benchmark(
  FILE p3m.py ARGUMENTS
  "--particles_per_core=10000;--volume_fraction=0.25;--prefactor=4")
python_benchmark(
  FILE lb.py ARGUMENTS
  "--particles_per_core=125;--volume_fraction=0.03;--lb_sites_per_particle=28")
python_benchmark(FILE ferrofluid.py ARGUMENTS "--particles_per_core=400")
python_benchmark(FILE mc_acid_base_reservoir.py ARGUMENTS
                 "--particles_per_core=500" RUN_WITH_MPI FALSE)
python_benchmark(FILE lb.py ARGUMENTS "--box_l=32;--single_precision")
python_benchmark(FILE lb.py ARGUMENTS "--box_l=32")
python_benchmark(FILE lb.py ARGUMENTS "--box_l=64;--single_precision")
python_benchmark(FILE lb.py ARGUMENTS "--box_l=64")
python_benchmark(FILE lb.py ARGUMENTS "--box_l=128;--single_precision")
python_benchmark(FILE lb.py ARGUMENTS "--box_l=128")
if(NOT ESPRESSO_BUILD_WITH_WALBERLA_USE_AVX)
  python_benchmark(FILE lb.py ARGUMENTS "--box_l=196;--single_precision")
  python_benchmark(FILE lb.py ARGUMENTS "--box_l=196")
endif()

add_custom_target(
  benchmarks_data
  COMMAND ${CMAKE_COMMAND} -E copy ${CMAKE_CURRENT_SOURCE_DIR}/benchmarks.py
          ${CMAKE_CURRENT_BINARY_DIR})
add_custom_target(
  benchmark_python
  COMMAND ${CMAKE_CTEST_COMMAND} --timeout ${ESPRESSO_TEST_TIMEOUT}
          ${ESPRESSO_CTEST_ARGS} --output-on-failure)

add_dependencies(benchmark_python pypresso benchmarks_data)
add_dependencies(benchmark benchmark_python)