more consistent naming of utilities

utils/sample-combine.cpp

@@ -0,0 +1,196 @@
+/*
+ * sample_combine.cpp, part of LatAnalyze 3
+ *
+ * Copyright (C) 2013 - 2016 Antonin Portelli
+ *
+ * LatAnalyze 3 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.
+ *
+ * LatAnalyze 3 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 LatAnalyze 3.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include <LatAnalyze/Io.hpp>
+#include <LatAnalyze/CompiledFunction.hpp>
+
+using namespace std;
+using namespace Latan;
+
+static void usage(const string &cmdName)
+{
+    cerr << "usage: " << cmdName;
+    cerr << " [-o <output sample>]";
+    cerr << " <n> <function> <sample 1> ... <sample n>";
+    cerr << endl;
+    exit(EXIT_FAILURE);
+}
+
+template <typename T>
+static void loadAndCheck(vector<T> &sample, const vector<string> &fileName)
+{
+    const unsigned int n = sample.size();
+    Index              nSample = 0;
+    
+    cout << "-- loading data..." << endl;
+    for (unsigned int i = 0; i < n; ++i)
+    {
+        sample[i] = Io::load<T>(fileName[i]);
+        if (i == 0)
+        {
+            nSample = sample[i].size();
+        }
+        else if (sample[i].size() != nSample)
+        {
+            cerr << "error: number of sample mismatch (between '";
+            cerr << fileName[0] << "' and '" << fileName[i] << "')" << endl;
+            abort();
+        }
+    }
+}
+
+template <typename T>
+static void combine(const string &outFileName __dumb,
+                    const vector<T> &sample __dumb, const string &code __dumb)
+{
+    abort();
+}
+
+template <>
+void combine(const string &outFileName, const vector<DSample> &sample,
+             const string &code)
+{
+    const unsigned int n = sample.size();
+    DoubleFunction     f = compile(code, n);
+    DSample            result(sample[0]);
+    DVec               buf(n);
+    
+    cout << "-- combining data..." << endl;
+    result = sample[0];
+    FOR_STAT_ARRAY(result, s)
+    {
+        for (unsigned int k = 0; k < n; ++k)
+        {
+            buf[k] = sample[k][s];
+        }
+        result[s] = f(buf);
+    }
+    cout << scientific;
+    cout << "central value:\n"      << result[central];
+    cout << endl;
+    cout << "standard deviation:\n" << sqrt(result.variance());
+    cout << endl;
+    if (!outFileName.empty())
+    {
+        Io::save<DSample>(result, outFileName);
+    }
+}
+
+template <>
+void combine(const string &outFileName, const vector<DMatSample> &sample,
+             const string &code)
+{
+    const unsigned int n = sample.size();
+    DoubleFunction     f = compile(code, n);
+    DVec               buf(n);
+    DMatSample         result(sample[0]);
+    
+    cout << "-- combining data..." << endl;
+    FOR_STAT_ARRAY(result, s)
+    {
+        FOR_MAT(result[s], i, j)
+        {
+            for (unsigned int k = 0; k < n; ++k)
+            {
+                buf[k] = sample[k][s](i,j);
+            }
+            result[s](i, j) = f(buf);
+        }
+    }
+    cout << scientific;
+    cout << "central value:\n"      << result[central];
+    cout << endl;
+    cout << "standard deviation:\n" << result.variance().cwiseSqrt();
+    cout << endl;
+    if (!outFileName.empty())
+    {
+        Io::save<DMatSample>(result, outFileName);
+    }
+}
+
+template <typename T>
+void process(const string &outFileName, const vector<string> &fileName,
+             const string &code)
+{
+    vector<T> sample(fileName.size());
+    
+    loadAndCheck(sample, fileName);
+    combine(outFileName, sample, code);
+}
+
+int main(int argc, char *argv[])
+{
+    // argument parsing ////////////////////////////////////////////////////////
+    string         cmdName, outFileName = "", code;
+    vector<string> fileName;
+    int            c;
+    unsigned int   n = 0;
+
+    opterr = 0;
+    cmdName = basename(argv[0]);
+    while ((c = getopt(argc, argv, "o:")) != -1)
+    {
+        switch (c)
+        {
+            case 'o':
+                outFileName = optarg;
+                break;
+            case '?':
+                cerr << "error parsing option -" << char(optopt) << endl;
+                usage(cmdName);
+                break;
+            default:
+                usage(cmdName);
+                break;
+        }
+    }
+    if (argc - optind >= 1)
+    {
+        n = strTo<unsigned int>(argv[optind]);
+        if (argc - optind == static_cast<int>(n + 2))
+        {
+            fileName.resize(n);
+            code = argv[optind + 1];
+            for (unsigned int i = 0; i < n; ++i)
+            {
+                fileName[i] = argv[optind + 2 + i];
+            }
+        }
+        else
+        {
+            usage(cmdName);
+        }
+    }
+    else
+    {
+        usage(cmdName);
+    }
+    
+    // process data ////////////////////////////////////////////////////////////
+    try
+    {
+        process<DSample>(outFileName, fileName, code);
+    }
+    catch (Exceptions::Definition)
+    {
+        process<DMatSample>(outFileName, fileName, code);
+    }
+    
+    return EXIT_SUCCESS;
+}