#include <gtest/gtest.h>

#include <vtkPolyData.h>
#include <vtkPolyDataMapper.h>

#include "actors/CurtainActor.hpp"
#include "geo/GeoLocalFrame.hpp"
#include "model/ColorScale.hpp"
#include "model/Field.hpp"

using namespace geopro::core;

// buildCurtain: 小 Grid(3,2) + lat/lon/y/values -> 非空 actor，经 banded contour 管线
// (vtkDataSetSurfaceFilter→vtkBandedPolyDataContourFilter→vtkPolyDataMapper) 产出非空 polydata。
TEST(Curtain, BuildsStructuredGridFromSmallGrid) {
    Grid g(3, 2);  // nx=3 (lat/lon 列), ny=2 (深度行)
    g.lat = {22.50, 22.50, 22.50};
    g.lon = {114.16, 114.17, 114.18};
    g.y = {0.0, -10.0};  // 两层深度
    double n = 0.0;
    for (int j = 0; j < 2; ++j)
        for (int i = 0; i < 3; ++i) g.valueAt(i, j) = n++;
    g.vmin = 0.0;
    g.vmax = 5.0;

    ColorScale cs;
    cs.addStop(0.0, Rgba{0, 0, 255, 255});
    cs.addStop(5.0, Rgba{255, 0, 0, 255});

    GeoLocalFrame frame(22.50, 114.16);

    auto actor = geopro::render::buildCurtain(g, cs, frame);
    ASSERT_NE(actor.GetPointer(), nullptr);

    auto* mapper = vtkPolyDataMapper::SafeDownCast(actor->GetMapper());
    ASSERT_NE(mapper, nullptr);
    ASSERT_NE(mapper->GetLookupTable(), nullptr);

    // 运行管线，断言 banded contour 产出非空 polydata（色带面）。
    mapper->Update();
    auto* out = mapper->GetInput();
    ASSERT_NE(out, nullptr);
    EXPECT_GT(out->GetNumberOfPoints(), 0);
    EXPECT_GT(out->GetNumberOfCells(), 0);
}