/*
  Copyright 2019 Equinor ASA.

  This file is part of the Open Porous Media project (OPM).

  OPM 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 3 of the License, or
  (at your option) any later version.

  OPM 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 OPM.  If not, see <http://www.gnu.org/licenses/>.
*/

#include <cmath>
#include <iomanip>
#include <iostream>
#include <tuple>
#include <getopt.h>
#include <sstream>

#include <opm/io/eclipse/ESmry.hpp>


static void printHelp() {

    std::cout << "\nsummary needs a minimum of two arguments. First is smspec filename and then list of vectors  \n"
              << "\nIn addition, the program takes these options (which must be given before the arguments):\n\n"
              << "-h Print help and exit.\n"
              << "-l list all summary vectors.\n"
              << "-r extract data only for report steps. \n\n";
}

void printHeader(const std::vector<std::string>& keyList){

    std::cout << "--" << std::setw(14) << keyList[0];

    for (size_t n= 1; n < keyList.size(); n++){
        std::cout << std::setw(16) << keyList[n]; 
    }

    std::cout << std::endl;
}

std::string formatString(float data){

    std::stringstream stream;
    
    if (std::fabs(data) < 1e6){
       stream << std::fixed << std::setw(16) << std::setprecision(6) << data;
    } else {
       stream << std::scientific << std::setw(16) << std::setprecision(6)  << data;
    }
    
    return stream.str();    
}

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

    int c                          = 0;
    bool reportStepsOnly           = false;
    bool listKeys                  = false;

    while ((c = getopt(argc, argv, "hrl")) != -1) {
        switch (c) {
        case 'h':
            printHelp();
            return 0;
        case 'r':
            reportStepsOnly=true;
            break;
        case 'l':
            listKeys=true;
            break;
        default:
            return EXIT_FAILURE;
        }
    }
    
    int argOffset = optind;

    std::string filename = argv[argOffset];
    Opm::EclIO::ESmry smryFile(filename);
    
    if (listKeys){
        auto list = smryFile.keywordList();
        
        for (size_t n = 0; n < list.size(); n++){
            std::cout << std::setw(20) << list[n];
            
            if (((n+1) % 5)==0){
                std::cout << std::endl;
            }
        }
        
        std::cout << std::endl;

        return 0;
    }

    std::vector<std::string> smryList;
    for (int i=0; i<argc - argOffset-1;i++){
       smryList.push_back(argv[i+argOffset+1]);
    }
    
    if (smryList.size()==0){
        std::string message = "No summary keys specified on command line";
        std::cout << "\n!Runtime Error \n >> " << message << "\n\n";
        return EXIT_FAILURE;
    }
    
    std::vector<std::vector<float>> smryData;

    for (auto key : smryList){
        
        if (!smryFile.hasKey(key)){
            std::string message = "Key " + key + " not found in summary file " + filename;
            std::cout << "\n!Runtime Error \n >> " << message << "\n\n";
            return EXIT_FAILURE;
        }
        
        std::vector<float> vect = reportStepsOnly ? smryFile.get_at_rstep(key) : smryFile.get(key);

        smryData.push_back(vect);
    }
    
    printHeader(smryList);
 
    for (size_t s=0; s<smryData[0].size(); s++){
        for (size_t n=0; n < smryData.size(); n++){
            std::cout << formatString(smryData[n][s]);
        }
        std::cout << std::endl;
    }   
     
    return 0;
}