generateSimplicityTables2D.cpp

#include <iostream>
#include <vector>
#include <bitset>
#include "DGtal/topology/Object.h"
#include "DGtal/shapes/Shapes.h"
#include "DGtal/helpers/StdDefs.h"
#include "DGtal/io/boards/Board2D.h"
#include "DGtal/io/Color.h"


using namespace std;
using namespace DGtal;


template <typename Object, typename Map>
void 
generateSimplicityTable( const typename Object::DigitalTopology & dt,
                         Map & map )
{
  typedef typename Object::DigitalTopology DigitalTopology;
  typedef typename Object::DigitalSet DigitalSet;
  typedef typename Object::Point Point;
  typedef typename DigitalSet::Domain Domain;
  typedef typename Domain::ConstIterator DomainConstIterator;

  Point p1 = Point::diagonal( -1 );
  Point p2 = Point::diagonal(  1 );
  Point c = Point::diagonal( 0 );
  Domain domain( p1, p2 );
  DigitalSet shapeSet( domain );
  Object shape( dt, shapeSet );
  unsigned int k = 0;
  for ( DomainConstIterator it = domain.begin(); it != domain.end(); ++it )
    if ( *it != c ) ++k;
  ASSERT( ( k < 32 )
          && "[generateSimplicityTable] number of configurations is too high." );
  unsigned int nbCfg = 1 << k;
  for ( unsigned int cfg = 0; cfg < nbCfg; ++cfg )
    {
      shape.pointSet().clear();
      shape.pointSet().insert( c );
      unsigned int mask = 1;
      for ( DomainConstIterator it = domain.begin(); it != domain.end(); ++it )
        {
          if ( *it != c )
            {
              if ( cfg & mask ) shape.pointSet().insert( *it );
              mask <<= 1;
            }
        }
      bool simple = shape.isSimple( c );
      map[ cfg ] = simple;
    }
}

template <typename Object, typename Map>
void 
displaySimplicityTable( Board2D & board, 
                        const typename Object::DigitalTopology & dt,
                        const Map & map )
{
  typedef typename Object::DigitalTopology DigitalTopology;
  typedef typename Object::DigitalSet DigitalSet;
  typedef typename Object::Point Point;
  typedef typename DigitalSet::Domain Domain;
  typedef typename Domain::ConstIterator DomainConstIterator;

  Point p1 = Point::diagonal( -1 );
  Point p2 = Point::diagonal(  1 );
  Point c = Point::diagonal( 0 );
  Domain domain( p1, p2 );

  Point q1 = Point::diagonal( -1 );
  Point q2 = Point::diagonal( 4*16-1 );
  Domain fullDomain( q1, q2 );
  board << SetMode( fullDomain.className(), "Paving" );
  unsigned int cfg = 0;
  for ( unsigned int y = 0; y < 16; ++y )
    for ( unsigned int x = 0; x < 16; ++x, ++cfg )
      {
        bool simple = map[ cfg ];
        Point base( x*4, y*4 );
        unsigned int mask = 1;
        for ( DomainConstIterator it = domain.begin(); 
              it != domain.end(); ++it )
          {
            Point q = base + (*it);
            if ( *it == c )
              board << CustomStyle( q.className(), 
                                    new CustomColors( Color( 0, 0, 0 ), 
                                                      Color( 30, 30, 30 ) ) );
            else 
              {
                if ( cfg & mask ) 
                  board << 
                    CustomStyle( q.className(), 
                                 simple 
                                 ? new CustomColors( Color( 0, 0, 0 ), 
                                                     Color( 10, 255, 10 ) )
                                 : new CustomColors( Color( 0, 0, 0 ), 
                                                     Color( 255, 10, 10 ) ) );
                else
                  board << 
                    CustomStyle( q.className(), 
                                 simple 
                                 ? new CustomColors( Color( 0, 0, 0 ), 
                                                     Color( 245, 255, 245 ) )
                                 : new CustomColors( Color( 0, 0, 0 ), 
                                                     Color( 255, 245, 245 ) ) );
                mask <<= 1;
              }
            board << q;
          }
      }
}

template <typename Map>
void 
outputTableAsArray( ostream & out,
                    const Map & map,
                    const string & tableName )
{
  typedef typename Map::const_iterator MapConstIterator;
  out << "const bool " << tableName << "[ " << map.size() << " ] = { ";
  for ( MapConstIterator it = map.begin(), it_end = map.end();
        it != it_end; )
    {
      out << *it;
      ++it;
      if ( it != it_end ) out << ", ";
    }
  out << " };" << std::endl;
}


int main( int argc, char** argv )
{
  typedef std::vector<bool> ConfigMap;

  using namespace Z2i;
  trace.beginBlock ( "Generate 2d table for 4-8 topology" );
  ConfigMap table4_8( 256 );
  generateSimplicityTable< Object4_8 >( dt4_8, table4_8 );
  trace.endBlock();

  trace.beginBlock ( "Generate 2d table for 8-4 topology" );
  ConfigMap table8_4( 256 );
  generateSimplicityTable< Object8_4 >( dt8_4, table8_4 );
  trace.endBlock();

  Board2D board;
  trace.beginBlock ( "Display 2d table for 4-8 topology" );
  displaySimplicityTable< Object4_8 >( board, dt4_8, table4_8 );
  board.saveEPS( "table4_8.eps" );
  trace.endBlock();

  board.clear();
  trace.beginBlock ( "Display 2d table for 8-4 topology" );
  displaySimplicityTable< Object8_4 >( board, dt8_4, table8_4 );
  board.saveEPS( "table8_4.eps" );
  trace.endBlock();

  outputTableAsArray( std::cout, table4_8, "simplicityTable4_8" );
  outputTableAsArray( std::cout, table8_4, "simplicityTable8_4" );

  return 0;
}
//                                                                           //