om_histogram.cpp

Go to the documentation of this file.

#include <openmodeller/om.hh>

#include <openmodeller/env_io/Raster.hh>

#include <string.h>
#include <stdio.h>
#include <stdlib.h>

#include <stdexcept>

#include <string>
#include <vector>
#include <map>
using std::string;
using std::vector;
using std::map;


/**************************************************************/
/**************************** main ****************************/

int
main( int argc, char **argv )
{
  // Reconfigure the global logger.
  Log::instance()->setLevel( Log::Info );
  Log::instance()->setPrefix( "" );

  if ( argc < 3 ) {

      Log::instance()->info( "\n%s <map_file> <num_intervals>\n\n", argv[0] );
      exit( 1 );
  }

  string map_file( argv[1] );
  int intervals = atoi( argv[2] );

  vector<int> histogram;
  vector<double> min_val;
  vector<double> max_val;
  map<double,int> values;

  if ( intervals > 0 ) {

    double interval = 1.0/intervals;

    for ( int i = 0; i < intervals; ++i ) {

      histogram.push_back( 0 );
      min_val.push_back( i*interval );
      max_val.push_back( min_val[i] + interval );
    }

    max_val[intervals-1] = max_val[intervals-1] + 0.1;
  }

  try {

    Map * map = new Map( RasterFactory::instance().create( map_file ) );

    Coord xmin, ymin, xmax, ymax, xstep, ystep;
    map->getRegion( &xmin, &ymin, &xmax, &ymax );
    map->getCell( &xstep, &ystep );

    Scalar val[ map->numBand() ];

    for ( Coord y = ymin; y < ymax; y += ystep ) {

      for ( Coord x = xmin; x < xmax; x += xstep ) {

        if ( map->get( x, y, val ) ) {

          if ( intervals > 0 ) {

            for ( int i = 0; i < intervals; ++i ) {

              if ( val[0] >= min_val[i] && val[0] < max_val[i] ) {

                ++histogram[i];
                break;
              }
            }
          }
          else {

            if ( values.find( val[0] ) != values.end() ) {

              ++values[val[0]];
            }
            else {

              values[val[0]] = 1;
            }
    }
        }
      }
    }

    Log::instance()->info( "Histogram:\n" );

    if ( intervals > 0 ) {

      for ( int i = 0; i < intervals; ++i ) {

        Log::instance()->info( "[%f-%f): %d\n", min_val[i], max_val[i], histogram[i] );
      }
    }
    else {

      for ( std::map<double, int>::const_iterator it = values.begin(); it != values.end(); it++ ) {

        Log::instance()->info( "%f: %d\n", (*it).first, (*it).second );
      }
    }

    delete map;
  }
  catch ( std::exception& e ) {
    Log::instance()->info( "Exception occurred\n" );
    Log::instance()->info( "Message is %s\n", e.what() );
  }
  catch ( ... ) {
    Log::instance()->info( "Unknown Error occurred\n" );
  }
}