TsSynchronizedRaster.cpp

Go to the documentation of this file.

/*  Copyright (C) 2008 National Institute For Space Research (INPE) - Brazil.

    This file is part of the TerraLib - a Framework for building GIS enabled applications.

    TerraLib 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.

    TerraLib 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 TerraLib. See COPYING. If not, write to
    TerraLib Team at <terralib-team@terralib.org>.
 */

/*!
  \file TsSynchronizedRaster.cpp
 
  \brief A test suit for the SynchronizedRaster Class.
 */

// TerraLib
#include <terralib_buildconfig.h>

#include <terralib/raster.h>
#include "../Config.h"

// STL
#include <memory>

// Boost
#include <boost/test/unit_test.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/thread.hpp>

BOOST_AUTO_TEST_SUITE ( synchronizedRaster_tests )

void CreateTestRaster( unsigned int nBands, unsigned int nLines,
  unsigned int nCols, boost::shared_ptr< te::rst::Raster >& rasterPointer )
{
  std::vector< te::rst::BandProperty * > bandsProps;
  for( unsigned int bandsPropsIdx = 0 ; bandsPropsIdx < nBands ; ++bandsPropsIdx )
  {
    bandsProps.push_back( new te::rst::BandProperty( bandsPropsIdx,
      te::dt::UINT32_TYPE ) );
  }
  
  rasterPointer.reset( te::rst::RasterFactory::make( "MEM",
    new te::rst::Grid( nCols, nLines ), bandsProps,
    std::map< std::string, std::string >(), 0, 0 ) );
    
  unsigned int band = 0;
  unsigned int line = 0;
  unsigned int col = 0;
  double pixelValue = 0;
  
  for( band = 0 ; band < nBands ; ++band )
    for( line = 0 ; line < nLines ; ++line )
      for( col = 0 ; col < nCols ; ++col )
      {
        rasterPointer->setValue( col, line, pixelValue, band );
        ++pixelValue;
      }
}

void threadEntry(te::rst::RasterSynchronizer* syncPtr)
{
  std::unique_ptr< te::rst::SynchronizedRaster > syncRasterPtr(
    new te::rst::SynchronizedRaster( 2, *syncPtr ) );
  
  const unsigned int nBands = syncRasterPtr->getNumberOfBands();
  const unsigned int nLines = syncRasterPtr->getNumberOfRows();
  const unsigned int nCols = syncRasterPtr->getNumberOfColumns();
  unsigned int band = 0;
  unsigned int line = 0;
  unsigned int col = 0;
  double pixelValue = 0;
  
  for( band = 0 ; band < nBands ; ++band )
  {
    boost::this_thread::sleep( boost::posix_time::milliseconds(100) );
    
    for( line = 0 ; line < nLines ; ++line )
    {
      for( col = 0 ; col < nCols ; ++col )
      {
        syncRasterPtr->getValue( col, line, pixelValue, band );
        syncRasterPtr->setValue( col, line, pixelValue + 10.0, band );
      }
    }
  }
}

BOOST_AUTO_TEST_CASE (singleThread_test)
{
  /* Create the input test raster */
  
  const unsigned int nBands = 10;
  const unsigned int nLines = 10;
  const unsigned int nCols = 10;

  boost::shared_ptr< te::rst::Raster > inputRasterPointer;
  CreateTestRaster( nBands, nLines, nCols, inputRasterPointer );
  
  /* Using the synchronized raster adaptor. */

  {
    std::unique_ptr< te::rst::RasterSynchronizer > syncPtr(
      new te::rst::RasterSynchronizer( *inputRasterPointer, te::common::RWAccess ) );
    std::unique_ptr< te::rst::SynchronizedRaster > syncRasterPtr(
      new te::rst::SynchronizedRaster( 2, *syncPtr ) );
    
    unsigned int band = 0;
    unsigned int line = 0;
    unsigned int col = 0;
    double pixelValue = 0;
    
    for( band = 0 ; band < nBands ; ++band )
    {
      for( line = 0 ; line < nLines ; ++line )
      {
        for( col = 0 ; col < nCols ; ++col )
        {
          syncRasterPtr->getValue( col, line, pixelValue, band );
          syncRasterPtr->setValue( col, line, pixelValue + 10.0, band );
        }
      }
    }
     
    syncRasterPtr.reset();
    syncPtr.reset();
  }
  
  /* Verifying the value */
  
  {
    unsigned int band = 0;
    unsigned int line = 0;
    unsigned int col = 0;
    double pixelValue = 0;
    double readPixelValue = 0;
    
    for( band = 0 ; band < nBands ; ++band )
      for( line = 0 ; line < nLines ; ++line )
        for( col = 0 ; col < nCols ; ++col )
        {
          inputRasterPointer->getValue( col, line, readPixelValue, band );
          BOOST_CHECK_CLOSE( pixelValue + 10.0, readPixelValue, 0.0000001 );
          ++pixelValue;
        }
  }
}

BOOST_AUTO_TEST_CASE (multiThread_test)
{
  /* Create the input test raster */
  
  const unsigned int nBands = 10;
  const unsigned int nLines = 100;
  const unsigned int nCols = 100;

  boost::shared_ptr< te::rst::Raster > inputRasterPointer;
  CreateTestRaster( nBands, nLines, nCols, inputRasterPointer );
  
  /* Using the synchronized raster adaptor. */

  {
    std::unique_ptr< te::rst::RasterSynchronizer > syncPtr(
      new te::rst::RasterSynchronizer( *inputRasterPointer, te::common::RWAccess ) );

    boost::thread_group threads;
    
    for( unsigned int threadIdx = 0 ; threadIdx < nBands ;
      ++threadIdx )
    {
      threads.add_thread( new boost::thread( threadEntry, syncPtr.get() ) );
    };
    
    threads.join_all();
     
    syncPtr.reset();
  }
  
  /* Verifying the values */
  
  {
    unsigned int band = 0;
    unsigned int line = 0;
    unsigned int col = 0;
    double pixelValue = 0;
    double readPixelValue = 0;
    
    for( band = 0 ; band < nBands ; ++band )
      for( line = 0 ; line < nLines ; ++line )
        for( col = 0 ; col < nCols ; ++col )
        {
          inputRasterPointer->getValue( col, line, readPixelValue, band );
          BOOST_CHECK_CLOSE( pixelValue + 100.0, readPixelValue, 0.0000001 );
          ++pixelValue;
        }
  }
}

BOOST_AUTO_TEST_SUITE_END()