#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>
#include <cmath>

using std::ifstream;
using std::vector;
using std::cout;
using std::endl;
using std::fabs;

int main(int argc, char* argv[]) 
{
  // No good solution
  // TODO
   bool isFortranFile;
   if ( 2 == argc ) {
     isFortranFile = argv[1];
   }

  // Read the milepede output and the file with shifts created by 
  // TestMillePede to compare if the correction have been correctly calculated

  int label;
  float value;
  float value2;
  vector<int> labelVector;   
  vector<float> valueVector;   

  if (!isFortranFile) {

    cout<<"Read file created by TestMillePede."<<endl;

    ifstream vInput;
    vInput.open("shifts.txt");
    if (!vInput.is_open()) {
      cout<<"Could not open the input file."<<endl;
      exit(1);
    }
    while (1) {
      vInput >> label >> value;
      if (!vInput.good()) break;
      labelVector.push_back(label);
      valueVector.push_back(value);
    }
    vInput.close();

  }else {

    cout<<"Read file created by pede."<<endl;

    ifstream vInput;
    vInput.open("shifts.txt");
    if (!vInput.is_open()) {
      cout<<"Could not open the input file."<<endl;
      exit(1);
    }
    while (1) {
      vInput >> label >> value >> value2;
      if (!vInput.good()) break;
      int label1=10+label*2;
      labelVector.push_back(label1);
      valueVector.push_back(value);
      label1=500+label;
      labelVector.push_back(label1);
      valueVector.push_back(value2);
    }
    vInput.close();
  }

  std::vector<int>::iterator result;
 
  result = std::max_element(labelVector.begin(), labelVector.end());
  
  int maxValue = *result;

  float origShifts[maxValue+1];
  float recoShifts[maxValue+1];

  for (int i=0; i<maxValue+1; ++i) {
    origShifts[i] = 0;
    recoShifts[i] = 0;
  }

  for (int i=0; i<labelVector.size();++i) {
    origShifts[labelVector[i]] = valueVector[i];
  }
  
  labelVector.clear();
  valueVector.clear();

  ifstream vInput1;
  float dummy1, dummy2, dummy3, dummy4;
  vInput1.open("millepede.res");
  if (!vInput1.is_open()) {
    cout<<"Could not open the input file."<<endl;
    exit(1);
  }
  while (1) {
    vInput1 >> label >> value >> dummy1 >> dummy2 >> dummy3 >> dummy4;
    if (!vInput1.good()) break;
    labelVector.push_back(label);
    valueVector.push_back(value);
  }
  vInput1.close();

  for (int i=0; i<labelVector.size();++i) {
    recoShifts[labelVector[i]] = valueVector[i];
  }

  int numBadValues = 0;
  for (int i=0; i<maxValue+1;++i) {
    float diff = recoShifts[i]+origShifts[i];
    float percent = fabs(diff/origShifts[i]);
    if ( percent > 0.01) {
      cout<<i<<" : "<<recoShifts[i]<<", "<<origShifts[i]<<", "<<diff<<", "<<percent<<endl;
      numBadValues++;
    }
  }
  cout<<numBadValues<<" corrections with a deviation > 1%"<<endl;
}