#include <gtest/gtest.h>
#include "core/algo/GprVolumeBuilder.hpp"
#include "io/gpr/RadarVolumeAssembler.hpp"

using geopro::io::gpr::RadarCubeDesc;
using geopro::io::gpr::assembleRadarVolume;

// 2 道 × 3 通道 × 4 采样，值 = 100*c + 10*t + s。不插值(chXOffsets 空)、coarse=1。
TEST(RadarVolumeAssembler, AxisMapAndQuantRoundTrip) {
  RadarCubeDesc d;
  d.channels = 3; d.traces = 2; d.samples = 4;
  d.dxBase = 0.1; d.dyWhenNotInterpolated = 0.5; d.dz = 0.05;
  auto sampler = [](int c, int t, int s) { return 100.0 * c + 10.0 * t + s; };

  const geopro::core::BuiltI16 b = assembleRadarVolume(d, sampler, /*coarse=*/1, /*targetDy=*/0.0);

  EXPECT_EQ(b.vol.nx(), 2);  // 道
  EXPECT_EQ(b.vol.ny(), 3);  // 通道(未插值=原通道数)
  EXPECT_EQ(b.vol.nz(), 4);  // 采样
  EXPECT_DOUBLE_EQ(b.spacing[0], 0.1);
  EXPECT_DOUBLE_EQ(b.spacing[1], 0.5);
  EXPECT_DOUBLE_EQ(b.spacing[2], 0.05);
  EXPECT_NEAR(b.vminPhys, 0.0, 1e-9);          // c0,t0,s0
  EXPECT_NEAR(b.vmaxPhys, 213.0, 1e-9);        // c2,t1,s3 = 200+10+3
  // 反量化对位：体素(道 t=1, 通道 c=2, 采样 s=3) 应≈213(量化误差内)。
  const double recon = b.quant.toPhys(b.vol.at(1, 2, 3));
  EXPECT_NEAR(recon, 213.0, b.quant.scale);
}

// coarse=2：4 道 → nxOut=2，dx×2。
TEST(RadarVolumeAssembler, CoarseDownsamplesTracesAndScalesDx) {
  RadarCubeDesc d;
  d.channels = 1; d.traces = 4; d.samples = 2; d.dxBase = 0.1;
  auto sampler = [](int, int t, int s) { return 10.0 * t + s; };
  const geopro::core::BuiltI16 b = assembleRadarVolume(d, sampler, /*coarse=*/2, 0.0);
  EXPECT_EQ(b.vol.nx(), 2);
  EXPECT_DOUBLE_EQ(b.spacing[0], 0.2);
  EXPECT_NEAR(b.quant.toPhys(b.vol.at(1, 0, 0)), 20.0, b.quant.scale);  // 输出道1 = 源道2
}