/*//////////////////////////////////////////////////////////////////////////////
// CMT Cosmic Muon Tomography project //////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////////

  Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
  All rights reserved

  Authors: Andrea Rigoni Garola < andrea.rigoni@pd.infn.it >

  ------------------------------------------------------------------
  This library is free software;  you  can  redistribute  it  and/or
  modify it  under the  terms  of  the  GNU  Lesser  General  Public
  License as published  by  the  Free  Software  Foundation;  either
  version 3.0 of the License, or (at your option) any later version.

  This library is  distributed in  the hope that it will  be useful,
  but  WITHOUT ANY WARRANTY;  without  even  the implied warranty of
  MERCHANTABILITY  or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  Lesser General Public License for more details.

  You should have received a copy of  the GNU Lesser General  Public
  License along with this library.

//////////////////////////////////////////////////////////////////////////////*/

#include "Vtk/Math/vtkVoxImage.h"
#include "Math/VoxImage.h"
#include "Vtk/uLibVtkViewer.h"

#define BOOST_TEST_MODULE VtkVoxImageTest
#include <boost/test/unit_test.hpp>

using namespace uLib;

struct TestVoxel {
  Scalarf Value;
  unsigned int Count;
};

BOOST_AUTO_TEST_CASE(vtkVoxImageConstruction) {
  TestVoxel zero = {0, 0};
  TestVoxel nonzero = {5.5f * 1e-6f, 100};

  VoxImage<TestVoxel> img(Vector3i(10, 10, 10));
  img.SetSpacing(Vector3f(3, 3, 3));
  img.InitVoxels(zero);
  img[Vector3i(3, 3, 3)] = nonzero;

  Vtk::vtkVoxImage vtk_img(img);
  vtk_img.SaveToXMLFile("test_vtkvoximage.vti");

  if (std::getenv("CTEST_PROJECT_NAME") == nullptr) {
    Vtk::Viewer viewer;
    viewer.AddPuppet(vtk_img);
    viewer.Start();
  }

  BOOST_CHECK(vtk_img.GetImageData() != nullptr);
}