/****************************************************************************
 * Copyright (c) 2025, ArborX authors                                       *
 * All rights reserved.                                                     *
 *                                                                          *
 * This file is part of the ArborX library. ArborX is                       *
 * distributed under a BSD 3-clause license. For the licensing terms see    *
 * the LICENSE file in the top-level directory.                             *
 *                                                                          *
 * SPDX-License-Identifier: BSD-3-Clause                                    *
 ****************************************************************************/
#include "data.hpp"

#include <Kokkos_Core.hpp>

#include <fstream>
#include <string>

#include "data_timpl.hpp"

namespace ArborXBenchmark
{

// Explicit instantiations
using MemorySpace = typename Kokkos::DefaultExecutionSpace::memory_space;
#define INSTANTIATE_LOADER(DIM, Coordinate)                                    \
  template Kokkos::View<ArborX::Point<DIM, Coordinate> *, MemorySpace>         \
  loadData<DIM, Coordinate, MemorySpace>(ArborXBenchmark::Parameters const &)
INSTANTIATE_LOADER(2, float);
INSTANTIATE_LOADER(3, float);
INSTANTIATE_LOADER(4, float);
INSTANTIATE_LOADER(5, float);
INSTANTIATE_LOADER(6, float);
INSTANTIATE_LOADER(2, double);
INSTANTIATE_LOADER(3, double);
INSTANTIATE_LOADER(4, double);
INSTANTIATE_LOADER(5, double);
INSTANTIATE_LOADER(6, double);
#undef INSTANTIATE_LOADER

#ifdef ARBORX_ENABLE_MPI
#define INSTANTIATE_MPI_LOADER(DIM, Coordinate)                                \
  template Kokkos::View<ArborX::Point<DIM, Coordinate> *, MemorySpace>         \
  loadData<DIM, Coordinate, MemorySpace>(MPI_Comm,                             \
                                         ArborXBenchmark::Parameters const &);
INSTANTIATE_MPI_LOADER(2, float);
INSTANTIATE_MPI_LOADER(3, float);
INSTANTIATE_MPI_LOADER(4, float);
INSTANTIATE_MPI_LOADER(5, float);
INSTANTIATE_MPI_LOADER(6, float);
INSTANTIATE_MPI_LOADER(2, double);
INSTANTIATE_MPI_LOADER(3, double);
INSTANTIATE_MPI_LOADER(4, double);
INSTANTIATE_MPI_LOADER(5, double);
INSTANTIATE_MPI_LOADER(6, double);
#undef INSTANTIATE_LOADER
#endif

int getDataDimension(std::string const &filename, bool binary)
{
  std::ifstream input;
  if (!binary)
    input.open(filename);
  else
    input.open(filename, std::ifstream::binary);
  if (!input.good())
    throw std::runtime_error("Error reading file \"" + filename + "\"");

  int num_points;
  int dim;
  if (!binary)
  {
    input >> num_points;
    input >> dim;
  }
  else
  {
    input.read(reinterpret_cast<char *>(&num_points), sizeof(int));
    input.read(reinterpret_cast<char *>(&dim), sizeof(int));
  }
  input.close();

  return dim;
}

} // namespace ArborXBenchmark