#include "polyfit.h"
#include <iostream>

/** Fit an n:th degree polynomial and remove if from data.  
 *  Uses discrete orthogonal polynomials for numerical stability. 
 *
 * Copyright Sören Molander, 2005
 * 
 * Feel free to use and modify the program at will. 
 *
 */ 

using namespace SIGMOD;
using namespace std;


int main(int argc, char** argv) {


  if (argc < 4) {
    cerr << "#############################"<<endl;
    cerr << "rempol: fit an n:th degree polynomial and remove it from data"<<endl;
    cerr << "Uses discrete orthogonal polynomials for numerical robustness"<<endl;
    cerr << "Usage: rempol <filename> <order> <column 1> <column 2>"<<endl;
    cerr << "Usage: rempol <filename> <order> <column>"<<endl;
    cerr << "The first version assumes the independent variable is in the first column"<<endl;
    cerr << "The second version just uses a running index"<<endl;
    cerr << "Output will be written to filename_fit.txt"<<endl;
    cerr << "#############################"<<endl;
    exit(-1);
  }
  
  string filename;
  int col1, col2, order;
  vector<int> cols;
  cols.clear();

  if (argc == 4) {
    filename = string(argv[1]);
    order = atoi(argv[2]);
    col1 = atoi(argv[3]);
    --col1; // columns are in C-standard, start at 0
    cols.push_back(col1);
  } else if (argc == 5) {
    filename = string(argv[1]);
    order = atoi(argv[2]);
    col1 = atoi(argv[3]);
    col2 = atoi(argv[4]);
    --col1;// columns are in C-standard, start at 0
    --col2;
    cols.push_back(col1);
    cols.push_back(col2);
  } else {
    cerr << "Error in argument list, try again"<<endl;
    exit(-1);
  }
  if (col1 < 0 || col2 < 0) {
    cerr << "Column numbers must be 1,2.."<<endl;
    exit(-1);
  }


  PolyFit polyfit(order, filename, cols);
  polyfit.removePolFromData();
  cerr << "stdv of data - polynomial = "<<polyfit.stdv()<<endl;
  polyfit.writeData();

  return 0;
}