New program summary based on OPM I/O classes

CMakeLists.txt

@@ -187,7 +187,11 @@ if(ENABLE_ECL_INPUT)
     test_util/convertECL.cpp
     )
 
-  foreach(target compareECL convertECL)
+  add_executable(summary
+    test_util/summary.cpp
+    )
+
+  foreach(target compareECL convertECL summary)
     target_link_libraries(${target} opmcommon)
     install(TARGETS ${target} DESTINATION bin)
   endforeach()

test_util/summary.cpp

@@ -0,0 +1,123 @@
+//#include <chrono>
+#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 (abs(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;
+}