WITKImageConversion_test.h

//---------------------------------------------------------------------------
//
// Project: OpenWalnut ( http://www.openwalnut.org )
//
// Copyright 2009 OpenWalnut Community, BSV@Uni-Leipzig and CNCF@MPI-CBS
// For more information see http://www.openwalnut.org/copying
//
// This file is part of OpenWalnut.
//
// OpenWalnut 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 of the License, or
// (at your option) any later version.
//
// OpenWalnut 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 OpenWalnut. If not, see <http://www.gnu.org/licenses/>.
//
//---------------------------------------------------------------------------
 
#ifndef WITKIMAGECONVERSION_TEST_H
#define WITKIMAGECONVERSION_TEST_H
 
#include <memory>
#include <vector>
 
#include <cxxtest/TestSuite.h>
 
#include "../../common/WLogger.h"
#include "../WITKImageConversion.h"
 
/**
 * Test functionality of WITKConversion class.
 */
class WITKImageConversionTest : public CxxTest::TestSuite
{
public:
    /**
     * Setup logger and other stuff for each test.
     */
    void setUp()
    {
        WLogger::startup();
    }
 
    /**
     * Converting a dataset into an itk image and then converting it back into a
     * dataset should yield the same dataset.
     */
    void testConversion()
    {
#ifdef OW_USE_ITK
        // build a dataset
        std::shared_ptr< std::vector< int > > values( new std::vector< int >( 27, 0 ) );
        for( int k = 0; k < 27; ++k )
        {
            ( *values )[ k ] = 27 - k;
        }
        std::shared_ptr< WGridRegular3D > g( new WGridRegular3D( 3, 3, 3 ) );
        std::shared_ptr< WValueSet< int > > v( new WValueSet< int >( 0, 1, values, W_DT_SIGNED_INT ) );
        std::shared_ptr< WDataSetScalar > ds( new WDataSetScalar( v, g ) );
 
        itk::Image< int, 3 >::Pointer i = makeImageFromDataSet< int >( ds );
        std::shared_ptr< WDataSetScalar > newds = makeDataSetFromImage< int >( i );
 
        TS_ASSERT_SAME_DATA( std::dynamic_pointer_cast< WValueSet< int > >( ds->getValueSet() )->rawData(),
                             std::dynamic_pointer_cast< WValueSet< int > >( newds->getValueSet() )->rawData(), sizeof( int ) * 27 );
#endif  // OW_USE_ITK
    }
};
 
#endif  // WITKIMAGECONVERSION_TEST_H