/* SPDX-FileCopyrightText: Copyright (c) 2016 - 2026, the adamantine authors.
 * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 */

#include "MaterialStates.hh"
#define BOOST_TEST_MODULE Integration_Data_Assimilation_Augmented

#include "../applications/adamantine.hh"

#include <boost/property_tree/info_parser.hpp>

#include <filesystem>
#include <fstream>

#include "main.cc"

namespace utf = boost::unit_test;

BOOST_AUTO_TEST_CASE(integration_3D_data_assimilation_augmented,
                     *utf::tolerance(5.))
{

  MPI_Comm communicator = MPI_COMM_WORLD;

  std::vector<adamantine::Timer> timers;
  initialize_timers(communicator, timers);

  // Read the input.
  std::string const filename = "bare_plate_L_da_augmented.info";
  adamantine::ASSERT_THROW(std::filesystem::exists(filename) == true,
                           "The file " + filename + " does not exist.");
  boost::property_tree::ptree database;
  boost::property_tree::info_parser::read_info(filename, database);

  // Run the simulation
  auto result =
      run_ensemble<3, -1, 3, adamantine::SolidLiquidPowder,
                   dealii::MemorySpace::Host>(communicator, database, timers);

  // Three ensemble members expected
  unsigned int local_result_size = result.size();
  unsigned int global_result_size =
      dealii::Utilities::MPI::sum(local_result_size, communicator);
  BOOST_TEST(global_result_size == 3);

  // Get the average absorption value for each ensemble member
  double sum = 0.0;
  for (unsigned int member = 0; member < result.size(); ++member)
  {
    sum += result.at(member).block(1).local_element(0);
  }
  double partial_average_value = sum / global_result_size;
  double average_value =
      dealii::Utilities::MPI::sum(partial_average_value, communicator);

  // Based on the reference solution, the expected absorption efficiency is 0.3
  double gold_solution = 0.3;
  BOOST_TEST(average_value == gold_solution);
}