source: trunk/test/test_wead.cc @ 58

// $Id: test_wead.cc 58 2004-04-14 07:07:08Z jari $

#include <iostream>
#include <fstream>
#include <string>

#include "FileIO.h"
#include "Wead.h"



// class for command line options.
class Parameter {
public:
  Parameter(const Parameter&);  // not implemented
  Parameter(const int argc,const char *argv[]);

  inline const std::string& matrix(void) const { return matrix_file_; }
  inline const std::string& flag(void) const { return flag_file_; }
  inline int neighbours(void) const { return neighbours_; }

private:
  void analyse(void);
  void defaults(void);

  std::string flag_file_;
  std::string matrix_file_;
  int neighbours_;
};

Parameter::Parameter(const int argc,const char *argv[])
{
  using namespace std;
  // check if help requested
  for (int i=1; i<argc; i++)
    if (string(argv[i])==string("-help") || string(argv[i])==string("-h")) {
      cout << "Please write some help information\n";
      exit(0);      // always exit after printing help
    }

  defaults();

  for (int i=1; i<argc; i++) {
    bool ok=false;
    string myargv(argv[i]);
    if (myargv==string("-matrix"))
      if ((i+1)<argc) {
        matrix_file_=argv[++i];
        ok=true;
      }
    if (myargv==string("-flag"))
      if ((i+1)<argc) {
        flag_file_=argv[++i];
        ok=true;
      }
    if (myargv==string("-neighbours"))
      if ((i+1)<argc) {
        neighbours_=atoi(argv[++i]);
        ok=true;
      }

    if (!ok)
      std::cerr << "# Parameter::Parameter     Invalid option: "
                << argv[i] << '\n';
  }

  analyse();
}

void Parameter::analyse(void)
{
  using namespace theplu::cpptools;
  bool ok=true;
  if (FileIO().access_rights(matrix(),"r")) {
    std::cerr << "Cannot access matrix file " << matrix() << std::endl;
    ok=false;
  }
  if (FileIO().access_rights(flag(),"r")) {
    std::cerr << "Cannot access flag file " << flag() << std::endl;
    ok=false;
  }
  if (!ok)
    std::exit(-1);
}

void Parameter::defaults(void)
{
  flag_file_="data/wead_flag.data";
  matrix_file_="data/wead_matrix.data";
  neighbours_=3;
}



int main(const int argc,const char* argv[])
{
  Parameter option(argc,argv);
  std::ifstream matrix_stream(option.matrix().c_str());
  std::ifstream flag_stream(option.flag().c_str());
  theplu::cpptools::Wead wead(matrix_stream,flag_stream,option.neighbours());
  wead.estimate();
  std::cout << wead.matrix() << std::endl;
  return 0; // normal exit
}