32 lines
1.5 KiB
C++
32 lines
1.5 KiB
C++
#include "source/WholeVolumeSource.hpp"
|
|
#include "data/store/ChunkedVolumeStore.hpp"
|
|
#include "core/algo/GprVolumeBuilder.hpp"
|
|
#include <vtkImageData.h>
|
|
#include <gtest/gtest.h>
|
|
#include <filesystem>
|
|
using namespace geopro;
|
|
|
|
TEST(WholeVolumeSource, ReassemblesFullVolumeFromStore) {
|
|
auto dir = (std::filesystem::temp_directory_path() / "gpr_wvs_test").string();
|
|
std::filesystem::remove_all(dir);
|
|
// 造一个可识别的体:值 = 全局 i+j+k(便于校验重组位置)
|
|
core::BuiltI16 b; b.vol = core::ScalarVolumeI16(100,40,30);
|
|
for (int k=0;k<30;k++)for(int j=0;j<40;j++)for(int i=0;i<100;i++) b.vol.at(i,j,k)=(short)((i+j+k)%1000);
|
|
b.quant={1.0,0.0}; b.origin={{1,2,3}}; b.spacing={{0.5,0.5,0.2}}; b.vminPhys=0; b.vmaxPhys=1000;
|
|
data::ChunkedVolumeStore::write(dir, b, 64); // 100/40/30 均非 64 整除→含边缘块
|
|
render::WholeVolumeSource src(dir);
|
|
EXPECT_EQ(src.meta().nx, 100);
|
|
auto imgs = src.currentImages();
|
|
ASSERT_EQ(imgs.size(), 1u);
|
|
auto* img = imgs[0].Get();
|
|
EXPECT_EQ(img->GetScalarType(), VTK_SHORT);
|
|
int dims[3]; img->GetDimensions(dims);
|
|
EXPECT_EQ(dims[0],100); EXPECT_EQ(dims[1],40); EXPECT_EQ(dims[2],30);
|
|
// 关键:重组位置正确(边缘块也对)
|
|
EXPECT_EQ(*static_cast<short*>(img->GetScalarPointer(99,39,29)), (short)((99+39+29)%1000));
|
|
EXPECT_EQ(*static_cast<short*>(img->GetScalarPointer(70,10,5)), (short)((70+10+5)%1000));
|
|
EXPECT_EQ(src.sliceSource(), img);
|
|
src.update(nullptr); // no-op 安全
|
|
std::filesystem::remove_all(dir);
|
|
}
|