raw

Thirdparty/Pangolin/tools/Plotter/main.cpp

@@ -0,0 +1,126 @@
+#include <pangolin/pangolin.h>
+#include <pangolin/utils/argagg.hpp>
+#include <pangolin/utils/file_utils.h>
+
+#include <functional>
+#include <thread>
+
+#include "csv_data_loader.h"
+
+namespace argagg{ namespace convert {
+
+template<>
+pangolin::Rangef arg<pangolin::Rangef>(char const* str)
+{
+    std::stringstream ss(str);
+    pangolin::Rangef r;
+    ss >> r.min; ss.get(); ss >> r.max;
+    return r;
+}
+}}
+
+int main( int argc, char* argv[] )
+{
+    // Parse command line
+    argagg::parser argparser {{
+        { "help", {"-h", "--help"}, "Print usage information and exit.", 0},
+        { "header", {"-H","--header"}, "Treat 1st row as column titles", 0},
+        { "x", {"-x"}, "X-axis series to plot, seperated by commas (default: '$i')", 1},
+        { "y", {"-y"}, "Y-axis series to plot, seperated by commas (eg: '$0,sin($1),sqrt($2+$3)' )", 1},
+        { "delim", {"-d"}, "Expected column delimitter (default: ',')", 1},
+        { "xrange", {"-X","--x-range"}, "X-Axis min:max view (default: '0:100')", 1},
+        { "yrange", {"-Y","--y-range"}, "Y-Axis min:max view (default: '0:100')", 1},
+        { "skip", {"-s","--skip"}, "Skip n rows of file, seperated by commas per file (default: '0,...')", 1},
+    }};
+
+    argagg::parser_results args = argparser.parse(argc, argv);
+    if ( (bool)args["help"] || !args.pos.size()) {
+        std::cerr << "Usage: Plotter [options] file1.csv [fileN.csv]*" << std::endl
+                  << argparser << std::endl
+                  << "    where: $i is a placeholder for the datum index," << std::endl
+                  << "           $0, $1, ... are placeholders for the 0th, 1st, ... sequential datum values over the input files" << std::endl;
+        return 0;
+    }
+
+    // Default values
+    const std::string xs = args["x"].as<std::string>("$i");
+    const std::string ys = args["y"].as<std::string>("$0");
+    const char delim = args["delim"].as<char>(',');
+    const pangolin::Rangef xrange = args["xrange"].as<>(pangolin::Rangef(0.0f,100.0f));
+    const pangolin::Rangef yrange = args["yrange"].as<>(pangolin::Rangef(0.0f,100.0f));
+    const std::string skips = args["skip"].as<std::string>("");
+    const std::vector<std::string> skipvecstr = pangolin::Split(skips,',');
+    std::vector<size_t> skipvec;
+    for(const std::string& s : skipvecstr) {
+        skipvec.push_back(std::stoul(s));
+    }
+    if( !(skipvec.size() == 0 || skipvec.size() == args.count()) )
+    {
+        std::cerr << "Skip argument must be empty or correspond to the number of files specified (" << args.count() << ")" << std::endl;
+        return -1;
+    }
+
+    pangolin::DataLog log;
+
+    CsvDataLoader csv_loader(args.all_as<std::string>(), delim);
+
+    if(args["header"]) {
+        std::vector<std::string> labels;
+        csv_loader.ReadRow(labels);
+        log.SetLabels(labels);
+    }
+
+    // Load asynchronously incase the file is large or is being read interactively from stdin
+    bool keep_loading = true;
+    std::thread data_thread([&](){
+        if(!csv_loader.SkipStreamRows(skipvec)) {
+            return;
+        }
+
+        std::vector<std::string> row;
+
+        while(keep_loading && csv_loader.ReadRow(row)) {
+            std::vector<float> row_num(row.size(), std::numeric_limits<float>::quiet_NaN() );
+            for(size_t i=0; i< row_num.size(); ++i) {
+                try{
+                    row_num[i] = std::stof(row[i]);
+                }catch(const std::invalid_argument& e){
+                    std::cerr << "Warning: couldn't parse '" << row[i] << "' as numeric data (use -H option to include header)" << std::endl;
+                }
+            }
+            log.Log(row_num);
+        }
+    });
+
+    pangolin::CreateWindowAndBind("Plotter", 640, 480);
+
+    pangolin::Plotter plotter(&log, xrange.min, xrange.max, yrange.min, yrange.max, 0.001, 0.001);
+    if( (bool)args["x"] || (bool)args["y"]) {
+        plotter.ClearSeries();
+        std::vector<std::string> xvec = pangolin::Split(xs,',');
+        std::vector<std::string> yvec = pangolin::Split(ys,',');
+
+        if( !(xvec.size() == 1 || xvec.size() == yvec.size()) ) {
+            std::cout << "x-series dimensions must be one, or equal to y-series dimensions" << std::endl;
+            return -1;
+        }
+
+        for(size_t i=0; i < yvec.size(); ++i) {
+            plotter.AddSeries( (xvec.size()==1) ? xvec[0] : xvec[i],yvec[i]);
+        }
+    }
+
+    plotter.SetBounds(0.0, 1.0, 0.0, 1.0);
+    pangolin::DisplayBase().AddDisplay(plotter);
+
+    while( !pangolin::ShouldQuit() )
+    {
+      glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
+      pangolin::FinishFrame();
+    }
+
+    keep_loading = false;
+    data_thread.join();
+
+    return 0;
+}