library(acousticTS)

test_that("numerical model tables retain data-frame recycling and metadata", {
  columns <- list(
    list(frequency = c(38000, 70000, 120000), f_bs = 1 + 2i, valid = TRUE),
    list(frequency = integer(), f_bs = complex()),
    list(frequency = c(low = 38000, high = 120000), sigma_bs = c(1, 2)),
    list(frequency = 1:4, sigma_bs = c(1, 2)),
    list(frequency = c(38000, 120000), medium = factor(c("water", "water"))),
    list(frequency = c(38000, 120000), coefficients = I(matrix(1:4, 2)))
  )
  for (args in columns) {
    expect_identical(
      do.call(acousticTS:::.model_data_frame, args),
      do.call(data.frame, args)
    )
  }
  expect_error(
    acousticTS:::.model_data_frame(frequency = 1:3, sigma_bs = 1:2),
    "differing number of rows"
  )
  expect_error(
    acousticTS:::.model_data_frame(frequency = numeric(), sigma_bs = 1),
    "differing number of rows"
  )
})

test_that("calibration_initialize works correctly", {
  # Test calibration_initialize ================================================
  cal_obj <- cal_generate()
  frequency <- c(38e3, 70e3, 120e3)

  # Initialize calibration model ===============================================
  cal_initialized <- calibration_initialize(cal_obj, frequency = frequency)

  # Check that object is still CAL class =======================================
  expect_s4_class(cal_initialized, "CAL")

  # Check that model parameters slot was updated ===============================
  expect_type(cal_initialized@model_parameters, "list")
  expect_true("calibration" %in% names(cal_initialized@model_parameters))

  # Check that model slot was updated ==========================================
  expect_type(cal_initialized@model, "list")
  expect_true("calibration" %in% names(cal_initialized@model))

  # Check that frequency was properly set ======================================
  expect_equal(cal_initialized@model$calibration$frequency, frequency)
})

test_that("sphms_initialize works correctly", {
  # Test mss_anderson_initialize for GAS objects ===============================
  gas_obj <- gas_generate(shape = sphere(radius_body = 1, n_segments = 80))
  frequency <- c(38e3, 120e3)

  # Initialize MSS Anderson model ==============================================
  gas_initialized <- sphms_initialize(gas_obj, frequency = frequency)

  # Check that object is still GAS class =======================================
  expect_s4_class(gas_initialized, "GAS")

  # Check that model parameters slot was updated ===============================
  expect_type(gas_initialized@model_parameters, "list")
  expect_true("SPHMS" %in% names(gas_initialized@model_parameters))

  # Check that model slot was updated ==========================================
  expect_type(gas_initialized@model, "list")
  expect_true("SPHMS" %in% names(gas_initialized@model))

  # Check that frequency was properly set ======================================
  expect_equal(gas_initialized@model$SPHMS$frequency, frequency)
})

test_that("essms_initialize works correctly", {
  # Test mss_goodman_stern_initialize for ESS objects ==========================
  ess_obj <- ess_generate(shape = sphere(radius_body = 1, n_segments = 80))
  frequency <- c(70e3, 200e3)

  # Initialize MSS Goodman-Stern model =========================================
  ess_initialized <- essms_initialize(ess_obj,
    frequency = frequency
  )

  # Check that object is still ESS class =======================================
  expect_s4_class(ess_initialized, "ESS")

  # Check that model parameters slot was updated ===============================
  expect_type(ess_initialized@model_parameters, "list")
  expect_true("ESSMS" %in% names(ess_initialized@model_parameters))

  # Check that model slot was updated ==========================================
  expect_type(ess_initialized@model, "list")
  expect_true("ESSMS" %in% names(ess_initialized@model))
})

test_that("dwba_initialize works correctly", {
  # Test dwba_initialize for FLS objects =======================================
  fls_obj <- fls_generate(
    x = 1, y = 1, z = 1, radius_body = 1,
    g_body = 1, h_body = 1
  )
  frequency <- c(38e3, 120e3)

  # Initialize DWBA model ======================================================
  fls_initialized <- dwba_initialize(fls_obj, frequency = frequency)

  # Check that object is still FLS class =======================================
  expect_s4_class(fls_initialized, "FLS")

  # Check that model parameters slot was updated ===============================
  expect_type(fls_initialized@model_parameters, "list")
  expect_true("DWBA" %in% names(fls_initialized@model_parameters))

  # Check that model slot was updated ==========================================
  expect_type(fls_initialized@model, "list")
  expect_true("DWBA" %in% names(fls_initialized@model))

  # Check that frequency was properly set ======================================
  expect_equal(fls_initialized@model$DWBA$frequency, frequency)
})

test_that("dwba_curved_initialize works correctly", {
  # Test dwba_curved_initialize for FLS objects ================================
  fls_obj <- fls_generate(
    x = seq(0, 1, length.out = 11),
    y = rep(0, 11),
    z = rep(0, 11),
    radius_body = rep(0.5, 11),
    g_body = 1, h_body = 1,
    radius_curvature_ratio = 3
  )
  frequency <- c(120e3)

  # Initialize DWBA curved model ===============================================
  expect_warning(
    fls_initialized <- dwba_curved_initialize(fls_obj, frequency = frequency),
    "deprecated"
  )

  # Check that object is still FLS class =======================================
  expect_s4_class(fls_initialized, "FLS")

  # Check that model parameters slot was updated ===============================
  expect_type(fls_initialized@model_parameters, "list")
  expect_true("DWBA_curved" %in% names(fls_initialized@model_parameters))

  # Check that model slot was updated ==========================================
  expect_type(fls_initialized@model, "list")
  expect_true("DWBA_curved" %in% names(fls_initialized@model))
})

# test_that("dcm_initialize works correctly", { ================================
# ==============================================================================
#   # Test dcm_initialize for FLS objects ======================================
#   fls_obj <- fls_generate( ===================================================
#     x = 1, y = 1, z = 1, radius_body = 1, ====================================
#     g_body = 1, h_body = 1 ===================================================
#   ) ==========================================================================
#   frequency <- c(38e3, 120e3) ================================================
# ==============================================================================
#   # Initialize DCM model =====================================================
#   fls_initialized <- dcm_initialize(fls_obj, frequency = frequency) ==========
# ==============================================================================
#   # Check that object is still FLS class =====================================
#   expect_s4_class(fls_initialized, "FLS") ====================================
# ==============================================================================
#   # Check that model parameters slot was updated =============================
#   expect_type(fls_initialized@model_parameters, "list") ======================
#   expect_true("DCM" %in% names(fls_initialized@model_parameters)) ============
# ==============================================================================
#   # Check that model slot was updated ========================================
#   expect_type(fls_initialized@model, "list") =================================
#   expect_true("DCM" %in% names(fls_initialized@model)) =======================
# }) ===========================================================================

test_that("sdwba_initialize works correctly", {
  # Test sdwba_initialize for FLS objects ======================================
  fls_obj <- fls_generate(
    x = seq(0, 1, length.out = 11),
    y = rep(0, 11),
    z = rep(0, 11),
    radius_body = rep(0.5, 11),
    g_body = 1, h_body = 1,
    radius_curvature_ratio = 3
  )
  frequency <- c(120e3, 200e3)

  # Initialize SDWBA model =====================================================
  fls_initialized <- sdwba_initialize(fls_obj, frequency = frequency)

  # Check that object is still FLS class =======================================
  expect_s4_class(fls_initialized, "FLS")

  # Check that model parameters slot was updated ===============================
  expect_type(fls_initialized@model_parameters, "list")
  expect_true("SDWBA" %in% names(fls_initialized@model_parameters))

  # Check that model slot was updated ==========================================
  expect_type(fls_initialized@model, "list")
  expect_true("SDWBA" %in% names(fls_initialized@model))
})

test_that("sdwba_curved_initialize works correctly", {
  # Test sdwba_curved_initialize for FLS objects ===============================
  fls_obj <- fls_generate(
    x = seq(0, 1, length.out = 11),
    y = rep(0, 11),
    z = rep(0, 11),
    radius_body = rep(0.5, 11),
    g_body = 1, h_body = 1,
    radius_curvature_ratio = 3
  )
  frequency <- c(120e3)

  # Initialize SDWBA curved model ==============================================
  expect_warning(
    fls_initialized <- sdwba_curved_initialize(fls_obj, frequency = frequency),
    "deprecated"
  )

  # Check that object is still FLS class =======================================
  expect_s4_class(fls_initialized, "FLS")

  # Check that model parameters slot was updated ===============================
  expect_type(fls_initialized@model_parameters, "list")
  expect_true("SDWBA_curved" %in% names(fls_initialized@model_parameters))

  # Check that model slot was updated ==========================================
  expect_type(fls_initialized@model, "list")
  expect_true("SDWBA_curved" %in% names(fls_initialized@model))
})

test_that("krm_initialize works correctly", {
  # Test krm_initialize for SBF objects ========================================
  # SBF should have both body and bladder ======================================
  x_body <- seq(0, 0.05, length.out = 6)
  w_body <- rep(0.005, 6)
  zU_body <- rep(0.0025, 6)
  zL_body <- rep(-0.0025, 6)
  x_bladder <- seq(0.01, 0.04, length.out = 4)
  w_bladder <- rep(0.002, 4)
  zU_bladder <- rep(0.001, 4)
  zL_bladder <- rep(-0.001, 4)

  sbf_obj <- sbf_generate(
    x_body = x_body, w_body = w_body, zU_body = zU_body, zL_body = zL_body,
    x_bladder = x_bladder, w_bladder = w_bladder,
    zU_bladder = zU_bladder, zL_bladder = zL_bladder,
    sound_speed_body = 1570, sound_speed_bladder = 345,
    density_body = 1070, density_bladder = 1.24
  )
  frequency <- c(38e3, 70e3)

  # Initialize KRM model =======================================================
  sbf_initialized <- krm_initialize(sbf_obj, frequency = frequency)

  # Check that object is still SBF class =======================================
  expect_s4_class(sbf_initialized, "SBF")

  # Check that model parameters slot was updated ===============================
  expect_type(sbf_initialized@model_parameters, "list")
  expect_true("KRM" %in% names(sbf_initialized@model_parameters))

  # Check that model slot was updated ==========================================
  expect_type(sbf_initialized@model, "list")
  expect_true("KRM" %in% names(sbf_initialized@model))
})


test_that("Initialization preserves object integrity", {
  # Test that initialization doesn't break the object structure ================
  cal_obj <- cal_generate()
  original_metadata <- cal_obj@metadata

  cal_initialized <- calibration_initialize(cal_obj, frequency = c(38e3))

  # Metadata should be preserved ===============================================
  expect_equal(cal_initialized@metadata, original_metadata)

  # Object should still have all required slots ================================
  expect_true("metadata" %in% slotNames(cal_initialized))
  expect_true("model_parameters" %in% slotNames(cal_initialized))
  expect_true("model" %in% slotNames(cal_initialized))
})