source: trunk/se/lu/thep/wenni/bin/NNIFileConverter/NNIFileConverter.cc @ 597

// $Id: NNIFileConverter.cc 597 2008-02-27 23:36:46Z jari $

/*
  Copyright (C) 2005, 2006, 2008 Jari Häkkinen

  This file is part of WeNNI,
  http://lev.thep.lu.se/trac/baseplugins/wiki/WeNNI

  WeNNI is free software; you can redistribute it and/or modify it
  under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2 of the License, or
  (at your option) any later version.

  WeNNI 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 General Public
  License for more details.

  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
  02111-1307, USA.
*/

#include "Parameter.h"
#include "weight.h"

#include <cmath>
#include <fstream>
#include <string>

#include <c++_tools/gslapi/matrix.h>


int main(const int argc,const char* argv[]) 
{
  theplu::wenni::nnifileconverter::Parameter option(argc,argv);
  std::ifstream is_f1(option.fg1().c_str());
  std::ifstream is_f2(option.fg2().c_str());
  std::ifstream is_s1(option.bgstd1().c_str());
  std::ifstream is_s2(option.bgstd2().c_str());
  std::ifstream* is_b1=NULL;
  std::ifstream* is_b2=NULL;
  if (option.datatype()==std::string("raw")) {
    is_b1=new std::ifstream(option.bg1().c_str());
    is_b2=new std::ifstream(option.bg2().c_str());
  }

  std::ofstream os_r(option.logratio().c_str());
  std::ofstream os_w(option.weight().c_str());
  std::string f1str;
  while (std::getline(is_f1,f1str)) {
    std::string f2str;
    std::string s1str;
    std::string s2str;
    std::getline(is_f2,f2str);
    std::getline(is_s1,s1str);
    std::getline(is_s2,s2str);
    if (is_f1.fail() || is_f2.fail() ||is_s1.fail() ||is_s2.fail()) {
      std::cerr << "Failed to read at least one file. Length problem?\n"
                << "   (fg1, fg2, bgstd1, or bgstd2)" << std::endl;
      exit(-1);
    }
    theplu::gslapi::vector f1(f1str);
    theplu::gslapi::vector f2(f2str);
    theplu::gslapi::vector s1(s1str);
    theplu::gslapi::vector s2(s2str);
    theplu::gslapi::vector b1(f1.size(),0.0);
    theplu::gslapi::vector b2(f2.size(),0.0);
    if (is_b1) {
      std::string b1str;
      std::getline(*is_b1,b1str);
      if (is_b1->fail()) {
        std::cerr << "Failed to read at bg1 file. Length problem?"
                  << std::endl;
        exit(-1);
      }
      b1=theplu::gslapi::vector(b1str);
    }
    if (is_b2) {
      std::string b2str;
      std::getline(*is_b2,b2str);
      if (is_b2->fail()) {
        std::cerr << "Failed to read at bg2 file. Length problem?"
                  << std::endl;
        exit(-1);
      }
      b2=theplu::gslapi::vector(b2str);
    }

    // Calculating log2ratio
    theplu::gslapi::vector log2ratio(f1.size());
    double log2=log(2);
    for (size_t i=0; i<log2ratio.size(); i++) 
      if (f1(i)>b1(i) && f2(i)>b2(i))
        log2ratio(i)=log( (f1(i)-b1(i)) / (f2(i)-b2(i)) ) / log2;

    // calculating weights
    theplu::gslapi::vector weight(f1.size());
    for (size_t i=0; i<weight.size(); i++) {
      using namespace theplu::wenni::weight;
      weight(i)=weight_SNR2(get_snr(f1(i),b1(i),s1(i)),
                            get_snr(f2(i),b2(i),s2(i)),option.beta());
    }

    os_r << log2ratio << '\n';
    os_w << weight << '\n';
  }

  is_f1.close();
  is_s1.close();
  is_f2.close();
  is_s2.close();
  if (is_b1) {
    is_b1->close();
    delete is_b1;
  }
  if (is_b2) {
    is_b2->close();
    delete is_b2;
  }
  os_r.close();
  os_w.close();

  return 0;
}